1 #include <u.h>
2 #include <fcntl.h>
3 #include <unistd.h>
4 #include "threadimpl.h"
5 
6 static void efork(int[3], int[2], char*, char**);
7 static void
8 _threadexec(Channel *pidc, int fd[3], char *prog, char *args[], int freeargs)
9 {
10 	int pfd[2];
11 	int n, pid;
12 	char exitstr[ERRMAX];
13 
14 	_threaddebug(DBGEXEC, "threadexec %s", prog);
15 	
16 	/*
17 	 * We want threadexec to behave like exec; if exec succeeds,
18 	 * never return, and if it fails, return with errstr set.
19 	 * Unfortunately, the exec happens in another proc since
20 	 * we have to wait for the exec'ed process to finish.
21 	 * To provide the semantics, we open a pipe with the 
22 	 * write end close-on-exec and hand it to the proc that
23 	 * is doing the exec.  If the exec succeeds, the pipe will
24 	 * close so that our read below fails.  If the exec fails,
25 	 * then the proc doing the exec sends the errstr down the
26 	 * pipe to us.
27 	 */
28 	if(pipe(pfd) < 0)
29 		goto Bad;
30 	if(fcntl(pfd[0], F_SETFD, 1) < 0)
31 		goto Bad;
32 	if(fcntl(pfd[1], F_SETFD, 1) < 0)
33 		goto Bad;
34 
35 	switch(pid = fork()){
36 	case -1:
37 		close(pfd[0]);
38 		close(pfd[1]);
39 		goto Bad;
40 	case 0:
41 		efork(fd, pfd, prog, args);
42 		_exit(0);
43 	default:
44 		if(freeargs)
45 			free(args);
46 		break;
47 	}
48 
49 	close(pfd[1]);
50 	if((n = read(pfd[0], exitstr, ERRMAX-1)) > 0){	/* exec failed */
51 		exitstr[n] = '\0';
52 		errstr(exitstr, ERRMAX);
53 		close(pfd[0]);
54 		goto Bad;
55 	}
56 	close(pfd[0]);
57 	close(fd[0]);
58 	if(fd[1] != fd[0])
59 		close(fd[1]);
60 	if(fd[2] != fd[1] && fd[2] != fd[0])
61 		close(fd[2]);
62 	if(pidc)
63 		sendul(pidc, pid);
64 
65 	_threaddebug(DBGEXEC, "threadexec schedexecwait");
66 	threadexits(0);
67 
68 Bad:
69 	_threaddebug(DBGEXEC, "threadexec bad %r");
70 	if(pidc)
71 		sendul(pidc, ~0);
72 }
73 
74 void
75 threadexec(Channel *pidc, int fd[3], char *prog, char *args[])
76 {
77 	_threadexec(pidc, fd, prog, args, 0);
78 }
79 
80 /*
81  * The &f+1 trick doesn't work on SunOS, so we might 
82  * as well bite the bullet and do this correctly.
83  */
84 void
85 threadexecl(Channel *pidc, int fd[3], char *f, ...)
86 {
87 	char **args, *s;
88 	int n;
89 	va_list arg;
90 
91 	va_start(arg, f);
92 	for(n=0; va_arg(arg, char*) != 0; n++)
93 		;
94 	n++;
95 	va_end(arg);
96 
97 	args = malloc(n*sizeof(args[0]));
98 	if(args == nil){
99 		if(pidc)
100 			sendul(pidc, ~0);
101 		return;
102 	}
103 
104 	va_start(arg, f);
105 	for(n=0; (s=va_arg(arg, char*)) != 0; n++)
106 		args[n] = s;
107 	args[n] = 0;
108 	va_end(arg);
109 
110 	_threadexec(pidc, fd, f, args, 1);
111 }
112 
113 static void
114 efork(int stdfd[3], int fd[2], char *prog, char **args)
115 {
116 	char buf[ERRMAX];
117 	int i;
118 
119 	_threaddebug(DBGEXEC, "_schedexec %s -- calling execv", prog);
120 	dup(stdfd[0], 0);
121 	dup(stdfd[1], 1);
122 	dup(stdfd[2], 2);
123 	for(i=3; i<40; i++)
124 		if(i != fd[1])
125 			close(i);
126 	rfork(RFNOTEG);
127 	execvp(prog, args);
128 	_threaddebug(DBGEXEC, "_schedexec failed: %r");
129 	rerrstr(buf, sizeof buf);
130 	if(buf[0]=='\0')
131 		strcpy(buf, "exec failed");
132 	write(fd[1], buf, strlen(buf));
133 	close(fd[1]);
134 	_exits(buf);
135 }
136