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