1 8f314994 2004-03-02 devnull #define NOPLAN9DEFINES
2 76193d7c 2003-09-30 devnull #include "mk.h"
3 76193d7c 2003-09-30 devnull #include <sys/wait.h>
4 76193d7c 2003-09-30 devnull #include <signal.h>
5 76193d7c 2003-09-30 devnull #include <sys/stat.h>
6 76193d7c 2003-09-30 devnull #include <sys/time.h>
8 76193d7c 2003-09-30 devnull char *shell = "/bin/sh";
9 76193d7c 2003-09-30 devnull char *shellname = "sh";
11 76193d7c 2003-09-30 devnull extern char **environ;
13 76193d7c 2003-09-30 devnull static void
14 76193d7c 2003-09-30 devnull mkperror(char *s)
16 76193d7c 2003-09-30 devnull fprint(2, "%s: %r\n", s);
20 76193d7c 2003-09-30 devnull readenv(void)
22 76193d7c 2003-09-30 devnull char **p, *s;
25 76193d7c 2003-09-30 devnull for(p = environ; *p; p++){
26 f1bfc54e 2004-05-14 devnull /* rsc 5/5/2004 -- This misparses fn#cd={whatever}
27 76193d7c 2003-09-30 devnull s = shname(*p);
28 76193d7c 2003-09-30 devnull if(*s == '=') {
30 f1bfc54e 2004-05-14 devnull w = newword(s+1);
32 f1bfc54e 2004-05-14 devnull w = newword("");
34 f1bfc54e 2004-05-14 devnull s = strchr(*p, '=');
37 76193d7c 2003-09-30 devnull w = newword(s+1);
39 76193d7c 2003-09-30 devnull w = newword("");
40 76193d7c 2003-09-30 devnull if (symlook(*p, S_INTERNAL, 0))
41 76193d7c 2003-09-30 devnull continue;
42 76193d7c 2003-09-30 devnull s = strdup(*p);
43 76193d7c 2003-09-30 devnull setvar(s, (void *)w);
44 76193d7c 2003-09-30 devnull symlook(s, S_EXPORTED, (void*)"")->value = (void*)"";
49 76193d7c 2003-09-30 devnull * done on child side of fork, so parent's env is not affected
50 76193d7c 2003-09-30 devnull * and we don't care about freeing memory because we're going
51 76193d7c 2003-09-30 devnull * to exec immediately after this.
54 76193d7c 2003-09-30 devnull exportenv(Envy *e)
57 76193d7c 2003-09-30 devnull char **p;
58 92a0a8b6 2004-04-21 devnull static char buf[16384];
61 76193d7c 2003-09-30 devnull for(i = 0; e->name; e++, i++) {
62 76193d7c 2003-09-30 devnull p = (char**) Realloc(p, (i+2)*sizeof(char*));
63 76193d7c 2003-09-30 devnull if(e->values)
64 92a0a8b6 2004-04-21 devnull snprint(buf, sizeof buf, "%s=%s", e->name, wtos(e->values, IWS));
66 92a0a8b6 2004-04-21 devnull snprint(buf, sizeof buf, "%s=", e->name);
67 76193d7c 2003-09-30 devnull p[i] = strdup(buf);
69 76193d7c 2003-09-30 devnull p[i] = 0;
70 76193d7c 2003-09-30 devnull environ = p;
74 76193d7c 2003-09-30 devnull waitfor(char *msg)
76 76193d7c 2003-09-30 devnull int status;
79 76193d7c 2003-09-30 devnull *msg = 0;
80 76193d7c 2003-09-30 devnull pid = wait(&status);
81 76193d7c 2003-09-30 devnull if(pid > 0) {
82 76193d7c 2003-09-30 devnull if(status&0x7f) {
83 76193d7c 2003-09-30 devnull if(status&0x80)
84 76193d7c 2003-09-30 devnull snprint(msg, ERRMAX, "signal %d, core dumped", status&0x7f);
86 76193d7c 2003-09-30 devnull snprint(msg, ERRMAX, "signal %d", status&0x7f);
87 76193d7c 2003-09-30 devnull } else if(status&0xff00)
88 76193d7c 2003-09-30 devnull snprint(msg, ERRMAX, "exit(%d)", (status>>8)&0xff);
90 76193d7c 2003-09-30 devnull return pid;
94 76193d7c 2003-09-30 devnull expunge(int pid, char *msg)
96 76193d7c 2003-09-30 devnull if(strcmp(msg, "interrupt"))
97 76193d7c 2003-09-30 devnull kill(pid, SIGINT);
99 76193d7c 2003-09-30 devnull kill(pid, SIGHUP);
102 92a0a8b6 2004-04-21 devnull int mypid;
105 76193d7c 2003-09-30 devnull execsh(char *args, char *cmd, Bufblock *buf, Envy *e)
107 76193d7c 2003-09-30 devnull char *p;
108 76193d7c 2003-09-30 devnull int tot, n, pid, in[2], out[2];
110 76193d7c 2003-09-30 devnull if(buf && pipe(out) < 0){
111 76193d7c 2003-09-30 devnull mkperror("pipe");
114 76193d7c 2003-09-30 devnull pid = fork();
115 92a0a8b6 2004-04-21 devnull mypid = getpid();
116 76193d7c 2003-09-30 devnull if(pid < 0){
117 76193d7c 2003-09-30 devnull mkperror("mk fork");
120 76193d7c 2003-09-30 devnull if(pid == 0){
122 76193d7c 2003-09-30 devnull close(out[0]);
123 76193d7c 2003-09-30 devnull if(pipe(in) < 0){
124 76193d7c 2003-09-30 devnull mkperror("pipe");
127 76193d7c 2003-09-30 devnull pid = fork();
128 76193d7c 2003-09-30 devnull if(pid < 0){
129 76193d7c 2003-09-30 devnull mkperror("mk fork");
132 76193d7c 2003-09-30 devnull if(pid != 0){
133 76193d7c 2003-09-30 devnull dup2(in[0], 0);
134 76193d7c 2003-09-30 devnull if(buf){
135 76193d7c 2003-09-30 devnull dup2(out[1], 1);
136 76193d7c 2003-09-30 devnull close(out[1]);
138 76193d7c 2003-09-30 devnull close(in[0]);
139 76193d7c 2003-09-30 devnull close(in[1]);
141 76193d7c 2003-09-30 devnull exportenv(e);
142 76193d7c 2003-09-30 devnull if(shflags)
143 76193d7c 2003-09-30 devnull execl(shell, shellname, shflags, args, 0);
145 76193d7c 2003-09-30 devnull execl(shell, shellname, args, 0);
146 76193d7c 2003-09-30 devnull mkperror(shell);
147 76193d7c 2003-09-30 devnull _exit(1);
149 76193d7c 2003-09-30 devnull close(out[1]);
150 76193d7c 2003-09-30 devnull close(in[0]);
151 76193d7c 2003-09-30 devnull if(DEBUG(D_EXEC))
152 76193d7c 2003-09-30 devnull fprint(1, "starting: %s\n", cmd);
153 76193d7c 2003-09-30 devnull p = cmd+strlen(cmd);
154 76193d7c 2003-09-30 devnull while(cmd < p){
155 76193d7c 2003-09-30 devnull n = write(in[1], cmd, p-cmd);
156 76193d7c 2003-09-30 devnull if(n < 0)
158 76193d7c 2003-09-30 devnull cmd += n;
160 76193d7c 2003-09-30 devnull close(in[1]);
161 76193d7c 2003-09-30 devnull _exit(0);
163 76193d7c 2003-09-30 devnull if(buf){
164 76193d7c 2003-09-30 devnull close(out[1]);
165 76193d7c 2003-09-30 devnull tot = 0;
166 76193d7c 2003-09-30 devnull for(;;){
167 76193d7c 2003-09-30 devnull if (buf->current >= buf->end)
168 76193d7c 2003-09-30 devnull growbuf(buf);
169 76193d7c 2003-09-30 devnull n = read(out[0], buf->current, buf->end-buf->current);
170 76193d7c 2003-09-30 devnull if(n <= 0)
172 76193d7c 2003-09-30 devnull buf->current += n;
173 76193d7c 2003-09-30 devnull tot += n;
175 76193d7c 2003-09-30 devnull if (tot && buf->current[-1] == '\n')
176 76193d7c 2003-09-30 devnull buf->current--;
177 76193d7c 2003-09-30 devnull close(out[0]);
179 76193d7c 2003-09-30 devnull return pid;
183 76193d7c 2003-09-30 devnull pipecmd(char *cmd, Envy *e, int *fd)
185 76193d7c 2003-09-30 devnull int pid, pfd[2];
187 76193d7c 2003-09-30 devnull if(DEBUG(D_EXEC))
188 76193d7c 2003-09-30 devnull fprint(1, "pipecmd='%s'\n", cmd);/**/
190 76193d7c 2003-09-30 devnull if(fd && pipe(pfd) < 0){
191 76193d7c 2003-09-30 devnull mkperror("pipe");
194 76193d7c 2003-09-30 devnull pid = fork();
195 76193d7c 2003-09-30 devnull if(pid < 0){
196 76193d7c 2003-09-30 devnull mkperror("mk fork");
199 76193d7c 2003-09-30 devnull if(pid == 0){
201 76193d7c 2003-09-30 devnull close(pfd[0]);
202 76193d7c 2003-09-30 devnull dup2(pfd[1], 1);
203 76193d7c 2003-09-30 devnull close(pfd[1]);
206 76193d7c 2003-09-30 devnull exportenv(e);
207 76193d7c 2003-09-30 devnull if(shflags)
208 76193d7c 2003-09-30 devnull execl(shell, shellname, shflags, "-c", cmd, 0);
210 76193d7c 2003-09-30 devnull execl(shell, shellname, "-c", cmd, 0);
211 76193d7c 2003-09-30 devnull mkperror(shell);
212 76193d7c 2003-09-30 devnull _exit(1);
215 76193d7c 2003-09-30 devnull close(pfd[1]);
216 76193d7c 2003-09-30 devnull *fd = pfd[0];
218 76193d7c 2003-09-30 devnull return pid;
222 76193d7c 2003-09-30 devnull Exit(void)
224 76193d7c 2003-09-30 devnull while(wait(0) >= 0)
226 76193d7c 2003-09-30 devnull exits("error");
229 76193d7c 2003-09-30 devnull static struct
231 76193d7c 2003-09-30 devnull int sig;
232 76193d7c 2003-09-30 devnull char *msg;
233 76193d7c 2003-09-30 devnull } sigmsgs[] =
235 76193d7c 2003-09-30 devnull SIGALRM, "alarm",
236 76193d7c 2003-09-30 devnull SIGFPE, "sys: fp: fptrap",
237 76193d7c 2003-09-30 devnull SIGPIPE, "sys: write on closed pipe",
238 76193d7c 2003-09-30 devnull SIGILL, "sys: trap: illegal instruction",
239 92a0a8b6 2004-04-21 devnull // SIGSEGV, "sys: segmentation violation",
243 76193d7c 2003-09-30 devnull static void
244 76193d7c 2003-09-30 devnull notifyf(int sig)
248 76193d7c 2003-09-30 devnull for(i = 0; sigmsgs[i].msg; i++)
249 76193d7c 2003-09-30 devnull if(sigmsgs[i].sig == sig)
250 76193d7c 2003-09-30 devnull killchildren(sigmsgs[i].msg);
252 76193d7c 2003-09-30 devnull /* should never happen */
253 76193d7c 2003-09-30 devnull signal(sig, SIG_DFL);
254 76193d7c 2003-09-30 devnull kill(getpid(), sig);
258 76193d7c 2003-09-30 devnull catchnotes()
262 76193d7c 2003-09-30 devnull for(i = 0; sigmsgs[i].msg; i++)
263 76193d7c 2003-09-30 devnull signal(sigmsgs[i].sig, notifyf);
267 76193d7c 2003-09-30 devnull maketmp(int *pfd)
269 76193d7c 2003-09-30 devnull static char temp[] = "/tmp/mkargXXXXXX";
270 76193d7c 2003-09-30 devnull static char buf[100];
273 76193d7c 2003-09-30 devnull strcpy(buf, temp);
274 76193d7c 2003-09-30 devnull fd = mkstemp(buf);
275 76193d7c 2003-09-30 devnull if(fd < 0)
276 76193d7c 2003-09-30 devnull return 0;
277 76193d7c 2003-09-30 devnull *pfd = fd;
278 76193d7c 2003-09-30 devnull return buf;
282 76193d7c 2003-09-30 devnull chgtime(char *name)
284 76193d7c 2003-09-30 devnull if(access(name, 0) >= 0)
285 76193d7c 2003-09-30 devnull return utimes(name, 0);
286 76193d7c 2003-09-30 devnull return close(creat(name, 0666));
290 76193d7c 2003-09-30 devnull rcopy(char **to, Resub *match, int n)
293 76193d7c 2003-09-30 devnull char *p;
295 76193d7c 2003-09-30 devnull *to = match->s.sp; /* stem0 matches complete target */
296 76193d7c 2003-09-30 devnull for(to++, match++; --n > 0; to++, match++){
297 76193d7c 2003-09-30 devnull if(match->s.sp && match->e.ep){
298 76193d7c 2003-09-30 devnull p = match->e.ep;
301 76193d7c 2003-09-30 devnull *to = strdup(match->s.sp);
305 76193d7c 2003-09-30 devnull *to = 0;
310 76193d7c 2003-09-30 devnull mkmtime(char *name)
312 76193d7c 2003-09-30 devnull struct stat st;
314 76193d7c 2003-09-30 devnull if(stat(name, &st) < 0)
315 76193d7c 2003-09-30 devnull return 0;
317 76193d7c 2003-09-30 devnull return st.st_mtime;