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) fcntl((FD), F_SETFD, FD_CLOEXEC)
56 # define _SET_CLOEXEC(FD)
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 fprintf(stderr
, 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
? 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
) < 0)
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
) == 0)
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 ;} */
251 panic(tr(153, "Invalid file pointer"));
265 for (p
= fp_head
; p
; p
= p
->link
)
275 wait_command(int pid
)
280 if (!wait_child(pid
, NULL
)) {
281 if (ok_blook(bsdcompat
) || ok_blook(bsdmsgs
))
282 fprintf(stderr
, tr(154, "Fatal error in process.\n"));
289 static struct child
*
295 for (cpp
= &_popen_child
; *cpp
!= NULL
&& (*cpp
)->pid
!= pid
;
299 *cpp
= smalloc(sizeof **cpp
);
301 (*cpp
)->done
= (*cpp
)->free
= 0;
309 delchild(struct child
*cp
)
314 for (cpp
= &_popen_child
; *cpp
!= cp
; cpp
= &(*cpp
)->link
)
322 safe_fopen(char const *file
, char const *oflags
, int *xflags
)
328 if (scan_mode(oflags
, &osflags
) < 0)
330 osflags
|= O_CLOEXEC
;
334 if ((fd
= open(file
, osflags
, 0666)) < 0)
338 fp
= fdopen(fd
, oflags
);
345 Fopen(char const *file
, char const *oflags
)
351 if ((fp
= safe_fopen(file
, oflags
, &osflags
)) != NULL
)
352 register_file(fp
, osflags
, 0, 0, FP_RAW
, NULL
, 0L);
358 Fdopen(int fd
, char const *oflags
)
364 scan_mode(oflags
, &osflags
);
365 osflags
|= O_CLOEXEC
;
367 if ((fp
= fdopen(fd
, oflags
)) != NULL
)
368 register_file(fp
, osflags
, 0, 0, FP_RAW
, NULL
, 0L);
379 if (unregister_file(fp
) == OKAY
)
384 return i
== 3 ? 0 : EOF
;
388 Zopen(char const *file
, char const *oflags
, int *compression
) /* FIXME MESS! */
391 int _compression
, osflags
, mode
, infd
;
397 if (compression
== NULL
)
398 compression
= &_compression
;
400 if (scan_mode(oflags
, &osflags
) < 0)
402 rof
= OF_RDWR
| OF_UNLINK
;
403 if (osflags
& O_APPEND
)
405 if (osflags
== O_RDONLY
) {
407 *compression
= FP_READONLY
;
414 if ((osflags
& O_APPEND
) && ((p
= which_protocol(file
)) == PROTO_IMAP
||
415 p
== PROTO_MAILDIR
)) {
416 *compression
|= (p
== PROTO_IMAP
) ? FP_IMAP
: FP_MAILDIR
;
417 osflags
= O_RDWR
| O_APPEND
| O_CREAT
;
422 if ((ext
= strrchr(file
, '.')) != NULL
) {
423 if (!strcmp(ext
, ".gz"))
424 *compression
|= FP_GZIP
;
425 else if (!strcmp(ext
, ".xz"))
426 *compression
|= FP_XZ
;
427 else if (!strcmp(ext
, ".bz2"))
428 *compression
|= FP_BZIP2
;
433 *compression
|= FP_RAW
;
434 rv
= Fopen(file
, oflags
);
437 if ((infd
= open(file
, (mode
& W_OK
) ? O_RDWR
: O_RDONLY
)) == -1 &&
438 (!(osflags
& O_CREAT
) || errno
!= ENOENT
))
442 if ((rv
= Ftmp(NULL
, "zopen", rof
, 0600)) == NULL
) {
443 perror(tr(167, "tmpfile"));
446 if (infd
>= 0 || (*compression
& FP_MASK
) == FP_IMAP
||
447 (*compression
& FP_MASK
) == FP_MAILDIR
) {
448 if (_decompress(*compression
, infd
, fileno(rv
)) < 0) {
458 if ((infd
= creat(file
, 0666)) == -1) {
468 if (!(osflags
& O_APPEND
))
470 if ((offset
= ftell(rv
)) == -1) {
475 register_file(rv
, osflags
, 0, 0, *compression
, file
, offset
);
482 Ftmp(char **fn
, char const *prefix
, enum oflags oflags
, int mode
)
490 cp
= smalloc(strlen(tempdir
) + 1 + sizeof("mail") + strlen(prefix
) + 7 +1);
491 cp
= sstpcpy(cp
, tempdir
);
493 cp
= sstpcpy(cp
, "mail");
496 cp
= sstpcpy(cp
, prefix
);
498 /* TODO Ftmp(): unroll our own creation loop with atoi(random()) */
499 sstpcpy(cp
, ".XXXXXX");
503 if ((fd
= mkstemp(cp_base
)) == -1)
505 if (mode
!= (S_IRUSR
| S_IWUSR
) && fchmod(fd
, mode
) == -1)
507 if (oflags
& OF_APPEND
) {
510 if ((f
= fcntl(fd
, F_GETFL
)) == -1 ||
511 fcntl(fd
, F_SETFL
, f
| O_APPEND
) == -1) {
517 if (!(oflags
& OF_REGISTER
))
518 fcntl(fd
, F_SETFD
, FD_CLOEXEC
);
520 if (mktemp(cp_base
) == NULL
)
522 if ((fd
= open(cp_base
, O_CREAT
| O_EXCL
| O_RDWR
| O_CLOEXEC
|
523 (oflags
& OF_APPEND
? O_APPEND
: 0), mode
)) < 0)
527 fp
= (*((oflags
& OF_REGISTER
) ? &Fdopen
: &fdopen
))(fd
,
528 (oflags
& OF_RDWR
? "w+" : "w"));
529 if (fp
== NULL
|| (oflags
& OF_UNLINK
)) {
538 if (fp
== NULL
|| !(oflags
& OF_HOLDSIGS
))
543 if ((cp
= cp_base
) != NULL
)
549 Ftmp_release(char **fn
)
578 pipe_cloexec(int fd
[2])
585 fcntl(fd
[0], F_SETFD
, FD_CLOEXEC
);
586 fcntl(fd
[1], F_SETFD
, FD_CLOEXEC
);
594 Popen(char const *cmd
, char const *mode
, char const *sh
, int newfd1
)
596 int p
[2], myside
, hisside
, fd0
, fd1
, pid
;
597 char mod
[2] = { '0', '\0' };
602 if (!pipe_cloexec(p
))
608 hisside
= fd1
= p
[WRITE
];
610 } else if (*mode
== 'W') {
612 hisside
= fd0
= p
[READ
];
617 hisside
= fd0
= p
[READ
];
623 pid
= start_command(cmd
, &nset
, fd0
, fd1
, NULL
, NULL
, NULL
);
625 pid
= start_command(sh
, &nset
, fd0
, fd1
, "-c", cmd
, NULL
);
633 if ((rv
= fdopen(myside
, mod
)) != NULL
)
634 register_file(rv
, 0, 1, pid
, FP_RAW
, NULL
, 0L);
641 Pclose(FILE *ptr
, bool_t dowait
)
651 unregister_file(ptr
);
655 sigaddset(&nset
, SIGINT
);
656 sigaddset(&nset
, SIGHUP
);
657 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
658 rv
= wait_child(pid
, NULL
);
659 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
670 close_all_files(void)
673 while (fp_head
!= NULL
)
675 Pclose(fp_head
->fp
, TRU1
);
682 run_command(char const *cmd
, sigset_t
*mask
, int infd
, int outfd
,
683 char const *a0
, char const *a1
, char const *a2
)
688 if ((rv
= start_command(cmd
, mask
, infd
, outfd
, a0
, a1
, a2
)) < 0)
691 rv
= wait_command(rv
);
697 start_command(char const *cmd
, sigset_t
*mask
, int infd
, int outfd
,
698 char const *a0
, char const *a1
, char const *a2
)
703 if ((rv
= fork()) < 0) {
706 } else if (rv
== 0) {
708 int i
= getrawlist(cmd
, strlen(cmd
), argv
, NELEM(argv
), 0);
710 if ((argv
[i
++] = UNCONST(a0
)) != NULL
&&
711 (argv
[i
++] = UNCONST(a1
)) != NULL
&&
712 (argv
[i
++] = UNCONST(a2
)) != NULL
)
714 prepare_child(mask
, infd
, outfd
);
715 execvp(argv
[0], argv
);
724 prepare_child(sigset_t
*nset
, int infd
, int outfd
)
730 /* All file descriptors other than 0, 1, and 2 are supposed to be cloexec */
737 for (i
= 1; i
< NSIG
; ++i
)
738 if (sigismember(nset
, i
))
739 safe_signal(i
, SIG_IGN
);
740 if (!sigismember(nset
, SIGINT
))
741 safe_signal(SIGINT
, SIG_DFL
);
745 sigprocmask(SIG_SETMASK
, &fset
, NULL
);
754 NYD_X
; /* Signal handler */
758 while ((pid
= waitpid(-1, &status
, WNOHANG
)) > 0) {
767 if (pid
== -1 && errno
== EINTR
)
779 sigaddset(&nset
, SIGCHLD
);
780 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
788 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
793 wait_child(int pid
, int *wait_status
)
802 sigaddset(&nset
, SIGCHLD
);
803 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
811 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
813 if (wait_status
!= NULL
)
815 rv
= (WIFEXITED(ws
) && WEXITSTATUS(ws
) == 0);
826 /* vim:set fenc=utf-8:s-it-mode */