--- parser3/src/targets/cgi/parser3.C 2001/03/13 17:54:14 1.2 +++ parser3/src/targets/cgi/parser3.C 2002/05/06 10:53:53 1.178 @@ -1,43 +1,528 @@ -/* - Parser - Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) - Author: Alexander Petrosyan (http://design.ru/paf) +/** @file + Parser: scripting and CGI main. - $Id: parser3.C,v 1.2 2001/03/13 17:54:14 paf Exp $ + Copyright(c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com) + Author: Alexandr Petrosian (http://paf.design.ru) + + $Id: parser3.C,v 1.178 2002/05/06 10:53:53 paf Exp $ */ -#include "core.h" -#include "pa_request.h" +#include "pa_config_includes.h" + +#if _MSC_VER +# include +# include +#else +# include "pa_config_paths.h" +#endif + +#include "pa_sapi.h" +#include "classes.h" #include "pa_common.h" +#include "pa_request.h" +#include "pa_socks.h" +#include "pa_version.h" +#include "pool_storage.h" + +#ifdef WIN32 +# include +#endif + +//#define DEBUG_POOL_MALLOC +//#define DEBUG_STRING_APPENDS_VS_EXPANDS + +#ifdef DEBUG_STRING_APPENDS_VS_EXPANDS +extern ulong + string_piece_appends, + wcontext_result_size, + total_alloc_size, + string_string_shortcut_economy; +#endif + +// consts + +#ifndef _PROFILE +extern const char *main_RCSIds[]; +#ifdef USE_SMTP +extern const char *smtp_RCSIds[]; +#endif +extern const char *gd_RCSIds[]; +extern const char *classes_RCSIds[]; +extern const char *types_RCSIds[]; +extern const char *parser3_RCSIds[]; +const char **RCSIds[]={ + main_RCSIds, +#ifdef USE_SMTP + smtp_RCSIds, +#endif + gd_RCSIds, + classes_RCSIds, + types_RCSIds, + parser3_RCSIds, + 0 +}; +#endif + +#define REDIRECT_PREFIX "REDIRECT_" +#define PARSER_ROOT_CONFIG_ENV_NAME "HTTP_PARSER_ROOT_CONFIG" +#define PARSER_SITE_CONFIG_ENV_NAME "HTTP_PARSER_SITE_CONFIG" + +/// IIS refuses to read bigger chunks +const size_t READ_POST_CHUNK_SIZE=0x400*0x400; // 1M + +const char *argv0; +Pool *pool; // global pool [dont describe to doxygen: it confuses it with param names] +bool cgi; ///< we were started as CGI? + +// SAPI + +static void log(const char *fmt, va_list args) { + bool opened; + FILE *f=0; + + if(argv0) { + // beside by binary + char file_spec[MAX_STRING]; + strncpy(file_spec, argv0, MAX_STRING-1); file_spec[MAX_STRING-1]=0; // filespec of my binary + rsplit(file_spec, '/'); rsplit(file_spec, '\\');// strip filename + strcat(file_spec, "/parser3.log"); + f=fopen(file_spec, "at"); + } + opened=f!=0; + if(!opened) + f=stderr; + + // prefix + time_t t=time(0); + if(const char *stamp=ctime(&t)) { // never saw that + if(size_t len=strlen(stamp)) // saw once stamp being ="" + fprintf(f, "[%.*s] ", len-1, stamp); + } + // message + + char buf[MAX_STRING]; + size_t size=vsnprintf(buf, MAX_STRING, fmt, args); + remove_crlf(buf, buf+size); + + fwrite(buf, size, 1, f); + // newline + fprintf(f, "\n"); + + if(opened) + fclose(f); + else + fflush(f); +} + +// appends to parser3.log located beside my binary if openable, to stderr otherwize +void SAPI::log(Pool& , const char *fmt, ...) { + va_list args; + va_start(args,fmt); + ::log(fmt, args); + va_end(args); +} + +void SAPI::die(const char *fmt, ...) { +#ifdef DEBUG_POOL_MALLOC + extern void log_pool_stats(Pool& pool); + log_pool_stats(*pool); +#endif + + va_list args; + va_start(args,fmt); + // log + + // logging is more important than user + // she can cancel download, we'd get SIG_PIPE, + // nothing would be logged then + ::log(fmt, args); + + // inform user + + char body[MAX_STRING]; + int content_length=vsnprintf(body, MAX_STRING, fmt, args); + + va_end(args); + + // prepare header + // let's be honest, that's bad we couldn't produce valid output + SAPI::add_header_attribute(*pool, "status", "500"); + SAPI::add_header_attribute(*pool, "content-type", "text/plain"); + char content_length_cstr[MAX_NUMBER]; + snprintf(content_length_cstr, sizeof(content_length_cstr), "%u", content_length); + SAPI::add_header_attribute(*pool, "content-length", content_length_cstr); + + // send header + SAPI::send_header(*pool); + + // body + SAPI::send_body(*pool, body, content_length); + + exit(1); +} + +const char *SAPI::get_env(Pool& , const char *name) { + return getenv(name); +} + +size_t SAPI::read_post(Pool& , char *buf, size_t max_bytes) { + size_t read_size=0; + do { + int chunk_size=read(fileno(stdin), + buf+read_size, min(READ_POST_CHUNK_SIZE, max_bytes-read_size)); + if(chunk_size<=0) + break; + read_size+=chunk_size; + } while(read_sizemalloc( + strlen(path_info)+1/*'?'*/+ + strlen(query_string)+1/*0*/); + strcpy(reconstructed_uri, path_info); + strcat(reconstructed_uri, "?"); + strcat(reconstructed_uri, query_string); + request_info.uri=reconstructed_uri; + } else + request_info.uri=path_info; + else + throw Exception("parser.runtime", + 0, + "CGI: no PATH_INFO defined(in reinventing REQUEST_URI)"); + +#ifndef WIN32 + // they've changed this under IIS5. + if(const char *script_name=SAPI::get_env(*pool, "SCRIPT_NAME")) { + size_t script_name_len=strlen(script_name); + size_t uri_len=strlen(request_info.uri); + if(strncmp(request_info.uri,script_name, script_name_len)==0 && + script_name_len != uri_len) // under IIS they are the same + SAPI::die("CGI: illegal call"); + } +#endif + } else + request_info.uri=""; + + request_info.content_type=SAPI::get_env(*pool, "CONTENT_TYPE"); + const char *content_length=SAPI::get_env(*pool, "CONTENT_LENGTH"); + request_info.content_length=(content_length?atoi(content_length):0); + request_info.cookie=SAPI::get_env(*pool, "HTTP_COOKIE"); + request_info.user_agent=SAPI::get_env(*pool, "HTTP_USER_AGENT"); + + // prepare to process request + Request request(*pool, + request_info, +#ifdef _DEBUG + String::UL_HTML|String::UL_OPTIMIZE_BIT +#else + cgi ? String::UL_HTML|String::UL_OPTIMIZE_BIT : String::UL_AS_IS +#endif + , + true /* status_allowed */); + + // some root-controlled location + const char *root_config_filespec_cstr; + char root_config_filespec_buf[MAX_STRING]; + const char *root_config_by_env=getenv(PARSER_ROOT_CONFIG_ENV_NAME); + if(!root_config_by_env) + root_config_by_env=getenv(REDIRECT_PREFIX PARSER_ROOT_CONFIG_ENV_NAME); + if(root_config_by_env) + root_config_filespec_cstr=root_config_by_env; + else { +#ifdef ROOT_CONFIG_DIR + root_config_filespec_cstr=ROOT_CONFIG_DIR "/" CONFIG_FILE_NAME; +#else +# ifdef WIN32 + // c:\windows + char windows_dir[MAX_STRING]; + GetWindowsDirectory(windows_dir, MAX_STRING); + + + snprintf(root_config_filespec_buf, MAX_STRING, + "%s/%s", + windows_dir, CONFIG_FILE_NAME); + + root_config_filespec_cstr=root_config_filespec_buf; +# else +#error must be compiled either configure/make or MSVC++ +# endif +#endif + } + + const char *site_config_filespec_cstr; + char site_config_filespec_buf[MAX_STRING]; + const char *site_config_by_env=getenv(PARSER_SITE_CONFIG_ENV_NAME); + if(!site_config_by_env) + site_config_by_env=getenv(REDIRECT_PREFIX PARSER_SITE_CONFIG_ENV_NAME); + if(site_config_by_env) + site_config_filespec_cstr=site_config_by_env; + else { + // beside by binary + // @todo full path, not ./! + 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; + } + snprintf(site_config_filespec_buf, MAX_STRING, + "%s/%s", + beside_binary_path, CONFIG_FILE_NAME); + site_config_filespec_cstr=site_config_filespec_buf; + } + + // process the request + request.core( + root_config_filespec_cstr, false /*fail_on_read_problem*/, + site_config_filespec_cstr, false /*fail_on_read_problem*/, + header_only); + + // + done_socks(); + +#ifdef DEBUG_POOL_MALLOC + extern void log_pool_stats(Pool& pool); + log_pool_stats(*pool); +#endif + +#ifdef DEBUG_STRING_APPENDS_VS_EXPANDS + SAPI::log(*pool, + "string piece appends=%lu, wcontext_result_size=%lu, string_string_shortcut_economy_closer=%lu, total_alloc_size=%lu", + string_piece_appends, + wcontext_result_size, + string_string_shortcut_economy, + total_alloc_size); +#endif + +} + +void call_real_parser_handler__do_SEH( + const char *filespec_to_process, + const char *request_method, bool header_only) { +#if _MSC_VER && !defined(_DEBUG) + LPEXCEPTION_POINTERS system_exception=0; + __try { +#endif + real_parser_handler( + filespec_to_process, + request_method, 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 +} + +#if _MSC_VER +int failed_new(size_t size) { + SAPI::die("out of memory in 'new', failed to allocated %u bytes", size); + return 0; // not reached +} +#endif + +#ifdef HAVE_SET_NEW_HANDLER +void failed_new() { + SAPI::die("out of memory in 'new'"); +} +#endif int main(int argc, char *argv[]) { - Pool pool; - core(pool); + Pool_storage global_pool_storage; + Pool global_pool(&global_pool_storage); + pool=&global_pool; + +#ifdef _DEBUG +// _crtBreakAlloc=33112; +#endif +// _asm int 3; + argv0=argv[0]; - bool cgi= + umask(2); + + // were we started as CGI? + cgi= getenv("SERVER_SOFTWARE") || getenv("SERVER_NAME") || getenv("GATEWAY_INTERFACE") || getenv("REQUEST_METHOD"); + + if(!cgi) { + if(argc<2) { + printf( + "Parser/%s Copyright(c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com)\n" + "Author: Alexandr Petrosian (http://paf.design.ru)\n" + "\n" + "Usage: %s \n", + PARSER_VERSION, + argv0?argv0:"parser3"); + return 1; + } + } - // TODO: ifdef WIN32 flip \\ to / - char *document_root="Y:/parser3/src/"; - char *page_filespec="Y:/parser3/src/test.p"; - - // request - Request request(pool, - cgi?String::Untaint_lang::HTML_TYPO:String::Untaint_lang::NO, // :) - document_root, - page_filespec - ); +#ifdef WIN32 + setmode(fileno(stdin), _O_BINARY); + setmode(fileno(stdout), _O_BINARY); + setmode(fileno(stderr), _O_BINARY); +#endif - bool error; - char *result=request.core(error); +#if _MSC_VER + _set_new_handler(failed_new); - if(error) - printf("ERROR: %s", result); - else - printf("%s", result); +#ifdef _DEBUG + // Get current flag + int tmpFlag = _CrtSetDbgFlag( _CRTDBG_REPORT_FLAG ); + + // Turn on leak-checking bit + tmpFlag |= _CRTDBG_LEAK_CHECK_DF; + + // Set flag to the new value + _CrtSetDbgFlag( tmpFlag ); +// _CrtSetBreakAlloc(471); + +#endif + +#endif + +#ifdef HAVE_SET_NEW_HANDLER + std::set_new_handler(failed_new); +#endif + + char *raw_filespec_to_process=cgi?getenv("PATH_TRANSLATED"):argv[1]; + char filespec_to_process[MAX_STRING]; + full_file_spec(raw_filespec_to_process, filespec_to_process, sizeof(filespec_to_process)); + + const char *request_method=getenv("REQUEST_METHOD"); + bool header_only=request_method && strcasecmp(request_method, "HEAD")==0; + + try { // global try + call_real_parser_handler__do_SEH( + filespec_to_process, + request_method, header_only); + } 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 + + SAPI::die("exception in request exception handler: %s", e.comment()); +#ifndef _DEBUG + } catch(...) { + SAPI::die(""); +#endif + } + +#ifndef WIN32 + // + if(!cgi) + SAPI::send_body(*pool, "\n", 1); +#endif +//_asm int 3; return 0; }