![[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|
Return to pa_exec.C CVS log | Up to [parser3project] / parser3 / src / main |
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.27 ! paf 7: $Id: pa_exec.C,v 1.26 2002/02/26 13:45:12 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:
73: // Create a child process (suspended)
1.9 parser 74: fCreated=CreateProcess(NULL,
1.26 paf 75: (LPTSTR)szCmdLine,
1.1 paf 76: NULL,
77: NULL,
78: TRUE,
1.9 parser 79: 0, //todo CREATE_NO_WINDOW,
1.1 paf 80: szEnv,
81: NULL,
82: &si,
1.9 parser 83: ppi);
1.1 paf 84:
1.9 parser 85: CloseHandle(hInRead);
86: CloseHandle(hOutWrite);
87: CloseHandle(hErrWrite);
1.1 paf 88:
1.9 parser 89: if(!fCreated)
1.1 paf 90: goto error;
91:
92: return TRUE;
93:
94: error:
1.9 parser 95: CloseHandle(*phInWrite);
96: CloseHandle(*phOutRead);
97: CloseHandle(*phErrRead);
1.1 paf 98:
99: return FALSE;
100: }
101:
1.2 paf 102: static void read_pipe(String& result, HANDLE hOutRead, const char *file_spec){
1.1 paf 103: while(true) {
104: char *buf=(char *)result.pool().malloc(MAX_STRING);
105: unsigned long size;
1.9 parser 106: ReadFile(hOutRead, buf, MAX_STRING, &size, NULL);
1.1 paf 107: if(!size)
108: break;
1.3 paf 109: result.APPEND_AS_IS(buf, size, file_spec, 0);
1.1 paf 110: }
111: }
112:
113:
114: static const char *buildCommand(Pool& pool,
115: const String& origin_string,
116: const char *file_spec_cstr, const Array *argv) {
1.8 parser 117: const char *result=file_spec_cstr;
1.4 paf 118: if(FILE *f=fopen(file_spec_cstr, "r")) {
1.1 paf 119: char buf[MAX_STRING];
120: size_t size=fread(buf, 1, MAX_STRING-1, f);
121: if(size>2) {
122: buf[size]=0;
123: if(strncmp(buf, "#!", 2)==0) {
1.4 paf 124: const char *begin=buf+2;
125: if(*begin==' ') // alx: were an old magic for some linux-es
126: begin++;
127: if(char *end=strchr(begin, '\n')) {
1.1 paf 128: String string(pool);
1.4 paf 129: string.APPEND_AS_IS(begin, end-begin,
1.1 paf 130: origin_string.origin().file, 0);
131: string << " " << file_spec_cstr;
132: if(argv)
133: for(int i=0; i<argv->size(); i++)
1.16 paf 134: string << argv->get_string(i)->cstr();
1.8 parser 135: result=string.cstr();
1.1 paf 136: }
137: }
138: }
139: fclose(f);
140: }
1.8 parser 141: if(argv) {
142: String buf(pool);
143: buf << result;
144: for(int i=0; i<argv->size(); i++) {
145: buf << " ";
146: buf << *argv->get_string(i);
147: }
148:
1.16 paf 149: result=buf.cstr();
1.8 parser 150: }
151:
152: return result;
1.1 paf 153: }
154:
155: #else
156:
1.27 ! paf 157: static int execve_piped(const char *path,
! 158: char * const argv[], char * const env[],
! 159: int *pipe_in, int *pipe_out, int *pipe_err) {
1.1 paf 160: int pid;
161: int in_fds[2];
162: int out_fds[2];
163: int err_fds[2];
164: int save_errno;
165:
166: if(pipe_in && pipe(in_fds)<0) {
167: save_errno=errno;
168: errno=save_errno;
169: return 0;
170: }
171:
172: if(pipe_out && pipe(out_fds)<0) {
173: save_errno=errno;
174: if(pipe_in) {
175: close(in_fds[0]); close(in_fds[1]);
176: }
177: errno=save_errno;
178: return 0;
179: }
180:
181: if(pipe_err && pipe(err_fds)<0) {
182: save_errno=errno;
183: if(pipe_in) {
184: close(in_fds[0]); close(in_fds[1]);
185: }
186: if(pipe_out) {
187: close(out_fds[0]); close(out_fds[1]);
188: }
189: errno=save_errno;
190: return 0;
191: }
192:
193: if((pid=fork())<0) {
194: save_errno=errno;
195: if(pipe_in) {
196: close(in_fds[0]); close(in_fds[1]);
197: }
198: if(pipe_out) {
199: close(out_fds[0]); close(out_fds[1]);
200: }
201: if(pipe_err) {
202: close(err_fds[0]); close(err_fds[1]);
203: }
204: errno=save_errno;
205: return 0;
206: }
207:
208: if(!pid) {
209: /* Child process */
210:
211: if(pipe_out) {
212: close(out_fds[0]);
213: dup2(out_fds[1], STDOUT_FILENO);
214: close(out_fds[1]);
215: }
216:
217: if(pipe_in) {
218: close(in_fds[1]);
219: dup2(in_fds[0], STDIN_FILENO);
220: close(in_fds[0]);
221: }
222:
223: if(pipe_err) {
224: close(err_fds[0]);
225: dup2(err_fds[1], STDERR_FILENO);
226: close(err_fds[1]);
227: }
228:
229: /* HP-UX SIGCHLD fix goes here, if someone will remind me what it is... */
230: signal(SIGCHLD, SIG_DFL); /* Was that it? */
231:
1.27 ! paf 232: execve(path, argv, env);
1.1 paf 233: exit(-errno);
234: }
235:
236: /* Parent process */
237:
238: if(pipe_out) {
239: close(out_fds[1]);
240: *pipe_out=out_fds[0];
241: }
242:
243: if(pipe_in) {
244: close(in_fds[0]);
245: *pipe_in=in_fds[1];
246: }
247:
248: if(pipe_err) {
249: close(err_fds[1]);
250: *pipe_err=err_fds[0];
251: }
252:
253: return pid;
254: }
255:
256: static int get_exit_status(int pid) {
257: int status;
258: if(!waitpid(pid, &status, 0))
259: return -1;
260: return WIFEXITED(status) ?
261: WEXITSTATUS(status) : -2;
262: }
263:
1.2 paf 264: static void read_pipe(String& result, int file, const char *file_spec){
1.1 paf 265: while(true) {
1.2 paf 266: char *buf=(char *)result.pool().malloc(MAX_STRING);
1.1 paf 267: size_t size=read(file, buf, MAX_STRING);
268: if(!size)
269: break;
1.3 paf 270: result.APPEND_AS_IS(buf, size, file_spec, 0);
1.1 paf 271: }
272: }
273:
274: #endif
275:
276: static void append_env_pair(const Hash::Key& key, Hash::Val *value, void *info) {
277: #ifdef WIN32
278: String& string=*static_cast<String *>(info);
279:
280: string << key << "=" << *static_cast<String *>(value);
1.3 paf 281: string.APPEND_AS_IS("", 1, 0, 0); // zero byte
1.1 paf 282: #else
283: String string(key.pool());
284: string << key << "=" << *static_cast<String *>(value);
285:
1.2 paf 286: char ***env_ref=static_cast<char ***>(info);
287: **env_ref=string.cstr(); (*env_ref)++;
1.1 paf 288: #endif
289: }
290: int pa_exec(const String& file_spec,
291: const Hash *env,
292: const Array *argv,
293: const String& in, String& out, String& err) {
294: Pool& pool=file_spec.pool();
295:
296: #ifdef WIN32
297:
298: char pwd[MAX_STRING];
299: GetCurrentDirectory(sizeof(pwd), pwd);
1.12 parser 300: char *dir=file_spec.cstr(String::UL_FILE_SPEC);
1.1 paf 301: rsplit(dir, '/'); SetCurrentDirectory(dir);
302:
303: PROCESS_INFORMATION pi;
304: HANDLE hInWrite, hOutRead, hErrRead;
1.12 parser 305: char *file_spec_cstr=file_spec.cstr(String::UL_FILE_SPEC);
1.1 paf 306: const char *cmd=buildCommand(file_spec.pool(), file_spec, file_spec_cstr, argv);
307: char *env_cstr=0;
308: if(env) {
309: String string(env->pool());
310: env->for_each(append_env_pair, &string);
1.16 paf 311: env_cstr=string.cstr();
1.1 paf 312: }
1.9 parser 313: if(CreateHiddenConsoleProcess(cmd, env_cstr, &pi, &hInWrite, &hOutRead, &hErrRead)) {
1.1 paf 314: SetCurrentDirectory(pwd);
315:
1.16 paf 316: const char *in_cstr=in.cstr();
1.5 paf 317: DWORD written_size;
318: WriteFile(hInWrite, in_cstr, in.size(), &written_size, NULL);
319: // EOF for stupid text reads
320: // normally they should read CONTENT_LENGTH bytes,
321: // without this char
322: WriteFile(hInWrite, "\x1A", 1, &written_size, NULL);
1.9 parser 323: CloseHandle(hInWrite);
1.1 paf 324: read_pipe(out, hOutRead, file_spec_cstr);
1.9 parser 325: CloseHandle(hOutRead);
1.5 paf 326: read_pipe(err, hErrRead, file_spec_cstr);
1.9 parser 327: CloseHandle(hErrRead);
1.1 paf 328: /*
329: 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
330:
331: * We must close the handles to the new process and its main thread
332: * to prevent handle and memory leaks.
333: */
334: CloseHandle(pi.hProcess);
335: CloseHandle(pi.hThread);
336: } else {
337: SetCurrentDirectory(pwd);
338:
339: DWORD error=GetLastError();
340: char szErrorDesc[MAX_STRING];
341: FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, error,
342: MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
343: szErrorDesc, sizeof(szErrorDesc), NULL);
344: size_t error_size=strlen(szErrorDesc);
345: if(error_size>3) // ".\r\n"
346: szErrorDesc[error_size-3]=0;
347:
1.15 parser 348: throw Exception(0, 0,
1.1 paf 349: &file_spec,
1.26 paf 350: "(real command line=\"%s\") exec failed - %s (%ld)",
1.11 parser 351: cmd,
1.10 parser 352: szErrorDesc, (long)error);
1.1 paf 353: }
354:
355: #else
356:
357: int pipe_write, pipe_read, pipe_err;
1.27 ! paf 358: char *file_spec_cstr=file_spec.cstr(String::UL_FILE_SPEC);
! 359: char *argv_cstrs[1+10+1]={file_spec_cstr, 0};
1.1 paf 360: if(argv) {
1.20 paf 361: const int argv_size=argv->size();
1.27 ! paf 362: const int argv_max=sizeof(argv_cstrs)/sizeof(argv_cstrs[0])-1-1;
1.20 paf 363: if(argv_size>argv_max)
364: throw Exception(0, 0,
365: &file_spec,
366: "too many arguments (%d > max %d)", argv_size, argv_max);
367: for(int i=0; i<argv_size; i++)
1.27 ! paf 368: argv_cstrs[1+i]=argv->get_string(i)->cstr();
! 369: argv_cstrs[1+argv_size]=0;
1.1 paf 370: }
1.2 paf 371: char **env_cstrs=0;
1.1 paf 372: if(env) {
1.2 paf 373: env_cstrs=
1.1 paf 374: (char **)env->pool().malloc(sizeof(char *)*(env->size()+1/*0*/));
1.2 paf 375: char **env_ref=env_cstrs;
376: env->for_each(append_env_pair, &env_ref);
377: *env_ref=0;
1.1 paf 378: }
1.21 paf 379: const char *pwd=getcwd(NULL, 0);
380: char dir[MAX_STRING];
381: strncpy(dir, file_spec_cstr, MAX_STRING);
382: rsplit(dir,'/'); // trim filename
383: chdir(dir);
1.27 ! paf 384: int pid=execve_piped(
1.1 paf 385: file_spec_cstr,
1.27 ! paf 386: argv_cstrs, env_cstrs,
1.22 paf 387: &pipe_write, &pipe_read, &pipe_err);
1.21 paf 388: if (pwd)
389: chdir(pwd);
390: if(pid) {
391: // in child
1.16 paf 392: const char *in_cstr=in.cstr();
1.1 paf 393: write(pipe_write, in_cstr, in.size());
394: close(pipe_write);
395: read_pipe(out, pipe_read, file_spec_cstr);
1.5 paf 396: close(pipe_read);
1.1 paf 397: read_pipe(err, pipe_err, file_spec_cstr);
398: close(pipe_err);
399:
400: return get_exit_status(pid); // negative may mean "-errno[execl()]"
401: } else
1.15 parser 402: throw Exception(0, 0,
1.2 paf 403: &file_spec,
1.1 paf 404: "pipe error");
405: #endif
406:
407: return 0;
408: }