1 /* Common code for executing a program in a sub-process.
2 Copyright (C) 2005, 2010 Free Software Foundation, Inc.
3 Written by Ian Lance Taylor <ian@airs.com>.
4
5 This file is part of the libiberty library.
6 Libiberty is free software; you can redistribute it and/or
7 modify it under the terms of the GNU Library General Public
8 License as published by the Free Software Foundation; either
9 version 2 of the License, or (at your option) any later version.
10
11 Libiberty is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 Library General Public License for more details.
15
16 You should have received a copy of the GNU Library General Public
17 License along with libiberty; see the file COPYING.LIB. If not,
18 write to the Free Software Foundation, Inc., 51 Franklin Street - Fifth Floor,
19 Boston, MA 02110-1301, USA. */
20
21 #include "config.h"
22 #include "libiberty.h"
23 #include "pex-common.h"
24
25 #include <stdio.h>
26 #include <errno.h>
27 #ifdef NEED_DECLARATION_ERRNO
28 extern int errno;
29 #endif
30 #ifdef HAVE_STDLIB_H
31 #include <stdlib.h>
32 #endif
33 #ifdef HAVE_STRING_H
34 #include <string.h>
35 #endif
36 #ifdef HAVE_UNISTD_H
37 #include <unistd.h>
38 #endif
39
40 extern int mkstemps (char *, int);
41
42 /* This file contains subroutines for the program execution routines
43 (pex_init, pex_run, etc.). This file is compiled on all
44 systems. */
45
46 static void pex_add_remove (struct pex_obj *, const char *, int);
47 static int pex_get_status_and_time (struct pex_obj *, int, const char **,
48 int *);
49
50 /* Initialize a pex_obj structure. */
51
52 struct pex_obj *
pex_init_common(int flags,const char * pname,const char * tempbase,const struct pex_funcs * funcs)53 pex_init_common (int flags, const char *pname, const char *tempbase,
54 const struct pex_funcs *funcs)
55 {
56 struct pex_obj *obj;
57
58 obj = XNEW (struct pex_obj);
59 obj->flags = flags;
60 obj->pname = pname;
61 obj->tempbase = tempbase;
62 obj->next_input = STDIN_FILE_NO;
63 obj->next_input_name = NULL;
64 obj->next_input_name_allocated = 0;
65 obj->stderr_pipe = -1;
66 obj->count = 0;
67 obj->children = NULL;
68 obj->status = NULL;
69 obj->time = NULL;
70 obj->number_waited = 0;
71 obj->input_file = NULL;
72 obj->read_output = NULL;
73 obj->read_err = NULL;
74 obj->remove_count = 0;
75 obj->remove = NULL;
76 obj->funcs = funcs;
77 obj->sysdep = NULL;
78 return obj;
79 }
80
81 /* Add a file to be removed when we are done. */
82
83 static void
pex_add_remove(struct pex_obj * obj,const char * name,int allocated)84 pex_add_remove (struct pex_obj *obj, const char *name, int allocated)
85 {
86 char *add;
87
88 ++obj->remove_count;
89 obj->remove = XRESIZEVEC (char *, obj->remove, obj->remove_count);
90 if (allocated)
91 add = (char *) name;
92 else
93 add = xstrdup (name);
94 obj->remove[obj->remove_count - 1] = add;
95 }
96
97 /* Generate a temporary file name based on OBJ, FLAGS, and NAME.
98 Return NULL if we were unable to reserve a temporary filename.
99
100 If non-NULL, the result is either allocated with malloc, or the
101 same pointer as NAME. */
102 static char *
temp_file(struct pex_obj * obj,int flags,char * name)103 temp_file (struct pex_obj *obj, int flags, char *name)
104 {
105 if (name == NULL)
106 {
107 if (obj->tempbase == NULL)
108 {
109 name = make_temp_file (NULL);
110 }
111 else
112 {
113 int len = strlen (obj->tempbase);
114 int out;
115
116 if (len >= 6
117 && strcmp (obj->tempbase + len - 6, "XXXXXX") == 0)
118 name = xstrdup (obj->tempbase);
119 else
120 name = concat (obj->tempbase, "XXXXXX", NULL);
121
122 out = mkstemps (name, 0);
123 if (out < 0)
124 {
125 free (name);
126 return NULL;
127 }
128
129 /* This isn't obj->funcs->close because we got the
130 descriptor from mkstemps, not from a function in
131 obj->funcs. Calling close here is just like what
132 make_temp_file does. */
133 close (out);
134 }
135 }
136 else if ((flags & PEX_SUFFIX) != 0)
137 {
138 if (obj->tempbase == NULL)
139 name = make_temp_file (name);
140 else
141 name = concat (obj->tempbase, name, NULL);
142 }
143
144 return name;
145 }
146
147
148 /* As for pex_run (), but permits the environment for the child process
149 to be specified. */
150
151 const char *
pex_run_in_environment(struct pex_obj * obj,int flags,const char * executable,char * const * argv,char * const * env,const char * orig_outname,const char * errname,int * err)152 pex_run_in_environment (struct pex_obj *obj, int flags, const char *executable,
153 char * const * argv, char * const * env,
154 const char *orig_outname, const char *errname,
155 int *err)
156 {
157 const char *errmsg;
158 int in, out, errdes;
159 char *outname;
160 int outname_allocated;
161 int p[2];
162 int toclose;
163 pid_t pid;
164
165 in = -1;
166 out = -1;
167 errdes = -1;
168 outname = (char *) orig_outname;
169 outname_allocated = 0;
170
171 /* If the user called pex_input_file, close the file now. */
172 if (obj->input_file)
173 {
174 if (fclose (obj->input_file) == EOF)
175 {
176 errmsg = "closing pipeline input file";
177 goto error_exit;
178 }
179 obj->input_file = NULL;
180 }
181
182 /* Set IN. */
183
184 if (obj->next_input_name != NULL)
185 {
186 /* We have to make sure that the previous process has completed
187 before we try to read the file. */
188 if (!pex_get_status_and_time (obj, 0, &errmsg, err))
189 goto error_exit;
190
191 in = obj->funcs->open_read (obj, obj->next_input_name,
192 (flags & PEX_BINARY_INPUT) != 0);
193 if (in < 0)
194 {
195 *err = errno;
196 errmsg = "open temporary file";
197 goto error_exit;
198 }
199 if (obj->next_input_name_allocated)
200 {
201 free (obj->next_input_name);
202 obj->next_input_name_allocated = 0;
203 }
204 obj->next_input_name = NULL;
205 }
206 else
207 {
208 in = obj->next_input;
209 if (in < 0)
210 {
211 *err = 0;
212 errmsg = "pipeline already complete";
213 goto error_exit;
214 }
215 }
216
217 /* Set OUT and OBJ->NEXT_INPUT/OBJ->NEXT_INPUT_NAME. */
218
219 if ((flags & PEX_LAST) != 0)
220 {
221 if (outname == NULL)
222 out = STDOUT_FILE_NO;
223 else if ((flags & PEX_SUFFIX) != 0)
224 {
225 outname = concat (obj->tempbase, outname, NULL);
226 outname_allocated = 1;
227 }
228 obj->next_input = -1;
229 }
230 else if ((obj->flags & PEX_USE_PIPES) == 0)
231 {
232 outname = temp_file (obj, flags, outname);
233 if (! outname)
234 {
235 *err = 0;
236 errmsg = "could not create temporary file";
237 goto error_exit;
238 }
239
240 if (outname != orig_outname)
241 outname_allocated = 1;
242
243 if ((obj->flags & PEX_SAVE_TEMPS) == 0)
244 {
245 pex_add_remove (obj, outname, outname_allocated);
246 outname_allocated = 0;
247 }
248
249 /* Hand off ownership of outname to the next stage. */
250 obj->next_input_name = outname;
251 obj->next_input_name_allocated = outname_allocated;
252 outname_allocated = 0;
253 }
254 else
255 {
256 if (obj->funcs->pipe (obj, p, (flags & PEX_BINARY_OUTPUT) != 0) < 0)
257 {
258 *err = errno;
259 errmsg = "pipe";
260 goto error_exit;
261 }
262
263 out = p[WRITE_PORT];
264 obj->next_input = p[READ_PORT];
265 }
266
267 if (out < 0)
268 {
269 out = obj->funcs->open_write (obj, outname,
270 (flags & PEX_BINARY_OUTPUT) != 0,
271 (flags & PEX_STDOUT_APPEND) != 0);
272 if (out < 0)
273 {
274 *err = errno;
275 errmsg = "open temporary output file";
276 goto error_exit;
277 }
278 }
279
280 if (outname_allocated)
281 {
282 free (outname);
283 outname_allocated = 0;
284 }
285
286 /* Set ERRDES. */
287
288 if (errname != NULL && (flags & PEX_STDERR_TO_PIPE) != 0)
289 {
290 *err = 0;
291 errmsg = "both ERRNAME and PEX_STDERR_TO_PIPE specified.";
292 goto error_exit;
293 }
294
295 if (obj->stderr_pipe != -1)
296 {
297 *err = 0;
298 errmsg = "PEX_STDERR_TO_PIPE used in the middle of pipeline";
299 goto error_exit;
300 }
301
302 if (errname == NULL)
303 {
304 if (flags & PEX_STDERR_TO_PIPE)
305 {
306 if (obj->funcs->pipe (obj, p, (flags & PEX_BINARY_ERROR) != 0) < 0)
307 {
308 *err = errno;
309 errmsg = "pipe";
310 goto error_exit;
311 }
312
313 errdes = p[WRITE_PORT];
314 obj->stderr_pipe = p[READ_PORT];
315 }
316 else
317 {
318 errdes = STDERR_FILE_NO;
319 }
320 }
321 else
322 {
323 errdes = obj->funcs->open_write (obj, errname,
324 (flags & PEX_BINARY_ERROR) != 0,
325 (flags & PEX_STDERR_APPEND) != 0);
326 if (errdes < 0)
327 {
328 *err = errno;
329 errmsg = "open error file";
330 goto error_exit;
331 }
332 }
333
334 /* If we are using pipes, the child process has to close the next
335 input pipe. */
336
337 if ((obj->flags & PEX_USE_PIPES) == 0)
338 toclose = -1;
339 else
340 toclose = obj->next_input;
341
342 /* Run the program. */
343
344 pid = obj->funcs->exec_child (obj, flags, executable, argv, env,
345 in, out, errdes, toclose, &errmsg, err);
346 if (pid < 0)
347 goto error_exit;
348
349 ++obj->count;
350 obj->children = XRESIZEVEC (pid_t, obj->children, obj->count);
351 obj->children[obj->count - 1] = pid;
352
353 return NULL;
354
355 error_exit:
356 if (in >= 0 && in != STDIN_FILE_NO)
357 obj->funcs->close (obj, in);
358 if (out >= 0 && out != STDOUT_FILE_NO)
359 obj->funcs->close (obj, out);
360 if (errdes >= 0 && errdes != STDERR_FILE_NO)
361 obj->funcs->close (obj, errdes);
362 if (outname_allocated)
363 free (outname);
364 return errmsg;
365 }
366
367 /* Run a program. */
368
369 const char *
pex_run(struct pex_obj * obj,int flags,const char * executable,char * const * argv,const char * orig_outname,const char * errname,int * err)370 pex_run (struct pex_obj *obj, int flags, const char *executable,
371 char * const * argv, const char *orig_outname, const char *errname,
372 int *err)
373 {
374 return pex_run_in_environment (obj, flags, executable, argv, NULL,
375 orig_outname, errname, err);
376 }
377
378 /* Return a FILE pointer for a temporary file to fill with input for
379 the pipeline. */
380 FILE *
pex_input_file(struct pex_obj * obj,int flags,const char * in_name)381 pex_input_file (struct pex_obj *obj, int flags, const char *in_name)
382 {
383 char *name = (char *) in_name;
384 FILE *f;
385
386 /* This must be called before the first pipeline stage is run, and
387 there must not have been any other input selected. */
388 if (obj->count != 0
389 || (obj->next_input >= 0 && obj->next_input != STDIN_FILE_NO)
390 || obj->next_input_name)
391 {
392 errno = EINVAL;
393 return NULL;
394 }
395
396 name = temp_file (obj, flags, name);
397 if (! name)
398 return NULL;
399
400 f = fopen (name, (flags & PEX_BINARY_OUTPUT) ? "wb" : "w");
401 if (! f)
402 {
403 free (name);
404 return NULL;
405 }
406
407 obj->input_file = f;
408 obj->next_input_name = name;
409 obj->next_input_name_allocated = (name != in_name);
410
411 return f;
412 }
413
414 /* Return a stream for a pipe connected to the standard input of the
415 first stage of the pipeline. */
416 FILE *
pex_input_pipe(struct pex_obj * obj,int binary)417 pex_input_pipe (struct pex_obj *obj, int binary)
418 {
419 int p[2];
420 FILE *f;
421
422 /* You must call pex_input_pipe before the first pex_run or pex_one. */
423 if (obj->count > 0)
424 goto usage_error;
425
426 /* You must be using pipes. Implementations that don't support
427 pipes clear this flag before calling pex_init_common. */
428 if (! (obj->flags & PEX_USE_PIPES))
429 goto usage_error;
430
431 /* If we have somehow already selected other input, that's a
432 mistake. */
433 if ((obj->next_input >= 0 && obj->next_input != STDIN_FILE_NO)
434 || obj->next_input_name)
435 goto usage_error;
436
437 if (obj->funcs->pipe (obj, p, binary != 0) < 0)
438 return NULL;
439
440 f = obj->funcs->fdopenw (obj, p[WRITE_PORT], binary != 0);
441 if (! f)
442 {
443 int saved_errno = errno;
444 obj->funcs->close (obj, p[READ_PORT]);
445 obj->funcs->close (obj, p[WRITE_PORT]);
446 errno = saved_errno;
447 return NULL;
448 }
449
450 obj->next_input = p[READ_PORT];
451
452 return f;
453
454 usage_error:
455 errno = EINVAL;
456 return NULL;
457 }
458
459 /* Return a FILE pointer for the output of the last program
460 executed. */
461
462 FILE *
pex_read_output(struct pex_obj * obj,int binary)463 pex_read_output (struct pex_obj *obj, int binary)
464 {
465 if (obj->next_input_name != NULL)
466 {
467 const char *errmsg;
468 int err;
469
470 /* We have to make sure that the process has completed before we
471 try to read the file. */
472 if (!pex_get_status_and_time (obj, 0, &errmsg, &err))
473 {
474 errno = err;
475 return NULL;
476 }
477
478 obj->read_output = fopen (obj->next_input_name, binary ? "rb" : "r");
479
480 if (obj->next_input_name_allocated)
481 {
482 free (obj->next_input_name);
483 obj->next_input_name_allocated = 0;
484 }
485 obj->next_input_name = NULL;
486 }
487 else
488 {
489 int o;
490
491 o = obj->next_input;
492 if (o < 0 || o == STDIN_FILE_NO)
493 return NULL;
494 obj->read_output = obj->funcs->fdopenr (obj, o, binary);
495 obj->next_input = -1;
496 }
497
498 return obj->read_output;
499 }
500
501 FILE *
pex_read_err(struct pex_obj * obj,int binary)502 pex_read_err (struct pex_obj *obj, int binary)
503 {
504 int o;
505
506 o = obj->stderr_pipe;
507 if (o < 0 || o == STDIN_FILE_NO)
508 return NULL;
509 obj->read_err = obj->funcs->fdopenr (obj, o, binary);
510 obj->stderr_pipe = -1;
511 return obj->read_err;
512 }
513
514 /* Get the exit status and, if requested, the resource time for all
515 the child processes. Return 0 on failure, 1 on success. */
516
517 static int
pex_get_status_and_time(struct pex_obj * obj,int done,const char ** errmsg,int * err)518 pex_get_status_and_time (struct pex_obj *obj, int done, const char **errmsg,
519 int *err)
520 {
521 int ret;
522 int i;
523
524 if (obj->number_waited == obj->count)
525 return 1;
526
527 obj->status = XRESIZEVEC (int, obj->status, obj->count);
528 if ((obj->flags & PEX_RECORD_TIMES) != 0)
529 obj->time = XRESIZEVEC (struct pex_time, obj->time, obj->count);
530
531 ret = 1;
532 for (i = obj->number_waited; i < obj->count; ++i)
533 {
534 if (obj->funcs->wait (obj, obj->children[i], &obj->status[i],
535 obj->time == NULL ? NULL : &obj->time[i],
536 done, errmsg, err) < 0)
537 ret = 0;
538 }
539 obj->number_waited = i;
540
541 return ret;
542 }
543
544 /* Get exit status of executed programs. */
545
546 int
pex_get_status(struct pex_obj * obj,int count,int * vector)547 pex_get_status (struct pex_obj *obj, int count, int *vector)
548 {
549 if (obj->status == NULL)
550 {
551 const char *errmsg;
552 int err;
553
554 if (!pex_get_status_and_time (obj, 0, &errmsg, &err))
555 return 0;
556 }
557
558 if (count > obj->count)
559 {
560 memset (vector + obj->count, 0, (count - obj->count) * sizeof (int));
561 count = obj->count;
562 }
563
564 memcpy (vector, obj->status, count * sizeof (int));
565
566 return 1;
567 }
568
569 /* Get process times of executed programs. */
570
571 int
pex_get_times(struct pex_obj * obj,int count,struct pex_time * vector)572 pex_get_times (struct pex_obj *obj, int count, struct pex_time *vector)
573 {
574 if (obj->status == NULL)
575 {
576 const char *errmsg;
577 int err;
578
579 if (!pex_get_status_and_time (obj, 0, &errmsg, &err))
580 return 0;
581 }
582
583 if (obj->time == NULL)
584 return 0;
585
586 if (count > obj->count)
587 {
588 memset (vector + obj->count, 0,
589 (count - obj->count) * sizeof (struct pex_time));
590 count = obj->count;
591 }
592
593 memcpy (vector, obj->time, count * sizeof (struct pex_time));
594
595 return 1;
596 }
597
598 /* Free a pex_obj structure. */
599
600 void
pex_free(struct pex_obj * obj)601 pex_free (struct pex_obj *obj)
602 {
603 /* Close pipe file descriptors corresponding to child's stdout and
604 stderr so that the child does not hang trying to output something
605 while we're waiting for it. */
606 if (obj->next_input >= 0 && obj->next_input != STDIN_FILE_NO)
607 obj->funcs->close (obj, obj->next_input);
608 if (obj->stderr_pipe >= 0 && obj->stderr_pipe != STDIN_FILE_NO)
609 obj->funcs->close (obj, obj->stderr_pipe);
610 if (obj->read_output != NULL)
611 fclose (obj->read_output);
612 if (obj->read_err != NULL)
613 fclose (obj->read_err);
614
615 /* If the caller forgot to wait for the children, we do it here, to
616 avoid zombies. */
617 if (obj->status == NULL)
618 {
619 const char *errmsg;
620 int err;
621
622 obj->flags &= ~ PEX_RECORD_TIMES;
623 pex_get_status_and_time (obj, 1, &errmsg, &err);
624 }
625
626 if (obj->next_input_name_allocated)
627 free (obj->next_input_name);
628 free (obj->children);
629 free (obj->status);
630 free (obj->time);
631
632 if (obj->remove_count > 0)
633 {
634 int i;
635
636 for (i = 0; i < obj->remove_count; ++i)
637 {
638 remove (obj->remove[i]);
639 free (obj->remove[i]);
640 }
641 free (obj->remove);
642 }
643
644 if (obj->funcs->cleanup != NULL)
645 obj->funcs->cleanup (obj);
646
647 free (obj);
648 }
649