Annotation of parser3/src/targets/isapi/parser3isapi.C, revision 1.7

1.1       paf         1: #ifndef _MSC_VER
                      2: #      error compile ISAPI module with MSVC
                      3: #endif
                      4: 
                      5: #include <windows.h>
                      6: #include <process.h>
                      7: 
                      8: #include <httpext.h>
                      9: 
                     10: #include "pa_globals.h"
                     11: #include "pa_request.h"
                     12: #include "pa_version.h"
                     13: 
                     14: #define MAX_STATUS_LENGTH sizeof("xxxx LONGEST STATUS DESCRIPTION")
                     15: 
                     16: //@{
                     17: /// service func decl
1.3       paf        18: 
                     19: struct Service_func_context {
                     20:        LPEXTENSION_CONTROL_BLOCK lpECB;
                     21:        String *header;
                     22:        DWORD http_response_code;
                     23: };
                     24: 
1.1       paf        25: static const char *get_env(Pool& pool, const char *name) {
1.3       paf        26:        Service_func_context& ctx=*static_cast<Service_func_context *>(pool.context());
1.1       paf        27: 
                     28:        char *variable_buf=(char *)pool.malloc(MAX_STRING);
                     29:        DWORD variable_len = MAX_STRING-1;
                     30: 
1.3       paf        31:        if(ctx.lpECB->GetServerVariable(ctx.lpECB->ConnID, const_cast<char *>(name), 
1.1       paf        32:                variable_buf, &variable_len)) {
                     33:                variable_buf[variable_len]=0;
                     34:                return variable_buf;
1.7     ! paf        35:        } else if (GetLastError()==ERROR_INSUFFICIENT_BUFFER) {
        !            36:                variable_buf=(char *)pool.malloc(variable_len+1);
        !            37:                
        !            38:                if(ctx.lpECB->GetServerVariable(ctx.lpECB->ConnID, const_cast<char *>(name), 
        !            39:                        variable_buf, &variable_len)) {
        !            40:                        variable_buf[variable_len]=0;
        !            41:                        return variable_buf;
        !            42:                }
1.1       paf        43:        }
1.7     ! paf        44:                
1.1       paf        45:        return 0;
                     46: }
                     47: 
                     48: static uint read_post(Pool& pool, char *buf, uint max_bytes) {
1.3       paf        49:        Service_func_context& ctx=*static_cast<Service_func_context *>(pool.context());
1.1       paf        50: 
                     51:        DWORD read_from_buf=0;
                     52:        DWORD read_from_input=0;
                     53:        DWORD total_read=0;
                     54: 
1.3       paf        55:        read_from_buf=min(ctx.lpECB->cbAvailable, max_bytes);
                     56:        memcpy(buf, ctx.lpECB->lpbData, read_from_buf);
1.1       paf        57:        total_read+=read_from_buf;
                     58: 
                     59:        if(read_from_buf<max_bytes &&
1.3       paf        60:                read_from_buf<ctx.lpECB->cbTotalBytes) {
1.1       paf        61:                DWORD cbRead=0, cbSize;
                     62: 
1.3       paf        63:                read_from_input=min(max_bytes-read_from_buf, 
                     64:                        ctx.lpECB->cbTotalBytes-read_from_buf);
1.1       paf        65:                while(cbRead < read_from_input) {
                     66:                        cbSize=read_from_input - cbRead;
1.3       paf        67:                        if(!ctx.lpECB->ReadClient(ctx.lpECB->ConnID, 
                     68:                                buf+read_from_buf+cbRead, &cbSize) || 
1.1       paf        69:                                cbSize==0) 
                     70:                                break;
                     71:                        cbRead+=cbSize;
                     72:                }
                     73:                total_read+=cbRead;
                     74:        }
                     75:        return total_read;
                     76: }
                     77: 
                     78: static void add_header_attribute(Pool& pool, const char *key, const char *value) {
1.3       paf        79:        Service_func_context& ctx=*static_cast<Service_func_context *>(pool.context());
                     80: 
                     81:        if(strcasecmp(key, "location")==0) 
                     82:                ctx.http_response_code=302;
                     83: 
                     84:        if(strcasecmp(key, "status")==0) 
                     85:                ctx.http_response_code=atoi(value);
                     86:        else {
                     87:                ctx.header->APPEND_CONST(key);
                     88:                ctx.header->APPEND_CONST(": ");
                     89:                ctx.header->APPEND_CONST(value);
                     90:                ctx.header->APPEND_CONST("\n");
                     91:        }
1.1       paf        92: }
                     93: 
