| 1 | /* Utilities to execute a program in a subprocess (possibly linked by pipes |
| 2 | with other subprocesses), and wait for it. Generic Win32 specialization. |
| 3 | Copyright (C) 1996, 1997, 1998, 1999, 2000, 2001, 2003, 2004, 2005 |
| 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, |
| 19 | write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, |
| 20 | Boston, MA 02111-1307, USA. */ |
| 21 | |
| 22 | #include "pex-common.h" |
| 23 | |
| 24 | #ifdef HAVE_STDLIB_H |
| 25 | #include <stdlib.h> |
| 26 | #endif |
| 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> |
| 41 | #include <sys/stat.h> |
| 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 * |
| 60 | fix_argv (char * const *argvec) |
| 61 | { |
| 62 | char **argv; |
| 63 | int i; |
| 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 | ; |
| 82 | argv = xmalloc ((i + 1) * sizeof (char *)); |
| 83 | for (i = 0; argvec[i] != NULL; i++) |
| 84 | argv[i] = xstrdup (argvec[i]); |
| 85 | argv[i] = NULL; |
| 86 | |
| 87 | /* Ensure that the executable pathname uses Win32 backslashes. This |
| 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++) |
| 94 | if (*command0 == '/') |
| 95 | *command0 = '\\'; |
| 96 | |
| 97 | for (i = 1; argv[i] != 0; i++) |
| 98 | { |
| 99 | int len, j; |
| 100 | char *temp, *newtemp; |
| 101 | |
| 102 | temp = argv[i]; |
| 103 | len = strlen (temp); |
| 104 | for (j = 0; j < len; j++) |
| 105 | { |
| 106 | if (temp[j] == '"') |
| 107 | { |
| 108 | newtemp = xmalloc (len + 2); |
| 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 | |
| 119 | if (argv[i] != temp) |
| 120 | { |
| 121 | free (argv[i]); |
| 122 | argv[i] = temp; |
| 123 | } |
| 124 | } |
| 125 | |
| 126 | for (i = 0; argv[i] != 0; i++) |
| 127 | { |
| 128 | if (strpbrk (argv[i], " \t")) |
| 129 | { |
| 130 | int len, trailing_backslash; |
| 131 | char *temp; |
| 132 | |
| 133 | len = strlen (argv[i]); |
| 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 |
| 144 | && argv[i][len-1] == '\\' |
| 145 | && argv[i][len-2] != '\\') |
| 146 | { |
| 147 | trailing_backslash = 1; |
| 148 | ++len; /* to escape the final backslash. */ |
| 149 | } |
| 150 | |
| 151 | len += 2; /* and for the enclosing quotes. */ |
| 152 | |
| 153 | temp = xmalloc (len + 1); |
| 154 | temp[0] = '"'; |
| 155 | strcpy (temp + 1, argv[i]); |
| 156 | if (trailing_backslash) |
| 157 | temp[len - 2] = '\\'; |
| 158 | temp[len - 1] = '"'; |
| 159 | temp[len] = '\0'; |
| 160 | |
| 161 | free (argv[i]); |
| 162 | argv[i] = temp; |
| 163 | } |
| 164 | } |
| 165 | |
| 166 | return (const char * const *) argv; |
| 167 | } |
| 168 | |
| 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 = |
| 183 | { |
| 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) |
| 249 | { |
| 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) |
| 264 | { |
| 265 | *err = errno; |
| 266 | *errmsg = "_close"; |
| 267 | return -1; |
| 268 | } |
| 269 | } |
| 270 | |
| 271 | if (out != STDOUT_FILE_NO) |
| 272 | { |
| 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 | } |
| 292 | } |
| 293 | |
| 294 | if (errdes != STDERR_FILE_NO |
| 295 | || (flags & PEX_STDERR_TO_STDOUT) != 0) |
| 296 | { |
| 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 | } |
| 320 | } |
| 321 | |
| 322 | pid = (((flags & PEX_SEARCH) != 0 ? _spawnvp : _spawnv) |
| 323 | (_P_NOWAIT, executable, fix_argv (argv))); |
| 324 | |
| 325 | if (pid == -1) |
| 326 | { |
| 327 | *err = errno; |
| 328 | *errmsg = ((flags & PEX_SEARCH) != 0) ? "_spawnvp" : "_spawnv"; |
| 329 | } |
| 330 | |
| 331 | if (in != STDIN_FILE_NO) |
| 332 | { |
| 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 | } |
| 345 | } |
| 346 | |
| 347 | if (out != STDOUT_FILE_NO) |
| 348 | { |
| 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 | } |
| 361 | } |
| 362 | |
| 363 | if (errdes != STDERR_FILE_NO |
| 364 | || (flags & PEX_STDERR_TO_STDOUT) != 0) |
| 365 | { |
| 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 | } |
| 378 | } |
| 379 | |
| 380 | return pid; |
| 381 | } |
| 382 | |
| 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) |
| 395 | { |
| 396 | int termstat; |
| 397 | |
| 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 | } |
| 410 | |
| 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. */ |
| 415 | |
| 416 | if (termstat == 3) |
| 417 | *status = SIGABRT; |
| 418 | else |
| 419 | *status = ((termstat & 0xff) << 8); |
| 420 | |
| 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"); |
| 440 | } |