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 75pid_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 183actions_inited: 184 errno = r; 185 posix_spawn_file_actions_destroy(&actions); 186#endif 187stdout_opened: 188 close(stdout_pipe[0]); 189 close(stdout_pipe[1]); 190stdin_opened: 191 close(stdin_pipe[0]); 192 close(stdin_pipe[1]); 193state_allocated: 194 __archive_cmdline_free(cmdline); 195 return -1; 196} 197 198void 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) */ 239