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 - 2014 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
40 #ifndef HAVE_AMALGAMATION
54 # define _SET_CLOEXEC(FD) do fcntl((FD), F_SETFD, FD_CLOEXEC); while (0)
56 # define _SET_CLOEXEC(FD) do {} while (0)
87 static struct fp
*fp_head
;
88 static struct child
*_popen_child
;
90 static int scan_mode(char const *mode
, int *omode
);
91 static void register_file(FILE *fp
, int omode
, int ispipe
, int pid
,
92 int compressed
, char const *realfile
, long offset
);
93 static enum okay
_compress(struct fp
*fpp
);
94 static int _decompress(int compression
, int infd
, int outfd
);
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 scan_mode(char const *mode
, int *omode
)
109 {"w", O_WRONLY
| O_CREAT
| O_TRUNC
},
110 {"wx", O_WRONLY
| O_CREAT
| O_EXCL
},
111 {"a", O_WRONLY
| O_APPEND
| O_CREAT
},
112 {"a+", O_RDWR
| O_APPEND
},
114 {"w+", O_RDWR
| O_CREAT
| O_EXCL
}
120 for (i
= 0; UICMP(z
, i
, <, NELEM(maps
)); ++i
)
121 if (!strcmp(maps
[i
].mode
, mode
)) {
122 *omode
= maps
[i
].omode
;
127 alert(tr(152, "Internal error: bad stdio open mode %s\n"), mode
);
129 *omode
= 0; /* (silence CC) */
137 register_file(FILE *fp
, int omode
, int ispipe
, int pid
, int compressed
,
138 char const *realfile
, long offset
)
143 fpp
= smalloc(sizeof *fpp
);
149 fpp
->compressed
= compressed
;
150 fpp
->realfile
= (realfile
!= NULL
) ? sstrdup(realfile
) : NULL
;
151 fpp
->offset
= offset
;
157 _compress(struct fp
*fpp
)
164 if (fpp
->omode
== O_RDONLY
) {
172 if (fseek(fpp
->fp
, fpp
->offset
, SEEK_SET
) == -1)
176 if ((fpp
->compressed
& FP_MASK
) == FP_IMAP
) {
177 rv
= imap_append(fpp
->realfile
, fpp
->fp
);
181 if ((fpp
->compressed
& FP_MASK
) == FP_MAILDIR
) {
182 rv
= maildir_append(fpp
->realfile
, fpp
->fp
);
186 outfd
= open(fpp
->realfile
, (fpp
->omode
| O_CREAT
) & ~O_EXCL
, 0666);
188 fprintf(stderr
, "Fatal: cannot create ");
189 perror(fpp
->realfile
);
192 if (!(fpp
->omode
& O_APPEND
))
194 switch (fpp
->compressed
& FP_MASK
) {
196 cmd
[0] = "gzip"; cmd
[1] = "-c"; break;
198 cmd
[0] = "bzip2"; cmd
[1] = "-c"; break;
200 cmd
[0] = "xz"; cmd
[1] = "-c"; break;
202 cmd
[0] = "cat"; cmd
[1] = NULL
; break;
204 if (run_command(cmd
[0], 0, fileno(fpp
->fp
), outfd
, cmd
[1], NULL
, NULL
) >= 0)
213 _decompress(int compression
, int infd
, int outfd
)
219 switch (compression
& FP_MASK
) {
220 case FP_GZIP
: cmd
[0] = "gzip"; cmd
[1] = "-cd"; break;
221 case FP_BZIP2
: cmd
[0] = "bzip2"; cmd
[1] = "-cd"; break;
222 case FP_XZ
: cmd
[0] = "xz"; cmd
[1] = "-cd"; break;
223 default: cmd
[0] = "cat"; cmd
[1] = NULL
; break;
229 rv
= run_command(cmd
[0], 0, infd
, outfd
, cmd
[1], NULL
, NULL
);
236 unregister_file(FILE *fp
)
242 for (pp
= &fp_head
; (p
= *pp
) != NULL
; pp
= &p
->link
)
244 if ((p
->compressed
& FP_MASK
) != FP_RAW
) /* TODO ;} */
255 (tr(153, "Invalid file pointer"));
270 for (p
= fp_head
; p
; p
= p
->link
)
280 wait_command(int pid
)
285 if (!wait_child(pid
, NULL
)) {
286 if (ok_blook(bsdcompat
) || ok_blook(bsdmsgs
))
287 fprintf(stderr
, tr(154, "Fatal error in process.\n"));
294 static struct child
*
300 for (cpp
= &_popen_child
; *cpp
!= NULL
&& (*cpp
)->pid
!= pid
;
304 *cpp
= smalloc(sizeof **cpp
);
306 (*cpp
)->done
= (*cpp
)->free
= 0;
314 delchild(struct child
*cp
)
319 for (cpp
= &_popen_child
; *cpp
!= cp
; cpp
= &(*cpp
)->link
)
327 safe_fopen(char const *file
, char const *oflags
, int *xflags
)
333 if (scan_mode(oflags
, &osflags
) < 0)
335 osflags
|= O_CLOEXEC
;
339 if ((fd
= open(file
, osflags
, 0666)) == -1)
343 fp
= fdopen(fd
, oflags
);
350 Fopen(char const *file
, char const *oflags
)
356 if ((fp
= safe_fopen(file
, oflags
, &osflags
)) != NULL
)
357 register_file(fp
, osflags
, 0, 0, FP_RAW
, NULL
, 0L);
363 Fdopen(int fd
, char const *oflags
)
369 scan_mode(oflags
, &osflags
);
370 osflags
|= O_CLOEXEC
;
372 if ((fp
= fdopen(fd
, oflags
)) != NULL
)
373 register_file(fp
, osflags
, 0, 0, FP_RAW
, NULL
, 0L);
384 if (unregister_file(fp
) == OKAY
)
389 return (i
== 3 ? 0 : EOF
);
393 Zopen(char const *file
, char const *oflags
, int *compression
) /* FIXME MESS! */
396 int _compression
, osflags
, mode
, infd
;
402 if (compression
== NULL
)
403 compression
= &_compression
;
405 if (scan_mode(oflags
, &osflags
) < 0)
407 rof
= OF_RDWR
| OF_UNLINK
;
408 if (osflags
& O_APPEND
)
410 if (osflags
== O_RDONLY
) {
412 *compression
= FP_READONLY
;
419 if ((osflags
& O_APPEND
) && ((p
= which_protocol(file
)) == PROTO_IMAP
||
420 p
== PROTO_MAILDIR
)) {
421 *compression
|= (p
== PROTO_IMAP
) ? FP_IMAP
: FP_MAILDIR
;
422 osflags
= O_RDWR
| O_APPEND
| O_CREAT
;
427 if ((ext
= strrchr(file
, '.')) != NULL
) {
428 if (!strcmp(ext
, ".gz"))
429 *compression
|= FP_GZIP
;
430 else if (!strcmp(ext
, ".xz"))
431 *compression
|= FP_XZ
;
432 else if (!strcmp(ext
, ".bz2"))
433 *compression
|= FP_BZIP2
;
438 *compression
|= FP_RAW
;
439 rv
= Fopen(file
, oflags
);
442 if ((infd
= open(file
, (mode
& W_OK
) ? O_RDWR
: O_RDONLY
)) == -1 &&
443 (!(osflags
& O_CREAT
) || errno
!= ENOENT
))
447 if ((rv
= Ftmp(NULL
, "zopen", rof
, 0600)) == NULL
) {
448 perror(tr(167, "tmpfile"));
451 if (infd
>= 0 || (*compression
& FP_MASK
) == FP_IMAP
||
452 (*compression
& FP_MASK
) == FP_MAILDIR
) {
453 if (_decompress(*compression
, infd
, fileno(rv
)) < 0) {
463 if ((infd
= creat(file
, 0666)) == -1) {
473 if (!(osflags
& O_APPEND
))
475 if ((offset
= ftell(rv
)) == -1) {
480 register_file(rv
, osflags
, 0, 0, *compression
, file
, offset
);
487 Ftmp(char **fn
, char const *prefix
, enum oflags oflags
, int mode
)
495 cp
= smalloc(strlen(tempdir
) + 1 + sizeof("mail") + strlen(prefix
) + 7 +1);
496 cp
= sstpcpy(cp
, tempdir
);
498 cp
= sstpcpy(cp
, "mail");
501 cp
= sstpcpy(cp
, prefix
);
503 /* TODO Ftmp(): unroll our own creation loop with atoi(random()) */
504 sstpcpy(cp
, ".XXXXXX");
508 if ((fd
= mkstemp(cp_base
)) == -1)
510 if (mode
!= (S_IRUSR
| S_IWUSR
) && fchmod(fd
, mode
) == -1)
512 if (oflags
& OF_APPEND
) {
515 if ((f
= fcntl(fd
, F_GETFL
)) == -1 ||
516 fcntl(fd
, F_SETFL
, f
| O_APPEND
) == -1) {
522 if (!(oflags
& OF_REGISTER
))
523 fcntl(fd
, F_SETFD
, FD_CLOEXEC
);
525 if (mktemp(cp_base
) == NULL
)
527 if ((fd
= open(cp_base
, O_CREAT
| O_EXCL
| O_RDWR
| O_CLOEXEC
|
528 (oflags
& OF_APPEND
? O_APPEND
: 0), mode
)) == -1)
530 if (!(oflags
& OF_REGISTER
))
534 fp
= (*((oflags
& OF_REGISTER
) ? &Fdopen
: &fdopen
))(fd
,
535 (oflags
& OF_RDWR
? "w+" : "w"));
536 if (fp
== NULL
|| (oflags
& OF_UNLINK
)) {
545 if (fp
== NULL
|| !(oflags
& OF_HOLDSIGS
))
550 if ((cp
= cp_base
) != NULL
)
556 Ftmp_release(char **fn
)
585 pipe_cloexec(int fd
[2])
591 if (pipe2(fd
, O_CLOEXEC
) == -1)
596 fcntl(fd
[0], F_SETFD
, FD_CLOEXEC
);
597 fcntl(fd
[1], F_SETFD
, FD_CLOEXEC
);
606 Popen(char const *cmd
, char const *mode
, char const *sh
, int newfd1
)
608 int p
[2], myside
, hisside
, fd0
, fd1
, pid
;
609 char mod
[2] = {'0', '\0'};
614 if (!pipe_cloexec(p
))
620 hisside
= fd1
= p
[WRITE
];
622 } else if (*mode
== 'W') {
624 hisside
= fd0
= p
[READ
];
629 hisside
= fd0
= p
[READ
];
635 pid
= start_command(cmd
, &nset
, fd0
, fd1
, NULL
, NULL
, NULL
);
637 pid
= start_command(sh
, &nset
, fd0
, fd1
, "-c", cmd
, NULL
);
645 if ((rv
= fdopen(myside
, mod
)) != NULL
)
646 register_file(rv
, 0, 1, pid
, FP_RAW
, NULL
, 0L);
653 Pclose(FILE *ptr
, bool_t dowait
)
663 unregister_file(ptr
);
667 sigaddset(&nset
, SIGINT
);
668 sigaddset(&nset
, SIGHUP
);
669 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
670 rv
= wait_child(pid
, NULL
);
671 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
682 close_all_files(void)
685 while (fp_head
!= NULL
)
687 Pclose(fp_head
->fp
, TRU1
);
694 run_command(char const *cmd
, sigset_t
*mask
, int infd
, int outfd
,
695 char const *a0
, char const *a1
, char const *a2
)
700 if ((rv
= start_command(cmd
, mask
, infd
, outfd
, a0
, a1
, a2
)) < 0)
703 rv
= wait_command(rv
);
709 start_command(char const *cmd
, sigset_t
*mask
, int infd
, int outfd
,
710 char const *a0
, char const *a1
, char const *a2
)
715 if ((rv
= fork()) == -1) {
718 } else if (rv
== 0) {
720 int i
= getrawlist(cmd
, strlen(cmd
), argv
, NELEM(argv
), 0);
722 if ((argv
[i
++] = UNCONST(a0
)) != NULL
&&
723 (argv
[i
++] = UNCONST(a1
)) != NULL
&&
724 (argv
[i
++] = UNCONST(a2
)) != NULL
)
726 prepare_child(mask
, infd
, outfd
);
727 execvp(argv
[0], argv
);
736 prepare_child(sigset_t
*nset
, int infd
, int outfd
)
742 /* All file descriptors other than 0, 1, and 2 are supposed to be cloexec */
749 for (i
= 1; i
< NSIG
; ++i
)
750 if (sigismember(nset
, i
))
751 safe_signal(i
, SIG_IGN
);
752 if (!sigismember(nset
, SIGINT
))
753 safe_signal(SIGINT
, SIG_DFL
);
757 sigprocmask(SIG_SETMASK
, &fset
, NULL
);
766 NYD_X
; /* Signal handler */
770 pid
= waitpid(-1, &status
, WNOHANG
);
794 sigaddset(&nset
, SIGCHLD
);
795 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
803 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
808 wait_child(int pid
, int *wait_status
)
817 sigaddset(&nset
, SIGCHLD
);
818 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
826 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
828 if (wait_status
!= NULL
)
830 rv
= (WIFEXITED(ws
) && WEXITSTATUS(ws
) == 0);
841 /* vim:set fenc=utf-8:s-it-mode */