contrib/libarchive/libarchive/filter_fork_posix.c

   1 /*-
   2  * Copyright (c) 2007 Joerg Sonnenberger
   3  * Copyright (c) 2012 Michihiro NAKAJIMA
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #include "archive_platform.h"
  28
  29 /* This capability is only available on POSIX systems. */
  30 #if defined(HAVE_PIPE) && defined(HAVE_FCNTL) && \
  31     (defined(HAVE_FORK) || defined(HAVE_VFORK) || defined(HAVE_POSIX_SPAWNP))
  32
  33 __FBSDID("$FreeBSD: head/lib/libarchive/filter_fork.c 182958 2008-09-12 05:33:00Z kientzle $");
  34
  35 #if defined(HAVE_SYS_TYPES_H)
  36 #  include <sys/types.h>
  37 #endif
  38 #ifdef HAVE_ERRNO_H
  39 #  include <errno.h>
  40 #endif
  41 #ifdef HAVE_STRING_H
  42 #  include <string.h>
  43 #endif
  44 #if defined(HAVE_POLL) && (defined(HAVE_POLL_H) || defined(HAVE_SYS_POLL_H))
  45 #  if defined(HAVE_POLL_H)
  46 #    include <poll.h>
  47 #  elif defined(HAVE_SYS_POLL_H)
  48 #    include <sys/poll.h>
  49 #  endif
  50 #elif defined(HAVE_SELECT)
  51 #  if defined(HAVE_SYS_SELECT_H)
  52 #    include <sys/select.h>
  53 #  elif defined(HAVE_UNISTD_H)
  54 #    include <unistd.h>
  55 #  endif
  56 #endif
  57 #ifdef HAVE_FCNTL_H
  58 #  include <fcntl.h>
  59 #endif
  60 #ifdef HAVE_SPAWN_H
  61 #  include <spawn.h>
  62 #endif
  63 #ifdef HAVE_STDLIB_H
  64 #  include <stdlib.h>
  65 #endif
  66 #ifdef HAVE_UNISTD_H
  67 #  include <unistd.h>
  68 #endif
  69
  70 #include "archive.h"
  71 #include "archive_cmdline_private.h"
  72
  73 #include "filter_fork.h"
  74
  75 pid_t
  76 __archive_create_child(const char *cmd, int *child_stdin, int *child_stdout)
  77 {
  78         pid_t child;
  79         int stdin_pipe[2], stdout_pipe[2], tmp;
  80 #if HAVE_POSIX_SPAWNP
  81         posix_spawn_file_actions_t actions;
  82         int r;
  83 #endif
  84         struct archive_cmdline *cmdline;
  85
  86         cmdline = __archive_cmdline_allocate();
  87         if (cmdline == NULL)
  88                 goto state_allocated;
  89         if (__archive_cmdline_parse(cmdline, cmd) != ARCHIVE_OK)
  90                 goto state_allocated;
  91
  92         if (pipe(stdin_pipe) == -1)
  93                 goto state_allocated;
  94         if (stdin_pipe[0] == 1 /* stdout */) {
  95                 if ((tmp = dup(stdin_pipe[0])) == -1)
  96                         goto stdin_opened;
  97                 close(stdin_pipe[0]);
  98                 stdin_pipe[0] = tmp;
  99         }
 100         if (pipe(stdout_pipe) == -1)
 101                 goto stdin_opened;
 102         if (stdout_pipe[1] == 0 /* stdin */) {
 103                 if ((tmp = dup(stdout_pipe[1])) == -1)
 104                         goto stdout_opened;
 105                 close(stdout_pipe[1]);
 106                 stdout_pipe[1] = tmp;
 107         }
 108
 109 #if HAVE_POSIX_SPAWNP
 110
 111         r = posix_spawn_file_actions_init(&actions);
 112         if (r != 0) {
 113                 errno = r;
 114                 goto stdout_opened;
 115         }
 116         r = posix_spawn_file_actions_addclose(&actions, stdin_pipe[1]);
 117         if (r != 0)
 118                 goto actions_inited;
 119         r = posix_spawn_file_actions_addclose(&actions, stdout_pipe[0]);
 120         if (r != 0)
 121                 goto actions_inited;
 122         /* Setup for stdin. */
 123         r = posix_spawn_file_actions_adddup2(&actions, stdin_pipe[0], 0);
 124         if (r != 0)
 125                 goto actions_inited;
 126         if (stdin_pipe[0] != 0 /* stdin */) {
 127                 r = posix_spawn_file_actions_addclose(&actions, stdin_pipe[0]);
 128                 if (r != 0)
 129                         goto actions_inited;
 130         }
 131         /* Setup for stdout. */
 132         r = posix_spawn_file_actions_adddup2(&actions, stdout_pipe[1], 1);
 133         if (r != 0)
 134                 goto actions_inited;
 135         if (stdout_pipe[1] != 1 /* stdout */) {
 136                 r = posix_spawn_file_actions_addclose(&actions, stdout_pipe[1]);
 137                 if (r != 0)
 138                         goto actions_inited;
 139         }
 140         r = posix_spawnp(&child, cmdline->path, &actions, NULL,
 141                 cmdline->argv, NULL);
 142         if (r != 0)
 143                 goto actions_inited;
 144         posix_spawn_file_actions_destroy(&actions);
 145
 146 #else /* HAVE_POSIX_SPAWNP */
 147
 148 #if HAVE_VFORK
 149         child = vfork();
 150 #else
 151         child = fork();
 152 #endif
 153         if (child == -1)
 154                 goto stdout_opened;
 155         if (child == 0) {
 156                 close(stdin_pipe[1]);
 157                 close(stdout_pipe[0]);
 158                 if (dup2(stdin_pipe[0], 0 /* stdin */) == -1)
 159                         _exit(254);
 160                 if (stdin_pipe[0] != 0 /* stdin */)
 161                         close(stdin_pipe[0]);
 162                 if (dup2(stdout_pipe[1], 1 /* stdout */) == -1)
 163                         _exit(254);
 164                 if (stdout_pipe[1] != 1 /* stdout */)
 165                         close(stdout_pipe[1]);
 166                 execvp(cmdline->path, cmdline->argv);
 167                 _exit(254);
 168         }
 169 #endif /* HAVE_POSIX_SPAWNP */
 170
 171         close(stdin_pipe[0]);
 172         close(stdout_pipe[1]);
 173
 174         *child_stdin = stdin_pipe[1];
 175         fcntl(*child_stdin, F_SETFL, O_NONBLOCK);
 176         *child_stdout = stdout_pipe[0];
 177         fcntl(*child_stdout, F_SETFL, O_NONBLOCK);
 178         __archive_cmdline_free(cmdline);
 179
 180         return child;
 181
 182 #if HAVE_POSIX_SPAWNP
 183 actions_inited:
 184         errno = r;
 185         posix_spawn_file_actions_destroy(&actions);
 186 #endif
 187 stdout_opened:
 188         close(stdout_pipe[0]);
 189         close(stdout_pipe[1]);
 190 stdin_opened:
 191         close(stdin_pipe[0]);
 192         close(stdin_pipe[1]);
 193 state_allocated:
 194         __archive_cmdline_free(cmdline);
 195         return -1;
 196 }
 197
 198 void
 199 __archive_check_child(int in, int out)
 200 {
 201 #if defined(HAVE_POLL) && (defined(HAVE_POLL_H) || defined(HAVE_SYS_POLL_H))
 202         struct pollfd fds[2];
 203         int idx;
 204
 205         idx = 0;
 206         if (in != -1) {
 207                 fds[idx].fd = in;
 208                 fds[idx].events = POLLOUT;
 209                 ++idx;
 210         }
 211         if (out != -1) {
 212                 fds[idx].fd = out;
 213                 fds[idx].events = POLLIN;
 214                 ++idx;
 215         }
 216
 217         poll(fds, idx, -1); /* -1 == INFTIM, wait forever */
 218 #elif defined(HAVE_SELECT)
 219         fd_set fds_in, fds_out, fds_error;
 220
 221         FD_ZERO(&fds_in);
 222         FD_ZERO(&fds_out);
 223         FD_ZERO(&fds_error);
 224         if (out != -1) {
 225                 FD_SET(out, &fds_in);
 226                 FD_SET(out, &fds_error);
 227         }
 228         if (in != -1) {
 229                 FD_SET(in, &fds_out);
 230                 FD_SET(in, &fds_error);
 231         }
 232         select(in < out ? out + 1 : in + 1, &fds_in, &fds_out, &fds_error, NULL);
 233 #else
 234         sleep(1);
 235 #endif
 236 }
 237
 238 #endif /* defined(HAVE_PIPE) && defined(HAVE_VFORK) && defined(HAVE_FCNTL) */