ServletMain.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#ifdef __WINNT__
#include <windows.h>
#endif
 
#include <wctype.h>
 
#include <dcl/Exception.h>
#include <dcl/StringReader.h>
#include <dcl/BufferedReader.h>
#include <dcl/FileReader.h>
#include <dcl/DateTime.h>
#include <dcl/IniFile.h>
#include <dcl/TextTemplate.h>
 
#include "ServletMain.h"
#include "Session.h"
#include "User.h"
#include "HtmlPage.h"
 
#if __DCL_HAVE_THIS_FILE__
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("fastpage/ServletMain.cpp");
#endif
 
__DCL_BEGIN_NAMESPACE
 
class ScriptException : public Exception
{
    DECLARE_CLASSINFO(ScriptException)
public:
    ScriptException(Exception* pCause);
 
    virtual String getMessage() const;
};
 
IMPLEMENT_CLASSINFO(ScriptException, Exception)
 
ScriptException::ScriptException(Exception* pCause)
    : Exception(pCause)
{
}
 
String ScriptException::getMessage() const
{
    return L"Invalid Script Contents";
}
 
SiteContext::SiteContext()
{
    uPostMaxLength = 1024 * 1024;     // 1 MBytes
    nErrorPageID = 1;
    nPermDeniedPageID = 2;
}
 
const SiteContext& SiteContext::operator = (const SiteContext& src)
{
    strDbConnect = src.strDbConnect;
 
    nErrorPageID = src.nErrorPageID;
    nPermDeniedPageID = src.nPermDeniedPageID;
 
    strSkinDir = src.strSkinDir;
    strAttachmentDir = src.strAttachmentDir;
    strThumbnailDir = src.strThumbnailDir;
 
    return *this;
}
 
HTTP_SERVLET_INSTANCE(
    FastPageServlet,
    L"DCL World-Wide-Web Components - Database based Html Page Assistant"
)
 
IMPLEMENT_CLASSINFO(FastPageServlet, HttpServletEx)
 
FastPageServlet::FastPageServlet()
{
}
 
void FastPageServlet::onInitialize()
    __DCL_THROWS1(Exception*)
{
    HttpServletEx::onInitialize();
 
#ifdef __WINNT__
    #define INI_FILENAME L"HSAFastPage_w.ini"
#else
    #define INI_FILENAME L"HSAFastPage.ini"
#endif
    String strIniFile = getIniFileName(INI_FILENAME);
    __DCL_TRACE1(L"%ls\n", strIniFile.data());
 
    SiteContext* pSite = NULL;
    BufferedReader in(new FileReader(strIniFile, new UTF8Decoder()));
    String strLine;
    while(in.readLine(strLine)) {
        strLine = strLine.trim();
        if (strLine.isEmpty())
            continue;
 
        if (strLine[0] == L';' || strLine[0] == L'#')
            continue;
        else if (strLine[0] == '[') {
            strLine = strLine.trim(L"[] ");
 
            pSite = new SiteContext();
            pSite->strSiteID = strLine;
 
            __listSiteCtx.addTail(pSite);
            __mapSiteCtx[pSite->strSiteID] = pSite;
        }
        else {
            if (pSite == NULL) {
                // invalid section
                continue;
            }
            // key = value
            size_t nPos = strLine.indexOf(L'=');
            if (nPos == (size_t) -1) {
                // invalid line
                continue;
            }
 
            String strName = strLine.left(nPos).trim();
            String strValue = strLine.right(strLine.length() - nPos - 1).trim();
            if (strName.isEmpty() || strValue.isEmpty()) {
                // ignore
                continue;
            }
 
            if (!strName.compareNoCase(L"POST_MAX_LENGTH")) {
                size_t nMul = 1;
                wchar_t chMul = strValue[strValue.length() - 1];
                if (!iswdigit((wint_t)chMul)) {
                    if (chMul == L'k' || chMul == L'K')
                        nMul = 1024;
                    else if (chMul == L'm' || chMul == L'M')
                        nMul = 1024 * 1024;
                    String strMul = chMul;
                    strValue = strValue.trimRight(strMul);
                }
                try {
                    size_t n = UInteger::parse(strValue, 10);
                    pSite->uPostMaxLength = n * nMul;
                }
                catch (Exception* e) {
                    e->destroy();
                }
            }
            else if (!strName.compareNoCase(L"SERVER_NAMES")) {
                StringArray list;
                if (strValue.split(L' ', list)) {
                    StringArray::Iterator itList = list.begin();
                    for( ; itList != list.end(); ++itList)
                        __mapSiteCtx[*itList] = pSite;
                }
            }
            else if (!strName.compareNoCase(L"DATABASE_DRIVER")) {
                pSite->strDbDriver = strValue;
            }
            else if (!strName.compareNoCase(L"DATABASE_CONNECT")) {
                pSite->strDbConnect = strValue;
            }
            else if (!strName.compareNoCase(L"ERROR_PAGE")) {
                pSite->nErrorPageID = Integer::parse(strValue, 10,
                    pSite->nErrorPageID);
            }
            else if (!strName.compareNoCase(L"PERM_DENIED_PAGE")) {
                pSite->nPermDeniedPageID = Integer::parse(strValue, 10,
                    pSite->nPermDeniedPageID);
            }
            else if (!strName.compareNoCase(L"SKIN_DIR")) {
                pSite->strSkinDir = strValue;
            }
            else if (!strName.compareNoCase(L"ATTACHMENT_DIR")) {
                pSite->strAttachmentDir = strValue;
            }
            else if (!strName.compareNoCase(L"THUMBNAIL_DIR")) {
                pSite->strThumbnailDir = strValue;
            }
            else {
                __DCL_TRACE1(L"Invalid Line: [%ls]\n", strLine.data());
            }
        }
    }
}
 
