1 /*@ S-nail - a mail user agent derived from Berkeley Mail.
2 *@ Handling of pipes, child processes, temporary files, file enwrapping
4 * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
5 * Copyright (c) 2012 - 2013 Steffen "Daode" Nurpmeso <sdaoden@users.sf.net>.
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
67 static struct fp
*fp_head
;
76 static struct child
*child
;
78 static int scan_mode(const char *mode
, int *omode
);
79 static void register_file(FILE *fp
, int omode
, int ispipe
, int pid
,
80 int compressed
, const char *realfile
, long offset
);
81 static enum okay
compress(struct fp
*fpp
);
82 static int decompress(int compression
, int input
, int output
);
83 static enum okay
unregister_file(FILE *fp
);
84 static int file_pid(FILE *fp
);
85 static int wait_command(int pid
);
86 static struct child
*findchild(int pid
);
87 static void delchild(struct child
*cp
);
90 * Provide BSD-like signal() on all systems.
93 safe_signal(int signum
, sighandler_type handler
)
95 struct sigaction nact
, oact
;
97 nact
.sa_handler
= handler
;
98 sigemptyset(&nact
.sa_mask
);
101 nact
.sa_flags
|= SA_RESTART
;
103 if (sigaction(signum
, &nact
, &oact
) != 0)
105 return oact
.sa_handler
;
109 scan_mode(const char *mode
, int *omode
)
112 if (!strcmp(mode
, "r")) {
114 } else if (!strcmp(mode
, "w")) {
115 *omode
= O_WRONLY
| O_CREAT
| O_TRUNC
;
116 } else if (!strcmp(mode
, "wx")) {
117 *omode
= O_WRONLY
| O_CREAT
| O_EXCL
;
118 } else if (!strcmp(mode
, "a")) {
119 *omode
= O_WRONLY
| O_APPEND
| O_CREAT
;
120 } else if (!strcmp(mode
, "a+")) {
121 *omode
= O_RDWR
| O_APPEND
;
122 } else if (!strcmp(mode
, "r+")) {
124 } else if (!strcmp(mode
, "w+")) {
125 *omode
= O_RDWR
| O_CREAT
| O_EXCL
;
127 fprintf(stderr
, tr(152,
128 "Internal error: bad stdio open mode %s\n"), mode
);
136 safe_fopen(const char *file
, const char *mode
, int *omode
)
140 if (scan_mode(mode
, omode
) < 0)
142 if ((fd
= open(file
, *omode
, 0666)) < 0)
144 return fdopen(fd
, mode
);
148 Fopen(const char *file
, const char *mode
)
153 if ((fp
= safe_fopen(file
, mode
, &omode
)) != NULL
) {
154 (void)fcntl(fileno(fp
), F_SETFD
, FD_CLOEXEC
);
155 register_file(fp
, omode
, 0, 0, FP_UNCOMPRESSED
, NULL
, 0L);
161 Fdopen(int fd
, const char *mode
)
166 scan_mode(mode
, &omode
);
167 if ((fp
= fdopen(fd
, mode
)) != NULL
) {
168 (void)fcntl(fileno(fp
), F_SETFD
, FD_CLOEXEC
);
169 register_file(fp
, omode
, 0, 0, FP_UNCOMPRESSED
, NULL
, 0L);
178 if (unregister_file(fp
) == OKAY
)
182 return i
== 3 ? 0 : EOF
;
186 Zopen(const char *file
, const char *mode
, int *compression
) /* TODO MESS!
187 * TODO maybe we shouldn't be simple and run commands but instead
188 * TODO links against any of available zlib.h, bzlib.h, lzma.h!
189 * TODO even libzip? Much faster, not that much more work, and we're
190 * TODO *so* fixed anyway!!
191 * TODO *or*: make it all hookable via BOXEXTENSION/DECOMPRESS etc.
192 * TODO by the user; but not like this, Coverity went grazy about it */
205 if (scan_mode(mode
, &omode
) < 0)
207 if (compression
== NULL
)
208 compression
= &_compression
;
209 bits
= R_OK
| (omode
== O_RDONLY
? 0 : W_OK
);
210 if (omode
& O_APPEND
&& ((p
= which_protocol(file
)) == PROTO_IMAP
||
211 p
== PROTO_MAILDIR
)) {
212 *compression
= p
== PROTO_IMAP
? FP_IMAP
: FP_MAILDIR
;
213 omode
= O_RDWR
| O_APPEND
| O_CREAT
;
218 if ((extension
= strrchr(file
, '.')) != NULL
) {
220 if (strcmp(extension
, ".gz") == 0)
222 if (strcmp(extension
, ".bz2") == 0)
226 if (access(file
, F_OK
) == 0) {
227 *compression
= FP_UNCOMPRESSED
;
228 return Fopen(file
, mode
);
229 } else if (access(rp
=savecat(file
, ".gz"), bits
) == 0) {
230 gzip
: *compression
= FP_GZIPPED
;
231 } else if (access(rp
=savecat(file
, ".bz2"), bits
) == 0) {
232 bz2
: *compression
= FP_BZIP2ED
;
234 *compression
= FP_UNCOMPRESSED
;
235 return Fopen(file
, mode
);
237 if (access(rp
, W_OK
) < 0)
238 *compression
|= FP_READONLY
;
240 if ((input
= open(rp
, bits
& W_OK
? O_RDWR
: O_RDONLY
)) < 0
241 && ((omode
&O_CREAT
) == 0 || errno
!= ENOENT
))
243 open
: if ((output
= Ftemp(&tempfn
, "Rz", "w+", 0600, 0)) == NULL
) {
244 perror(tr(167, "tmpfile"));
252 if (input
>= 0 || (*compression
&FP_MASK
) == FP_IMAP
||
253 (*compression
&FP_MASK
) == FP_MAILDIR
) {
254 if (decompress(*compression
, input
, fileno(output
)) < 0) {
261 if ((input
= creat(rp
, 0666)) < 0) {
270 if (omode
& O_APPEND
) {
273 if ((flags
= fcntl(fileno(output
), F_GETFL
)) < 0 ||
274 fcntl(fileno(output
), F_SETFL
, flags
|O_APPEND
)
275 < 0 || (offset
= ftell(output
)) < 0) {
284 register_file(output
, omode
, 0, 0, *compression
, rp
, offset
);
289 Ftemp(char **fn
, char const *prefix
, char const *mode
, int bits
,
297 cp
= smalloc(strlen(tempdir
) + 1 + sizeof("mail") + strlen(prefix
) +
299 cp
= sstpcpy(cp
, tempdir
);
301 cp
= sstpcpy(cp
, "mail");
304 cp
= sstpcpy(cp
, prefix
);
306 sstpcpy(cp
, ".XXXXXX");
309 if ((fd
= mkstemp(*fn
)) < 0)
311 if (bits
!= (S_IRUSR
|S_IWUSR
) && fchmod(fd
, bits
) < 0)
314 if (mktemp(*fn
) == NULL
)
316 if ((fd
= open(*fn
, O_CREAT
|O_EXCL
|O_RDWR
, bits
)) < 0)
321 fp
= Fdopen(fd
, mode
);
323 (void)fcntl(fd
, F_SETFD
, FD_CLOEXEC
);
324 fp
= fdopen(fd
, mode
);
343 pipe_cloexec(int fd
[2])
349 fcntl(fd
[0], F_SETFD
, FD_CLOEXEC
);
350 fcntl(fd
[1], F_SETFD
, FD_CLOEXEC
);
357 Popen(const char *cmd
, const char *mode
, const char *sh
, int newfd1
)
360 int myside
, hisside
, fd0
, fd1
;
362 char mod
[2] = { '0', '\0' };
366 if (! pipe_cloexec(p
))
372 hisside
= fd1
= p
[WRITE
];
374 } else if (*mode
== 'W') {
376 hisside
= fd0
= p
[READ
];
381 hisside
= fd0
= p
[READ
];
387 pid
= start_command(cmd
, &nset
, fd0
, fd1
, NULL
, NULL
, NULL
);
389 pid
= start_command(sh
, &nset
, fd0
, fd1
, "-c", cmd
, NULL
);
397 if ((fp
= fdopen(myside
, mod
)) != NULL
)
398 register_file(fp
, 0, 1, pid
, FP_UNCOMPRESSED
, NULL
, 0L);
403 Pclose(FILE *ptr
, bool_t dowait
)
411 unregister_file(ptr
);
415 sigaddset(&nset
, SIGINT
);
416 sigaddset(&nset
, SIGHUP
);
417 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
418 pid
= wait_child(pid
);
419 sigprocmask(SIG_SETMASK
, &oset
, (sigset_t
*)NULL
);
426 close_all_files(void)
430 Pclose(fp_head
->fp
, TRU1
);
436 register_file(FILE *fp
, int omode
, int ispipe
, int pid
, int compressed
,
437 const char *realfile
, long offset
)
441 fpp
= (struct fp
*)smalloc(sizeof *fpp
);
447 fpp
->compressed
= compressed
;
448 fpp
->realfile
= realfile
? sstrdup(realfile
) : NULL
;
449 fpp
->offset
= offset
;
454 compress(struct fp
*fpp
)
457 char const *command
[2];
460 if (fpp
->omode
== O_RDONLY
)
464 if (fseek(fpp
->fp
, fpp
->offset
, SEEK_SET
) < 0)
467 if ((fpp
->compressed
&FP_MASK
) == FP_IMAP
) {
468 return imap_append(fpp
->realfile
, fpp
->fp
);
471 if ((fpp
->compressed
&FP_MASK
) == FP_MAILDIR
) {
472 return maildir_append(fpp
->realfile
, fpp
->fp
);
474 if ((output
= open(fpp
->realfile
,
475 (fpp
->omode
|O_CREAT
)&~O_EXCL
,
477 fprintf(stderr
, "Fatal: cannot create ");
478 perror(fpp
->realfile
);
481 if ((fpp
->omode
& O_APPEND
) == 0)
482 ftruncate(output
, 0);
483 switch (fpp
->compressed
& FP_MASK
) {
485 command
[0] = "gzip"; command
[1] = "-c"; break;
487 command
[0] = "bzip2"; command
[1] = "-c"; break;
489 command
[0] = "cat"; command
[1] = NULL
; break;
491 if (run_command(command
[0], 0, fileno(fpp
->fp
), output
,
492 command
[1], NULL
, NULL
) < 0)
501 decompress(int compression
, int input
, int output
)
503 char const *command
[2];
506 * Note that it is not possible to handle 'pack' or 'compress'
507 * formats because appending data does not work with them.
509 switch (compression
& FP_MASK
) {
510 case FP_GZIPPED
: command
[0] = "gzip"; command
[1] = "-cd"; break;
511 case FP_BZIP2ED
: command
[0] = "bzip2"; command
[1] = "-cd"; break;
512 case FP_IMAP
: return 0;
513 case FP_MAILDIR
: return 0;
514 default: command
[0] = "cat"; command
[1] = NULL
;
516 return run_command(command
[0], 0, input
, output
,
517 command
[1], NULL
, NULL
);
521 unregister_file(FILE *fp
)
526 for (pp
= &fp_head
; (p
= *pp
) != (struct fp
*)NULL
; pp
= &p
->link
)
528 if ((p
->compressed
&FP_MASK
) != FP_UNCOMPRESSED
)
534 panic(tr(153, "Invalid file pointer"));
544 for (p
= fp_head
; p
; p
= p
->link
)
551 * Run a command without a shell, with optional arguments and splicing
552 * of stdin and stdout. The command name can be a sequence of words.
553 * Signals must be handled by the caller.
554 * "Mask" contains the signals to ignore in the new process.
555 * SIGINT is enabled unless it's in the mask.
559 run_command(char const *cmd
, sigset_t
*mask
, int infd
, int outfd
,
560 char const *a0
, char const *a1
, char const *a2
)
564 if ((pid
= start_command(cmd
, mask
, infd
, outfd
, a0
, a1
, a2
)) < 0)
566 return wait_command(pid
);
571 start_command(const char *cmd
, sigset_t
*mask
, int infd
, int outfd
,
572 const char *a0
, const char *a1
, const char *a2
)
576 if ((pid
= fork()) < 0) {
582 int i
= getrawlist(cmd
, strlen(cmd
),
583 argv
, sizeof argv
/ sizeof *argv
, 0);
585 if ((argv
[i
++] = UNCONST(a0
)) != NULL
&&
586 (argv
[i
++] = UNCONST(a1
)) != NULL
&&
587 (argv
[i
++] = UNCONST(a2
)) != NULL
)
589 prepare_child(mask
, infd
, outfd
);
590 execvp(argv
[0], argv
);
598 prepare_child(sigset_t
*nset
, int infd
, int outfd
)
604 * All file descriptors other than 0, 1, and 2 are supposed to be
612 for (i
= 1; i
< NSIG
; i
++)
613 if (sigismember(nset
, i
))
614 safe_signal(i
, SIG_IGN
);
615 if (!sigismember(nset
, SIGINT
))
616 safe_signal(SIGINT
, SIG_DFL
);
619 sigprocmask(SIG_UNBLOCK
, &fset
, (sigset_t
*)NULL
);
623 wait_command(int pid
)
626 if (wait_child(pid
) < 0 && (value("bsdcompat") || value("bsdmsgs"))) {
627 printf(tr(154, "Fatal error in process.\n"));
633 static struct child
*
638 for (cpp
= &child
; *cpp
!= (struct child
*)NULL
&& (*cpp
)->pid
!= pid
;
641 if (*cpp
== (struct child
*)NULL
) {
642 *cpp
= (struct child
*) smalloc(sizeof (struct child
));
644 (*cpp
)->done
= (*cpp
)->free
= 0;
645 (*cpp
)->link
= (struct child
*)NULL
;
651 delchild(struct child
*cp
)
655 for (cpp
= &child
; *cpp
!= cp
; cpp
= &(*cpp
)->link
)
671 while ((pid
= waitpid(-1, (int*)&status
, WNOHANG
)) > 0) {
680 if (pid
== -1 && errno
== EINTR
)
687 * Mark a child as don't care.
693 struct child
*cp
= findchild(pid
);
695 sigaddset(&nset
, SIGCHLD
);
696 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
702 sigprocmask(SIG_SETMASK
, &oset
, (sigset_t
*)NULL
);
706 * Wait for a specific child to die.
710 * This version is correct code, but causes harm on some loosing
711 * systems. So we use the second one instead.
717 struct child
*cp
= findchild(pid
);
719 sigaddset(&nset
, SIGCHLD
);
720 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
724 wait_status
= cp
->status
;
726 sigprocmask(SIG_SETMASK
, &oset
, (sigset_t
*)NULL
);
728 if (WIFEXITED(wait_status
) && (WEXITSTATUS(wait_status
) == 0))
738 struct sigaction nact
, oact
;
740 nact
.sa_handler
= SIG_DFL
;
741 sigemptyset(&nact
.sa_mask
);
742 nact
.sa_flags
= SA_NOCLDSTOP
;
743 sigaction(SIGCHLD
, &nact
, &oact
);
748 term
= wait(&wait_status
);
749 if (term
== -1 && errno
== EINTR
)
751 if (term
== 0 || term
== -1)
753 cp
= findchild(term
);
754 if (cp
->free
|| term
== pid
) {
758 cp
->status
= wait_status
;
760 } while (term
!= pid
);
762 wait_status
= cp
->status
;
766 sigaction(SIGCHLD
, &oact
, NULL
);
768 * Make sure no zombies are left.
772 if (WIFEXITED(wait_status
) && (WEXITSTATUS(wait_status
) == 0))