The first a bug in pax and should be commited to FBSD, too.
[dragonfly.git] / contrib / lukemftpd / src / popen.c
1 /*      $NetBSD: popen.c,v 1.27 2001/12/01 10:25:30 lukem Exp $ */
2
3 /*-
4  * Copyright (c) 1999-2001 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Luke Mewburn.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 3. All advertising materials mentioning features or use of this software
19  *    must display the following acknowledgement:
20  *        This product includes software developed by the NetBSD
21  *        Foundation, Inc. and its contributors.
22  * 4. Neither the name of The NetBSD Foundation nor the names of its
23  *    contributors may be used to endorse or promote products derived
24  *    from this software without specific prior written permission.
25  *
26  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36  * POSSIBILITY OF SUCH DAMAGE.
37  */
38
39 /*
40  * Copyright (c) 1988, 1993, 1994
41  *      The Regents of the University of California.  All rights reserved.
42  *
43  * This code is derived from software written by Ken Arnold and
44  * published in UNIX Review, Vol. 6, No. 8.
45  *
46  * Redistribution and use in source and binary forms, with or without
47  * modification, are permitted provided that the following conditions
48  * are met:
49  * 1. Redistributions of source code must retain the above copyright
50  *    notice, this list of conditions and the following disclaimer.
51  * 2. Redistributions in binary form must reproduce the above copyright
52  *    notice, this list of conditions and the following disclaimer in the
53  *    documentation and/or other materials provided with the distribution.
54  * 3. All advertising materials mentioning features or use of this software
55  *    must display the following acknowledgement:
56  *      This product includes software developed by the University of
57  *      California, Berkeley and its contributors.
58  * 4. Neither the name of the University nor the names of its contributors
59  *    may be used to endorse or promote products derived from this software
60  *    without specific prior written permission.
61  *
62  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
63  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
64  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
65  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
66  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
67  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
68  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
69  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
70  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
71  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
72  * SUCH DAMAGE.
73  *
74  */
75
76 #include "lukemftpd.h"
77
78 #include "extern.h"
79
80 #define INCR    100
81 /*
82  * Special version of popen which avoids call to shell.  This ensures no-one
83  * may create a pipe to a hidden program as a side effect of a list or dir
84  * command.
85  * If stderrfd != -1, then send stderr of a read command there,
86  * otherwise close stderr.
87  */
88 static int *pids;
89 static int fds;
90
91 extern int ls_main(int, char *[]);
92
93 FILE *
94 ftpd_popen(char *argv[], const char *ptype, int stderrfd)
95 {
96         FILE *iop;
97         int argc, pdes[2], pid, isls;
98         char **pop;
99         StringList *sl;
100
101         iop = NULL;
102         isls = 0;
103         if ((*ptype != 'r' && *ptype != 'w') || ptype[1])
104                 return (NULL);
105
106         if (!pids) {
107                 if ((fds = getdtablesize()) <= 0)
108                         return (NULL);
109                 if ((pids = (int *)malloc((u_int)(fds * sizeof(int)))) == NULL)
110                         return (NULL);
111                 memset(pids, 0, fds * sizeof(int));
112         }
113         if (pipe(pdes) < 0)
114                 return (NULL);
115
116         if ((sl = sl_init()) == NULL)
117                 goto pfree;
118
119                                         /* glob each piece */
120         if (sl_add(sl, xstrdup(argv[0])) == -1)
121                 goto pfree;
122         for (argc = 1; argv[argc]; argc++) {
123                 glob_t gl;
124                 int flags = GLOB_BRACE|GLOB_NOCHECK|GLOB_TILDE|GLOB_LIMIT;
125
126                 memset(&gl, 0, sizeof(gl));
127                 if (glob(argv[argc], flags, NULL, &gl)) {
128                         if (sl_add(sl, xstrdup(argv[argc])) == -1) {
129                                 globfree(&gl);
130                                 goto pfree;
131                         }
132                 } else {
133                         for (pop = gl.gl_pathv; *pop; pop++) {
134                                 if (sl_add(sl, xstrdup(*pop)) == -1) {
135                                         globfree(&gl);
136                                         goto pfree;
137                                 }
138                         }
139                 }
140                 globfree(&gl);
141         }
142         if (sl_add(sl, NULL) == -1)
143                 goto pfree;
144
145 #ifndef NO_INTERNAL_LS
146         isls = (strcmp(sl->sl_str[0], INTERNAL_LS) == 0);
147 #endif
148
149 #if HAVE_VFORK
150         pid = isls ? fork() : vfork();
151 #else
152         pid = fork();
153 #endif
154         switch (pid) {
155         case -1:                        /* error */
156                 (void)close(pdes[0]);
157                 (void)close(pdes[1]);
158                 goto pfree;
159                 /* NOTREACHED */
160         case 0:                         /* child */
161                 if (*ptype == 'r') {
162                         if (pdes[1] != STDOUT_FILENO) {
163                                 dup2(pdes[1], STDOUT_FILENO);
164                                 (void)close(pdes[1]);
165                         }
166                         if (stderrfd == -1)
167                                 (void)close(STDERR_FILENO);
168                         else
169                                 dup2(stderrfd, STDERR_FILENO);
170                         (void)close(pdes[0]);
171                 } else {
172                         if (pdes[0] != STDIN_FILENO) {
173                                 dup2(pdes[0], STDIN_FILENO);
174                                 (void)close(pdes[0]);
175                         }
176                         (void)close(pdes[1]);
177                 }
178 #ifndef NO_INTERNAL_LS
179                 if (isls) {     /* use internal ls */
180 #if HAVE_OPTRESET
181                         optreset = 1;
182 #endif
183                         optind = optopt = 1;
184                         closelog();
185                         exit(ls_main(sl->sl_cur - 1, sl->sl_str));
186                 }
187 #endif
188
189                 execv(sl->sl_str[0], sl->sl_str);
190                 _exit(1);
191         }
192         /* parent; assume fdopen can't fail...  */
193         if (*ptype == 'r') {
194                 iop = fdopen(pdes[0], ptype);
195                 (void)close(pdes[1]);
196         } else {
197                 iop = fdopen(pdes[1], ptype);
198                 (void)close(pdes[0]);
199         }
200         pids[fileno(iop)] = pid;
201
202  pfree:
203         if (sl)
204                 sl_free(sl, 1);
205         return (iop);
206 }
207
208 int
209 ftpd_pclose(FILE *iop)
210 {
211         int fdes, status;
212         pid_t pid;
213         sigset_t sigset, osigset;
214
215         /*
216          * pclose returns -1 if stream is not associated with a
217          * `popened' command, or, if already `pclosed'.
218          */
219         if (pids == 0 || pids[fdes = fileno(iop)] == 0)
220                 return (-1);
221         (void)fclose(iop);
222         sigemptyset(&sigset);
223         sigaddset(&sigset, SIGINT);
224         sigaddset(&sigset, SIGQUIT);
225         sigaddset(&sigset, SIGHUP);
226         sigprocmask(SIG_BLOCK, &sigset, &osigset);
227         while ((pid = waitpid(pids[fdes], &status, 0)) < 0 && errno == EINTR)
228                 continue;
229         sigprocmask(SIG_SETMASK, &osigset, NULL);
230         pids[fdes] = 0;
231         if (pid < 0)
232                 return (pid);
233         if (WIFEXITED(status))
234                 return (WEXITSTATUS(status));
235         return (1);
236 }