NetBSD/usr.bin/mail/popen.c

402 lines
7.9 KiB
C
Raw Normal View History

/* $NetBSD: popen.c,v 1.6 1997/05/13 06:48:42 mikel Exp $ */
1993-03-21 12:45:37 +03:00
/*
1994-06-29 09:09:04 +04:00
* Copyright (c) 1980, 1993
* The Regents of the University of California. All rights reserved.
1993-03-21 12:45:37 +03:00
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the University of
* California, Berkeley and its contributors.
* 4. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#ifndef lint
#if 0
static char sccsid[] = "@(#)popen.c 8.1 (Berkeley) 6/6/93";
#else
static char rcsid[] = "$NetBSD: popen.c,v 1.6 1997/05/13 06:48:42 mikel Exp $";
#endif
1993-03-21 12:45:37 +03:00
#endif /* not lint */
#include "rcv.h"
#include <sys/wait.h>
1994-06-29 09:09:04 +04:00
#include <fcntl.h>
#include "extern.h"
1993-03-21 12:45:37 +03:00
#define READ 0
#define WRITE 1
struct fp {
FILE *fp;
int pipe;
1994-06-29 09:09:04 +04:00
int pid;
1993-03-21 12:45:37 +03:00
struct fp *link;
};
static struct fp *fp_head;
1994-06-29 09:09:04 +04:00
struct child {
int pid;
char done;
char free;
union wait status;
struct child *link;
};
static struct child *child;
static struct child *findchild __P((int));
static void delchild __P((struct child *));
static int file_pid __P((FILE *));
1994-06-29 09:09:04 +04:00
1993-03-21 12:45:37 +03:00
FILE *
Fopen(file, mode)
char *file, *mode;
{
FILE *fp;
1994-06-29 09:09:04 +04:00
if ((fp = fopen(file, mode)) != NULL) {
register_file(fp, 0, 0);
(void) fcntl(fileno(fp), F_SETFD, 1);
}
1993-03-21 12:45:37 +03:00
return fp;
}
FILE *
Fdopen(fd, mode)
1994-06-29 09:09:04 +04:00
int fd;
1993-03-21 12:45:37 +03:00
char *mode;
{
FILE *fp;
1994-06-29 09:09:04 +04:00
if ((fp = fdopen(fd, mode)) != NULL) {
register_file(fp, 0, 0);
(void) fcntl(fileno(fp), F_SETFD, 1);
}
1993-03-21 12:45:37 +03:00
return fp;
}
1994-06-29 09:09:04 +04:00
int
1993-03-21 12:45:37 +03:00
Fclose(fp)
FILE *fp;
{
unregister_file(fp);
return fclose(fp);
}
FILE *
Popen(cmd, mode)
char *cmd;
char *mode;
{
int p[2];
int myside, hisside, fd0, fd1;
1994-06-29 09:09:04 +04:00
int pid;
sigset_t nset;
1993-03-21 12:45:37 +03:00
FILE *fp;
if (pipe(p) < 0)
return NULL;
1994-06-29 09:09:04 +04:00
(void) fcntl(p[READ], F_SETFD, 1);
(void) fcntl(p[WRITE], F_SETFD, 1);
1993-03-21 12:45:37 +03:00
if (*mode == 'r') {
myside = p[READ];
fd0 = -1;
hisside = fd1 = p[WRITE];
} else {
myside = p[WRITE];
hisside = fd0 = p[READ];
fd1 = -1;
}
sigemptyset(&nset);
if ((pid = start_command(cmd, &nset, fd0, fd1, NOSTR, NOSTR, NOSTR)) < 0) {
1993-03-21 12:45:37 +03:00
close(p[READ]);
close(p[WRITE]);
return NULL;
}
(void) close(hisside);
if ((fp = fdopen(myside, mode)) != NULL)
1994-06-29 09:09:04 +04:00
register_file(fp, 1, pid);
1993-03-21 12:45:37 +03:00
return fp;
}
1994-06-29 09:09:04 +04:00
int
1993-03-21 12:45:37 +03:00
Pclose(ptr)
FILE *ptr;
{
int i;
sigset_t nset, oset;
1993-03-21 12:45:37 +03:00
1994-06-29 09:09:04 +04:00
i = file_pid(ptr);
1993-03-21 12:45:37 +03:00
unregister_file(ptr);
(void) fclose(ptr);
sigemptyset(&nset);
sigaddset(&nset, SIGINT);
sigaddset(&nset, SIGHUP);
sigprocmask(SIG_BLOCK, &nset, &oset);
1994-06-29 09:09:04 +04:00
i = wait_child(i);
sigprocmask(SIG_SETMASK, &oset, NULL);
1993-03-21 12:45:37 +03:00
return i;
}
1994-06-29 09:09:04 +04:00
void
1993-03-21 12:45:37 +03:00
close_all_files()
{
while (fp_head)
if (fp_head->pipe)
(void) Pclose(fp_head->fp);
else
(void) Fclose(fp_head->fp);
}
1994-06-29 09:09:04 +04:00
void
register_file(fp, pipe, pid)
1993-03-21 12:45:37 +03:00
FILE *fp;
1994-06-29 09:09:04 +04:00
int pipe, pid;
1993-03-21 12:45:37 +03:00
{
struct fp *fpp;
if ((fpp = (struct fp *) malloc(sizeof *fpp)) == NULL)
panic("Out of memory");
fpp->fp = fp;
fpp->pipe = pipe;
1994-06-29 09:09:04 +04:00
fpp->pid = pid;
1993-03-21 12:45:37 +03:00
fpp->link = fp_head;
fp_head = fpp;
}
1994-06-29 09:09:04 +04:00
void
1993-03-21 12:45:37 +03:00
unregister_file(fp)
FILE *fp;
{
struct fp **pp, *p;
for (pp = &fp_head; (p = *pp) != NULL; pp = &p->link)
1993-03-21 12:45:37 +03:00
if (p->fp == fp) {
*pp = p->link;
free((char *) p);
return;
}
panic("Invalid file pointer");
1994-06-29 09:09:04 +04:00
}
static int
1994-06-29 09:09:04 +04:00
file_pid(fp)
FILE *fp;
{
struct fp *p;
for (p = fp_head; p; p = p->link)
if (p->fp == fp)
return (p->pid);
panic("Invalid file pointer");
/*NOTREACHED*/
1993-03-21 12:45:37 +03:00
}
/*
* Run a command without a shell, with optional arguments and splicing
* of stdin and stdout. The command name can be a sequence of words.
* Signals must be handled by the caller.
* "Mask" contains the signals to ignore in the new process.
* SIGINT is enabled unless it's in the mask.
*/
/*VARARGS4*/
1994-06-29 09:09:04 +04:00
int
1993-03-21 12:45:37 +03:00
run_command(cmd, mask, infd, outfd, a0, a1, a2)
char *cmd;
sigset_t *mask;
int infd, outfd;
1993-03-21 12:45:37 +03:00
char *a0, *a1, *a2;
{
int pid;
if ((pid = start_command(cmd, mask, infd, outfd, a0, a1, a2)) < 0)
return -1;
return wait_command(pid);
}
/*VARARGS4*/
1994-06-29 09:09:04 +04:00
int
1993-03-21 12:45:37 +03:00
start_command(cmd, mask, infd, outfd, a0, a1, a2)
char *cmd;
sigset_t *mask;
int infd, outfd;
1993-03-21 12:45:37 +03:00
char *a0, *a1, *a2;
{
int pid;
if ((pid = vfork()) < 0) {
perror("fork");
return -1;
}
if (pid == 0) {
char *argv[100];
int i = getrawlist(cmd, argv, sizeof argv / sizeof *argv);
if ((argv[i++] = a0) != NOSTR &&
(argv[i++] = a1) != NOSTR &&
(argv[i++] = a2) != NOSTR)
argv[i] = NOSTR;
prepare_child(mask, infd, outfd);
execvp(argv[0], argv);
perror(argv[0]);
_exit(1);
}
return pid;
}
1994-06-29 09:09:04 +04:00
void
prepare_child(nset, infd, outfd)
sigset_t *nset;
int infd, outfd;
1993-03-21 12:45:37 +03:00
{
int i;
sigset_t fset;
1993-03-21 12:45:37 +03:00
1994-06-29 09:09:04 +04:00
/*
* All file descriptors other than 0, 1, and 2 are supposed to be
* close-on-exec.
*/
1993-03-21 12:45:37 +03:00
if (infd >= 0)
dup2(infd, 0);
if (outfd >= 0)
dup2(outfd, 1);
if (nset == NULL)
return;
if (nset != NULL) {
for (i = 1; i < NSIG; i++)
if (sigismember(nset, i))
(void) signal(i, SIG_IGN);
}
if (nset == NULL || !sigismember(nset, SIGINT))
1993-03-21 12:45:37 +03:00
(void) signal(SIGINT, SIG_DFL);
sigfillset(&fset);
(void) sigprocmask(SIG_UNBLOCK, &fset, NULL);
1993-03-21 12:45:37 +03:00
}
1994-06-29 09:09:04 +04:00
int
1993-03-21 12:45:37 +03:00
wait_command(pid)
int pid;
{
if (wait_child(pid) < 0) {
printf("Fatal error in process.\n");
return -1;
}
return 0;
}
1994-06-29 09:09:04 +04:00
static struct child *
1993-03-21 12:45:37 +03:00
findchild(pid)
int pid;
{
register struct child **cpp;
for (cpp = &child; *cpp != NULL && (*cpp)->pid != pid;
cpp = &(*cpp)->link)
;
if (*cpp == NULL) {
*cpp = (struct child *) malloc(sizeof (struct child));
(*cpp)->pid = pid;
(*cpp)->done = (*cpp)->free = 0;
(*cpp)->link = NULL;
}
return *cpp;
}
1994-06-29 09:09:04 +04:00
static void
1993-03-21 12:45:37 +03:00
delchild(cp)
register struct child *cp;
{
register struct child **cpp;
for (cpp = &child; *cpp != cp; cpp = &(*cpp)->link)
;
*cpp = cp->link;
free((char *) cp);
}
void
1994-06-29 09:09:04 +04:00
sigchild(signo)
int signo;
1993-03-21 12:45:37 +03:00
{
int pid;
union wait status;
register struct child *cp;
while ((pid =
wait3((int *)&status, WNOHANG, (struct rusage *)0)) > 0) {
cp = findchild(pid);
if (cp->free)
delchild(cp);
else {
cp->done = 1;
cp->status = status;
}
}
}
union wait wait_status;
/*
* Wait for a specific child to die.
*/
1994-06-29 09:09:04 +04:00
int
1993-03-21 12:45:37 +03:00
wait_child(pid)
int pid;
{
sigset_t nset, oset;
1993-03-21 12:45:37 +03:00
register struct child *cp = findchild(pid);
sigemptyset(&nset);
sigaddset(&nset, SIGCHLD);
sigprocmask(SIG_BLOCK, &nset, &oset);
1993-03-21 12:45:37 +03:00
while (!cp->done)
sigsuspend(&oset);
1993-03-21 12:45:37 +03:00
wait_status = cp->status;
delchild(cp);
sigprocmask(SIG_SETMASK, &oset, NULL);
1993-03-21 12:45:37 +03:00
return wait_status.w_status ? -1 : 0;
}
/*
* Mark a child as don't care.
*/
1994-06-29 09:09:04 +04:00
void
1993-03-21 12:45:37 +03:00
free_child(pid)
int pid;
{
sigset_t nset, oset;
1993-03-21 12:45:37 +03:00
register struct child *cp = findchild(pid);
sigemptyset(&nset);
sigaddset(&nset, SIGCHLD);
sigprocmask(SIG_BLOCK, &nset, &oset);
1993-03-21 12:45:37 +03:00
if (cp->done)
delchild(cp);
else
cp->free = 1;
sigprocmask(SIG_SETMASK, &oset, NULL);
1993-03-21 12:45:37 +03:00
}