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 0158bcee 2020-01-12 rsc static void procmain(Proc*);
20 1fdb4a54 2005-01-23 devnull static void procscheduler(Proc*);
21 6dde87f8 2005-09-26 devnull static int threadinfo(void*, char*);
23 2c87dda8 2004-12-28 devnull static void
24 2c87dda8 2004-12-28 devnull _threaddebug(char *fmt, ...)
26 2c87dda8 2004-12-28 devnull va_list arg;
27 2c87dda8 2004-12-28 devnull char buf[128];
28 2c87dda8 2004-12-28 devnull _Thread *t;
30 af89fc10 2005-01-06 devnull static int fd = -1;
32 5b092791 2005-09-26 devnull if(_threaddebuglevel == 0)
35 af89fc10 2005-01-06 devnull if(fd < 0){
36 af89fc10 2005-01-06 devnull p = strrchr(argv0, '/');
40 af89fc10 2005-01-06 devnull p = argv0;
41 af89fc10 2005-01-06 devnull snprint(buf, sizeof buf, "/tmp/%s.tlog", p);
42 af89fc10 2005-01-06 devnull if((fd = create(buf, OWRITE, 0666)) < 0)
43 af89fc10 2005-01-06 devnull fd = open("/dev/null", OWRITE);
44 d93cc14e 2006-06-12 devnull if(fd >= 0 && fd != 2){
45 d93cc14e 2006-06-12 devnull dup(fd, 2);
46 d93cc14e 2006-06-12 devnull close(fd);
51 2c87dda8 2004-12-28 devnull va_start(arg, fmt);
52 2c87dda8 2004-12-28 devnull vsnprint(buf, sizeof buf, fmt, arg);
53 2c87dda8 2004-12-28 devnull va_end(arg);
54 2c87dda8 2004-12-28 devnull t = proc()->thread;
56 af89fc10 2005-01-06 devnull fprint(fd, "%d.%d: %s\n", getpid(), t->id, buf);
58 af89fc10 2005-01-06 devnull fprint(fd, "%d._: %s\n", getpid(), buf);
61 1544f909 2004-12-25 devnull static _Thread*
62 1544f909 2004-12-25 devnull getthreadnow(void)
64 1544f909 2004-12-25 devnull return proc()->thread;
66 1544f909 2004-12-25 devnull _Thread *(*threadnow)(void) = getthreadnow;
68 1544f909 2004-12-25 devnull static Proc*
69 1544f909 2004-12-25 devnull procalloc(void)
73 1544f909 2004-12-25 devnull p = malloc(sizeof *p);
74 73722a8b 2004-12-27 devnull if(p == nil)
75 73722a8b 2004-12-27 devnull sysfatal("procalloc malloc: %r");
76 1544f909 2004-12-25 devnull memset(p, 0, sizeof *p);
77 4dbefdd4 2004-12-27 devnull addproc(p);
78 1544f909 2004-12-25 devnull lock(&threadnproclock);
79 1544f909 2004-12-25 devnull threadnproc++;
80 1544f909 2004-12-25 devnull unlock(&threadnproclock);
81 1544f909 2004-12-25 devnull return p;
84 1544f909 2004-12-25 devnull static void
85 86502504 2005-01-18 devnull threadstart(uint y, uint x)
87 1544f909 2004-12-25 devnull _Thread *t;
90 04e0a9bb 2011-10-14 rsc //print("threadstart\n");
91 df1ee4e1 2008-06-01 rsc z = (ulong)x << 16; /* hide undefined 32-bit shift from 32-bit compilers */
92 86502504 2005-01-18 devnull z <<= 16;
94 86502504 2005-01-18 devnull t = (_Thread*)z;
96 fc567f47 2011-08-02 rsc //print("threadstart sp=%p arg=%p startfn=%p t=%p\n", &t, t, t->startfn, t->startarg);
97 1544f909 2004-12-25 devnull t->startfn(t->startarg);
98 cbeb0b26 2006-04-01 devnull /*print("threadexits %p\n", v); */
99 4dbefdd4 2004-12-27 devnull threadexits(nil);
100 cbeb0b26 2006-04-01 devnull /*print("not reacehd\n"); */
103 1544f909 2004-12-25 devnull static _Thread*
104 1544f909 2004-12-25 devnull threadalloc(void (*fn)(void*), void *arg, uint stack)
106 1544f909 2004-12-25 devnull _Thread *t;
107 1544f909 2004-12-25 devnull sigset_t zero;
108 86502504 2005-01-18 devnull uint x, y;
109 86502504 2005-01-18 devnull ulong z;
111 1544f909 2004-12-25 devnull /* allocate the task and stack together */
112 8c573cab 2020-01-14 rsc t = malloc(sizeof *t);
113 73722a8b 2004-12-27 devnull if(t == nil)
114 73722a8b 2004-12-27 devnull sysfatal("threadalloc malloc: %r");
115 1544f909 2004-12-25 devnull memset(t, 0, sizeof *t);
116 1544f909 2004-12-25 devnull t->id = incref(&threadidref);
117 fc567f47 2011-08-02 rsc //print("fn=%p arg=%p\n", fn, arg);
118 1544f909 2004-12-25 devnull t->startfn = fn;
119 1544f909 2004-12-25 devnull t->startarg = arg;
120 fc567f47 2011-08-02 rsc //print("makecontext sp=%p t=%p startfn=%p\n", (char*)t->stk+t->stksize, t, t->startfn);
122 1544f909 2004-12-25 devnull /* do a reasonable initialization */
123 0158bcee 2020-01-12 rsc if(stack == 0)
125 8c573cab 2020-01-14 rsc t->stk = _threadstkalloc(stack);
126 8c573cab 2020-01-14 rsc if(t->stk == nil)
127 8c573cab 2020-01-14 rsc sysfatal("threadalloc malloc stack: %r");
128 0158bcee 2020-01-12 rsc t->stksize = stack;
129 1544f909 2004-12-25 devnull memset(&t->context.uc, 0, sizeof t->context.uc);
130 1544f909 2004-12-25 devnull sigemptyset(&zero);
131 1544f909 2004-12-25 devnull sigprocmask(SIG_BLOCK, &zero, &t->context.uc.uc_sigmask);
132 fc567f47 2011-08-02 rsc //print("makecontext sp=%p t=%p startfn=%p\n", (char*)t->stk+t->stksize, t, t->startfn);
134 d54ead7f 2004-12-28 devnull /* must initialize with current context */
135 37cc72d3 2005-01-11 devnull if(getcontext(&t->context.uc) < 0)
136 37cc72d3 2005-01-11 devnull sysfatal("threadalloc getcontext: %r");
137 fc567f47 2011-08-02 rsc //print("makecontext sp=%p t=%p startfn=%p\n", (char*)t->stk+t->stksize, t, t->startfn);
140 4197af41 2020-01-24 rsc * Call makecontext to do the real work.
141 4197af41 2020-01-24 rsc * To avoid various mistakes on other system software,
142 4197af41 2020-01-24 rsc * debuggers, and so on, don't get too close to both
143 4197af41 2020-01-24 rsc * ends of the stack. Just staying away is much easier
144 4197af41 2020-01-24 rsc * than debugging everything (outside our control)
145 4197af41 2020-01-24 rsc * that has off-by-one errors.
147 4197af41 2020-01-24 rsc t->context.uc.uc_stack.ss_sp = (void*)(t->stk+64);
148 4197af41 2020-01-24 rsc t->context.uc.uc_stack.ss_size = t->stksize-2*64;
149 d08f4066 2005-07-27 devnull #if defined(__sun__) && !defined(__MAKECONTEXT_V2_SOURCE) /* sigh */
150 37cc72d3 2005-01-11 devnull /* can avoid this with __MAKECONTEXT_V2_SOURCE but only on SunOS 5.9 */
151 fa325e9b 2020-01-10 cross t->context.uc.uc_stack.ss_sp =
152 2b3b1e89 2005-01-14 devnull (char*)t->context.uc.uc_stack.ss_sp
153 2b3b1e89 2005-01-14 devnull +t->context.uc.uc_stack.ss_size;
156 86502504 2005-01-18 devnull * All this magic is because you have to pass makecontext a
157 86502504 2005-01-18 devnull * function that takes some number of word-sized variables,
158 86502504 2005-01-18 devnull * and on 64-bit machines pointers are bigger than words.
160 fc567f47 2011-08-02 rsc //print("makecontext sp=%p t=%p startfn=%p\n", (char*)t->stk+t->stksize, t, t->startfn);
161 86502504 2005-01-18 devnull z = (ulong)t;
163 86502504 2005-01-18 devnull z >>= 16; /* hide undefined 32-bit shift from 32-bit compilers */
164 86502504 2005-01-18 devnull x = z>>16;
165 2b604081 2005-05-07 devnull makecontext(&t->context.uc, (void(*)(void))threadstart, 2, y, x);
167 1544f909 2004-12-25 devnull return t;
170 1544f909 2004-12-25 devnull _Thread*
171 1544f909 2004-12-25 devnull _threadcreate(Proc *p, void (*fn)(void*), void *arg, uint stack)
173 1544f909 2004-12-25 devnull _Thread *t;
175 0829f75b 2009-08-18 rsc /* defend against bad C libraries */
176 0829f75b 2009-08-18 rsc if(stack < (256<<10))
177 0829f75b 2009-08-18 rsc stack = 256<<10;
179 0158bcee 2020-01-12 rsc if(p->nthread == 0)
180 0158bcee 2020-01-12 rsc stack = 0; // not using it
181 1544f909 2004-12-25 devnull t = threadalloc(fn, arg, stack);
182 1544f909 2004-12-25 devnull t->proc = p;
183 1544f909 2004-12-25 devnull addthreadinproc(p, t);
184 1544f909 2004-12-25 devnull p->nthread++;
185 1544f909 2004-12-25 devnull _threadready(t);
186 1544f909 2004-12-25 devnull return t;
190 1544f909 2004-12-25 devnull threadcreate(void (*fn)(void*), void *arg, uint stack)
192 1544f909 2004-12-25 devnull _Thread *t;
194 1544f909 2004-12-25 devnull t = _threadcreate(proc(), fn, arg, stack);
195 1544f909 2004-12-25 devnull return t->id;
199 1544f909 2004-12-25 devnull proccreate(void (*fn)(void*), void *arg, uint stack)
202 1544f909 2004-12-25 devnull _Thread *t;
203 1544f909 2004-12-25 devnull Proc *p;
205 1544f909 2004-12-25 devnull p = procalloc();
206 1544f909 2004-12-25 devnull t = _threadcreate(p, fn, arg, stack);
207 803292c6 2005-01-18 devnull id = t->id; /* t might be freed after _procstart */
208 0158bcee 2020-01-12 rsc _procstart(p, procmain);
209 803292c6 2005-01-18 devnull return id;
213 1544f909 2004-12-25 devnull _threadswitch(void)
215 1544f909 2004-12-25 devnull Proc *p;
217 c8b6342d 2005-01-13 devnull needstack(0);
218 1544f909 2004-12-25 devnull p = proc();
219 cbeb0b26 2006-04-01 devnull /*print("threadswtch %p\n", p); */
220 0158bcee 2020-01-12 rsc if(p->thread->stk == nil)
221 0158bcee 2020-01-12 rsc procscheduler(p);
223 0158bcee 2020-01-12 rsc contextswitch(&p->thread->context, &p->schedcontext);
227 1544f909 2004-12-25 devnull _threadready(_Thread *t)
229 1544f909 2004-12-25 devnull Proc *p;
231 1544f909 2004-12-25 devnull p = t->proc;
232 1544f909 2004-12-25 devnull lock(&p->lock);
233 a0a331aa 2005-01-06 devnull p->runrend.l = &p->lock;
234 1544f909 2004-12-25 devnull addthread(&p->runqueue, t);
235 cbeb0b26 2006-04-01 devnull /*print("%d wake for job %d->%d\n", time(0), getpid(), p->osprocid); */
236 8cbd854a 2004-12-27 devnull if(p != proc())
237 a0a331aa 2005-01-06 devnull _procwakeupandunlock(&p->runrend);
239 a0a331aa 2005-01-06 devnull unlock(&p->lock);
243 615e0f9f 2006-02-07 devnull threadidle(void)
246 615e0f9f 2006-02-07 devnull Proc *p;
248 615e0f9f 2006-02-07 devnull p = proc();
249 615e0f9f 2006-02-07 devnull n = p->nswitch;
250 615e0f9f 2006-02-07 devnull lock(&p->lock);
251 615e0f9f 2006-02-07 devnull p->runrend.l = &p->lock;
252 615e0f9f 2006-02-07 devnull addthread(&p->idlequeue, p->thread);
253 615e0f9f 2006-02-07 devnull unlock(&p->lock);
254 615e0f9f 2006-02-07 devnull _threadswitch();
255 615e0f9f 2006-02-07 devnull return p->nswitch - n;
259 1544f909 2004-12-25 devnull threadyield(void)
262 8cbd854a 2004-12-27 devnull Proc *p;
264 8cbd854a 2004-12-27 devnull p = proc();
265 8cbd854a 2004-12-27 devnull n = p->nswitch;
266 8cbd854a 2004-12-27 devnull _threadready(p->thread);
267 1544f909 2004-12-25 devnull _threadswitch();
268 8cbd854a 2004-12-27 devnull return p->nswitch - n;
272 1544f909 2004-12-25 devnull threadexits(char *msg)
274 1544f909 2004-12-25 devnull Proc *p;
276 1544f909 2004-12-25 devnull p = proc();
277 4dbefdd4 2004-12-27 devnull if(msg == nil)
278 4dbefdd4 2004-12-27 devnull msg = "";
279 1544f909 2004-12-25 devnull utfecpy(p->msg, p->msg+sizeof p->msg, msg);
280 4dbefdd4 2004-12-27 devnull proc()->thread->exiting = 1;
281 4dbefdd4 2004-12-27 devnull _threadswitch();
285 df970459 2006-06-26 devnull threadpin(void)
287 df970459 2006-06-26 devnull Proc *p;
289 df970459 2006-06-26 devnull p = proc();
290 df970459 2006-06-26 devnull if(p->pinthread){
291 df970459 2006-06-26 devnull fprint(2, "already pinning a thread - %p %p\n", p->pinthread, p->thread);
292 df970459 2006-06-26 devnull assert(0);
294 df970459 2006-06-26 devnull p->pinthread = p->thread;
298 df970459 2006-06-26 devnull threadunpin(void)
300 df970459 2006-06-26 devnull Proc *p;
302 df970459 2006-06-26 devnull p = proc();
303 df970459 2006-06-26 devnull if(p->pinthread != p->thread){
304 df970459 2006-06-26 devnull fprint(2, "wrong pinthread - %p %p\n", p->pinthread, p->thread);
305 df970459 2006-06-26 devnull assert(0);
307 df970459 2006-06-26 devnull p->pinthread = nil;
311 d19769ae 2008-10-26 rsc threadsysfatal(char *fmt, va_list arg)
313 69d1259c 2008-07-20 devnull char buf[256];
315 69d1259c 2008-07-20 devnull vseprint(buf, buf+sizeof(buf), fmt, arg);
316 69d1259c 2008-07-20 devnull __fixargv0();
317 69d1259c 2008-07-20 devnull fprint(2, "%s: %s\n", argv0 ? argv0 : "<prog>", buf);
318 69d1259c 2008-07-20 devnull threadexitsall(buf);
321 1544f909 2004-12-25 devnull static void
322 1544f909 2004-12-25 devnull contextswitch(Context *from, Context *to)
324 1544f909 2004-12-25 devnull if(swapcontext(&from->uc, &to->uc) < 0){
325 1544f909 2004-12-25 devnull fprint(2, "swapcontext failed: %r\n");
326 1544f909 2004-12-25 devnull assert(0);
330 1544f909 2004-12-25 devnull static void
331 0158bcee 2020-01-12 rsc procmain(Proc *p)
335 0158bcee 2020-01-12 rsc _threadsetproc(p);
337 0158bcee 2020-01-12 rsc /* take out first thread to run on system stack */
338 0158bcee 2020-01-12 rsc t = p->runqueue.head;
339 0158bcee 2020-01-12 rsc delthread(&p->runqueue, t);
340 0158bcee 2020-01-12 rsc memset(&t->context.uc, 0, sizeof t->context.uc);
342 0158bcee 2020-01-12 rsc /* run it */
343 0158bcee 2020-01-12 rsc p->thread = t;
344 0158bcee 2020-01-12 rsc t->startfn(t->startarg);
345 0158bcee 2020-01-12 rsc if(p->nthread != 0)
346 0158bcee 2020-01-12 rsc threadexits(nil);
350 1fdb4a54 2005-01-23 devnull procscheduler(Proc *p)
352 1544f909 2004-12-25 devnull _Thread *t;
354 2c87dda8 2004-12-28 devnull _threaddebug("scheduler enter");
355 04e0a9bb 2011-10-14 rsc //print("s %p\n", p);
357 1544f909 2004-12-25 devnull lock(&p->lock);
358 0158bcee 2020-01-12 rsc t = p->thread;
359 0158bcee 2020-01-12 rsc p->thread = nil;
360 0158bcee 2020-01-12 rsc if(t->exiting){
361 0158bcee 2020-01-12 rsc delthreadinproc(p, t);
362 0158bcee 2020-01-12 rsc p->nthread--;
363 0158bcee 2020-01-12 rsc /*print("nthread %d\n", p->nthread); */
364 8c573cab 2020-01-14 rsc _threadstkfree(t->stk, t->stksize);
368 1544f909 2004-12-25 devnull for(;;){
369 df970459 2006-06-26 devnull if((t = p->pinthread) != nil){
370 df970459 2006-06-26 devnull while(!onlist(&p->runqueue, t)){
371 df970459 2006-06-26 devnull p->runrend.l = &p->lock;
372 df970459 2006-06-26 devnull _threaddebug("scheduler sleep (pin)");
373 df970459 2006-06-26 devnull _procsleep(&p->runrend);
374 df970459 2006-06-26 devnull _threaddebug("scheduler wake (pin)");
377 1544f909 2004-12-25 devnull while((t = p->runqueue.head) == nil){
378 1544f909 2004-12-25 devnull if(p->nthread == 0)
379 1544f909 2004-12-25 devnull goto Out;
380 615e0f9f 2006-02-07 devnull if((t = p->idlequeue.head) != nil){
382 615e0f9f 2006-02-07 devnull * Run all the idling threads once.
384 615e0f9f 2006-02-07 devnull while((t = p->idlequeue.head) != nil){
385 615e0f9f 2006-02-07 devnull delthread(&p->idlequeue, t);
386 615e0f9f 2006-02-07 devnull addthread(&p->runqueue, t);
388 615e0f9f 2006-02-07 devnull continue;
390 1544f909 2004-12-25 devnull p->runrend.l = &p->lock;
391 2c87dda8 2004-12-28 devnull _threaddebug("scheduler sleep");
392 1544f909 2004-12-25 devnull _procsleep(&p->runrend);
393 2c87dda8 2004-12-28 devnull _threaddebug("scheduler wake");
395 df970459 2006-06-26 devnull if(p->pinthread && p->pinthread != t)
396 df970459 2006-06-26 devnull fprint(2, "p->pinthread %p t %p\n", p->pinthread, t);
397 df970459 2006-06-26 devnull assert(p->pinthread == nil || p->pinthread == t);
398 1544f909 2004-12-25 devnull delthread(&p->runqueue, t);
399 1544f909 2004-12-25 devnull unlock(&p->lock);
400 1544f909 2004-12-25 devnull p->thread = t;
401 8cbd854a 2004-12-27 devnull p->nswitch++;
402 2c87dda8 2004-12-28 devnull _threaddebug("run %d (%s)", t->id, t->name);
403 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);
404 0158bcee 2020-01-12 rsc if(t->stk == nil)
406 1544f909 2004-12-25 devnull contextswitch(&p->schedcontext, &t->context);
407 cbeb0b26 2006-04-01 devnull /*print("back in scheduler\n"); */
412 2c87dda8 2004-12-28 devnull _threaddebug("scheduler exit");
413 8ee6ad4d 2005-02-14 devnull if(p->mainproc){
415 8ee6ad4d 2005-02-14 devnull * Stupid bug - on Linux 2.6 and maybe elsewhere,
416 8ee6ad4d 2005-02-14 devnull * if the main thread exits then the others keep running
417 8ee6ad4d 2005-02-14 devnull * but the process shows up as a zombie in ps and is not
418 8ee6ad4d 2005-02-14 devnull * attachable with ptrace. We'll just sit around pretending
419 8ee6ad4d 2005-02-14 devnull * to be a system proc instead of exiting.
421 8ee6ad4d 2005-02-14 devnull _threaddaemonize();
422 8ee6ad4d 2005-02-14 devnull lock(&threadnproclock);
423 8ee6ad4d 2005-02-14 devnull if(++threadnsysproc == threadnproc)
424 8ee6ad4d 2005-02-14 devnull threadexitsall(p->msg);
425 8ee6ad4d 2005-02-14 devnull p->sysproc = 1;
426 8ee6ad4d 2005-02-14 devnull unlock(&threadnproclock);
428 8ee6ad4d 2005-02-14 devnull sleep(1000);
431 4dbefdd4 2004-12-27 devnull delproc(p);
432 1544f909 2004-12-25 devnull lock(&threadnproclock);
433 1544f909 2004-12-25 devnull if(p->sysproc)
434 1544f909 2004-12-25 devnull --threadnsysproc;
435 1544f909 2004-12-25 devnull if(--threadnproc == threadnsysproc)
436 4dbefdd4 2004-12-27 devnull threadexitsall(p->msg);
437 1544f909 2004-12-25 devnull unlock(&threadnproclock);
438 1544f909 2004-12-25 devnull unlock(&p->lock);
439 d93cc14e 2006-06-12 devnull _threadsetproc(nil);
440 1544f909 2004-12-25 devnull free(p);
441 481b596d 2020-01-14 rsc _threadpexit();
445 1544f909 2004-12-25 devnull _threadsetsysproc(void)
447 1544f909 2004-12-25 devnull lock(&threadnproclock);
448 1544f909 2004-12-25 devnull if(++threadnsysproc == threadnproc)
449 8ee6ad4d 2005-02-14 devnull threadexitsall(nil);
450 1544f909 2004-12-25 devnull unlock(&threadnproclock);
451 1544f909 2004-12-25 devnull proc()->sysproc = 1;
455 4dbefdd4 2004-12-27 devnull procdata(void)
457 4dbefdd4 2004-12-27 devnull return &proc()->udata;
461 8dd8a81f 2005-03-18 devnull threaddata(void)
463 8dd8a81f 2005-03-18 devnull return &proc()->thread->udata;
466 4dbefdd4 2004-12-27 devnull extern Jmp *(*_notejmpbuf)(void);
467 4dbefdd4 2004-12-27 devnull static Jmp*
468 4dbefdd4 2004-12-27 devnull threadnotejmp(void)
470 4dbefdd4 2004-12-27 devnull return &proc()->sigjmp;
474 1544f909 2004-12-25 devnull * debugging
477 1544f909 2004-12-25 devnull threadsetname(char *fmt, ...)
479 1544f909 2004-12-25 devnull va_list arg;
480 1544f909 2004-12-25 devnull _Thread *t;
482 1544f909 2004-12-25 devnull t = proc()->thread;
483 1544f909 2004-12-25 devnull va_start(arg, fmt);
484 1544f909 2004-12-25 devnull vsnprint(t->name, sizeof t->name, fmt, arg);
485 1544f909 2004-12-25 devnull va_end(arg);
489 515b6e5b 2005-01-18 devnull threadgetname(void)
491 515b6e5b 2005-01-18 devnull return proc()->thread->name;
495 1544f909 2004-12-25 devnull threadsetstate(char *fmt, ...)
497 1544f909 2004-12-25 devnull va_list arg;
498 1544f909 2004-12-25 devnull _Thread *t;
500 1544f909 2004-12-25 devnull t = proc()->thread;
501 1544f909 2004-12-25 devnull va_start(arg, fmt);
502 1544f909 2004-12-25 devnull vsnprint(t->state, sizeof t->name, fmt, arg);
503 1544f909 2004-12-25 devnull va_end(arg);
507 9eda38e5 2006-02-05 devnull threadid(void)
509 9eda38e5 2006-02-05 devnull _Thread *t;
511 9eda38e5 2006-02-05 devnull t = proc()->thread;
512 9eda38e5 2006-02-05 devnull return t->id;
516 c8b6342d 2005-01-13 devnull needstack(int n)
518 c8b6342d 2005-01-13 devnull _Thread *t;
520 c8b6342d 2005-01-13 devnull t = proc()->thread;
521 8c573cab 2020-01-14 rsc if(t->stk == nil)
524 c8b6342d 2005-01-13 devnull if((char*)&t <= (char*)t->stk
525 c8b6342d 2005-01-13 devnull || (char*)&t - (char*)t->stk < 256+n){
526 1fdb4a54 2005-01-23 devnull fprint(2, "thread stack overflow: &t=%p tstk=%p n=%d\n", &t, t->stk, 256+n);
527 c8b6342d 2005-01-13 devnull abort();
532 faf1fb6c 2008-07-09 rsc singlethreaded(void)
534 faf1fb6c 2008-07-09 rsc return threadnproc == 1 && _threadprocs->nthread == 1;
538 1544f909 2004-12-25 devnull * locking
540 1544f909 2004-12-25 devnull static int
541 1544f909 2004-12-25 devnull threadqlock(QLock *l, int block, ulong pc)
543 cbeb0b26 2006-04-01 devnull /*print("threadqlock %p\n", l); */
544 1544f909 2004-12-25 devnull lock(&l->l);
545 1544f909 2004-12-25 devnull if(l->owner == nil){
546 1544f909 2004-12-25 devnull l->owner = (*threadnow)();
547 cbeb0b26 2006-04-01 devnull /*print("qlock %p @%#x by %p\n", l, pc, l->owner); */
548 1544f909 2004-12-25 devnull unlock(&l->l);
549 1544f909 2004-12-25 devnull return 1;
551 1544f909 2004-12-25 devnull if(!block){
552 1544f909 2004-12-25 devnull unlock(&l->l);
553 1544f909 2004-12-25 devnull return 0;
556 faf1fb6c 2008-07-09 rsc if(singlethreaded()){
557 faf1fb6c 2008-07-09 rsc fprint(2, "qlock deadlock\n");
561 cbeb0b26 2006-04-01 devnull /*print("qsleep %p @%#x by %p\n", l, pc, (*threadnow)()); */
562 1544f909 2004-12-25 devnull addthread(&l->waiting, (*threadnow)());
563 1544f909 2004-12-25 devnull unlock(&l->l);
565 1544f909 2004-12-25 devnull _threadswitch();
567 1544f909 2004-12-25 devnull if(l->owner != (*threadnow)()){
568 15cd8255 2005-01-04 devnull fprint(2, "%s: qlock pc=0x%lux owner=%p self=%p oops\n",
569 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
570 1544f909 2004-12-25 devnull abort();
572 cbeb0b26 2006-04-01 devnull /*print("qlock wakeup %p @%#x by %p\n", l, pc, (*threadnow)()); */
573 1544f909 2004-12-25 devnull return 1;
576 1544f909 2004-12-25 devnull static void
577 1544f909 2004-12-25 devnull threadqunlock(QLock *l, ulong pc)
579 80b8842f 2005-02-15 devnull _Thread *ready;
581 1544f909 2004-12-25 devnull lock(&l->l);
582 cbeb0b26 2006-04-01 devnull /*print("qlock unlock %p @%#x by %p (owner %p)\n", l, pc, (*threadnow)(), l->owner); */
583 1aa9c533 2005-01-07 devnull if(l->owner == 0){
584 15cd8255 2005-01-04 devnull fprint(2, "%s: qunlock pc=0x%lux owner=%p self=%p oops\n",
585 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
586 1aa9c533 2005-01-07 devnull abort();
588 80b8842f 2005-02-15 devnull if((l->owner = ready = l->waiting.head) != nil)
589 1544f909 2004-12-25 devnull delthread(&l->waiting, l->owner);
591 80b8842f 2005-02-15 devnull * N.B. Cannot call _threadready() before unlocking l->l,
592 80b8842f 2005-02-15 devnull * because the thread we are readying might:
593 80b8842f 2005-02-15 devnull * - be in another proc
594 80b8842f 2005-02-15 devnull * - start running immediately
595 80b8842f 2005-02-15 devnull * - and free l before we get a chance to run again
597 1544f909 2004-12-25 devnull unlock(&l->l);
598 80b8842f 2005-02-15 devnull if(ready)
599 80b8842f 2005-02-15 devnull _threadready(l->owner);
602 1544f909 2004-12-25 devnull static int
603 1544f909 2004-12-25 devnull threadrlock(RWLock *l, int block, ulong pc)
605 1544f909 2004-12-25 devnull USED(pc);
607 1544f909 2004-12-25 devnull lock(&l->l);
608 1544f909 2004-12-25 devnull if(l->writer == nil && l->wwaiting.head == nil){
609 1544f909 2004-12-25 devnull l->readers++;
610 1544f909 2004-12-25 devnull unlock(&l->l);
611 1544f909 2004-12-25 devnull return 1;
613 1544f909 2004-12-25 devnull if(!block){
614 1544f909 2004-12-25 devnull unlock(&l->l);
615 1544f909 2004-12-25 devnull return 0;
617 faf1fb6c 2008-07-09 rsc if(singlethreaded()){
618 faf1fb6c 2008-07-09 rsc fprint(2, "rlock deadlock\n");
621 1544f909 2004-12-25 devnull addthread(&l->rwaiting, (*threadnow)());
622 1544f909 2004-12-25 devnull unlock(&l->l);
623 1544f909 2004-12-25 devnull _threadswitch();
627 1544f909 2004-12-25 devnull static int
628 1544f909 2004-12-25 devnull threadwlock(RWLock *l, int block, ulong pc)
630 1544f909 2004-12-25 devnull USED(pc);
632 1544f909 2004-12-25 devnull lock(&l->l);
633 1544f909 2004-12-25 devnull if(l->writer == nil && l->readers == 0){
634 1544f909 2004-12-25 devnull l->writer = (*threadnow)();
635 1544f909 2004-12-25 devnull unlock(&l->l);
636 1544f909 2004-12-25 devnull return 1;
638 1544f909 2004-12-25 devnull if(!block){
639 1544f909 2004-12-25 devnull unlock(&l->l);
640 1544f909 2004-12-25 devnull return 0;
642 faf1fb6c 2008-07-09 rsc if(singlethreaded()){
643 faf1fb6c 2008-07-09 rsc fprint(2, "wlock deadlock\n");
646 1544f909 2004-12-25 devnull addthread(&l->wwaiting, (*threadnow)());
647 1544f909 2004-12-25 devnull unlock(&l->l);
648 1544f909 2004-12-25 devnull _threadswitch();
649 1544f909 2004-12-25 devnull return 1;
652 1544f909 2004-12-25 devnull static void
653 1544f909 2004-12-25 devnull threadrunlock(RWLock *l, ulong pc)
655 1544f909 2004-12-25 devnull _Thread *t;
657 1544f909 2004-12-25 devnull USED(pc);
658 80b8842f 2005-02-15 devnull t = nil;
659 1544f909 2004-12-25 devnull lock(&l->l);
660 1544f909 2004-12-25 devnull --l->readers;
661 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
662 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
663 1544f909 2004-12-25 devnull l->writer = t;
665 1544f909 2004-12-25 devnull unlock(&l->l);
667 80b8842f 2005-02-15 devnull _threadready(t);
671 1544f909 2004-12-25 devnull static void
672 1544f909 2004-12-25 devnull threadwunlock(RWLock *l, ulong pc)
674 1544f909 2004-12-25 devnull _Thread *t;
676 1544f909 2004-12-25 devnull USED(pc);
677 1544f909 2004-12-25 devnull lock(&l->l);
678 1544f909 2004-12-25 devnull l->writer = nil;
679 1544f909 2004-12-25 devnull assert(l->readers == 0);
680 1544f909 2004-12-25 devnull while((t = l->rwaiting.head) != nil){
681 1544f909 2004-12-25 devnull delthread(&l->rwaiting, t);
682 1544f909 2004-12-25 devnull l->readers++;
683 1544f909 2004-12-25 devnull _threadready(t);
685 80b8842f 2005-02-15 devnull t = nil;
686 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
687 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
688 1544f909 2004-12-25 devnull l->writer = t;
690 1544f909 2004-12-25 devnull unlock(&l->l);
692 80b8842f 2005-02-15 devnull _threadready(t);
696 1544f909 2004-12-25 devnull * sleep and wakeup
698 1544f909 2004-12-25 devnull static void
699 1544f909 2004-12-25 devnull threadrsleep(Rendez *r, ulong pc)
701 faf1fb6c 2008-07-09 rsc if(singlethreaded()){
702 faf1fb6c 2008-07-09 rsc fprint(2, "rsleep deadlock\n");
705 1544f909 2004-12-25 devnull addthread(&r->waiting, proc()->thread);
706 1544f909 2004-12-25 devnull qunlock(r->l);
707 1544f909 2004-12-25 devnull _threadswitch();
708 1544f909 2004-12-25 devnull qlock(r->l);
711 1544f909 2004-12-25 devnull static int
712 1544f909 2004-12-25 devnull threadrwakeup(Rendez *r, int all, ulong pc)
715 1544f909 2004-12-25 devnull _Thread *t;
717 1544f909 2004-12-25 devnull for(i=0;; i++){
718 1544f909 2004-12-25 devnull if(i==1 && !all)
720 1544f909 2004-12-25 devnull if((t = r->waiting.head) == nil)
722 1544f909 2004-12-25 devnull delthread(&r->waiting, t);
723 fa325e9b 2020-01-10 cross _threadready(t);
725 1544f909 2004-12-25 devnull return i;
729 4dbefdd4 2004-12-27 devnull * startup
732 4dbefdd4 2004-12-27 devnull static int threadargc;
733 4dbefdd4 2004-12-27 devnull static char **threadargv;
734 4dbefdd4 2004-12-27 devnull int mainstacksize;
735 d93cc14e 2006-06-12 devnull extern int _p9usepwlibrary; /* getgrgid etc. smash the stack - tell _p9dir just say no */
736 4dbefdd4 2004-12-27 devnull static void
737 4dbefdd4 2004-12-27 devnull threadmainstart(void *v)
739 4dbefdd4 2004-12-27 devnull USED(v);
742 0b561695 2005-01-16 devnull * N.B. This call to proc() is a program's first call (indirectly) to a
743 0b561695 2005-01-16 devnull * pthreads function while executing on a non-pthreads-allocated
744 0b561695 2005-01-16 devnull * stack. If the pthreads implementation is using the stack pointer
745 0b561695 2005-01-16 devnull * to locate the per-thread data, then this call will blow up.
746 0b561695 2005-01-16 devnull * This means the pthread implementation is not suitable for
747 0b561695 2005-01-16 devnull * running under libthread. Time to write your own. Sorry.
749 d93cc14e 2006-06-12 devnull _p9usepwlibrary = 0;
750 4dbefdd4 2004-12-27 devnull threadmainproc = proc();
751 4dbefdd4 2004-12-27 devnull threadmain(threadargc, threadargv);
754 d19769ae 2008-10-26 rsc extern void (*_sysfatal)(char*, va_list);
757 4dbefdd4 2004-12-27 devnull main(int argc, char **argv)
759 4dbefdd4 2004-12-27 devnull Proc *p;
761 1d2533d0 2004-12-28 devnull argv0 = argv[0];
763 615e0f9f 2006-02-07 devnull if(getenv("NOLIBTHREADDAEMONIZE") == nil)
764 615e0f9f 2006-02-07 devnull _threadsetupdaemonize();
766 4dbefdd4 2004-12-27 devnull threadargc = argc;
767 4dbefdd4 2004-12-27 devnull threadargv = argv;
770 4dbefdd4 2004-12-27 devnull * Install locking routines into C library.
772 4dbefdd4 2004-12-27 devnull _lock = _threadlock;
773 4dbefdd4 2004-12-27 devnull _unlock = _threadunlock;
774 4dbefdd4 2004-12-27 devnull _qlock = threadqlock;
775 4dbefdd4 2004-12-27 devnull _qunlock = threadqunlock;
776 4dbefdd4 2004-12-27 devnull _rlock = threadrlock;
777 4dbefdd4 2004-12-27 devnull _runlock = threadrunlock;
778 4dbefdd4 2004-12-27 devnull _wlock = threadwlock;
779 4dbefdd4 2004-12-27 devnull _wunlock = threadwunlock;
780 4dbefdd4 2004-12-27 devnull _rsleep = threadrsleep;
781 4dbefdd4 2004-12-27 devnull _rwakeup = threadrwakeup;
782 4dbefdd4 2004-12-27 devnull _notejmpbuf = threadnotejmp;
783 df970459 2006-06-26 devnull _pin = threadpin;
784 df970459 2006-06-26 devnull _unpin = threadunpin;
785 69d1259c 2008-07-20 devnull _sysfatal = threadsysfatal;
787 4dbefdd4 2004-12-27 devnull _pthreadinit();
788 4dbefdd4 2004-12-27 devnull p = procalloc();
789 8ee6ad4d 2005-02-14 devnull p->mainproc = 1;
790 4dbefdd4 2004-12-27 devnull _threadsetproc(p);
791 4dbefdd4 2004-12-27 devnull if(mainstacksize == 0)
792 a0e8d02d 2005-01-23 devnull mainstacksize = 256*1024;
793 af6e459f 2006-04-21 devnull atnotify(threadinfo, 1);
794 4dbefdd4 2004-12-27 devnull _threadcreate(p, threadmainstart, nil, mainstacksize);
795 0158bcee 2020-01-12 rsc procmain(p);
796 d7688cce 2005-02-16 devnull sysfatal("procscheduler returned in threadmain!");
797 8ee6ad4d 2005-02-14 devnull /* does not return */
798 15cd8255 2005-01-04 devnull return 0;
802 1544f909 2004-12-25 devnull * hooray for linked lists
804 1544f909 2004-12-25 devnull static void
805 1544f909 2004-12-25 devnull addthread(_Threadlist *l, _Thread *t)
807 1544f909 2004-12-25 devnull if(l->tail){
808 1544f909 2004-12-25 devnull l->tail->next = t;
809 1544f909 2004-12-25 devnull t->prev = l->tail;
811 1544f909 2004-12-25 devnull l->head = t;
812 1544f909 2004-12-25 devnull t->prev = nil;
814 1544f909 2004-12-25 devnull l->tail = t;
815 1544f909 2004-12-25 devnull t->next = nil;
818 1544f909 2004-12-25 devnull static void
819 1544f909 2004-12-25 devnull delthread(_Threadlist *l, _Thread *t)
821 1544f909 2004-12-25 devnull if(t->prev)
822 1544f909 2004-12-25 devnull t->prev->next = t->next;
824 1544f909 2004-12-25 devnull l->head = t->next;
825 1544f909 2004-12-25 devnull if(t->next)
826 1544f909 2004-12-25 devnull t->next->prev = t->prev;
828 1544f909 2004-12-25 devnull l->tail = t->prev;
831 df970459 2006-06-26 devnull /* inefficient but rarely used */
832 df970459 2006-06-26 devnull static int
833 df970459 2006-06-26 devnull onlist(_Threadlist *l, _Thread *t)
835 df970459 2006-06-26 devnull _Thread *tt;
837 df970459 2006-06-26 devnull for(tt = l->head; tt; tt=tt->next)
838 df970459 2006-06-26 devnull if(tt == t)
839 df970459 2006-06-26 devnull return 1;
840 df970459 2006-06-26 devnull return 0;
843 1544f909 2004-12-25 devnull static void
844 1544f909 2004-12-25 devnull addthreadinproc(Proc *p, _Thread *t)
846 1544f909 2004-12-25 devnull _Threadlist *l;
848 1544f909 2004-12-25 devnull l = &p->allthreads;
849 1544f909 2004-12-25 devnull if(l->tail){
850 1544f909 2004-12-25 devnull l->tail->allnext = t;
851 1544f909 2004-12-25 devnull t->allprev = l->tail;
853 1544f909 2004-12-25 devnull l->head = t;
854 1544f909 2004-12-25 devnull t->allprev = nil;
856 1544f909 2004-12-25 devnull l->tail = t;
857 1544f909 2004-12-25 devnull t->allnext = nil;
860 1544f909 2004-12-25 devnull static void
861 1544f909 2004-12-25 devnull delthreadinproc(Proc *p, _Thread *t)
863 1544f909 2004-12-25 devnull _Threadlist *l;
865 1544f909 2004-12-25 devnull l = &p->allthreads;
866 1544f909 2004-12-25 devnull if(t->allprev)
867 1544f909 2004-12-25 devnull t->allprev->allnext = t->allnext;
869 1544f909 2004-12-25 devnull l->head = t->allnext;
870 1544f909 2004-12-25 devnull if(t->allnext)
871 1544f909 2004-12-25 devnull t->allnext->allprev = t->allprev;
873 1544f909 2004-12-25 devnull l->tail = t->allprev;
876 4dbefdd4 2004-12-27 devnull Proc *_threadprocs;
877 4dbefdd4 2004-12-27 devnull Lock _threadprocslock;
878 4dbefdd4 2004-12-27 devnull static Proc *_threadprocstail;
880 1544f909 2004-12-25 devnull static void
881 4dbefdd4 2004-12-27 devnull addproc(Proc *p)
883 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
884 4dbefdd4 2004-12-27 devnull if(_threadprocstail){
885 4dbefdd4 2004-12-27 devnull _threadprocstail->next = p;
886 4dbefdd4 2004-12-27 devnull p->prev = _threadprocstail;
888 4dbefdd4 2004-12-27 devnull _threadprocs = p;
889 4dbefdd4 2004-12-27 devnull p->prev = nil;
891 4dbefdd4 2004-12-27 devnull _threadprocstail = p;
892 4dbefdd4 2004-12-27 devnull p->next = nil;
893 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
896 4dbefdd4 2004-12-27 devnull static void
897 4dbefdd4 2004-12-27 devnull delproc(Proc *p)
899 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
900 4dbefdd4 2004-12-27 devnull if(p->prev)
901 4dbefdd4 2004-12-27 devnull p->prev->next = p->next;
903 4dbefdd4 2004-12-27 devnull _threadprocs = p->next;
904 4dbefdd4 2004-12-27 devnull if(p->next)
905 4dbefdd4 2004-12-27 devnull p->next->prev = p->prev;
907 4dbefdd4 2004-12-27 devnull _threadprocstail = p->prev;
908 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
912 15cd8255 2005-01-04 devnull * notify - for now just use the usual mechanisms
915 15cd8255 2005-01-04 devnull threadnotify(int (*f)(void*, char*), int in)
917 15cd8255 2005-01-04 devnull atnotify(f, in);
920 6dde87f8 2005-09-26 devnull static int
921 6dde87f8 2005-09-26 devnull onrunqueue(Proc *p, _Thread *t)
923 6dde87f8 2005-09-26 devnull _Thread *tt;
925 6dde87f8 2005-09-26 devnull for(tt=p->runqueue.head; tt; tt=tt->next)
926 6dde87f8 2005-09-26 devnull if(tt == t)
927 6dde87f8 2005-09-26 devnull return 1;
928 6dde87f8 2005-09-26 devnull return 0;
932 6dde87f8 2005-09-26 devnull * print state - called from SIGINFO
934 6dde87f8 2005-09-26 devnull static int
935 6dde87f8 2005-09-26 devnull threadinfo(void *v, char *s)
937 6dde87f8 2005-09-26 devnull Proc *p;
938 6dde87f8 2005-09-26 devnull _Thread *t;
940 6dde87f8 2005-09-26 devnull if(strcmp(s, "quit") != 0 && strcmp(s, "sys: status request") != 0)
941 6dde87f8 2005-09-26 devnull return 0;
943 6dde87f8 2005-09-26 devnull for(p=_threadprocs; p; p=p->next){
944 6dde87f8 2005-09-26 devnull fprint(2, "proc %p %s%s\n", (void*)p->osprocid, p->msg,
945 6dde87f8 2005-09-26 devnull p->sysproc ? " (sysproc)": "");
946 6dde87f8 2005-09-26 devnull for(t=p->allthreads.head; t; t=t->allnext){
947 fa325e9b 2020-01-10 cross fprint(2, "\tthread %d %s: %s %s\n",
949 fa325e9b 2020-01-10 cross t == p->thread ? "Running" :
950 6dde87f8 2005-09-26 devnull onrunqueue(p, t) ? "Ready" : "Sleeping",
951 6dde87f8 2005-09-26 devnull t->state, t->name);
954 6dde87f8 2005-09-26 devnull return 1;