Blame


1 a0a331aa 2005-01-06 devnull #include <u.h>
2 a0a331aa 2005-01-06 devnull #include <sys/time.h>
3 a0a331aa 2005-01-06 devnull #include <sys/resource.h>
4 ddb664da 2004-12-28 devnull #include "threadimpl.h"
5 ddb664da 2004-12-28 devnull
6 230ead2f 2005-01-04 devnull #undef waitpid
7 ddb664da 2004-12-28 devnull #undef pipe
8 ddb664da 2004-12-28 devnull #undef wait
9 ddb664da 2004-12-28 devnull
10 ddb664da 2004-12-28 devnull static int sigpid;
11 ddb664da 2004-12-28 devnull static int threadpassfd;
12 e9dbe11d 2005-01-11 devnull static int gotsigchld;
13 ddb664da 2004-12-28 devnull
14 ddb664da 2004-12-28 devnull static void
15 ddb664da 2004-12-28 devnull child(void)
16 ddb664da 2004-12-28 devnull {
17 1d2533d0 2004-12-28 devnull int status, pid;
18 a0a331aa 2005-01-06 devnull struct rlimit rl;
19 230ead2f 2005-01-04 devnull
20 230ead2f 2005-01-04 devnull notedisable("sys: child");
21 230ead2f 2005-01-04 devnull pid = waitpid(sigpid, &status, 0);
22 230ead2f 2005-01-04 devnull if(pid < 0){
23 230ead2f 2005-01-04 devnull fprint(2, "%s: wait: %r\n", argv0);
24 230ead2f 2005-01-04 devnull _exit(97);
25 230ead2f 2005-01-04 devnull }
26 230ead2f 2005-01-04 devnull if(WIFEXITED(status))
27 1d2533d0 2004-12-28 devnull _exit(WEXITSTATUS(status));
28 230ead2f 2005-01-04 devnull if(WIFSIGNALED(status)){
29 a0a331aa 2005-01-06 devnull /*
30 a0a331aa 2005-01-06 devnull * Make sure we don't scribble over the nice
31 a0a331aa 2005-01-06 devnull * core file that our child just wrote out.
32 a0a331aa 2005-01-06 devnull */
33 a0a331aa 2005-01-06 devnull rl.rlim_cur = 0;
34 a0a331aa 2005-01-06 devnull rl.rlim_max = 0;
35 a0a331aa 2005-01-06 devnull setrlimit(RLIMIT_CORE, &rl);
36 a0a331aa 2005-01-06 devnull
37 230ead2f 2005-01-04 devnull signal(WTERMSIG(status), SIG_DFL);
38 230ead2f 2005-01-04 devnull raise(WTERMSIG(status));
39 230ead2f 2005-01-04 devnull _exit(98); /* not reached */
40 230ead2f 2005-01-04 devnull }
41 37cc72d3 2005-01-11 devnull if(WIFSTOPPED(status)){
42 37cc72d3 2005-01-11 devnull fprint(2, "%s: wait pid %d stopped\n", argv0, pid);
43 37cc72d3 2005-01-11 devnull return;
44 37cc72d3 2005-01-11 devnull }
45 c8b6342d 2005-01-13 devnull #ifdef WIFCONTINUED
46 37cc72d3 2005-01-11 devnull if(WIFCONTINUED(status)){
47 37cc72d3 2005-01-11 devnull fprint(2, "%s: wait pid %d continued\n", argv0, pid);
48 37cc72d3 2005-01-11 devnull return;
49 37cc72d3 2005-01-11 devnull }
50 c8b6342d 2005-01-13 devnull #endif
51 37cc72d3 2005-01-11 devnull fprint(2, "%s: wait pid %d status 0x%ux\n", argv0, pid, status);
52 230ead2f 2005-01-04 devnull _exit(99);
53 ddb664da 2004-12-28 devnull }
54 ddb664da 2004-12-28 devnull
55 ddb664da 2004-12-28 devnull static void
56 ddb664da 2004-12-28 devnull sigpass(int sig)
57 ddb664da 2004-12-28 devnull {
58 e9dbe11d 2005-01-11 devnull if(sigpid == 1){
59 e9dbe11d 2005-01-11 devnull gotsigchld = 1;
60 e9dbe11d 2005-01-11 devnull return;
61 e9dbe11d 2005-01-11 devnull }
62 e9dbe11d 2005-01-11 devnull
63 ddb664da 2004-12-28 devnull if(sig == SIGCHLD)
64 ddb664da 2004-12-28 devnull child();
65 ddb664da 2004-12-28 devnull else
66 ddb664da 2004-12-28 devnull kill(sigpid, sig);
67 ddb664da 2004-12-28 devnull }
68 ddb664da 2004-12-28 devnull
69 e9dbe11d 2005-01-11 devnull static int sigs[] =
70 e9dbe11d 2005-01-11 devnull {
71 e9dbe11d 2005-01-11 devnull SIGHUP, SIGINT, SIGQUIT, SIGILL,
72 e9dbe11d 2005-01-11 devnull SIGTRAP, SIGABRT, SIGBUS, SIGFPE,
73 e9dbe11d 2005-01-11 devnull SIGUSR1, SIGSEGV, SIGUSR2, SIGPIPE,
74 e9dbe11d 2005-01-11 devnull SIGALRM, SIGTERM, SIGCHLD, SIGSTOP,
75 e9dbe11d 2005-01-11 devnull SIGTSTP, SIGTTIN, SIGTTOU, SIGURG,
76 e9dbe11d 2005-01-11 devnull SIGXCPU, SIGXFSZ, SIGVTALRM, SIGPROF,
77 1ac19816 2005-01-14 devnull SIGWINCH, SIGIO,
78 1ac19816 2005-01-14 devnull #ifdef SIGEMT
79 1ac19816 2005-01-14 devnull SIGEMT,
80 1ac19816 2005-01-14 devnull #endif
81 1ac19816 2005-01-14 devnull #ifdef SIGPWR
82 1ac19816 2005-01-14 devnull SIGPWR,
83 1ac19816 2005-01-14 devnull #endif
84 1ac19816 2005-01-14 devnull #ifdef SIGINFO
85 1ac19816 2005-01-14 devnull SIGINFO,
86 1ac19816 2005-01-14 devnull #endif
87 1ac19816 2005-01-14 devnull SIGSYS
88 e9dbe11d 2005-01-11 devnull };
89 e9dbe11d 2005-01-11 devnull
90 ddb664da 2004-12-28 devnull void
91 ddb664da 2004-12-28 devnull _threadsetupdaemonize(void)
92 ddb664da 2004-12-28 devnull {
93 ddb664da 2004-12-28 devnull int i, n, pid;
94 ddb664da 2004-12-28 devnull int p[2];
95 ddb664da 2004-12-28 devnull char buf[20];
96 ddb664da 2004-12-28 devnull
97 ddb664da 2004-12-28 devnull sigpid = 1;
98 ddb664da 2004-12-28 devnull
99 ddb664da 2004-12-28 devnull if(pipe(p) < 0)
100 ddb664da 2004-12-28 devnull sysfatal("passer pipe: %r");
101 ddb664da 2004-12-28 devnull
102 ddb664da 2004-12-28 devnull /* hide these somewhere they won't cause harm */
103 a0d2173c 2005-01-16 devnull /* can't go too high: NetBSD max is 64, for example */
104 a0d2173c 2005-01-16 devnull if(dup(p[0], 28) < 0 || dup(p[1], 29) < 0)
105 ddb664da 2004-12-28 devnull sysfatal("passer pipe dup: %r");
106 ddb664da 2004-12-28 devnull close(p[0]);
107 ddb664da 2004-12-28 devnull close(p[1]);
108 a0d2173c 2005-01-16 devnull p[0] = 28;
109 a0d2173c 2005-01-16 devnull p[1] = 29;
110 ddb664da 2004-12-28 devnull
111 ddb664da 2004-12-28 devnull /* close on exec */
112 ddb664da 2004-12-28 devnull if(fcntl(p[0], F_SETFD, 1) < 0 || fcntl(p[1], F_SETFD, 1) < 0)
113 ddb664da 2004-12-28 devnull sysfatal("passer pipe pipe fcntl: %r");
114 ddb664da 2004-12-28 devnull
115 230ead2f 2005-01-04 devnull noteenable("sys: child");
116 1d2533d0 2004-12-28 devnull signal(SIGCHLD, sigpass);
117 ddb664da 2004-12-28 devnull switch(pid = fork()){
118 ddb664da 2004-12-28 devnull case -1:
119 ddb664da 2004-12-28 devnull sysfatal("passer fork: %r");
120 ddb664da 2004-12-28 devnull default:
121 ddb664da 2004-12-28 devnull close(p[1]);
122 ddb664da 2004-12-28 devnull break;
123 ddb664da 2004-12-28 devnull case 0:
124 230ead2f 2005-01-04 devnull notedisable("sys: child");
125 1d2533d0 2004-12-28 devnull signal(SIGCHLD, SIG_DFL);
126 e9dbe11d 2005-01-11 devnull /* rfork(RFNOTEG); */
127 ddb664da 2004-12-28 devnull close(p[0]);
128 ddb664da 2004-12-28 devnull threadpassfd = p[1];
129 ddb664da 2004-12-28 devnull return;
130 ddb664da 2004-12-28 devnull }
131 ddb664da 2004-12-28 devnull
132 ddb664da 2004-12-28 devnull sigpid = pid;
133 e9dbe11d 2005-01-11 devnull if(gotsigchld)
134 e9dbe11d 2005-01-11 devnull sigpass(SIGCHLD);
135 e9dbe11d 2005-01-11 devnull
136 e9dbe11d 2005-01-11 devnull for(i=0; i<nelem(sigs); i++){
137 ddb664da 2004-12-28 devnull struct sigaction sa;
138 ddb664da 2004-12-28 devnull
139 ddb664da 2004-12-28 devnull memset(&sa, 0, sizeof sa);
140 ddb664da 2004-12-28 devnull sa.sa_handler = sigpass;
141 ddb664da 2004-12-28 devnull sa.sa_flags |= SA_RESTART;
142 e9dbe11d 2005-01-11 devnull sigaction(sigs[i], &sa, nil);
143 ddb664da 2004-12-28 devnull }
144 ddb664da 2004-12-28 devnull
145 ddb664da 2004-12-28 devnull for(;;){
146 ddb664da 2004-12-28 devnull n = read(p[0], buf, sizeof buf-1);
147 230ead2f 2005-01-04 devnull if(n == 0){ /* program exited */
148 ddb664da 2004-12-28 devnull child();
149 230ead2f 2005-01-04 devnull }
150 ddb664da 2004-12-28 devnull if(n > 0)
151 ddb664da 2004-12-28 devnull break;
152 d54ead7f 2004-12-28 devnull print("passer read: %r\n");
153 ddb664da 2004-12-28 devnull }
154 ddb664da 2004-12-28 devnull buf[n] = 0;
155 ddb664da 2004-12-28 devnull _exit(atoi(buf));
156 ddb664da 2004-12-28 devnull }
157 ddb664da 2004-12-28 devnull
158 ddb664da 2004-12-28 devnull void
159 230ead2f 2005-01-04 devnull _threaddaemonize(void)
160 ddb664da 2004-12-28 devnull {
161 1d2533d0 2004-12-28 devnull if(threadpassfd >= 0){
162 1d2533d0 2004-12-28 devnull write(threadpassfd, "0", 1);
163 1d2533d0 2004-12-28 devnull close(threadpassfd);
164 1d2533d0 2004-12-28 devnull threadpassfd = -1;
165 1d2533d0 2004-12-28 devnull }
166 ddb664da 2004-12-28 devnull }