void FastPageServlet::onCleanup()
    __DCL_THROWS1(Exception*)
{
 
    for(PointerList::Iterator itList = __listSiteCtx.begin();
            itList != __listSiteCtx.end(); itList++
    ) {
        SiteContext* pSite = (SiteContext*)(*itList);
        delete (pSite);
    }
    HttpServletEx::onCleanup();
}
 
void FastPageServlet::onService(HttpServletContextEx& ctx)
    __DCL_THROWS1(Exception*)
{
    DateTime dtStart = DateTime::getCurrentLocalTime();
 
    const SiteContext* pSite = NULL;
    int nPageID = -1;
    int nServiceUserID = Session::SYSTEM_USER_ID;
 
    // SERVER_NAME 이 구성되어 있지 않으면 404
    {
        String strServerName = AsciiDecoder::decode(
            ctx.getCgiVariable("SERVER_NAME"));
        __DCL_TRACE1(L"SERVER_NAME: [%ls]\n", strServerName.data());
        StringToPointerMap::ConstIterator itMap = __mapSiteCtx.find(strServerName);
        if (itMap == __mapSiteCtx.end()) {
            __DCL_TRACE0(L"404:\n");
            ctx.setStatusCode(HTTP_STATUS_NOT_FOUND);
            return;
        }
        pSite = (SiteContext*)(*itMap).value;
 
        __DCL_TRACE1(L"HOST ID: [%ls]\n", pSite->strSiteID.data());
        __DCL_TRACE1(L"POST_MAX_LENGTH: [%u]\n", pSite->uPostMaxLength);
        __DCL_TRACE1(L"DATABASE_DRIVER: [%ls]\n", pSite->strDbDriver.data());
        __DCL_TRACE1(L"DATABASE_CONNECT: [%ls]\n", pSite->strDbConnect.data());
        __DCL_TRACE1(L"ERROR_PAGE_ID: [%d]\n", pSite->nErrorPageID);
        __DCL_TRACE1(L"PERM_DENIED_PAGE_ID: [%d]\n", pSite->nPermDeniedPageID);
        __DCL_TRACE1(L"SKIN_DIR: [%ls]\n", pSite->strSkinDir.data());
        __DCL_TRACE1(L"ATTACHMENT_DIR: [%ls]\n", pSite->strAttachmentDir.data());
        __DCL_TRACE1(L"THUMBNAIL_DIR: [%ls]\n", pSite->strThumbnailDir.data());
    }
 
    // GET, POST 메소드만 서비스 한다.
    if (!(ctx.methodNo() == HTTP_METHOD_GET || ctx.methodNo() == HTTP_METHOD_POST)) {
        ctx.setStatusCode(HTTP_STATUS_NOT_IMPLEMENTED);
        return;
    }
 
    // 스크립트에서 PAGE_ID, SERVICE_USER_ID를 읽는다. 
    try {
        String script = ctx.scriptData();
        __DCL_TRACE1(L"script data[%ls]\n", script.data());
 
        StringArray lines;
        script.split_r(L"\r?\n", true, lines);
        for (size_t i = 0; i < lines.size(); i++) {
            String strLine = lines[i];
            StringArray list;
            if (strLine.split(L'=', list) == 2) {
                StringArray::Iterator itList = list.begin();
                (*itList).trim();
                if (!(*itList).compareNoCase(L"PAGE")) {
                    ++itList;
                    nPageID = Integer::parse(*itList, 10);
                }
                else if (!(*itList).compareNoCase(L"SERVICE_USER")) {
                    ++itList;
                    nServiceUserID = Integer::parse(*itList, 10);
                }
            }
        }
        __DCL_TRACE2(L"SCRIPT PAGE_ID=%d, SERVICE_USER_ID=%d\n",
            nPageID, nServiceUserID);
    }
    catch(Exception* _cause) {
        throw new ScriptException(_cause);
    }
 
    // QUERY_STRING에 page파라미터가 있으면 PAGE_ID를 오버라이드 한다.
    {
        ListedStringToStringArrayMap& mapQuery = ctx.__queryMap;
        ListedStringToStringArrayMap::ConstIterator itMap
            = mapQuery.find(L"page");
        if (itMap != mapQuery.end() && (*itMap).value.size() > 0) {
            nPageID = Integer::parse((*itMap).value[0], 10, nPageID);
        }
    }
 
    SQLConnection conn(pSite->strDbDriver);
    conn.open(pSite->strDbConnect);
    __DCL_TRACE1(L"database server info: %ls\n",
        conn.getServerInfo().data());
    try {
        if (conn.canTransact()) {
            conn.startTrans();
        }
 
        SQLQuery query(&conn);
 
        Session session(ctx, query, nServiceUserID);
        session.check();
 
        HtmlPage page(nPageID, pSite);
 
        __DCL_TRACE3(
            L"nPageID: %d, nPermID: %d, RoleID: %d\n",
            page.__nPageID,
            page.__nPermID,
            session.__nRoleID
        );
 
        if (page.init(query)) {
#define __PERMITTED     0
#define __DENIED        1
#define __YET_GROUP     2
#define __PENDING_GROUP 3
 
            //int fAuth = __PERMITTED;
            int fAuth = __DENIED;
 
            if (page.__nPermID == Session::GUEST_ROLE_ID)
                fAuth = __PERMITTED;
            else if (session.__nRoleID != Session::GUEST_ROLE_ID) {
                if (nServiceUserID == Session::SYSTEM_USER_ID) {
                    if (page.__nPermID >= session.__nRoleID)
                        fAuth = __PERMITTED;
                }
                else {
                    if (session.getMemberRole(nServiceUserID)) {
                        if (page.__nPermID >= session.__nMemberRoleID)
                            fAuth = __PERMITTED;
                        else if (session.__nMemberRoleID == Session::GUEST_ROLE_ID)
                            fAuth = __PENDING_GROUP;
                    }
                    else
                        fAuth = __YET_GROUP;
                }
            }
 
            switch (fAuth) {
                case __PERMITTED: {
                    page.invoke(session);
                    break;
                }
                case __YET_GROUP:
                case __PENDING_GROUP:
                case __DENIED:
                default: {
                    HtmlPage permDenied(
                        pSite->nPermDeniedPageID,
                        pSite
                    );
                    permDenied.init(query);
                    permDenied.print(session);
                }
            }
        }
        else {
            page.printError(session, L"페이지 없음", L"요청하신 페이지가 없습니다.");
        }
 
        if (conn.inTransaction()) {
            conn.commitTrans();
        }
    }
    catch (Exception* _e) {
        if (conn.inTransaction()) {
            conn.rollbackTrans();
        }
        throw _e;
    }
 
    DateTime dtEnd = DateTime::getCurrentLocalTime();
 
    if (ctx.resContentType().compareNoCase(L"text", 4) == 0) {
        ctx.writer() << L"<div>Elapsed Time: "
            << (dtEnd - dtStart).toString()
            << L"</div>\n";
    }
}
 
__DCL_END_NAMESPACE
 
#ifdef __WINNT_NEW_DELETE_OVERRIDE
#undef new
__WINNT_NEW_DELETE_OVERRIDE
#endif