Differential D5912 Diff 15080 contrib/blacklist/port/popenve.c

Changeset View

Standalone View

contrib/blacklist/port/popenve.c

This file was added.

Property	Old Value	New Value
svn:eol-style	null	native \ No newline at end of property
svn:keywords	null	FreeBSD=%H \ No newline at end of property
svn:mime-type	null	text/plain \ No newline at end of property

				/* $NetBSD: popenve.c,v 1.2 2015/01/22 03:10:50 christos Exp $ */

				/*
				* Copyright (c) 1988, 1993
				* The Regents of the University of California. All rights reserved.
				*
				* This code is derived from software written by Ken Arnold and
				* published in UNIX Review, Vol. 6, No. 8.
				*
				* 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. 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.
				*/

				#ifdef HAVE_CONFIG_H
				#include "config.h"
				#endif

				#include <sys/cdefs.h>
				#if defined(LIBC_SCCS) && !defined(lint)
				#if 0
				static char sccsid[] = "@(#)popen.c 8.3 (Berkeley) 5/3/95";
				#else
				__RCSID("$NetBSD: popenve.c,v 1.2 2015/01/22 03:10:50 christos Exp $");
				#endif
				#endif /* LIBC_SCCS and not lint */

				#include <sys/param.h>
				#include <sys/wait.h>
				#include <sys/socket.h>

				#include <assert.h>
				#include <errno.h>
				#include <paths.h>
				#include <signal.h>
				#include <stdio.h>
				#include <stdlib.h>
				#include <string.h>
				#include <unistd.h>
				#include <fcntl.h>

				#ifdef __weak_alias
				__weak_alias(popen,_popen)
				__weak_alias(pclose,_pclose)
				#endif

				static struct pid {
				struct pid *next;
				FILE *fp;
				#ifdef _REENTRANT
				int fd;
				#endif
				pid_t pid;
				} *pidlist;

				#ifdef _REENTRANT
				static rwlock_t pidlist_lock = RWLOCK_INITIALIZER;
				#endif

				static struct pid *
				pdes_get(int pdes, const char *type)
				{
				struct pid *cur;
				int flags = strchr(*type, 'e') ? O_CLOEXEC : 0;
				int serrno;

				if (strchr(*type, '+')) {
				#ifndef SOCK_CLOEXEC
				#define SOCK_CLOEXEC 0
				#endif
				int stype = flags ? (SOCK_STREAM \| SOCK_CLOEXEC) : SOCK_STREAM;
				*type = "r+";
				if (socketpair(AF_LOCAL, stype, 0, pdes) < 0)
				return NULL;
				#if SOCK_CLOEXEC == 0
				fcntl(pdes[0], F_SETFD, FD_CLOEXEC);
				fcntl(pdes[1], F_SETFD, FD_CLOEXEC);
				#endif
				} else {
				type = strrchr(type, 'r') ? "r" : "w";
				#if SOCK_CLOEXEC != 0
				if (pipe2(pdes, flags) == -1)
				return NULL;
				#else
				if (pipe(pdes) == -1)
				return NULL;
				fcntl(pdes[0], F_SETFL, fcntl(pdes[0], F_GETFL) \| flags);
				fcntl(pdes[1], F_SETFL, fcntl(pdes[1], F_GETFL) \| flags);
				#endif
				}

				if ((cur = malloc(sizeof(*cur))) != NULL)
				return cur;
				serrno = errno;
				(void)close(pdes[0]);
				(void)close(pdes[1]);
				errno = serrno;
				return NULL;
				}

				static void
				pdes_child(int pdes, const char type)
				{
				struct pid *old;

				/* POSIX.2 B.3.2.2 "popen() shall ensure that any streams
				from previous popen() calls that remain open in the
				parent process are closed in the new child process. */
				for (old = pidlist; old; old = old->next)
				#ifdef _REENTRANT
				(void)close(old->fd); /* don't allow a flush */
				#else
				(void)close(fileno(old->fp)); /* don't allow a flush */
				#endif

				if (type[0] == 'r') {
				(void)close(pdes[0]);
				if (pdes[1] != STDOUT_FILENO) {
				(void)dup2(pdes[1], STDOUT_FILENO);
				(void)close(pdes[1]);
				}
				if (type[1] == '+')
				(void)dup2(STDOUT_FILENO, STDIN_FILENO);
				} else {
				(void)close(pdes[1]);
				if (pdes[0] != STDIN_FILENO) {
				(void)dup2(pdes[0], STDIN_FILENO);
				(void)close(pdes[0]);
				}
				}
				}

				static void
				pdes_parent(int pdes, struct pid cur, pid_t pid, const char *type)
				{
				FILE *iop;

				/* Parent; assume fdopen can't fail. */
				if (*type == 'r') {
				iop = fdopen(pdes[0], type);
				#ifdef _REENTRANT
				cur->fd = pdes[0];
				#endif
				(void)close(pdes[1]);
				} else {
				iop = fdopen(pdes[1], type);
				#ifdef _REENTRANT
				cur->fd = pdes[1];
				#endif
				(void)close(pdes[0]);
				}

				/* Link into list of file descriptors. */
				cur->fp = iop;
				cur->pid = pid;
				cur->next = pidlist;
				pidlist = cur;
				}

				static void
				pdes_error(int pdes, struct pid cur)
				{
				free(cur);
				(void)close(pdes[0]);
				(void)close(pdes[1]);
				}

				FILE *
				popenve(const char cmd, char const argv, char const envp, const char type)
				{
				struct pid *cur;
				int pdes[2], serrno;
				pid_t pid;

				if ((cur = pdes_get(pdes, &type)) == NULL)
				return NULL;

				#ifdef _REENTRANT
				(void)rwlock_rdlock(&pidlist_lock);
				#endif
				switch (pid = vfork()) {
				case -1: /* Error. */
				serrno = errno;
				#ifdef _REENTRANT
				(void)rwlock_unlock(&pidlist_lock);
				#endif
				pdes_error(pdes, cur);
				errno = serrno;
				return NULL;
				/* NOTREACHED */
				case 0: /* Child. */
				pdes_child(pdes, type);
				execve(cmd, argv, envp);
				_exit(127);
				/* NOTREACHED */
				}

				pdes_parent(pdes, cur, pid, type);

				#ifdef _REENTRANT
				(void)rwlock_unlock(&pidlist_lock);
				#endif

				return cur->fp;
				}

				/*
				* pclose --
				* Pclose returns -1 if stream is not associated with a `popened' command,
				* if already `pclosed', or waitpid returns an error.
				*/
				int
				pcloseve(FILE *iop)
				{
				struct pid cur, last;
				int pstat;
				pid_t pid;

				#ifdef _REENTRANT
				rwlock_wrlock(&pidlist_lock);
				#endif

				/* Find the appropriate file pointer. */
				for (last = NULL, cur = pidlist; cur; last = cur, cur = cur->next)
				if (cur->fp == iop)
				break;
				if (cur == NULL) {
				#ifdef _REENTRANT
				(void)rwlock_unlock(&pidlist_lock);
				#endif
				errno = ESRCH;
				return -1;
				}

				(void)fclose(iop);

				/* Remove the entry from the linked list. */
				if (last == NULL)
				pidlist = cur->next;
				else
				last->next = cur->next;

				#ifdef _REENTRANT
				(void)rwlock_unlock(&pidlist_lock);
				#endif

				do {
				pid = waitpid(cur->pid, &pstat, 0);
				} while (pid == -1 && errno == EINTR);

				free(cur);

				return pid == -1 ? -1 : pstat;
				}