Annotation of parser3/src/main/pa_exec.C, revision 1.48.2.8
1.1 paf 1: /** @file
2: Parser: program executing for different OS-es.
3:
1.48.2.5 paf 4: Copyright(c) 2000,2001-2003 ArtLebedev Group(http://www.artlebedev.com)
1.25 paf 5: Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.1 paf 6:
1.18 paf 7: @todo setrlimit
1.1 paf 8: */
1.38 paf 9:
1.48.2.8! paf 10: static const char* IDENT_EXEC_C="$Date: 2003/02/04 10:57:16 $";
1.1 paf 11:
12: #include "pa_config_includes.h"
13:
1.19 paf 14: #include "pa_exec.h"
15: #include "pa_exception.h"
16: #include "pa_common.h"
1.48.2.3 paf 17: #include "pa_value_includes.h"
1.19 paf 18:
1.1 paf 19: #ifdef WIN32
20: # include <windows.h>
21: #else
22: # include <signal.h>
23: # include <sys/types.h>
24: # include <sys/wait.h>
25: #endif
26:
27: #ifdef WIN32
28:
29: /// this func from http://www.ccas.ru/~posp/popov/spawn.htm
1.36 paf 30: static DWORD CreateHiddenConsoleProcess(LPCTSTR szCmdLine,
1.1 paf 31: char *szEnv,
32: PROCESS_INFORMATION* ppi,
33: LPHANDLE phInWrite,
34: LPHANDLE phOutRead,
1.9 parser 35: LPHANDLE phErrRead)
1.1 paf 36: {
1.36 paf 37: DWORD result=0;
38: BOOL fCreated;
1.1 paf 39: STARTUPINFO si;
40: SECURITY_ATTRIBUTES sa={0};
41: HANDLE hInRead;
42: HANDLE hOutWrite;
43: HANDLE hErrWrite;
44:
45: // Create pipes
46: // initialize security attributes for handle inheritance (for WinNT)
1.9 parser 47: sa.nLength=sizeof(sa);
48: sa.bInheritHandle=TRUE;
49: sa.lpSecurityDescriptor=NULL;
1.1 paf 50:
51: // create STDIN pipe
1.9 parser 52: if(!CreatePipe(&hInRead, phInWrite, &sa, 0))
1.1 paf 53: goto error;
54:
55: // create STDOUT pipe
1.9 parser 56: if(!CreatePipe(phOutRead, &hOutWrite, &sa, 0))
1.1 paf 57: goto error;
58:
59: // create STDERR pipe
1.9 parser 60: if(!CreatePipe(phErrRead, &hErrWrite, &sa, 0))
1.1 paf 61: goto error;
62:
63: // process startup information
1.9 parser 64: memset(&si, 0, sizeof(si));
65: si.cb=sizeof(si);
66: si.dwFlags=STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
1.1 paf 67: // child process' console must be hidden for Win95 compatibility
1.9 parser 68: si.wShowWindow=SW_HIDE;
1.1 paf 69: // assign "other" sides of pipes
1.9 parser 70: si.hStdInput=hInRead;
71: si.hStdOutput=hOutWrite;
72: si.hStdError=hErrWrite;
1.1 paf 73:
1.31 paf 74: // calculating script's directory
75: char dir[MAX_STRING];
76: strncpy(dir, szCmdLine, MAX_STRING-1); dir[MAX_STRING-1]=0;
1.33 paf 77: lsplit(dir,' '); // trim arguments
1.31 paf 78: rsplit(dir,'/'); rsplit(dir,'\\'); // trim filename
79: chdir(dir);
80:
1.1 paf 81: // Create a child process (suspended)
1.9 parser 82: fCreated=CreateProcess(NULL,
1.26 paf 83: (LPTSTR)szCmdLine,
1.1 paf 84: NULL,
85: NULL,
86: TRUE,
1.30 paf 87: CREATE_NO_WINDOW,
1.1 paf 88: szEnv,
1.31 paf 89: dir,
1.1 paf 90: &si,
1.9 parser 91: ppi);
1.36 paf 92: if(!fCreated)
93: result=GetLastError();
1.1 paf 94:
1.9 parser 95: CloseHandle(hInRead);
96: CloseHandle(hOutWrite);
97: CloseHandle(hErrWrite);
1.1 paf 98:
1.9 parser 99: if(!fCreated)
1.1 paf 100: goto error;
101:
1.36 paf 102: return result;
1.1 paf 103:
104: error:
1.36 paf 105: if(!result/*yet*/)
106: result=GetLastError(); // get it
107:
1.9 parser 108: CloseHandle(*phInWrite);
109: CloseHandle(*phOutRead);
110: CloseHandle(*phErrRead);
1.1 paf 111:
1.36 paf 112: return result;
1.1 paf 113: }
114:
1.48.2.5 paf 115: static void read_pipe(Pool& pool, StringPtr result, HANDLE hOutRead, const char* file_spec,
1.35 paf 116: String::Untaint_lang lang){
1.1 paf 117: while(true) {
1.48.2.2 paf 118: char *buf=new(pool) char[MAX_STRING];
1.1 paf 119: unsigned long size;
1.30 paf 120: if(!ReadFile(hOutRead, buf, MAX_STRING, &size, NULL) || !size)
1.1 paf 121: break;
1.48.2.2 paf 122: result->APPEND(buf, size, lang, file_spec, 0);
1.1 paf 123: }
124: }
125:
1.48.2.4 paf 126: static CharPtr buildCommand(CharPtr file_spec_cstr, Array<StringPtr>& argv) {
1.48.2.2 paf 127: CharPtr result=file_spec_cstr;
1.4 paf 128: if(FILE *f=fopen(file_spec_cstr, "r")) {
1.48.2.2 paf 129: try {
1.1 paf 130: char buf[MAX_STRING];
131: size_t size=fread(buf, 1, MAX_STRING-1, f);
132: if(size>2) {
133: buf[size]=0;
134: if(strncmp(buf, "#!", 2)==0) {
1.48.2.5 paf 135: const char* begin=buf+2;
1.4 paf 136: if(*begin==' ') // alx: were an old magic for some linux-es
137: begin++;
138: if(char *end=strchr(begin, '\n')) {
1.48.2.2 paf 139: String string(begin, end-begin);
1.1 paf 140: string << " " << file_spec_cstr;
1.8 parser 141: result=string.cstr();
1.1 paf 142: }
143: }
144: }
1.48.2.2 paf 145: } catch(...) {
146: fclose(f);
147: rethrow;
148: }
1.1 paf 149: fclose(f);
150: }
1.48.2.2 paf 151: { // appending argv
152: String string(result);
153: for(int i=0; i<argv.count(); i++) {
154: string << " ";
155: string << *argv[i];
1.8 parser 156: }
157:
1.48.2.2 paf 158: result=string.cstr();
1.8 parser 159: }
160:
161: return result;
1.1 paf 162: }
163:
164: #else
165:
1.48.2.5 paf 166: static pid_t execve_piped(const char* file_spec_cstr,
1.27 paf 167: char * const argv[], char * const env[],
168: int *pipe_in, int *pipe_out, int *pipe_err) {
1.46 paf 169: pid_t pid;
1.1 paf 170: int in_fds[2];
171: int out_fds[2];
172: int err_fds[2];
173: int save_errno;
174:
175: if(pipe_in && pipe(in_fds)<0) {
176: save_errno=errno;
177: errno=save_errno;
178: return 0;
179: }
180:
181: if(pipe_out && pipe(out_fds)<0) {
182: save_errno=errno;
183: if(pipe_in) {
184: close(in_fds[0]); close(in_fds[1]);
185: }
186: errno=save_errno;
187: return 0;
188: }
189:
190: if(pipe_err && pipe(err_fds)<0) {
191: save_errno=errno;
192: if(pipe_in) {
193: close(in_fds[0]); close(in_fds[1]);
194: }
195: if(pipe_out) {
196: close(out_fds[0]); close(out_fds[1]);
197: }
198: errno=save_errno;
199: return 0;
200: }
201:
202: if((pid=fork())<0) {
203: save_errno=errno;
204: if(pipe_in) {
205: close(in_fds[0]); close(in_fds[1]);
206: }
207: if(pipe_out) {
208: close(out_fds[0]); close(out_fds[1]);
209: }
210: if(pipe_err) {
211: close(err_fds[0]); close(err_fds[1]);
212: }
213: errno=save_errno;
1.46 paf 214: return -1;
1.1 paf 215: }
216:
217: if(!pid) {
218: /* Child process */
219:
220: if(pipe_out) {
221: close(out_fds[0]);
222: dup2(out_fds[1], STDOUT_FILENO);
223: close(out_fds[1]);
224: }
225:
226: if(pipe_in) {
227: close(in_fds[1]);
228: dup2(in_fds[0], STDIN_FILENO);
229: close(in_fds[0]);
230: }
231:
232: if(pipe_err) {
233: close(err_fds[0]);
234: dup2(err_fds[1], STDERR_FILENO);
235: close(err_fds[1]);
236: }
237:
1.42 paf 238: /* grabbed this from Apache source: */
239: /* HP-UX SIGCHLD fix goes here, if someone will remind me what it is... */
1.1 paf 240: signal(SIGCHLD, SIG_DFL); /* Was that it? */
241:
1.31 paf 242: // chdir to script's directory
243: char dir[MAX_STRING];
244: strncpy(dir, file_spec_cstr, MAX_STRING-1); dir[MAX_STRING-1]=0;
245: rsplit(dir,'/'); // trim filename
246: chdir(dir);
247:
248: // execute
249: execve(file_spec_cstr, argv, env);
1.1 paf 250: exit(-errno);
251: }
252:
253: /* Parent process */
254:
255: if(pipe_out) {
256: close(out_fds[1]);
257: *pipe_out=out_fds[0];
258: }
259:
260: if(pipe_in) {
261: close(in_fds[0]);
262: *pipe_in=in_fds[1];
263: }
264:
265: if(pipe_err) {
266: close(err_fds[1]);
267: *pipe_err=err_fds[0];
268: }
269:
270: return pid;
271: }
272:
273: static int get_exit_status(int pid) {
274: int status;
275: if(!waitpid(pid, &status, 0))
276: return -1;
277: return WIFEXITED(status) ?
278: WEXITSTATUS(status) : -2;
279: }
280:
1.48.2.8! paf 281: static void read_pipe(Pool& pool, StringPtr result, int file, const char* file_spec, String::Untaint_lang lang){
1.1 paf 282: while(true) {
1.48.2.8! paf 283: char *buf=new(pool) char[MAX_STRING];
1.40 paf 284: ssize_t size=read(file, buf, MAX_STRING);
285: if(size<=0)
1.1 paf 286: break;
1.48.2.8! paf 287: result->APPEND(buf, size, lang, file_spec, 0);
1.1 paf 288: }
289: }
290:
291: #endif
292:
1.48.2.8! paf 293: #ifndef DOXYGEN
! 294: struct Append_env_pair_info {
1.1 paf 295: #ifdef WIN32
1.48.2.8! paf 296: String *string;
1.1 paf 297: #else
1.48.2.8! paf 298: Pool *pool;
! 299: char **env_ref;
1.48.2.2 paf 300: #endif
1.48.2.8! paf 301: };
! 302: #endif
! 303: ///@test maybe here and at argv construction --- cstr(String::UL_UNSPECIFIED
! 304: static void append_env_pair(HashStringString::key_type key, HashStringString::value_type value,
! 305: Append_env_pair_info *info) {
1.48.2.2 paf 306: #ifdef WIN32
1.48.2.8! paf 307: *info->string << key << "=" << value;
! 308: info->string->APPEND_AS_IS("", 1, 0, 0); // zero byte
1.48.2.2 paf 309: #else
310: String string;
1.48.2.1 paf 311: string << key << "=" << value;
1.1 paf 312:
1.48.2.8! paf 313: *(info->env_ref++)=string.cstr(*info->pool, String::UL_UNSPECIFIED);
1.1 paf 314: #endif
315: }
1.31 paf 316:
1.48.2.2 paf 317: /// @todonow unix part to smart_ptr
318: int pa_exec(Pool& pool,
319: bool forced_allow,
1.48.2.4 paf 320: StringPtr file_spec,
1.48.2.2 paf 321: HashStringString& env,
1.48.2.4 paf 322: Array<StringPtr>& argv,
323: StringPtr in, StringPtr out, StringPtr err) {
1.28 paf 324: #ifdef NO_PA_EXECS
1.32 paf 325: if(!forced_allow)
1.34 paf 326: throw Exception("parser.runtime",
1.48.2.2 paf 327: file_spec,
1.32 paf 328: "parser execs are disabled [recompile parser without --disable-execs configure option]");
329: #endif
1.28 paf 330:
1.48.2.8! paf 331: char* file_spec_cstr=file_spec->cstr(pool, String::UL_FILE_SPEC);
1.1 paf 332: #ifdef WIN32
333:
334: PROCESS_INFORMATION pi;
335: HANDLE hInWrite, hOutRead, hErrRead;
1.48.2.2 paf 336: CharPtr cmd=buildCommand(file_spec_cstr, argv);
337: CharPtr env_cstr(0);
338: { // env
339: String string;
340: env.for_each(append_env_pair, &string);
1.45 paf 341: env_cstr=string.cstr(String::UL_UNSPECIFIED);
1.1 paf 342: }
1.36 paf 343: if(DWORD error=CreateHiddenConsoleProcess(cmd, env_cstr, &pi, &hInWrite, &hOutRead, &hErrRead)) {
344: char szErrorDesc[MAX_STRING];
1.48.2.6 paf 345: const char *param="the file you tried to run";
1.36 paf 346: size_t error_size=FormatMessage(
347: FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_ARGUMENT_ARRAY , NULL, error,
348: MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
1.48.2.7 paf 349: szErrorDesc, sizeof(szErrorDesc), (va_list *)¶m);
1.36 paf 350: if(error_size>3) // ".\r\n"
351: szErrorDesc[error_size-3]=0;
352:
353: throw Exception(0,
1.48.2.2 paf 354: file_spec,
1.36 paf 355: "exec failed - %s (%ld). Consider adding shbang line (#!x:\\interpreter\\command line)",
356: error_size?szErrorDesc:"<unknown>", (long)error);
357: } else {
1.48.2.2 paf 358: CharPtr in_cstr=in->cstr();
1.5 paf 359: DWORD written_size;
1.48.2.2 paf 360: WriteFile(hInWrite, in_cstr, in->size(), &written_size, NULL);
1.5 paf 361: // EOF for stupid text reads
362: // normally they should read CONTENT_LENGTH bytes,
363: // without this char
364: WriteFile(hInWrite, "\x1A", 1, &written_size, NULL);
1.9 parser 365: CloseHandle(hInWrite);
1.48.2.2 paf 366: read_pipe(pool, out, hOutRead, file_spec_cstr, String::UL_AS_IS);
1.9 parser 367: CloseHandle(hOutRead);
1.48.2.2 paf 368: read_pipe(pool, err, hErrRead, file_spec_cstr, String::UL_TAINTED);
1.9 parser 369: CloseHandle(hErrRead);
1.1 paf 370: /*
371: 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
372:
373: * We must close the handles to the new process and its main thread
374: * to prevent handle and memory leaks.
375: */
376: CloseHandle(pi.hProcess);
377: CloseHandle(pi.hThread);
378: }
379:
380: #else
381:
382: int pipe_write, pipe_read, pipe_err;
1.31 paf 383:
1.47 paf 384: #ifdef PA_SAFE_MODE
1.32 paf 385: if(!forced_allow) {
386: struct stat finfo;
387: if(stat(file_spec_cstr, &finfo)!=0)
1.34 paf 388: throw Exception("file.missing",
1.48.2.8! paf 389: file_spec,
1.32 paf 390: "stat failed: %s (%d), actual filename '%s'",
391: strerror(errno), errno, file_spec_cstr);
1.31 paf 392:
1.47 paf 393: if(finfo.st_uid/*foreign?*/!=geteuid()
394: && finfo.st_gid/*foreign?*/!=getegid())
395: throw Exception("parser.runtime",
1.48.2.8! paf 396: file_spec,
1.47 paf 397: "parser is in safe mode: executing files of foreign group and user disabled [recompile parser with --disable-safe-mode configure option], actual filename '%s'",
1.32 paf 398: file_spec_cstr);
399: }
1.31 paf 400: #endif
401:
1.48.2.8! paf 402: char * argv_cstrs[1+10+1]={file_spec_cstr, 0};
! 403: const int argv_size=argv.count();
! 404: const int argv_max=sizeof(argv_cstrs)/sizeof(argv_cstrs[0])-1-1;
! 405: if(argv_size>argv_max)
! 406: throw Exception("parser.runtime",
! 407: file_spec,
! 408: "too many arguments (%d > max %d)", argv_size, argv_max);
! 409: for(int i=0; i<argv_size; i++)
! 410: argv_cstrs[1+i]=argv[i]->cstr(pool);
! 411: argv_cstrs[1+argv_size]=0;
! 412:
! 413: char **env_cstrs=new(pool) char *[env.count()+1/*0*/];
! 414: {
! 415: Append_env_pair_info info={&pool, env_cstrs};
! 416: env.for_each(append_env_pair, &info);
! 417: *info.env_ref=0;
1.1 paf 418: }
1.31 paf 419:
1.46 paf 420: pid_t pid=execve_piped(
1.1 paf 421: file_spec_cstr,
1.27 paf 422: argv_cstrs, env_cstrs,
1.22 paf 423: &pipe_write, &pipe_read, &pipe_err);
1.46 paf 424: if(pid>0) {
1.21 paf 425: // in child
1.48.2.8! paf 426: if(in && in->size()) {// there is some in data
! 427: CharPtr in_cstr=in->cstr();
! 428: write(pipe_write, in_cstr, in->size());
! 429: }
1.1 paf 430: close(pipe_write);
1.48.2.8! paf 431: read_pipe(pool, out, pipe_read, file_spec_cstr, String::UL_AS_IS);
1.5 paf 432: close(pipe_read);
1.48.2.8! paf 433: read_pipe(pool, err, pipe_err, file_spec_cstr, String::UL_TAINTED);
1.1 paf 434: close(pipe_err);
435:
436: return get_exit_status(pid); // negative may mean "-errno[execl()]"
437: } else
1.34 paf 438: throw Exception(0,
1.48.2.8! paf 439: file_spec,
1.46 paf 440: "%s error: %s (%d)", pid<0?"fork":"pipe", strerror(errno), errno);
1.1 paf 441: #endif
442:
443: return 0;
444: }
E-mail:
![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to pa_exec.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / parser3 / src / main