1*7c478bd9Sstevel@tonic-gate /*
2*7c478bd9Sstevel@tonic-gate * CDDL HEADER START
3*7c478bd9Sstevel@tonic-gate *
4*7c478bd9Sstevel@tonic-gate * The contents of this file are subject to the terms of the
5*7c478bd9Sstevel@tonic-gate * Common Development and Distribution License, Version 1.0 only
6*7c478bd9Sstevel@tonic-gate * (the "License"). You may not use this file except in compliance
7*7c478bd9Sstevel@tonic-gate * with the License.
8*7c478bd9Sstevel@tonic-gate *
9*7c478bd9Sstevel@tonic-gate * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*7c478bd9Sstevel@tonic-gate * or http://www.opensolaris.org/os/licensing.
11*7c478bd9Sstevel@tonic-gate * See the License for the specific language governing permissions
12*7c478bd9Sstevel@tonic-gate * and limitations under the License.
13*7c478bd9Sstevel@tonic-gate *
14*7c478bd9Sstevel@tonic-gate * When distributing Covered Code, include this CDDL HEADER in each
15*7c478bd9Sstevel@tonic-gate * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*7c478bd9Sstevel@tonic-gate * If applicable, add the following below this CDDL HEADER, with the
17*7c478bd9Sstevel@tonic-gate * fields enclosed by brackets "[]" replaced with your own identifying
18*7c478bd9Sstevel@tonic-gate * information: Portions Copyright [yyyy] [name of copyright owner]
19*7c478bd9Sstevel@tonic-gate *
20*7c478bd9Sstevel@tonic-gate * CDDL HEADER END
21*7c478bd9Sstevel@tonic-gate */
22*7c478bd9Sstevel@tonic-gate /*
23*7c478bd9Sstevel@tonic-gate * Copyright 2004 Sun Microsystems, Inc. All rights reserved.
24*7c478bd9Sstevel@tonic-gate * Use is subject to license terms.
25*7c478bd9Sstevel@tonic-gate */
26*7c478bd9Sstevel@tonic-gate
27*7c478bd9Sstevel@tonic-gate /*
28*7c478bd9Sstevel@tonic-gate * Utility functions for buffering output to stdout, stderr while
29*7c478bd9Sstevel@tonic-gate * process is grabbed. Prevents infamous deadlocks due to pfiles `pgrep xterm`
30*7c478bd9Sstevel@tonic-gate * and other varients.
31*7c478bd9Sstevel@tonic-gate */
32*7c478bd9Sstevel@tonic-gate
33*7c478bd9Sstevel@tonic-gate #include "libproc.h"
34*7c478bd9Sstevel@tonic-gate #include <stdio.h>
35*7c478bd9Sstevel@tonic-gate
36*7c478bd9Sstevel@tonic-gate static int cached_stdout_fd = -1;
37*7c478bd9Sstevel@tonic-gate static int cached_stderr_fd = -1;
38*7c478bd9Sstevel@tonic-gate static int initialized = 0;
39*7c478bd9Sstevel@tonic-gate
40*7c478bd9Sstevel@tonic-gate static char stdout_name[] = "/tmp/.stdoutXXXXXX";
41*7c478bd9Sstevel@tonic-gate static char stderr_name[] = "/tmp/.stderrXXXXXX";
42*7c478bd9Sstevel@tonic-gate
43*7c478bd9Sstevel@tonic-gate int
proc_initstdio(void)44*7c478bd9Sstevel@tonic-gate proc_initstdio(void)
45*7c478bd9Sstevel@tonic-gate {
46*7c478bd9Sstevel@tonic-gate int fd;
47*7c478bd9Sstevel@tonic-gate
48*7c478bd9Sstevel@tonic-gate (void) fflush(stdout);
49*7c478bd9Sstevel@tonic-gate (void) fflush(stderr);
50*7c478bd9Sstevel@tonic-gate
51*7c478bd9Sstevel@tonic-gate if ((cached_stdout_fd = dup(1)) < 0) {
52*7c478bd9Sstevel@tonic-gate return (-1);
53*7c478bd9Sstevel@tonic-gate }
54*7c478bd9Sstevel@tonic-gate
55*7c478bd9Sstevel@tonic-gate if ((cached_stderr_fd = dup(2)) < 0) {
56*7c478bd9Sstevel@tonic-gate (void) close(cached_stdout_fd);
57*7c478bd9Sstevel@tonic-gate return (-1);
58*7c478bd9Sstevel@tonic-gate }
59*7c478bd9Sstevel@tonic-gate
60*7c478bd9Sstevel@tonic-gate if ((fd = mkstemp(stdout_name)) < 0) {
61*7c478bd9Sstevel@tonic-gate (void) close(cached_stdout_fd);
62*7c478bd9Sstevel@tonic-gate (void) close(cached_stderr_fd);
63*7c478bd9Sstevel@tonic-gate return (-1);
64*7c478bd9Sstevel@tonic-gate }
65*7c478bd9Sstevel@tonic-gate
66*7c478bd9Sstevel@tonic-gate (void) unlink(stdout_name);
67*7c478bd9Sstevel@tonic-gate
68*7c478bd9Sstevel@tonic-gate if (dup2(fd, 1) < 0) {
69*7c478bd9Sstevel@tonic-gate (void) close(fd);
70*7c478bd9Sstevel@tonic-gate (void) close(cached_stdout_fd);
71*7c478bd9Sstevel@tonic-gate (void) close(cached_stderr_fd);
72*7c478bd9Sstevel@tonic-gate return (-1);
73*7c478bd9Sstevel@tonic-gate }
74*7c478bd9Sstevel@tonic-gate
75*7c478bd9Sstevel@tonic-gate (void) close(fd);
76*7c478bd9Sstevel@tonic-gate
77*7c478bd9Sstevel@tonic-gate
78*7c478bd9Sstevel@tonic-gate if ((fd = mkstemp(stderr_name)) < 0) {
79*7c478bd9Sstevel@tonic-gate (void) dup2(cached_stdout_fd, 1);
80*7c478bd9Sstevel@tonic-gate (void) close(cached_stdout_fd);
81*7c478bd9Sstevel@tonic-gate (void) close(cached_stderr_fd);
82*7c478bd9Sstevel@tonic-gate return (-1);
83*7c478bd9Sstevel@tonic-gate }
84*7c478bd9Sstevel@tonic-gate
85*7c478bd9Sstevel@tonic-gate (void) unlink(stderr_name);
86*7c478bd9Sstevel@tonic-gate
87*7c478bd9Sstevel@tonic-gate if (dup2(fd, 2) < 0) {
88*7c478bd9Sstevel@tonic-gate (void) close(fd);
89*7c478bd9Sstevel@tonic-gate (void) dup2(cached_stdout_fd, 1);
90*7c478bd9Sstevel@tonic-gate (void) close(cached_stdout_fd);
91*7c478bd9Sstevel@tonic-gate (void) dup2(cached_stderr_fd, 2);
92*7c478bd9Sstevel@tonic-gate (void) close(cached_stderr_fd);
93*7c478bd9Sstevel@tonic-gate (void) close(fd);
94*7c478bd9Sstevel@tonic-gate return (-1);
95*7c478bd9Sstevel@tonic-gate }
96*7c478bd9Sstevel@tonic-gate
97*7c478bd9Sstevel@tonic-gate (void) close(fd);
98*7c478bd9Sstevel@tonic-gate
99*7c478bd9Sstevel@tonic-gate initialized = 1;
100*7c478bd9Sstevel@tonic-gate
101*7c478bd9Sstevel@tonic-gate return (0);
102*7c478bd9Sstevel@tonic-gate }
103*7c478bd9Sstevel@tonic-gate
104*7c478bd9Sstevel@tonic-gate static int
copy_fd(int out,FILE * in,size_t len)105*7c478bd9Sstevel@tonic-gate copy_fd(int out, FILE *in, size_t len)
106*7c478bd9Sstevel@tonic-gate {
107*7c478bd9Sstevel@tonic-gate char buffer[8192];
108*7c478bd9Sstevel@tonic-gate int rlen, alen;
109*7c478bd9Sstevel@tonic-gate int errors = 0;
110*7c478bd9Sstevel@tonic-gate
111*7c478bd9Sstevel@tonic-gate rewind(in);
112*7c478bd9Sstevel@tonic-gate while (len > 0 && !errors) {
113*7c478bd9Sstevel@tonic-gate rlen = (len > sizeof (buffer)) ? sizeof (buffer) : len;
114*7c478bd9Sstevel@tonic-gate alen = read(fileno(in), buffer, rlen);
115*7c478bd9Sstevel@tonic-gate if (alen == rlen) {
116*7c478bd9Sstevel@tonic-gate if (write(out, buffer, alen) < alen)
117*7c478bd9Sstevel@tonic-gate errors++;
118*7c478bd9Sstevel@tonic-gate else
119*7c478bd9Sstevel@tonic-gate len -= alen;
120*7c478bd9Sstevel@tonic-gate }
121*7c478bd9Sstevel@tonic-gate else
122*7c478bd9Sstevel@tonic-gate errors++;
123*7c478bd9Sstevel@tonic-gate }
124*7c478bd9Sstevel@tonic-gate rewind(in);
125*7c478bd9Sstevel@tonic-gate return (errors);
126*7c478bd9Sstevel@tonic-gate }
127*7c478bd9Sstevel@tonic-gate
128*7c478bd9Sstevel@tonic-gate int
proc_flushstdio(void)129*7c478bd9Sstevel@tonic-gate proc_flushstdio(void)
130*7c478bd9Sstevel@tonic-gate {
131*7c478bd9Sstevel@tonic-gate size_t len;
132*7c478bd9Sstevel@tonic-gate int errors = 0;
133*7c478bd9Sstevel@tonic-gate
134*7c478bd9Sstevel@tonic-gate /*
135*7c478bd9Sstevel@tonic-gate * flush any pending IO
136*7c478bd9Sstevel@tonic-gate */
137*7c478bd9Sstevel@tonic-gate
138*7c478bd9Sstevel@tonic-gate if (!initialized)
139*7c478bd9Sstevel@tonic-gate return (-1);
140*7c478bd9Sstevel@tonic-gate
141*7c478bd9Sstevel@tonic-gate (void) fflush(stdout);
142*7c478bd9Sstevel@tonic-gate (void) fflush(stderr);
143*7c478bd9Sstevel@tonic-gate
144*7c478bd9Sstevel@tonic-gate if ((len = ftell(stdout)) > 0)
145*7c478bd9Sstevel@tonic-gate errors += copy_fd(cached_stdout_fd, stdout, len);
146*7c478bd9Sstevel@tonic-gate
147*7c478bd9Sstevel@tonic-gate
148*7c478bd9Sstevel@tonic-gate if ((len = ftell(stderr)) > 0)
149*7c478bd9Sstevel@tonic-gate errors += copy_fd(cached_stderr_fd, stderr, len);
150*7c478bd9Sstevel@tonic-gate
151*7c478bd9Sstevel@tonic-gate return (errors?-1:0);
152*7c478bd9Sstevel@tonic-gate }
153*7c478bd9Sstevel@tonic-gate
154*7c478bd9Sstevel@tonic-gate int
proc_finistdio(void)155*7c478bd9Sstevel@tonic-gate proc_finistdio(void)
156*7c478bd9Sstevel@tonic-gate {
157*7c478bd9Sstevel@tonic-gate if (!initialized)
158*7c478bd9Sstevel@tonic-gate return (-1);
159*7c478bd9Sstevel@tonic-gate
160*7c478bd9Sstevel@tonic-gate if (proc_flushstdio() != 0)
161*7c478bd9Sstevel@tonic-gate return (-1);
162*7c478bd9Sstevel@tonic-gate
163*7c478bd9Sstevel@tonic-gate (void) dup2(cached_stdout_fd, 1);
164*7c478bd9Sstevel@tonic-gate (void) close(cached_stdout_fd);
165*7c478bd9Sstevel@tonic-gate (void) dup2(cached_stderr_fd, 2);
166*7c478bd9Sstevel@tonic-gate (void) close(cached_stderr_fd);
167*7c478bd9Sstevel@tonic-gate
168*7c478bd9Sstevel@tonic-gate return (0);
169*7c478bd9Sstevel@tonic-gate }
170