source: heimdal/trunk/lib/roken/simple_exec.c

Last change on this file was 1, checked in by Paul Smedley, 10 years ago

Initial commit of Heimdal 1.5.3
File size: 7.5 KB
Line 
1/*
2 * Copyright (c) 1998 - 2001, 2004 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 *
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 *
17 * 3. Neither the name of the Institute 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 INSTITUTE 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 INSTITUTE 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 <config.h>
35
36#include <stdarg.h>
37#include <stdlib.h>
38#ifdef HAVE_SYS_TYPES_H
39#include <sys/types.h>
40#endif
41#ifdef HAVE_SYS_WAIT_H
42#include <sys/wait.h>
43#endif
44#ifdef HAVE_UNISTD_H
45#include <unistd.h>
46#endif
47#include <errno.h>
48
49#include "roken.h"
50
51#define EX_NOEXEC 126
52#define EX_NOTFOUND 127
53
54/* return values:
55 SE_E_UNSPECIFIED on `unspecified' system errors
56 SE_E_FORKFAILED on fork failures
57 SE_E_WAITPIDFAILED on waitpid errors
58 SE_E_EXECTIMEOUT exec timeout
59 0- is return value from subprocess
60 SE_E_NOEXEC if the program couldn't be executed
61 SE_E_NOTFOUND if the program couldn't be found
62 128- is 128 + signal that killed subprocess
63
64 possible values `func' can return:
65 ((time_t)-2) exit loop w/o killing child and return
66 `exec timeout'/-4 from simple_exec
67 ((time_t)-1) kill child with SIGTERM and wait for child to exit
68 0 don't timeout again
69 n seconds to next timeout
70 */
71
72static int sig_alarm;
73
74static RETSIGTYPE
75sigtimeout(int sig)
76{
77 sig_alarm = 1;
78 SIGRETURN(0);
79}
80
81ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
82wait_for_process_timed(pid_t pid, time_t (*func)(void *),
83 void *ptr, time_t timeout)
84{
85 RETSIGTYPE (*old_func)(int sig) = NULL;
86 unsigned int oldtime = 0;
87 int ret;
88
89 sig_alarm = 0;
90
91 if (func) {
92 old_func = signal(SIGALRM, sigtimeout);
93 oldtime = alarm(timeout);
94 }
95
96 while(1) {
97 int status;
98
99 while(waitpid(pid, &status, 0) < 0) {
100 if (errno != EINTR) {
101 ret = SE_E_WAITPIDFAILED;
102 goto out;
103 }
104 if (func == NULL)
105 continue;
106 if (sig_alarm == 0)
107 continue;
108 timeout = (*func)(ptr);
109 if (timeout == (time_t)-1) {
110 kill(pid, SIGTERM);
111 continue;
112 } else if (timeout == (time_t)-2) {
113 ret = SE_E_EXECTIMEOUT;
114 goto out;
115 }
116 alarm(timeout);
117 }
118 if(WIFSTOPPED(status))
119 continue;
120 if(WIFEXITED(status)) {
121 ret = WEXITSTATUS(status);
122 break;
123 }
124 if(WIFSIGNALED(status)) {
125 ret = WTERMSIG(status) + 128;
126 break;
127 }
128 }
129 out:
130 if (func) {
131 signal(SIGALRM, old_func);
132 alarm(oldtime);
133 }
134 return ret;
135}
136
137ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
138wait_for_process(pid_t pid)
139{
140 return wait_for_process_timed(pid, NULL, NULL, 0);
141}
142
143ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
144pipe_execv(FILE **stdin_fd, FILE **stdout_fd, FILE **stderr_fd,
145 const char *file, ...)
146{
147 int in_fd[2], out_fd[2], err_fd[2];
148 pid_t pid;
149 va_list ap;
150 char **argv;
151
152 if(stdin_fd != NULL)
153 pipe(in_fd);
154 if(stdout_fd != NULL)
155 pipe(out_fd);
156 if(stderr_fd != NULL)
157 pipe(err_fd);
158 pid = fork();
159 switch(pid) {
160 case 0:
161 va_start(ap, file);
162 argv = vstrcollect(&ap);
163 va_end(ap);
164 if(argv == NULL)
165 exit(-1);
166
167 /* close pipes we're not interested in */
168 if(stdin_fd != NULL)
169 close(in_fd[1]);
170 if(stdout_fd != NULL)
171 close(out_fd[0]);
172 if(stderr_fd != NULL)
173 close(err_fd[0]);
174
175 /* pipe everything caller doesn't care about to /dev/null */
176 if(stdin_fd == NULL)
177 in_fd[0] = open(_PATH_DEVNULL, O_RDONLY);
178 if(stdout_fd == NULL)
179 out_fd[1] = open(_PATH_DEVNULL, O_WRONLY);
180 if(stderr_fd == NULL)
181 err_fd[1] = open(_PATH_DEVNULL, O_WRONLY);
182
183 /* move to proper descriptors */
184 if(in_fd[0] != STDIN_FILENO) {
185 dup2(in_fd[0], STDIN_FILENO);
186 close(in_fd[0]);
187 }
188 if(out_fd[1] != STDOUT_FILENO) {
189 dup2(out_fd[1], STDOUT_FILENO);
190 close(out_fd[1]);
191 }
192 if(err_fd[1] != STDERR_FILENO) {
193 dup2(err_fd[1], STDERR_FILENO);
194 close(err_fd[1]);
195 }
196
197 closefrom(3);
198
199 execv(file, argv);
200 exit((errno == ENOENT) ? EX_NOTFOUND : EX_NOEXEC);
201 case -1:
202 if(stdin_fd != NULL) {
203 close(in_fd[0]);
204 close(in_fd[1]);
205 }
206 if(stdout_fd != NULL) {
207 close(out_fd[0]);
208 close(out_fd[1]);
209 }
210 if(stderr_fd != NULL) {
211 close(err_fd[0]);
212 close(err_fd[1]);
213 }
214 return SE_E_FORKFAILED;
215 default:
216 if(stdin_fd != NULL) {
217 close(in_fd[0]);
218 *stdin_fd = fdopen(in_fd[1], "w");
219 }
220 if(stdout_fd != NULL) {
221 close(out_fd[1]);
222 *stdout_fd = fdopen(out_fd[0], "r");
223 }
224 if(stderr_fd != NULL) {
225 close(err_fd[1]);
226 *stderr_fd = fdopen(err_fd[0], "r");
227 }
228 }
229 return pid;
230}
231
232ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
233simple_execvp_timed(const char *file, char *const args[],
234 time_t (*func)(void *), void *ptr, time_t timeout)
235{
236 pid_t pid = fork();
237 switch(pid){
238 case -1:
239 return SE_E_FORKFAILED;
240 case 0:
241 execvp(file, args);
242 exit((errno == ENOENT) ? EX_NOTFOUND : EX_NOEXEC);
243 default:
244 return wait_for_process_timed(pid, func, ptr, timeout);
245 }
246}
247
248ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
249simple_execvp(const char *file, char *const args[])
250{
251 return simple_execvp_timed(file, args, NULL, NULL, 0);
252}
253
254/* gee, I'd like a execvpe */
255ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
256simple_execve_timed(const char *file, char *const args[], char *const envp[],
257 time_t (*func)(void *), void *ptr, time_t timeout)
258{
259 pid_t pid = fork();
260 switch(pid){
261 case -1:
262 return SE_E_FORKFAILED;
263 case 0:
264 execve(file, args, envp);
265 exit((errno == ENOENT) ? EX_NOTFOUND : EX_NOEXEC);
266 default:
267 return wait_for_process_timed(pid, func, ptr, timeout);
268 }
269}
270
271ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
272simple_execve(const char *file, char *const args[], char *const envp[])
273{
274 return simple_execve_timed(file, args, envp, NULL, NULL, 0);
275}
276
277ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
278simple_execlp(const char *file, ...)
279{
280 va_list ap;
281 char **argv;
282 int ret;
283
284 va_start(ap, file);
285 argv = vstrcollect(&ap);
286 va_end(ap);
287 if(argv == NULL)
288 return SE_E_UNSPECIFIED;
289 ret = simple_execvp(file, argv);
290 free(argv);
291 return ret;
292}
293
294ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
295simple_execle(const char *file, ... /* ,char *const envp[] */)
296{
297 va_list ap;
298 char **argv;
299 char *const* envp;
300 int ret;
301
302 va_start(ap, file);
303 argv = vstrcollect(&ap);
304 envp = va_arg(ap, char **);
305 va_end(ap);
306 if(argv == NULL)
307 return SE_E_UNSPECIFIED;
308 ret = simple_execve(file, argv, envp);
309 free(argv);
310 return ret;
311}
Note: See TracBrowser for help on using the repository browser.