--- parser3/src/targets/cgi/parser3.C 2001/03/13 17:54:14 1.2 +++ parser3/src/targets/cgi/parser3.C 2002/01/31 11:11:45 1.152 @@ -1,43 +1,472 @@ -/* - 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 ArtLebedev Group(http://www.artlebedev.com) + Author: Alexander Petrosyan (http://paf.design.ru) + + $Id: parser3.C,v 1.152 2002/01/31 11:11:45 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 + +// consts + +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 +}; + +/// IIS refuses to read bigger chunks +const size_t READ_POST_CHUNK_SIZE=0x400*0x400; // 1M + +const char *argv0; +Pool_storage pool_storage; +Pool pool(&pool_storage); // 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); + const char *stamp=ctime(&t); + fprintf(f, "[%.*s] ", strlen(stamp)-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, MAX_NUMBER, "%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_sizeExceptionRecord) + throw Exception(0, 0, + 0, + "Exception 0x%08X at 0x%08X", er->ExceptionCode, er->ExceptionAddress); + else + throw Exception(0, 0, 0, "Exception "); + else + throw Exception(0, 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); +// _asm int 3; + argv0=argv[0]; + + umask(2); - bool cgi= + // 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 ArtLebedev Group(http://www.artlebedev.com)\n" + "Author: Alexander Petrosyan (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"; +#ifdef WIN32 + setmode(fileno(stdin), _O_BINARY); + setmode(fileno(stdout), _O_BINARY); + setmode(fileno(stderr), _O_BINARY); +#endif - // request - Request request(pool, - cgi?String::Untaint_lang::HTML_TYPO:String::Untaint_lang::NO, // :) - document_root, - page_filespec - ); +#if _MSC_VER + _set_new_handler(failed_new); - bool error; - char *result=request.core(error); +#ifdef _DEBUG + // Get current flag + int tmpFlag = _CrtSetDbgFlag( _CRTDBG_REPORT_FLAG ); - if(error) - printf("ERROR: %s", result); - else - printf("%s", result); + // 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 *filespec_to_process=cgi?getenv("PATH_TRANSLATED"):argv[1]; +#ifdef WIN32 + back_slashes_to_slashes(filespec_to_process); +#endif + filespec_to_process=full_file_spec(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 return 0; }