Blame


1 f1e8706d 2004-12-27 devnull #include "threadimpl.h"
2 f1e8706d 2004-12-27 devnull
3 f1e8706d 2004-12-27 devnull /*
4 f1e8706d 2004-12-27 devnull * spin locks
5 f1e8706d 2004-12-27 devnull */
6 f1e8706d 2004-12-27 devnull extern int _tas(int*);
7 f1e8706d 2004-12-27 devnull
8 f1e8706d 2004-12-27 devnull void
9 f1e8706d 2004-12-27 devnull _threadunlock(Lock *l, ulong pc)
10 f1e8706d 2004-12-27 devnull {
11 f1e8706d 2004-12-27 devnull USED(pc);
12 f1e8706d 2004-12-27 devnull
13 f1e8706d 2004-12-27 devnull l->held = 0;
14 f1e8706d 2004-12-27 devnull }
15 f1e8706d 2004-12-27 devnull
16 f1e8706d 2004-12-27 devnull int
17 f1e8706d 2004-12-27 devnull _threadlock(Lock *l, int block, ulong pc)
18 f1e8706d 2004-12-27 devnull {
19 f1e8706d 2004-12-27 devnull int i;
20 f1e8706d 2004-12-27 devnull
21 f1e8706d 2004-12-27 devnull USED(pc);
22 f1e8706d 2004-12-27 devnull
23 f1e8706d 2004-12-27 devnull /* once fast */
24 f1e8706d 2004-12-27 devnull if(!_tas(&l->held))
25 f1e8706d 2004-12-27 devnull return 1;
26 f1e8706d 2004-12-27 devnull if(!block)
27 f1e8706d 2004-12-27 devnull return 0;
28 f1e8706d 2004-12-27 devnull
29 f1e8706d 2004-12-27 devnull /* a thousand times pretty fast */
30 f1e8706d 2004-12-27 devnull for(i=0; i<1000; i++){
31 f1e8706d 2004-12-27 devnull if(!_tas(&l->held))
32 f1e8706d 2004-12-27 devnull return 1;
33 f1e8706d 2004-12-27 devnull sched_yield();
34 f1e8706d 2004-12-27 devnull }
35 f1e8706d 2004-12-27 devnull /* now nice and slow */
36 f1e8706d 2004-12-27 devnull for(i=0; i<1000; i++){
37 f1e8706d 2004-12-27 devnull if(!_tas(&l->held))
38 f1e8706d 2004-12-27 devnull return 1;
39 f1e8706d 2004-12-27 devnull usleep(100*1000);
40 f1e8706d 2004-12-27 devnull }
41 f1e8706d 2004-12-27 devnull /* take your time */
42 f1e8706d 2004-12-27 devnull while(_tas(&l->held))
43 f1e8706d 2004-12-27 devnull usleep(1000*1000);
44 f1e8706d 2004-12-27 devnull return 1;
45 f1e8706d 2004-12-27 devnull }
46 f1e8706d 2004-12-27 devnull
47 f1e8706d 2004-12-27 devnull /*
48 f1e8706d 2004-12-27 devnull * sleep and wakeup
49 f1e8706d 2004-12-27 devnull */
50 f1e8706d 2004-12-27 devnull static void
51 f1e8706d 2004-12-27 devnull ign(int x)
52 f1e8706d 2004-12-27 devnull {
53 f1e8706d 2004-12-27 devnull USED(x);
54 f1e8706d 2004-12-27 devnull }
55 f1e8706d 2004-12-27 devnull
56 f1e8706d 2004-12-27 devnull static void /*__attribute__((constructor))*/
57 f1e8706d 2004-12-27 devnull ignusr1(int restart)
58 f1e8706d 2004-12-27 devnull {
59 f1e8706d 2004-12-27 devnull struct sigaction sa;
60 f1e8706d 2004-12-27 devnull
61 f1e8706d 2004-12-27 devnull memset(&sa, 0, sizeof sa);
62 f1e8706d 2004-12-27 devnull sa.sa_handler = ign;
63 f1e8706d 2004-12-27 devnull sigemptyset(&sa.sa_mask);
64 f1e8706d 2004-12-27 devnull sigaddset(&sa.sa_mask, SIGUSR1);
65 f1e8706d 2004-12-27 devnull if(restart)
66 f1e8706d 2004-12-27 devnull sa.sa_flags = SA_RESTART;
67 f1e8706d 2004-12-27 devnull sigaction(SIGUSR1, &sa, nil);
68 f1e8706d 2004-12-27 devnull }
69 f1e8706d 2004-12-27 devnull
70 f1e8706d 2004-12-27 devnull void
71 f1e8706d 2004-12-27 devnull _procsleep(_Procrendez *r)
72 f1e8706d 2004-12-27 devnull {
73 f1e8706d 2004-12-27 devnull sigset_t mask;
74 f1e8706d 2004-12-27 devnull
75 f1e8706d 2004-12-27 devnull /*
76 f1e8706d 2004-12-27 devnull * Go to sleep.
77 f1e8706d 2004-12-27 devnull *
78 f1e8706d 2004-12-27 devnull * Block USR1, set the handler to interrupt system calls,
79 f1e8706d 2004-12-27 devnull * unlock the vouslock so our waker can wake us,
80 f1e8706d 2004-12-27 devnull * and then suspend.
81 f1e8706d 2004-12-27 devnull */
82 955a2ca7 2004-12-27 devnull again:
83 f1e8706d 2004-12-27 devnull r->asleep = 1;
84 f1e8706d 2004-12-27 devnull r->pid = getpid();
85 f1e8706d 2004-12-27 devnull
86 f1e8706d 2004-12-27 devnull sigprocmask(SIG_SETMASK, nil, &mask);
87 f1e8706d 2004-12-27 devnull sigaddset(&mask, SIGUSR1);
88 f1e8706d 2004-12-27 devnull sigprocmask(SIG_SETMASK, &mask, nil);
89 f1e8706d 2004-12-27 devnull ignusr1(0);
90 f1e8706d 2004-12-27 devnull unlock(r->l);
91 f1e8706d 2004-12-27 devnull sigdelset(&mask, SIGUSR1);
92 f1e8706d 2004-12-27 devnull sigsuspend(&mask);
93 f1e8706d 2004-12-27 devnull
94 f1e8706d 2004-12-27 devnull /*
95 f1e8706d 2004-12-27 devnull * We're awake. Make USR1 not interrupt system calls.
96 f1e8706d 2004-12-27 devnull */
97 f1e8706d 2004-12-27 devnull lock(r->l);
98 955a2ca7 2004-12-27 devnull ignusr1(1);
99 955a2ca7 2004-12-27 devnull if(r->asleep && r->pid == getpid()){
100 955a2ca7 2004-12-27 devnull /* Didn't really wake up - signal from something else */
101 955a2ca7 2004-12-27 devnull goto again;
102 955a2ca7 2004-12-27 devnull }
103 f1e8706d 2004-12-27 devnull }
104 f1e8706d 2004-12-27 devnull
105 f1e8706d 2004-12-27 devnull void
106 f1e8706d 2004-12-27 devnull _procwakeup(_Procrendez *r)
107 f1e8706d 2004-12-27 devnull {
108 f1e8706d 2004-12-27 devnull if(r->asleep){
109 f1e8706d 2004-12-27 devnull r->asleep = 0;
110 f1e8706d 2004-12-27 devnull assert(r->pid >= 1);
111 f1e8706d 2004-12-27 devnull kill(r->pid, SIGUSR1);
112 f1e8706d 2004-12-27 devnull }
113 f1e8706d 2004-12-27 devnull }
114 f1e8706d 2004-12-27 devnull
115 f1e8706d 2004-12-27 devnull /*
116 f1e8706d 2004-12-27 devnull * process creation and exit
117 f1e8706d 2004-12-27 devnull */
118 f1e8706d 2004-12-27 devnull typedef struct Stackfree Stackfree;
119 f1e8706d 2004-12-27 devnull struct Stackfree
120 f1e8706d 2004-12-27 devnull {
121 f1e8706d 2004-12-27 devnull Stackfree *next;
122 f1e8706d 2004-12-27 devnull int pid;
123 f1e8706d 2004-12-27 devnull };
124 f1e8706d 2004-12-27 devnull static Lock stacklock;
125 f1e8706d 2004-12-27 devnull static Stackfree *stackfree;
126 f1e8706d 2004-12-27 devnull
127 f1e8706d 2004-12-27 devnull static void
128 f1e8706d 2004-12-27 devnull delayfreestack(uchar *stk)
129 f1e8706d 2004-12-27 devnull {
130 f1e8706d 2004-12-27 devnull Stackfree *sf;
131 f1e8706d 2004-12-27 devnull
132 f1e8706d 2004-12-27 devnull sf = (Stackfree*)stk;
133 f1e8706d 2004-12-27 devnull sf->pid = getpid();
134 f1e8706d 2004-12-27 devnull lock(&stacklock);
135 f1e8706d 2004-12-27 devnull sf->next = stackfree;
136 f1e8706d 2004-12-27 devnull stackfree = sf;
137 f1e8706d 2004-12-27 devnull unlock(&stacklock);
138 f1e8706d 2004-12-27 devnull }
139 f1e8706d 2004-12-27 devnull
140 f1e8706d 2004-12-27 devnull static void
141 f1e8706d 2004-12-27 devnull dofreestacks(void)
142 f1e8706d 2004-12-27 devnull {
143 f1e8706d 2004-12-27 devnull Stackfree *sf, *last, *next;
144 f1e8706d 2004-12-27 devnull
145 f1e8706d 2004-12-27 devnull if(stackfree==nil || !canlock(&stacklock))
146 f1e8706d 2004-12-27 devnull return;
147 f1e8706d 2004-12-27 devnull
148 f1e8706d 2004-12-27 devnull for(last=nil,sf=stackfree; sf; last=sf,sf=next){
149 f1e8706d 2004-12-27 devnull next = sf->next;
150 f1e8706d 2004-12-27 devnull if(sf->pid >= 1 && kill(sf->pid, 0) < 0 && errno == ESRCH){
151 f1e8706d 2004-12-27 devnull free(sf);
152 2c87dda8 2004-12-28 devnull if(last)
153 2c87dda8 2004-12-28 devnull last->next = next;
154 2c87dda8 2004-12-28 devnull else
155 2c87dda8 2004-12-28 devnull stackfree = next;
156 f1e8706d 2004-12-27 devnull sf = last;
157 f1e8706d 2004-12-27 devnull }
158 f1e8706d 2004-12-27 devnull }
159 2c87dda8 2004-12-28 devnull unlock(&stacklock);
160 2c87dda8 2004-12-28 devnull }
161 f1e8706d 2004-12-27 devnull
162 f1e8706d 2004-12-27 devnull static int
163 f1e8706d 2004-12-27 devnull startprocfn(void *v)
164 f1e8706d 2004-12-27 devnull {
165 f1e8706d 2004-12-27 devnull void **a;
166 f1e8706d 2004-12-27 devnull uchar *stk;
167 f1e8706d 2004-12-27 devnull void (*fn)(void*);
168 f1e8706d 2004-12-27 devnull Proc *p;
169 f1e8706d 2004-12-27 devnull
170 f1e8706d 2004-12-27 devnull a = (void**)v;
171 f1e8706d 2004-12-27 devnull fn = a[0];
172 f1e8706d 2004-12-27 devnull p = a[1];
173 f1e8706d 2004-12-27 devnull stk = a[2];
174 f1e8706d 2004-12-27 devnull free(a);
175 f1e8706d 2004-12-27 devnull p->osprocid = getpid();
176 f1e8706d 2004-12-27 devnull
177 f1e8706d 2004-12-27 devnull (*fn)(p);
178 f1e8706d 2004-12-27 devnull
179 f1e8706d 2004-12-27 devnull delayfreestack(stk);
180 f1e8706d 2004-12-27 devnull _exit(0);
181 f1e8706d 2004-12-27 devnull return 0;
182 f1e8706d 2004-12-27 devnull }
183 f1e8706d 2004-12-27 devnull
184 f1e8706d 2004-12-27 devnull /*
185 f1e8706d 2004-12-27 devnull * indirect through here so that parent need not wait for child zombie
186 f1e8706d 2004-12-27 devnull *
187 f1e8706d 2004-12-27 devnull * slight race - if child exits and then another process starts before we
188 f1e8706d 2004-12-27 devnull * manage to exit, we'll be running on a freed stack.
189 f1e8706d 2004-12-27 devnull */
190 f1e8706d 2004-12-27 devnull static int
191 f1e8706d 2004-12-27 devnull trampnowait(void *v)
192 f1e8706d 2004-12-27 devnull {
193 f1e8706d 2004-12-27 devnull void **a;
194 f1e8706d 2004-12-27 devnull
195 f1e8706d 2004-12-27 devnull a = (void*)v;
196 f1e8706d 2004-12-27 devnull *(int*)a[3] = clone(startprocfn, a[2]+65536-512, CLONE_VM|CLONE_FILES, a);
197 f1e8706d 2004-12-27 devnull _exit(0);
198 f1e8706d 2004-12-27 devnull return 0;
199 f1e8706d 2004-12-27 devnull }
200 f1e8706d 2004-12-27 devnull
201 f1e8706d 2004-12-27 devnull void
202 f1e8706d 2004-12-27 devnull _procstart(Proc *p, void (*fn)(Proc*))
203 f1e8706d 2004-12-27 devnull {
204 f1e8706d 2004-12-27 devnull void **a;
205 f1e8706d 2004-12-27 devnull uchar *stk;
206 f1e8706d 2004-12-27 devnull int pid, kidpid, status;
207 f1e8706d 2004-12-27 devnull
208 f1e8706d 2004-12-27 devnull dofreestacks();
209 f1e8706d 2004-12-27 devnull a = malloc(4*sizeof a[0]);
210 f1e8706d 2004-12-27 devnull if(a == nil)
211 f1e8706d 2004-12-27 devnull sysfatal("_procstart malloc: %r");
212 f1e8706d 2004-12-27 devnull stk = malloc(65536);
213 f1e8706d 2004-12-27 devnull if(stk == nil)
214 f1e8706d 2004-12-27 devnull sysfatal("_procstart malloc stack: %r");
215 f1e8706d 2004-12-27 devnull
216 f1e8706d 2004-12-27 devnull a[0] = fn;
217 f1e8706d 2004-12-27 devnull a[1] = p;
218 f1e8706d 2004-12-27 devnull a[2] = stk;
219 f1e8706d 2004-12-27 devnull a[3] = &kidpid;
220 f1e8706d 2004-12-27 devnull kidpid = -1;
221 f1e8706d 2004-12-27 devnull
222 f1e8706d 2004-12-27 devnull pid = clone(trampnowait, stk+65536-16, CLONE_VM|CLONE_FILES, a);
223 f1e8706d 2004-12-27 devnull if(pid > 0)
224 f1e8706d 2004-12-27 devnull if(wait4(pid, &status, __WALL, 0) < 0)
225 f1e8706d 2004-12-27 devnull fprint(2, "ffork wait4: %r\n");
226 f1e8706d 2004-12-27 devnull if(pid < 0 || kidpid < 0){
227 f1e8706d 2004-12-27 devnull fprint(2, "_procstart clone: %r\n");
228 f1e8706d 2004-12-27 devnull abort();
229 f1e8706d 2004-12-27 devnull }
230 f1e8706d 2004-12-27 devnull }
231 f1e8706d 2004-12-27 devnull
232 f1e8706d 2004-12-27 devnull static char *threadexitsmsg;
233 f1e8706d 2004-12-27 devnull void
234 f1e8706d 2004-12-27 devnull sigusr2handler(int s)
235 f1e8706d 2004-12-27 devnull {
236 2c87dda8 2004-12-28 devnull /* fprint(2, "%d usr2 %d\n", time(0), getpid()); */
237 f1e8706d 2004-12-27 devnull if(threadexitsmsg)
238 f1e8706d 2004-12-27 devnull _exits(threadexitsmsg);
239 f1e8706d 2004-12-27 devnull }
240 f1e8706d 2004-12-27 devnull
241 f1e8706d 2004-12-27 devnull void
242 f1e8706d 2004-12-27 devnull threadexitsall(char *msg)
243 f1e8706d 2004-12-27 devnull {
244 f1e8706d 2004-12-27 devnull static int pid[1024];
245 f1e8706d 2004-12-27 devnull int i, npid, mypid;
246 f1e8706d 2004-12-27 devnull Proc *p;
247 f1e8706d 2004-12-27 devnull
248 f1e8706d 2004-12-27 devnull if(msg == nil)
249 f1e8706d 2004-12-27 devnull msg = "";
250 f1e8706d 2004-12-27 devnull mypid = getpid();
251 f1e8706d 2004-12-27 devnull lock(&_threadprocslock);
252 f1e8706d 2004-12-27 devnull threadexitsmsg = msg;
253 f1e8706d 2004-12-27 devnull npid = 0;
254 f1e8706d 2004-12-27 devnull for(p=_threadprocs; p; p=p->next)
255 f1e8706d 2004-12-27 devnull if(p->osprocid != mypid && p->osprocid >= 1)
256 f1e8706d 2004-12-27 devnull pid[npid++] = p->osprocid;
257 f1e8706d 2004-12-27 devnull for(i=0; i<npid; i++)
258 f1e8706d 2004-12-27 devnull kill(pid[i], SIGUSR2);
259 f1e8706d 2004-12-27 devnull unlock(&_threadprocslock);
260 f1e8706d 2004-12-27 devnull exits(msg);
261 f1e8706d 2004-12-27 devnull }
262 f1e8706d 2004-12-27 devnull
263 f1e8706d 2004-12-27 devnull /*
264 f1e8706d 2004-12-27 devnull * per-process data, indexed by pid
265 f1e8706d 2004-12-27 devnull *
266 f1e8706d 2004-12-27 devnull * could use modify_ldt and a segment register
267 f1e8706d 2004-12-27 devnull * to avoid the many calls to getpid(), but i don't
268 f1e8706d 2004-12-27 devnull * care -- this is compatibility code. linux 2.6 with
269 f1e8706d 2004-12-27 devnull * nptl is a good enough pthreads to avoid this whole file.
270 f1e8706d 2004-12-27 devnull */
271 f1e8706d 2004-12-27 devnull typedef struct Perproc Perproc;
272 f1e8706d 2004-12-27 devnull struct Perproc
273 f1e8706d 2004-12-27 devnull {
274 f1e8706d 2004-12-27 devnull int pid;
275 f1e8706d 2004-12-27 devnull Proc *proc;
276 f1e8706d 2004-12-27 devnull };
277 f1e8706d 2004-12-27 devnull
278 f1e8706d 2004-12-27 devnull static Lock perlock;
279 f1e8706d 2004-12-27 devnull static Perproc perproc[1024];
280 f1e8706d 2004-12-27 devnull #define P ((Proc*)-1)
281 f1e8706d 2004-12-27 devnull
282 f1e8706d 2004-12-27 devnull static Perproc*
283 f1e8706d 2004-12-27 devnull myperproc(void)
284 f1e8706d 2004-12-27 devnull {
285 f1e8706d 2004-12-27 devnull int i, pid, h;
286 f1e8706d 2004-12-27 devnull Perproc *p;
287 f1e8706d 2004-12-27 devnull
288 f1e8706d 2004-12-27 devnull pid = getpid();
289 f1e8706d 2004-12-27 devnull h = pid%nelem(perproc);
290 f1e8706d 2004-12-27 devnull for(i=0; i<nelem(perproc); i++){
291 f1e8706d 2004-12-27 devnull p = &perproc[(i+h)%nelem(perproc)];
292 f1e8706d 2004-12-27 devnull if(p->pid == pid)
293 f1e8706d 2004-12-27 devnull return p;
294 f1e8706d 2004-12-27 devnull if(p->pid == 0){
295 f1e8706d 2004-12-27 devnull print("found 0 at %d (h=%d)\n", (i+h)%nelem(perproc), h);
296 f1e8706d 2004-12-27 devnull break;
297 f1e8706d 2004-12-27 devnull }
298 f1e8706d 2004-12-27 devnull }
299 f1e8706d 2004-12-27 devnull fprint(2, "myperproc %d: cannot find self\n", pid);
300 f1e8706d 2004-12-27 devnull abort();
301 f1e8706d 2004-12-27 devnull return nil;
302 f1e8706d 2004-12-27 devnull }
303 f1e8706d 2004-12-27 devnull
304 f1e8706d 2004-12-27 devnull static Perproc*
305 f1e8706d 2004-12-27 devnull newperproc(void)
306 f1e8706d 2004-12-27 devnull {
307 f1e8706d 2004-12-27 devnull int i, pid, h;
308 f1e8706d 2004-12-27 devnull Perproc *p;
309 f1e8706d 2004-12-27 devnull
310 f1e8706d 2004-12-27 devnull lock(&perlock);
311 f1e8706d 2004-12-27 devnull pid = getpid();
312 f1e8706d 2004-12-27 devnull h = pid%nelem(perproc);
313 f1e8706d 2004-12-27 devnull for(i=0; i<nelem(perproc); i++){
314 f1e8706d 2004-12-27 devnull p = &perproc[(i+h)%nelem(perproc)];
315 f1e8706d 2004-12-27 devnull if(p->pid == pid || p->pid == -1 || p->pid == 0){
316 f1e8706d 2004-12-27 devnull p->pid = pid;
317 f1e8706d 2004-12-27 devnull unlock(&perlock);
318 f1e8706d 2004-12-27 devnull return p;
319 f1e8706d 2004-12-27 devnull }
320 f1e8706d 2004-12-27 devnull }
321 f1e8706d 2004-12-27 devnull fprint(2, "newperproc %d: out of procs\n", pid);
322 f1e8706d 2004-12-27 devnull abort();
323 f1e8706d 2004-12-27 devnull return nil;
324 f1e8706d 2004-12-27 devnull }
325 f1e8706d 2004-12-27 devnull
326 f1e8706d 2004-12-27 devnull Proc*
327 f1e8706d 2004-12-27 devnull _threadproc(void)
328 f1e8706d 2004-12-27 devnull {
329 f1e8706d 2004-12-27 devnull return myperproc()->proc;
330 f1e8706d 2004-12-27 devnull }
331 f1e8706d 2004-12-27 devnull
332 f1e8706d 2004-12-27 devnull void
333 f1e8706d 2004-12-27 devnull _threadsetproc(Proc *p)
334 f1e8706d 2004-12-27 devnull {
335 f1e8706d 2004-12-27 devnull Perproc *pp;
336 f1e8706d 2004-12-27 devnull
337 2c87dda8 2004-12-28 devnull if(p)
338 2c87dda8 2004-12-28 devnull p->osprocid = getpid();
339 f1e8706d 2004-12-27 devnull pp = newperproc();
340 f1e8706d 2004-12-27 devnull pp->proc = p;
341 f1e8706d 2004-12-27 devnull if(p == nil)
342 f1e8706d 2004-12-27 devnull pp->pid = -1;
343 f1e8706d 2004-12-27 devnull }
344 f1e8706d 2004-12-27 devnull
345 f1e8706d 2004-12-27 devnull void
346 f1e8706d 2004-12-27 devnull _pthreadinit(void)
347 f1e8706d 2004-12-27 devnull {
348 f1e8706d 2004-12-27 devnull signal(SIGUSR2, sigusr2handler);
349 f1e8706d 2004-12-27 devnull }
350 f1e8706d 2004-12-27 devnull
351 1d2533d0 2004-12-28 devnull void
352 1d2533d0 2004-12-28 devnull _threadpexit(void)
353 1d2533d0 2004-12-28 devnull {
354 1d2533d0 2004-12-28 devnull _exit(0);
355 1d2533d0 2004-12-28 devnull }