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 df970459 2006-06-26 devnull static int onlist(_Threadlist*, _Thread*);
16 1544f909 2004-12-25 devnull static void addthreadinproc(Proc*, _Thread*);
17 1544f909 2004-12-25 devnull static void delthreadinproc(Proc*, _Thread*);
18 1544f909 2004-12-25 devnull static void contextswitch(Context *from, Context *to);
19 1fdb4a54 2005-01-23 devnull static void procscheduler(Proc*);
20 6dde87f8 2005-09-26 devnull static int threadinfo(void*, char*);
22 2c87dda8 2004-12-28 devnull static void
23 2c87dda8 2004-12-28 devnull _threaddebug(char *fmt, ...)
25 2c87dda8 2004-12-28 devnull va_list arg;
26 2c87dda8 2004-12-28 devnull char buf[128];
27 2c87dda8 2004-12-28 devnull _Thread *t;
29 af89fc10 2005-01-06 devnull static int fd = -1;
31 5b092791 2005-09-26 devnull if(_threaddebuglevel == 0)
34 af89fc10 2005-01-06 devnull if(fd < 0){
35 af89fc10 2005-01-06 devnull p = strrchr(argv0, '/');
39 af89fc10 2005-01-06 devnull p = argv0;
40 af89fc10 2005-01-06 devnull snprint(buf, sizeof buf, "/tmp/%s.tlog", p);
41 af89fc10 2005-01-06 devnull if((fd = create(buf, OWRITE, 0666)) < 0)
42 af89fc10 2005-01-06 devnull fd = open("/dev/null", OWRITE);
43 d93cc14e 2006-06-12 devnull if(fd >= 0 && fd != 2){
44 d93cc14e 2006-06-12 devnull dup(fd, 2);
45 d93cc14e 2006-06-12 devnull close(fd);
50 2c87dda8 2004-12-28 devnull va_start(arg, fmt);
51 2c87dda8 2004-12-28 devnull vsnprint(buf, sizeof buf, fmt, arg);
52 2c87dda8 2004-12-28 devnull va_end(arg);
53 2c87dda8 2004-12-28 devnull t = proc()->thread;
55 af89fc10 2005-01-06 devnull fprint(fd, "%d.%d: %s\n", getpid(), t->id, buf);
57 af89fc10 2005-01-06 devnull fprint(fd, "%d._: %s\n", getpid(), buf);
60 1544f909 2004-12-25 devnull static _Thread*
61 1544f909 2004-12-25 devnull getthreadnow(void)
63 1544f909 2004-12-25 devnull return proc()->thread;
65 1544f909 2004-12-25 devnull _Thread *(*threadnow)(void) = getthreadnow;
67 1544f909 2004-12-25 devnull static Proc*
68 1544f909 2004-12-25 devnull procalloc(void)
72 1544f909 2004-12-25 devnull p = malloc(sizeof *p);
73 73722a8b 2004-12-27 devnull if(p == nil)
74 73722a8b 2004-12-27 devnull sysfatal("procalloc malloc: %r");
75 1544f909 2004-12-25 devnull memset(p, 0, sizeof *p);
76 4dbefdd4 2004-12-27 devnull addproc(p);
77 1544f909 2004-12-25 devnull lock(&threadnproclock);
78 1544f909 2004-12-25 devnull threadnproc++;
79 1544f909 2004-12-25 devnull unlock(&threadnproclock);
80 1544f909 2004-12-25 devnull return p;
83 1544f909 2004-12-25 devnull static void
84 86502504 2005-01-18 devnull threadstart(uint y, uint x)
86 1544f909 2004-12-25 devnull _Thread *t;
89 04e0a9bb 2011-10-14 rsc //print("threadstart\n");
90 df1ee4e1 2008-06-01 rsc z = (ulong)x << 16; /* hide undefined 32-bit shift from 32-bit compilers */
91 86502504 2005-01-18 devnull z <<= 16;
93 86502504 2005-01-18 devnull t = (_Thread*)z;
95 fc567f47 2011-08-02 rsc //print("threadstart sp=%p arg=%p startfn=%p t=%p\n", &t, t, t->startfn, t->startarg);
96 1544f909 2004-12-25 devnull t->startfn(t->startarg);
97 cbeb0b26 2006-04-01 devnull /*print("threadexits %p\n", v); */
98 4dbefdd4 2004-12-27 devnull threadexits(nil);
99 cbeb0b26 2006-04-01 devnull /*print("not reacehd\n"); */
102 1544f909 2004-12-25 devnull static _Thread*
103 1544f909 2004-12-25 devnull threadalloc(void (*fn)(void*), void *arg, uint stack)
105 1544f909 2004-12-25 devnull _Thread *t;
106 1544f909 2004-12-25 devnull sigset_t zero;
107 86502504 2005-01-18 devnull uint x, y;
108 86502504 2005-01-18 devnull ulong z;
110 1544f909 2004-12-25 devnull /* allocate the task and stack together */
111 1544f909 2004-12-25 devnull t = malloc(sizeof *t+stack);
112 73722a8b 2004-12-27 devnull if(t == nil)
113 73722a8b 2004-12-27 devnull sysfatal("threadalloc malloc: %r");
114 1544f909 2004-12-25 devnull memset(t, 0, sizeof *t);
115 1544f909 2004-12-25 devnull t->stk = (uchar*)(t+1);
116 1544f909 2004-12-25 devnull t->stksize = stack;
117 1544f909 2004-12-25 devnull t->id = incref(&threadidref);
118 fc567f47 2011-08-02 rsc //print("fn=%p arg=%p\n", fn, arg);
119 1544f909 2004-12-25 devnull t->startfn = fn;
120 1544f909 2004-12-25 devnull t->startarg = arg;
121 fc567f47 2011-08-02 rsc //print("makecontext sp=%p t=%p startfn=%p\n", (char*)t->stk+t->stksize, t, t->startfn);
123 1544f909 2004-12-25 devnull /* do a reasonable initialization */
124 1544f909 2004-12-25 devnull memset(&t->context.uc, 0, sizeof t->context.uc);
125 1544f909 2004-12-25 devnull sigemptyset(&zero);
126 1544f909 2004-12-25 devnull sigprocmask(SIG_BLOCK, &zero, &t->context.uc.uc_sigmask);
127 fc567f47 2011-08-02 rsc //print("makecontext sp=%p t=%p startfn=%p\n", (char*)t->stk+t->stksize, t, t->startfn);
129 d54ead7f 2004-12-28 devnull /* must initialize with current context */
130 37cc72d3 2005-01-11 devnull if(getcontext(&t->context.uc) < 0)
131 37cc72d3 2005-01-11 devnull sysfatal("threadalloc getcontext: %r");
132 fc567f47 2011-08-02 rsc //print("makecontext sp=%p t=%p startfn=%p\n", (char*)t->stk+t->stksize, t, t->startfn);
134 1544f909 2004-12-25 devnull /* call makecontext to do the real work. */
135 1544f909 2004-12-25 devnull /* leave a few words open on both ends */
136 0ac67058 2008-12-23 rsc t->context.uc.uc_stack.ss_sp = (void*)(t->stk+8);
137 2c87dda8 2004-12-28 devnull t->context.uc.uc_stack.ss_size = t->stksize-64;
138 d08f4066 2005-07-27 devnull #if defined(__sun__) && !defined(__MAKECONTEXT_V2_SOURCE) /* sigh */
139 37cc72d3 2005-01-11 devnull /* can avoid this with __MAKECONTEXT_V2_SOURCE but only on SunOS 5.9 */
140 2b3b1e89 2005-01-14 devnull t->context.uc.uc_stack.ss_sp =
141 2b3b1e89 2005-01-14 devnull (char*)t->context.uc.uc_stack.ss_sp
142 2b3b1e89 2005-01-14 devnull +t->context.uc.uc_stack.ss_size;
145 86502504 2005-01-18 devnull * All this magic is because you have to pass makecontext a
146 86502504 2005-01-18 devnull * function that takes some number of word-sized variables,
147 86502504 2005-01-18 devnull * and on 64-bit machines pointers are bigger than words.
149 fc567f47 2011-08-02 rsc //print("makecontext sp=%p t=%p startfn=%p\n", (char*)t->stk+t->stksize, t, t->startfn);
150 86502504 2005-01-18 devnull z = (ulong)t;
152 86502504 2005-01-18 devnull z >>= 16; /* hide undefined 32-bit shift from 32-bit compilers */
153 86502504 2005-01-18 devnull x = z>>16;
154 2b604081 2005-05-07 devnull makecontext(&t->context.uc, (void(*)(void))threadstart, 2, y, x);
156 1544f909 2004-12-25 devnull return t;
159 1544f909 2004-12-25 devnull _Thread*
160 1544f909 2004-12-25 devnull _threadcreate(Proc *p, void (*fn)(void*), void *arg, uint stack)
162 1544f909 2004-12-25 devnull _Thread *t;
164 0829f75b 2009-08-18 rsc /* defend against bad C libraries */
165 0829f75b 2009-08-18 rsc if(stack < (256<<10))
166 0829f75b 2009-08-18 rsc stack = 256<<10;
168 1544f909 2004-12-25 devnull t = threadalloc(fn, arg, stack);
169 1544f909 2004-12-25 devnull t->proc = p;
170 1544f909 2004-12-25 devnull addthreadinproc(p, t);
171 1544f909 2004-12-25 devnull p->nthread++;
172 1544f909 2004-12-25 devnull _threadready(t);
173 1544f909 2004-12-25 devnull return t;
177 1544f909 2004-12-25 devnull threadcreate(void (*fn)(void*), void *arg, uint stack)
179 1544f909 2004-12-25 devnull _Thread *t;
181 1544f909 2004-12-25 devnull t = _threadcreate(proc(), fn, arg, stack);
182 1544f909 2004-12-25 devnull return t->id;
186 1544f909 2004-12-25 devnull proccreate(void (*fn)(void*), void *arg, uint stack)
189 1544f909 2004-12-25 devnull _Thread *t;
190 1544f909 2004-12-25 devnull Proc *p;
192 1544f909 2004-12-25 devnull p = procalloc();
193 1544f909 2004-12-25 devnull t = _threadcreate(p, fn, arg, stack);
194 803292c6 2005-01-18 devnull id = t->id; /* t might be freed after _procstart */
195 1fdb4a54 2005-01-23 devnull _procstart(p, procscheduler);
196 803292c6 2005-01-18 devnull return id;
200 1544f909 2004-12-25 devnull _threadswitch(void)
202 1544f909 2004-12-25 devnull Proc *p;
204 c8b6342d 2005-01-13 devnull needstack(0);
205 1544f909 2004-12-25 devnull p = proc();
206 cbeb0b26 2006-04-01 devnull /*print("threadswtch %p\n", p); */
207 1544f909 2004-12-25 devnull contextswitch(&p->thread->context, &p->schedcontext);
211 1544f909 2004-12-25 devnull _threadready(_Thread *t)
213 1544f909 2004-12-25 devnull Proc *p;
215 1544f909 2004-12-25 devnull p = t->proc;
216 1544f909 2004-12-25 devnull lock(&p->lock);
217 a0a331aa 2005-01-06 devnull p->runrend.l = &p->lock;
218 1544f909 2004-12-25 devnull addthread(&p->runqueue, t);
219 cbeb0b26 2006-04-01 devnull /*print("%d wake for job %d->%d\n", time(0), getpid(), p->osprocid); */
220 8cbd854a 2004-12-27 devnull if(p != proc())
221 a0a331aa 2005-01-06 devnull _procwakeupandunlock(&p->runrend);
223 a0a331aa 2005-01-06 devnull unlock(&p->lock);
227 615e0f9f 2006-02-07 devnull threadidle(void)
230 615e0f9f 2006-02-07 devnull Proc *p;
232 615e0f9f 2006-02-07 devnull p = proc();
233 615e0f9f 2006-02-07 devnull n = p->nswitch;
234 615e0f9f 2006-02-07 devnull lock(&p->lock);
235 615e0f9f 2006-02-07 devnull p->runrend.l = &p->lock;
236 615e0f9f 2006-02-07 devnull addthread(&p->idlequeue, p->thread);
237 615e0f9f 2006-02-07 devnull unlock(&p->lock);
238 615e0f9f 2006-02-07 devnull _threadswitch();
239 615e0f9f 2006-02-07 devnull return p->nswitch - n;
243 1544f909 2004-12-25 devnull threadyield(void)
246 8cbd854a 2004-12-27 devnull Proc *p;
248 8cbd854a 2004-12-27 devnull p = proc();
249 8cbd854a 2004-12-27 devnull n = p->nswitch;
250 8cbd854a 2004-12-27 devnull _threadready(p->thread);
251 1544f909 2004-12-25 devnull _threadswitch();
252 8cbd854a 2004-12-27 devnull return p->nswitch - n;
256 1544f909 2004-12-25 devnull threadexits(char *msg)
258 1544f909 2004-12-25 devnull Proc *p;
260 1544f909 2004-12-25 devnull p = proc();
261 4dbefdd4 2004-12-27 devnull if(msg == nil)
262 4dbefdd4 2004-12-27 devnull msg = "";
263 1544f909 2004-12-25 devnull utfecpy(p->msg, p->msg+sizeof p->msg, msg);
264 4dbefdd4 2004-12-27 devnull proc()->thread->exiting = 1;
265 4dbefdd4 2004-12-27 devnull _threadswitch();
269 df970459 2006-06-26 devnull threadpin(void)
271 df970459 2006-06-26 devnull Proc *p;
273 df970459 2006-06-26 devnull p = proc();
274 df970459 2006-06-26 devnull if(p->pinthread){
275 df970459 2006-06-26 devnull fprint(2, "already pinning a thread - %p %p\n", p->pinthread, p->thread);
276 df970459 2006-06-26 devnull assert(0);
278 df970459 2006-06-26 devnull p->pinthread = p->thread;
282 df970459 2006-06-26 devnull threadunpin(void)
284 df970459 2006-06-26 devnull Proc *p;
286 df970459 2006-06-26 devnull p = proc();
287 df970459 2006-06-26 devnull if(p->pinthread != p->thread){
288 df970459 2006-06-26 devnull fprint(2, "wrong pinthread - %p %p\n", p->pinthread, p->thread);
289 df970459 2006-06-26 devnull assert(0);
291 df970459 2006-06-26 devnull p->pinthread = nil;
295 d19769ae 2008-10-26 rsc threadsysfatal(char *fmt, va_list arg)
297 69d1259c 2008-07-20 devnull char buf[256];
299 69d1259c 2008-07-20 devnull vseprint(buf, buf+sizeof(buf), fmt, arg);
300 69d1259c 2008-07-20 devnull __fixargv0();
301 69d1259c 2008-07-20 devnull fprint(2, "%s: %s\n", argv0 ? argv0 : "<prog>", buf);
302 69d1259c 2008-07-20 devnull threadexitsall(buf);
305 1544f909 2004-12-25 devnull static void
306 1544f909 2004-12-25 devnull contextswitch(Context *from, Context *to)
308 1544f909 2004-12-25 devnull if(swapcontext(&from->uc, &to->uc) < 0){
309 1544f909 2004-12-25 devnull fprint(2, "swapcontext failed: %r\n");
310 1544f909 2004-12-25 devnull assert(0);
314 1544f909 2004-12-25 devnull static void
315 1fdb4a54 2005-01-23 devnull procscheduler(Proc *p)
317 1544f909 2004-12-25 devnull _Thread *t;
319 1544f909 2004-12-25 devnull setproc(p);
320 2c87dda8 2004-12-28 devnull _threaddebug("scheduler enter");
321 04e0a9bb 2011-10-14 rsc //print("s %p\n", p);
322 1544f909 2004-12-25 devnull lock(&p->lock);
323 1544f909 2004-12-25 devnull for(;;){
324 df970459 2006-06-26 devnull if((t = p->pinthread) != nil){
325 df970459 2006-06-26 devnull while(!onlist(&p->runqueue, t)){
326 df970459 2006-06-26 devnull p->runrend.l = &p->lock;
327 df970459 2006-06-26 devnull _threaddebug("scheduler sleep (pin)");
328 df970459 2006-06-26 devnull _procsleep(&p->runrend);
329 df970459 2006-06-26 devnull _threaddebug("scheduler wake (pin)");
332 1544f909 2004-12-25 devnull while((t = p->runqueue.head) == nil){
333 1544f909 2004-12-25 devnull if(p->nthread == 0)
334 1544f909 2004-12-25 devnull goto Out;
335 615e0f9f 2006-02-07 devnull if((t = p->idlequeue.head) != nil){
337 615e0f9f 2006-02-07 devnull * Run all the idling threads once.
339 615e0f9f 2006-02-07 devnull while((t = p->idlequeue.head) != nil){
340 615e0f9f 2006-02-07 devnull delthread(&p->idlequeue, t);
341 615e0f9f 2006-02-07 devnull addthread(&p->runqueue, t);
343 615e0f9f 2006-02-07 devnull continue;
345 1544f909 2004-12-25 devnull p->runrend.l = &p->lock;
346 2c87dda8 2004-12-28 devnull _threaddebug("scheduler sleep");
347 1544f909 2004-12-25 devnull _procsleep(&p->runrend);
348 2c87dda8 2004-12-28 devnull _threaddebug("scheduler wake");
350 df970459 2006-06-26 devnull if(p->pinthread && p->pinthread != t)
351 df970459 2006-06-26 devnull fprint(2, "p->pinthread %p t %p\n", p->pinthread, t);
352 df970459 2006-06-26 devnull assert(p->pinthread == nil || p->pinthread == t);
353 1544f909 2004-12-25 devnull delthread(&p->runqueue, t);
354 1544f909 2004-12-25 devnull unlock(&p->lock);
355 1544f909 2004-12-25 devnull p->thread = t;
356 8cbd854a 2004-12-27 devnull p->nswitch++;
357 2c87dda8 2004-12-28 devnull _threaddebug("run %d (%s)", t->id, t->name);
358 04e0a9bb 2011-10-14 rsc //print("run %p %p %p %p\n", t, *(uintptr*)(t->context.uc.mc.sp), t->context.uc.mc.di, t->context.uc.mc.si);
359 1544f909 2004-12-25 devnull contextswitch(&p->schedcontext, &t->context);
360 cbeb0b26 2006-04-01 devnull /*print("back in scheduler\n"); */
361 1544f909 2004-12-25 devnull p->thread = nil;
362 1544f909 2004-12-25 devnull lock(&p->lock);
363 1544f909 2004-12-25 devnull if(t->exiting){
364 1544f909 2004-12-25 devnull delthreadinproc(p, t);
365 1544f909 2004-12-25 devnull p->nthread--;
366 cbeb0b26 2006-04-01 devnull /*print("nthread %d\n", p->nthread); */
367 1544f909 2004-12-25 devnull free(t);
372 2c87dda8 2004-12-28 devnull _threaddebug("scheduler exit");
373 8ee6ad4d 2005-02-14 devnull if(p->mainproc){
375 8ee6ad4d 2005-02-14 devnull * Stupid bug - on Linux 2.6 and maybe elsewhere,
376 8ee6ad4d 2005-02-14 devnull * if the main thread exits then the others keep running
377 8ee6ad4d 2005-02-14 devnull * but the process shows up as a zombie in ps and is not
378 8ee6ad4d 2005-02-14 devnull * attachable with ptrace. We'll just sit around pretending
379 8ee6ad4d 2005-02-14 devnull * to be a system proc instead of exiting.
381 8ee6ad4d 2005-02-14 devnull _threaddaemonize();
382 8ee6ad4d 2005-02-14 devnull lock(&threadnproclock);
383 8ee6ad4d 2005-02-14 devnull if(++threadnsysproc == threadnproc)
384 8ee6ad4d 2005-02-14 devnull threadexitsall(p->msg);
385 8ee6ad4d 2005-02-14 devnull p->sysproc = 1;
386 8ee6ad4d 2005-02-14 devnull unlock(&threadnproclock);
388 8ee6ad4d 2005-02-14 devnull sleep(1000);
391 4dbefdd4 2004-12-27 devnull delproc(p);
392 1544f909 2004-12-25 devnull lock(&threadnproclock);
393 1544f909 2004-12-25 devnull if(p->sysproc)
394 1544f909 2004-12-25 devnull --threadnsysproc;
395 1544f909 2004-12-25 devnull if(--threadnproc == threadnsysproc)
396 4dbefdd4 2004-12-27 devnull threadexitsall(p->msg);
397 1544f909 2004-12-25 devnull unlock(&threadnproclock);
398 1544f909 2004-12-25 devnull unlock(&p->lock);
399 d93cc14e 2006-06-12 devnull _threadsetproc(nil);
400 1544f909 2004-12-25 devnull free(p);
404 1544f909 2004-12-25 devnull _threadsetsysproc(void)
406 1544f909 2004-12-25 devnull lock(&threadnproclock);
407 1544f909 2004-12-25 devnull if(++threadnsysproc == threadnproc)
408 8ee6ad4d 2005-02-14 devnull threadexitsall(nil);
409 1544f909 2004-12-25 devnull unlock(&threadnproclock);
410 1544f909 2004-12-25 devnull proc()->sysproc = 1;
414 4dbefdd4 2004-12-27 devnull procdata(void)
416 4dbefdd4 2004-12-27 devnull return &proc()->udata;
420 8dd8a81f 2005-03-18 devnull threaddata(void)
422 8dd8a81f 2005-03-18 devnull return &proc()->thread->udata;
425 4dbefdd4 2004-12-27 devnull extern Jmp *(*_notejmpbuf)(void);
426 4dbefdd4 2004-12-27 devnull static Jmp*
427 4dbefdd4 2004-12-27 devnull threadnotejmp(void)
429 4dbefdd4 2004-12-27 devnull return &proc()->sigjmp;
433 1544f909 2004-12-25 devnull * debugging
436 1544f909 2004-12-25 devnull threadsetname(char *fmt, ...)
438 1544f909 2004-12-25 devnull va_list arg;
439 1544f909 2004-12-25 devnull _Thread *t;
441 1544f909 2004-12-25 devnull t = proc()->thread;
442 1544f909 2004-12-25 devnull va_start(arg, fmt);
443 1544f909 2004-12-25 devnull vsnprint(t->name, sizeof t->name, fmt, arg);
444 1544f909 2004-12-25 devnull va_end(arg);
448 515b6e5b 2005-01-18 devnull threadgetname(void)
450 515b6e5b 2005-01-18 devnull return proc()->thread->name;
454 1544f909 2004-12-25 devnull threadsetstate(char *fmt, ...)
456 1544f909 2004-12-25 devnull va_list arg;
457 1544f909 2004-12-25 devnull _Thread *t;
459 1544f909 2004-12-25 devnull t = proc()->thread;
460 1544f909 2004-12-25 devnull va_start(arg, fmt);
461 1544f909 2004-12-25 devnull vsnprint(t->state, sizeof t->name, fmt, arg);
462 1544f909 2004-12-25 devnull va_end(arg);
466 9eda38e5 2006-02-05 devnull threadid(void)
468 9eda38e5 2006-02-05 devnull _Thread *t;
470 9eda38e5 2006-02-05 devnull t = proc()->thread;
471 9eda38e5 2006-02-05 devnull return t->id;
475 c8b6342d 2005-01-13 devnull needstack(int n)
477 c8b6342d 2005-01-13 devnull _Thread *t;
479 c8b6342d 2005-01-13 devnull t = proc()->thread;
481 c8b6342d 2005-01-13 devnull if((char*)&t <= (char*)t->stk
482 c8b6342d 2005-01-13 devnull || (char*)&t - (char*)t->stk < 256+n){
483 1fdb4a54 2005-01-23 devnull fprint(2, "thread stack overflow: &t=%p tstk=%p n=%d\n", &t, t->stk, 256+n);
484 c8b6342d 2005-01-13 devnull abort();
489 faf1fb6c 2008-07-09 rsc singlethreaded(void)
491 faf1fb6c 2008-07-09 rsc return threadnproc == 1 && _threadprocs->nthread == 1;
495 1544f909 2004-12-25 devnull * locking
497 1544f909 2004-12-25 devnull static int
498 1544f909 2004-12-25 devnull threadqlock(QLock *l, int block, ulong pc)
500 cbeb0b26 2006-04-01 devnull /*print("threadqlock %p\n", l); */
501 1544f909 2004-12-25 devnull lock(&l->l);
502 1544f909 2004-12-25 devnull if(l->owner == nil){
503 1544f909 2004-12-25 devnull l->owner = (*threadnow)();
504 cbeb0b26 2006-04-01 devnull /*print("qlock %p @%#x by %p\n", l, pc, l->owner); */
505 1544f909 2004-12-25 devnull unlock(&l->l);
506 1544f909 2004-12-25 devnull return 1;
508 1544f909 2004-12-25 devnull if(!block){
509 1544f909 2004-12-25 devnull unlock(&l->l);
510 1544f909 2004-12-25 devnull return 0;
513 faf1fb6c 2008-07-09 rsc if(singlethreaded()){
514 faf1fb6c 2008-07-09 rsc fprint(2, "qlock deadlock\n");
518 cbeb0b26 2006-04-01 devnull /*print("qsleep %p @%#x by %p\n", l, pc, (*threadnow)()); */
519 1544f909 2004-12-25 devnull addthread(&l->waiting, (*threadnow)());
520 1544f909 2004-12-25 devnull unlock(&l->l);
522 1544f909 2004-12-25 devnull _threadswitch();
524 1544f909 2004-12-25 devnull if(l->owner != (*threadnow)()){
525 15cd8255 2005-01-04 devnull fprint(2, "%s: qlock pc=0x%lux owner=%p self=%p oops\n",
526 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
527 1544f909 2004-12-25 devnull abort();
529 cbeb0b26 2006-04-01 devnull /*print("qlock wakeup %p @%#x by %p\n", l, pc, (*threadnow)()); */
530 1544f909 2004-12-25 devnull return 1;
533 1544f909 2004-12-25 devnull static void
534 1544f909 2004-12-25 devnull threadqunlock(QLock *l, ulong pc)
536 80b8842f 2005-02-15 devnull _Thread *ready;
538 1544f909 2004-12-25 devnull lock(&l->l);
539 cbeb0b26 2006-04-01 devnull /*print("qlock unlock %p @%#x by %p (owner %p)\n", l, pc, (*threadnow)(), l->owner); */
540 1aa9c533 2005-01-07 devnull if(l->owner == 0){
541 15cd8255 2005-01-04 devnull fprint(2, "%s: qunlock pc=0x%lux owner=%p self=%p oops\n",
542 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
543 1aa9c533 2005-01-07 devnull abort();
545 80b8842f 2005-02-15 devnull if((l->owner = ready = l->waiting.head) != nil)
546 1544f909 2004-12-25 devnull delthread(&l->waiting, l->owner);
548 80b8842f 2005-02-15 devnull * N.B. Cannot call _threadready() before unlocking l->l,
549 80b8842f 2005-02-15 devnull * because the thread we are readying might:
550 80b8842f 2005-02-15 devnull * - be in another proc
551 80b8842f 2005-02-15 devnull * - start running immediately
552 80b8842f 2005-02-15 devnull * - and free l before we get a chance to run again
554 1544f909 2004-12-25 devnull unlock(&l->l);
555 80b8842f 2005-02-15 devnull if(ready)
556 80b8842f 2005-02-15 devnull _threadready(l->owner);
559 1544f909 2004-12-25 devnull static int
560 1544f909 2004-12-25 devnull threadrlock(RWLock *l, int block, ulong pc)
562 1544f909 2004-12-25 devnull USED(pc);
564 1544f909 2004-12-25 devnull lock(&l->l);
565 1544f909 2004-12-25 devnull if(l->writer == nil && l->wwaiting.head == nil){
566 1544f909 2004-12-25 devnull l->readers++;
567 1544f909 2004-12-25 devnull unlock(&l->l);
568 1544f909 2004-12-25 devnull return 1;
570 1544f909 2004-12-25 devnull if(!block){
571 1544f909 2004-12-25 devnull unlock(&l->l);
572 1544f909 2004-12-25 devnull return 0;
574 faf1fb6c 2008-07-09 rsc if(singlethreaded()){
575 faf1fb6c 2008-07-09 rsc fprint(2, "rlock deadlock\n");
578 1544f909 2004-12-25 devnull addthread(&l->rwaiting, (*threadnow)());
579 1544f909 2004-12-25 devnull unlock(&l->l);
580 1544f909 2004-12-25 devnull _threadswitch();
581 1544f909 2004-12-25 devnull return 1;
584 1544f909 2004-12-25 devnull static int
585 1544f909 2004-12-25 devnull threadwlock(RWLock *l, int block, ulong pc)
587 1544f909 2004-12-25 devnull USED(pc);
589 1544f909 2004-12-25 devnull lock(&l->l);
590 1544f909 2004-12-25 devnull if(l->writer == nil && l->readers == 0){
591 1544f909 2004-12-25 devnull l->writer = (*threadnow)();
592 1544f909 2004-12-25 devnull unlock(&l->l);
593 1544f909 2004-12-25 devnull return 1;
595 1544f909 2004-12-25 devnull if(!block){
596 1544f909 2004-12-25 devnull unlock(&l->l);
597 1544f909 2004-12-25 devnull return 0;
599 faf1fb6c 2008-07-09 rsc if(singlethreaded()){
600 faf1fb6c 2008-07-09 rsc fprint(2, "wlock deadlock\n");
603 1544f909 2004-12-25 devnull addthread(&l->wwaiting, (*threadnow)());
604 1544f909 2004-12-25 devnull unlock(&l->l);
605 1544f909 2004-12-25 devnull _threadswitch();
606 1544f909 2004-12-25 devnull return 1;
609 1544f909 2004-12-25 devnull static void
610 1544f909 2004-12-25 devnull threadrunlock(RWLock *l, ulong pc)
612 1544f909 2004-12-25 devnull _Thread *t;
614 1544f909 2004-12-25 devnull USED(pc);
615 80b8842f 2005-02-15 devnull t = nil;
616 1544f909 2004-12-25 devnull lock(&l->l);
617 1544f909 2004-12-25 devnull --l->readers;
618 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
619 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
620 1544f909 2004-12-25 devnull l->writer = t;
622 1544f909 2004-12-25 devnull unlock(&l->l);
624 80b8842f 2005-02-15 devnull _threadready(t);
628 1544f909 2004-12-25 devnull static void
629 1544f909 2004-12-25 devnull threadwunlock(RWLock *l, ulong pc)
631 1544f909 2004-12-25 devnull _Thread *t;
633 1544f909 2004-12-25 devnull USED(pc);
634 1544f909 2004-12-25 devnull lock(&l->l);
635 1544f909 2004-12-25 devnull l->writer = nil;
636 1544f909 2004-12-25 devnull assert(l->readers == 0);
637 1544f909 2004-12-25 devnull while((t = l->rwaiting.head) != nil){
638 1544f909 2004-12-25 devnull delthread(&l->rwaiting, t);
639 1544f909 2004-12-25 devnull l->readers++;
640 1544f909 2004-12-25 devnull _threadready(t);
642 80b8842f 2005-02-15 devnull t = nil;
643 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
644 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
645 1544f909 2004-12-25 devnull l->writer = t;
647 1544f909 2004-12-25 devnull unlock(&l->l);
649 80b8842f 2005-02-15 devnull _threadready(t);
653 1544f909 2004-12-25 devnull * sleep and wakeup
655 1544f909 2004-12-25 devnull static void
656 1544f909 2004-12-25 devnull threadrsleep(Rendez *r, ulong pc)
658 faf1fb6c 2008-07-09 rsc if(singlethreaded()){
659 faf1fb6c 2008-07-09 rsc fprint(2, "rsleep deadlock\n");
662 1544f909 2004-12-25 devnull addthread(&r->waiting, proc()->thread);
663 1544f909 2004-12-25 devnull qunlock(r->l);
664 1544f909 2004-12-25 devnull _threadswitch();
665 1544f909 2004-12-25 devnull qlock(r->l);
668 1544f909 2004-12-25 devnull static int
669 1544f909 2004-12-25 devnull threadrwakeup(Rendez *r, int all, ulong pc)
672 1544f909 2004-12-25 devnull _Thread *t;
674 1544f909 2004-12-25 devnull for(i=0;; i++){
675 1544f909 2004-12-25 devnull if(i==1 && !all)
677 1544f909 2004-12-25 devnull if((t = r->waiting.head) == nil)
679 1544f909 2004-12-25 devnull delthread(&r->waiting, t);
680 1544f909 2004-12-25 devnull _threadready(t);
682 1544f909 2004-12-25 devnull return i;
686 4dbefdd4 2004-12-27 devnull * startup
689 4dbefdd4 2004-12-27 devnull static int threadargc;
690 4dbefdd4 2004-12-27 devnull static char **threadargv;
691 4dbefdd4 2004-12-27 devnull int mainstacksize;
692 d93cc14e 2006-06-12 devnull extern int _p9usepwlibrary; /* getgrgid etc. smash the stack - tell _p9dir just say no */
693 4dbefdd4 2004-12-27 devnull static void
694 4dbefdd4 2004-12-27 devnull threadmainstart(void *v)
696 4dbefdd4 2004-12-27 devnull USED(v);
699 0b561695 2005-01-16 devnull * N.B. This call to proc() is a program's first call (indirectly) to a
700 0b561695 2005-01-16 devnull * pthreads function while executing on a non-pthreads-allocated
701 0b561695 2005-01-16 devnull * stack. If the pthreads implementation is using the stack pointer
702 0b561695 2005-01-16 devnull * to locate the per-thread data, then this call will blow up.
703 0b561695 2005-01-16 devnull * This means the pthread implementation is not suitable for
704 0b561695 2005-01-16 devnull * running under libthread. Time to write your own. Sorry.
706 d93cc14e 2006-06-12 devnull _p9usepwlibrary = 0;
707 4dbefdd4 2004-12-27 devnull threadmainproc = proc();
708 4dbefdd4 2004-12-27 devnull threadmain(threadargc, threadargv);
711 d19769ae 2008-10-26 rsc extern void (*_sysfatal)(char*, va_list);
714 4dbefdd4 2004-12-27 devnull main(int argc, char **argv)
716 4dbefdd4 2004-12-27 devnull Proc *p;
718 1d2533d0 2004-12-28 devnull argv0 = argv[0];
720 615e0f9f 2006-02-07 devnull if(getenv("NOLIBTHREADDAEMONIZE") == nil)
721 615e0f9f 2006-02-07 devnull _threadsetupdaemonize();
723 4dbefdd4 2004-12-27 devnull threadargc = argc;
724 4dbefdd4 2004-12-27 devnull threadargv = argv;
727 4dbefdd4 2004-12-27 devnull * Install locking routines into C library.
729 4dbefdd4 2004-12-27 devnull _lock = _threadlock;
730 4dbefdd4 2004-12-27 devnull _unlock = _threadunlock;
731 4dbefdd4 2004-12-27 devnull _qlock = threadqlock;
732 4dbefdd4 2004-12-27 devnull _qunlock = threadqunlock;
733 4dbefdd4 2004-12-27 devnull _rlock = threadrlock;
734 4dbefdd4 2004-12-27 devnull _runlock = threadrunlock;
735 4dbefdd4 2004-12-27 devnull _wlock = threadwlock;
736 4dbefdd4 2004-12-27 devnull _wunlock = threadwunlock;
737 4dbefdd4 2004-12-27 devnull _rsleep = threadrsleep;
738 4dbefdd4 2004-12-27 devnull _rwakeup = threadrwakeup;
739 4dbefdd4 2004-12-27 devnull _notejmpbuf = threadnotejmp;
740 df970459 2006-06-26 devnull _pin = threadpin;
741 df970459 2006-06-26 devnull _unpin = threadunpin;
742 69d1259c 2008-07-20 devnull _sysfatal = threadsysfatal;
744 4dbefdd4 2004-12-27 devnull _pthreadinit();
745 4dbefdd4 2004-12-27 devnull p = procalloc();
746 8ee6ad4d 2005-02-14 devnull p->mainproc = 1;
747 4dbefdd4 2004-12-27 devnull _threadsetproc(p);
748 4dbefdd4 2004-12-27 devnull if(mainstacksize == 0)
749 a0e8d02d 2005-01-23 devnull mainstacksize = 256*1024;
750 af6e459f 2006-04-21 devnull atnotify(threadinfo, 1);
751 4dbefdd4 2004-12-27 devnull _threadcreate(p, threadmainstart, nil, mainstacksize);
752 1fdb4a54 2005-01-23 devnull procscheduler(p);
753 d7688cce 2005-02-16 devnull sysfatal("procscheduler returned in threadmain!");
754 8ee6ad4d 2005-02-14 devnull /* does not return */
755 15cd8255 2005-01-04 devnull return 0;
759 1544f909 2004-12-25 devnull * hooray for linked lists
761 1544f909 2004-12-25 devnull static void
762 1544f909 2004-12-25 devnull addthread(_Threadlist *l, _Thread *t)
764 1544f909 2004-12-25 devnull if(l->tail){
765 1544f909 2004-12-25 devnull l->tail->next = t;
766 1544f909 2004-12-25 devnull t->prev = l->tail;
768 1544f909 2004-12-25 devnull l->head = t;
769 1544f909 2004-12-25 devnull t->prev = nil;
771 1544f909 2004-12-25 devnull l->tail = t;
772 1544f909 2004-12-25 devnull t->next = nil;
775 1544f909 2004-12-25 devnull static void
776 1544f909 2004-12-25 devnull delthread(_Threadlist *l, _Thread *t)
778 1544f909 2004-12-25 devnull if(t->prev)
779 1544f909 2004-12-25 devnull t->prev->next = t->next;
781 1544f909 2004-12-25 devnull l->head = t->next;
782 1544f909 2004-12-25 devnull if(t->next)
783 1544f909 2004-12-25 devnull t->next->prev = t->prev;
785 1544f909 2004-12-25 devnull l->tail = t->prev;
788 df970459 2006-06-26 devnull /* inefficient but rarely used */
789 df970459 2006-06-26 devnull static int
790 df970459 2006-06-26 devnull onlist(_Threadlist *l, _Thread *t)
792 df970459 2006-06-26 devnull _Thread *tt;
794 df970459 2006-06-26 devnull for(tt = l->head; tt; tt=tt->next)
795 df970459 2006-06-26 devnull if(tt == t)
796 df970459 2006-06-26 devnull return 1;
797 df970459 2006-06-26 devnull return 0;
800 1544f909 2004-12-25 devnull static void
801 1544f909 2004-12-25 devnull addthreadinproc(Proc *p, _Thread *t)
803 1544f909 2004-12-25 devnull _Threadlist *l;
805 1544f909 2004-12-25 devnull l = &p->allthreads;
806 1544f909 2004-12-25 devnull if(l->tail){
807 1544f909 2004-12-25 devnull l->tail->allnext = t;
808 1544f909 2004-12-25 devnull t->allprev = l->tail;
810 1544f909 2004-12-25 devnull l->head = t;
811 1544f909 2004-12-25 devnull t->allprev = nil;
813 1544f909 2004-12-25 devnull l->tail = t;
814 1544f909 2004-12-25 devnull t->allnext = nil;
817 1544f909 2004-12-25 devnull static void
818 1544f909 2004-12-25 devnull delthreadinproc(Proc *p, _Thread *t)
820 1544f909 2004-12-25 devnull _Threadlist *l;
822 1544f909 2004-12-25 devnull l = &p->allthreads;
823 1544f909 2004-12-25 devnull if(t->allprev)
824 1544f909 2004-12-25 devnull t->allprev->allnext = t->allnext;
826 1544f909 2004-12-25 devnull l->head = t->allnext;
827 1544f909 2004-12-25 devnull if(t->allnext)
828 1544f909 2004-12-25 devnull t->allnext->allprev = t->allprev;
830 1544f909 2004-12-25 devnull l->tail = t->allprev;
833 4dbefdd4 2004-12-27 devnull Proc *_threadprocs;
834 4dbefdd4 2004-12-27 devnull Lock _threadprocslock;
835 4dbefdd4 2004-12-27 devnull static Proc *_threadprocstail;
837 1544f909 2004-12-25 devnull static void
838 4dbefdd4 2004-12-27 devnull addproc(Proc *p)
840 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
841 4dbefdd4 2004-12-27 devnull if(_threadprocstail){
842 4dbefdd4 2004-12-27 devnull _threadprocstail->next = p;
843 4dbefdd4 2004-12-27 devnull p->prev = _threadprocstail;
845 4dbefdd4 2004-12-27 devnull _threadprocs = p;
846 4dbefdd4 2004-12-27 devnull p->prev = nil;
848 4dbefdd4 2004-12-27 devnull _threadprocstail = p;
849 4dbefdd4 2004-12-27 devnull p->next = nil;
850 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
853 4dbefdd4 2004-12-27 devnull static void
854 4dbefdd4 2004-12-27 devnull delproc(Proc *p)
856 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
857 4dbefdd4 2004-12-27 devnull if(p->prev)
858 4dbefdd4 2004-12-27 devnull p->prev->next = p->next;
860 4dbefdd4 2004-12-27 devnull _threadprocs = p->next;
861 4dbefdd4 2004-12-27 devnull if(p->next)
862 4dbefdd4 2004-12-27 devnull p->next->prev = p->prev;
864 4dbefdd4 2004-12-27 devnull _threadprocstail = p->prev;
865 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
869 15cd8255 2005-01-04 devnull * notify - for now just use the usual mechanisms
872 15cd8255 2005-01-04 devnull threadnotify(int (*f)(void*, char*), int in)
874 15cd8255 2005-01-04 devnull atnotify(f, in);
877 6dde87f8 2005-09-26 devnull static int
878 6dde87f8 2005-09-26 devnull onrunqueue(Proc *p, _Thread *t)
880 6dde87f8 2005-09-26 devnull _Thread *tt;
882 6dde87f8 2005-09-26 devnull for(tt=p->runqueue.head; tt; tt=tt->next)
883 6dde87f8 2005-09-26 devnull if(tt == t)
884 6dde87f8 2005-09-26 devnull return 1;
885 6dde87f8 2005-09-26 devnull return 0;
889 6dde87f8 2005-09-26 devnull * print state - called from SIGINFO
891 6dde87f8 2005-09-26 devnull static int
892 6dde87f8 2005-09-26 devnull threadinfo(void *v, char *s)
894 6dde87f8 2005-09-26 devnull Proc *p;
895 6dde87f8 2005-09-26 devnull _Thread *t;
897 6dde87f8 2005-09-26 devnull if(strcmp(s, "quit") != 0 && strcmp(s, "sys: status request") != 0)
898 6dde87f8 2005-09-26 devnull return 0;
900 6dde87f8 2005-09-26 devnull for(p=_threadprocs; p; p=p->next){
901 6dde87f8 2005-09-26 devnull fprint(2, "proc %p %s%s\n", (void*)p->osprocid, p->msg,
902 6dde87f8 2005-09-26 devnull p->sysproc ? " (sysproc)": "");
903 6dde87f8 2005-09-26 devnull for(t=p->allthreads.head; t; t=t->allnext){
904 6dde87f8 2005-09-26 devnull fprint(2, "\tthread %d %s: %s %s\n",
906 6dde87f8 2005-09-26 devnull t == p->thread ? "Running" :
907 6dde87f8 2005-09-26 devnull onrunqueue(p, t) ? "Ready" : "Sleeping",
908 6dde87f8 2005-09-26 devnull t->state, t->name);
911 6dde87f8 2005-09-26 devnull return 1;