1 1544f909 2004-12-25 devnull #include "threadimpl.h"
3 1544f909 2004-12-25 devnull int _threaddebuglevel;
5 1544f909 2004-12-25 devnull static uint threadnproc;
6 1544f909 2004-12-25 devnull static uint threadnsysproc;
7 1544f909 2004-12-25 devnull static Lock threadnproclock;
8 1544f909 2004-12-25 devnull static Ref threadidref;
9 4dbefdd4 2004-12-27 devnull static Proc *threadmainproc;
11 4dbefdd4 2004-12-27 devnull static void addproc(Proc*);
12 4dbefdd4 2004-12-27 devnull static void delproc(Proc*);
13 1544f909 2004-12-25 devnull static void addthread(_Threadlist*, _Thread*);
14 1544f909 2004-12-25 devnull static void delthread(_Threadlist*, _Thread*);
15 1544f909 2004-12-25 devnull static void addthreadinproc(Proc*, _Thread*);
16 1544f909 2004-12-25 devnull static void delthreadinproc(Proc*, _Thread*);
17 1544f909 2004-12-25 devnull static void contextswitch(Context *from, Context *to);
18 1fdb4a54 2005-01-23 devnull static void procscheduler(Proc*);
19 6dde87f8 2005-09-26 devnull static int threadinfo(void*, char*);
21 2c87dda8 2004-12-28 devnull static void
22 2c87dda8 2004-12-28 devnull _threaddebug(char *fmt, ...)
24 2c87dda8 2004-12-28 devnull va_list arg;
25 2c87dda8 2004-12-28 devnull char buf[128];
26 2c87dda8 2004-12-28 devnull _Thread *t;
28 af89fc10 2005-01-06 devnull static int fd = -1;
30 5b092791 2005-09-26 devnull if(_threaddebuglevel == 0)
33 af89fc10 2005-01-06 devnull if(fd < 0){
34 af89fc10 2005-01-06 devnull p = strrchr(argv0, '/');
38 af89fc10 2005-01-06 devnull p = argv0;
39 af89fc10 2005-01-06 devnull snprint(buf, sizeof buf, "/tmp/%s.tlog", p);
40 af89fc10 2005-01-06 devnull if((fd = create(buf, OWRITE, 0666)) < 0)
41 af89fc10 2005-01-06 devnull fd = open("/dev/null", OWRITE);
44 2c87dda8 2004-12-28 devnull va_start(arg, fmt);
45 2c87dda8 2004-12-28 devnull vsnprint(buf, sizeof buf, fmt, arg);
46 2c87dda8 2004-12-28 devnull va_end(arg);
47 2c87dda8 2004-12-28 devnull t = proc()->thread;
49 af89fc10 2005-01-06 devnull fprint(fd, "%d.%d: %s\n", getpid(), t->id, buf);
51 af89fc10 2005-01-06 devnull fprint(fd, "%d._: %s\n", getpid(), buf);
54 1544f909 2004-12-25 devnull static _Thread*
55 1544f909 2004-12-25 devnull getthreadnow(void)
57 1544f909 2004-12-25 devnull return proc()->thread;
59 1544f909 2004-12-25 devnull _Thread *(*threadnow)(void) = getthreadnow;
61 1544f909 2004-12-25 devnull static Proc*
62 1544f909 2004-12-25 devnull procalloc(void)
66 1544f909 2004-12-25 devnull p = malloc(sizeof *p);
67 73722a8b 2004-12-27 devnull if(p == nil)
68 73722a8b 2004-12-27 devnull sysfatal("procalloc malloc: %r");
69 1544f909 2004-12-25 devnull memset(p, 0, sizeof *p);
70 4dbefdd4 2004-12-27 devnull addproc(p);
71 1544f909 2004-12-25 devnull lock(&threadnproclock);
72 1544f909 2004-12-25 devnull threadnproc++;
73 1544f909 2004-12-25 devnull unlock(&threadnproclock);
74 1544f909 2004-12-25 devnull return p;
77 1544f909 2004-12-25 devnull static void
78 86502504 2005-01-18 devnull threadstart(uint y, uint x)
80 1544f909 2004-12-25 devnull _Thread *t;
83 86502504 2005-01-18 devnull z = x<<16; /* hide undefined 32-bit shift from 32-bit compilers */
84 86502504 2005-01-18 devnull z <<= 16;
86 86502504 2005-01-18 devnull t = (_Thread*)z;
88 b8f742db 2005-01-11 devnull //print("threadstart %p\n", v);
89 1544f909 2004-12-25 devnull t->startfn(t->startarg);
90 b8f742db 2005-01-11 devnull //print("threadexits %p\n", v);
91 4dbefdd4 2004-12-27 devnull threadexits(nil);
92 b8f742db 2005-01-11 devnull //print("not reacehd\n");
95 1544f909 2004-12-25 devnull static _Thread*
96 1544f909 2004-12-25 devnull threadalloc(void (*fn)(void*), void *arg, uint stack)
98 1544f909 2004-12-25 devnull _Thread *t;
99 1544f909 2004-12-25 devnull sigset_t zero;
100 86502504 2005-01-18 devnull uint x, y;
101 86502504 2005-01-18 devnull ulong z;
103 1544f909 2004-12-25 devnull /* allocate the task and stack together */
104 1544f909 2004-12-25 devnull t = malloc(sizeof *t+stack);
105 73722a8b 2004-12-27 devnull if(t == nil)
106 73722a8b 2004-12-27 devnull sysfatal("threadalloc malloc: %r");
107 1544f909 2004-12-25 devnull memset(t, 0, sizeof *t);
108 1544f909 2004-12-25 devnull t->stk = (uchar*)(t+1);
109 1544f909 2004-12-25 devnull t->stksize = stack;
110 1544f909 2004-12-25 devnull t->id = incref(&threadidref);
111 1544f909 2004-12-25 devnull t->startfn = fn;
112 1544f909 2004-12-25 devnull t->startarg = arg;
114 1544f909 2004-12-25 devnull /* do a reasonable initialization */
115 1544f909 2004-12-25 devnull memset(&t->context.uc, 0, sizeof t->context.uc);
116 1544f909 2004-12-25 devnull sigemptyset(&zero);
117 1544f909 2004-12-25 devnull sigprocmask(SIG_BLOCK, &zero, &t->context.uc.uc_sigmask);
119 d54ead7f 2004-12-28 devnull /* must initialize with current context */
120 37cc72d3 2005-01-11 devnull if(getcontext(&t->context.uc) < 0)
121 37cc72d3 2005-01-11 devnull sysfatal("threadalloc getcontext: %r");
123 1544f909 2004-12-25 devnull /* call makecontext to do the real work. */
124 1544f909 2004-12-25 devnull /* leave a few words open on both ends */
125 1544f909 2004-12-25 devnull t->context.uc.uc_stack.ss_sp = t->stk+8;
126 2c87dda8 2004-12-28 devnull t->context.uc.uc_stack.ss_size = t->stksize-64;
127 d08f4066 2005-07-27 devnull #if defined(__sun__) && !defined(__MAKECONTEXT_V2_SOURCE) /* sigh */
128 37cc72d3 2005-01-11 devnull /* can avoid this with __MAKECONTEXT_V2_SOURCE but only on SunOS 5.9 */
129 2b3b1e89 2005-01-14 devnull t->context.uc.uc_stack.ss_sp =
130 2b3b1e89 2005-01-14 devnull (char*)t->context.uc.uc_stack.ss_sp
131 2b3b1e89 2005-01-14 devnull +t->context.uc.uc_stack.ss_size;
134 86502504 2005-01-18 devnull * All this magic is because you have to pass makecontext a
135 86502504 2005-01-18 devnull * function that takes some number of word-sized variables,
136 86502504 2005-01-18 devnull * and on 64-bit machines pointers are bigger than words.
138 86502504 2005-01-18 devnull z = (ulong)t;
140 86502504 2005-01-18 devnull z >>= 16; /* hide undefined 32-bit shift from 32-bit compilers */
141 86502504 2005-01-18 devnull x = z>>16;
142 2b604081 2005-05-07 devnull makecontext(&t->context.uc, (void(*)(void))threadstart, 2, y, x);
144 1544f909 2004-12-25 devnull return t;
147 1544f909 2004-12-25 devnull _Thread*
148 1544f909 2004-12-25 devnull _threadcreate(Proc *p, void (*fn)(void*), void *arg, uint stack)
150 1544f909 2004-12-25 devnull _Thread *t;
152 1544f909 2004-12-25 devnull t = threadalloc(fn, arg, stack);
153 1544f909 2004-12-25 devnull t->proc = p;
154 1544f909 2004-12-25 devnull addthreadinproc(p, t);
155 1544f909 2004-12-25 devnull p->nthread++;
156 1544f909 2004-12-25 devnull _threadready(t);
157 1544f909 2004-12-25 devnull return t;
161 1544f909 2004-12-25 devnull threadcreate(void (*fn)(void*), void *arg, uint stack)
163 1544f909 2004-12-25 devnull _Thread *t;
165 1544f909 2004-12-25 devnull t = _threadcreate(proc(), fn, arg, stack);
166 1544f909 2004-12-25 devnull return t->id;
170 1544f909 2004-12-25 devnull proccreate(void (*fn)(void*), void *arg, uint stack)
173 1544f909 2004-12-25 devnull _Thread *t;
174 1544f909 2004-12-25 devnull Proc *p;
176 1544f909 2004-12-25 devnull p = procalloc();
177 1544f909 2004-12-25 devnull t = _threadcreate(p, fn, arg, stack);
178 803292c6 2005-01-18 devnull id = t->id; /* t might be freed after _procstart */
179 1fdb4a54 2005-01-23 devnull _procstart(p, procscheduler);
180 803292c6 2005-01-18 devnull return id;
184 1544f909 2004-12-25 devnull _threadswitch(void)
186 1544f909 2004-12-25 devnull Proc *p;
188 c8b6342d 2005-01-13 devnull needstack(0);
189 1544f909 2004-12-25 devnull p = proc();
190 b8f742db 2005-01-11 devnull //print("threadswtch %p\n", p);
191 1544f909 2004-12-25 devnull contextswitch(&p->thread->context, &p->schedcontext);
195 1544f909 2004-12-25 devnull _threadready(_Thread *t)
197 1544f909 2004-12-25 devnull Proc *p;
199 1544f909 2004-12-25 devnull p = t->proc;
200 1544f909 2004-12-25 devnull lock(&p->lock);
201 a0a331aa 2005-01-06 devnull p->runrend.l = &p->lock;
202 1544f909 2004-12-25 devnull addthread(&p->runqueue, t);
203 8cbd854a 2004-12-27 devnull //print("%d wake for job %d->%d\n", time(0), getpid(), p->osprocid);
204 8cbd854a 2004-12-27 devnull if(p != proc())
205 a0a331aa 2005-01-06 devnull _procwakeupandunlock(&p->runrend);
207 a0a331aa 2005-01-06 devnull unlock(&p->lock);
211 615e0f9f 2006-02-07 devnull threadidle(void)
214 615e0f9f 2006-02-07 devnull Proc *p;
216 615e0f9f 2006-02-07 devnull p = proc();
217 615e0f9f 2006-02-07 devnull n = p->nswitch;
218 615e0f9f 2006-02-07 devnull lock(&p->lock);
219 615e0f9f 2006-02-07 devnull p->runrend.l = &p->lock;
220 615e0f9f 2006-02-07 devnull addthread(&p->idlequeue, p->thread);
221 615e0f9f 2006-02-07 devnull unlock(&p->lock);
222 615e0f9f 2006-02-07 devnull _threadswitch();
223 615e0f9f 2006-02-07 devnull return p->nswitch - n;
227 1544f909 2004-12-25 devnull threadyield(void)
230 8cbd854a 2004-12-27 devnull Proc *p;
232 8cbd854a 2004-12-27 devnull p = proc();
233 8cbd854a 2004-12-27 devnull n = p->nswitch;
234 8cbd854a 2004-12-27 devnull _threadready(p->thread);
235 1544f909 2004-12-25 devnull _threadswitch();
236 8cbd854a 2004-12-27 devnull return p->nswitch - n;
240 1544f909 2004-12-25 devnull threadexits(char *msg)
242 1544f909 2004-12-25 devnull Proc *p;
244 1544f909 2004-12-25 devnull p = proc();
245 4dbefdd4 2004-12-27 devnull if(msg == nil)
246 4dbefdd4 2004-12-27 devnull msg = "";
247 1544f909 2004-12-25 devnull utfecpy(p->msg, p->msg+sizeof p->msg, msg);
248 4dbefdd4 2004-12-27 devnull proc()->thread->exiting = 1;
249 4dbefdd4 2004-12-27 devnull _threadswitch();
252 1544f909 2004-12-25 devnull static void
253 1544f909 2004-12-25 devnull contextswitch(Context *from, Context *to)
255 1544f909 2004-12-25 devnull if(swapcontext(&from->uc, &to->uc) < 0){
256 1544f909 2004-12-25 devnull fprint(2, "swapcontext failed: %r\n");
257 1544f909 2004-12-25 devnull assert(0);
261 1544f909 2004-12-25 devnull static void
262 1fdb4a54 2005-01-23 devnull procscheduler(Proc *p)
264 1544f909 2004-12-25 devnull _Thread *t;
266 1544f909 2004-12-25 devnull setproc(p);
267 2c87dda8 2004-12-28 devnull _threaddebug("scheduler enter");
268 b8f742db 2005-01-11 devnull // print("s %p\n", p);
269 1544f909 2004-12-25 devnull lock(&p->lock);
270 1544f909 2004-12-25 devnull for(;;){
271 1544f909 2004-12-25 devnull while((t = p->runqueue.head) == nil){
272 1544f909 2004-12-25 devnull if(p->nthread == 0)
273 1544f909 2004-12-25 devnull goto Out;
274 615e0f9f 2006-02-07 devnull if((t = p->idlequeue.head) != nil){
276 615e0f9f 2006-02-07 devnull * Run all the idling threads once.
278 615e0f9f 2006-02-07 devnull while((t = p->idlequeue.head) != nil){
279 615e0f9f 2006-02-07 devnull delthread(&p->idlequeue, t);
280 615e0f9f 2006-02-07 devnull addthread(&p->runqueue, t);
282 615e0f9f 2006-02-07 devnull continue;
284 1544f909 2004-12-25 devnull p->runrend.l = &p->lock;
285 2c87dda8 2004-12-28 devnull _threaddebug("scheduler sleep");
286 1544f909 2004-12-25 devnull _procsleep(&p->runrend);
287 2c87dda8 2004-12-28 devnull _threaddebug("scheduler wake");
289 1544f909 2004-12-25 devnull delthread(&p->runqueue, t);
290 1544f909 2004-12-25 devnull unlock(&p->lock);
291 1544f909 2004-12-25 devnull p->thread = t;
292 8cbd854a 2004-12-27 devnull p->nswitch++;
293 2c87dda8 2004-12-28 devnull _threaddebug("run %d (%s)", t->id, t->name);
294 1544f909 2004-12-25 devnull contextswitch(&p->schedcontext, &t->context);
295 b8f742db 2005-01-11 devnull //print("back in scheduler\n");
296 1544f909 2004-12-25 devnull p->thread = nil;
297 1544f909 2004-12-25 devnull lock(&p->lock);
298 1544f909 2004-12-25 devnull if(t->exiting){
299 1544f909 2004-12-25 devnull delthreadinproc(p, t);
300 1544f909 2004-12-25 devnull p->nthread--;
301 615e0f9f 2006-02-07 devnull //print("nthread %d\n", p->nthread);
302 1544f909 2004-12-25 devnull free(t);
307 2c87dda8 2004-12-28 devnull _threaddebug("scheduler exit");
308 8ee6ad4d 2005-02-14 devnull if(p->mainproc){
310 8ee6ad4d 2005-02-14 devnull * Stupid bug - on Linux 2.6 and maybe elsewhere,
311 8ee6ad4d 2005-02-14 devnull * if the main thread exits then the others keep running
312 8ee6ad4d 2005-02-14 devnull * but the process shows up as a zombie in ps and is not
313 8ee6ad4d 2005-02-14 devnull * attachable with ptrace. We'll just sit around pretending
314 8ee6ad4d 2005-02-14 devnull * to be a system proc instead of exiting.
316 8ee6ad4d 2005-02-14 devnull _threaddaemonize();
317 8ee6ad4d 2005-02-14 devnull lock(&threadnproclock);
318 8ee6ad4d 2005-02-14 devnull if(++threadnsysproc == threadnproc)
319 8ee6ad4d 2005-02-14 devnull threadexitsall(p->msg);
320 8ee6ad4d 2005-02-14 devnull p->sysproc = 1;
321 8ee6ad4d 2005-02-14 devnull unlock(&threadnproclock);
323 8ee6ad4d 2005-02-14 devnull sleep(1000);
326 4dbefdd4 2004-12-27 devnull delproc(p);
327 1544f909 2004-12-25 devnull lock(&threadnproclock);
328 1544f909 2004-12-25 devnull if(p->sysproc)
329 1544f909 2004-12-25 devnull --threadnsysproc;
330 1544f909 2004-12-25 devnull if(--threadnproc == threadnsysproc)
331 4dbefdd4 2004-12-27 devnull threadexitsall(p->msg);
332 1544f909 2004-12-25 devnull unlock(&threadnproclock);
333 1544f909 2004-12-25 devnull unlock(&p->lock);
334 1544f909 2004-12-25 devnull free(p);
338 1544f909 2004-12-25 devnull _threadsetsysproc(void)
340 1544f909 2004-12-25 devnull lock(&threadnproclock);
341 1544f909 2004-12-25 devnull if(++threadnsysproc == threadnproc)
342 8ee6ad4d 2005-02-14 devnull threadexitsall(nil);
343 1544f909 2004-12-25 devnull unlock(&threadnproclock);
344 1544f909 2004-12-25 devnull proc()->sysproc = 1;
348 4dbefdd4 2004-12-27 devnull procdata(void)
350 4dbefdd4 2004-12-27 devnull return &proc()->udata;
354 8dd8a81f 2005-03-18 devnull threaddata(void)
356 8dd8a81f 2005-03-18 devnull return &proc()->thread->udata;
359 4dbefdd4 2004-12-27 devnull extern Jmp *(*_notejmpbuf)(void);
360 4dbefdd4 2004-12-27 devnull static Jmp*
361 4dbefdd4 2004-12-27 devnull threadnotejmp(void)
363 4dbefdd4 2004-12-27 devnull return &proc()->sigjmp;
367 1544f909 2004-12-25 devnull * debugging
370 1544f909 2004-12-25 devnull threadsetname(char *fmt, ...)
372 1544f909 2004-12-25 devnull va_list arg;
373 1544f909 2004-12-25 devnull _Thread *t;
375 1544f909 2004-12-25 devnull t = proc()->thread;
376 1544f909 2004-12-25 devnull va_start(arg, fmt);
377 1544f909 2004-12-25 devnull vsnprint(t->name, sizeof t->name, fmt, arg);
378 1544f909 2004-12-25 devnull va_end(arg);
382 515b6e5b 2005-01-18 devnull threadgetname(void)
384 515b6e5b 2005-01-18 devnull return proc()->thread->name;
388 1544f909 2004-12-25 devnull threadsetstate(char *fmt, ...)
390 1544f909 2004-12-25 devnull va_list arg;
391 1544f909 2004-12-25 devnull _Thread *t;
393 1544f909 2004-12-25 devnull t = proc()->thread;
394 1544f909 2004-12-25 devnull va_start(arg, fmt);
395 1544f909 2004-12-25 devnull vsnprint(t->state, sizeof t->name, fmt, arg);
396 1544f909 2004-12-25 devnull va_end(arg);
400 9eda38e5 2006-02-05 devnull threadid(void)
402 9eda38e5 2006-02-05 devnull _Thread *t;
404 9eda38e5 2006-02-05 devnull t = proc()->thread;
405 9eda38e5 2006-02-05 devnull return t->id;
409 c8b6342d 2005-01-13 devnull needstack(int n)
411 c8b6342d 2005-01-13 devnull _Thread *t;
413 c8b6342d 2005-01-13 devnull t = proc()->thread;
415 c8b6342d 2005-01-13 devnull if((char*)&t <= (char*)t->stk
416 c8b6342d 2005-01-13 devnull || (char*)&t - (char*)t->stk < 256+n){
417 1fdb4a54 2005-01-23 devnull fprint(2, "thread stack overflow: &t=%p tstk=%p n=%d\n", &t, t->stk, 256+n);
418 c8b6342d 2005-01-13 devnull abort();
423 1544f909 2004-12-25 devnull * locking
425 1544f909 2004-12-25 devnull static int
426 1544f909 2004-12-25 devnull threadqlock(QLock *l, int block, ulong pc)
428 b8f742db 2005-01-11 devnull //print("threadqlock %p\n", l);
429 1544f909 2004-12-25 devnull lock(&l->l);
430 1544f909 2004-12-25 devnull if(l->owner == nil){
431 1544f909 2004-12-25 devnull l->owner = (*threadnow)();
432 1544f909 2004-12-25 devnull //print("qlock %p @%#x by %p\n", l, pc, l->owner);
433 1544f909 2004-12-25 devnull unlock(&l->l);
434 1544f909 2004-12-25 devnull return 1;
436 1544f909 2004-12-25 devnull if(!block){
437 1544f909 2004-12-25 devnull unlock(&l->l);
438 1544f909 2004-12-25 devnull return 0;
440 1544f909 2004-12-25 devnull //print("qsleep %p @%#x by %p\n", l, pc, (*threadnow)());
441 1544f909 2004-12-25 devnull addthread(&l->waiting, (*threadnow)());
442 1544f909 2004-12-25 devnull unlock(&l->l);
444 1544f909 2004-12-25 devnull _threadswitch();
446 1544f909 2004-12-25 devnull if(l->owner != (*threadnow)()){
447 15cd8255 2005-01-04 devnull fprint(2, "%s: qlock pc=0x%lux owner=%p self=%p oops\n",
448 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
449 1544f909 2004-12-25 devnull abort();
451 1544f909 2004-12-25 devnull //print("qlock wakeup %p @%#x by %p\n", l, pc, (*threadnow)());
452 1544f909 2004-12-25 devnull return 1;
455 1544f909 2004-12-25 devnull static void
456 1544f909 2004-12-25 devnull threadqunlock(QLock *l, ulong pc)
458 80b8842f 2005-02-15 devnull _Thread *ready;
460 1544f909 2004-12-25 devnull lock(&l->l);
461 1544f909 2004-12-25 devnull //print("qlock unlock %p @%#x by %p (owner %p)\n", l, pc, (*threadnow)(), l->owner);
462 1aa9c533 2005-01-07 devnull if(l->owner == 0){
463 15cd8255 2005-01-04 devnull fprint(2, "%s: qunlock pc=0x%lux owner=%p self=%p oops\n",
464 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
465 1aa9c533 2005-01-07 devnull abort();
467 80b8842f 2005-02-15 devnull if((l->owner = ready = l->waiting.head) != nil)
468 1544f909 2004-12-25 devnull delthread(&l->waiting, l->owner);
470 80b8842f 2005-02-15 devnull * N.B. Cannot call _threadready() before unlocking l->l,
471 80b8842f 2005-02-15 devnull * because the thread we are readying might:
472 80b8842f 2005-02-15 devnull * - be in another proc
473 80b8842f 2005-02-15 devnull * - start running immediately
474 80b8842f 2005-02-15 devnull * - and free l before we get a chance to run again
476 1544f909 2004-12-25 devnull unlock(&l->l);
477 80b8842f 2005-02-15 devnull if(ready)
478 80b8842f 2005-02-15 devnull _threadready(l->owner);
481 1544f909 2004-12-25 devnull static int
482 1544f909 2004-12-25 devnull threadrlock(RWLock *l, int block, ulong pc)
484 1544f909 2004-12-25 devnull USED(pc);
486 1544f909 2004-12-25 devnull lock(&l->l);
487 1544f909 2004-12-25 devnull if(l->writer == nil && l->wwaiting.head == nil){
488 1544f909 2004-12-25 devnull l->readers++;
489 1544f909 2004-12-25 devnull unlock(&l->l);
490 1544f909 2004-12-25 devnull return 1;
492 1544f909 2004-12-25 devnull if(!block){
493 1544f909 2004-12-25 devnull unlock(&l->l);
494 1544f909 2004-12-25 devnull return 0;
496 1544f909 2004-12-25 devnull addthread(&l->rwaiting, (*threadnow)());
497 1544f909 2004-12-25 devnull unlock(&l->l);
498 1544f909 2004-12-25 devnull _threadswitch();
499 1544f909 2004-12-25 devnull return 1;
502 1544f909 2004-12-25 devnull static int
503 1544f909 2004-12-25 devnull threadwlock(RWLock *l, int block, ulong pc)
505 1544f909 2004-12-25 devnull USED(pc);
507 1544f909 2004-12-25 devnull lock(&l->l);
508 1544f909 2004-12-25 devnull if(l->writer == nil && l->readers == 0){
509 1544f909 2004-12-25 devnull l->writer = (*threadnow)();
510 1544f909 2004-12-25 devnull unlock(&l->l);
511 1544f909 2004-12-25 devnull return 1;
513 1544f909 2004-12-25 devnull if(!block){
514 1544f909 2004-12-25 devnull unlock(&l->l);
515 1544f909 2004-12-25 devnull return 0;
517 1544f909 2004-12-25 devnull addthread(&l->wwaiting, (*threadnow)());
518 1544f909 2004-12-25 devnull unlock(&l->l);
519 1544f909 2004-12-25 devnull _threadswitch();
520 1544f909 2004-12-25 devnull return 1;
523 1544f909 2004-12-25 devnull static void
524 1544f909 2004-12-25 devnull threadrunlock(RWLock *l, ulong pc)
526 1544f909 2004-12-25 devnull _Thread *t;
528 1544f909 2004-12-25 devnull USED(pc);
529 80b8842f 2005-02-15 devnull t = nil;
530 1544f909 2004-12-25 devnull lock(&l->l);
531 1544f909 2004-12-25 devnull --l->readers;
532 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
533 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
534 1544f909 2004-12-25 devnull l->writer = t;
536 1544f909 2004-12-25 devnull unlock(&l->l);
538 80b8842f 2005-02-15 devnull _threadready(t);
542 1544f909 2004-12-25 devnull static void
543 1544f909 2004-12-25 devnull threadwunlock(RWLock *l, ulong pc)
545 1544f909 2004-12-25 devnull _Thread *t;
547 1544f909 2004-12-25 devnull USED(pc);
548 1544f909 2004-12-25 devnull lock(&l->l);
549 1544f909 2004-12-25 devnull l->writer = nil;
550 1544f909 2004-12-25 devnull assert(l->readers == 0);
551 1544f909 2004-12-25 devnull while((t = l->rwaiting.head) != nil){
552 1544f909 2004-12-25 devnull delthread(&l->rwaiting, t);
553 1544f909 2004-12-25 devnull l->readers++;
554 1544f909 2004-12-25 devnull _threadready(t);
556 80b8842f 2005-02-15 devnull t = nil;
557 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
558 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
559 1544f909 2004-12-25 devnull l->writer = t;
561 1544f909 2004-12-25 devnull unlock(&l->l);
563 80b8842f 2005-02-15 devnull _threadready(t);
567 1544f909 2004-12-25 devnull * sleep and wakeup
569 1544f909 2004-12-25 devnull static void
570 1544f909 2004-12-25 devnull threadrsleep(Rendez *r, ulong pc)
572 1544f909 2004-12-25 devnull addthread(&r->waiting, proc()->thread);
573 1544f909 2004-12-25 devnull qunlock(r->l);
574 1544f909 2004-12-25 devnull _threadswitch();
575 1544f909 2004-12-25 devnull qlock(r->l);
578 1544f909 2004-12-25 devnull static int
579 1544f909 2004-12-25 devnull threadrwakeup(Rendez *r, int all, ulong pc)
582 1544f909 2004-12-25 devnull _Thread *t;
584 1544f909 2004-12-25 devnull for(i=0;; i++){
585 1544f909 2004-12-25 devnull if(i==1 && !all)
587 1544f909 2004-12-25 devnull if((t = r->waiting.head) == nil)
589 1544f909 2004-12-25 devnull delthread(&r->waiting, t);
590 1544f909 2004-12-25 devnull _threadready(t);
592 1544f909 2004-12-25 devnull return i;
596 4dbefdd4 2004-12-27 devnull * startup
599 4dbefdd4 2004-12-27 devnull static int threadargc;
600 4dbefdd4 2004-12-27 devnull static char **threadargv;
601 4dbefdd4 2004-12-27 devnull int mainstacksize;
603 4dbefdd4 2004-12-27 devnull static void
604 4dbefdd4 2004-12-27 devnull threadmainstart(void *v)
606 4dbefdd4 2004-12-27 devnull USED(v);
609 0b561695 2005-01-16 devnull * N.B. This call to proc() is a program's first call (indirectly) to a
610 0b561695 2005-01-16 devnull * pthreads function while executing on a non-pthreads-allocated
611 0b561695 2005-01-16 devnull * stack. If the pthreads implementation is using the stack pointer
612 0b561695 2005-01-16 devnull * to locate the per-thread data, then this call will blow up.
613 0b561695 2005-01-16 devnull * This means the pthread implementation is not suitable for
614 0b561695 2005-01-16 devnull * running under libthread. Time to write your own. Sorry.
616 4dbefdd4 2004-12-27 devnull threadmainproc = proc();
617 4dbefdd4 2004-12-27 devnull threadmain(threadargc, threadargv);
621 4dbefdd4 2004-12-27 devnull main(int argc, char **argv)
623 4dbefdd4 2004-12-27 devnull Proc *p;
625 1d2533d0 2004-12-28 devnull argv0 = argv[0];
627 615e0f9f 2006-02-07 devnull if(getenv("NOLIBTHREADDAEMONIZE") == nil)
628 615e0f9f 2006-02-07 devnull _threadsetupdaemonize();
630 4dbefdd4 2004-12-27 devnull threadargc = argc;
631 4dbefdd4 2004-12-27 devnull threadargv = argv;
634 4dbefdd4 2004-12-27 devnull * Install locking routines into C library.
636 4dbefdd4 2004-12-27 devnull _lock = _threadlock;
637 4dbefdd4 2004-12-27 devnull _unlock = _threadunlock;
638 4dbefdd4 2004-12-27 devnull _qlock = threadqlock;
639 4dbefdd4 2004-12-27 devnull _qunlock = threadqunlock;
640 4dbefdd4 2004-12-27 devnull _rlock = threadrlock;
641 4dbefdd4 2004-12-27 devnull _runlock = threadrunlock;
642 4dbefdd4 2004-12-27 devnull _wlock = threadwlock;
643 4dbefdd4 2004-12-27 devnull _wunlock = threadwunlock;
644 4dbefdd4 2004-12-27 devnull _rsleep = threadrsleep;
645 4dbefdd4 2004-12-27 devnull _rwakeup = threadrwakeup;
646 4dbefdd4 2004-12-27 devnull _notejmpbuf = threadnotejmp;
648 4dbefdd4 2004-12-27 devnull _pthreadinit();
649 4dbefdd4 2004-12-27 devnull p = procalloc();
650 8ee6ad4d 2005-02-14 devnull p->mainproc = 1;
651 4dbefdd4 2004-12-27 devnull _threadsetproc(p);
652 4dbefdd4 2004-12-27 devnull if(mainstacksize == 0)
653 a0e8d02d 2005-01-23 devnull mainstacksize = 256*1024;
654 6dde87f8 2005-09-26 devnull atnotify(threadinfo, 1);
655 4dbefdd4 2004-12-27 devnull _threadcreate(p, threadmainstart, nil, mainstacksize);
656 1fdb4a54 2005-01-23 devnull procscheduler(p);
657 d7688cce 2005-02-16 devnull sysfatal("procscheduler returned in threadmain!");
658 8ee6ad4d 2005-02-14 devnull /* does not return */
659 15cd8255 2005-01-04 devnull return 0;
663 1544f909 2004-12-25 devnull * hooray for linked lists
665 1544f909 2004-12-25 devnull static void
666 1544f909 2004-12-25 devnull addthread(_Threadlist *l, _Thread *t)
668 1544f909 2004-12-25 devnull if(l->tail){
669 1544f909 2004-12-25 devnull l->tail->next = t;
670 1544f909 2004-12-25 devnull t->prev = l->tail;
672 1544f909 2004-12-25 devnull l->head = t;
673 1544f909 2004-12-25 devnull t->prev = nil;
675 1544f909 2004-12-25 devnull l->tail = t;
676 1544f909 2004-12-25 devnull t->next = nil;
679 1544f909 2004-12-25 devnull static void
680 1544f909 2004-12-25 devnull delthread(_Threadlist *l, _Thread *t)
682 1544f909 2004-12-25 devnull if(t->prev)
683 1544f909 2004-12-25 devnull t->prev->next = t->next;
685 1544f909 2004-12-25 devnull l->head = t->next;
686 1544f909 2004-12-25 devnull if(t->next)
687 1544f909 2004-12-25 devnull t->next->prev = t->prev;
689 1544f909 2004-12-25 devnull l->tail = t->prev;
692 1544f909 2004-12-25 devnull static void
693 1544f909 2004-12-25 devnull addthreadinproc(Proc *p, _Thread *t)
695 1544f909 2004-12-25 devnull _Threadlist *l;
697 1544f909 2004-12-25 devnull l = &p->allthreads;
698 1544f909 2004-12-25 devnull if(l->tail){
699 1544f909 2004-12-25 devnull l->tail->allnext = t;
700 1544f909 2004-12-25 devnull t->allprev = l->tail;
702 1544f909 2004-12-25 devnull l->head = t;
703 1544f909 2004-12-25 devnull t->allprev = nil;
705 1544f909 2004-12-25 devnull l->tail = t;
706 1544f909 2004-12-25 devnull t->allnext = nil;
709 1544f909 2004-12-25 devnull static void
710 1544f909 2004-12-25 devnull delthreadinproc(Proc *p, _Thread *t)
712 1544f909 2004-12-25 devnull _Threadlist *l;
714 1544f909 2004-12-25 devnull l = &p->allthreads;
715 1544f909 2004-12-25 devnull if(t->allprev)
716 1544f909 2004-12-25 devnull t->allprev->allnext = t->allnext;
718 1544f909 2004-12-25 devnull l->head = t->allnext;
719 1544f909 2004-12-25 devnull if(t->allnext)
720 1544f909 2004-12-25 devnull t->allnext->allprev = t->allprev;
722 1544f909 2004-12-25 devnull l->tail = t->allprev;
725 4dbefdd4 2004-12-27 devnull Proc *_threadprocs;
726 4dbefdd4 2004-12-27 devnull Lock _threadprocslock;
727 4dbefdd4 2004-12-27 devnull static Proc *_threadprocstail;
729 1544f909 2004-12-25 devnull static void
730 4dbefdd4 2004-12-27 devnull addproc(Proc *p)
732 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
733 4dbefdd4 2004-12-27 devnull if(_threadprocstail){
734 4dbefdd4 2004-12-27 devnull _threadprocstail->next = p;
735 4dbefdd4 2004-12-27 devnull p->prev = _threadprocstail;
737 4dbefdd4 2004-12-27 devnull _threadprocs = p;
738 4dbefdd4 2004-12-27 devnull p->prev = nil;
740 4dbefdd4 2004-12-27 devnull _threadprocstail = p;
741 4dbefdd4 2004-12-27 devnull p->next = nil;
742 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
745 4dbefdd4 2004-12-27 devnull static void
746 4dbefdd4 2004-12-27 devnull delproc(Proc *p)
748 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
749 4dbefdd4 2004-12-27 devnull if(p->prev)
750 4dbefdd4 2004-12-27 devnull p->prev->next = p->next;
752 4dbefdd4 2004-12-27 devnull _threadprocs = p->next;
753 4dbefdd4 2004-12-27 devnull if(p->next)
754 4dbefdd4 2004-12-27 devnull p->next->prev = p->prev;
756 4dbefdd4 2004-12-27 devnull _threadprocstail = p->prev;
757 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
761 15cd8255 2005-01-04 devnull * notify - for now just use the usual mechanisms
764 15cd8255 2005-01-04 devnull threadnotify(int (*f)(void*, char*), int in)
766 15cd8255 2005-01-04 devnull atnotify(f, in);
769 6dde87f8 2005-09-26 devnull static int
770 6dde87f8 2005-09-26 devnull onrunqueue(Proc *p, _Thread *t)
772 6dde87f8 2005-09-26 devnull _Thread *tt;
774 6dde87f8 2005-09-26 devnull for(tt=p->runqueue.head; tt; tt=tt->next)
775 6dde87f8 2005-09-26 devnull if(tt == t)
776 6dde87f8 2005-09-26 devnull return 1;
777 6dde87f8 2005-09-26 devnull return 0;
781 6dde87f8 2005-09-26 devnull * print state - called from SIGINFO
783 6dde87f8 2005-09-26 devnull static int
784 6dde87f8 2005-09-26 devnull threadinfo(void *v, char *s)
786 6dde87f8 2005-09-26 devnull Proc *p;
787 6dde87f8 2005-09-26 devnull _Thread *t;
789 6dde87f8 2005-09-26 devnull if(strcmp(s, "quit") != 0 && strcmp(s, "sys: status request") != 0)
790 6dde87f8 2005-09-26 devnull return 0;
792 6dde87f8 2005-09-26 devnull for(p=_threadprocs; p; p=p->next){
793 6dde87f8 2005-09-26 devnull fprint(2, "proc %p %s%s\n", (void*)p->osprocid, p->msg,
794 6dde87f8 2005-09-26 devnull p->sysproc ? " (sysproc)": "");
795 6dde87f8 2005-09-26 devnull for(t=p->allthreads.head; t; t=t->allnext){
796 6dde87f8 2005-09-26 devnull fprint(2, "\tthread %d %s: %s %s\n",
798 6dde87f8 2005-09-26 devnull t == p->thread ? "Running" :
799 6dde87f8 2005-09-26 devnull onrunqueue(p, t) ? "Ready" : "Sleeping",
800 6dde87f8 2005-09-26 devnull t->state, t->name);
803 6dde87f8 2005-09-26 devnull return 1;