[BACK]Return to popen.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / mail

Annotation of src/usr.bin/mail/popen.c, Revision 1.1

1.1     ! deraadt     1: /*
        !             2:  * Copyright (c) 1980, 1993
        !             3:  *     The Regents of the University of California.  All rights reserved.
        !             4:  *
        !             5:  * Redistribution and use in source and binary forms, with or without
        !             6:  * modification, are permitted provided that the following conditions
        !             7:  * are met:
        !             8:  * 1. Redistributions of source code must retain the above copyright
        !             9:  *    notice, this list of conditions and the following disclaimer.
        !            10:  * 2. Redistributions in binary form must reproduce the above copyright
        !            11:  *    notice, this list of conditions and the following disclaimer in the
        !            12:  *    documentation and/or other materials provided with the distribution.
        !            13:  * 3. All advertising materials mentioning features or use of this software
        !            14:  *    must display the following acknowledgement:
        !            15:  *     This product includes software developed by the University of
        !            16:  *     California, Berkeley and its contributors.
        !            17:  * 4. Neither the name of the University nor the names of its contributors
        !            18:  *    may be used to endorse or promote products derived from this software
        !            19:  *    without specific prior written permission.
        !            20:  *
        !            21:  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
        !            22:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
        !            23:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
        !            24:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
        !            25:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
        !            26:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
        !            27:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
        !            28:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
        !            29:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
        !            30:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
        !            31:  * SUCH DAMAGE.
        !            32:  */
        !            33:
        !            34: #ifndef lint
        !            35: static char sccsid[] = "from: @(#)popen.c      8.1 (Berkeley) 6/6/93";
        !            36: static char rcsid[] = "$Id: popen.c,v 1.3 1994/06/29 05:09:38 deraadt Exp $";
        !            37: #endif /* not lint */
        !            38:
        !            39: #include "rcv.h"
        !            40: #include <sys/wait.h>
        !            41: #include <fcntl.h>
        !            42: #include "extern.h"
        !            43:
        !            44: #define READ 0
        !            45: #define WRITE 1
        !            46:
        !            47: struct fp {
        !            48:        FILE *fp;
        !            49:        int pipe;
        !            50:        int pid;
        !            51:        struct fp *link;
        !            52: };
        !            53: static struct fp *fp_head;
        !            54:
        !            55: struct child {
        !            56:        int pid;
        !            57:        char done;
        !            58:        char free;
        !            59:        union wait status;
        !            60:        struct child *link;
        !            61: };
        !            62: static struct child *child;
        !            63: static struct child *findchild __P((int));
        !            64: static void delchild __P((struct child *));
        !            65:
        !            66: FILE *
        !            67: Fopen(file, mode)
        !            68:        char *file, *mode;
        !            69: {
        !            70:        FILE *fp;
        !            71:
        !            72:        if ((fp = fopen(file, mode)) != NULL) {
        !            73:                register_file(fp, 0, 0);
        !            74:                (void) fcntl(fileno(fp), F_SETFD, 1);
        !            75:        }
        !            76:        return fp;
        !            77: }
        !            78:
        !            79: FILE *
        !            80: Fdopen(fd, mode)
        !            81:        int fd;
        !            82:        char *mode;
        !            83: {
        !            84:        FILE *fp;
        !            85:
        !            86:        if ((fp = fdopen(fd, mode)) != NULL) {
        !            87:                register_file(fp, 0, 0);
        !            88:                (void) fcntl(fileno(fp), F_SETFD, 1);
        !            89:        }
        !            90:        return fp;
        !            91: }
        !            92:
        !            93: int
        !            94: Fclose(fp)
        !            95:        FILE *fp;
        !            96: {
        !            97:        unregister_file(fp);
        !            98:        return fclose(fp);
        !            99: }
        !           100:
        !           101: FILE *
        !           102: Popen(cmd, mode)
        !           103:        char *cmd;
        !           104:        char *mode;
        !           105: {
        !           106:        int p[2];
        !           107:        int myside, hisside, fd0, fd1;
        !           108:        int pid;
        !           109:        FILE *fp;
        !           110:
        !           111:        if (pipe(p) < 0)
        !           112:                return NULL;
        !           113:        (void) fcntl(p[READ], F_SETFD, 1);
        !           114:        (void) fcntl(p[WRITE], F_SETFD, 1);
        !           115:        if (*mode == 'r') {
        !           116:                myside = p[READ];
        !           117:                fd0 = -1;
        !           118:                hisside = fd1 = p[WRITE];
        !           119:        } else {
        !           120:                myside = p[WRITE];
        !           121:                hisside = fd0 = p[READ];
        !           122:                fd1 = -1;
        !           123:        }
        !           124:        if ((pid = start_command(cmd, 0, fd0, fd1, NOSTR, NOSTR, NOSTR)) < 0) {
        !           125:                close(p[READ]);
        !           126:                close(p[WRITE]);
        !           127:                return NULL;
        !           128:        }
        !           129:        (void) close(hisside);
        !           130:        if ((fp = fdopen(myside, mode)) != NULL)
        !           131:                register_file(fp, 1, pid);
        !           132:        return fp;
        !           133: }
        !           134:
        !           135: int
        !           136: Pclose(ptr)
        !           137:        FILE *ptr;
        !           138: {
        !           139:        int i;
        !           140:        int omask;
        !           141:
        !           142:        i = file_pid(ptr);
        !           143:        unregister_file(ptr);
        !           144:        (void) fclose(ptr);
        !           145:        omask = sigblock(sigmask(SIGINT)|sigmask(SIGHUP));
        !           146:        i = wait_child(i);
        !           147:        sigsetmask(omask);
        !           148:        return i;
        !           149: }
        !           150:
        !           151: void
        !           152: close_all_files()
        !           153: {
        !           154:
        !           155:        while (fp_head)
        !           156:                if (fp_head->pipe)
        !           157:                        (void) Pclose(fp_head->fp);
        !           158:                else
        !           159:                        (void) Fclose(fp_head->fp);
        !           160: }
        !           161:
        !           162: void
        !           163: register_file(fp, pipe, pid)
        !           164:        FILE *fp;
        !           165:        int pipe, pid;
        !           166: {
        !           167:        struct fp *fpp;
        !           168:
        !           169:        if ((fpp = (struct fp *) malloc(sizeof *fpp)) == NULL)
        !           170:                panic("Out of memory");
        !           171:        fpp->fp = fp;
        !           172:        fpp->pipe = pipe;
        !           173:        fpp->pid = pid;
        !           174:        fpp->link = fp_head;
        !           175:        fp_head = fpp;
        !           176: }
        !           177:
        !           178: void
        !           179: unregister_file(fp)
        !           180:        FILE *fp;
        !           181: {
        !           182:        struct fp **pp, *p;
        !           183:
        !           184:        for (pp = &fp_head; p = *pp; pp = &p->link)
        !           185:                if (p->fp == fp) {
        !           186:                        *pp = p->link;
        !           187:                        free((char *) p);
        !           188:                        return;
        !           189:                }
        !           190:        panic("Invalid file pointer");
        !           191: }
        !           192:
        !           193: file_pid(fp)
        !           194:        FILE *fp;
        !           195: {
        !           196:        struct fp *p;
        !           197:
        !           198:        for (p = fp_head; p; p = p->link)
        !           199:                if (p->fp == fp)
        !           200:                        return (p->pid);
        !           201:        panic("Invalid file pointer");
        !           202:        /*NOTREACHED*/
        !           203: }
        !           204:
        !           205: /*
        !           206:  * Run a command without a shell, with optional arguments and splicing
        !           207:  * of stdin and stdout.  The command name can be a sequence of words.
        !           208:  * Signals must be handled by the caller.
        !           209:  * "Mask" contains the signals to ignore in the new process.
        !           210:  * SIGINT is enabled unless it's in the mask.
        !           211:  */
        !           212: /*VARARGS4*/
        !           213: int
        !           214: run_command(cmd, mask, infd, outfd, a0, a1, a2)
        !           215:        char *cmd;
        !           216:        int mask, infd, outfd;
        !           217:        char *a0, *a1, *a2;
        !           218: {
        !           219:        int pid;
        !           220:
        !           221:        if ((pid = start_command(cmd, mask, infd, outfd, a0, a1, a2)) < 0)
        !           222:                return -1;
        !           223:        return wait_command(pid);
        !           224: }
        !           225:
        !           226: /*VARARGS4*/
        !           227: int
        !           228: start_command(cmd, mask, infd, outfd, a0, a1, a2)
        !           229:        char *cmd;
        !           230:        int mask, infd, outfd;
        !           231:        char *a0, *a1, *a2;
        !           232: {
        !           233:        int pid;
        !           234:
        !           235:        if ((pid = vfork()) < 0) {
        !           236:                perror("fork");
        !           237:                return -1;
        !           238:        }
        !           239:        if (pid == 0) {
        !           240:                char *argv[100];
        !           241:                int i = getrawlist(cmd, argv, sizeof argv / sizeof *argv);
        !           242:
        !           243:                if ((argv[i++] = a0) != NOSTR &&
        !           244:                    (argv[i++] = a1) != NOSTR &&
        !           245:                    (argv[i++] = a2) != NOSTR)
        !           246:                        argv[i] = NOSTR;
        !           247:                prepare_child(mask, infd, outfd);
        !           248:                execvp(argv[0], argv);
        !           249:                perror(argv[0]);
        !           250:                _exit(1);
        !           251:        }
        !           252:        return pid;
        !           253: }
        !           254:
        !           255: void
        !           256: prepare_child(mask, infd, outfd)
        !           257:        int mask, infd, outfd;
        !           258: {
        !           259:        int i;
        !           260:
        !           261:        /*
        !           262:         * All file descriptors other than 0, 1, and 2 are supposed to be
        !           263:         * close-on-exec.
        !           264:         */
        !           265:        if (infd >= 0)
        !           266:                dup2(infd, 0);
        !           267:        if (outfd >= 0)
        !           268:                dup2(outfd, 1);
        !           269:        for (i = 1; i <= NSIG; i++)
        !           270:                if (mask & sigmask(i))
        !           271:                        (void) signal(i, SIG_IGN);
        !           272:        if ((mask & sigmask(SIGINT)) == 0)
        !           273:                (void) signal(SIGINT, SIG_DFL);
        !           274:        (void) sigsetmask(0);
        !           275: }
        !           276:
        !           277: int
        !           278: wait_command(pid)
        !           279:        int pid;
        !           280: {
        !           281:
        !           282:        if (wait_child(pid) < 0) {
        !           283:                printf("Fatal error in process.\n");
        !           284:                return -1;
        !           285:        }
        !           286:        return 0;
        !           287: }
        !           288:
        !           289: static struct child *
        !           290: findchild(pid)
        !           291:        int pid;
        !           292: {
        !           293:        register struct child **cpp;
        !           294:
        !           295:        for (cpp = &child; *cpp != NULL && (*cpp)->pid != pid;
        !           296:             cpp = &(*cpp)->link)
        !           297:                        ;
        !           298:        if (*cpp == NULL) {
        !           299:                *cpp = (struct child *) malloc(sizeof (struct child));
        !           300:                (*cpp)->pid = pid;
        !           301:                (*cpp)->done = (*cpp)->free = 0;
        !           302:                (*cpp)->link = NULL;
        !           303:        }
        !           304:        return *cpp;
        !           305: }
        !           306:
        !           307: static void
        !           308: delchild(cp)
        !           309:        register struct child *cp;
        !           310: {
        !           311:        register struct child **cpp;
        !           312:
        !           313:        for (cpp = &child; *cpp != cp; cpp = &(*cpp)->link)
        !           314:                ;
        !           315:        *cpp = cp->link;
        !           316:        free((char *) cp);
        !           317: }
        !           318:
        !           319: void
        !           320: sigchild(signo)
        !           321:        int signo;
        !           322: {
        !           323:        int pid;
        !           324:        union wait status;
        !           325:        register struct child *cp;
        !           326:
        !           327:        while ((pid =
        !           328:            wait3((int *)&status, WNOHANG, (struct rusage *)0)) > 0) {
        !           329:                cp = findchild(pid);
        !           330:                if (cp->free)
        !           331:                        delchild(cp);
        !           332:                else {
        !           333:                        cp->done = 1;
        !           334:                        cp->status = status;
        !           335:                }
        !           336:        }
        !           337: }
        !           338:
        !           339: union wait wait_status;
        !           340:
        !           341: /*
        !           342:  * Wait for a specific child to die.
        !           343:  */
        !           344: int
        !           345: wait_child(pid)
        !           346:        int pid;
        !           347: {
        !           348:        int mask = sigblock(sigmask(SIGCHLD));
        !           349:        register struct child *cp = findchild(pid);
        !           350:
        !           351:        while (!cp->done)
        !           352:                sigpause(mask);
        !           353:        wait_status = cp->status;
        !           354:        delchild(cp);
        !           355:        sigsetmask(mask);
        !           356:        return wait_status.w_status ? -1 : 0;
        !           357: }
        !           358:
        !           359: /*
        !           360:  * Mark a child as don't care.
        !           361:  */
        !           362: void
        !           363: free_child(pid)
        !           364:        int pid;
        !           365: {
        !           366:        int mask = sigblock(sigmask(SIGCHLD));
        !           367:        register struct child *cp = findchild(pid);
        !           368:
        !           369:        if (cp->done)
        !           370:                delchild(cp);
        !           371:        else
        !           372:                cp->free = 1;
        !           373:        sigsetmask(mask);
        !           374: }