popen.inc

   1 /*      $OpenBSD: popen.c,v 1.17 2005/08/08 08:05:34 espie Exp $ */
   2 /*
   3  * Copyright (c) 1988, 1993
   4  *      The Regents of the University of California.  All rights reserved.
   5  *
   6  * This code is derived from software written by Ken Arnold and
   7  * published in UNIX Review, Vol. 6, No. 8.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  * 3. 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 #include <sys/param.h>
  35 #include <sys/wait.h>
  36
  37 #include <signal.h>
  38 #include <errno.h>
  39 #include <unistd.h>
  40 #include <stdio.h>
  41 #include <stdlib.h>
  42 #include <string.h>
  43 #include <paths.h>
  44
  45 __RCSID("$MirOS: contrib/code/jupp/popen.inc,v 1.3 2005/09/22 20:40:00 tg Exp $");
  46
  47 static struct pid {
  48         struct pid *next;
  49         FILE *fp;
  50         pid_t pid;
  51 } *pidlist;
  52
  53 FILE *
  54 popen(const char *program, const char *type)
  55 {
  56         struct pid * volatile cur;
  57         FILE *iop;
  58         int pdes[2];
  59         pid_t pid;
  60
  61         if ((*type != 'r' && *type != 'w') || type[1] != '\0') {
  62                 errno = EINVAL;
  63                 return (NULL);
  64         }
  65
  66         if ((cur = malloc(sizeof(struct pid))) == NULL)
  67                 return (NULL);
  68
  69         if (pipe(pdes) < 0) {
  70                 free(cur);
  71                 return (NULL);
  72         }
  73
  74         switch (pid = vfork()) {
  75         case -1:                        /* Error. */
  76                 (void)close(pdes[0]);
  77                 (void)close(pdes[1]);
  78                 free(cur);
  79                 return (NULL);
  80                 /* NOTREACHED */
  81         case 0:                         /* Child. */
  82             {
  83                 struct pid *pcur;
  84                 /*
  85                  * because vfork() instead of fork(), must leak FILE *,
  86                  * but luckily we are terminally headed for an execl()
  87                  */
  88                 for (pcur = pidlist; pcur; pcur = pcur->next)
  89                         close(fileno(pcur->fp));
  90
  91                 if (*type == 'r') {
  92                         int tpdes1 = pdes[1];
  93
  94                         (void) close(pdes[0]);
  95                         /*
  96                          * We must NOT modify pdes, due to the
  97                          * semantics of vfork.
  98                          */
  99                         if (tpdes1 != STDOUT_FILENO) {
 100                                 (void)dup2(tpdes1, STDOUT_FILENO);
 101                                 (void)close(tpdes1);
 102                                 tpdes1 = STDOUT_FILENO;
 103                         }
 104                 } else {
 105                         (void)close(pdes[1]);
 106                         if (pdes[0] != STDIN_FILENO) {
 107                                 (void)dup2(pdes[0], STDIN_FILENO);
 108                                 (void)close(pdes[0]);
 109                         }
 110                 }
 111                 execl(_PATH_BSHELL, "sh", "-c", program, (char *)NULL);
 112                 _exit(127);
 113                 /* NOTREACHED */
 114             }
 115         }
 116
 117         /* Parent; assume fdopen can't fail. */
 118         if (*type == 'r') {
 119                 iop = fdopen(pdes[0], type);
 120                 (void)close(pdes[1]);
 121         } else {
 122                 iop = fdopen(pdes[1], type);
 123                 (void)close(pdes[0]);
 124         }
 125
 126         /* Link into list of file descriptors. */
 127         cur->fp = iop;
 128         cur->pid =  pid;
 129         cur->next = pidlist;
 130         pidlist = cur;
 131
 132         return (iop);
 133 }
 134
 135 /*
 136  * pclose --
 137  *      Pclose returns -1 if stream is not associated with a `popened' command,
 138  *      if already `pclosed', or waitpid returns an error.
 139  */
 140 int
 141 pclose(FILE *iop)
 142 {
 143         struct pid *cur, *last;
 144         int pstat;
 145         pid_t pid;
 146
 147         /* Find the appropriate file pointer. */
 148         for (last = NULL, cur = pidlist; cur; last = cur, cur = cur->next)
 149                 if (cur->fp == iop)
 150                         break;
 151
 152         if (cur == NULL)
 153                 return (-1);
 154
 155         (void)fclose(iop);
 156
 157         do {
 158                 pid = waitpid(cur->pid, &pstat, 0);
 159         } while (pid == -1 && errno == EINTR);
 160
 161         /* Remove the entry from the linked list. */
 162         if (last == NULL)
 163                 pidlist = cur->next;
 164         else
 165                 last->next = cur->next;
 166         free(cur);
 167
 168         return (pid == -1 ? -1 : pstat);
 169 }