1.3       paf        94: /// @todo intelligent cache-control
1.1       paf        95: static void send_header(Pool& pool) {
1.3       paf        96:        Service_func_context& ctx=*static_cast<Service_func_context *>(pool.context());
1.1       paf        97: 
1.6       paf        98:        ctx.header->APPEND_CONST(
                     99:                "Connection: keep-alive\n"
                    100:                "Cache-Control: no-cache\n"
                    101:                "\n");
1.1       paf       102:        HSE_SEND_HEADER_EX_INFO header_info;
                    103: 
                    104:        char status_buf[MAX_STATUS_LENGTH];
1.3       paf       105:        switch(ctx.http_response_code) {
1.1       paf       106:                case 200:
                    107:                        header_info.pszStatus="200 OK";
                    108:                        break;
                    109:                case 302:
                    110:                        header_info.pszStatus="302 Moved Temporarily";
                    111:                        break;
1.3       paf       112:                /*case 401:
1.1       paf       113:                        header_info.pszStatus="401 Authorization Required";
1.3       paf       114:                        break;*/
1.1       paf       115:                default:
1.3       paf       116:                        snprintf(status_buf, MAX_STATUS_LENGTH, 
                    117:                                "%d Undescribed", ctx.http_response_code);
1.1       paf       118:                        header_info.pszStatus=status_buf;
                    119:                        break;
                    120:        }
                    121:        header_info.cchStatus=strlen(header_info.pszStatus);
1.3       paf       122:        header_info.pszHeader=ctx.header->cstr();
                    123:        header_info.cchHeader=ctx.header->size();
1.5       paf       124:        header_info.fKeepConn=true;
1.1       paf       125: 
1.3       paf       126:        ctx.lpECB->dwHttpStatusCode=ctx.http_response_code;
1.1       paf       127: 
1.3       paf       128:        ctx.lpECB->ServerSupportFunction(ctx.lpECB->ConnID, 
                    129:                HSE_REQ_SEND_RESPONSE_HEADER_EX, &header_info, NULL, NULL);
1.1       paf       130: }
                    131: 
                    132: static void send_body(Pool& pool, const char *buf, size_t size) {
1.3       paf       133:        Service_func_context& ctx=*static_cast<Service_func_context *>(pool.context());
1.1       paf       134: 
                    135:        DWORD num_bytes=size;
1.3       paf       136:        ctx.lpECB->WriteClient(ctx.lpECB->ConnID, 
                    137:                const_cast<char *>(buf), &num_bytes, HSE_IO_SYNC);
1.1       paf       138: }
                    139: //@}
                    140: 
                    141: /// Service funcs 
                    142: Service_funcs service_funcs={
                    143:        get_env,
                    144:        read_post,
                    145:        add_header_attribute,
                    146:        send_header,
                    147:        send_body
                    148: };
                    149: 
                    150: // 
                    151: 
                    152: static void parser_init() {
                    153:        static bool globals_inited=false;
                    154:        if(globals_inited)
                    155:                return;
                    156:        globals_inited=true;
                    157: 
                    158:        static Pool pool; // global pool
                    159:        PTRY {
                    160:                // init global variables
                    161:                globals_init(pool);
                    162:                
                    163:                //...
                    164:        } PCATCH(e) { // global problem 
                    165:                const char *body=e.comment();
                    166:                // TODO: somehow report that error
                    167:        }
                    168:        PEND_CATCH
                    169: }
                    170: 
                    171: /// ISAPI //
                    172: BOOL WINAPI GetExtensionVersion(HSE_VERSION_INFO *pVer) {
                    173:        pVer->dwExtensionVersion = HSE_VERSION;
                    174:        strncpy(pVer->lpszExtensionDesc, "Parser " PARSER_VERSION, HSE_MAX_EXT_DLL_NAME_LEN);
                    175:        return TRUE;
                    176: }
                    177: 
1.6       paf       178: /** 
                    179:        ISAPI // main workhorse
                    180: 
                    181:        @todo 
                    182:                think of a better way than @c APPL_PHYSICAL_PATH
                    183:                of obtaining the @c DOCUMENT_ROOT
                    184:                because this only gets "the place where last IIS Application was set"
                    185:                and if someone would redefine Application settings below the /
                    186:                all ^table:load[/test] would open not /test but /below/test
                    187: */
