1 f1e8706d 2004-12-27 devnull #include "threadimpl.h"
3 a620761c 2005-01-06 devnull static int
4 a620761c 2005-01-06 devnull timefmt(Fmt *fmt)
6 a620761c 2005-01-06 devnull static char *mon[] = { "Jan", "Feb", "Mar", "Apr", "May", "Jun",
7 a620761c 2005-01-06 devnull "Jul", "Aug", "Sep", "Oct", "Nov", "Dec" };
10 a620761c 2005-01-06 devnull ns = nsec();
11 a620761c 2005-01-06 devnull tm = *localtime(time(0));
12 a620761c 2005-01-06 devnull return fmtprint(fmt, "%s %2d %02d:%02d:%02d.%03d",
13 a620761c 2005-01-06 devnull mon[tm.mon], tm.mday, tm.hour, tm.min, tm.sec,
14 a620761c 2005-01-06 devnull (int)(ns%1000000000)/1000000);
18 f1e8706d 2004-12-27 devnull * spin locks
20 f1e8706d 2004-12-27 devnull extern int _tas(int*);
23 f1e8706d 2004-12-27 devnull _threadunlock(Lock *l, ulong pc)
25 f1e8706d 2004-12-27 devnull USED(pc);
27 f1e8706d 2004-12-27 devnull l->held = 0;
31 f1e8706d 2004-12-27 devnull _threadlock(Lock *l, int block, ulong pc)
34 a620761c 2005-01-06 devnull static int first=1;
35 a0a331aa 2005-01-06 devnull if(first) {first=0; fmtinstall('\001', timefmt);}
37 f1e8706d 2004-12-27 devnull USED(pc);
39 f1e8706d 2004-12-27 devnull /* once fast */
40 f1e8706d 2004-12-27 devnull if(!_tas(&l->held))
41 f1e8706d 2004-12-27 devnull return 1;
42 f1e8706d 2004-12-27 devnull if(!block)
43 f1e8706d 2004-12-27 devnull return 0;
45 f1e8706d 2004-12-27 devnull /* a thousand times pretty fast */
46 815552b9 2005-01-06 devnull for(i=0; i<1000; i++){
47 f1e8706d 2004-12-27 devnull if(!_tas(&l->held))
48 f1e8706d 2004-12-27 devnull return 1;
49 f1e8706d 2004-12-27 devnull sched_yield();
51 a0a331aa 2005-01-06 devnull /* now increasingly slow */
52 a620761c 2005-01-06 devnull for(i=0; i<10; i++){
53 a620761c 2005-01-06 devnull if(!_tas(&l->held))
54 a620761c 2005-01-06 devnull return 1;
55 a620761c 2005-01-06 devnull usleep(1);
57 a0a331aa 2005-01-06 devnull fprint(2, "%\001 %s: lock loop1 %p from %lux\n", argv0, l, pc);
58 a620761c 2005-01-06 devnull for(i=0; i<10; i++){
59 a620761c 2005-01-06 devnull if(!_tas(&l->held))
60 a620761c 2005-01-06 devnull return 1;
61 a620761c 2005-01-06 devnull usleep(10);
63 a0a331aa 2005-01-06 devnull fprint(2, "%\001 %s: lock loop2 %p from %lux\n", argv0, l, pc);
64 a620761c 2005-01-06 devnull for(i=0; i<10; i++){
65 a620761c 2005-01-06 devnull if(!_tas(&l->held))
66 a620761c 2005-01-06 devnull return 1;
67 a620761c 2005-01-06 devnull usleep(100);
69 a0a331aa 2005-01-06 devnull fprint(2, "%\001 %s: lock loop3 %p from %lux\n", argv0, l, pc);
70 a620761c 2005-01-06 devnull for(i=0; i<10; i++){
71 a620761c 2005-01-06 devnull if(!_tas(&l->held))
72 a620761c 2005-01-06 devnull return 1;
73 a620761c 2005-01-06 devnull usleep(1000);
75 a0a331aa 2005-01-06 devnull fprint(2, "%\001 %s: lock loop4 %p from %lux\n", argv0, l, pc);
76 a620761c 2005-01-06 devnull for(i=0; i<10; i++){
77 a620761c 2005-01-06 devnull if(!_tas(&l->held))
78 a620761c 2005-01-06 devnull return 1;
79 a620761c 2005-01-06 devnull usleep(10*1000);
81 a0a331aa 2005-01-06 devnull fprint(2, "%\001 %s: lock loop5 %p from %lux\n", argv0, l, pc);
82 f1e8706d 2004-12-27 devnull for(i=0; i<1000; i++){
83 f1e8706d 2004-12-27 devnull if(!_tas(&l->held))
84 f1e8706d 2004-12-27 devnull return 1;
85 f1e8706d 2004-12-27 devnull usleep(100*1000);
87 a0a331aa 2005-01-06 devnull fprint(2, "%\001 %s: lock loop6 %p from %lux\n", argv0, l, pc);
88 f1e8706d 2004-12-27 devnull /* take your time */
89 f1e8706d 2004-12-27 devnull while(_tas(&l->held))
90 f1e8706d 2004-12-27 devnull usleep(1000*1000);
91 f1e8706d 2004-12-27 devnull return 1;
95 f1e8706d 2004-12-27 devnull * sleep and wakeup
97 f1e8706d 2004-12-27 devnull static void
98 f1e8706d 2004-12-27 devnull ign(int x)
100 f1e8706d 2004-12-27 devnull USED(x);
103 f1e8706d 2004-12-27 devnull static void /*__attribute__((constructor))*/
104 f1e8706d 2004-12-27 devnull ignusr1(int restart)
106 f1e8706d 2004-12-27 devnull struct sigaction sa;
108 f1e8706d 2004-12-27 devnull memset(&sa, 0, sizeof sa);
109 f1e8706d 2004-12-27 devnull sa.sa_handler = ign;
110 f1e8706d 2004-12-27 devnull sigemptyset(&sa.sa_mask);
111 f1e8706d 2004-12-27 devnull sigaddset(&sa.sa_mask, SIGUSR1);
112 f1e8706d 2004-12-27 devnull if(restart)
113 f1e8706d 2004-12-27 devnull sa.sa_flags = SA_RESTART;
114 f1e8706d 2004-12-27 devnull sigaction(SIGUSR1, &sa, nil);
118 f1e8706d 2004-12-27 devnull _procsleep(_Procrendez *r)
120 f1e8706d 2004-12-27 devnull sigset_t mask;
123 f1e8706d 2004-12-27 devnull * Go to sleep.
125 f1e8706d 2004-12-27 devnull * Block USR1, set the handler to interrupt system calls,
126 f1e8706d 2004-12-27 devnull * unlock the vouslock so our waker can wake us,
127 f1e8706d 2004-12-27 devnull * and then suspend.
130 f1e8706d 2004-12-27 devnull r->asleep = 1;
131 f1e8706d 2004-12-27 devnull r->pid = getpid();
133 f1e8706d 2004-12-27 devnull sigprocmask(SIG_SETMASK, nil, &mask);
134 f1e8706d 2004-12-27 devnull sigaddset(&mask, SIGUSR1);
135 f1e8706d 2004-12-27 devnull sigprocmask(SIG_SETMASK, &mask, nil);
136 f1e8706d 2004-12-27 devnull ignusr1(0);
137 f1e8706d 2004-12-27 devnull unlock(r->l);
138 f1e8706d 2004-12-27 devnull sigdelset(&mask, SIGUSR1);
139 f1e8706d 2004-12-27 devnull sigsuspend(&mask);
142 f1e8706d 2004-12-27 devnull * We're awake. Make USR1 not interrupt system calls.
144 f1e8706d 2004-12-27 devnull lock(r->l);
145 955a2ca7 2004-12-27 devnull ignusr1(1);
146 955a2ca7 2004-12-27 devnull if(r->asleep && r->pid == getpid()){
147 955a2ca7 2004-12-27 devnull /* Didn't really wake up - signal from something else */
148 955a2ca7 2004-12-27 devnull goto again;
153 a0a331aa 2005-01-06 devnull _procwakeupandunlock(_Procrendez *r)
155 a0a331aa 2005-01-06 devnull int pid;
157 a0a331aa 2005-01-06 devnull pid = 0;
158 f1e8706d 2004-12-27 devnull if(r->asleep){
159 f1e8706d 2004-12-27 devnull r->asleep = 0;
160 f1e8706d 2004-12-27 devnull assert(r->pid >= 1);
161 a0a331aa 2005-01-06 devnull pid = r->pid;
163 a0a331aa 2005-01-06 devnull assert(r->l);
164 a0a331aa 2005-01-06 devnull unlock(r->l);
166 a0a331aa 2005-01-06 devnull kill(pid, SIGUSR1);
170 f1e8706d 2004-12-27 devnull * process creation and exit
172 f1e8706d 2004-12-27 devnull typedef struct Stackfree Stackfree;
173 f1e8706d 2004-12-27 devnull struct Stackfree
175 f1e8706d 2004-12-27 devnull Stackfree *next;
176 f1e8706d 2004-12-27 devnull int pid;
177 a620761c 2005-01-06 devnull int pid1;
179 f1e8706d 2004-12-27 devnull static Lock stacklock;
180 f1e8706d 2004-12-27 devnull static Stackfree *stackfree;
182 f1e8706d 2004-12-27 devnull static void
183 a620761c 2005-01-06 devnull delayfreestack(uchar *stk, int pid, int pid1)
185 f1e8706d 2004-12-27 devnull Stackfree *sf;
187 f1e8706d 2004-12-27 devnull sf = (Stackfree*)stk;
188 a620761c 2005-01-06 devnull sf->pid = pid;
189 a620761c 2005-01-06 devnull sf->pid1 = pid1;
190 f1e8706d 2004-12-27 devnull lock(&stacklock);
191 f1e8706d 2004-12-27 devnull sf->next = stackfree;
192 f1e8706d 2004-12-27 devnull stackfree = sf;
193 f1e8706d 2004-12-27 devnull unlock(&stacklock);
196 f1e8706d 2004-12-27 devnull static void
197 f1e8706d 2004-12-27 devnull dofreestacks(void)
199 f1e8706d 2004-12-27 devnull Stackfree *sf, *last, *next;
201 f1e8706d 2004-12-27 devnull if(stackfree==nil || !canlock(&stacklock))
204 f1e8706d 2004-12-27 devnull for(last=nil,sf=stackfree; sf; last=sf,sf=next){
205 f1e8706d 2004-12-27 devnull next = sf->next;
206 a620761c 2005-01-06 devnull if(sf->pid >= 1 && kill(sf->pid, 0) < 0 && errno == ESRCH)
207 a620761c 2005-01-06 devnull if(sf->pid1 >= 1 && kill(sf->pid1, 0) < 0 && errno == ESRCH){
208 f1e8706d 2004-12-27 devnull free(sf);
209 2c87dda8 2004-12-28 devnull if(last)
210 2c87dda8 2004-12-28 devnull last->next = next;
212 2c87dda8 2004-12-28 devnull stackfree = next;
213 f1e8706d 2004-12-27 devnull sf = last;
216 2c87dda8 2004-12-28 devnull unlock(&stacklock);
219 f1e8706d 2004-12-27 devnull static int
220 f1e8706d 2004-12-27 devnull startprocfn(void *v)
222 f1e8706d 2004-12-27 devnull void **a;
223 f1e8706d 2004-12-27 devnull uchar *stk;
224 f1e8706d 2004-12-27 devnull void (*fn)(void*);
225 f1e8706d 2004-12-27 devnull Proc *p;
226 a620761c 2005-01-06 devnull int pid0, pid1;
228 f1e8706d 2004-12-27 devnull a = (void**)v;
229 f1e8706d 2004-12-27 devnull fn = a[0];
230 f1e8706d 2004-12-27 devnull p = a[1];
231 f1e8706d 2004-12-27 devnull stk = a[2];
232 a620761c 2005-01-06 devnull pid0 = (int)a[4];
233 a620761c 2005-01-06 devnull pid1 = getpid();
234 f1e8706d 2004-12-27 devnull free(a);
235 a620761c 2005-01-06 devnull p->osprocid = pid1;
237 f1e8706d 2004-12-27 devnull (*fn)(p);
239 a620761c 2005-01-06 devnull delayfreestack(stk, pid0, pid1);
240 f1e8706d 2004-12-27 devnull _exit(0);
241 f1e8706d 2004-12-27 devnull return 0;
245 f1e8706d 2004-12-27 devnull * indirect through here so that parent need not wait for child zombie
247 f1e8706d 2004-12-27 devnull * slight race - if child exits and then another process starts before we
248 f1e8706d 2004-12-27 devnull * manage to exit, we'll be running on a freed stack.
250 f1e8706d 2004-12-27 devnull static int
251 f1e8706d 2004-12-27 devnull trampnowait(void *v)
253 f1e8706d 2004-12-27 devnull void **a;
254 a620761c 2005-01-06 devnull int *kidpid;
256 f1e8706d 2004-12-27 devnull a = (void*)v;
257 a620761c 2005-01-06 devnull kidpid = a[3];
258 a620761c 2005-01-06 devnull a[4] = (void*)getpid();
259 a620761c 2005-01-06 devnull *kidpid = clone(startprocfn, a[2]+65536-512, CLONE_VM|CLONE_FILES, a);
260 f1e8706d 2004-12-27 devnull _exit(0);
261 f1e8706d 2004-12-27 devnull return 0;
265 f1e8706d 2004-12-27 devnull _procstart(Proc *p, void (*fn)(Proc*))
267 f1e8706d 2004-12-27 devnull void **a;
268 f1e8706d 2004-12-27 devnull uchar *stk;
269 f1e8706d 2004-12-27 devnull int pid, kidpid, status;
271 f1e8706d 2004-12-27 devnull dofreestacks();
272 a620761c 2005-01-06 devnull a = malloc(5*sizeof a[0]);
273 f1e8706d 2004-12-27 devnull if(a == nil)
274 f1e8706d 2004-12-27 devnull sysfatal("_procstart malloc: %r");
275 f1e8706d 2004-12-27 devnull stk = malloc(65536);
276 f1e8706d 2004-12-27 devnull if(stk == nil)
277 f1e8706d 2004-12-27 devnull sysfatal("_procstart malloc stack: %r");
279 f1e8706d 2004-12-27 devnull a[0] = fn;
280 f1e8706d 2004-12-27 devnull a[1] = p;
281 f1e8706d 2004-12-27 devnull a[2] = stk;
282 f1e8706d 2004-12-27 devnull a[3] = &kidpid;
283 f1e8706d 2004-12-27 devnull kidpid = -1;
285 f1e8706d 2004-12-27 devnull pid = clone(trampnowait, stk+65536-16, CLONE_VM|CLONE_FILES, a);
286 f1e8706d 2004-12-27 devnull if(pid > 0)
287 f1e8706d 2004-12-27 devnull if(wait4(pid, &status, __WALL, 0) < 0)
288 f1e8706d 2004-12-27 devnull fprint(2, "ffork wait4: %r\n");
289 f1e8706d 2004-12-27 devnull if(pid < 0 || kidpid < 0){
290 f1e8706d 2004-12-27 devnull fprint(2, "_procstart clone: %r\n");
291 f1e8706d 2004-12-27 devnull abort();
295 f1e8706d 2004-12-27 devnull static char *threadexitsmsg;
297 f1e8706d 2004-12-27 devnull sigusr2handler(int s)
299 2c87dda8 2004-12-28 devnull /* fprint(2, "%d usr2 %d\n", time(0), getpid()); */
300 f1e8706d 2004-12-27 devnull if(threadexitsmsg)
301 f1e8706d 2004-12-27 devnull _exits(threadexitsmsg);
305 f1e8706d 2004-12-27 devnull threadexitsall(char *msg)
307 f1e8706d 2004-12-27 devnull static int pid[1024];
308 f1e8706d 2004-12-27 devnull int i, npid, mypid;
309 f1e8706d 2004-12-27 devnull Proc *p;
311 f1e8706d 2004-12-27 devnull if(msg == nil)
312 f1e8706d 2004-12-27 devnull msg = "";
313 f1e8706d 2004-12-27 devnull mypid = getpid();
314 f1e8706d 2004-12-27 devnull lock(&_threadprocslock);
315 f1e8706d 2004-12-27 devnull threadexitsmsg = msg;
316 f1e8706d 2004-12-27 devnull npid = 0;
317 f1e8706d 2004-12-27 devnull for(p=_threadprocs; p; p=p->next)
318 f1e8706d 2004-12-27 devnull if(p->osprocid != mypid && p->osprocid >= 1)
319 f1e8706d 2004-12-27 devnull pid[npid++] = p->osprocid;
320 f1e8706d 2004-12-27 devnull for(i=0; i<npid; i++)
321 f1e8706d 2004-12-27 devnull kill(pid[i], SIGUSR2);
322 f1e8706d 2004-12-27 devnull unlock(&_threadprocslock);
323 f1e8706d 2004-12-27 devnull exits(msg);
327 f1e8706d 2004-12-27 devnull * per-process data, indexed by pid
329 f1e8706d 2004-12-27 devnull * could use modify_ldt and a segment register
330 f1e8706d 2004-12-27 devnull * to avoid the many calls to getpid(), but i don't
331 f1e8706d 2004-12-27 devnull * care -- this is compatibility code. linux 2.6 with
332 f1e8706d 2004-12-27 devnull * nptl is a good enough pthreads to avoid this whole file.
334 f1e8706d 2004-12-27 devnull typedef struct Perproc Perproc;
335 f1e8706d 2004-12-27 devnull struct Perproc
337 f1e8706d 2004-12-27 devnull int pid;
338 f1e8706d 2004-12-27 devnull Proc *proc;
341 f1e8706d 2004-12-27 devnull static Lock perlock;
342 f1e8706d 2004-12-27 devnull static Perproc perproc[1024];
343 f1e8706d 2004-12-27 devnull #define P ((Proc*)-1)
345 f1e8706d 2004-12-27 devnull static Perproc*
346 f1e8706d 2004-12-27 devnull myperproc(void)
348 f1e8706d 2004-12-27 devnull int i, pid, h;
349 f1e8706d 2004-12-27 devnull Perproc *p;
351 f1e8706d 2004-12-27 devnull pid = getpid();
352 f1e8706d 2004-12-27 devnull h = pid%nelem(perproc);
353 f1e8706d 2004-12-27 devnull for(i=0; i<nelem(perproc); i++){
354 f1e8706d 2004-12-27 devnull p = &perproc[(i+h)%nelem(perproc)];
355 f1e8706d 2004-12-27 devnull if(p->pid == pid)
356 f1e8706d 2004-12-27 devnull return p;
357 f1e8706d 2004-12-27 devnull if(p->pid == 0){
358 f1e8706d 2004-12-27 devnull print("found 0 at %d (h=%d)\n", (i+h)%nelem(perproc), h);
362 f1e8706d 2004-12-27 devnull fprint(2, "myperproc %d: cannot find self\n", pid);
363 f1e8706d 2004-12-27 devnull abort();
364 f1e8706d 2004-12-27 devnull return nil;
367 f1e8706d 2004-12-27 devnull static Perproc*
368 f1e8706d 2004-12-27 devnull newperproc(void)
370 f1e8706d 2004-12-27 devnull int i, pid, h;
371 f1e8706d 2004-12-27 devnull Perproc *p;
373 f1e8706d 2004-12-27 devnull lock(&perlock);
374 f1e8706d 2004-12-27 devnull pid = getpid();
375 f1e8706d 2004-12-27 devnull h = pid%nelem(perproc);
376 f1e8706d 2004-12-27 devnull for(i=0; i<nelem(perproc); i++){
377 f1e8706d 2004-12-27 devnull p = &perproc[(i+h)%nelem(perproc)];
378 f1e8706d 2004-12-27 devnull if(p->pid == pid || p->pid == -1 || p->pid == 0){
379 f1e8706d 2004-12-27 devnull p->pid = pid;
380 f1e8706d 2004-12-27 devnull unlock(&perlock);
381 f1e8706d 2004-12-27 devnull return p;
384 f1e8706d 2004-12-27 devnull fprint(2, "newperproc %d: out of procs\n", pid);
385 f1e8706d 2004-12-27 devnull abort();
386 f1e8706d 2004-12-27 devnull return nil;
390 f1e8706d 2004-12-27 devnull _threadproc(void)
392 f1e8706d 2004-12-27 devnull return myperproc()->proc;
396 f1e8706d 2004-12-27 devnull _threadsetproc(Proc *p)
398 f1e8706d 2004-12-27 devnull Perproc *pp;
401 2c87dda8 2004-12-28 devnull p->osprocid = getpid();
402 f1e8706d 2004-12-27 devnull pp = newperproc();
403 f1e8706d 2004-12-27 devnull pp->proc = p;
404 f1e8706d 2004-12-27 devnull if(p == nil)
405 f1e8706d 2004-12-27 devnull pp->pid = -1;
409 f1e8706d 2004-12-27 devnull _pthreadinit(void)
411 f1e8706d 2004-12-27 devnull signal(SIGUSR2, sigusr2handler);
415 1d2533d0 2004-12-28 devnull _threadpexit(void)
417 1d2533d0 2004-12-28 devnull _exit(0);