Commit | Line | Data |
---|---|---|
5a17353c DD |
1 | /* Utilities to execute a program in a subprocess (possibly linked by pipes |
2 | with other subprocesses), and wait for it. Generic Win32 specialization. | |
1e45deed | 3 | Copyright (C) 1996, 1997, 1998, 1999, 2000, 2001, 2003, 2004, 2005 |
5a17353c DD |
4 | Free Software Foundation, Inc. |
5 | ||
6 | This file is part of the libiberty library. | |
7 | Libiberty is free software; you can redistribute it and/or | |
8 | modify it under the terms of the GNU Library General Public | |
9 | License as published by the Free Software Foundation; either | |
10 | version 2 of the License, or (at your option) any later version. | |
11 | ||
12 | Libiberty is distributed in the hope that it will be useful, | |
13 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 | Library General Public License for more details. | |
16 | ||
17 | You should have received a copy of the GNU Library General Public | |
18 | License along with libiberty; see the file COPYING.LIB. If not, | |
979c05d3 NC |
19 | write to the Free Software Foundation, Inc., 51 Franklin Street - Fifth Floor, |
20 | Boston, MA 02110-1301, USA. */ | |
5a17353c DD |
21 | |
22 | #include "pex-common.h" | |
23 | ||
b109e79a ILT |
24 | #ifdef HAVE_STDLIB_H |
25 | #include <stdlib.h> | |
26 | #endif | |
5a17353c DD |
27 | #ifdef HAVE_STRING_H |
28 | #include <string.h> | |
29 | #endif | |
30 | #ifdef HAVE_UNISTD_H | |
31 | #include <unistd.h> | |
32 | #endif | |
33 | #ifdef HAVE_SYS_WAIT_H | |
34 | #include <sys/wait.h> | |
35 | #endif | |
36 | ||
37 | #include <process.h> | |
38 | #include <io.h> | |
39 | #include <fcntl.h> | |
40 | #include <signal.h> | |
b109e79a | 41 | #include <sys/stat.h> |
5a17353c DD |
42 | |
43 | /* mingw32 headers may not define the following. */ | |
44 | ||
45 | #ifndef _P_WAIT | |
46 | # define _P_WAIT 0 | |
47 | # define _P_NOWAIT 1 | |
48 | # define _P_OVERLAY 2 | |
49 | # define _P_NOWAITO 3 | |
50 | # define _P_DETACH 4 | |
51 | ||
52 | # define WAIT_CHILD 0 | |
53 | # define WAIT_GRANDCHILD 1 | |
54 | #endif | |
55 | ||
56 | /* This is a kludge to get around the Microsoft C spawn functions' propensity | |
57 | to remove the outermost set of double quotes from all arguments. */ | |
58 | ||
59 | static const char * const * | |
b109e79a | 60 | fix_argv (char * const *argvec) |
5a17353c | 61 | { |
b109e79a | 62 | char **argv; |
5a17353c | 63 | int i; |
b109e79a ILT |
64 | char *command0; |
65 | ||
66 | /* See whether we need to change anything. */ | |
67 | for (command0 = argvec[0]; *command0 != '\0'; command0++) | |
68 | if (*command0 == '/') | |
69 | break; | |
70 | if (*command0 == '\0') | |
71 | { | |
72 | for (i = 1; argvec[i] != NULL; i++) | |
73 | if (strpbrk (argvec[i], "\" \t") != NULL) | |
74 | break; | |
75 | ||
76 | if (argvec[i] == NULL) | |
77 | return (const char * const *) argvec; | |
78 | } | |
79 | ||
80 | for (i = 0; argvec[i] != NULL; i++) | |
81 | ; | |
abf6a75b | 82 | argv = XNEWVEC (char *, i + 1); |
b109e79a ILT |
83 | for (i = 0; argvec[i] != NULL; i++) |
84 | argv[i] = xstrdup (argvec[i]); | |
85 | argv[i] = NULL; | |
5a17353c | 86 | |
9f7d3e7d | 87 | /* Ensure that the executable pathname uses Win32 backslashes. This |
b109e79a ILT |
88 | is not necessary on NT, but on W9x, forward slashes causes |
89 | failure of spawn* and exec* functions (and probably any function | |
90 | that calls CreateProcess) *iff* the executable pathname (argv[0]) | |
91 | is a quoted string. And quoting is necessary in case a pathname | |
92 | contains embedded white space. You can't win. */ | |
93 | for (command0 = argv[0]; *command0 != '\0'; command0++) | |
ab8c7bb1 DD |
94 | if (*command0 == '/') |
95 | *command0 = '\\'; | |
b109e79a ILT |
96 | |
97 | for (i = 1; argv[i] != 0; i++) | |
5a17353c DD |
98 | { |
99 | int len, j; | |
100 | char *temp, *newtemp; | |
101 | ||
b109e79a | 102 | temp = argv[i]; |
5a17353c DD |
103 | len = strlen (temp); |
104 | for (j = 0; j < len; j++) | |
105 | { | |
106 | if (temp[j] == '"') | |
107 | { | |
abf6a75b | 108 | newtemp = XNEWVEC (char, len + 2); |
5a17353c DD |
109 | strncpy (newtemp, temp, j); |
110 | newtemp [j] = '\\'; | |
111 | strncpy (&newtemp [j+1], &temp [j], len-j); | |
112 | newtemp [len+1] = 0; | |
113 | temp = newtemp; | |
114 | len++; | |
115 | j++; | |
116 | } | |
117 | } | |
118 | ||
b109e79a ILT |
119 | if (argv[i] != temp) |
120 | { | |
121 | free (argv[i]); | |
122 | argv[i] = temp; | |
123 | } | |
124 | } | |
5a17353c | 125 | |
b109e79a | 126 | for (i = 0; argv[i] != 0; i++) |
5a17353c | 127 | { |
b109e79a | 128 | if (strpbrk (argv[i], " \t")) |
5a17353c DD |
129 | { |
130 | int len, trailing_backslash; | |
131 | char *temp; | |
132 | ||
b109e79a | 133 | len = strlen (argv[i]); |
5a17353c DD |
134 | trailing_backslash = 0; |
135 | ||
136 | /* There is an added complication when an arg with embedded white | |
137 | space ends in a backslash (such as in the case of -iprefix arg | |
138 | passed to cpp). The resulting quoted strings gets misinterpreted | |
139 | by the command interpreter -- it thinks that the ending quote | |
140 | is escaped by the trailing backslash and things get confused. | |
141 | We handle this case by escaping the trailing backslash, provided | |
142 | it was not escaped in the first place. */ | |
143 | if (len > 1 | |
b109e79a ILT |
144 | && argv[i][len-1] == '\\' |
145 | && argv[i][len-2] != '\\') | |
5a17353c DD |
146 | { |
147 | trailing_backslash = 1; | |
148 | ++len; /* to escape the final backslash. */ | |
149 | } | |
150 | ||
151 | len += 2; /* and for the enclosing quotes. */ | |
152 | ||
abf6a75b | 153 | temp = XNEWVEC (char, len + 1); |
5a17353c | 154 | temp[0] = '"'; |
b109e79a | 155 | strcpy (temp + 1, argv[i]); |
5a17353c | 156 | if (trailing_backslash) |
b109e79a ILT |
157 | temp[len - 2] = '\\'; |
158 | temp[len - 1] = '"'; | |
5a17353c DD |
159 | temp[len] = '\0'; |
160 | ||
b109e79a ILT |
161 | free (argv[i]); |
162 | argv[i] = temp; | |
5a17353c DD |
163 | } |
164 | } | |
165 | ||
b109e79a | 166 | return (const char * const *) argv; |
5a17353c DD |
167 | } |
168 | ||
b109e79a ILT |
169 | static int pex_win32_open_read (struct pex_obj *, const char *, int); |
170 | static int pex_win32_open_write (struct pex_obj *, const char *, int); | |
171 | static long pex_win32_exec_child (struct pex_obj *, int, const char *, | |
172 | char * const *, int, int, int, | |
173 | const char **, int *); | |
174 | static int pex_win32_close (struct pex_obj *, int); | |
175 | static int pex_win32_wait (struct pex_obj *, long, int *, | |
176 | struct pex_time *, int, const char **, int *); | |
177 | static int pex_win32_pipe (struct pex_obj *, int *, int); | |
178 | static FILE *pex_win32_fdopenr (struct pex_obj *, int, int); | |
179 | ||
180 | /* The list of functions we pass to the common routines. */ | |
181 | ||
182 | const struct pex_funcs funcs = | |
5a17353c | 183 | { |
b109e79a ILT |
184 | pex_win32_open_read, |
185 | pex_win32_open_write, | |
186 | pex_win32_exec_child, | |
187 | pex_win32_close, | |
188 | pex_win32_wait, | |
189 | pex_win32_pipe, | |
190 | pex_win32_fdopenr, | |
191 | NULL /* cleanup */ | |
192 | }; | |
193 | ||
194 | /* Return a newly initialized pex_obj structure. */ | |
195 | ||
196 | struct pex_obj * | |
197 | pex_init (int flags, const char *pname, const char *tempbase) | |
198 | { | |
199 | return pex_init_common (flags, pname, tempbase, &funcs); | |
200 | } | |
201 | ||
202 | /* Open a file for reading. */ | |
203 | ||
204 | static int | |
205 | pex_win32_open_read (struct pex_obj *obj ATTRIBUTE_UNUSED, const char *name, | |
206 | int binary) | |
207 | { | |
208 | return _open (name, _O_RDONLY | (binary ? _O_BINARY : _O_TEXT)); | |
209 | } | |
210 | ||
211 | /* Open a file for writing. */ | |
212 | ||
213 | static int | |
214 | pex_win32_open_write (struct pex_obj *obj ATTRIBUTE_UNUSED, const char *name, | |
215 | int binary) | |
216 | { | |
217 | /* Note that we can't use O_EXCL here because gcc may have already | |
218 | created the temporary file via make_temp_file. */ | |
219 | return _open (name, | |
220 | (_O_WRONLY | _O_CREAT | _O_TRUNC | |
221 | | (binary ? _O_BINARY : _O_TEXT)), | |
222 | _S_IREAD | _S_IWRITE); | |
223 | } | |
224 | ||
225 | /* Close a file. */ | |
226 | ||
227 | static int | |
228 | pex_win32_close (struct pex_obj *obj ATTRIBUTE_UNUSED, int fd) | |
229 | { | |
230 | return _close (fd); | |
231 | } | |
232 | ||
233 | /* Execute a child. */ | |
234 | ||
235 | static long | |
236 | pex_win32_exec_child (struct pex_obj *obj ATTRIBUTE_UNUSED, int flags, | |
237 | const char *executable, char * const * argv, | |
238 | int in, int out, int errdes, const char **errmsg, | |
239 | int *err) | |
240 | { | |
241 | int org_in, org_out, org_errdes; | |
242 | long pid; | |
243 | ||
244 | org_in = -1; | |
245 | org_out = -1; | |
246 | org_errdes = -1; | |
247 | ||
248 | if (in != STDIN_FILE_NO) | |
5a17353c | 249 | { |
b109e79a ILT |
250 | org_in = _dup (STDIN_FILE_NO); |
251 | if (org_in < 0) | |
252 | { | |
253 | *err = errno; | |
254 | *errmsg = "_dup"; | |
255 | return -1; | |
256 | } | |
257 | if (_dup2 (in, STDIN_FILE_NO) < 0) | |
258 | { | |
259 | *err = errno; | |
260 | *errmsg = "_dup2"; | |
261 | return -1; | |
262 | } | |
263 | if (_close (in) < 0) | |
5a17353c | 264 | { |
b109e79a ILT |
265 | *err = errno; |
266 | *errmsg = "_close"; | |
5a17353c DD |
267 | return -1; |
268 | } | |
5a17353c | 269 | } |
b109e79a ILT |
270 | |
271 | if (out != STDOUT_FILE_NO) | |
5a17353c | 272 | { |
b109e79a ILT |
273 | org_out = _dup (STDOUT_FILE_NO); |
274 | if (org_out < 0) | |
275 | { | |
276 | *err = errno; | |
277 | *errmsg = "_dup"; | |
278 | return -1; | |
279 | } | |
280 | if (_dup2 (out, STDOUT_FILE_NO) < 0) | |
281 | { | |
282 | *err = errno; | |
283 | *errmsg = "_dup2"; | |
284 | return -1; | |
285 | } | |
286 | if (_close (out) < 0) | |
287 | { | |
288 | *err = errno; | |
289 | *errmsg = "_close"; | |
290 | return -1; | |
291 | } | |
5a17353c DD |
292 | } |
293 | ||
b109e79a ILT |
294 | if (errdes != STDERR_FILE_NO |
295 | || (flags & PEX_STDERR_TO_STDOUT) != 0) | |
5a17353c | 296 | { |
b109e79a ILT |
297 | org_errdes = _dup (STDERR_FILE_NO); |
298 | if (org_errdes < 0) | |
299 | { | |
300 | *err = errno; | |
301 | *errmsg = "_dup"; | |
302 | return -1; | |
303 | } | |
304 | if (_dup2 ((flags & PEX_STDERR_TO_STDOUT) != 0 ? STDOUT_FILE_NO : errdes, | |
305 | STDERR_FILE_NO) < 0) | |
306 | { | |
307 | *err = errno; | |
308 | *errmsg = "_dup2"; | |
309 | return -1; | |
310 | } | |
311 | if (errdes != STDERR_FILE_NO) | |
312 | { | |
313 | if (_close (errdes) < 0) | |
314 | { | |
315 | *err = errno; | |
316 | *errmsg = "_close"; | |
317 | return -1; | |
318 | } | |
319 | } | |
5a17353c DD |
320 | } |
321 | ||
b109e79a ILT |
322 | pid = (((flags & PEX_SEARCH) != 0 ? _spawnvp : _spawnv) |
323 | (_P_NOWAIT, executable, fix_argv (argv))); | |
324 | ||
325 | if (pid == -1) | |
5a17353c | 326 | { |
b109e79a ILT |
327 | *err = errno; |
328 | *errmsg = ((flags & PEX_SEARCH) != 0) ? "_spawnvp" : "_spawnv"; | |
5a17353c DD |
329 | } |
330 | ||
b109e79a | 331 | if (in != STDIN_FILE_NO) |
5a17353c | 332 | { |
b109e79a ILT |
333 | if (_dup2 (org_in, STDIN_FILE_NO) < 0) |
334 | { | |
335 | *err = errno; | |
336 | *errmsg = "_dup2"; | |
337 | return -1; | |
338 | } | |
339 | if (_close (org_in) < 0) | |
340 | { | |
341 | *err = errno; | |
342 | *errmsg = "_close"; | |
343 | return -1; | |
344 | } | |
5a17353c DD |
345 | } |
346 | ||
b109e79a | 347 | if (out != STDOUT_FILE_NO) |
5a17353c | 348 | { |
b109e79a ILT |
349 | if (_dup2 (org_out, STDOUT_FILE_NO) < 0) |
350 | { | |
351 | *err = errno; | |
352 | *errmsg = "_dup2"; | |
353 | return -1; | |
354 | } | |
355 | if (_close (org_out) < 0) | |
356 | { | |
357 | *err = errno; | |
358 | *errmsg = "_close"; | |
359 | return -1; | |
360 | } | |
5a17353c DD |
361 | } |
362 | ||
b109e79a ILT |
363 | if (errdes != STDERR_FILE_NO |
364 | || (flags & PEX_STDERR_TO_STDOUT) != 0) | |
5a17353c | 365 | { |
b109e79a ILT |
366 | if (_dup2 (org_errdes, STDERR_FILE_NO) < 0) |
367 | { | |
368 | *err = errno; | |
369 | *errmsg = "_dup2"; | |
370 | return -1; | |
371 | } | |
372 | if (_close (org_errdes) < 0) | |
373 | { | |
374 | *err = errno; | |
375 | *errmsg = "_close"; | |
376 | return -1; | |
377 | } | |
5a17353c DD |
378 | } |
379 | ||
380 | return pid; | |
381 | } | |
382 | ||
b109e79a ILT |
383 | /* Wait for a child process to complete. MS CRTDLL doesn't return |
384 | enough information in status to decide if the child exited due to a | |
385 | signal or not, rather it simply returns an integer with the exit | |
386 | code of the child; eg., if the child exited with an abort() call | |
387 | and didn't have a handler for SIGABRT, it simply returns with | |
388 | status == 3. We fix the status code to conform to the usual WIF* | |
389 | macros. Note that WIFSIGNALED will never be true under CRTDLL. */ | |
390 | ||
391 | static int | |
392 | pex_win32_wait (struct pex_obj *obj ATTRIBUTE_UNUSED, long pid, | |
393 | int *status, struct pex_time *time, int done ATTRIBUTE_UNUSED, | |
394 | const char **errmsg, int *err) | |
5a17353c DD |
395 | { |
396 | int termstat; | |
397 | ||
b109e79a ILT |
398 | if (time != NULL) |
399 | memset (time, 0, sizeof *time); | |
400 | ||
401 | /* FIXME: If done is non-zero, we should probably try to kill the | |
402 | process. */ | |
403 | ||
404 | if (_cwait (&termstat, pid, WAIT_CHILD) < 0) | |
405 | { | |
406 | *err = errno; | |
407 | *errmsg = "_cwait"; | |
408 | return -1; | |
409 | } | |
5a17353c | 410 | |
b109e79a ILT |
411 | /* cwait returns the child process exit code in termstat. A value |
412 | of 3 indicates that the child caught a signal, but not which one. | |
413 | Since only SIGABRT, SIGFPE and SIGINT do anything, we report | |
414 | SIGABRT. */ | |
5a17353c | 415 | |
5a17353c DD |
416 | if (termstat == 3) |
417 | *status = SIGABRT; | |
418 | else | |
b109e79a | 419 | *status = ((termstat & 0xff) << 8); |
5a17353c | 420 | |
b109e79a ILT |
421 | return 0; |
422 | } | |
423 | ||
424 | /* Create a pipe. */ | |
425 | ||
426 | static int | |
427 | pex_win32_pipe (struct pex_obj *obj ATTRIBUTE_UNUSED, int *p, | |
428 | int binary) | |
429 | { | |
430 | return _pipe (p, 256, binary ? _O_BINARY : _O_TEXT); | |
431 | } | |
432 | ||
433 | /* Get a FILE pointer to read from a file descriptor. */ | |
434 | ||
435 | static FILE * | |
436 | pex_win32_fdopenr (struct pex_obj *obj ATTRIBUTE_UNUSED, int fd, | |
437 | int binary) | |
438 | { | |
439 | return fdopen (fd, binary ? "rb" : "r"); | |
5a17353c | 440 | } |