2 * Heirloom mailx - a mail user agent derived from Berkeley Mail.
4 * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
5 * Copyright (c) 2012 Steffen "Daode" Nurpmeso.
8 * Copyright (c) 1980, 1993
9 * The Regents of the University of California. All rights reserved.
11 * Redistribution and use in source and binary forms, with or without
12 * modification, are permitted provided that the following conditions
14 * 1. Redistributions of source code must retain the above copyright
15 * notice, this list of conditions and the following disclaimer.
16 * 2. Redistributions in binary form must reproduce the above copyright
17 * notice, this list of conditions and the following disclaimer in the
18 * documentation and/or other materials provided with the distribution.
19 * 3. All advertising materials mentioning features or use of this software
20 * must display the following acknowledgement:
21 * This product includes software developed by the University of
22 * California, Berkeley and its contributors.
23 * 4. Neither the name of the University nor the names of its contributors
24 * may be used to endorse or promote products derived from this software
25 * without specific prior written permission.
27 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
28 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
29 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
30 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
31 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
32 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
33 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
34 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
35 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
36 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
42 static char sccsid
[] = "@(#)popen.c 2.20 (gritter) 3/4/06";
79 static struct fp
*fp_head
;
88 static struct child
*child
;
90 static int scan_mode(const char *mode
, int *omode
);
91 static void register_file(FILE *fp
, int omode
, int pipe
, int pid
,
92 int compressed
, const char *realfile
, long offset
);
93 static enum okay
compress(struct fp
*fpp
);
94 static int decompress(int compression
, int input
, int output
);
95 static enum okay
unregister_file(FILE *fp
);
96 static int file_pid(FILE *fp
);
97 static int wait_command(int pid
);
98 static struct child
*findchild(int pid
);
99 static void delchild(struct child
*cp
);
102 * Provide BSD-like signal() on all systems.
105 safe_signal(int signum
, sighandler_type handler
)
107 struct sigaction nact
, oact
;
109 nact
.sa_handler
= handler
;
110 sigemptyset(&nact
.sa_mask
);
113 nact
.sa_flags
|= SA_RESTART
;
115 if (sigaction(signum
, &nact
, &oact
) != 0)
117 return oact
.sa_handler
;
121 scan_mode(const char *mode
, int *omode
)
124 if (!strcmp(mode
, "r")) {
126 } else if (!strcmp(mode
, "w")) {
127 *omode
= O_WRONLY
| O_CREAT
| O_TRUNC
;
128 } else if (!strcmp(mode
, "wx")) {
129 *omode
= O_WRONLY
| O_CREAT
| O_EXCL
;
130 } else if (!strcmp(mode
, "a")) {
131 *omode
= O_WRONLY
| O_APPEND
| O_CREAT
;
132 } else if (!strcmp(mode
, "a+")) {
133 *omode
= O_RDWR
| O_APPEND
;
134 } else if (!strcmp(mode
, "r+")) {
136 } else if (!strcmp(mode
, "w+")) {
137 *omode
= O_RDWR
| O_CREAT
| O_EXCL
;
139 fprintf(stderr
, catgets(catd
, CATSET
, 152,
140 "Internal error: bad stdio open mode %s\n"), mode
);
148 safe_fopen(const char *file
, const char *mode
, int *omode
)
152 if (scan_mode(mode
, omode
) < 0)
154 if ((fd
= open(file
, *omode
, 0666)) < 0)
156 return fdopen(fd
, mode
);
160 Fopen(const char *file
, const char *mode
)
165 if ((fp
= safe_fopen(file
, mode
, &omode
)) != NULL
) {
166 register_file(fp
, omode
, 0, 0, FP_UNCOMPRESSED
, NULL
, 0L);
167 fcntl(fileno(fp
), F_SETFD
, FD_CLOEXEC
);
173 Fdopen(int fd
, const char *mode
)
178 scan_mode(mode
, &omode
);
179 if ((fp
= fdopen(fd
, mode
)) != NULL
) {
180 register_file(fp
, omode
, 0, 0, FP_UNCOMPRESSED
, NULL
, 0L);
181 fcntl(fileno(fp
), F_SETFD
, FD_CLOEXEC
);
190 if (unregister_file(fp
) == OKAY
)
194 return i
== 3 ? 0 : EOF
;
198 Zopen(const char *file
, const char *mode
, int *compression
)
211 if (scan_mode(mode
, &omode
) < 0)
213 if (compression
== NULL
)
214 compression
= &_compression
;
215 bits
= R_OK
| (omode
== O_RDONLY
? 0 : W_OK
);
216 if (omode
& O_APPEND
&& ((p
= which_protocol(file
)) == PROTO_IMAP
||
217 p
== PROTO_MAILDIR
)) {
218 *compression
= p
== PROTO_IMAP
? FP_IMAP
: FP_MAILDIR
;
219 omode
= O_RDWR
| O_APPEND
| O_CREAT
;
224 if ((extension
= strrchr(file
, '.')) != NULL
) {
226 if (strcmp(extension
, ".gz") == 0)
228 if (strcmp(extension
, ".bz2") == 0)
231 if (access(file
, F_OK
) == 0) {
232 *compression
= FP_UNCOMPRESSED
;
233 return Fopen(file
, mode
);
234 } else if (access(rp
=savecat(file
, ".gz"), bits
) == 0) {
235 gzip
: *compression
= FP_GZIPPED
;
236 } else if (access(rp
=savecat(file
, ".bz2"), bits
) == 0) {
237 bz2
: *compression
= FP_BZIP2ED
;
239 *compression
= FP_UNCOMPRESSED
;
240 return Fopen(file
, mode
);
242 if (access(rp
, W_OK
) < 0)
243 *compression
|= FP_READONLY
;
244 if ((input
= open(rp
, bits
& W_OK
? O_RDWR
: O_RDONLY
)) < 0
245 && ((omode
&O_CREAT
) == 0 || errno
!= ENOENT
))
247 open
: if ((output
= Ftemp(&tempfn
, "Rz", "w+", 0600, 0)) == NULL
) {
248 perror(catgets(catd
, CATSET
, 167, "tmpfile"));
253 if (input
>= 0 || (*compression
&FP_MASK
) == FP_IMAP
||
254 (*compression
&FP_MASK
) == FP_MAILDIR
) {
255 if (decompress(*compression
, input
, fileno(output
)) < 0) {
261 if ((input
= creat(rp
, 0666)) < 0) {
268 if (omode
& O_APPEND
) {
271 if ((flags
= fcntl(fileno(output
), F_GETFL
)) != -1)
272 fcntl(fileno(output
), F_SETFL
, flags
| O_APPEND
);
273 offset
= ftell(output
);
278 register_file(output
, omode
, 0, 0, *compression
, rp
, offset
);
283 Popen(const char *cmd
, const char *mode
, const char *shell
, int newfd1
)
286 int myside
, hisside
, fd0
, fd1
;
288 char mod
[2] = { '0', '\0' };
294 fcntl(p
[READ
], F_SETFD
, FD_CLOEXEC
);
295 fcntl(p
[WRITE
], F_SETFD
, FD_CLOEXEC
);
299 hisside
= fd1
= p
[WRITE
];
301 } else if (*mode
== 'W') {
303 hisside
= fd0
= p
[READ
];
308 hisside
= fd0
= p
[READ
];
314 pid
= start_command(cmd
, &nset
, fd0
, fd1
, NULL
, NULL
, NULL
);
316 pid
= start_command(shell
, &nset
, fd0
, fd1
, "-c", cmd
, NULL
);
324 if ((fp
= fdopen(myside
, mod
)) != NULL
)
325 register_file(fp
, 0, 1, pid
, FP_UNCOMPRESSED
, NULL
, 0L);
338 unregister_file(ptr
);
341 sigaddset(&nset
, SIGINT
);
342 sigaddset(&nset
, SIGHUP
);
343 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
345 sigprocmask(SIG_SETMASK
, &oset
, (sigset_t
*)NULL
);
350 close_all_files(void)
361 register_file(FILE *fp
, int omode
, int pipe
, int pid
, int compressed
,
362 const char *realfile
, long offset
)
366 fpp
= (struct fp
*)smalloc(sizeof *fpp
);
372 fpp
->compressed
= compressed
;
373 fpp
->realfile
= realfile
? sstrdup(realfile
) : NULL
;
374 fpp
->offset
= offset
;
379 compress(struct fp
*fpp
)
385 if (fpp
->omode
== O_RDONLY
)
389 fseek(fpp
->fp
, fpp
->offset
, SEEK_SET
);
390 if ((fpp
->compressed
&FP_MASK
) == FP_IMAP
) {
391 return imap_append(fpp
->realfile
, fpp
->fp
);
393 if ((fpp
->compressed
&FP_MASK
) == FP_MAILDIR
) {
394 return maildir_append(fpp
->realfile
, fpp
->fp
);
396 if ((output
= open(fpp
->realfile
,
397 (fpp
->omode
|O_CREAT
)&~O_EXCL
,
399 fprintf(stderr
, "Fatal: cannot create ");
400 perror(fpp
->realfile
);
403 if ((fpp
->omode
& O_APPEND
) == 0)
404 ftruncate(output
, 0);
405 switch (fpp
->compressed
& FP_MASK
) {
407 command
[0] = "gzip"; command
[1] = "-c"; break;
409 command
[0] = "bzip2"; command
[1] = "-c"; break;
411 command
[0] = "cat"; command
[1] = NULL
; break;
413 if (run_command(command
[0], 0, fileno(fpp
->fp
), output
,
414 command
[1], NULL
, NULL
) < 0)
423 decompress(int compression
, int input
, int output
)
428 * Note that it is not possible to handle 'pack' or 'compress'
429 * formats because appending data does not work with them.
431 switch (compression
& FP_MASK
) {
432 case FP_GZIPPED
: command
[0] = "gzip"; command
[1] = "-cd"; break;
433 case FP_BZIP2ED
: command
[0] = "bzip2"; command
[1] = "-cd"; break;
434 case FP_IMAP
: return 0;
435 case FP_MAILDIR
: return 0;
436 default: command
[0] = "cat"; command
[1] = NULL
;
438 return run_command(command
[0], 0, input
, output
,
439 command
[1], NULL
, NULL
);
443 unregister_file(FILE *fp
)
448 for (pp
= &fp_head
; (p
= *pp
) != (struct fp
*)NULL
; pp
= &p
->link
)
450 if ((p
->compressed
&FP_MASK
) != FP_UNCOMPRESSED
)
456 panic(catgets(catd
, CATSET
, 153, "Invalid file pointer"));
466 for (p
= fp_head
; p
; p
= p
->link
)
473 * Run a command without a shell, with optional arguments and splicing
474 * of stdin and stdout. The command name can be a sequence of words.
475 * Signals must be handled by the caller.
476 * "Mask" contains the signals to ignore in the new process.
477 * SIGINT is enabled unless it's in the mask.
481 run_command(char *cmd
, sigset_t
*mask
, int infd
, int outfd
,
482 char *a0
, char *a1
, char *a2
)
486 if ((pid
= start_command(cmd
, mask
, infd
, outfd
, a0
, a1
, a2
)) < 0)
488 return wait_command(pid
);
493 start_command(const char *cmd
, sigset_t
*mask
, int infd
, int outfd
,
494 const char *a0
, const char *a1
, const char *a2
)
498 if ((pid
= fork()) < 0) {
504 int i
= getrawlist(cmd
, strlen(cmd
),
505 argv
, sizeof argv
/ sizeof *argv
, 0);
507 if ((argv
[i
++] = (char *)a0
) != NULL
&&
508 (argv
[i
++] = (char *)a1
) != NULL
&&
509 (argv
[i
++] = (char *)a2
) != NULL
)
511 prepare_child(mask
, infd
, outfd
);
512 execvp(argv
[0], argv
);
520 prepare_child(sigset_t
*nset
, int infd
, int outfd
)
526 * All file descriptors other than 0, 1, and 2 are supposed to be
534 for (i
= 1; i
< NSIG
; i
++)
535 if (sigismember(nset
, i
))
536 safe_signal(i
, SIG_IGN
);
537 if (!sigismember(nset
, SIGINT
))
538 safe_signal(SIGINT
, SIG_DFL
);
541 sigprocmask(SIG_UNBLOCK
, &fset
, (sigset_t
*)NULL
);
545 wait_command(int pid
)
548 if (wait_child(pid
) < 0 && (value("bsdcompat") || value("bsdmsgs"))) {
549 printf(catgets(catd
, CATSET
, 154, "Fatal error in process.\n"));
555 static struct child
*
560 for (cpp
= &child
; *cpp
!= (struct child
*)NULL
&& (*cpp
)->pid
!= pid
;
563 if (*cpp
== (struct child
*)NULL
) {
564 *cpp
= (struct child
*) smalloc(sizeof (struct child
));
566 (*cpp
)->done
= (*cpp
)->free
= 0;
567 (*cpp
)->link
= (struct child
*)NULL
;
573 delchild(struct child
*cp
)
577 for (cpp
= &child
; *cpp
!= cp
; cpp
= &(*cpp
)->link
)
593 while ((pid
= waitpid(-1, (int*)&status
, WNOHANG
)) > 0) {
602 if (pid
== -1 && errno
== EINTR
)
609 * Mark a child as don't care.
615 struct child
*cp
= findchild(pid
);
617 sigaddset(&nset
, SIGCHLD
);
618 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
624 sigprocmask(SIG_SETMASK
, &oset
, (sigset_t
*)NULL
);
628 * Wait for a specific child to die.
632 * This version is correct code, but causes harm on some loosing
633 * systems. So we use the second one instead.
639 struct child
*cp
= findchild(pid
);
641 sigaddset(&nset
, SIGCHLD
);
642 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
646 wait_status
= cp
->status
;
648 sigprocmask(SIG_SETMASK
, &oset
, (sigset_t
*)NULL
);
650 if (WIFEXITED(wait_status
) && (WEXITSTATUS(wait_status
) == 0))
660 struct sigaction nact
, oact
;
662 nact
.sa_handler
= SIG_DFL
;
663 sigemptyset(&nact
.sa_mask
);
664 nact
.sa_flags
= SA_NOCLDSTOP
;
665 sigaction(SIGCHLD
, &nact
, &oact
);
670 term
= wait(&wait_status
);
671 if (term
== -1 && errno
== EINTR
)
673 if (term
== 0 || term
== -1)
675 cp
= findchild(term
);
676 if (cp
->free
|| term
== pid
) {
680 cp
->status
= wait_status
;
682 } while (term
!= pid
);
684 wait_status
= cp
->status
;
688 sigaction(SIGCHLD
, &oact
, NULL
);
690 * Make sure no zombies are left.
694 if (WIFEXITED(wait_status
) && (WEXITSTATUS(wait_status
) == 0))