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 df1ee4e1 2008-06-01 rsc z = (ulong)x << 16; /* hide undefined 32-bit shift from 32-bit compilers */
90 86502504 2005-01-18 devnull z <<= 16;
92 86502504 2005-01-18 devnull t = (_Thread*)z;
94 cbeb0b26 2006-04-01 devnull /*print("threadstart %p\n", v); */
95 1544f909 2004-12-25 devnull t->startfn(t->startarg);
96 cbeb0b26 2006-04-01 devnull /*print("threadexits %p\n", v); */
97 4dbefdd4 2004-12-27 devnull threadexits(nil);
98 cbeb0b26 2006-04-01 devnull /*print("not reacehd\n"); */
101 1544f909 2004-12-25 devnull static _Thread*
102 1544f909 2004-12-25 devnull threadalloc(void (*fn)(void*), void *arg, uint stack)
104 1544f909 2004-12-25 devnull _Thread *t;
105 1544f909 2004-12-25 devnull sigset_t zero;
106 86502504 2005-01-18 devnull uint x, y;
107 86502504 2005-01-18 devnull ulong z;
109 1544f909 2004-12-25 devnull /* allocate the task and stack together */
110 1544f909 2004-12-25 devnull t = malloc(sizeof *t+stack);
111 73722a8b 2004-12-27 devnull if(t == nil)
112 73722a8b 2004-12-27 devnull sysfatal("threadalloc malloc: %r");
113 1544f909 2004-12-25 devnull memset(t, 0, sizeof *t);
114 1544f909 2004-12-25 devnull t->stk = (uchar*)(t+1);
115 1544f909 2004-12-25 devnull t->stksize = stack;
116 1544f909 2004-12-25 devnull t->id = incref(&threadidref);
117 1544f909 2004-12-25 devnull t->startfn = fn;
118 1544f909 2004-12-25 devnull t->startarg = arg;
120 1544f909 2004-12-25 devnull /* do a reasonable initialization */
121 1544f909 2004-12-25 devnull memset(&t->context.uc, 0, sizeof t->context.uc);
122 1544f909 2004-12-25 devnull sigemptyset(&zero);
123 1544f909 2004-12-25 devnull sigprocmask(SIG_BLOCK, &zero, &t->context.uc.uc_sigmask);
125 d54ead7f 2004-12-28 devnull /* must initialize with current context */
126 37cc72d3 2005-01-11 devnull if(getcontext(&t->context.uc) < 0)
127 37cc72d3 2005-01-11 devnull sysfatal("threadalloc getcontext: %r");
129 1544f909 2004-12-25 devnull /* call makecontext to do the real work. */
130 1544f909 2004-12-25 devnull /* leave a few words open on both ends */
131 1544f909 2004-12-25 devnull t->context.uc.uc_stack.ss_sp = t->stk+8;
132 2c87dda8 2004-12-28 devnull t->context.uc.uc_stack.ss_size = t->stksize-64;
133 d08f4066 2005-07-27 devnull #if defined(__sun__) && !defined(__MAKECONTEXT_V2_SOURCE) /* sigh */
134 37cc72d3 2005-01-11 devnull /* can avoid this with __MAKECONTEXT_V2_SOURCE but only on SunOS 5.9 */
135 2b3b1e89 2005-01-14 devnull t->context.uc.uc_stack.ss_sp =
136 2b3b1e89 2005-01-14 devnull (char*)t->context.uc.uc_stack.ss_sp
137 2b3b1e89 2005-01-14 devnull +t->context.uc.uc_stack.ss_size;
140 86502504 2005-01-18 devnull * All this magic is because you have to pass makecontext a
141 86502504 2005-01-18 devnull * function that takes some number of word-sized variables,
142 86502504 2005-01-18 devnull * and on 64-bit machines pointers are bigger than words.
144 86502504 2005-01-18 devnull z = (ulong)t;
146 86502504 2005-01-18 devnull z >>= 16; /* hide undefined 32-bit shift from 32-bit compilers */
147 86502504 2005-01-18 devnull x = z>>16;
148 2b604081 2005-05-07 devnull makecontext(&t->context.uc, (void(*)(void))threadstart, 2, y, x);
150 1544f909 2004-12-25 devnull return t;
153 1544f909 2004-12-25 devnull _Thread*
154 1544f909 2004-12-25 devnull _threadcreate(Proc *p, void (*fn)(void*), void *arg, uint stack)
156 1544f909 2004-12-25 devnull _Thread *t;
158 1544f909 2004-12-25 devnull t = threadalloc(fn, arg, stack);
159 1544f909 2004-12-25 devnull t->proc = p;
160 1544f909 2004-12-25 devnull addthreadinproc(p, t);
161 1544f909 2004-12-25 devnull p->nthread++;
162 1544f909 2004-12-25 devnull _threadready(t);
163 1544f909 2004-12-25 devnull return t;
167 1544f909 2004-12-25 devnull threadcreate(void (*fn)(void*), void *arg, uint stack)
169 1544f909 2004-12-25 devnull _Thread *t;
171 1544f909 2004-12-25 devnull t = _threadcreate(proc(), fn, arg, stack);
172 1544f909 2004-12-25 devnull return t->id;
176 1544f909 2004-12-25 devnull proccreate(void (*fn)(void*), void *arg, uint stack)
179 1544f909 2004-12-25 devnull _Thread *t;
180 1544f909 2004-12-25 devnull Proc *p;
182 1544f909 2004-12-25 devnull p = procalloc();
183 1544f909 2004-12-25 devnull t = _threadcreate(p, fn, arg, stack);
184 803292c6 2005-01-18 devnull id = t->id; /* t might be freed after _procstart */
185 1fdb4a54 2005-01-23 devnull _procstart(p, procscheduler);
186 803292c6 2005-01-18 devnull return id;
190 1544f909 2004-12-25 devnull _threadswitch(void)
192 1544f909 2004-12-25 devnull Proc *p;
194 c8b6342d 2005-01-13 devnull needstack(0);
195 1544f909 2004-12-25 devnull p = proc();
196 cbeb0b26 2006-04-01 devnull /*print("threadswtch %p\n", p); */
197 1544f909 2004-12-25 devnull contextswitch(&p->thread->context, &p->schedcontext);
201 1544f909 2004-12-25 devnull _threadready(_Thread *t)
203 1544f909 2004-12-25 devnull Proc *p;
205 1544f909 2004-12-25 devnull p = t->proc;
206 1544f909 2004-12-25 devnull lock(&p->lock);
207 a0a331aa 2005-01-06 devnull p->runrend.l = &p->lock;
208 1544f909 2004-12-25 devnull addthread(&p->runqueue, t);
209 cbeb0b26 2006-04-01 devnull /*print("%d wake for job %d->%d\n", time(0), getpid(), p->osprocid); */
210 8cbd854a 2004-12-27 devnull if(p != proc())
211 a0a331aa 2005-01-06 devnull _procwakeupandunlock(&p->runrend);
213 a0a331aa 2005-01-06 devnull unlock(&p->lock);
217 615e0f9f 2006-02-07 devnull threadidle(void)
220 615e0f9f 2006-02-07 devnull Proc *p;
222 615e0f9f 2006-02-07 devnull p = proc();
223 615e0f9f 2006-02-07 devnull n = p->nswitch;
224 615e0f9f 2006-02-07 devnull lock(&p->lock);
225 615e0f9f 2006-02-07 devnull p->runrend.l = &p->lock;
226 615e0f9f 2006-02-07 devnull addthread(&p->idlequeue, p->thread);
227 615e0f9f 2006-02-07 devnull unlock(&p->lock);
228 615e0f9f 2006-02-07 devnull _threadswitch();
229 615e0f9f 2006-02-07 devnull return p->nswitch - n;
233 1544f909 2004-12-25 devnull threadyield(void)
236 8cbd854a 2004-12-27 devnull Proc *p;
238 8cbd854a 2004-12-27 devnull p = proc();
239 8cbd854a 2004-12-27 devnull n = p->nswitch;
240 8cbd854a 2004-12-27 devnull _threadready(p->thread);
241 1544f909 2004-12-25 devnull _threadswitch();
242 8cbd854a 2004-12-27 devnull return p->nswitch - n;
246 1544f909 2004-12-25 devnull threadexits(char *msg)
248 1544f909 2004-12-25 devnull Proc *p;
250 1544f909 2004-12-25 devnull p = proc();
251 4dbefdd4 2004-12-27 devnull if(msg == nil)
252 4dbefdd4 2004-12-27 devnull msg = "";
253 1544f909 2004-12-25 devnull utfecpy(p->msg, p->msg+sizeof p->msg, msg);
254 4dbefdd4 2004-12-27 devnull proc()->thread->exiting = 1;
255 4dbefdd4 2004-12-27 devnull _threadswitch();
259 df970459 2006-06-26 devnull threadpin(void)
261 df970459 2006-06-26 devnull Proc *p;
263 df970459 2006-06-26 devnull p = proc();
264 df970459 2006-06-26 devnull if(p->pinthread){
265 df970459 2006-06-26 devnull fprint(2, "already pinning a thread - %p %p\n", p->pinthread, p->thread);
266 df970459 2006-06-26 devnull assert(0);
268 df970459 2006-06-26 devnull p->pinthread = p->thread;
272 df970459 2006-06-26 devnull threadunpin(void)
274 df970459 2006-06-26 devnull Proc *p;
276 df970459 2006-06-26 devnull p = proc();
277 df970459 2006-06-26 devnull if(p->pinthread != p->thread){
278 df970459 2006-06-26 devnull fprint(2, "wrong pinthread - %p %p\n", p->pinthread, p->thread);
279 df970459 2006-06-26 devnull assert(0);
281 df970459 2006-06-26 devnull p->pinthread = nil;
284 1544f909 2004-12-25 devnull static void
285 1544f909 2004-12-25 devnull contextswitch(Context *from, Context *to)
287 1544f909 2004-12-25 devnull if(swapcontext(&from->uc, &to->uc) < 0){
288 1544f909 2004-12-25 devnull fprint(2, "swapcontext failed: %r\n");
289 1544f909 2004-12-25 devnull assert(0);
293 1544f909 2004-12-25 devnull static void
294 1fdb4a54 2005-01-23 devnull procscheduler(Proc *p)
296 1544f909 2004-12-25 devnull _Thread *t;
298 1544f909 2004-12-25 devnull setproc(p);
299 2c87dda8 2004-12-28 devnull _threaddebug("scheduler enter");
300 cbeb0b26 2006-04-01 devnull /* print("s %p\n", p); */
301 1544f909 2004-12-25 devnull lock(&p->lock);
302 1544f909 2004-12-25 devnull for(;;){
303 df970459 2006-06-26 devnull if((t = p->pinthread) != nil){
304 df970459 2006-06-26 devnull while(!onlist(&p->runqueue, t)){
305 df970459 2006-06-26 devnull p->runrend.l = &p->lock;
306 df970459 2006-06-26 devnull _threaddebug("scheduler sleep (pin)");
307 df970459 2006-06-26 devnull _procsleep(&p->runrend);
308 df970459 2006-06-26 devnull _threaddebug("scheduler wake (pin)");
311 1544f909 2004-12-25 devnull while((t = p->runqueue.head) == nil){
312 1544f909 2004-12-25 devnull if(p->nthread == 0)
313 1544f909 2004-12-25 devnull goto Out;
314 615e0f9f 2006-02-07 devnull if((t = p->idlequeue.head) != nil){
316 615e0f9f 2006-02-07 devnull * Run all the idling threads once.
318 615e0f9f 2006-02-07 devnull while((t = p->idlequeue.head) != nil){
319 615e0f9f 2006-02-07 devnull delthread(&p->idlequeue, t);
320 615e0f9f 2006-02-07 devnull addthread(&p->runqueue, t);
322 615e0f9f 2006-02-07 devnull continue;
324 1544f909 2004-12-25 devnull p->runrend.l = &p->lock;
325 2c87dda8 2004-12-28 devnull _threaddebug("scheduler sleep");
326 1544f909 2004-12-25 devnull _procsleep(&p->runrend);
327 2c87dda8 2004-12-28 devnull _threaddebug("scheduler wake");
329 df970459 2006-06-26 devnull if(p->pinthread && p->pinthread != t)
330 df970459 2006-06-26 devnull fprint(2, "p->pinthread %p t %p\n", p->pinthread, t);
331 df970459 2006-06-26 devnull assert(p->pinthread == nil || p->pinthread == t);
332 1544f909 2004-12-25 devnull delthread(&p->runqueue, t);
333 1544f909 2004-12-25 devnull unlock(&p->lock);
334 1544f909 2004-12-25 devnull p->thread = t;
335 8cbd854a 2004-12-27 devnull p->nswitch++;
336 2c87dda8 2004-12-28 devnull _threaddebug("run %d (%s)", t->id, t->name);
337 1544f909 2004-12-25 devnull contextswitch(&p->schedcontext, &t->context);
338 cbeb0b26 2006-04-01 devnull /*print("back in scheduler\n"); */
339 1544f909 2004-12-25 devnull p->thread = nil;
340 1544f909 2004-12-25 devnull lock(&p->lock);
341 1544f909 2004-12-25 devnull if(t->exiting){
342 1544f909 2004-12-25 devnull delthreadinproc(p, t);
343 1544f909 2004-12-25 devnull p->nthread--;
344 cbeb0b26 2006-04-01 devnull /*print("nthread %d\n", p->nthread); */
345 1544f909 2004-12-25 devnull free(t);
350 2c87dda8 2004-12-28 devnull _threaddebug("scheduler exit");
351 8ee6ad4d 2005-02-14 devnull if(p->mainproc){
353 8ee6ad4d 2005-02-14 devnull * Stupid bug - on Linux 2.6 and maybe elsewhere,
354 8ee6ad4d 2005-02-14 devnull * if the main thread exits then the others keep running
355 8ee6ad4d 2005-02-14 devnull * but the process shows up as a zombie in ps and is not
356 8ee6ad4d 2005-02-14 devnull * attachable with ptrace. We'll just sit around pretending
357 8ee6ad4d 2005-02-14 devnull * to be a system proc instead of exiting.
359 8ee6ad4d 2005-02-14 devnull _threaddaemonize();
360 8ee6ad4d 2005-02-14 devnull lock(&threadnproclock);
361 8ee6ad4d 2005-02-14 devnull if(++threadnsysproc == threadnproc)
362 8ee6ad4d 2005-02-14 devnull threadexitsall(p->msg);
363 8ee6ad4d 2005-02-14 devnull p->sysproc = 1;
364 8ee6ad4d 2005-02-14 devnull unlock(&threadnproclock);
366 8ee6ad4d 2005-02-14 devnull sleep(1000);
369 4dbefdd4 2004-12-27 devnull delproc(p);
370 1544f909 2004-12-25 devnull lock(&threadnproclock);
371 1544f909 2004-12-25 devnull if(p->sysproc)
372 1544f909 2004-12-25 devnull --threadnsysproc;
373 1544f909 2004-12-25 devnull if(--threadnproc == threadnsysproc)
374 4dbefdd4 2004-12-27 devnull threadexitsall(p->msg);
375 1544f909 2004-12-25 devnull unlock(&threadnproclock);
376 1544f909 2004-12-25 devnull unlock(&p->lock);
377 d93cc14e 2006-06-12 devnull _threadsetproc(nil);
378 1544f909 2004-12-25 devnull free(p);
382 1544f909 2004-12-25 devnull _threadsetsysproc(void)
384 1544f909 2004-12-25 devnull lock(&threadnproclock);
385 1544f909 2004-12-25 devnull if(++threadnsysproc == threadnproc)
386 8ee6ad4d 2005-02-14 devnull threadexitsall(nil);
387 1544f909 2004-12-25 devnull unlock(&threadnproclock);
388 1544f909 2004-12-25 devnull proc()->sysproc = 1;
392 4dbefdd4 2004-12-27 devnull procdata(void)
394 4dbefdd4 2004-12-27 devnull return &proc()->udata;
398 8dd8a81f 2005-03-18 devnull threaddata(void)
400 8dd8a81f 2005-03-18 devnull return &proc()->thread->udata;
403 4dbefdd4 2004-12-27 devnull extern Jmp *(*_notejmpbuf)(void);
404 4dbefdd4 2004-12-27 devnull static Jmp*
405 4dbefdd4 2004-12-27 devnull threadnotejmp(void)
407 4dbefdd4 2004-12-27 devnull return &proc()->sigjmp;
411 1544f909 2004-12-25 devnull * debugging
414 1544f909 2004-12-25 devnull threadsetname(char *fmt, ...)
416 1544f909 2004-12-25 devnull va_list arg;
417 1544f909 2004-12-25 devnull _Thread *t;
419 1544f909 2004-12-25 devnull t = proc()->thread;
420 1544f909 2004-12-25 devnull va_start(arg, fmt);
421 1544f909 2004-12-25 devnull vsnprint(t->name, sizeof t->name, fmt, arg);
422 1544f909 2004-12-25 devnull va_end(arg);
426 515b6e5b 2005-01-18 devnull threadgetname(void)
428 515b6e5b 2005-01-18 devnull return proc()->thread->name;
432 1544f909 2004-12-25 devnull threadsetstate(char *fmt, ...)
434 1544f909 2004-12-25 devnull va_list arg;
435 1544f909 2004-12-25 devnull _Thread *t;
437 1544f909 2004-12-25 devnull t = proc()->thread;
438 1544f909 2004-12-25 devnull va_start(arg, fmt);
439 1544f909 2004-12-25 devnull vsnprint(t->state, sizeof t->name, fmt, arg);
440 1544f909 2004-12-25 devnull va_end(arg);
444 9eda38e5 2006-02-05 devnull threadid(void)
446 9eda38e5 2006-02-05 devnull _Thread *t;
448 9eda38e5 2006-02-05 devnull t = proc()->thread;
449 9eda38e5 2006-02-05 devnull return t->id;
453 c8b6342d 2005-01-13 devnull needstack(int n)
455 c8b6342d 2005-01-13 devnull _Thread *t;
457 c8b6342d 2005-01-13 devnull t = proc()->thread;
459 c8b6342d 2005-01-13 devnull if((char*)&t <= (char*)t->stk
460 c8b6342d 2005-01-13 devnull || (char*)&t - (char*)t->stk < 256+n){
461 1fdb4a54 2005-01-23 devnull fprint(2, "thread stack overflow: &t=%p tstk=%p n=%d\n", &t, t->stk, 256+n);
462 c8b6342d 2005-01-13 devnull abort();
467 1544f909 2004-12-25 devnull * locking
469 1544f909 2004-12-25 devnull static int
470 1544f909 2004-12-25 devnull threadqlock(QLock *l, int block, ulong pc)
472 cbeb0b26 2006-04-01 devnull /*print("threadqlock %p\n", l); */
473 1544f909 2004-12-25 devnull lock(&l->l);
474 1544f909 2004-12-25 devnull if(l->owner == nil){
475 1544f909 2004-12-25 devnull l->owner = (*threadnow)();
476 cbeb0b26 2006-04-01 devnull /*print("qlock %p @%#x by %p\n", l, pc, l->owner); */
477 1544f909 2004-12-25 devnull unlock(&l->l);
478 1544f909 2004-12-25 devnull return 1;
480 1544f909 2004-12-25 devnull if(!block){
481 1544f909 2004-12-25 devnull unlock(&l->l);
482 1544f909 2004-12-25 devnull return 0;
484 cbeb0b26 2006-04-01 devnull /*print("qsleep %p @%#x by %p\n", l, pc, (*threadnow)()); */
485 1544f909 2004-12-25 devnull addthread(&l->waiting, (*threadnow)());
486 1544f909 2004-12-25 devnull unlock(&l->l);
488 1544f909 2004-12-25 devnull _threadswitch();
490 1544f909 2004-12-25 devnull if(l->owner != (*threadnow)()){
491 15cd8255 2005-01-04 devnull fprint(2, "%s: qlock pc=0x%lux owner=%p self=%p oops\n",
492 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
493 1544f909 2004-12-25 devnull abort();
495 cbeb0b26 2006-04-01 devnull /*print("qlock wakeup %p @%#x by %p\n", l, pc, (*threadnow)()); */
496 1544f909 2004-12-25 devnull return 1;
499 1544f909 2004-12-25 devnull static void
500 1544f909 2004-12-25 devnull threadqunlock(QLock *l, ulong pc)
502 80b8842f 2005-02-15 devnull _Thread *ready;
504 1544f909 2004-12-25 devnull lock(&l->l);
505 cbeb0b26 2006-04-01 devnull /*print("qlock unlock %p @%#x by %p (owner %p)\n", l, pc, (*threadnow)(), l->owner); */
506 1aa9c533 2005-01-07 devnull if(l->owner == 0){
507 15cd8255 2005-01-04 devnull fprint(2, "%s: qunlock pc=0x%lux owner=%p self=%p oops\n",
508 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
509 1aa9c533 2005-01-07 devnull abort();
511 80b8842f 2005-02-15 devnull if((l->owner = ready = l->waiting.head) != nil)
512 1544f909 2004-12-25 devnull delthread(&l->waiting, l->owner);
514 80b8842f 2005-02-15 devnull * N.B. Cannot call _threadready() before unlocking l->l,
515 80b8842f 2005-02-15 devnull * because the thread we are readying might:
516 80b8842f 2005-02-15 devnull * - be in another proc
517 80b8842f 2005-02-15 devnull * - start running immediately
518 80b8842f 2005-02-15 devnull * - and free l before we get a chance to run again
520 1544f909 2004-12-25 devnull unlock(&l->l);
521 80b8842f 2005-02-15 devnull if(ready)
522 80b8842f 2005-02-15 devnull _threadready(l->owner);
525 1544f909 2004-12-25 devnull static int
526 1544f909 2004-12-25 devnull threadrlock(RWLock *l, int block, ulong pc)
528 1544f909 2004-12-25 devnull USED(pc);
530 1544f909 2004-12-25 devnull lock(&l->l);
531 1544f909 2004-12-25 devnull if(l->writer == nil && l->wwaiting.head == nil){
532 1544f909 2004-12-25 devnull l->readers++;
533 1544f909 2004-12-25 devnull unlock(&l->l);
534 1544f909 2004-12-25 devnull return 1;
536 1544f909 2004-12-25 devnull if(!block){
537 1544f909 2004-12-25 devnull unlock(&l->l);
538 1544f909 2004-12-25 devnull return 0;
540 1544f909 2004-12-25 devnull addthread(&l->rwaiting, (*threadnow)());
541 1544f909 2004-12-25 devnull unlock(&l->l);
542 1544f909 2004-12-25 devnull _threadswitch();
543 1544f909 2004-12-25 devnull return 1;
546 1544f909 2004-12-25 devnull static int
547 1544f909 2004-12-25 devnull threadwlock(RWLock *l, int block, ulong pc)
549 1544f909 2004-12-25 devnull USED(pc);
551 1544f909 2004-12-25 devnull lock(&l->l);
552 1544f909 2004-12-25 devnull if(l->writer == nil && l->readers == 0){
553 1544f909 2004-12-25 devnull l->writer = (*threadnow)();
554 1544f909 2004-12-25 devnull unlock(&l->l);
555 1544f909 2004-12-25 devnull return 1;
557 1544f909 2004-12-25 devnull if(!block){
558 1544f909 2004-12-25 devnull unlock(&l->l);
559 1544f909 2004-12-25 devnull return 0;
561 1544f909 2004-12-25 devnull addthread(&l->wwaiting, (*threadnow)());
562 1544f909 2004-12-25 devnull unlock(&l->l);
563 1544f909 2004-12-25 devnull _threadswitch();
564 1544f909 2004-12-25 devnull return 1;
567 1544f909 2004-12-25 devnull static void
568 1544f909 2004-12-25 devnull threadrunlock(RWLock *l, ulong pc)
570 1544f909 2004-12-25 devnull _Thread *t;
572 1544f909 2004-12-25 devnull USED(pc);
573 80b8842f 2005-02-15 devnull t = nil;
574 1544f909 2004-12-25 devnull lock(&l->l);
575 1544f909 2004-12-25 devnull --l->readers;
576 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
577 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
578 1544f909 2004-12-25 devnull l->writer = t;
580 1544f909 2004-12-25 devnull unlock(&l->l);
582 80b8842f 2005-02-15 devnull _threadready(t);
586 1544f909 2004-12-25 devnull static void
587 1544f909 2004-12-25 devnull threadwunlock(RWLock *l, ulong pc)
589 1544f909 2004-12-25 devnull _Thread *t;
591 1544f909 2004-12-25 devnull USED(pc);
592 1544f909 2004-12-25 devnull lock(&l->l);
593 1544f909 2004-12-25 devnull l->writer = nil;
594 1544f909 2004-12-25 devnull assert(l->readers == 0);
595 1544f909 2004-12-25 devnull while((t = l->rwaiting.head) != nil){
596 1544f909 2004-12-25 devnull delthread(&l->rwaiting, t);
597 1544f909 2004-12-25 devnull l->readers++;
598 1544f909 2004-12-25 devnull _threadready(t);
600 80b8842f 2005-02-15 devnull t = nil;
601 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
602 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
603 1544f909 2004-12-25 devnull l->writer = t;
605 1544f909 2004-12-25 devnull unlock(&l->l);
607 80b8842f 2005-02-15 devnull _threadready(t);
611 1544f909 2004-12-25 devnull * sleep and wakeup
613 1544f909 2004-12-25 devnull static void
614 1544f909 2004-12-25 devnull threadrsleep(Rendez *r, ulong pc)
616 1544f909 2004-12-25 devnull addthread(&r->waiting, proc()->thread);
617 1544f909 2004-12-25 devnull qunlock(r->l);
618 1544f909 2004-12-25 devnull _threadswitch();
619 1544f909 2004-12-25 devnull qlock(r->l);
622 1544f909 2004-12-25 devnull static int
623 1544f909 2004-12-25 devnull threadrwakeup(Rendez *r, int all, ulong pc)
626 1544f909 2004-12-25 devnull _Thread *t;
628 1544f909 2004-12-25 devnull for(i=0;; i++){
629 1544f909 2004-12-25 devnull if(i==1 && !all)
631 1544f909 2004-12-25 devnull if((t = r->waiting.head) == nil)
633 1544f909 2004-12-25 devnull delthread(&r->waiting, t);
634 1544f909 2004-12-25 devnull _threadready(t);
636 1544f909 2004-12-25 devnull return i;
640 4dbefdd4 2004-12-27 devnull * startup
643 4dbefdd4 2004-12-27 devnull static int threadargc;
644 4dbefdd4 2004-12-27 devnull static char **threadargv;
645 4dbefdd4 2004-12-27 devnull int mainstacksize;
646 d93cc14e 2006-06-12 devnull extern int _p9usepwlibrary; /* getgrgid etc. smash the stack - tell _p9dir just say no */
647 4dbefdd4 2004-12-27 devnull static void
648 4dbefdd4 2004-12-27 devnull threadmainstart(void *v)
650 4dbefdd4 2004-12-27 devnull USED(v);
653 0b561695 2005-01-16 devnull * N.B. This call to proc() is a program's first call (indirectly) to a
654 0b561695 2005-01-16 devnull * pthreads function while executing on a non-pthreads-allocated
655 0b561695 2005-01-16 devnull * stack. If the pthreads implementation is using the stack pointer
656 0b561695 2005-01-16 devnull * to locate the per-thread data, then this call will blow up.
657 0b561695 2005-01-16 devnull * This means the pthread implementation is not suitable for
658 0b561695 2005-01-16 devnull * running under libthread. Time to write your own. Sorry.
660 d93cc14e 2006-06-12 devnull _p9usepwlibrary = 0;
661 4dbefdd4 2004-12-27 devnull threadmainproc = proc();
662 4dbefdd4 2004-12-27 devnull threadmain(threadargc, threadargv);
666 4dbefdd4 2004-12-27 devnull main(int argc, char **argv)
668 4dbefdd4 2004-12-27 devnull Proc *p;
670 1d2533d0 2004-12-28 devnull argv0 = argv[0];
672 615e0f9f 2006-02-07 devnull if(getenv("NOLIBTHREADDAEMONIZE") == nil)
673 615e0f9f 2006-02-07 devnull _threadsetupdaemonize();
675 4dbefdd4 2004-12-27 devnull threadargc = argc;
676 4dbefdd4 2004-12-27 devnull threadargv = argv;
679 4dbefdd4 2004-12-27 devnull * Install locking routines into C library.
681 4dbefdd4 2004-12-27 devnull _lock = _threadlock;
682 4dbefdd4 2004-12-27 devnull _unlock = _threadunlock;
683 4dbefdd4 2004-12-27 devnull _qlock = threadqlock;
684 4dbefdd4 2004-12-27 devnull _qunlock = threadqunlock;
685 4dbefdd4 2004-12-27 devnull _rlock = threadrlock;
686 4dbefdd4 2004-12-27 devnull _runlock = threadrunlock;
687 4dbefdd4 2004-12-27 devnull _wlock = threadwlock;
688 4dbefdd4 2004-12-27 devnull _wunlock = threadwunlock;
689 4dbefdd4 2004-12-27 devnull _rsleep = threadrsleep;
690 4dbefdd4 2004-12-27 devnull _rwakeup = threadrwakeup;
691 4dbefdd4 2004-12-27 devnull _notejmpbuf = threadnotejmp;
692 df970459 2006-06-26 devnull _pin = threadpin;
693 df970459 2006-06-26 devnull _unpin = threadunpin;
695 4dbefdd4 2004-12-27 devnull _pthreadinit();
696 4dbefdd4 2004-12-27 devnull p = procalloc();
697 8ee6ad4d 2005-02-14 devnull p->mainproc = 1;
698 4dbefdd4 2004-12-27 devnull _threadsetproc(p);
699 4dbefdd4 2004-12-27 devnull if(mainstacksize == 0)
700 a0e8d02d 2005-01-23 devnull mainstacksize = 256*1024;
701 af6e459f 2006-04-21 devnull atnotify(threadinfo, 1);
702 4dbefdd4 2004-12-27 devnull _threadcreate(p, threadmainstart, nil, mainstacksize);
703 1fdb4a54 2005-01-23 devnull procscheduler(p);
704 d7688cce 2005-02-16 devnull sysfatal("procscheduler returned in threadmain!");
705 8ee6ad4d 2005-02-14 devnull /* does not return */
706 15cd8255 2005-01-04 devnull return 0;
710 1544f909 2004-12-25 devnull * hooray for linked lists
712 1544f909 2004-12-25 devnull static void
713 1544f909 2004-12-25 devnull addthread(_Threadlist *l, _Thread *t)
715 1544f909 2004-12-25 devnull if(l->tail){
716 1544f909 2004-12-25 devnull l->tail->next = t;
717 1544f909 2004-12-25 devnull t->prev = l->tail;
719 1544f909 2004-12-25 devnull l->head = t;
720 1544f909 2004-12-25 devnull t->prev = nil;
722 1544f909 2004-12-25 devnull l->tail = t;
723 1544f909 2004-12-25 devnull t->next = nil;
726 1544f909 2004-12-25 devnull static void
727 1544f909 2004-12-25 devnull delthread(_Threadlist *l, _Thread *t)
729 1544f909 2004-12-25 devnull if(t->prev)
730 1544f909 2004-12-25 devnull t->prev->next = t->next;
732 1544f909 2004-12-25 devnull l->head = t->next;
733 1544f909 2004-12-25 devnull if(t->next)
734 1544f909 2004-12-25 devnull t->next->prev = t->prev;
736 1544f909 2004-12-25 devnull l->tail = t->prev;
739 df970459 2006-06-26 devnull /* inefficient but rarely used */
740 df970459 2006-06-26 devnull static int
741 df970459 2006-06-26 devnull onlist(_Threadlist *l, _Thread *t)
743 df970459 2006-06-26 devnull _Thread *tt;
745 df970459 2006-06-26 devnull for(tt = l->head; tt; tt=tt->next)
746 df970459 2006-06-26 devnull if(tt == t)
747 df970459 2006-06-26 devnull return 1;
748 df970459 2006-06-26 devnull return 0;
751 1544f909 2004-12-25 devnull static void
752 1544f909 2004-12-25 devnull addthreadinproc(Proc *p, _Thread *t)
754 1544f909 2004-12-25 devnull _Threadlist *l;
756 1544f909 2004-12-25 devnull l = &p->allthreads;
757 1544f909 2004-12-25 devnull if(l->tail){
758 1544f909 2004-12-25 devnull l->tail->allnext = t;
759 1544f909 2004-12-25 devnull t->allprev = l->tail;
761 1544f909 2004-12-25 devnull l->head = t;
762 1544f909 2004-12-25 devnull t->allprev = nil;
764 1544f909 2004-12-25 devnull l->tail = t;
765 1544f909 2004-12-25 devnull t->allnext = nil;
768 1544f909 2004-12-25 devnull static void
769 1544f909 2004-12-25 devnull delthreadinproc(Proc *p, _Thread *t)
771 1544f909 2004-12-25 devnull _Threadlist *l;
773 1544f909 2004-12-25 devnull l = &p->allthreads;
774 1544f909 2004-12-25 devnull if(t->allprev)
775 1544f909 2004-12-25 devnull t->allprev->allnext = t->allnext;
777 1544f909 2004-12-25 devnull l->head = t->allnext;
778 1544f909 2004-12-25 devnull if(t->allnext)
779 1544f909 2004-12-25 devnull t->allnext->allprev = t->allprev;
781 1544f909 2004-12-25 devnull l->tail = t->allprev;
784 4dbefdd4 2004-12-27 devnull Proc *_threadprocs;
785 4dbefdd4 2004-12-27 devnull Lock _threadprocslock;
786 4dbefdd4 2004-12-27 devnull static Proc *_threadprocstail;
788 1544f909 2004-12-25 devnull static void
789 4dbefdd4 2004-12-27 devnull addproc(Proc *p)
791 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
792 4dbefdd4 2004-12-27 devnull if(_threadprocstail){
793 4dbefdd4 2004-12-27 devnull _threadprocstail->next = p;
794 4dbefdd4 2004-12-27 devnull p->prev = _threadprocstail;
796 4dbefdd4 2004-12-27 devnull _threadprocs = p;
797 4dbefdd4 2004-12-27 devnull p->prev = nil;
799 4dbefdd4 2004-12-27 devnull _threadprocstail = p;
800 4dbefdd4 2004-12-27 devnull p->next = nil;
801 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
804 4dbefdd4 2004-12-27 devnull static void
805 4dbefdd4 2004-12-27 devnull delproc(Proc *p)
807 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
808 4dbefdd4 2004-12-27 devnull if(p->prev)
809 4dbefdd4 2004-12-27 devnull p->prev->next = p->next;
811 4dbefdd4 2004-12-27 devnull _threadprocs = p->next;
812 4dbefdd4 2004-12-27 devnull if(p->next)
813 4dbefdd4 2004-12-27 devnull p->next->prev = p->prev;
815 4dbefdd4 2004-12-27 devnull _threadprocstail = p->prev;
816 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
820 15cd8255 2005-01-04 devnull * notify - for now just use the usual mechanisms
823 15cd8255 2005-01-04 devnull threadnotify(int (*f)(void*, char*), int in)
825 15cd8255 2005-01-04 devnull atnotify(f, in);
828 6dde87f8 2005-09-26 devnull static int
829 6dde87f8 2005-09-26 devnull onrunqueue(Proc *p, _Thread *t)
831 6dde87f8 2005-09-26 devnull _Thread *tt;
833 6dde87f8 2005-09-26 devnull for(tt=p->runqueue.head; tt; tt=tt->next)
834 6dde87f8 2005-09-26 devnull if(tt == t)
835 6dde87f8 2005-09-26 devnull return 1;
836 6dde87f8 2005-09-26 devnull return 0;
840 6dde87f8 2005-09-26 devnull * print state - called from SIGINFO
842 6dde87f8 2005-09-26 devnull static int
843 6dde87f8 2005-09-26 devnull threadinfo(void *v, char *s)
845 6dde87f8 2005-09-26 devnull Proc *p;
846 6dde87f8 2005-09-26 devnull _Thread *t;
848 6dde87f8 2005-09-26 devnull if(strcmp(s, "quit") != 0 && strcmp(s, "sys: status request") != 0)
849 6dde87f8 2005-09-26 devnull return 0;
851 6dde87f8 2005-09-26 devnull for(p=_threadprocs; p; p=p->next){
852 6dde87f8 2005-09-26 devnull fprint(2, "proc %p %s%s\n", (void*)p->osprocid, p->msg,
853 6dde87f8 2005-09-26 devnull p->sysproc ? " (sysproc)": "");
854 6dde87f8 2005-09-26 devnull for(t=p->allthreads.head; t; t=t->allnext){
855 6dde87f8 2005-09-26 devnull fprint(2, "\tthread %d %s: %s %s\n",
857 6dde87f8 2005-09-26 devnull t == p->thread ? "Running" :
858 6dde87f8 2005-09-26 devnull onrunqueue(p, t) ? "Ready" : "Sleeping",
859 6dde87f8 2005-09-26 devnull t->state, t->name);
862 6dde87f8 2005-09-26 devnull return 1;
