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
);
99 static void _sigchld(int signo
);
101 static int wait_command(int pid
);
102 static struct child
*findchild(int pid
);
103 static void delchild(struct child
*cp
);
106 scan_mode(char const *mode
, int *omode
)
113 {"w", O_WRONLY
| O_CREAT
| O_TRUNC
},
114 {"wx", O_WRONLY
| O_CREAT
| O_EXCL
},
115 {"a", O_WRONLY
| O_APPEND
| O_CREAT
},
116 {"a+", O_RDWR
| O_APPEND
},
118 {"w+", O_RDWR
| O_CREAT
| O_EXCL
}
124 for (i
= 0; UICMP(z
, i
, <, NELEM(maps
)); ++i
)
125 if (!strcmp(maps
[i
].mode
, mode
)) {
126 *omode
= maps
[i
].omode
;
131 alert(tr(152, "Internal error: bad stdio open mode %s\n"), mode
);
133 *omode
= 0; /* (silence CC) */
141 register_file(FILE *fp
, int omode
, int ispipe
, int pid
, int compressed
,
142 char const *realfile
, long offset
)
147 fpp
= smalloc(sizeof *fpp
);
153 fpp
->compressed
= compressed
;
154 fpp
->realfile
= (realfile
!= NULL
) ? sstrdup(realfile
) : NULL
;
155 fpp
->offset
= offset
;
161 _compress(struct fp
*fpp
)
168 if (fpp
->omode
== O_RDONLY
) {
176 if (fseek(fpp
->fp
, fpp
->offset
, SEEK_SET
) == -1)
180 if ((fpp
->compressed
& FP_MASK
) == FP_IMAP
) {
181 rv
= imap_append(fpp
->realfile
, fpp
->fp
);
185 if ((fpp
->compressed
& FP_MASK
) == FP_MAILDIR
) {
186 rv
= maildir_append(fpp
->realfile
, fpp
->fp
);
190 outfd
= open(fpp
->realfile
, (fpp
->omode
| O_CREAT
) & ~O_EXCL
, 0666);
192 fprintf(stderr
, "Fatal: cannot create ");
193 perror(fpp
->realfile
);
196 if (!(fpp
->omode
& O_APPEND
))
198 switch (fpp
->compressed
& FP_MASK
) {
200 cmd
[0] = "gzip"; cmd
[1] = "-c"; break;
202 cmd
[0] = "bzip2"; cmd
[1] = "-c"; break;
204 cmd
[0] = "xz"; cmd
[1] = "-c"; break;
206 cmd
[0] = "cat"; cmd
[1] = NULL
; break;
208 if (run_command(cmd
[0], 0, fileno(fpp
->fp
), outfd
, cmd
[1], NULL
, NULL
) >= 0)
217 _decompress(int compression
, int infd
, int outfd
)
223 switch (compression
& FP_MASK
) {
224 case FP_GZIP
: cmd
[0] = "gzip"; cmd
[1] = "-cd"; break;
225 case FP_BZIP2
: cmd
[0] = "bzip2"; cmd
[1] = "-cd"; break;
226 case FP_XZ
: cmd
[0] = "xz"; cmd
[1] = "-cd"; break;
227 default: cmd
[0] = "cat"; cmd
[1] = NULL
; break;
233 rv
= run_command(cmd
[0], 0, infd
, outfd
, cmd
[1], NULL
, NULL
);
240 unregister_file(FILE *fp
)
246 for (pp
= &fp_head
; (p
= *pp
) != NULL
; pp
= &p
->link
)
248 if ((p
->compressed
& FP_MASK
) != FP_RAW
) /* TODO ;} */
259 (tr(153, "Invalid file pointer"));
274 for (p
= fp_head
; p
; p
= p
->link
)
288 NYD_X
; /* Signal handler */
292 pid
= (int)waitpid((pid_t
)-1, &status
, WNOHANG
);
294 if (pid
== -1 && errno
== EINTR
)
309 wait_command(int pid
)
314 if (!wait_child(pid
, NULL
)) {
315 if (ok_blook(bsdcompat
) || ok_blook(bsdmsgs
))
316 fprintf(stderr
, tr(154, "Fatal error in process.\n"));
323 static struct child
*
329 for (cpp
= &_popen_child
; *cpp
!= NULL
&& (*cpp
)->pid
!= pid
;
333 *cpp
= smalloc(sizeof **cpp
);
335 (*cpp
)->done
= (*cpp
)->free
= 0;
343 delchild(struct child
*cp
)
348 for (cpp
= &_popen_child
; *cpp
!= cp
; cpp
= &(*cpp
)->link
)
356 command_manager_start(void)
358 struct sigaction nact
, oact
;
361 nact
.sa_handler
= &_sigchld
;
362 sigemptyset(&nact
.sa_mask
);
371 if (sigaction(SIGCHLD
, &nact
, &oact
) != 0)
372 panic("Cannot install signal handler for child process management");
377 safe_fopen(char const *file
, char const *oflags
, int *xflags
)
383 if (scan_mode(oflags
, &osflags
) < 0)
385 osflags
|= O_CLOEXEC
;
389 if ((fd
= open(file
, osflags
, 0666)) == -1)
393 fp
= fdopen(fd
, oflags
);
400 Fopen(char const *file
, char const *oflags
)
406 if ((fp
= safe_fopen(file
, oflags
, &osflags
)) != NULL
)
407 register_file(fp
, osflags
, 0, 0, FP_RAW
, NULL
, 0L);
413 Fdopen(int fd
, char const *oflags
)
419 scan_mode(oflags
, &osflags
);
420 osflags
|= O_CLOEXEC
;
422 if ((fp
= fdopen(fd
, oflags
)) != NULL
)
423 register_file(fp
, osflags
, 0, 0, FP_RAW
, NULL
, 0L);
434 if (unregister_file(fp
) == OKAY
)
439 return (i
== 3 ? 0 : EOF
);
443 Zopen(char const *file
, char const *oflags
, int *compression
) /* FIXME MESS! */
446 int _compression
, osflags
, mode
, infd
;
452 if (compression
== NULL
)
453 compression
= &_compression
;
455 if (scan_mode(oflags
, &osflags
) < 0)
457 rof
= OF_RDWR
| OF_UNLINK
;
458 if (osflags
& O_APPEND
)
460 if (osflags
== O_RDONLY
) {
462 *compression
= FP_READONLY
;
469 if ((osflags
& O_APPEND
) && ((p
= which_protocol(file
)) == PROTO_IMAP
||
470 p
== PROTO_MAILDIR
)) {
471 *compression
|= (p
== PROTO_IMAP
) ? FP_IMAP
: FP_MAILDIR
;
472 osflags
= O_RDWR
| O_APPEND
| O_CREAT
;
477 if ((ext
= strrchr(file
, '.')) != NULL
) {
478 if (!strcmp(ext
, ".gz"))
479 *compression
|= FP_GZIP
;
480 else if (!strcmp(ext
, ".xz"))
481 *compression
|= FP_XZ
;
482 else if (!strcmp(ext
, ".bz2"))
483 *compression
|= FP_BZIP2
;
488 *compression
|= FP_RAW
;
489 rv
= Fopen(file
, oflags
);
492 if ((infd
= open(file
, (mode
& W_OK
) ? O_RDWR
: O_RDONLY
)) == -1 &&
493 (!(osflags
& O_CREAT
) || errno
!= ENOENT
))
497 if ((rv
= Ftmp(NULL
, "zopen", rof
, 0600)) == NULL
) {
498 perror(tr(167, "tmpfile"));
501 if (infd
>= 0 || (*compression
& FP_MASK
) == FP_IMAP
||
502 (*compression
& FP_MASK
) == FP_MAILDIR
) {
503 if (_decompress(*compression
, infd
, fileno(rv
)) < 0) {
513 if ((infd
= creat(file
, 0666)) == -1) {
523 if (!(osflags
& O_APPEND
))
525 if ((offset
= ftell(rv
)) == -1) {
530 register_file(rv
, osflags
, 0, 0, *compression
, file
, offset
);
537 Ftmp(char **fn
, char const *prefix
, enum oflags oflags
, int mode
)
545 cp
= smalloc(strlen(tempdir
) + 1 + sizeof("mail") + strlen(prefix
) + 7 +1);
546 cp
= sstpcpy(cp
, tempdir
);
548 cp
= sstpcpy(cp
, "mail");
551 cp
= sstpcpy(cp
, prefix
);
553 /* TODO Ftmp(): unroll our own creation loop with atoi(random()) */
554 sstpcpy(cp
, ".XXXXXX");
558 if ((fd
= mkstemp(cp_base
)) == -1)
560 if (mode
!= (S_IRUSR
| S_IWUSR
) && fchmod(fd
, mode
) == -1)
562 if (oflags
& OF_APPEND
) {
565 if ((f
= fcntl(fd
, F_GETFL
)) == -1 ||
566 fcntl(fd
, F_SETFL
, f
| O_APPEND
) == -1) {
572 if (!(oflags
& OF_REGISTER
))
573 fcntl(fd
, F_SETFD
, FD_CLOEXEC
);
575 if (mktemp(cp_base
) == NULL
)
577 if ((fd
= open(cp_base
, O_CREAT
| O_EXCL
| O_RDWR
| O_CLOEXEC
|
578 (oflags
& OF_APPEND
? O_APPEND
: 0), mode
)) == -1)
580 if (!(oflags
& OF_REGISTER
))
584 fp
= (*((oflags
& OF_REGISTER
) ? &Fdopen
: &fdopen
))(fd
,
585 (oflags
& OF_RDWR
? "w+" : "w"));
586 if (fp
== NULL
|| (oflags
& OF_UNLINK
)) {
595 if (fp
== NULL
|| !(oflags
& OF_HOLDSIGS
))
600 if ((cp
= cp_base
) != NULL
)
606 Ftmp_release(char **fn
)
635 pipe_cloexec(int fd
[2])
641 if (pipe2(fd
, O_CLOEXEC
) == -1)
646 fcntl(fd
[0], F_SETFD
, FD_CLOEXEC
);
647 fcntl(fd
[1], F_SETFD
, FD_CLOEXEC
);
656 Popen(char const *cmd
, char const *mode
, char const *sh
, int newfd1
)
658 int p
[2], myside
, hisside
, fd0
, fd1
, pid
;
659 char mod
[2] = {'0', '\0'};
664 if (!pipe_cloexec(p
))
670 hisside
= fd1
= p
[WRITE
];
672 } else if (*mode
== 'W') {
674 hisside
= fd0
= p
[READ
];
679 hisside
= fd0
= p
[READ
];
685 pid
= start_command(cmd
, &nset
, fd0
, fd1
, NULL
, NULL
, NULL
);
687 pid
= start_command(sh
, &nset
, fd0
, fd1
, "-c", cmd
, NULL
);
695 if ((rv
= fdopen(myside
, mod
)) != NULL
)
696 register_file(rv
, 0, 1, pid
, FP_RAW
, NULL
, 0L);
703 Pclose(FILE *ptr
, bool_t dowait
)
713 unregister_file(ptr
);
717 sigaddset(&nset
, SIGINT
);
718 sigaddset(&nset
, SIGHUP
);
719 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
720 rv
= wait_child(pid
, NULL
);
721 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
732 close_all_files(void)
735 while (fp_head
!= NULL
)
737 Pclose(fp_head
->fp
, TRU1
);
744 run_command(char const *cmd
, sigset_t
*mask
, int infd
, int outfd
,
745 char const *a0
, char const *a1
, char const *a2
)
750 if ((rv
= start_command(cmd
, mask
, infd
, outfd
, a0
, a1
, a2
)) < 0)
753 rv
= wait_command(rv
);
759 start_command(char const *cmd
, sigset_t
*mask
, int infd
, int outfd
,
760 char const *a0
, char const *a1
, char const *a2
)
765 if ((rv
= fork()) == -1) {
768 } else if (rv
== 0) {
770 int i
= getrawlist(cmd
, strlen(cmd
), argv
, NELEM(argv
), 0);
772 if ((argv
[i
++] = UNCONST(a0
)) != NULL
&&
773 (argv
[i
++] = UNCONST(a1
)) != NULL
&&
774 (argv
[i
++] = UNCONST(a2
)) != NULL
)
776 prepare_child(mask
, infd
, outfd
);
777 execvp(argv
[0], argv
);
786 prepare_child(sigset_t
*nset
, int infd
, int outfd
)
792 /* All file descriptors other than 0, 1, and 2 are supposed to be cloexec */
799 for (i
= 1; i
< NSIG
; ++i
)
800 if (sigismember(nset
, i
))
801 safe_signal(i
, SIG_IGN
);
802 if (!sigismember(nset
, SIGINT
))
803 safe_signal(SIGINT
, SIG_DFL
);
807 sigprocmask(SIG_SETMASK
, &fset
, NULL
);
819 sigaddset(&nset
, SIGCHLD
);
820 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
828 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
833 wait_child(int pid
, int *wait_status
)
842 sigaddset(&nset
, SIGCHLD
);
843 sigprocmask(SIG_BLOCK
, &nset
, &oset
);
851 sigprocmask(SIG_SETMASK
, &oset
, NULL
);
853 if (wait_status
!= NULL
)
855 rv
= (WIFEXITED(ws
) && WEXITSTATUS(ws
) == 0);
866 /* vim:set fenc=utf-8:s-it-mode */