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