xref: /onnv-gate/usr/src/cmd/cmd-inet/usr.bin/ftp/pclose.c (revision 0:68f95e015346)
1*0Sstevel@tonic-gate /*
2*0Sstevel@tonic-gate  * CDDL HEADER START
3*0Sstevel@tonic-gate  *
4*0Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*0Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*0Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*0Sstevel@tonic-gate  * with the License.
8*0Sstevel@tonic-gate  *
9*0Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*0Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*0Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*0Sstevel@tonic-gate  * and limitations under the License.
13*0Sstevel@tonic-gate  *
14*0Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*0Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*0Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*0Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*0Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*0Sstevel@tonic-gate  *
20*0Sstevel@tonic-gate  * CDDL HEADER END
21*0Sstevel@tonic-gate  */
22*0Sstevel@tonic-gate /*
23*0Sstevel@tonic-gate  *	Copyright 1996 Sun Microsystems, Inc.  All rights reserved.
24*0Sstevel@tonic-gate  *	Use is subject to license terms.
25*0Sstevel@tonic-gate  */
26*0Sstevel@tonic-gate 
27*0Sstevel@tonic-gate /*	Copyright (c) 1983, 1984, 1985, 1986, 1987, 1988, 1989 AT&T	*/
28*0Sstevel@tonic-gate /*	All Rights Reserved  	*/
29*0Sstevel@tonic-gate 
30*0Sstevel@tonic-gate /*
31*0Sstevel@tonic-gate  *	University Copyright- Copyright (c) 1982, 1986, 1988
32*0Sstevel@tonic-gate  *	The Regents of the University of California
33*0Sstevel@tonic-gate  *	All Rights Reserved
34*0Sstevel@tonic-gate  *
35*0Sstevel@tonic-gate  *	University Acknowledgment- Portions of this document are derived from
36*0Sstevel@tonic-gate  *	software developed by the University of California, Berkeley, and its
37*0Sstevel@tonic-gate  *	contributors.
38*0Sstevel@tonic-gate  */
39*0Sstevel@tonic-gate 
40*0Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
41*0Sstevel@tonic-gate 
42*0Sstevel@tonic-gate #include "ftp_var.h"
43*0Sstevel@tonic-gate 
44*0Sstevel@tonic-gate #ifndef sigmask
45*0Sstevel@tonic-gate #define	sigmask(m)	(1 << ((m)-1))
46*0Sstevel@tonic-gate #endif
47*0Sstevel@tonic-gate 
48*0Sstevel@tonic-gate #define	set2mask(setp) ((setp)->__sigbits[0])
49*0Sstevel@tonic-gate #define	mask2set(mask, setp) \
50*0Sstevel@tonic-gate 	((mask) == -1 ? sigfillset(setp) : (((setp)->__sigbits[0]) = (mask)))
51*0Sstevel@tonic-gate 
52*0Sstevel@tonic-gate 
53*0Sstevel@tonic-gate static int
sigsetmask(int mask)54*0Sstevel@tonic-gate sigsetmask(int mask)
55*0Sstevel@tonic-gate {
56*0Sstevel@tonic-gate 	sigset_t oset;
57*0Sstevel@tonic-gate 	sigset_t nset;
58*0Sstevel@tonic-gate 
59*0Sstevel@tonic-gate 	(void) sigprocmask(0, (sigset_t *)0, &nset);
60*0Sstevel@tonic-gate 	mask2set(mask, &nset);
61*0Sstevel@tonic-gate 	(void) sigprocmask(SIG_SETMASK, &nset, &oset);
62*0Sstevel@tonic-gate 	return (set2mask(&oset));
63*0Sstevel@tonic-gate }
64*0Sstevel@tonic-gate 
65*0Sstevel@tonic-gate static int
sigblock(int mask)66*0Sstevel@tonic-gate sigblock(int mask)
67*0Sstevel@tonic-gate {
68*0Sstevel@tonic-gate 	sigset_t oset;
69*0Sstevel@tonic-gate 	sigset_t nset;
70*0Sstevel@tonic-gate 
71*0Sstevel@tonic-gate 	(void) sigprocmask(0, (sigset_t *)0, &nset);
72*0Sstevel@tonic-gate 	mask2set(mask, &nset);
73*0Sstevel@tonic-gate 	(void) sigprocmask(SIG_BLOCK, &nset, &oset);
74*0Sstevel@tonic-gate 	return (set2mask(&oset));
75*0Sstevel@tonic-gate }
76*0Sstevel@tonic-gate 
77*0Sstevel@tonic-gate #define	signal(s, f)	sigset(s, f)
78*0Sstevel@tonic-gate 
79*0Sstevel@tonic-gate #define	tst(a, b)	(*mode == 'r'? (b) : (a))
80*0Sstevel@tonic-gate #define	RDR		0
81*0Sstevel@tonic-gate #define	WTR		1
82*0Sstevel@tonic-gate #define	NOFILES		20	/* just in case */
83*0Sstevel@tonic-gate 
84*0Sstevel@tonic-gate static	pid_t *popen_pid;
85*0Sstevel@tonic-gate static	rlim_t nfiles = 0;
86*0Sstevel@tonic-gate 
87*0Sstevel@tonic-gate FILE *
mypopen(char * cmd,char * mode)88*0Sstevel@tonic-gate mypopen(char *cmd, char *mode)
89*0Sstevel@tonic-gate {
90*0Sstevel@tonic-gate 	int p[2];
91*0Sstevel@tonic-gate 	pid_t pid;
92*0Sstevel@tonic-gate 	int myside, remside, i;
93*0Sstevel@tonic-gate 	struct rlimit rl;
94*0Sstevel@tonic-gate 
95*0Sstevel@tonic-gate 	if (nfiles <= 0) {
96*0Sstevel@tonic-gate 		if (getrlimit(RLIMIT_NOFILE, &rl) == 0)
97*0Sstevel@tonic-gate 			nfiles = rl.rlim_max;
98*0Sstevel@tonic-gate 		else
99*0Sstevel@tonic-gate 			nfiles = NOFILES;
100*0Sstevel@tonic-gate 	}
101*0Sstevel@tonic-gate 	if (popen_pid == NULL) {
102*0Sstevel@tonic-gate 		popen_pid = (pid_t *)malloc((unsigned)nfiles *
103*0Sstevel@tonic-gate 		    sizeof (*popen_pid));
104*0Sstevel@tonic-gate 		if (popen_pid == NULL)
105*0Sstevel@tonic-gate 			return (NULL);
106*0Sstevel@tonic-gate 		for (i = 0; i < nfiles; i++)
107*0Sstevel@tonic-gate 			popen_pid[i] = (pid_t)-1;
108*0Sstevel@tonic-gate 	}
109*0Sstevel@tonic-gate 	if (pipe(p) < 0)
110*0Sstevel@tonic-gate 		return (NULL);
111*0Sstevel@tonic-gate 	myside = tst(p[WTR], p[RDR]);
112*0Sstevel@tonic-gate 	remside = tst(p[RDR], p[WTR]);
113*0Sstevel@tonic-gate 	if ((pid = vfork()) == 0) {
114*0Sstevel@tonic-gate 		/* myside and remside reverse roles in child */
115*0Sstevel@tonic-gate 		(void) close(myside);
116*0Sstevel@tonic-gate 		if (remside != tst(0, 1)) {
117*0Sstevel@tonic-gate 			(void) dup2(remside, tst(0, 1));
118*0Sstevel@tonic-gate 			(void) close(remside);
119*0Sstevel@tonic-gate 		}
120*0Sstevel@tonic-gate 		execl("/bin/sh", "sh", "-c", cmd, (char *)NULL);
121*0Sstevel@tonic-gate 		_exit(127);
122*0Sstevel@tonic-gate 	}
123*0Sstevel@tonic-gate 	if (pid == (pid_t)-1) {
124*0Sstevel@tonic-gate 		(void) close(myside);
125*0Sstevel@tonic-gate 		(void) close(remside);
126*0Sstevel@tonic-gate 		return (NULL);
127*0Sstevel@tonic-gate 	}
128*0Sstevel@tonic-gate 	popen_pid[myside] = pid;
129*0Sstevel@tonic-gate 	(void) close(remside);
130*0Sstevel@tonic-gate 	return (fdopen(myside, mode));
131*0Sstevel@tonic-gate }
132*0Sstevel@tonic-gate 
133*0Sstevel@tonic-gate /*ARGSUSED*/
134*0Sstevel@tonic-gate static void
pabort(int sig)135*0Sstevel@tonic-gate pabort(int sig)
136*0Sstevel@tonic-gate {
137*0Sstevel@tonic-gate 	extern int mflag;
138*0Sstevel@tonic-gate 
139*0Sstevel@tonic-gate 	mflag = 0;
140*0Sstevel@tonic-gate }
141*0Sstevel@tonic-gate 
142*0Sstevel@tonic-gate int
mypclose(FILE * ptr)143*0Sstevel@tonic-gate mypclose(FILE *ptr)
144*0Sstevel@tonic-gate {
145*0Sstevel@tonic-gate 	pid_t child, pid;
146*0Sstevel@tonic-gate 	int omask;
147*0Sstevel@tonic-gate 	void (*istat)();
148*0Sstevel@tonic-gate 	int status;
149*0Sstevel@tonic-gate 
150*0Sstevel@tonic-gate 	child = popen_pid[fileno(ptr)];
151*0Sstevel@tonic-gate 	popen_pid[fileno(ptr)] = (pid_t)-1;
152*0Sstevel@tonic-gate 	(void) fclose(ptr);
153*0Sstevel@tonic-gate 	if (child == (pid_t)-1)
154*0Sstevel@tonic-gate 		return (-1);
155*0Sstevel@tonic-gate 	istat = signal(SIGINT, pabort);
156*0Sstevel@tonic-gate 	omask = sigblock(sigmask(SIGQUIT)|sigmask(SIGHUP));
157*0Sstevel@tonic-gate 	while ((pid = wait(&status)) != child && pid != (pid_t)-1)
158*0Sstevel@tonic-gate 		;
159*0Sstevel@tonic-gate 	(void) sigsetmask(omask);
160*0Sstevel@tonic-gate 	(void) signal(SIGINT, istat);
161*0Sstevel@tonic-gate 	return (pid == (pid_t)-1 ? -1 : 0);
162*0Sstevel@tonic-gate }
163