--- parser3/src/main/pa_exec.C 2004/07/07 11:29:06 1.63 +++ parser3/src/main/pa_exec.C 2009/10/02 01:18:27 1.78 @@ -1,13 +1,13 @@ /** @file Parser: program executing for different OS-es. - Copyright(c) 2000,2001-2004 ArtLebedev Group(http://www.artlebedev.com) + Copyright(c) 2000-2009 ArtLebedev Group(http://www.artlebedev.com) Author: Alexandr Petrosian (http://paf.design.ru) @todo setrlimit */ -static const char * const IDENT_EXEC_C="$Date: 2004/07/07 11:29:06 $"; +static const char * const IDENT_EXEC_C="$Date: 2009/10/02 01:18:27 $"; #include "pa_config_includes.h" @@ -27,11 +27,12 @@ static const char * const IDENT_EXEC_C=" /// this func from http://www.ccas.ru/~posp/popov/spawn.htm static DWORD CreateHiddenConsoleProcess(LPCTSTR szCmdLine, + LPCTSTR szScriptFileSpec, char *szEnv, - PROCESS_INFORMATION* ppi, - LPHANDLE phInWrite, - LPHANDLE phOutRead, - LPHANDLE phErrRead) + PROCESS_INFORMATION* ppi, + LPHANDLE phInWrite, + LPHANDLE phOutRead, + LPHANDLE phErrRead) { DWORD result=0; BOOL fCreated; @@ -72,10 +73,9 @@ static DWORD CreateHiddenConsoleProcess( // calculating script's directory char dir[MAX_STRING]; - strncpy(dir, szCmdLine, MAX_STRING-1); dir[MAX_STRING-1]=0; + strncpy(dir, szScriptFileSpec, MAX_STRING-1); dir[MAX_STRING-1]=0; lsplit(dir,' '); // trim arguments rsplit(dir,'/'); rsplit(dir,'\\'); // trim filename - chdir(dir); // Create a child process (suspended) fCreated=CreateProcess(NULL, @@ -113,15 +113,38 @@ error: static void read_pipe(String& result, HANDLE hOutRead, String::Language lang){ while(true) { - char *buf=new(PointerFreeGC) char[MAX_STRING]; - unsigned long size; - if(!ReadFile(hOutRead, buf, MAX_STRING-1, &size, NULL) || !size) + char *buf=new(PointerFreeGC) char[MAX_STRING+1]; + DWORD size=0; + if(!ReadFile(hOutRead, buf, MAX_STRING, &size, NULL) || !size) break; buf[size]=0; result.append_know_length(buf, size, lang); } } +static void read_pipe(File_read_result& result, HANDLE hOutRead){ + + char *buf=(char*)pa_malloc(MAX_STRING+1); + DWORD bufsize = MAX_STRING; + + result.headers = 0; + result.length = 0; + result.str = 0; + result.success = false; + + while(true) { + DWORD size=0; + if(!ReadFile(hOutRead, buf + result.length, bufsize - result.length, &size, NULL) || !size) + break; + result.length += size; + if(result.length >= bufsize){ + bufsize *= 2; + buf=(char*)pa_realloc(buf, bufsize+1); + } + result.str=buf; + } +} + static const char* buildCommand(const char* file_spec_cstr, const ArrayString& argv) { const char* result=file_spec_cstr; if(FILE *f=fopen(file_spec_cstr, "r")) { @@ -132,9 +155,9 @@ static const char* buildCommand(const ch buf[size]=0; if(strncmp(buf, "#!", 2)==0) { const char* begin=buf+2; - if(*begin==' ') // alx: were an old magic for some linux-es + while(*begin==' ') // alx: were an old magic for some linux-es begin++; - if(char *end=strchr(begin, '\n')) { + if(const char *end=strchr(begin, '\n')) { String string(pa_strdup(begin, end-begin)); string << " " << file_spec_cstr; result=string.cstr(); @@ -281,8 +304,8 @@ static int get_exit_status(int pid) { static void read_pipe(String& result, int file, String::Language lang){ while(true) { - char *buf=new(PointerFreeGC) char[MAX_STRING]; - ssize_t length=read(file, buf, MAX_STRING-1); + char *buf=new(PointerFreeGC) char[MAX_STRING+1]; + ssize_t length=read(file, buf, MAX_STRING); if(length<=0) break; buf[length]=0; @@ -290,6 +313,28 @@ static void read_pipe(String& result, in } } +static void read_pipe(File_read_result& result, int file){ + char *buf=(char*)pa_malloc(MAX_STRING+1); + ssize_t bufsize = MAX_STRING; + + result.headers = 0; + result.length = 0; + result.str = 0; + result.success = false; + + while(true) { + ssize_t size=read(file, buf + result.length, bufsize - result.length); + if(size <= 0) + break; + result.length += size; + if(result.length >= bufsize){ + bufsize *= 2; + buf=(char*)pa_realloc(buf, bufsize+1); + } + result.str=buf; + } +} + #endif #ifndef DOXYGEN @@ -302,7 +347,7 @@ struct Append_env_pair_info { #endif }; #endif -///@test maybe here and at argv construction --- cstr(String::L_UNSPECIFIED +///@test maybe here and at argv construction --- untaint_cstr(String::L_AS_IS static void append_env_pair(HashStringString::key_type key, HashStringString::value_type value, Append_env_pair_info *info) { #ifdef WIN32 @@ -330,7 +375,7 @@ PA_exec_result pa_exec( #ifdef NO_PA_EXECS if(!forced_allow) - throw Exception("parser.runtime", + throw Exception(PARSER_RUNTIME, &file_spec, "parser execs are disabled [recompile parser without --disable-execs configure option]"); #endif @@ -339,18 +384,19 @@ PA_exec_result pa_exec( PROCESS_INFORMATION pi; HANDLE hInWrite, hOutRead, hErrRead; - const char* cmd=buildCommand(file_spec.cstr(String::L_FILE_SPEC), argv); + const char* script_spec_cstr=file_spec.taint_cstr(String::L_FILE_SPEC); + const char* cmd=buildCommand(script_spec_cstr, argv); char* env_cstr=0; if(env) { String::Body body; Append_env_pair_info info(body); - env->for_each(append_env_pair, &info); + env->for_each(append_env_pair, &info); env_cstr=info.body.cstrm(); for(char* replacer=env_cstr; *replacer; replacer++) if(*replacer=='\1') *replacer=0; } - if(DWORD error=CreateHiddenConsoleProcess(cmd, env_cstr, &pi, &hInWrite, &hOutRead, &hErrRead)) { + if(DWORD error=CreateHiddenConsoleProcess(cmd, script_spec_cstr, env_cstr, &pi, &hInWrite, &hOutRead, &hErrRead)) { char szErrorDesc[MAX_STRING]; const char* param="the file you tried to run"; size_t error_size=FormatMessage( @@ -359,8 +405,8 @@ PA_exec_result pa_exec( szErrorDesc, sizeof(szErrorDesc), (va_list *)¶m); if(error_size>3) // ".\r\n" szErrorDesc[error_size-3]=0; - - throw Exception(0, + + throw Exception("file.execute", &file_spec, "exec failed - %s (%u). Consider adding shbang line (#!x:\\interpreter\\command line)", error_size?szErrorDesc:"", error); @@ -373,7 +419,7 @@ PA_exec_result pa_exec( // without this char WriteFile(hInWrite, "\x1A", 1, &written_size, NULL); CloseHandle(hInWrite); - read_pipe(result.out, hOutRead, String::L_AS_IS); + read_pipe(result.out, hOutRead); CloseHandle(hOutRead); read_pipe(result.err, hErrRead, String::L_TAINTED); CloseHandle(hErrRead); @@ -381,16 +427,16 @@ PA_exec_result pa_exec( from http://www.apache.org/websrc/cvsweb.cgi/apache-1.3/src/main/util_script.c?rev=1.151&content-type=text/vnd.viewcvs-markup * We must close the handles to the new process and its main thread - * to prevent handle and memory leaks. + * to prevent handle and memory leaks. */ CloseHandle(pi.hProcess); - CloseHandle(pi.hThread); + CloseHandle(pi.hThread); } #else // execve needs non const - char* file_spec_cstr=file_spec.cstrm(String::L_FILE_SPEC); + char* file_spec_cstr=file_spec.taint_cstrm(String::L_FILE_SPEC); int pipe_write, pipe_read, pipe_err; @@ -407,11 +453,11 @@ from http://www.apache.org/websrc/cvsweb } #endif - char* argv_cstrs[1+10+1]={file_spec_cstr, 0}; + char* argv_cstrs[1+100+1]={file_spec_cstr, 0}; const int argv_size=argv.count(); const int argv_max=sizeof(argv_cstrs)/sizeof(argv_cstrs[0])-1-1; if(argv_size>argv_max) - throw Exception("parser.runtime", + throw Exception(PARSER_RUNTIME, &file_spec, "too many arguments (%d > max %d)", argv_size, argv_max); for(int i=0; i0) { // in child - if(in.length()) {// there is some in data + if(!in.is_empty()) {// there is some in data const char* in_cstr=in.cstr(); write(pipe_write, in_cstr, in.length()); } close(pipe_write); - read_pipe(result.out, pipe_read, String::L_AS_IS); + read_pipe(result.out, pipe_read); close(pipe_read); read_pipe(result.err, pipe_err, String::L_TAINTED); close(pipe_err); @@ -446,7 +492,7 @@ from http://www.apache.org/websrc/cvsweb result.status=get_exit_status(pid); // negative may mean "-errno[execl()]" } else { const char* str=strerror(errno); - throw Exception(0, + throw Exception("file.execute", &file_spec, "%s error: %s (%d)", pid<0?"fork":"pipe", str?str:"", errno); }