Blame


1 1544f909 2004-12-25 devnull #include "threadimpl.h"
2 1544f909 2004-12-25 devnull
3 1544f909 2004-12-25 devnull int _threaddebuglevel;
4 1544f909 2004-12-25 devnull
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;
10 1544f909 2004-12-25 devnull
11 4dbefdd4 2004-12-27 devnull static void addproc(Proc*);
12 4dbefdd4 2004-12-27 devnull static void delproc(Proc*);
13 1544f909 2004-12-25 devnull static void addthread(_Threadlist*, _Thread*);
14 1544f909 2004-12-25 devnull static void delthread(_Threadlist*, _Thread*);
15 1544f909 2004-12-25 devnull static void addthreadinproc(Proc*, _Thread*);
16 1544f909 2004-12-25 devnull static void delthreadinproc(Proc*, _Thread*);
17 1544f909 2004-12-25 devnull static void contextswitch(Context *from, Context *to);
18 1fdb4a54 2005-01-23 devnull static void procscheduler(Proc*);
19 1544f909 2004-12-25 devnull
20 2c87dda8 2004-12-28 devnull static void
21 2c87dda8 2004-12-28 devnull _threaddebug(char *fmt, ...)
22 2c87dda8 2004-12-28 devnull {
23 2c87dda8 2004-12-28 devnull va_list arg;
24 2c87dda8 2004-12-28 devnull char buf[128];
25 2c87dda8 2004-12-28 devnull _Thread *t;
26 af89fc10 2005-01-06 devnull char *p;
27 af89fc10 2005-01-06 devnull static int fd = -1;
28 2c87dda8 2004-12-28 devnull
29 af89fc10 2005-01-06 devnull return;
30 b8f742db 2005-01-11 devnull va_start(arg, fmt);
31 b8f742db 2005-01-11 devnull vfprint(1, fmt, arg);
32 b8f742db 2005-01-11 devnull va_end(arg);
33 b8f742db 2005-01-11 devnull return;
34 b8f742db 2005-01-11 devnull
35 af89fc10 2005-01-06 devnull if(fd < 0){
36 af89fc10 2005-01-06 devnull p = strrchr(argv0, '/');
37 af89fc10 2005-01-06 devnull if(p)
38 af89fc10 2005-01-06 devnull p++;
39 af89fc10 2005-01-06 devnull else
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 af89fc10 2005-01-06 devnull }
45 2c87dda8 2004-12-28 devnull
46 2c87dda8 2004-12-28 devnull va_start(arg, fmt);
47 2c87dda8 2004-12-28 devnull vsnprint(buf, sizeof buf, fmt, arg);
48 2c87dda8 2004-12-28 devnull va_end(arg);
49 2c87dda8 2004-12-28 devnull t = proc()->thread;
50 2c87dda8 2004-12-28 devnull if(t)
51 af89fc10 2005-01-06 devnull fprint(fd, "%d.%d: %s\n", getpid(), t->id, buf);
52 2c87dda8 2004-12-28 devnull else
53 af89fc10 2005-01-06 devnull fprint(fd, "%d._: %s\n", getpid(), buf);
54 2c87dda8 2004-12-28 devnull }
55 2c87dda8 2004-12-28 devnull
56 1544f909 2004-12-25 devnull static _Thread*
57 1544f909 2004-12-25 devnull getthreadnow(void)
58 1544f909 2004-12-25 devnull {
59 1544f909 2004-12-25 devnull return proc()->thread;
60 1544f909 2004-12-25 devnull }
61 1544f909 2004-12-25 devnull _Thread *(*threadnow)(void) = getthreadnow;
62 1544f909 2004-12-25 devnull
63 1544f909 2004-12-25 devnull static Proc*
64 1544f909 2004-12-25 devnull procalloc(void)
65 1544f909 2004-12-25 devnull {
66 1544f909 2004-12-25 devnull Proc *p;
67 1544f909 2004-12-25 devnull
68 1544f909 2004-12-25 devnull p = malloc(sizeof *p);
69 73722a8b 2004-12-27 devnull if(p == nil)
70 73722a8b 2004-12-27 devnull sysfatal("procalloc malloc: %r");
71 1544f909 2004-12-25 devnull memset(p, 0, sizeof *p);
72 4dbefdd4 2004-12-27 devnull addproc(p);
73 1544f909 2004-12-25 devnull lock(&threadnproclock);
74 1544f909 2004-12-25 devnull threadnproc++;
75 1544f909 2004-12-25 devnull unlock(&threadnproclock);
76 1544f909 2004-12-25 devnull return p;
77 1544f909 2004-12-25 devnull }
78 1544f909 2004-12-25 devnull
79 1544f909 2004-12-25 devnull static void
80 86502504 2005-01-18 devnull threadstart(uint y, uint x)
81 1544f909 2004-12-25 devnull {
82 1544f909 2004-12-25 devnull _Thread *t;
83 86502504 2005-01-18 devnull ulong z;
84 1544f909 2004-12-25 devnull
85 86502504 2005-01-18 devnull z = x<<16; /* hide undefined 32-bit shift from 32-bit compilers */
86 86502504 2005-01-18 devnull z <<= 16;
87 86502504 2005-01-18 devnull z |= y;
88 86502504 2005-01-18 devnull t = (_Thread*)z;
89 86502504 2005-01-18 devnull
90 b8f742db 2005-01-11 devnull //print("threadstart %p\n", v);
91 1544f909 2004-12-25 devnull t->startfn(t->startarg);
92 b8f742db 2005-01-11 devnull //print("threadexits %p\n", v);
93 4dbefdd4 2004-12-27 devnull threadexits(nil);
94 b8f742db 2005-01-11 devnull //print("not reacehd\n");
95 1544f909 2004-12-25 devnull }
96 1544f909 2004-12-25 devnull
97 1544f909 2004-12-25 devnull static _Thread*
98 1544f909 2004-12-25 devnull threadalloc(void (*fn)(void*), void *arg, uint stack)
99 1544f909 2004-12-25 devnull {
100 1544f909 2004-12-25 devnull _Thread *t;
101 1544f909 2004-12-25 devnull sigset_t zero;
102 86502504 2005-01-18 devnull uint x, y;
103 86502504 2005-01-18 devnull ulong z;
104 1544f909 2004-12-25 devnull
105 1544f909 2004-12-25 devnull /* allocate the task and stack together */
106 1544f909 2004-12-25 devnull t = malloc(sizeof *t+stack);
107 73722a8b 2004-12-27 devnull if(t == nil)
108 73722a8b 2004-12-27 devnull sysfatal("threadalloc malloc: %r");
109 1544f909 2004-12-25 devnull memset(t, 0, sizeof *t);
110 1544f909 2004-12-25 devnull t->stk = (uchar*)(t+1);
111 1544f909 2004-12-25 devnull t->stksize = stack;
112 1544f909 2004-12-25 devnull t->id = incref(&threadidref);
113 1544f909 2004-12-25 devnull t->startfn = fn;
114 1544f909 2004-12-25 devnull t->startarg = arg;
115 1544f909 2004-12-25 devnull
116 1544f909 2004-12-25 devnull /* do a reasonable initialization */
117 1544f909 2004-12-25 devnull memset(&t->context.uc, 0, sizeof t->context.uc);
118 1544f909 2004-12-25 devnull sigemptyset(&zero);
119 1544f909 2004-12-25 devnull sigprocmask(SIG_BLOCK, &zero, &t->context.uc.uc_sigmask);
120 1544f909 2004-12-25 devnull
121 d54ead7f 2004-12-28 devnull /* must initialize with current context */
122 37cc72d3 2005-01-11 devnull if(getcontext(&t->context.uc) < 0)
123 37cc72d3 2005-01-11 devnull sysfatal("threadalloc getcontext: %r");
124 1544f909 2004-12-25 devnull
125 1544f909 2004-12-25 devnull /* call makecontext to do the real work. */
126 1544f909 2004-12-25 devnull /* leave a few words open on both ends */
127 1544f909 2004-12-25 devnull t->context.uc.uc_stack.ss_sp = t->stk+8;
128 2c87dda8 2004-12-28 devnull t->context.uc.uc_stack.ss_size = t->stksize-64;
129 d08f4066 2005-07-27 devnull #if defined(__sun__) && !defined(__MAKECONTEXT_V2_SOURCE) /* sigh */
130 37cc72d3 2005-01-11 devnull /* can avoid this with __MAKECONTEXT_V2_SOURCE but only on SunOS 5.9 */
131 2b3b1e89 2005-01-14 devnull t->context.uc.uc_stack.ss_sp =
132 2b3b1e89 2005-01-14 devnull (char*)t->context.uc.uc_stack.ss_sp
133 2b3b1e89 2005-01-14 devnull +t->context.uc.uc_stack.ss_size;
134 37cc72d3 2005-01-11 devnull #endif
135 86502504 2005-01-18 devnull /*
136 86502504 2005-01-18 devnull * All this magic is because you have to pass makecontext a
137 86502504 2005-01-18 devnull * function that takes some number of word-sized variables,
138 86502504 2005-01-18 devnull * and on 64-bit machines pointers are bigger than words.
139 86502504 2005-01-18 devnull */
140 86502504 2005-01-18 devnull z = (ulong)t;
141 86502504 2005-01-18 devnull y = z;
142 86502504 2005-01-18 devnull z >>= 16; /* hide undefined 32-bit shift from 32-bit compilers */
143 86502504 2005-01-18 devnull x = z>>16;
144 2b604081 2005-05-07 devnull makecontext(&t->context.uc, (void(*)(void))threadstart, 2, y, x);
145 1544f909 2004-12-25 devnull
146 1544f909 2004-12-25 devnull return t;
147 1544f909 2004-12-25 devnull }
148 1544f909 2004-12-25 devnull
149 1544f909 2004-12-25 devnull _Thread*
150 1544f909 2004-12-25 devnull _threadcreate(Proc *p, void (*fn)(void*), void *arg, uint stack)
151 1544f909 2004-12-25 devnull {
152 1544f909 2004-12-25 devnull _Thread *t;
153 1544f909 2004-12-25 devnull
154 1544f909 2004-12-25 devnull t = threadalloc(fn, arg, stack);
155 1544f909 2004-12-25 devnull t->proc = p;
156 1544f909 2004-12-25 devnull addthreadinproc(p, t);
157 1544f909 2004-12-25 devnull p->nthread++;
158 1544f909 2004-12-25 devnull _threadready(t);
159 1544f909 2004-12-25 devnull return t;
160 1544f909 2004-12-25 devnull }
161 1544f909 2004-12-25 devnull
162 1544f909 2004-12-25 devnull int
163 1544f909 2004-12-25 devnull threadcreate(void (*fn)(void*), void *arg, uint stack)
164 1544f909 2004-12-25 devnull {
165 1544f909 2004-12-25 devnull _Thread *t;
166 1544f909 2004-12-25 devnull
167 1544f909 2004-12-25 devnull t = _threadcreate(proc(), fn, arg, stack);
168 1544f909 2004-12-25 devnull return t->id;
169 1544f909 2004-12-25 devnull }
170 1544f909 2004-12-25 devnull
171 1544f909 2004-12-25 devnull int
172 1544f909 2004-12-25 devnull proccreate(void (*fn)(void*), void *arg, uint stack)
173 1544f909 2004-12-25 devnull {
174 803292c6 2005-01-18 devnull int id;
175 1544f909 2004-12-25 devnull _Thread *t;
176 1544f909 2004-12-25 devnull Proc *p;
177 1544f909 2004-12-25 devnull
178 1544f909 2004-12-25 devnull p = procalloc();
179 1544f909 2004-12-25 devnull t = _threadcreate(p, fn, arg, stack);
180 803292c6 2005-01-18 devnull id = t->id; /* t might be freed after _procstart */
181 1fdb4a54 2005-01-23 devnull _procstart(p, procscheduler);
182 803292c6 2005-01-18 devnull return id;
183 1544f909 2004-12-25 devnull }
184 1544f909 2004-12-25 devnull
185 1544f909 2004-12-25 devnull void
186 1544f909 2004-12-25 devnull _threadswitch(void)
187 1544f909 2004-12-25 devnull {
188 1544f909 2004-12-25 devnull Proc *p;
189 1544f909 2004-12-25 devnull
190 c8b6342d 2005-01-13 devnull needstack(0);
191 1544f909 2004-12-25 devnull p = proc();
192 b8f742db 2005-01-11 devnull //print("threadswtch %p\n", p);
193 1544f909 2004-12-25 devnull contextswitch(&p->thread->context, &p->schedcontext);
194 1544f909 2004-12-25 devnull }
195 1544f909 2004-12-25 devnull
196 1544f909 2004-12-25 devnull void
197 1544f909 2004-12-25 devnull _threadready(_Thread *t)
198 1544f909 2004-12-25 devnull {
199 1544f909 2004-12-25 devnull Proc *p;
200 1544f909 2004-12-25 devnull
201 1544f909 2004-12-25 devnull p = t->proc;
202 1544f909 2004-12-25 devnull lock(&p->lock);
203 a0a331aa 2005-01-06 devnull p->runrend.l = &p->lock;
204 1544f909 2004-12-25 devnull addthread(&p->runqueue, t);
205 8cbd854a 2004-12-27 devnull //print("%d wake for job %d->%d\n", time(0), getpid(), p->osprocid);
206 8cbd854a 2004-12-27 devnull if(p != proc())
207 a0a331aa 2005-01-06 devnull _procwakeupandunlock(&p->runrend);
208 a0a331aa 2005-01-06 devnull else
209 a0a331aa 2005-01-06 devnull unlock(&p->lock);
210 1544f909 2004-12-25 devnull }
211 1544f909 2004-12-25 devnull
212 8cbd854a 2004-12-27 devnull int
213 1544f909 2004-12-25 devnull threadyield(void)
214 1544f909 2004-12-25 devnull {
215 8cbd854a 2004-12-27 devnull int n;
216 8cbd854a 2004-12-27 devnull Proc *p;
217 8cbd854a 2004-12-27 devnull
218 8cbd854a 2004-12-27 devnull p = proc();
219 8cbd854a 2004-12-27 devnull n = p->nswitch;
220 8cbd854a 2004-12-27 devnull _threadready(p->thread);
221 1544f909 2004-12-25 devnull _threadswitch();
222 8cbd854a 2004-12-27 devnull return p->nswitch - n;
223 1544f909 2004-12-25 devnull }
224 1544f909 2004-12-25 devnull
225 1544f909 2004-12-25 devnull void
226 1544f909 2004-12-25 devnull threadexits(char *msg)
227 1544f909 2004-12-25 devnull {
228 1544f909 2004-12-25 devnull Proc *p;
229 1544f909 2004-12-25 devnull
230 1544f909 2004-12-25 devnull p = proc();
231 4dbefdd4 2004-12-27 devnull if(msg == nil)
232 4dbefdd4 2004-12-27 devnull msg = "";
233 1544f909 2004-12-25 devnull utfecpy(p->msg, p->msg+sizeof p->msg, msg);
234 4dbefdd4 2004-12-27 devnull proc()->thread->exiting = 1;
235 4dbefdd4 2004-12-27 devnull _threadswitch();
236 1544f909 2004-12-25 devnull }
237 1544f909 2004-12-25 devnull
238 1544f909 2004-12-25 devnull static void
239 1544f909 2004-12-25 devnull contextswitch(Context *from, Context *to)
240 1544f909 2004-12-25 devnull {
241 1544f909 2004-12-25 devnull if(swapcontext(&from->uc, &to->uc) < 0){
242 1544f909 2004-12-25 devnull fprint(2, "swapcontext failed: %r\n");
243 1544f909 2004-12-25 devnull assert(0);
244 1544f909 2004-12-25 devnull }
245 1544f909 2004-12-25 devnull }
246 1544f909 2004-12-25 devnull
247 1544f909 2004-12-25 devnull static void
248 1fdb4a54 2005-01-23 devnull procscheduler(Proc *p)
249 1544f909 2004-12-25 devnull {
250 1544f909 2004-12-25 devnull _Thread *t;
251 1544f909 2004-12-25 devnull
252 1544f909 2004-12-25 devnull setproc(p);
253 2c87dda8 2004-12-28 devnull _threaddebug("scheduler enter");
254 b8f742db 2005-01-11 devnull // print("s %p\n", p);
255 1544f909 2004-12-25 devnull lock(&p->lock);
256 1544f909 2004-12-25 devnull for(;;){
257 1544f909 2004-12-25 devnull while((t = p->runqueue.head) == nil){
258 1544f909 2004-12-25 devnull if(p->nthread == 0)
259 1544f909 2004-12-25 devnull goto Out;
260 1544f909 2004-12-25 devnull p->runrend.l = &p->lock;
261 2c87dda8 2004-12-28 devnull _threaddebug("scheduler sleep");
262 1544f909 2004-12-25 devnull _procsleep(&p->runrend);
263 2c87dda8 2004-12-28 devnull _threaddebug("scheduler wake");
264 1544f909 2004-12-25 devnull }
265 1544f909 2004-12-25 devnull delthread(&p->runqueue, t);
266 1544f909 2004-12-25 devnull unlock(&p->lock);
267 1544f909 2004-12-25 devnull p->thread = t;
268 8cbd854a 2004-12-27 devnull p->nswitch++;
269 2c87dda8 2004-12-28 devnull _threaddebug("run %d (%s)", t->id, t->name);
270 1544f909 2004-12-25 devnull contextswitch(&p->schedcontext, &t->context);
271 b8f742db 2005-01-11 devnull //print("back in scheduler\n");
272 1544f909 2004-12-25 devnull p->thread = nil;
273 1544f909 2004-12-25 devnull lock(&p->lock);
274 1544f909 2004-12-25 devnull if(t->exiting){
275 1544f909 2004-12-25 devnull delthreadinproc(p, t);
276 1544f909 2004-12-25 devnull p->nthread--;
277 b8f742db 2005-01-11 devnull //print("ntrhead %d\n", p->nthread);
278 1544f909 2004-12-25 devnull free(t);
279 1544f909 2004-12-25 devnull }
280 1544f909 2004-12-25 devnull }
281 1544f909 2004-12-25 devnull
282 1544f909 2004-12-25 devnull Out:
283 2c87dda8 2004-12-28 devnull _threaddebug("scheduler exit");
284 8ee6ad4d 2005-02-14 devnull if(p->mainproc){
285 8ee6ad4d 2005-02-14 devnull /*
286 8ee6ad4d 2005-02-14 devnull * Stupid bug - on Linux 2.6 and maybe elsewhere,
287 8ee6ad4d 2005-02-14 devnull * if the main thread exits then the others keep running
288 8ee6ad4d 2005-02-14 devnull * but the process shows up as a zombie in ps and is not
289 8ee6ad4d 2005-02-14 devnull * attachable with ptrace. We'll just sit around pretending
290 8ee6ad4d 2005-02-14 devnull * to be a system proc instead of exiting.
291 8ee6ad4d 2005-02-14 devnull */
292 8ee6ad4d 2005-02-14 devnull _threaddaemonize();
293 8ee6ad4d 2005-02-14 devnull lock(&threadnproclock);
294 8ee6ad4d 2005-02-14 devnull if(++threadnsysproc == threadnproc)
295 8ee6ad4d 2005-02-14 devnull threadexitsall(p->msg);
296 8ee6ad4d 2005-02-14 devnull p->sysproc = 1;
297 8ee6ad4d 2005-02-14 devnull unlock(&threadnproclock);
298 8ee6ad4d 2005-02-14 devnull for(;;)
299 8ee6ad4d 2005-02-14 devnull sleep(1000);
300 8ee6ad4d 2005-02-14 devnull }
301 8ee6ad4d 2005-02-14 devnull
302 4dbefdd4 2004-12-27 devnull delproc(p);
303 1544f909 2004-12-25 devnull lock(&threadnproclock);
304 1544f909 2004-12-25 devnull if(p->sysproc)
305 1544f909 2004-12-25 devnull --threadnsysproc;
306 1544f909 2004-12-25 devnull if(--threadnproc == threadnsysproc)
307 4dbefdd4 2004-12-27 devnull threadexitsall(p->msg);
308 1544f909 2004-12-25 devnull unlock(&threadnproclock);
309 1544f909 2004-12-25 devnull unlock(&p->lock);
310 1544f909 2004-12-25 devnull free(p);
311 1544f909 2004-12-25 devnull }
312 1544f909 2004-12-25 devnull
313 1544f909 2004-12-25 devnull void
314 1544f909 2004-12-25 devnull _threadsetsysproc(void)
315 1544f909 2004-12-25 devnull {
316 1544f909 2004-12-25 devnull lock(&threadnproclock);
317 1544f909 2004-12-25 devnull if(++threadnsysproc == threadnproc)
318 8ee6ad4d 2005-02-14 devnull threadexitsall(nil);
319 1544f909 2004-12-25 devnull unlock(&threadnproclock);
320 1544f909 2004-12-25 devnull proc()->sysproc = 1;
321 1544f909 2004-12-25 devnull }
322 1544f909 2004-12-25 devnull
323 4dbefdd4 2004-12-27 devnull void**
324 4dbefdd4 2004-12-27 devnull procdata(void)
325 4dbefdd4 2004-12-27 devnull {
326 4dbefdd4 2004-12-27 devnull return &proc()->udata;
327 8dd8a81f 2005-03-18 devnull }
328 8dd8a81f 2005-03-18 devnull
329 8dd8a81f 2005-03-18 devnull void**
330 8dd8a81f 2005-03-18 devnull threaddata(void)
331 8dd8a81f 2005-03-18 devnull {
332 8dd8a81f 2005-03-18 devnull return &proc()->thread->udata;
333 4dbefdd4 2004-12-27 devnull }
334 4dbefdd4 2004-12-27 devnull
335 4dbefdd4 2004-12-27 devnull extern Jmp *(*_notejmpbuf)(void);
336 4dbefdd4 2004-12-27 devnull static Jmp*
337 4dbefdd4 2004-12-27 devnull threadnotejmp(void)
338 4dbefdd4 2004-12-27 devnull {
339 4dbefdd4 2004-12-27 devnull return &proc()->sigjmp;
340 4dbefdd4 2004-12-27 devnull }
341 4dbefdd4 2004-12-27 devnull
342 1544f909 2004-12-25 devnull /*
343 1544f909 2004-12-25 devnull * debugging
344 1544f909 2004-12-25 devnull */
345 1544f909 2004-12-25 devnull void
346 1544f909 2004-12-25 devnull threadsetname(char *fmt, ...)
347 1544f909 2004-12-25 devnull {
348 1544f909 2004-12-25 devnull va_list arg;
349 1544f909 2004-12-25 devnull _Thread *t;
350 1544f909 2004-12-25 devnull
351 1544f909 2004-12-25 devnull t = proc()->thread;
352 1544f909 2004-12-25 devnull va_start(arg, fmt);
353 1544f909 2004-12-25 devnull vsnprint(t->name, sizeof t->name, fmt, arg);
354 1544f909 2004-12-25 devnull va_end(arg);
355 515b6e5b 2005-01-18 devnull }
356 515b6e5b 2005-01-18 devnull
357 515b6e5b 2005-01-18 devnull char*
358 515b6e5b 2005-01-18 devnull threadgetname(void)
359 515b6e5b 2005-01-18 devnull {
360 515b6e5b 2005-01-18 devnull return proc()->thread->name;
361 1544f909 2004-12-25 devnull }
362 1544f909 2004-12-25 devnull
363 1544f909 2004-12-25 devnull void
364 1544f909 2004-12-25 devnull threadsetstate(char *fmt, ...)
365 1544f909 2004-12-25 devnull {
366 1544f909 2004-12-25 devnull va_list arg;
367 1544f909 2004-12-25 devnull _Thread *t;
368 1544f909 2004-12-25 devnull
369 1544f909 2004-12-25 devnull t = proc()->thread;
370 1544f909 2004-12-25 devnull va_start(arg, fmt);
371 1544f909 2004-12-25 devnull vsnprint(t->state, sizeof t->name, fmt, arg);
372 1544f909 2004-12-25 devnull va_end(arg);
373 c8b6342d 2005-01-13 devnull }
374 c8b6342d 2005-01-13 devnull
375 c8b6342d 2005-01-13 devnull void
376 c8b6342d 2005-01-13 devnull needstack(int n)
377 c8b6342d 2005-01-13 devnull {
378 c8b6342d 2005-01-13 devnull _Thread *t;
379 c8b6342d 2005-01-13 devnull
380 c8b6342d 2005-01-13 devnull t = proc()->thread;
381 c8b6342d 2005-01-13 devnull
382 c8b6342d 2005-01-13 devnull if((char*)&t <= (char*)t->stk
383 c8b6342d 2005-01-13 devnull || (char*)&t - (char*)t->stk < 256+n){
384 1fdb4a54 2005-01-23 devnull fprint(2, "thread stack overflow: &t=%p tstk=%p n=%d\n", &t, t->stk, 256+n);
385 c8b6342d 2005-01-13 devnull abort();
386 c8b6342d 2005-01-13 devnull }
387 1544f909 2004-12-25 devnull }
388 1544f909 2004-12-25 devnull
389 1544f909 2004-12-25 devnull /*
390 1544f909 2004-12-25 devnull * locking
391 1544f909 2004-12-25 devnull */
392 1544f909 2004-12-25 devnull static int
393 1544f909 2004-12-25 devnull threadqlock(QLock *l, int block, ulong pc)
394 1544f909 2004-12-25 devnull {
395 b8f742db 2005-01-11 devnull //print("threadqlock %p\n", l);
396 1544f909 2004-12-25 devnull lock(&l->l);
397 1544f909 2004-12-25 devnull if(l->owner == nil){
398 1544f909 2004-12-25 devnull l->owner = (*threadnow)();
399 1544f909 2004-12-25 devnull //print("qlock %p @%#x by %p\n", l, pc, l->owner);
400 1544f909 2004-12-25 devnull unlock(&l->l);
401 1544f909 2004-12-25 devnull return 1;
402 1544f909 2004-12-25 devnull }
403 1544f909 2004-12-25 devnull if(!block){
404 1544f909 2004-12-25 devnull unlock(&l->l);
405 1544f909 2004-12-25 devnull return 0;
406 1544f909 2004-12-25 devnull }
407 1544f909 2004-12-25 devnull //print("qsleep %p @%#x by %p\n", l, pc, (*threadnow)());
408 1544f909 2004-12-25 devnull addthread(&l->waiting, (*threadnow)());
409 1544f909 2004-12-25 devnull unlock(&l->l);
410 1544f909 2004-12-25 devnull
411 1544f909 2004-12-25 devnull _threadswitch();
412 1544f909 2004-12-25 devnull
413 1544f909 2004-12-25 devnull if(l->owner != (*threadnow)()){
414 15cd8255 2005-01-04 devnull fprint(2, "%s: qlock pc=0x%lux owner=%p self=%p oops\n",
415 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
416 1544f909 2004-12-25 devnull abort();
417 1544f909 2004-12-25 devnull }
418 1544f909 2004-12-25 devnull //print("qlock wakeup %p @%#x by %p\n", l, pc, (*threadnow)());
419 1544f909 2004-12-25 devnull return 1;
420 1544f909 2004-12-25 devnull }
421 1544f909 2004-12-25 devnull
422 1544f909 2004-12-25 devnull static void
423 1544f909 2004-12-25 devnull threadqunlock(QLock *l, ulong pc)
424 1544f909 2004-12-25 devnull {
425 80b8842f 2005-02-15 devnull _Thread *ready;
426 80b8842f 2005-02-15 devnull
427 1544f909 2004-12-25 devnull lock(&l->l);
428 1544f909 2004-12-25 devnull //print("qlock unlock %p @%#x by %p (owner %p)\n", l, pc, (*threadnow)(), l->owner);
429 1aa9c533 2005-01-07 devnull if(l->owner == 0){
430 15cd8255 2005-01-04 devnull fprint(2, "%s: qunlock pc=0x%lux owner=%p self=%p oops\n",
431 15cd8255 2005-01-04 devnull argv0, pc, l->owner, (*threadnow)());
432 1aa9c533 2005-01-07 devnull abort();
433 1544f909 2004-12-25 devnull }
434 80b8842f 2005-02-15 devnull if((l->owner = ready = l->waiting.head) != nil)
435 1544f909 2004-12-25 devnull delthread(&l->waiting, l->owner);
436 80b8842f 2005-02-15 devnull /*
437 80b8842f 2005-02-15 devnull * N.B. Cannot call _threadready() before unlocking l->l,
438 80b8842f 2005-02-15 devnull * because the thread we are readying might:
439 80b8842f 2005-02-15 devnull * - be in another proc
440 80b8842f 2005-02-15 devnull * - start running immediately
441 80b8842f 2005-02-15 devnull * - and free l before we get a chance to run again
442 80b8842f 2005-02-15 devnull */
443 1544f909 2004-12-25 devnull unlock(&l->l);
444 80b8842f 2005-02-15 devnull if(ready)
445 80b8842f 2005-02-15 devnull _threadready(l->owner);
446 1544f909 2004-12-25 devnull }
447 1544f909 2004-12-25 devnull
448 1544f909 2004-12-25 devnull static int
449 1544f909 2004-12-25 devnull threadrlock(RWLock *l, int block, ulong pc)
450 1544f909 2004-12-25 devnull {
451 1544f909 2004-12-25 devnull USED(pc);
452 1544f909 2004-12-25 devnull
453 1544f909 2004-12-25 devnull lock(&l->l);
454 1544f909 2004-12-25 devnull if(l->writer == nil && l->wwaiting.head == nil){
455 1544f909 2004-12-25 devnull l->readers++;
456 1544f909 2004-12-25 devnull unlock(&l->l);
457 1544f909 2004-12-25 devnull return 1;
458 1544f909 2004-12-25 devnull }
459 1544f909 2004-12-25 devnull if(!block){
460 1544f909 2004-12-25 devnull unlock(&l->l);
461 1544f909 2004-12-25 devnull return 0;
462 1544f909 2004-12-25 devnull }
463 1544f909 2004-12-25 devnull addthread(&l->rwaiting, (*threadnow)());
464 1544f909 2004-12-25 devnull unlock(&l->l);
465 1544f909 2004-12-25 devnull _threadswitch();
466 1544f909 2004-12-25 devnull return 1;
467 1544f909 2004-12-25 devnull }
468 1544f909 2004-12-25 devnull
469 1544f909 2004-12-25 devnull static int
470 1544f909 2004-12-25 devnull threadwlock(RWLock *l, int block, ulong pc)
471 1544f909 2004-12-25 devnull {
472 1544f909 2004-12-25 devnull USED(pc);
473 1544f909 2004-12-25 devnull
474 1544f909 2004-12-25 devnull lock(&l->l);
475 1544f909 2004-12-25 devnull if(l->writer == nil && l->readers == 0){
476 1544f909 2004-12-25 devnull l->writer = (*threadnow)();
477 1544f909 2004-12-25 devnull unlock(&l->l);
478 1544f909 2004-12-25 devnull return 1;
479 1544f909 2004-12-25 devnull }
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;
483 1544f909 2004-12-25 devnull }
484 1544f909 2004-12-25 devnull addthread(&l->wwaiting, (*threadnow)());
485 1544f909 2004-12-25 devnull unlock(&l->l);
486 1544f909 2004-12-25 devnull _threadswitch();
487 1544f909 2004-12-25 devnull return 1;
488 1544f909 2004-12-25 devnull }
489 1544f909 2004-12-25 devnull
490 1544f909 2004-12-25 devnull static void
491 1544f909 2004-12-25 devnull threadrunlock(RWLock *l, ulong pc)
492 1544f909 2004-12-25 devnull {
493 1544f909 2004-12-25 devnull _Thread *t;
494 1544f909 2004-12-25 devnull
495 1544f909 2004-12-25 devnull USED(pc);
496 80b8842f 2005-02-15 devnull t = nil;
497 1544f909 2004-12-25 devnull lock(&l->l);
498 1544f909 2004-12-25 devnull --l->readers;
499 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
500 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
501 1544f909 2004-12-25 devnull l->writer = t;
502 1544f909 2004-12-25 devnull }
503 1544f909 2004-12-25 devnull unlock(&l->l);
504 80b8842f 2005-02-15 devnull if(t)
505 80b8842f 2005-02-15 devnull _threadready(t);
506 80b8842f 2005-02-15 devnull
507 1544f909 2004-12-25 devnull }
508 1544f909 2004-12-25 devnull
509 1544f909 2004-12-25 devnull static void
510 1544f909 2004-12-25 devnull threadwunlock(RWLock *l, ulong pc)
511 1544f909 2004-12-25 devnull {
512 1544f909 2004-12-25 devnull _Thread *t;
513 1544f909 2004-12-25 devnull
514 1544f909 2004-12-25 devnull USED(pc);
515 1544f909 2004-12-25 devnull lock(&l->l);
516 1544f909 2004-12-25 devnull l->writer = nil;
517 1544f909 2004-12-25 devnull assert(l->readers == 0);
518 1544f909 2004-12-25 devnull while((t = l->rwaiting.head) != nil){
519 1544f909 2004-12-25 devnull delthread(&l->rwaiting, t);
520 1544f909 2004-12-25 devnull l->readers++;
521 1544f909 2004-12-25 devnull _threadready(t);
522 1544f909 2004-12-25 devnull }
523 80b8842f 2005-02-15 devnull t = nil;
524 1544f909 2004-12-25 devnull if(l->readers == 0 && (t = l->wwaiting.head) != nil){
525 1544f909 2004-12-25 devnull delthread(&l->wwaiting, t);
526 1544f909 2004-12-25 devnull l->writer = t;
527 1544f909 2004-12-25 devnull }
528 1544f909 2004-12-25 devnull unlock(&l->l);
529 80b8842f 2005-02-15 devnull if(t)
530 80b8842f 2005-02-15 devnull _threadready(t);
531 1544f909 2004-12-25 devnull }
532 1544f909 2004-12-25 devnull
533 1544f909 2004-12-25 devnull /*
534 1544f909 2004-12-25 devnull * sleep and wakeup
535 1544f909 2004-12-25 devnull */
536 1544f909 2004-12-25 devnull static void
537 1544f909 2004-12-25 devnull threadrsleep(Rendez *r, ulong pc)
538 1544f909 2004-12-25 devnull {
539 1544f909 2004-12-25 devnull addthread(&r->waiting, proc()->thread);
540 1544f909 2004-12-25 devnull qunlock(r->l);
541 1544f909 2004-12-25 devnull _threadswitch();
542 1544f909 2004-12-25 devnull qlock(r->l);
543 1544f909 2004-12-25 devnull }
544 1544f909 2004-12-25 devnull
545 1544f909 2004-12-25 devnull static int
546 1544f909 2004-12-25 devnull threadrwakeup(Rendez *r, int all, ulong pc)
547 1544f909 2004-12-25 devnull {
548 1544f909 2004-12-25 devnull int i;
549 1544f909 2004-12-25 devnull _Thread *t;
550 1544f909 2004-12-25 devnull
551 1544f909 2004-12-25 devnull for(i=0;; i++){
552 1544f909 2004-12-25 devnull if(i==1 && !all)
553 1544f909 2004-12-25 devnull break;
554 1544f909 2004-12-25 devnull if((t = r->waiting.head) == nil)
555 1544f909 2004-12-25 devnull break;
556 1544f909 2004-12-25 devnull delthread(&r->waiting, t);
557 1544f909 2004-12-25 devnull _threadready(t);
558 1544f909 2004-12-25 devnull }
559 1544f909 2004-12-25 devnull return i;
560 1544f909 2004-12-25 devnull }
561 1544f909 2004-12-25 devnull
562 1544f909 2004-12-25 devnull /*
563 4dbefdd4 2004-12-27 devnull * startup
564 4dbefdd4 2004-12-27 devnull */
565 4dbefdd4 2004-12-27 devnull
566 4dbefdd4 2004-12-27 devnull static int threadargc;
567 4dbefdd4 2004-12-27 devnull static char **threadargv;
568 4dbefdd4 2004-12-27 devnull int mainstacksize;
569 4dbefdd4 2004-12-27 devnull
570 4dbefdd4 2004-12-27 devnull static void
571 4dbefdd4 2004-12-27 devnull threadmainstart(void *v)
572 4dbefdd4 2004-12-27 devnull {
573 4dbefdd4 2004-12-27 devnull USED(v);
574 0b561695 2005-01-16 devnull
575 0b561695 2005-01-16 devnull /*
576 0b561695 2005-01-16 devnull * N.B. This call to proc() is a program's first call (indirectly) to a
577 0b561695 2005-01-16 devnull * pthreads function while executing on a non-pthreads-allocated
578 0b561695 2005-01-16 devnull * stack. If the pthreads implementation is using the stack pointer
579 0b561695 2005-01-16 devnull * to locate the per-thread data, then this call will blow up.
580 0b561695 2005-01-16 devnull * This means the pthread implementation is not suitable for
581 0b561695 2005-01-16 devnull * running under libthread. Time to write your own. Sorry.
582 0b561695 2005-01-16 devnull */
583 4dbefdd4 2004-12-27 devnull threadmainproc = proc();
584 4dbefdd4 2004-12-27 devnull threadmain(threadargc, threadargv);
585 2c87dda8 2004-12-28 devnull }
586 2c87dda8 2004-12-28 devnull
587 4dbefdd4 2004-12-27 devnull int
588 4dbefdd4 2004-12-27 devnull main(int argc, char **argv)
589 4dbefdd4 2004-12-27 devnull {
590 4dbefdd4 2004-12-27 devnull Proc *p;
591 1d2533d0 2004-12-28 devnull
592 1d2533d0 2004-12-28 devnull argv0 = argv[0];
593 4dbefdd4 2004-12-27 devnull
594 2c87dda8 2004-12-28 devnull _threadsetupdaemonize();
595 2c87dda8 2004-12-28 devnull
596 4dbefdd4 2004-12-27 devnull threadargc = argc;
597 4dbefdd4 2004-12-27 devnull threadargv = argv;
598 4dbefdd4 2004-12-27 devnull
599 4dbefdd4 2004-12-27 devnull /*
600 4dbefdd4 2004-12-27 devnull * Install locking routines into C library.
601 4dbefdd4 2004-12-27 devnull */
602 4dbefdd4 2004-12-27 devnull _lock = _threadlock;
603 4dbefdd4 2004-12-27 devnull _unlock = _threadunlock;
604 4dbefdd4 2004-12-27 devnull _qlock = threadqlock;
605 4dbefdd4 2004-12-27 devnull _qunlock = threadqunlock;
606 4dbefdd4 2004-12-27 devnull _rlock = threadrlock;
607 4dbefdd4 2004-12-27 devnull _runlock = threadrunlock;
608 4dbefdd4 2004-12-27 devnull _wlock = threadwlock;
609 4dbefdd4 2004-12-27 devnull _wunlock = threadwunlock;
610 4dbefdd4 2004-12-27 devnull _rsleep = threadrsleep;
611 4dbefdd4 2004-12-27 devnull _rwakeup = threadrwakeup;
612 4dbefdd4 2004-12-27 devnull _notejmpbuf = threadnotejmp;
613 4dbefdd4 2004-12-27 devnull
614 4dbefdd4 2004-12-27 devnull _pthreadinit();
615 4dbefdd4 2004-12-27 devnull p = procalloc();
616 8ee6ad4d 2005-02-14 devnull p->mainproc = 1;
617 4dbefdd4 2004-12-27 devnull _threadsetproc(p);
618 4dbefdd4 2004-12-27 devnull if(mainstacksize == 0)
619 a0e8d02d 2005-01-23 devnull mainstacksize = 256*1024;
620 4dbefdd4 2004-12-27 devnull _threadcreate(p, threadmainstart, nil, mainstacksize);
621 1fdb4a54 2005-01-23 devnull procscheduler(p);
622 d7688cce 2005-02-16 devnull sysfatal("procscheduler returned in threadmain!");
623 8ee6ad4d 2005-02-14 devnull /* does not return */
624 15cd8255 2005-01-04 devnull return 0;
625 4dbefdd4 2004-12-27 devnull }
626 4dbefdd4 2004-12-27 devnull
627 4dbefdd4 2004-12-27 devnull /*
628 1544f909 2004-12-25 devnull * hooray for linked lists
629 1544f909 2004-12-25 devnull */
630 1544f909 2004-12-25 devnull static void
631 1544f909 2004-12-25 devnull addthread(_Threadlist *l, _Thread *t)
632 1544f909 2004-12-25 devnull {
633 1544f909 2004-12-25 devnull if(l->tail){
634 1544f909 2004-12-25 devnull l->tail->next = t;
635 1544f909 2004-12-25 devnull t->prev = l->tail;
636 1544f909 2004-12-25 devnull }else{
637 1544f909 2004-12-25 devnull l->head = t;
638 1544f909 2004-12-25 devnull t->prev = nil;
639 1544f909 2004-12-25 devnull }
640 1544f909 2004-12-25 devnull l->tail = t;
641 1544f909 2004-12-25 devnull t->next = nil;
642 1544f909 2004-12-25 devnull }
643 1544f909 2004-12-25 devnull
644 1544f909 2004-12-25 devnull static void
645 1544f909 2004-12-25 devnull delthread(_Threadlist *l, _Thread *t)
646 1544f909 2004-12-25 devnull {
647 1544f909 2004-12-25 devnull if(t->prev)
648 1544f909 2004-12-25 devnull t->prev->next = t->next;
649 1544f909 2004-12-25 devnull else
650 1544f909 2004-12-25 devnull l->head = t->next;
651 1544f909 2004-12-25 devnull if(t->next)
652 1544f909 2004-12-25 devnull t->next->prev = t->prev;
653 1544f909 2004-12-25 devnull else
654 1544f909 2004-12-25 devnull l->tail = t->prev;
655 1544f909 2004-12-25 devnull }
656 1544f909 2004-12-25 devnull
657 1544f909 2004-12-25 devnull static void
658 1544f909 2004-12-25 devnull addthreadinproc(Proc *p, _Thread *t)
659 1544f909 2004-12-25 devnull {
660 1544f909 2004-12-25 devnull _Threadlist *l;
661 1544f909 2004-12-25 devnull
662 1544f909 2004-12-25 devnull l = &p->allthreads;
663 1544f909 2004-12-25 devnull if(l->tail){
664 1544f909 2004-12-25 devnull l->tail->allnext = t;
665 1544f909 2004-12-25 devnull t->allprev = l->tail;
666 1544f909 2004-12-25 devnull }else{
667 1544f909 2004-12-25 devnull l->head = t;
668 1544f909 2004-12-25 devnull t->allprev = nil;
669 1544f909 2004-12-25 devnull }
670 1544f909 2004-12-25 devnull l->tail = t;
671 1544f909 2004-12-25 devnull t->allnext = nil;
672 1544f909 2004-12-25 devnull }
673 1544f909 2004-12-25 devnull
674 1544f909 2004-12-25 devnull static void
675 1544f909 2004-12-25 devnull delthreadinproc(Proc *p, _Thread *t)
676 1544f909 2004-12-25 devnull {
677 1544f909 2004-12-25 devnull _Threadlist *l;
678 1544f909 2004-12-25 devnull
679 1544f909 2004-12-25 devnull l = &p->allthreads;
680 1544f909 2004-12-25 devnull if(t->allprev)
681 1544f909 2004-12-25 devnull t->allprev->allnext = t->allnext;
682 1544f909 2004-12-25 devnull else
683 1544f909 2004-12-25 devnull l->head = t->allnext;
684 1544f909 2004-12-25 devnull if(t->allnext)
685 1544f909 2004-12-25 devnull t->allnext->allprev = t->allprev;
686 1544f909 2004-12-25 devnull else
687 1544f909 2004-12-25 devnull l->tail = t->allprev;
688 1544f909 2004-12-25 devnull }
689 1544f909 2004-12-25 devnull
690 4dbefdd4 2004-12-27 devnull Proc *_threadprocs;
691 4dbefdd4 2004-12-27 devnull Lock _threadprocslock;
692 4dbefdd4 2004-12-27 devnull static Proc *_threadprocstail;
693 1544f909 2004-12-25 devnull
694 1544f909 2004-12-25 devnull static void
695 4dbefdd4 2004-12-27 devnull addproc(Proc *p)
696 1544f909 2004-12-25 devnull {
697 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
698 4dbefdd4 2004-12-27 devnull if(_threadprocstail){
699 4dbefdd4 2004-12-27 devnull _threadprocstail->next = p;
700 4dbefdd4 2004-12-27 devnull p->prev = _threadprocstail;
701 4dbefdd4 2004-12-27 devnull }else{
702 4dbefdd4 2004-12-27 devnull _threadprocs = p;
703 4dbefdd4 2004-12-27 devnull p->prev = nil;
704 4dbefdd4 2004-12-27 devnull }
705 4dbefdd4 2004-12-27 devnull _threadprocstail = p;
706 4dbefdd4 2004-12-27 devnull p->next = nil;
707 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
708 e1dc7e45 2004-12-27 devnull }
709 e1dc7e45 2004-12-27 devnull
710 4dbefdd4 2004-12-27 devnull static void
711 4dbefdd4 2004-12-27 devnull delproc(Proc *p)
712 e1dc7e45 2004-12-27 devnull {
713 4dbefdd4 2004-12-27 devnull lock(&_threadprocslock);
714 4dbefdd4 2004-12-27 devnull if(p->prev)
715 4dbefdd4 2004-12-27 devnull p->prev->next = p->next;
716 4dbefdd4 2004-12-27 devnull else
717 4dbefdd4 2004-12-27 devnull _threadprocs = p->next;
718 4dbefdd4 2004-12-27 devnull if(p->next)
719 4dbefdd4 2004-12-27 devnull p->next->prev = p->prev;
720 4dbefdd4 2004-12-27 devnull else
721 4dbefdd4 2004-12-27 devnull _threadprocstail = p->prev;
722 4dbefdd4 2004-12-27 devnull unlock(&_threadprocslock);
723 1544f909 2004-12-25 devnull }
724 15cd8255 2005-01-04 devnull
725 15cd8255 2005-01-04 devnull /*
726 15cd8255 2005-01-04 devnull * notify - for now just use the usual mechanisms
727 15cd8255 2005-01-04 devnull */
728 15cd8255 2005-01-04 devnull void
729 15cd8255 2005-01-04 devnull threadnotify(int (*f)(void*, char*), int in)
730 15cd8255 2005-01-04 devnull {
731 15cd8255 2005-01-04 devnull atnotify(f, in);
732 15cd8255 2005-01-04 devnull }