|
|
| version 1.2, 2001/03/22 16:38:23 | version 1.82.2.6.2.1, 2003/03/18 15:14:20 |
|---|---|
| Line 1 | Line 1 |
| /** @file | |
| Parser: IIS extension. | |
| Copyright (c) 2000,2001-2003 ArtLebedev Group (http://www.artlebedev.com) | |
| Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru) | |
| */ | |
| static const char* IDENT_PARSER3ISAPI_C="$Date$"; | |
| #ifndef _MSC_VER | #ifndef _MSC_VER |
| # error compile ISAPI module with MSVC | # error compile ISAPI module with MSVC [no urge for now to make it autoconf-ed (PAF)] |
| #endif | #endif |
| #include <windows.h> | #include "pa_config_includes.h" |
| #include <process.h> | |
| #include <httpext.h> | |
| #include <httpfilt.h> | |
| #include <httpext.h> | |
| #include "pa_sapi.h" | |
| #include "pa_globals.h" | #include "pa_globals.h" |
| #include "pa_request.h" | #include "pa_request.h" |
| #include "pa_version.h" | #include "pa_version.h" |
| #include "pool_storage.h" | |
| #include "pa_socks.h" | |
| #include <windows.h> | |
| #include <process.h> | |
| #include <httpext.h> | |
| #define MAX_STATUS_LENGTH sizeof("xxxx LONGEST STATUS DESCRIPTION") | #define MAX_STATUS_LENGTH sizeof("xxxx LONGEST STATUS DESCRIPTION") |
| //@{ | // consts |
| /// service func decl | |
| static const char *get_env(Pool& pool, const char *name) { | |
| LPEXTENSION_CONTROL_BLOCK lpECB=static_cast<LPEXTENSION_CONTROL_BLOCK>(pool.context()); | |
| char *variable_buf=(char *)pool.malloc(MAX_STRING); | const char* IIS51vars[]={ |
| "APPL_MD_PATH", "APPL_PHYSICAL_PATH", | |
| "AUTH_PASSWORD", "AUTH_TYPE", "AUTH_USER", | |
| "CERT_COOKIE", "CERT_FLAGS", "CERT_ISSUER", "CERT_KEYSIZE", "CERT_SECRETKEYSIZE", | |
| "CERT_SERIALNUMBER", "CERT_SERVER_ISSUER", "CERT_SERVER_SUBJECT", "CERT_SUBJECT", | |
| "CONTENT_LENGTH", "CONTENT_TYPE", | |
| "LOGON_USER", | |
| "HTTPS", "HTTPS_KEYSIZE", "HTTPS_SECRETKEYSIZE", "HTTPS_SERVER_ISSUER", "HTTPS_SERVER_SUBJECT", | |
| "INSTANCE_ID", "INSTANCE_META_PATH", | |
| "PATH_INFO", "PATH_TRANSLATED", | |
| "QUERY_STRING", | |
| "REMOTE_ADDR", "REMOTE_HOST", "REMOTE_USER", "REQUEST_METHOD", | |
| "SCRIPT_NAME", | |
| "SERVER_NAME", "SERVER_PORT", "SERVER_PORT_SECURE", "SERVER_PROTOCOL", "SERVER_SOFTWARE", | |
| "URL", | |
| }; | |
| const int IIS51var_count=sizeof(IIS51vars)/sizeof(*IIS51vars); | |
| // globals | |
| char argv0[MAX_STRING]=""; | |
| // SAPI | |
| #ifndef DOXYGEN | |
| /* | |
| ISAPI SAPI functions receive this context information. | |
| see Pool::set_context | |
| */ | |
| struct SAPI_func_context { | |
| LPEXTENSION_CONTROL_BLOCK lpECB; | |
| String *header; | |
| DWORD http_response_code; | |
| }; | |
| #endif | |
| // goes to 'cs-uri-query' log file field. webmaster: switch it ON[default OFF]. | |
| void SAPI::logconst char* fmt, ...) { | |
| SAPI_func_context& ctx=*static_cast<SAPI_func_context *>(pool.get_context()); | |
| va_list args; | |
| va_start(args,fmt); | |
| char buf[MAX_STRING]; | |
| const char* prefix="PARSER_ERROR:"; | |
| strcpy(buf, prefix); | |
| char *start=buf+strlen(prefix); | |
| DWORD size=vsnprintf(start, MAX_STRING-strlen(prefix), fmt, args); | |
| remove_crlf(start, start+size); | |
| ctx.lpECB->ServerSupportFunction(ctx.lpECB->ConnID, | |
| HSE_APPEND_LOG_PARAMETER, buf, &size, 0); | |
| } | |
| /// @todo event log | |
| static void die_or_abort(const char* fmt, va_list args, bool write_core) { | |
| if(FILE *log=fopen("c:\\parser3die.log", "at")) { | |
| vfprintf(log, fmt, args); | |
| fclose(log); | |
| } | |
| // exit & try to produce core dump | |
| abort(); | |
| } | |
| void SAPI::die(const char* fmt, ...) { | |
| va_list args; | |
| va_start(args, fmt); | |
| die_or_abort(fmt, args, false/*write core?*/); | |
| va_end(args); | |
| } | |
| void SAPI::abort(const char* fmt, ...) { | |
| va_list args; | |
| va_start(args, fmt); | |
| die_or_abort(fmt, args, true/*write core?*/); | |
| va_end(args); | |
| } | |
| const char* SAPI::get_envconst char* name) { | |
| SAPI_func_context& ctx=*static_cast<SAPI_func_context *>(pool.get_context()); | |
| char *variable_buf=new char[MAX_STRING]; | |
| DWORD variable_len = MAX_STRING-1; | DWORD variable_len = MAX_STRING-1; |
| if(lpECB->GetServerVariable(lpECB->ConnID, const_cast<char *>(name), | if(ctx.lpECB->GetServerVariable(ctx.lpECB->ConnID, const_cast<char *>(name), |
| variable_buf, &variable_len)) { | variable_buf, &variable_len)) { |
| variable_buf[variable_len]=0; | if(*variable_buf) { // saw returning len=1 && *buf=0 :( |
| return variable_buf; | variable_buf[variable_len]=0; |
| return variable_buf; | |
| } | |
| } else if (GetLastError()==ERROR_INSUFFICIENT_BUFFER) { | |
| variable_buf=new char[variable_len+1]; | |
| if(ctx.lpECB->GetServerVariable(ctx.lpECB->ConnID, const_cast<char *>(name), | |
| variable_buf, &variable_len)) { | |
| if(*variable_buf) { | |
| variable_buf[variable_len]=0; | |
| return variable_buf; | |
| } | |
| } | |
| } | } |
| return 0; | |
| } | |
| variable_buf=(char *)pool.malloc(variable_len+1); | static int grep_char(const char* s, char c) { |
| int result=0; | |
| if(s) { | |
| while(s=strchr(s, c)) { | |
| s++; // skip found c | |
| result++; | |
| } | |
| } | |
| return result; | |
| } | |
| static const char* mk_env_pairconst char* key, const char* value) { | |
| char *result=new char[strlen(key)+1/*=*/+strlen(value)+1/*0*/]; | |
| strcpy(result, key); strcat(result, "="); strcat(result, value); | |
| return result; | |
| } | |
| const char* const *SAPI::environment(SAPI_Info& info, ) { | |
| // we know this buf is writable | |
| const char* all_http_vars=SAPI::get_env(pool, "ALL_HTTP"); | |
| const int http_var_count=grep_char(all_http_vars, '\n')+1/*\n for theoretical(never saw) this \0*/; | |
| if(lpECB->GetServerVariable(lpECB->ConnID, const_cast<char *>(name), | const char* *result=new const char*[IIS51var_count+http_var_count+1/*0*/]; |
| variable_buf, &variable_len)) { | const char* *cur=result; |
| variable_buf[variable_len]=0; | |
| return variable_buf; | // IIS5.1 vars |
| for(int i=0; i<IIS51var_count; i++) { | |
| const char* key=IIS51vars[i]; | |
| if(const char* value=SAPI::get_env(pool, key)) | |
| *cur++=mk_env_pair(pool, key, value); | |
| } | } |
| return 0; | // HTTP_* vars |
| if(char *s=all_http_vars) { | |
| while(char *key=lsplit(&s, '\n')) | |
| if(char *value=lsplit(key, ':')) | |
| *cur++=mk_env_pair(pool, key, value); | |
| } | |
| // mark EOE | |
| *cur=0; | |
| return result; | |
| } | } |
| static uint read_post(Pool& pool, char *buf, uint max_bytes) { | size_t SAPI::read_postchar *buf, size_t max_bytes) { |
| LPEXTENSION_CONTROL_BLOCK lpECB=static_cast<LPEXTENSION_CONTROL_BLOCK>(pool.context()); | SAPI_func_context& ctx=*static_cast<SAPI_func_context *>(pool.get_context()); |
| DWORD read_from_buf=0; | DWORD read_from_buf=0; |
| DWORD read_from_input=0; | DWORD read_from_input=0; |
| DWORD total_read=0; | DWORD total_read=0; |
| read_from_buf=min(lpECB->cbAvailable, max_bytes); | read_from_buf=min(ctx.lpECB->cbAvailable, max_bytes); |
| memcpy(buf, lpECB->lpbData, read_from_buf); | memcpy(buf, ctx.lpECB->lpbData, read_from_buf); |
| total_read+=read_from_buf; | total_read+=read_from_buf; |
| if(read_from_buf<max_bytes && | if(read_from_buf<max_bytes && |
| read_from_buf<lpECB->cbTotalBytes) { | read_from_buf<ctx.lpECB->cbTotalBytes) { |
| DWORD cbRead=0, cbSize; | DWORD cbRead=0, cbSize; |
| read_from_input=min(max_bytes-read_from_buf, lpECB->cbTotalBytes-read_from_buf); | read_from_input=min(max_bytes-read_from_buf, |
| ctx.lpECB->cbTotalBytes-read_from_buf); | |
| while(cbRead < read_from_input) { | while(cbRead < read_from_input) { |
| cbSize=read_from_input - cbRead; | cbSize=read_from_input - cbRead; |
| if(!lpECB->ReadClient(lpECB->ConnID, buf+read_from_buf+cbRead, &cbSize) || | if(!ctx.lpECB->ReadClient(ctx.lpECB->ConnID, |
| buf+read_from_buf+cbRead, &cbSize) || | |
| cbSize==0) | cbSize==0) |
| break; | break; |
| cbRead+=cbSize; | cbRead+=cbSize; |
| Line 68 static uint read_post(Pool& pool, char * | Line 205 static uint read_post(Pool& pool, char * |
| return total_read; | return total_read; |
| } | } |
| static void add_header_attribute(Pool& pool, const char *key, const char *value) { | void SAPI::add_header_attribute |
| LPEXTENSION_CONTROL_BLOCK lpECB=static_cast<LPEXTENSION_CONTROL_BLOCK>(pool.context()); | const char* dont_store_key, const char* dont_store_value) { |
| String *header=static_cast<String *>(pool.tag()); | SAPI_func_context& ctx=*static_cast<SAPI_func_context *>(pool.get_context()); |
| if(!header) | |
| pool.set_tag(header=new(pool) String(pool)); | if(strcasecmp(key, "location")==0) |
| ctx.http_response_code=302; | |
| header->APPEND_CONST(key); | |
| header->APPEND_CONST(": "); | if(strcasecmp(key, "status")==0) |
| header->APPEND_CONST(value); | ctx.http_response_code=atoi(dont_store_value); |
| header->APPEND_CONST("\r\n"); | else { |
| } | todo: copy dont_store_ to nonvilotile |
| ctx.header->APPEND_CONST(dont_store_key); | |
| static void send_header(Pool& pool) { | ctx.header->APPEND_CONST(": "); |
| LPEXTENSION_CONTROL_BLOCK lpECB=static_cast<LPEXTENSION_CONTROL_BLOCK>(pool.context()); | ctx.header->APPEND_CONST(dont_store_value); |
| String *header=static_cast<String *>(pool.tag()); | ctx.header->APPEND_CONST("\r\n"); |
| if(!header) // never | } |
| return; | } |
| header->APPEND_CONST("\r\n"); | /// @todo intelligent cache-control |
| HSE_SEND_HEADER_EX_INFO header_info; | void SAPI::send_header() { |
| SAPI_func_context& ctx=*static_cast<SAPI_func_context *>(pool.get_context()); | |
| int http_response_code=200; // todo: dig from headers | HSE_SEND_HEADER_EX_INFO header_info; |
| char status_buf[MAX_STATUS_LENGTH]; | char status_buf[MAX_STATUS_LENGTH]; |
| switch(http_response_code) { | switch(ctx.http_response_code) { |
| case 200: | case 200: |
| header_info.pszStatus="200 OK"; | header_info.pszStatus="200 OK"; |
| break; | break; |
| case 302: | case 302: |
| header_info.pszStatus="302 Moved Temporarily"; | header_info.pszStatus="302 Moved Temporarily"; |
| break; | break; |
| case 401: | case 401:// useless untill parser auth mech |
| header_info.pszStatus="401 Authorization Required"; | header_info.pszStatus="401 Authorization Required"; |
| break; | break; |
| default: | default: |
| snprintf(status_buf, MAX_STATUS_LENGTH, "%d Undescribed", http_response_code); | snprintf(status_buf, MAX_STATUS_LENGTH, |
| "%d Undescribed", ctx.http_response_code); | |
| header_info.pszStatus=status_buf; | header_info.pszStatus=status_buf; |
| break; | break; |
| } | } |
| header_info.cchStatus=strlen(header_info.pszStatus); | header_info.cchStatus=strlen(header_info.pszStatus); |
| header_info.pszHeader=header->cstr(); | *ctx.header << "\r\n"; // ISAPI v<5 did quite well without it |
| header_info.cchHeader=header->size(); | header_info.pszHeader=ctx.header->cstr(); |
| header_info.cchHeader=ctx.header->size(); | |
| header_info.fKeepConn=true; | |
| lpECB->dwHttpStatusCode=http_response_code; | ctx.lpECB->dwHttpStatusCode=ctx.http_response_code; |
| lpECB->ServerSupportFunction(lpECB->ConnID, HSE_REQ_SEND_RESPONSE_HEADER_EX, | ctx.lpECB->ServerSupportFunction(ctx.lpECB->ConnID, |
| &header_info, NULL, NULL); | HSE_REQ_SEND_RESPONSE_HEADER_EX, &header_info, NULL, NULL); |
| } | } |
| static void send_body(Pool& pool, const char *buf, size_t size) { | void SAPI::send_bodyconst void *buf, size_t size) { |
| LPEXTENSION_CONTROL_BLOCK lpECB=static_cast<LPEXTENSION_CONTROL_BLOCK>(pool.context()); | SAPI_func_context& ctx=*static_cast<SAPI_func_context *>(pool.get_context()); |
| DWORD num_bytes=size; | DWORD num_bytes=size; |
| lpECB->WriteClient(lpECB->ConnID, const_cast<char *>(buf), &num_bytes, HSE_IO_SYNC); | ctx.lpECB->WriteClient(ctx.lpECB->ConnID, |
| const_cast<void *>(buf), &num_bytes, HSE_IO_SYNC); | |
| } | } |
| //@} | |
| /// Service funcs | |
| Service_funcs service_funcs={ | |
| get_env, | |
| read_post, | |
| add_header_attribute, | |
| send_header, | |
| send_body | |
| }; | |
| // | |
| static void parser_init() { | static bool parser_init() { |
| static bool globals_inited=false; | static bool globals_inited=false; |
| if(globals_inited) | if(globals_inited) |
| return; | return true; |
| globals_inited=true; | globals_inited=true; |
| static Pool pool; // global pool | static Pool pool(&pool_storage); // global pool |
| PTRY { | try { |
| // init socks | |
| pa_init_socks(); | |
| // init global classes | |
| init_methoded_array(); | |
| // init global variables | // init global variables |
| globals_init(pool); | pa_globals_init(); |
| // successful finish | |
| return true; | |
| } catch(const Exception& e) { // global problem | |
| const char* body=e.comment(); | |
| //... | // unsuccessful finish |
| } PCATCH(e) { // global problem | return false; |
| const char *body=e.comment(); | |
| // TODO: somehow report that error | |
| } | } |
| PEND_CATCH | |
| } | } |
| /// ISAPI // | /// ISAPI // |
| BOOL WINAPI GetExtensionVersion(HSE_VERSION_INFO *pVer) { | BOOL WINAPI GetExtensionVersion(HSE_VERSION_INFO *pVer) { |
| pVer->dwExtensionVersion = HSE_VERSION; | pVer->dwExtensionVersion = HSE_VERSION; |
| strncpy(pVer->lpszExtensionDesc, "Parser " PARSER_VERSION, HSE_MAX_EXT_DLL_NAME_LEN); | strncpy(pVer->lpszExtensionDesc, "Parser "PARSER_VERSION, HSE_MAX_EXT_DLL_NAME_LEN-1); |
| return TRUE; | pVer->lpszExtensionDesc[HSE_MAX_EXT_DLL_NAME_LEN-1]=0; |
| return parser_init(); | |
| } | } |
| /// ISAPI // main workhorse | /** |
| DWORD WINAPI HttpExtensionProc(LPEXTENSION_CONTROL_BLOCK lpECB) { | ISAPI // main workhorse |
| Pool pool; | |
| pool.set_context(lpECB); | @todo |
| IIS: remove trailing default-document[index.html] from $request.uri. | |
| to do that we need to consult metabase, | |
| wich is tested&works but seems slow runtime | |
| and not could-be-quickly-implemented if prepared. | |
| @test | |
| PARSER_VERSION from outside | |
| */ | |
| void real_parser_handlerLPEXTENSION_CONTROL_BLOCK lpECB, bool header_only) { | |
| static_cast<SAPI_func_context *>(pool.get_context())->header=new String(); | |
| // TODO r->no_cache=1; | // Request info |
| Request::Info request_info; | |
| size_t path_translated_buf_size=strlen(lpECB->lpszPathTranslated)+1; | |
| char *filespec_to_process=pool.copy(lpECB->lpszPathTranslated, path_translated_buf_size); | |
| #ifdef WIN32 | |
| back_slashes_to_slashes(filespec_to_process); | |
| #endif | |
| if(const char* path_info=SAPI::get_env(pool, "PATH_INFO")) { | |
| // IIS | |
| size_t len=strlen(filespec_to_process)-strlen(path_info); | |
| char *buf=new char[len+1]; | |
| strncpy(buf, filespec_to_process, len); buf[len]=0; | |
| request_info.document_root=buf; | |
| } else | |
| throw Exception("parser.runtime", | |
| 0, | |
| "ISAPI: no PATH_INFO defined (in reinventing DOCUMENT_ROOT)"); | |
| request_info.path_translated=filespec_to_process; | |
| request_info.method=lpECB->lpszMethod; | |
| request_info.query_string=lpECB->lpszQueryString; | |
| if(lpECB->lpszQueryString && *lpECB->lpszQueryString) { | |
| char *reconstructed_uri=new char[ | |
| strlen(lpECB->lpszPathInfo)+1/*'?'*/+ | |
| strlen(lpECB->lpszQueryString)+1/*0*/]; | |
| strcpy(reconstructed_uri, lpECB->lpszPathInfo); | |
| strcat(reconstructed_uri, "?"); | |
| strcat(reconstructed_uri, lpECB->lpszQueryString); | |
| request_info.uri=reconstructed_uri; | |
| } else | |
| request_info.uri=lpECB->lpszPathInfo; | |
| bool header_only=strcasecmp(lpECB->lpszMethod, "HEAD")==0; | request_info.content_type=lpECB->lpszContentType; |
| PTRY { // global try | request_info.content_length=lpECB->cbTotalBytes; |
| // must be first in PTRY{}PCATCH | request_info.cookie=SAPI::get_env(pool, "HTTP_COOKIE"); |
| request_info.mail_received=false; | |
| const char *filespec_to_process=lpECB->lpszPathTranslated; | |
| // prepare to process request | |
| // Request info | Request request(pool, |
| Request::Info request_info; | request_info, |
| String::UL_HTML|String::UL_OPTIMIZE_BIT, | |
| const char *document_root=0; // todo: get from somewhere? | #ifdef _DEBUG |
| if(!document_root) { | true |
| static char fake_document_root[MAX_STRING]; | #else |
| strncpy(fake_document_root, filespec_to_process, MAX_STRING); | false |
| rsplit(fake_document_root, '/'); rsplit(fake_document_root, '\\');// strip filename | #endif |
| document_root=fake_document_root; | /* status_allowed */); |
| } | |
| request_info.document_root=document_root; | // beside by binary |
| request_info.path_translated=filespec_to_process; | static char beside_binary_path[MAX_STRING]; |
| request_info.method=lpECB->lpszMethod; | strncpy(beside_binary_path, argv0, MAX_STRING-1); beside_binary_path[MAX_STRING-1]=0; // filespec of my binary |
| request_info.query_string=lpECB->lpszQueryString; | if(!( |
| char reconstructed_uri[MAX_STRING]; | rsplit(beside_binary_path, '/') || |
| if(lpECB->lpszQueryString && *lpECB->lpszQueryString) { | rsplit(beside_binary_path, '\\'))) { // strip filename |
| strncpy(reconstructed_uri, lpECB->lpszPathInfo, | // no path, just filename |
| MAX_STRING-1/*'?'*/-strlen(lpECB->lpszQueryString)); | beside_binary_path[0]='.'; beside_binary_path[1]=0; |
| strcat(reconstructed_uri, "?"); | } |
| strcat(reconstructed_uri, lpECB->lpszPathInfo); | char config_filespec[MAX_STRING]; |
| request_info.uri=reconstructed_uri; | snprintf(config_filespec, MAX_STRING, |
| } else | "%s/%s", |
| request_info.uri=lpECB->lpszPathInfo; | beside_binary_path, AUTO_FILE_NAME); |
| bool fail_on_config_read_problem=entry_exists(config_filespec); | |
| request_info.content_type=lpECB->lpszContentType; | |
| request_info.content_length=lpECB->cbTotalBytes; | // process the request |
| // cookie | request.core( |
| char cookie_buf[MAX_STRING]; | config_filespec, fail_on_config_read_problem, // /path/to/first/auto.p |
| { | header_only); |
| DWORD cookie_len = MAX_STRING-1; | } |
| if (lpECB->GetServerVariable(lpECB->ConnID, "HTTP_COOKIE", cookie_buf, &cookie_len)) { | void call_real_parser_handler__do_SEH |
| cookie_buf[cookie_len]=0; | LPEXTENSION_CONTROL_BLOCK lpECB, |
| request_info.cookie=cookie_buf; | bool header_only) { |
| } else if (GetLastError()==ERROR_INSUFFICIENT_BUFFER) { | #if _MSC_VER & !defined(_DEBUG) |
| char *tmp_cookie_buf=(char *)pool.malloc(cookie_len+1); | LPEXCEPTION_POINTERS system_exception=0; |
| __try { | |
| if (lpECB->GetServerVariable(lpECB->ConnID, "HTTP_COOKIE", tmp_cookie_buf, &cookie_len)) { | #endif |
| tmp_cookie_buf[cookie_len]=0; | real_parser_handler(pool, lpECB, header_only); |
| request_info.cookie=tmp_cookie_buf; | |
| } else { | |
| request_info.cookie=0; | |
| } | |
| } | |
| } | |
| // prepare to process request | |
| Request request(pool, | |
| request_info, | |
| String::UL_HTML_TYPO | |
| ); | |
| // some root-controlled location | #if _MSC_VER & !defined(_DEBUG) |
| char *root_auto_path; | } __except ( |
| // c:\windows | (system_exception=GetExceptionInformation()), |
| root_auto_path=(char *)pool.malloc(MAX_STRING); | EXCEPTION_EXECUTE_HANDLER) { |
| GetWindowsDirectory(root_auto_path, MAX_STRING); | |
| /* | |
| char *fuck_it="fff"; | |
| DWORD num_bytes=strlen(fuck_it);; | |
| lpECB->WriteClient(lpECB->ConnID, fuck_it, &num_bytes, HSE_IO_SYNC); | |
| return HSE_STATUS_SUCCESS; | |
| */ | |
| // process the request | |
| request.core( | |
| root_auto_path, false/*may be abcent*/, // /path/to/admin/auto.p | |
| 0/*parser_site_auto_path*/, false, // /path/to/site/auto.p | |
| header_only); | |
| if(system_exception) | |
| if(_EXCEPTION_RECORD *er=system_exception->ExceptionRecord) | |
| throw Exception(0, | |
| 0, | |
| "Exception 0x%08X at 0x%08X", er->ExceptionCode, er->ExceptionAddress); | |
| else | |
| throw Exception(0, 0, "Exception <no exception record>"); | |
| else | |
| throw Exception(0, 0, "Exception <no exception information>"); | |
| } | |
| #endif | |
| } | |
| inline DWORD RealHttpExtensionProc(LPEXTENSION_CONTROL_BLOCK lpECB) { | |
| Pool_storage pool_storage; | |
| Pool pool(&pool_storage); // no allocations until assigned context [for reporting] | |
| SAPI_func_context ctx={ | |
| lpECB, | |
| 0, // filling later: so that if there would be error pool would have ctx | |
| 200 // default http_response_code [lpECB->dwHttpStatusCode seems to be always 0, even on 404 redirect to /404.html] | |
| }; | |
| pool.set_context(&ctx);// no allocations before this line! | |
| bool header_only=strcasecmp(lpECB->lpszMethod, "HEAD")==0; | |
| try { // global try | |
| call_real_parser_handler__do_SEH(pool, lpECB, header_only); | |
| // successful finish | // successful finish |
| } PCATCH(e) { // global problem | } catch(const Exception& e) { // global problem |
| const char *body=e.comment(); | // don't allocate anything on pool here: |
| int content_length=strlen(body); | // possible pool' exception not catch-ed now |
| // and there could be out-of-memory exception | |
| const char* body=e.comment(); | |
| // log it | |
| SAPI::log(pool, "exception in request exception handler: %s", body); | |
| // prepare header | // |
| (*service_funcs.add_header_attribute)(pool, "content-type", "text/plain"); | int content_length=strlen(body); |
| char content_length_cstr[MAX_NUMBER]; | |
| snprintf(content_length_cstr, MAX_NUMBER, "%d", content_length); | |
| (*service_funcs.add_header_attribute)(pool, "content-length", | |
| content_length_cstr); | |
| // prepare header // not using SAPI func wich allocates on pool | |
| char header_buf[MAX_STRING]; | |
| int header_len=snprintf(header_buf, MAX_STRING, | |
| "content-type: text/plain\r\n" | |
| "content-length: %lu\r\n" | |
| // "expires: Fri, 23 Mar 2001 09:32:23 GMT\r\n" | |
| "\r\n", | |
| content_length); | |
| HSE_SEND_HEADER_EX_INFO header_info; | |
| header_info.pszStatus="200 OK"; | |
| header_info.cchStatus=strlen(header_info.pszStatus); | |
| header_info.pszHeader=header_buf; | |
| header_info.cchHeader=header_len; | |
| header_info.fKeepConn=true; | |
| // send header | // send header |
| (*service_funcs.send_header)(pool); | lpECB->dwHttpStatusCode=200; |
| lpECB->ServerSupportFunction(lpECB->ConnID, | |
| HSE_REQ_SEND_RESPONSE_HEADER_EX, &header_info, NULL, NULL); | |
| // send body | // send body |
| if(!header_only) | if(!header_only) |
| (*service_funcs.send_body)(pool, body, content_length); | SAPI::send_body(pool, body, content_length); |
| // unsuccessful finish | // unsuccessful finish |
| _endthread(); | |
| } | } |
| PEND_CATCH | /* |
| const char* body="test"; | |
| return HSE_STATUS_SUCCESS; | |
| // | |
| int content_length=strlen(body); | |
| // prepare header // not using SAPI func wich allocates on pool | |
| char header_buf[MAX_STRING]; | |
| int header_len=snprintf(header_buf, MAX_STRING, | |
| "content-type: text/plain\r\n" | |
| "content-length: %lu\r\n" | |
| "expires: Fri, 23 Mar 2001 09:32:23 GMT\r\n" | |
| "\r\n", | |
| content_length); | |
| HSE_SEND_HEADER_EX_INFO header_info; | |
| header_info.pszStatus="200 OK"; | |
| header_info.cchStatus=strlen(header_info.pszStatus); | |
| header_info.pszHeader=header_buf; | |
| header_info.cchHeader=header_len; | |
| header_info.fKeepConn=true; | |
| // send header | |
| lpECB->dwHttpStatusCode=200; | |
| lpECB->ServerSupportFunction(lpECB->ConnID, | |
| HSE_REQ_SEND_RESPONSE_HEADER_EX, &header_info, NULL, NULL); | |
| // send body | |
| DWORD num_bytes=content_length; | |
| lpECB->WriteClient(lpECB->ConnID, | |
| (void *)body, &num_bytes, HSE_IO_SYNC); | |
| */ | |
| return HSE_STATUS_SUCCESS_AND_KEEP_CONN; | |
| } | } |
| #ifdef _DEBUG | |
| //for memory leaks detection only | |
| #undef _WINDOWS_ | |
| #include <afx.h> | |
| #endif | |
| DWORD WINAPI HttpExtensionProc(LPEXTENSION_CONTROL_BLOCK lpECB) { | |
| // Declare the variables needed | |
| #ifdef _DEBUG | |
| // _crtBreakAlloc=97779;//97777; //997; | |
| CMemoryState oldMemState, newMemState, diffMemState; | |
| oldMemState.Checkpoint(); | |
| //global_pool_storagep->fbreak_on_alloc=true; | |
| #endif | |
| DWORD result=RealHttpExtensionProc(lpECB); | |
| BOOL APIENTRY DllMain(HANDLE hModule, | #ifdef _DEBUG |
| DWORD ul_reason_for_call, | newMemState.Checkpoint(); |
| LPVOID lpReserved | if( diffMemState.Difference( oldMemState, newMemState ) ) |
| ) { | { |
| switch (ul_reason_for_call) { | TRACE( "Memory leaked!\n" ); |
| case DLL_PROCESS_ATTACH: | diffMemState.DumpStatistics( ); |
| parser_init(); | diffMemState.DumpAllObjectsSince(); |
| break; | |
| case DLL_THREAD_ATTACH: | |
| break; | |
| case DLL_THREAD_DETACH: | |
| break; | |
| case DLL_PROCESS_DETACH: | |
| break; | |
| } | } |
| return TRUE; | #endif |
| return result; | |
| } | } |
| BOOL WINAPI DllMain( | |
| HINSTANCE hinstDLL, // handle to the DLL module | |
| DWORD fdwReason, // reason for calling function | |
| LPVOID lpvReserved // reserved | |
| ) { | |
| GetModuleFileName( | |
| hinstDLL, // handle to module | |
| argv0, // file name of module | |
| sizeof(argv0) // size of buffer | |
| ); | |
| return TRUE; | |
| } | |