1.1       paf       188: DWORD WINAPI HttpExtensionProc(LPEXTENSION_CONTROL_BLOCK lpECB) {
                    189:        Pool pool;
1.2       paf       190:        
1.1       paf       191:        bool header_only=strcasecmp(lpECB->lpszMethod, "HEAD")==0;
                    192:        PTRY { // global try
1.3       paf       193:                Service_func_context ctx={
                    194:                        lpECB,
                    195:                        new(pool) String(pool),
                    196:                        200
                    197:                };
                    198:                pool.set_context(&ctx);
                    199:                
1.2       paf       200:                // Request info
                    201:                Request::Info request_info;
                    202:                
1.7     ! paf       203:                if(!(request_info.document_root=get_env(pool, "APPL_PHYSICAL_PATH")))
1.6       paf       204:                        PTHROW(0, 0,
                    205:                                0,
1.7     ! paf       206:                                "can not get server variable APPL_PHYSICAL_PATH (error #%lu)",
1.6       paf       207:                                        GetLastError()); // never
                    208: 
                    209:                request_info.path_translated=lpECB->lpszPathTranslated;
1.2       paf       210:                request_info.method=lpECB->lpszMethod;
                    211:                request_info.query_string=lpECB->lpszQueryString;
                    212:                if(lpECB->lpszQueryString && *lpECB->lpszQueryString) {
1.7     ! paf       213:                        char *reconstructed_uri=(char *)malloc(
        !           214:                                strlen(lpECB->lpszPathInfo)+1/*'?'*/+
        !           215:                                strlen(lpECB->lpszQueryString)+1/*0*/);
        !           216:                        strcpy(reconstructed_uri, lpECB->lpszPathInfo);
1.2       paf       217:                        strcat(reconstructed_uri, "?");
1.7     ! paf       218:                        strcat(reconstructed_uri, lpECB->lpszQueryString);
1.2       paf       219:                        request_info.uri=reconstructed_uri;
                    220:                } else
                    221:                        request_info.uri=lpECB->lpszPathInfo;
                    222:                
                    223:                request_info.content_type=lpECB->lpszContentType;
                    224:                request_info.content_length=lpECB->cbTotalBytes;
1.7     ! paf       225:                request_info.cookie=get_env(pool, "HTTP_COOKIE");
1.2       paf       226:                
                    227:                // prepare to process request
                    228:                Request request(pool,
                    229:                        request_info,
                    230:                        String::UL_HTML_TYPO
                    231:                        );
                    232:                
                    233:                // some root-controlled location
1.7     ! paf       234:                //   c:\windows
        !           235:                // must be dynamic: rethrowing from request.core 
        !           236:                //   may return 'source' which can be inside of 'root auto.p@exeception'
        !           237:                char *root_auto_path=(char *)pool.malloc(MAX_STRING);
1.2       paf       238:                GetWindowsDirectory(root_auto_path, MAX_STRING);
1.4       paf       239: 
1.2       paf       240:                // process the request
                    241:                request.core(
                    242:                        root_auto_path, false/*may be abcent*/, // /path/to/admin/auto.p
                    243:                        0/*parser_site_auto_path*/, false, // /path/to/site/auto.p
                    244:                        header_only);
                    245:                
                    246:                // successful finish
1.1       paf       247:        } PCATCH(e) { // global problem 
                    248:                const char *body=e.comment();
                    249:                int content_length=strlen(body);
                    250: 
                    251:                // prepare header
1.7     ! paf       252:                add_header_attribute(pool, "content-type", "text/plain");
1.1       paf       253:                char content_length_cstr[MAX_NUMBER];
1.6       paf       254:                snprintf(content_length_cstr, MAX_NUMBER, "%lu", content_length);
1.7     ! paf       255:                add_header_attribute(pool, "content-length", content_length_cstr);
1.1       paf       256: 
                    257:                // send header
1.7     ! paf       258:                send_header(pool);
1.1       paf       259: 
                    260:                // send body
                    261:                if(!header_only)
1.7     ! paf       262:                        send_body(pool, body, content_length);
1.1       paf       263: 
                    264:                // unsuccessful finish
                    265:                _endthread();
                    266:        }
                    267:        PEND_CATCH
                    268:        
1.6       paf       269:        return HSE_STATUS_SUCCESS_AND_KEEP_CONN;
1.1       paf       270: }
                    271: 
                    272: 
                    273: BOOL APIENTRY DllMain(HANDLE hModule, 
                    274:                                          DWORD  ul_reason_for_call, 
                    275:                                          LPVOID lpReserved
                    276:                                          ) {
                    277:     switch (ul_reason_for_call) {
                    278:                case DLL_PROCESS_ATTACH:
                    279:                        parser_init();
                    280:                        break;
                    281:                case DLL_THREAD_ATTACH:
                    282:                        break;
                    283:                case DLL_THREAD_DETACH:
                    284:                        break;
                    285:                case DLL_PROCESS_DETACH:
                    286:                        break;
                    287:     }
                    288:     return TRUE;
                    289: }

E-mail: