--- parser3/src/targets/isapi/parser3isapi.C 2001/03/22 16:38:23 1.2 +++ parser3/src/targets/isapi/parser3isapi.C 2002/08/01 11:41:21 1.78 @@ -1,64 +1,191 @@ +/** @file + Parser: IIS extension. + + Copyright (c) 2000,2001, 2002 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexandr Petrosian (http://paf.design.ru) +*/ + +static const char* IDENT_PARSER3ISAPI_C="$Date: 2002/08/01 11:41:21 $"; + #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 -#include -#include - -#include -#include -#include +#include "pa_config_includes.h" +#include "pa_sapi.h" #include "pa_globals.h" #include "pa_request.h" #include "pa_version.h" +#include "pool_storage.h" +#include "pa_socks.h" + +#include +#include +#include + +#include #define MAX_STATUS_LENGTH sizeof("xxxx LONGEST STATUS DESCRIPTION") -//@{ -/// service func decl -static const char *get_env(Pool& pool, const char *name) { - LPEXTENSION_CONTROL_BLOCK lpECB=static_cast(pool.context()); +// consts + +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::log(Pool& pool, const char *fmt, ...) { + SAPI_func_context& ctx=*static_cast(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 +void SAPI::die(const char *fmt, ...) { + if(FILE *log=fopen("c:\\die.log", "at")) { + va_list args; + va_start(args,fmt); + vfprintf(log, fmt, args); + fclose(log); + } + exit(1); +} + +const char *SAPI::get_env(Pool& pool, const char *name) { + SAPI_func_context& ctx=*static_cast(pool.get_context()); char *variable_buf=(char *)pool.malloc(MAX_STRING); DWORD variable_len = MAX_STRING-1; - if(lpECB->GetServerVariable(lpECB->ConnID, const_cast(name), + if(ctx.lpECB->GetServerVariable(ctx.lpECB->ConnID, const_cast(name), variable_buf, &variable_len)) { - variable_buf[variable_len]=0; - return variable_buf; + if(*variable_buf) { // saw returning len=1 && *buf=0 :( + variable_buf[variable_len]=0; + return variable_buf; + } + } else if (GetLastError()==ERROR_INSUFFICIENT_BUFFER) { + variable_buf=(char *)pool.malloc(variable_len+1); + + if(ctx.lpECB->GetServerVariable(ctx.lpECB->ConnID, const_cast(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_pair(Pool& pool, const char *key, const char *value) { + char *result=(char *)pool.malloc(strlen(key)+1/*=*/+strlen(value)+1/*0*/); + strcpy(result, key); strcat(result, "="); strcat(result, value); + return result; +} +const char *const *SAPI::environment(Pool& pool) { + // we know this buf is writable + char *all_http_vars=const_cast(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(name), - variable_buf, &variable_len)) { - variable_buf[variable_len]=0; - return variable_buf; + const char **result= + (const char **)pool.malloc(sizeof(char *)*(IIS51var_count+http_var_count+1/*0*/)); + const char **cur=result; + + // IIS5.1 vars + for(int i=0; i(pool.context()); +size_t SAPI::read_post(Pool& pool, char *buf, size_t max_bytes) { + SAPI_func_context& ctx=*static_cast(pool.get_context()); DWORD read_from_buf=0; DWORD read_from_input=0; DWORD total_read=0; - read_from_buf=min(lpECB->cbAvailable, max_bytes); - memcpy(buf, lpECB->lpbData, read_from_buf); + read_from_buf=min(ctx.lpECB->cbAvailable, max_bytes); + memcpy(buf, ctx.lpECB->lpbData, read_from_buf); total_read+=read_from_buf; if(read_from_bufcbTotalBytes) { + read_from_bufcbTotalBytes) { 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) { 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) break; cbRead+=cbSize; @@ -68,224 +195,353 @@ static uint read_post(Pool& pool, char * return total_read; } -static void add_header_attribute(Pool& pool, const char *key, const char *value) { - LPEXTENSION_CONTROL_BLOCK lpECB=static_cast(pool.context()); - String *header=static_cast(pool.tag()); - if(!header) - pool.set_tag(header=new(pool) String(pool)); - - header->APPEND_CONST(key); - header->APPEND_CONST(": "); - header->APPEND_CONST(value); - header->APPEND_CONST("\r\n"); -} - -static void send_header(Pool& pool) { - LPEXTENSION_CONTROL_BLOCK lpECB=static_cast(pool.context()); - String *header=static_cast(pool.tag()); - if(!header) // never - return; +void SAPI::add_header_attribute(Pool& pool, const char *key, const char *value) { + SAPI_func_context& ctx=*static_cast(pool.get_context()); - header->APPEND_CONST("\r\n"); - HSE_SEND_HEADER_EX_INFO header_info; + if(strcasecmp(key, "location")==0) + ctx.http_response_code=302; + + if(strcasecmp(key, "status")==0) + ctx.http_response_code=atoi(value); + else { + ctx.header->APPEND_CONST(key); + ctx.header->APPEND_CONST(": "); + ctx.header->APPEND_CONST(value); + ctx.header->APPEND_CONST("\r\n"); + } +} + +/// @todo intelligent cache-control +void SAPI::send_header(Pool& pool) { + SAPI_func_context& ctx=*static_cast(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]; - switch(http_response_code) { + switch(ctx.http_response_code) { case 200: header_info.pszStatus="200 OK"; break; case 302: header_info.pszStatus="302 Moved Temporarily"; break; - case 401: + case 401:// useless untill parser auth mech header_info.pszStatus="401 Authorization Required"; break; 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; break; } header_info.cchStatus=strlen(header_info.pszStatus); - header_info.pszHeader=header->cstr(); - header_info.cchHeader=header->size(); + *ctx.header << "\r\n"; // ISAPI v<5 did quite well without it + 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, - &header_info, NULL, NULL); + ctx.lpECB->ServerSupportFunction(ctx.lpECB->ConnID, + HSE_REQ_SEND_RESPONSE_HEADER_EX, &header_info, NULL, NULL); } -static void send_body(Pool& pool, const char *buf, size_t size) { - LPEXTENSION_CONTROL_BLOCK lpECB=static_cast(pool.context()); +void SAPI::send_body(Pool& pool, const void *buf, size_t size) { + SAPI_func_context& ctx=*static_cast(pool.get_context()); DWORD num_bytes=size; - lpECB->WriteClient(lpECB->ConnID, const_cast(buf), &num_bytes, HSE_IO_SYNC); + ctx.lpECB->WriteClient(ctx.lpECB->ConnID, + const_cast(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() { +int failed_new(size_t size) { + SAPI::die("out of memory in 'new', failed to allocated %u bytes", size); + return 0; // not reached +} + +#ifdef _DEBUG +static Pool_storage *global_pool_storagep; +#endif +static bool parser_init() { static bool globals_inited=false; if(globals_inited) - return; + return true; globals_inited=true; - static Pool pool; // global pool - PTRY { + _set_new_handler(failed_new); + + static Pool_storage pool_storage; +#ifdef _DEBUG + global_pool_storagep=&pool_storage; +#endif + static Pool pool(&pool_storage); // global pool + try { + // init socks + init_socks(pool); + // init global classes + init_methoded_array(pool); // init global variables - globals_init(pool); - - //... - } PCATCH(e) { // global problem + pa_globals_init(pool); + + // successful finish + return true; + } catch(const Exception& e) { // global problem const char *body=e.comment(); - // TODO: somehow report that error + + // unsuccessful finish + return false; } - PEND_CATCH } /// ISAPI // BOOL WINAPI GetExtensionVersion(HSE_VERSION_INFO *pVer) { pVer->dwExtensionVersion = HSE_VERSION; - strncpy(pVer->lpszExtensionDesc, "Parser " PARSER_VERSION, HSE_MAX_EXT_DLL_NAME_LEN); - return TRUE; + strncpy(pVer->lpszExtensionDesc, "Parser "PARSER_VERSION, HSE_MAX_EXT_DLL_NAME_LEN-1); + pVer->lpszExtensionDesc[HSE_MAX_EXT_DLL_NAME_LEN-1]=0; + return parser_init(); } -/// ISAPI // main workhorse -DWORD WINAPI HttpExtensionProc(LPEXTENSION_CONTROL_BLOCK lpECB) { - Pool pool; - pool.set_context(lpECB); +/** + ISAPI // main workhorse + + @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_handler(Pool& pool, LPEXTENSION_CONTROL_BLOCK lpECB, bool header_only) { + static_cast(pool.get_context())->header=new(pool) String(pool); - // 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=(char *)pool.malloc(path_translated_buf_size); + memcpy(filespec_to_process, 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=(char *)pool.malloc(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=(char *)pool.malloc( + 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; - PTRY { // global try - // must be first in PTRY{}PCATCH - - const char *filespec_to_process=lpECB->lpszPathTranslated; - - // Request info - Request::Info request_info; - - const char *document_root=0; // todo: get from somewhere? - if(!document_root) { - static char fake_document_root[MAX_STRING]; - strncpy(fake_document_root, filespec_to_process, MAX_STRING); - rsplit(fake_document_root, '/'); rsplit(fake_document_root, '\\');// strip filename - document_root=fake_document_root; - } - request_info.document_root=document_root; - request_info.path_translated=filespec_to_process; - request_info.method=lpECB->lpszMethod; - request_info.query_string=lpECB->lpszQueryString; - char reconstructed_uri[MAX_STRING]; - if(lpECB->lpszQueryString && *lpECB->lpszQueryString) { - strncpy(reconstructed_uri, lpECB->lpszPathInfo, - MAX_STRING-1/*'?'*/-strlen(lpECB->lpszQueryString)); - strcat(reconstructed_uri, "?"); - strcat(reconstructed_uri, lpECB->lpszPathInfo); - request_info.uri=reconstructed_uri; - } else - request_info.uri=lpECB->lpszPathInfo; - - request_info.content_type=lpECB->lpszContentType; - request_info.content_length=lpECB->cbTotalBytes; - // cookie - char cookie_buf[MAX_STRING]; - { - DWORD cookie_len = MAX_STRING-1; - - if (lpECB->GetServerVariable(lpECB->ConnID, "HTTP_COOKIE", cookie_buf, &cookie_len)) { - cookie_buf[cookie_len]=0; - request_info.cookie=cookie_buf; - } else if (GetLastError()==ERROR_INSUFFICIENT_BUFFER) { - char *tmp_cookie_buf=(char *)pool.malloc(cookie_len+1); - - if (lpECB->GetServerVariable(lpECB->ConnID, "HTTP_COOKIE", tmp_cookie_buf, &cookie_len)) { - tmp_cookie_buf[cookie_len]=0; - request_info.cookie=tmp_cookie_buf; - } else { - request_info.cookie=0; - } - } - } - - // prepare to process request - Request request(pool, - request_info, - String::UL_HTML_TYPO - ); + request_info.content_type=lpECB->lpszContentType; + request_info.content_length=lpECB->cbTotalBytes; + request_info.cookie=SAPI::get_env(pool, "HTTP_COOKIE"); + request_info.mail_received=false; + + // prepare to process request + Request request(pool, + request_info, + String::UL_HTML|String::UL_OPTIMIZE_BIT, +#ifdef _DEBUG + true +#else + false +#endif + /* status_allowed */); + + // beside by binary + static char beside_binary_path[MAX_STRING]; + strncpy(beside_binary_path, argv0, MAX_STRING-1); beside_binary_path[MAX_STRING-1]=0; // filespec of my binary + if(!( + rsplit(beside_binary_path, '/') || + rsplit(beside_binary_path, '\\'))) { // strip filename + // no path, just filename + beside_binary_path[0]='.'; beside_binary_path[1]=0; + } + char config_filespec[MAX_STRING]; + snprintf(config_filespec, MAX_STRING, + "%s/%s", + beside_binary_path, AUTO_FILE_NAME); + + // process the request + request.core( + config_filespec, false /*fail_on_read_problem*/, // /path/to/first/auto.p + header_only); +} + +void call_real_parser_handler__do_SEH(Pool& pool, + LPEXTENSION_CONTROL_BLOCK lpECB, + bool header_only) { +#if _MSC_VER & !defined(_DEBUG) + LPEXCEPTION_POINTERS system_exception=0; + __try { +#endif + real_parser_handler(pool, lpECB, header_only); - // some root-controlled location - char *root_auto_path; - // c:\windows - root_auto_path=(char *)pool.malloc(MAX_STRING); - 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 _MSC_VER & !defined(_DEBUG) + } __except ( + (system_exception=GetExceptionInformation()), + EXCEPTION_EXECUTE_HANDLER) { + 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 "); + else + throw Exception(0, 0, "Exception "); + } +#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 + }; + 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 - } PCATCH(e) { // global problem + } catch(const Exception& e) { // global problem + // don't allocate anything on pool here: + // possible pool' exception not catch-ed now + // and there could be out-of-memory exception const char *body=e.comment(); - int content_length=strlen(body); + // log it + SAPI::log(pool, "exception in request exception handler: %s", body); - // prepare header - (*service_funcs.add_header_attribute)(pool, "content-type", "text/plain"); - 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); + // + 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 - (*service_funcs.send_header)(pool); + lpECB->dwHttpStatusCode=200; + lpECB->ServerSupportFunction(lpECB->ConnID, + HSE_REQ_SEND_RESPONSE_HEADER_EX, &header_info, NULL, NULL); // send body if(!header_only) - (*service_funcs.send_body)(pool, body, content_length); + SAPI::send_body(pool, body, content_length); // unsuccessful finish - _endthread(); } - PEND_CATCH - - return HSE_STATUS_SUCCESS; +/* + const char *body="test"; + + // + 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 +#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, - DWORD ul_reason_for_call, - LPVOID lpReserved - ) { - switch (ul_reason_for_call) { - case DLL_PROCESS_ATTACH: - parser_init(); - break; - case DLL_THREAD_ATTACH: - break; - case DLL_THREAD_DETACH: - break; - case DLL_PROCESS_DETACH: - break; +#ifdef _DEBUG + newMemState.Checkpoint(); + if( diffMemState.Difference( oldMemState, newMemState ) ) + { + TRACE( "Memory leaked!\n" ); + diffMemState.DumpStatistics( ); + diffMemState.DumpAllObjectsSince(); } - 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; +} \ No newline at end of file