1 31cf0ab1 2004-12-28 devnull #include "threadimpl.h"
3 4753f069 2005-01-17 devnull #undef exits
4 4753f069 2005-01-17 devnull #undef _exits
6 31cf0ab1 2004-12-28 devnull extern int __isthreaded;
9 31cf0ab1 2004-12-28 devnull * spin locks
11 31cf0ab1 2004-12-28 devnull extern int _tas(int*);
14 31cf0ab1 2004-12-28 devnull _threadunlock(Lock *l, ulong pc)
16 31cf0ab1 2004-12-28 devnull USED(pc);
18 31cf0ab1 2004-12-28 devnull l->held = 0;
22 31cf0ab1 2004-12-28 devnull _threadlock(Lock *l, int block, ulong pc)
26 31cf0ab1 2004-12-28 devnull USED(pc);
28 31cf0ab1 2004-12-28 devnull /* once fast */
29 31cf0ab1 2004-12-28 devnull if(!_tas(&l->held))
30 31cf0ab1 2004-12-28 devnull return 1;
31 31cf0ab1 2004-12-28 devnull if(!block)
32 31cf0ab1 2004-12-28 devnull return 0;
34 31cf0ab1 2004-12-28 devnull /* a thousand times pretty fast */
35 31cf0ab1 2004-12-28 devnull for(i=0; i<1000; i++){
36 31cf0ab1 2004-12-28 devnull if(!_tas(&l->held))
37 31cf0ab1 2004-12-28 devnull return 1;
38 31cf0ab1 2004-12-28 devnull sched_yield();
40 31cf0ab1 2004-12-28 devnull /* now nice and slow */
41 31cf0ab1 2004-12-28 devnull for(i=0; i<1000; i++){
42 31cf0ab1 2004-12-28 devnull if(!_tas(&l->held))
43 31cf0ab1 2004-12-28 devnull return 1;
44 31cf0ab1 2004-12-28 devnull usleep(100*1000);
46 31cf0ab1 2004-12-28 devnull /* take your time */
47 31cf0ab1 2004-12-28 devnull while(_tas(&l->held))
48 31cf0ab1 2004-12-28 devnull usleep(1000*1000);
49 31cf0ab1 2004-12-28 devnull return 1;
53 31cf0ab1 2004-12-28 devnull * For FreeBSD libc.
56 31cf0ab1 2004-12-28 devnull typedef struct {
57 31cf0ab1 2004-12-28 devnull volatile long access_lock;
58 31cf0ab1 2004-12-28 devnull volatile long lock_owner;
59 31cf0ab1 2004-12-28 devnull volatile char *fname;
60 31cf0ab1 2004-12-28 devnull volatile int lineno;
61 31cf0ab1 2004-12-28 devnull } spinlock_t;
64 31cf0ab1 2004-12-28 devnull _spinlock(spinlock_t *lk)
66 31cf0ab1 2004-12-28 devnull lock((Lock*)&lk->access_lock);
70 31cf0ab1 2004-12-28 devnull * sleep and wakeup
72 31cf0ab1 2004-12-28 devnull static void
73 31cf0ab1 2004-12-28 devnull ign(int x)
78 31cf0ab1 2004-12-28 devnull static void /*__attribute__((constructor))*/
79 31cf0ab1 2004-12-28 devnull ignusr1(int restart)
81 31cf0ab1 2004-12-28 devnull struct sigaction sa;
83 31cf0ab1 2004-12-28 devnull memset(&sa, 0, sizeof sa);
84 31cf0ab1 2004-12-28 devnull sa.sa_handler = ign;
85 31cf0ab1 2004-12-28 devnull sigemptyset(&sa.sa_mask);
86 31cf0ab1 2004-12-28 devnull sigaddset(&sa.sa_mask, SIGUSR1);
87 31cf0ab1 2004-12-28 devnull if(restart)
88 31cf0ab1 2004-12-28 devnull sa.sa_flags = SA_RESTART;
89 31cf0ab1 2004-12-28 devnull sigaction(SIGUSR1, &sa, nil);
93 31cf0ab1 2004-12-28 devnull _procsleep(_Procrendez *r)
95 31cf0ab1 2004-12-28 devnull sigset_t mask;
98 31cf0ab1 2004-12-28 devnull * Go to sleep.
100 31cf0ab1 2004-12-28 devnull * Block USR1, set the handler to interrupt system calls,
101 31cf0ab1 2004-12-28 devnull * unlock the vouslock so our waker can wake us,
102 31cf0ab1 2004-12-28 devnull * and then suspend.
105 31cf0ab1 2004-12-28 devnull r->asleep = 1;
106 31cf0ab1 2004-12-28 devnull r->pid = getpid();
108 31cf0ab1 2004-12-28 devnull sigprocmask(SIG_SETMASK, nil, &mask);
109 31cf0ab1 2004-12-28 devnull sigaddset(&mask, SIGUSR1);
110 31cf0ab1 2004-12-28 devnull sigprocmask(SIG_SETMASK, &mask, nil);
111 31cf0ab1 2004-12-28 devnull ignusr1(0);
112 31cf0ab1 2004-12-28 devnull unlock(r->l);
113 31cf0ab1 2004-12-28 devnull sigdelset(&mask, SIGUSR1);
114 31cf0ab1 2004-12-28 devnull sigsuspend(&mask);
117 31cf0ab1 2004-12-28 devnull * We're awake. Make USR1 not interrupt system calls.
119 31cf0ab1 2004-12-28 devnull lock(r->l);
120 31cf0ab1 2004-12-28 devnull ignusr1(1);
121 31cf0ab1 2004-12-28 devnull if(r->asleep && r->pid == getpid()){
122 31cf0ab1 2004-12-28 devnull /* Didn't really wake up - signal from something else */
123 31cf0ab1 2004-12-28 devnull goto again;
128 31cf0ab1 2004-12-28 devnull _procwakeup(_Procrendez *r)
130 31cf0ab1 2004-12-28 devnull if(r->asleep){
131 31cf0ab1 2004-12-28 devnull r->asleep = 0;
132 31cf0ab1 2004-12-28 devnull assert(r->pid >= 1);
133 31cf0ab1 2004-12-28 devnull kill(r->pid, SIGUSR1);
138 4a8edd8d 2005-01-14 devnull _procwakeupandunlock(_Procrendez *r)
140 4a8edd8d 2005-01-14 devnull _procwakeup(r);
141 4a8edd8d 2005-01-14 devnull unlock(r->l);
146 31cf0ab1 2004-12-28 devnull * process creation and exit
148 31cf0ab1 2004-12-28 devnull typedef struct Stackfree Stackfree;
149 31cf0ab1 2004-12-28 devnull struct Stackfree
151 31cf0ab1 2004-12-28 devnull Stackfree *next;
152 31cf0ab1 2004-12-28 devnull int pid;
154 31cf0ab1 2004-12-28 devnull static Lock stacklock;
155 31cf0ab1 2004-12-28 devnull static Stackfree *stackfree;
157 31cf0ab1 2004-12-28 devnull static void
158 31cf0ab1 2004-12-28 devnull delayfreestack(uchar *stk)
160 31cf0ab1 2004-12-28 devnull Stackfree *sf;
162 31cf0ab1 2004-12-28 devnull sf = (Stackfree*)stk;
163 31cf0ab1 2004-12-28 devnull sf->pid = getpid();
164 31cf0ab1 2004-12-28 devnull lock(&stacklock);
165 31cf0ab1 2004-12-28 devnull sf->next = stackfree;
166 31cf0ab1 2004-12-28 devnull stackfree = sf;
167 31cf0ab1 2004-12-28 devnull unlock(&stacklock);
170 31cf0ab1 2004-12-28 devnull static void
171 31cf0ab1 2004-12-28 devnull dofreestacks(void)
173 31cf0ab1 2004-12-28 devnull Stackfree *sf, *last, *next;
175 31cf0ab1 2004-12-28 devnull if(stackfree==nil || !canlock(&stacklock))
178 31cf0ab1 2004-12-28 devnull for(last=nil,sf=stackfree; sf; last=sf,sf=next){
179 31cf0ab1 2004-12-28 devnull next = sf->next;
180 31cf0ab1 2004-12-28 devnull if(sf->pid >= 1 && kill(sf->pid, 0) < 0 && errno == ESRCH){
181 31cf0ab1 2004-12-28 devnull free(sf);
182 31cf0ab1 2004-12-28 devnull if(last)
183 31cf0ab1 2004-12-28 devnull last->next = next;
185 31cf0ab1 2004-12-28 devnull stackfree = next;
186 31cf0ab1 2004-12-28 devnull sf = last;
189 31cf0ab1 2004-12-28 devnull unlock(&stacklock);
192 31cf0ab1 2004-12-28 devnull static int
193 31cf0ab1 2004-12-28 devnull startprocfn(void *v)
195 31cf0ab1 2004-12-28 devnull void **a;
196 31cf0ab1 2004-12-28 devnull uchar *stk;
197 31cf0ab1 2004-12-28 devnull void (*fn)(void*);
198 31cf0ab1 2004-12-28 devnull Proc *p;
200 31cf0ab1 2004-12-28 devnull a = (void**)v;
201 31cf0ab1 2004-12-28 devnull fn = a[0];
202 31cf0ab1 2004-12-28 devnull p = a[1];
203 31cf0ab1 2004-12-28 devnull stk = a[2];
204 31cf0ab1 2004-12-28 devnull free(a);
205 31cf0ab1 2004-12-28 devnull p->osprocid = getpid();
207 31cf0ab1 2004-12-28 devnull (*fn)(p);
209 31cf0ab1 2004-12-28 devnull delayfreestack(stk);
210 31cf0ab1 2004-12-28 devnull _exit(0);
211 31cf0ab1 2004-12-28 devnull return 0;
215 31cf0ab1 2004-12-28 devnull _procstart(Proc *p, void (*fn)(Proc*))
217 31cf0ab1 2004-12-28 devnull void **a;
218 31cf0ab1 2004-12-28 devnull uchar *stk;
219 31cf0ab1 2004-12-28 devnull int pid;
221 31cf0ab1 2004-12-28 devnull dofreestacks();
222 31cf0ab1 2004-12-28 devnull a = malloc(3*sizeof a[0]);
223 31cf0ab1 2004-12-28 devnull if(a == nil)
224 31cf0ab1 2004-12-28 devnull sysfatal("_procstart malloc: %r");
225 31cf0ab1 2004-12-28 devnull stk = malloc(65536);
226 31cf0ab1 2004-12-28 devnull if(stk == nil)
227 31cf0ab1 2004-12-28 devnull sysfatal("_procstart malloc stack: %r");
229 31cf0ab1 2004-12-28 devnull a[0] = fn;
230 31cf0ab1 2004-12-28 devnull a[1] = p;
231 31cf0ab1 2004-12-28 devnull a[2] = stk;
233 31cf0ab1 2004-12-28 devnull pid = rfork_thread(RFPROC|RFMEM|RFNOWAIT, stk+65536-64, startprocfn, a);
234 31cf0ab1 2004-12-28 devnull if(pid < 0){
235 31cf0ab1 2004-12-28 devnull fprint(2, "_procstart rfork_thread: %r\n");
236 31cf0ab1 2004-12-28 devnull abort();
240 31cf0ab1 2004-12-28 devnull static char *threadexitsmsg;
242 31cf0ab1 2004-12-28 devnull sigusr2handler(int s)
244 31cf0ab1 2004-12-28 devnull /* fprint(2, "%d usr2 %d\n", time(0), getpid()); */
245 31cf0ab1 2004-12-28 devnull if(threadexitsmsg)
246 31cf0ab1 2004-12-28 devnull _exits(threadexitsmsg);
250 31cf0ab1 2004-12-28 devnull threadexitsall(char *msg)
252 31cf0ab1 2004-12-28 devnull static int pid[1024];
253 31cf0ab1 2004-12-28 devnull int i, npid, mypid;
254 31cf0ab1 2004-12-28 devnull Proc *p;
256 31cf0ab1 2004-12-28 devnull if(msg == nil)
257 31cf0ab1 2004-12-28 devnull msg = "";
258 31cf0ab1 2004-12-28 devnull mypid = getpid();
259 31cf0ab1 2004-12-28 devnull lock(&_threadprocslock);
260 31cf0ab1 2004-12-28 devnull threadexitsmsg = msg;
261 31cf0ab1 2004-12-28 devnull npid = 0;
262 31cf0ab1 2004-12-28 devnull for(p=_threadprocs; p; p=p->next)
263 31cf0ab1 2004-12-28 devnull if(p->osprocid != mypid && p->osprocid >= 1)
264 31cf0ab1 2004-12-28 devnull pid[npid++] = p->osprocid;
265 31cf0ab1 2004-12-28 devnull for(i=0; i<npid; i++)
266 31cf0ab1 2004-12-28 devnull kill(pid[i], SIGUSR2);
267 31cf0ab1 2004-12-28 devnull unlock(&_threadprocslock);
268 31cf0ab1 2004-12-28 devnull exits(msg);
272 31cf0ab1 2004-12-28 devnull * per-process data, indexed by pid
274 31cf0ab1 2004-12-28 devnull * could use modify_ldt and a segment register
275 31cf0ab1 2004-12-28 devnull * to avoid the many calls to getpid(), but i don't
276 31cf0ab1 2004-12-28 devnull * care -- this is compatibility code. linux 2.6 with
277 31cf0ab1 2004-12-28 devnull * nptl is a good enough pthreads to avoid this whole file.
279 31cf0ab1 2004-12-28 devnull typedef struct Perproc Perproc;
280 31cf0ab1 2004-12-28 devnull struct Perproc
282 31cf0ab1 2004-12-28 devnull int pid;
283 31cf0ab1 2004-12-28 devnull Proc *proc;
286 31cf0ab1 2004-12-28 devnull static Lock perlock;
287 31cf0ab1 2004-12-28 devnull static Perproc perproc[1024];
288 31cf0ab1 2004-12-28 devnull #define P ((Proc*)-1)
290 31cf0ab1 2004-12-28 devnull static Perproc*
291 31cf0ab1 2004-12-28 devnull myperproc(void)
293 31cf0ab1 2004-12-28 devnull int i, pid, h;
294 31cf0ab1 2004-12-28 devnull Perproc *p;
296 31cf0ab1 2004-12-28 devnull pid = getpid();
297 31cf0ab1 2004-12-28 devnull h = pid%nelem(perproc);
298 31cf0ab1 2004-12-28 devnull for(i=0; i<nelem(perproc); i++){
299 31cf0ab1 2004-12-28 devnull p = &perproc[(i+h)%nelem(perproc)];
300 31cf0ab1 2004-12-28 devnull if(p->pid == pid)
301 31cf0ab1 2004-12-28 devnull return p;
302 31cf0ab1 2004-12-28 devnull if(p->pid == 0){
303 31cf0ab1 2004-12-28 devnull print("found 0 at %d (h=%d)\n", (i+h)%nelem(perproc), h);
307 31cf0ab1 2004-12-28 devnull fprint(2, "myperproc %d: cannot find self\n", pid);
308 31cf0ab1 2004-12-28 devnull abort();
309 31cf0ab1 2004-12-28 devnull return nil;
312 31cf0ab1 2004-12-28 devnull static Perproc*
313 31cf0ab1 2004-12-28 devnull newperproc(void)
315 31cf0ab1 2004-12-28 devnull int i, pid, h;
316 31cf0ab1 2004-12-28 devnull Perproc *p;
318 31cf0ab1 2004-12-28 devnull lock(&perlock);
319 31cf0ab1 2004-12-28 devnull pid = getpid();
320 31cf0ab1 2004-12-28 devnull h = pid%nelem(perproc);
321 31cf0ab1 2004-12-28 devnull for(i=0; i<nelem(perproc); i++){
322 31cf0ab1 2004-12-28 devnull p = &perproc[(i+h)%nelem(perproc)];
323 31cf0ab1 2004-12-28 devnull if(p->pid == pid || p->pid == -1 || p->pid == 0){
324 31cf0ab1 2004-12-28 devnull p->pid = pid;
325 31cf0ab1 2004-12-28 devnull unlock(&perlock);
326 31cf0ab1 2004-12-28 devnull return p;
329 31cf0ab1 2004-12-28 devnull fprint(2, "newperproc %d: out of procs\n", pid);
330 31cf0ab1 2004-12-28 devnull abort();
331 31cf0ab1 2004-12-28 devnull return nil;
335 31cf0ab1 2004-12-28 devnull _threadproc(void)
337 31cf0ab1 2004-12-28 devnull return myperproc()->proc;
341 31cf0ab1 2004-12-28 devnull _threadsetproc(Proc *p)
343 31cf0ab1 2004-12-28 devnull Perproc *pp;
346 31cf0ab1 2004-12-28 devnull p->osprocid = getpid();
347 31cf0ab1 2004-12-28 devnull pp = newperproc();
348 31cf0ab1 2004-12-28 devnull pp->proc = p;
349 31cf0ab1 2004-12-28 devnull if(p == nil)
350 31cf0ab1 2004-12-28 devnull pp->pid = -1;
354 31cf0ab1 2004-12-28 devnull _pthreadinit(void)
356 d54ead7f 2004-12-28 devnull __isthreaded = 1;
357 31cf0ab1 2004-12-28 devnull signal(SIGUSR2, sigusr2handler);
361 1d2533d0 2004-12-28 devnull _threadpexit(void)
363 1d2533d0 2004-12-28 devnull _exit(0);
368 d54ead7f 2004-12-28 devnull * FreeBSD 4 and earlier needs the context functions.
371 d54ead7f 2004-12-28 devnull makecontext(ucontext_t *ucp, void (*func)(void), int argc, ...)
373 d54ead7f 2004-12-28 devnull int *sp;
375 d54ead7f 2004-12-28 devnull sp = (int*)ucp->uc_stack.ss_sp+ucp->uc_stack.ss_size/4;
376 d54ead7f 2004-12-28 devnull sp -= argc;
377 d54ead7f 2004-12-28 devnull memmove(sp, &argc+1, argc*sizeof(int));
378 d54ead7f 2004-12-28 devnull *--sp = 0; /* return address */
379 d54ead7f 2004-12-28 devnull ucp->uc_mcontext.mc_eip = (long)func;
380 d54ead7f 2004-12-28 devnull ucp->uc_mcontext.mc_esp = (int)sp;
383 d54ead7f 2004-12-28 devnull extern int getmcontext(mcontext_t*);
384 d54ead7f 2004-12-28 devnull extern int setmcontext(mcontext_t*);
387 d54ead7f 2004-12-28 devnull getcontext(ucontext_t *uc)
389 d54ead7f 2004-12-28 devnull return getmcontext(&uc->uc_mcontext);
393 d54ead7f 2004-12-28 devnull setcontext(ucontext_t *uc)
395 d54ead7f 2004-12-28 devnull setmcontext(&uc->uc_mcontext);
399 d54ead7f 2004-12-28 devnull swapcontext(ucontext_t *oucp, ucontext_t *ucp)
401 d54ead7f 2004-12-28 devnull if(getcontext(oucp) == 0)
402 d54ead7f 2004-12-28 devnull setcontext(ucp);
403 d54ead7f 2004-12-28 devnull return 0;