1 1544f909 2004-12-25 devnull #include "u.h"
  2 1544f909 2004-12-25 devnull #include "libc.h"
  3 1544f909 2004-12-25 devnull #include "thread.h"
  4 1544f909 2004-12-25 devnull #include "threadimpl.h"
  5 1544f909 2004-12-25 devnull 
  6 1544f909 2004-12-25 devnull int	_threaddebuglevel;
  7 1544f909 2004-12-25 devnull 
  8 1544f909 2004-12-25 devnull static	uint		threadnproc;
  9 1544f909 2004-12-25 devnull static	uint		threadnsysproc;
 10 1544f909 2004-12-25 devnull static	Lock		threadnproclock;
 11 1544f909 2004-12-25 devnull static	Ref		threadidref;
 12 4dbefdd4 2004-12-27 devnull static	Proc		*threadmainproc;
 13 1544f909 2004-12-25 devnull 
 14 4dbefdd4 2004-12-27 devnull static	void		addproc(Proc*);
 15 4dbefdd4 2004-12-27 devnull static	void		delproc(Proc*);
 16 1544f909 2004-12-25 devnull static	void		addthread(_Threadlist*, _Thread*);
 17 1544f909 2004-12-25 devnull static	void		delthread(_Threadlist*, _Thread*);
 18 1544f909 2004-12-25 devnull static	void		addthreadinproc(Proc*, _Thread*);
 19 1544f909 2004-12-25 devnull static	void		delthreadinproc(Proc*, _Thread*);
 20 1544f909 2004-12-25 devnull static	void		contextswitch(Context *from, Context *to);
 21 73722a8b 2004-12-27 devnull static	void		scheduler(Proc*);
 22 1544f909 2004-12-25 devnull 
 23 2c87dda8 2004-12-28 devnull static void
 24 2c87dda8 2004-12-28 devnull _threaddebug(char *fmt, ...)
 25 2c87dda8 2004-12-28 devnull {
 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;
 29 2c87dda8 2004-12-28 devnull 
 30 2c87dda8 2004-12-28 devnull 	return;
 31 2c87dda8 2004-12-28 devnull 
 32 2c87dda8 2004-12-28 devnull 	va_start(arg, fmt);
 33 2c87dda8 2004-12-28 devnull 	vsnprint(buf, sizeof buf, fmt, arg);
 34 2c87dda8 2004-12-28 devnull 	va_end(arg);
 35 2c87dda8 2004-12-28 devnull 	t = proc()->thread;
 36 2c87dda8 2004-12-28 devnull 	if(t)
 37 2c87dda8 2004-12-28 devnull 		fprint(2, "%d.%d: %s\n", getpid(), t->id, buf);
 38 2c87dda8 2004-12-28 devnull 	else
 39 2c87dda8 2004-12-28 devnull 		fprint(2, "%d._: %s\n", getpid(), buf);
 40 2c87dda8 2004-12-28 devnull }
 41 2c87dda8 2004-12-28 devnull 
 42 1544f909 2004-12-25 devnull static _Thread*
 43 1544f909 2004-12-25 devnull getthreadnow(void)
 44 1544f909 2004-12-25 devnull {
 45 1544f909 2004-12-25 devnull 	return proc()->thread;
 46 1544f909 2004-12-25 devnull }
 47 1544f909 2004-12-25 devnull _Thread	*(*threadnow)(void) = getthreadnow;
 48 1544f909 2004-12-25 devnull 
 49 1544f909 2004-12-25 devnull static Proc*
 50 1544f909 2004-12-25 devnull procalloc(void)
 51 1544f909 2004-12-25 devnull {
 52 1544f909 2004-12-25 devnull 	Proc *p;
 53 1544f909 2004-12-25 devnull 
 54 1544f909 2004-12-25 devnull 	p = malloc(sizeof *p);
 55 73722a8b 2004-12-27 devnull 	if(p == nil)
 56 73722a8b 2004-12-27 devnull 		sysfatal("procalloc malloc: %r");
 57 1544f909 2004-12-25 devnull 	memset(p, 0, sizeof *p);
 58 4dbefdd4 2004-12-27 devnull 	addproc(p);
 59 1544f909 2004-12-25 devnull 	lock(&threadnproclock);
 60 1544f909 2004-12-25 devnull 	threadnproc++;
 61 1544f909 2004-12-25 devnull 	unlock(&threadnproclock);
 62 1544f909 2004-12-25 devnull 	return p;
 63 1544f909 2004-12-25 devnull }
 64 1544f909 2004-12-25 devnull 
 65 1544f909 2004-12-25 devnull static void
 66 1544f909 2004-12-25 devnull threadstart(void *v)
 67 1544f909 2004-12-25 devnull {
 68 1544f909 2004-12-25 devnull 	_Thread *t;
 69 1544f909 2004-12-25 devnull 
 70 1544f909 2004-12-25 devnull 	t = v;
 71 1544f909 2004-12-25 devnull 	t->startfn(t->startarg);
 72 2c87dda8 2004-12-28 devnull 	memset(&v, 0xff, 32);	/* try to cut off stack traces */
 73 4dbefdd4 2004-12-27 devnull 	threadexits(nil);
 74 1544f909 2004-12-25 devnull }
 75 1544f909 2004-12-25 devnull 
 76 1544f909 2004-12-25 devnull static _Thread*
 77 1544f909 2004-12-25 devnull threadalloc(void (*fn)(void*), void *arg, uint stack)
 78 1544f909 2004-12-25 devnull {
 79 1544f909 2004-12-25 devnull 	_Thread *t;
 80 1544f909 2004-12-25 devnull 	sigset_t zero;
 81 1544f909 2004-12-25 devnull 
 82 1544f909 2004-12-25 devnull 	/* allocate the task and stack together */
 83 1544f909 2004-12-25 devnull 	t = malloc(sizeof *t+stack);
 84 73722a8b 2004-12-27 devnull 	if(t == nil)
 85 73722a8b 2004-12-27 devnull 		sysfatal("threadalloc malloc: %r");
 86 1544f909 2004-12-25 devnull 	memset(t, 0, sizeof *t);
 87 1544f909 2004-12-25 devnull 	t->stk = (uchar*)(t+1);
 88 1544f909 2004-12-25 devnull 	t->stksize = stack;
 89 1544f909 2004-12-25 devnull 	t->id = incref(&threadidref);
 90 1544f909 2004-12-25 devnull 	t->startfn = fn;
 91 1544f909 2004-12-25 devnull 	t->startarg = arg;
 92 1544f909 2004-12-25 devnull 
 93 1544f909 2004-12-25 devnull 	/* do a reasonable initialization */
 94 1544f909 2004-12-25 devnull 	memset(&t->context.uc, 0, sizeof t->context.uc);
 95 1544f909 2004-12-25 devnull 	sigemptyset(&zero);
 96 1544f909 2004-12-25 devnull 	sigprocmask(SIG_BLOCK, &zero, &t->context.uc.uc_sigmask);
 97 1544f909 2004-12-25 devnull 
 98 1544f909 2004-12-25 devnull 	/* on Linux makecontext neglects floating point */
 99 1544f909 2004-12-25 devnull 	getcontext(&t->context.uc);
100 1544f909 2004-12-25 devnull 
101 1544f909 2004-12-25 devnull 	/* call makecontext to do the real work. */
102 1544f909 2004-12-25 devnull 	/* leave a few words open on both ends */
103 1544f909 2004-12-25 devnull 	t->context.uc.uc_stack.ss_sp = t->stk+8;
104 2c87dda8 2004-12-28 devnull 	t->context.uc.uc_stack.ss_size = t->stksize-64;
105 1544f909 2004-12-25 devnull 	makecontext(&t->context.uc, (void(*)())threadstart, 1, t);
106 1544f909 2004-12-25 devnull 
107 1544f909 2004-12-25 devnull 	return t;
108 1544f909 2004-12-25 devnull }
109 1544f909 2004-12-25 devnull 
110 1544f909 2004-12-25 devnull _Thread*
111 1544f909 2004-12-25 devnull _threadcreate(Proc *p, void (*fn)(void*), void *arg, uint stack)
112 1544f909 2004-12-25 devnull {
113 1544f909 2004-12-25 devnull 	_Thread *t;
114 1544f909 2004-12-25 devnull 
115 1544f909 2004-12-25 devnull 	t = threadalloc(fn, arg, stack);
116 1544f909 2004-12-25 devnull 	t->proc = p;
117 1544f909 2004-12-25 devnull 	addthreadinproc(p, t);
118 1544f909 2004-12-25 devnull 	p->nthread++;
119 1544f909 2004-12-25 devnull 	_threadready(t);
120 1544f909 2004-12-25 devnull 	return t;
121 1544f909 2004-12-25 devnull }
122 1544f909 2004-12-25 devnull 
123 1544f909 2004-12-25 devnull int
124 1544f909 2004-12-25 devnull threadcreate(void (*fn)(void*), void *arg, uint stack)
125 1544f909 2004-12-25 devnull {
126 1544f909 2004-12-25 devnull 	_Thread *t;
127 1544f909 2004-12-25 devnull 
128 1544f909 2004-12-25 devnull 	t = _threadcreate(proc(), fn, arg, stack);
129 1544f909 2004-12-25 devnull 	return t->id;
130 1544f909 2004-12-25 devnull }
131 1544f909 2004-12-25 devnull 
132 1544f909 2004-12-25 devnull int
133 1544f909 2004-12-25 devnull proccreate(void (*fn)(void*), void *arg, uint stack)
134 1544f909 2004-12-25 devnull {
135 1544f909 2004-12-25 devnull 	_Thread *t;
136 1544f909 2004-12-25 devnull 	Proc *p;
137 1544f909 2004-12-25 devnull 
138 1544f909 2004-12-25 devnull 	p = procalloc();
139 1544f909 2004-12-25 devnull 	t = _threadcreate(p, fn, arg, stack);
140 1544f909 2004-12-25 devnull 	_procstart(p, scheduler);
141 1544f909 2004-12-25 devnull 	return t->id;
142 1544f909 2004-12-25 devnull }
143 1544f909 2004-12-25 devnull 
144 1544f909 2004-12-25 devnull void
145 1544f909 2004-12-25 devnull _threadswitch(void)
146 1544f909 2004-12-25 devnull {
147 1544f909 2004-12-25 devnull 	Proc *p;
148 1544f909 2004-12-25 devnull 
149 1544f909 2004-12-25 devnull 	p = proc();
150 1544f909 2004-12-25 devnull 	contextswitch(&p->thread->context, &p->schedcontext);
151 1544f909 2004-12-25 devnull }
152 1544f909 2004-12-25 devnull 
153 1544f909 2004-12-25 devnull void
154 1544f909 2004-12-25 devnull _threadready(_Thread *t)
155 1544f909 2004-12-25 devnull {
156 1544f909 2004-12-25 devnull 	Proc *p;
157 1544f909 2004-12-25 devnull 
158 1544f909 2004-12-25 devnull 	p = t->proc;
159 1544f909 2004-12-25 devnull 	lock(&p->lock);
160 1544f909 2004-12-25 devnull 	addthread(&p->runqueue, t);
161 8cbd854a 2004-12-27 devnull //print("%d wake for job %d->%d\n", time(0), getpid(), p->osprocid);
162 8cbd854a 2004-12-27 devnull 	if(p != proc())
163 8cbd854a 2004-12-27 devnull 		_procwakeup(&p->runrend);
164 1544f909 2004-12-25 devnull 	unlock(&p->lock);
165 1544f909 2004-12-25 devnull }
166 1544f909 2004-12-25 devnull 
167 8cbd854a 2004-12-27 devnull int
168 1544f909 2004-12-25 devnull threadyield(void)
169 1544f909 2004-12-25 devnull {
170 8cbd854a 2004-12-27 devnull 	int n;
171 8cbd854a 2004-12-27 devnull 	Proc *p;
172 8cbd854a 2004-12-27 devnull 
173 8cbd854a 2004-12-27 devnull 	p = proc();
174 8cbd854a 2004-12-27 devnull 	n = p->nswitch;
175 8cbd854a 2004-12-27 devnull 	_threadready(p->thread);
176 1544f909 2004-12-25 devnull 	_threadswitch();
177 8cbd854a 2004-12-27 devnull 	return p->nswitch - n;
178 1544f909 2004-12-25 devnull }
179 1544f909 2004-12-25 devnull 
180 1544f909 2004-12-25 devnull void
181 1544f909 2004-12-25 devnull threadexits(char *msg)
182 1544f909 2004-12-25 devnull {
183 1544f909 2004-12-25 devnull 	Proc *p;
184 1544f909 2004-12-25 devnull 
185 1544f909 2004-12-25 devnull 	p = proc();
186 4dbefdd4 2004-12-27 devnull 	if(msg == nil)
187 4dbefdd4 2004-12-27 devnull 		msg = "";
188 1544f909 2004-12-25 devnull 	utfecpy(p->msg, p->msg+sizeof p->msg, msg);
189 4dbefdd4 2004-12-27 devnull 	proc()->thread->exiting = 1;
190 4dbefdd4 2004-12-27 devnull 	_threadswitch();
191 1544f909 2004-12-25 devnull }
192 1544f909 2004-12-25 devnull 
193 1544f909 2004-12-25 devnull static void
194 1544f909 2004-12-25 devnull contextswitch(Context *from, Context *to)
195 1544f909 2004-12-25 devnull {
196 1544f909 2004-12-25 devnull 	if(swapcontext(&from->uc, &to->uc) < 0){
197 1544f909 2004-12-25 devnull 		fprint(2, "swapcontext failed: %r\n");
198 1544f909 2004-12-25 devnull 		assert(0);
199 1544f909 2004-12-25 devnull 	}
200 1544f909 2004-12-25 devnull }
201 1544f909 2004-12-25 devnull 
202 1544f909 2004-12-25 devnull static void
203 73722a8b 2004-12-27 devnull scheduler(Proc *p)
204 1544f909 2004-12-25 devnull {
205 1544f909 2004-12-25 devnull 	_Thread *t;
206 1544f909 2004-12-25 devnull 
207 1544f909 2004-12-25 devnull 	setproc(p);
208 2c87dda8 2004-12-28 devnull 	_threaddebug("scheduler enter");
209 e1dc7e45 2004-12-27 devnull 	// print("s %p %d\n", p, gettid());
210 1544f909 2004-12-25 devnull 	lock(&p->lock);
211 1544f909 2004-12-25 devnull 	for(;;){
212 1544f909 2004-12-25 devnull 		while((t = p->runqueue.head) == nil){
213 1544f909 2004-12-25 devnull 			if(p->nthread == 0)
214 1544f909 2004-12-25 devnull 				goto Out;
215 1544f909 2004-12-25 devnull 			p->runrend.l = &p->lock;
216 2c87dda8 2004-12-28 devnull 			_threaddebug("scheduler sleep");
217 1544f909 2004-12-25 devnull 			_procsleep(&p->runrend);
218 2c87dda8 2004-12-28 devnull 			_threaddebug("scheduler wake");
219 1544f909 2004-12-25 devnull 		}
220 1544f909 2004-12-25 devnull 		delthread(&p->runqueue, t);
221 1544f909 2004-12-25 devnull 		unlock(&p->lock);
222 1544f909 2004-12-25 devnull 		p->thread = t;
223 8cbd854a 2004-12-27 devnull 		p->nswitch++;
224 2c87dda8 2004-12-28 devnull 		_threaddebug("run %d (%s)", t->id, t->name);
225 1544f909 2004-12-25 devnull 		contextswitch(&p->schedcontext, &t->context);
226 1544f909 2004-12-25 devnull 		p->thread = nil;
227 1544f909 2004-12-25 devnull 		lock(&p->lock);
228 1544f909 2004-12-25 devnull 		if(t->exiting){
229 1544f909 2004-12-25 devnull 			delthreadinproc(p, t);
230 1544f909 2004-12-25 devnull 			p->nthread--;
231 1544f909 2004-12-25 devnull 			free(t);
232 1544f909 2004-12-25 devnull 		}
233 1544f909 2004-12-25 devnull 	}
234 1544f909 2004-12-25 devnull 
235 1544f909 2004-12-25 devnull Out:
236 2c87dda8 2004-12-28 devnull 	_threaddebug("scheduler exit");
237 4dbefdd4 2004-12-27 devnull 	delproc(p);
238 1544f909 2004-12-25 devnull 	lock(&threadnproclock);
239 1544f909 2004-12-25 devnull 	if(p->sysproc)
240 1544f909 2004-12-25 devnull 		--threadnsysproc;
241 1544f909 2004-12-25 devnull 	if(--threadnproc == threadnsysproc)
242 4dbefdd4 2004-12-27 devnull 		threadexitsall(p->msg);
243 1544f909 2004-12-25 devnull 	unlock(&threadnproclock);
244 1544f909 2004-12-25 devnull 	unlock(&p->lock);
245 1544f909 2004-12-25 devnull 	free(p);
246 1544f909 2004-12-25 devnull 	setproc(0);
247 1544f909 2004-12-25 devnull }
248 1544f909 2004-12-25 devnull 
249 1544f909 2004-12-25 devnull void
250 1544f909 2004-12-25 devnull _threadsetsysproc(void)
251 1544f909 2004-12-25 devnull {
252 1544f909 2004-12-25 devnull 	lock(&threadnproclock);
253 1544f909 2004-12-25 devnull 	if(++threadnsysproc == threadnproc)
254 1544f909 2004-12-25 devnull 		exit(0);
255 1544f909 2004-12-25 devnull 	unlock(&threadnproclock);
256 1544f909 2004-12-25 devnull 	proc()->sysproc = 1;
257 1544f909 2004-12-25 devnull }
258 1544f909 2004-12-25 devnull 
259 4dbefdd4 2004-12-27 devnull void**
260 4dbefdd4 2004-12-27 devnull procdata(void)
261 4dbefdd4 2004-12-27 devnull {
262 4dbefdd4 2004-12-27 devnull 	return &proc()->udata;
263 4dbefdd4 2004-12-27 devnull }
264 4dbefdd4 2004-12-27 devnull 
265 4dbefdd4 2004-12-27 devnull extern Jmp *(*_notejmpbuf)(void);
266 4dbefdd4 2004-12-27 devnull static Jmp*
267 4dbefdd4 2004-12-27 devnull threadnotejmp(void)
268 4dbefdd4 2004-12-27 devnull {
269 4dbefdd4 2004-12-27 devnull 	return &proc()->sigjmp;
270 4dbefdd4 2004-12-27 devnull }
271 4dbefdd4 2004-12-27 devnull 
272 1544f909 2004-12-25 devnull /*
273 1544f909 2004-12-25 devnull  * debugging
274 1544f909 2004-12-25 devnull  */
275 1544f909 2004-12-25 devnull void
276 1544f909 2004-12-25 devnull threadsetname(char *fmt, ...)
277 1544f909 2004-12-25 devnull {
278 1544f909 2004-12-25 devnull 	va_list arg;
279 1544f909 2004-12-25 devnull 	_Thread *t;
280 1544f909 2004-12-25 devnull 
281 1544f909 2004-12-25 devnull 	t = proc()->thread;
282 1544f909 2004-12-25 devnull 	va_start(arg, fmt);
283 1544f909 2004-12-25 devnull 	vsnprint(t->name, sizeof t->name, fmt, arg);
284 1544f909 2004-12-25 devnull 	va_end(arg);
285 1544f909 2004-12-25 devnull }
286 1544f909 2004-12-25 devnull 
287 1544f909 2004-12-25 devnull void
288 1544f909 2004-12-25 devnull threadsetstate(char *fmt, ...)
289 1544f909 2004-12-25 devnull {
290 1544f909 2004-12-25 devnull 	va_list arg;
291 1544f909 2004-12-25 devnull 	_Thread *t;
292 1544f909 2004-12-25 devnull 
293 1544f909 2004-12-25 devnull 	t = proc()->thread;
294 1544f909 2004-12-25 devnull 	va_start(arg, fmt);
295 1544f909 2004-12-25 devnull 	vsnprint(t->state, sizeof t->name, fmt, arg);
296 1544f909 2004-12-25 devnull 	va_end(arg);
297 1544f909 2004-12-25 devnull }
298 1544f909 2004-12-25 devnull 
299 1544f909 2004-12-25 devnull /*
300 1544f909 2004-12-25 devnull  * locking
301 1544f909 2004-12-25 devnull  */
302 1544f909 2004-12-25 devnull static int
303 1544f909 2004-12-25 devnull threadqlock(QLock *l, int block, ulong pc)
304 1544f909 2004-12-25 devnull {
305 1544f909 2004-12-25 devnull 	lock(&l->l);
306 1544f909 2004-12-25 devnull 	if(l->owner == nil){
307 1544f909 2004-12-25 devnull 		l->owner = (*threadnow)();
308 1544f909 2004-12-25 devnull //print("qlock %p @%#x by %p\n", l, pc, l->owner);
309 1544f909 2004-12-25 devnull 		unlock(&l->l);
310 1544f909 2004-12-25 devnull 		return 1;
311 1544f909 2004-12-25 devnull 	}
312 1544f909 2004-12-25 devnull 	if(!block){
313 1544f909 2004-12-25 devnull 		unlock(&l->l);
314 1544f909 2004-12-25 devnull 		return 0;
315 1544f909 2004-12-25 devnull 	}
316 1544f909 2004-12-25 devnull //print("qsleep %p @%#x by %p\n", l, pc, (*threadnow)());
317 1544f909 2004-12-25 devnull 	addthread(&l->waiting, (*threadnow)());
318 1544f909 2004-12-25 devnull 	unlock(&l->l);
319 1544f909 2004-12-25 devnull 
320 1544f909 2004-12-25 devnull 	_threadswitch();
321 1544f909 2004-12-25 devnull 
322 1544f909 2004-12-25 devnull 	if(l->owner != (*threadnow)()){
323 1544f909 2004-12-25 devnull 		fprint(2, "qlock pc=0x%lux owner=%p self=%p oops\n", pc, l->owner, (*threadnow)());
324 1544f909 2004-12-25 devnull 		abort();
325 1544f909 2004-12-25 devnull 	}
326 1544f909 2004-12-25 devnull //print("qlock wakeup %p @%#x by %p\n", l, pc, (*threadnow)());
327 1544f909 2004-12-25 devnull 	return 1;
328 1544f909 2004-12-25 devnull }
329 1544f909 2004-12-25 devnull 
330 1544f909 2004-12-25 devnull static void
331 1544f909 2004-12-25 devnull threadqunlock(QLock *l, ulong pc)
332 1544f909 2004-12-25 devnull {
333 1544f909 2004-12-25 devnull 	lock(&l->l);
334 1544f909 2004-12-25 devnull //print("qlock unlock %p @%#x by %p (owner %p)\n", l, pc, (*threadnow)(), l->owner);
335 1544f909 2004-12-25 devnull 	if(l->owner == nil){
336 1544f909 2004-12-25 devnull 		fprint(2, "qunlock pc=0x%lux owner=%p self=%p oops\n",
337 1544f909 2004-12-25 devnull 			pc, l->owner, (*threadnow)());
338 1544f909 2004-12-25 devnull 		abort();
339 1544f909 2004-12-25 devnull 	}
340 1544f909 2004-12-25 devnull 	if((l->owner = l->waiting.head) != nil){
341 1544f909 2004-12-25 devnull 		delthread(&l->waiting, l->owner);
342 1544f909 2004-12-25 devnull 		_threadready(l->owner);
343 1544f909 2004-12-25 devnull 	}
344 1544f909 2004-12-25 devnull 	unlock(&l->l);
345 1544f909 2004-12-25 devnull }
346 1544f909 2004-12-25 devnull 
347 1544f909 2004-12-25 devnull static int
348 1544f909 2004-12-25 devnull threadrlock(RWLock *l, int block, ulong pc)
349 1544f909 2004-12-25 devnull {
350 1544f909 2004-12-25 devnull 	USED(pc);
351 1544f909 2004-12-25 devnull 
352 1544f909 2004-12-25 devnull 	lock(&l->l);
353 1544f909 2004-12-25 devnull 	if(l->writer == nil && l->wwaiting.head == nil){
354 1544f909 2004-12-25 devnull 		l->readers++;
355 1544f909 2004-12-25 devnull 		unlock(&l->l);
356 1544f909 2004-12-25 devnull 		return 1;
357 1544f909 2004-12-25 devnull 	}
358 1544f909 2004-12-25 devnull 	if(!block){
359 1544f909 2004-12-25 devnull 		unlock(&l->l);
360 1544f909 2004-12-25 devnull 		return 0;
361 1544f909 2004-12-25 devnull 	}
362 1544f909 2004-12-25 devnull 	addthread(&l->rwaiting, (*threadnow)());
363 1544f909 2004-12-25 devnull 	unlock(&l->l);
364 1544f909 2004-12-25 devnull 	_threadswitch();
365 1544f909 2004-12-25 devnull 	return 1;	
366 1544f909 2004-12-25 devnull }
367 1544f909 2004-12-25 devnull 
368 1544f909 2004-12-25 devnull static int
369 1544f909 2004-12-25 devnull threadwlock(RWLock *l, int block, ulong pc)
370 1544f909 2004-12-25 devnull {
371 1544f909 2004-12-25 devnull 	USED(pc);
372 1544f909 2004-12-25 devnull 
373 1544f909 2004-12-25 devnull 	lock(&l->l);
374 1544f909 2004-12-25 devnull 	if(l->writer == nil && l->readers == 0){
375 1544f909 2004-12-25 devnull 		l->writer = (*threadnow)();
376 1544f909 2004-12-25 devnull 		unlock(&l->l);
377 1544f909 2004-12-25 devnull 		return 1;
378 1544f909 2004-12-25 devnull 	}
379 1544f909 2004-12-25 devnull 	if(!block){
380 1544f909 2004-12-25 devnull 		unlock(&l->l);
381 1544f909 2004-12-25 devnull 		return 0;
382 1544f909 2004-12-25 devnull 	}
383 1544f909 2004-12-25 devnull 	addthread(&l->wwaiting, (*threadnow)());
384 1544f909 2004-12-25 devnull 	unlock(&l->l);
385 1544f909 2004-12-25 devnull 	_threadswitch();
386 1544f909 2004-12-25 devnull 	return 1;
387 1544f909 2004-12-25 devnull }
388 1544f909 2004-12-25 devnull 
389 1544f909 2004-12-25 devnull static void
390 1544f909 2004-12-25 devnull threadrunlock(RWLock *l, ulong pc)
391 1544f909 2004-12-25 devnull {
392 1544f909 2004-12-25 devnull 	_Thread *t;
393 1544f909 2004-12-25 devnull 
394 1544f909 2004-12-25 devnull 	USED(pc);
395 1544f909 2004-12-25 devnull 	lock(&l->l);
396 1544f909 2004-12-25 devnull 	--l->readers;
397 1544f909 2004-12-25 devnull 	if(l->readers == 0 && (t = l->wwaiting.head) != nil){
398 1544f909 2004-12-25 devnull 		delthread(&l->wwaiting, t);
399 1544f909 2004-12-25 devnull 		l->writer = t;
400 1544f909 2004-12-25 devnull 		_threadready(t);
401 1544f909 2004-12-25 devnull 	}
402 1544f909 2004-12-25 devnull 	unlock(&l->l);
403 1544f909 2004-12-25 devnull }
404 1544f909 2004-12-25 devnull 
405 1544f909 2004-12-25 devnull static void
406 1544f909 2004-12-25 devnull threadwunlock(RWLock *l, ulong pc)
407 1544f909 2004-12-25 devnull {
408 1544f909 2004-12-25 devnull 	_Thread *t;
409 1544f909 2004-12-25 devnull 
410 1544f909 2004-12-25 devnull 	USED(pc);
411 1544f909 2004-12-25 devnull 	lock(&l->l);
412 1544f909 2004-12-25 devnull 	l->writer = nil;
413 1544f909 2004-12-25 devnull 	assert(l->readers == 0);
414 1544f909 2004-12-25 devnull 	while((t = l->rwaiting.head) != nil){
415 1544f909 2004-12-25 devnull 		delthread(&l->rwaiting, t);
416 1544f909 2004-12-25 devnull 		l->readers++;
417 1544f909 2004-12-25 devnull 		_threadready(t);
418 1544f909 2004-12-25 devnull 	}
419 1544f909 2004-12-25 devnull 	if(l->readers == 0 && (t = l->wwaiting.head) != nil){
420 1544f909 2004-12-25 devnull 		delthread(&l->wwaiting, t);
421 1544f909 2004-12-25 devnull 		l->writer = t;
422 1544f909 2004-12-25 devnull 		_threadready(t);
423 1544f909 2004-12-25 devnull 	}
424 1544f909 2004-12-25 devnull 	unlock(&l->l);
425 1544f909 2004-12-25 devnull }
426 1544f909 2004-12-25 devnull 
427 1544f909 2004-12-25 devnull /*
428 1544f909 2004-12-25 devnull  * sleep and wakeup
429 1544f909 2004-12-25 devnull  */
430 1544f909 2004-12-25 devnull static void
431 1544f909 2004-12-25 devnull threadrsleep(Rendez *r, ulong pc)
432 1544f909 2004-12-25 devnull {
433 1544f909 2004-12-25 devnull 	addthread(&r->waiting, proc()->thread);
434 1544f909 2004-12-25 devnull 	qunlock(r->l);
435 1544f909 2004-12-25 devnull 	_threadswitch();
436 1544f909 2004-12-25 devnull 	qlock(r->l);
437 1544f909 2004-12-25 devnull }
438 1544f909 2004-12-25 devnull 
439 1544f909 2004-12-25 devnull static int
440 1544f909 2004-12-25 devnull threadrwakeup(Rendez *r, int all, ulong pc)
441 1544f909 2004-12-25 devnull {
442 1544f909 2004-12-25 devnull 	int i;
443 1544f909 2004-12-25 devnull 	_Thread *t;
444 1544f909 2004-12-25 devnull 
445 1544f909 2004-12-25 devnull 	for(i=0;; i++){
446 1544f909 2004-12-25 devnull 		if(i==1 && !all)
447 1544f909 2004-12-25 devnull 			break;
448 1544f909 2004-12-25 devnull 		if((t = r->waiting.head) == nil)
449 1544f909 2004-12-25 devnull 			break;
450 1544f909 2004-12-25 devnull 		delthread(&r->waiting, t);
451 1544f909 2004-12-25 devnull 		_threadready(t);	
452 1544f909 2004-12-25 devnull 	}
453 1544f909 2004-12-25 devnull 	return i;
454 1544f909 2004-12-25 devnull }
455 1544f909 2004-12-25 devnull 
456 1544f909 2004-12-25 devnull /*
457 4dbefdd4 2004-12-27 devnull  * startup
458 4dbefdd4 2004-12-27 devnull  */
459 4dbefdd4 2004-12-27 devnull 
460 4dbefdd4 2004-12-27 devnull static int threadargc;
461 4dbefdd4 2004-12-27 devnull static char **threadargv;
462 4dbefdd4 2004-12-27 devnull int mainstacksize;
463 4dbefdd4 2004-12-27 devnull 
464 4dbefdd4 2004-12-27 devnull static void
465 4dbefdd4 2004-12-27 devnull threadmainstart(void *v)
466 4dbefdd4 2004-12-27 devnull {
467 4dbefdd4 2004-12-27 devnull 	USED(v);
468 4dbefdd4 2004-12-27 devnull 	threadmainproc = proc();
469 4dbefdd4 2004-12-27 devnull 	threadmain(threadargc, threadargv);
470 2c87dda8 2004-12-28 devnull }
471 2c87dda8 2004-12-28 devnull 
472 2c87dda8 2004-12-28 devnull void
473 2c87dda8 2004-12-28 devnull threadlinklibrary(void)
474 2c87dda8 2004-12-28 devnull {
475 4dbefdd4 2004-12-27 devnull }
476 4dbefdd4 2004-12-27 devnull 
477 4dbefdd4 2004-12-27 devnull int
478 4dbefdd4 2004-12-27 devnull main(int argc, char **argv)
479 4dbefdd4 2004-12-27 devnull {
480 4dbefdd4 2004-12-27 devnull 	Proc *p;
481 4dbefdd4 2004-12-27 devnull 
482 2c87dda8 2004-12-28 devnull 	_threadsetupdaemonize();
483 2c87dda8 2004-12-28 devnull 
484 4dbefdd4 2004-12-27 devnull 	threadargc = argc;
485 4dbefdd4 2004-12-27 devnull 	threadargv = argv;
486 4dbefdd4 2004-12-27 devnull 
487 4dbefdd4 2004-12-27 devnull 	/*
488 4dbefdd4 2004-12-27 devnull 	 * Install locking routines into C library.
489 4dbefdd4 2004-12-27 devnull 	 */
490 4dbefdd4 2004-12-27 devnull 	_lock = _threadlock;
491 4dbefdd4 2004-12-27 devnull 	_unlock = _threadunlock;
492 4dbefdd4 2004-12-27 devnull 	_qlock = threadqlock;
493 4dbefdd4 2004-12-27 devnull 	_qunlock = threadqunlock;
494 4dbefdd4 2004-12-27 devnull 	_rlock = threadrlock;
495 4dbefdd4 2004-12-27 devnull 	_runlock = threadrunlock;
496 4dbefdd4 2004-12-27 devnull 	_wlock = threadwlock;
497 4dbefdd4 2004-12-27 devnull 	_wunlock = threadwunlock;
498 4dbefdd4 2004-12-27 devnull 	_rsleep = threadrsleep;
499 4dbefdd4 2004-12-27 devnull 	_rwakeup = threadrwakeup;
500 4dbefdd4 2004-12-27 devnull 	_notejmpbuf = threadnotejmp;
501 4dbefdd4 2004-12-27 devnull 
502 4dbefdd4 2004-12-27 devnull 	_pthreadinit();
503 4dbefdd4 2004-12-27 devnull 	p = procalloc();
504 4dbefdd4 2004-12-27 devnull 	_threadsetproc(p);
505 4dbefdd4 2004-12-27 devnull 	if(mainstacksize == 0)
506 4dbefdd4 2004-12-27 devnull 		mainstacksize = 65536;
507 4dbefdd4 2004-12-27 devnull 	_threadcreate(p, threadmainstart, nil, mainstacksize);
508 4dbefdd4 2004-12-27 devnull 	scheduler(p);
509 4dbefdd4 2004-12-27 devnull 	return 0;	/* not reached */
510 4dbefdd4 2004-12-27 devnull }
511 4dbefdd4 2004-12-27 devnull 
512 4dbefdd4 2004-12-27 devnull /*
513 1544f909 2004-12-25 devnull  * hooray for linked lists
514 1544f909 2004-12-25 devnull  */
515 1544f909 2004-12-25 devnull static void
516 1544f909 2004-12-25 devnull addthread(_Threadlist *l, _Thread *t)
517 1544f909 2004-12-25 devnull {
518 1544f909 2004-12-25 devnull 	if(l->tail){
519 1544f909 2004-12-25 devnull 		l->tail->next = t;
520 1544f909 2004-12-25 devnull 		t->prev = l->tail;
521 1544f909 2004-12-25 devnull 	}else{
522 1544f909 2004-12-25 devnull 		l->head = t;
523 1544f909 2004-12-25 devnull 		t->prev = nil;
524 1544f909 2004-12-25 devnull 	}
525 1544f909 2004-12-25 devnull 	l->tail = t;
526 1544f909 2004-12-25 devnull 	t->next = nil;
527 1544f909 2004-12-25 devnull }
528 1544f909 2004-12-25 devnull 
529 1544f909 2004-12-25 devnull static void
530 1544f909 2004-12-25 devnull delthread(_Threadlist *l, _Thread *t)
531 1544f909 2004-12-25 devnull {
532 1544f909 2004-12-25 devnull 	if(t->prev)
533 1544f909 2004-12-25 devnull 		t->prev->next = t->next;
534 1544f909 2004-12-25 devnull 	else
535 1544f909 2004-12-25 devnull 		l->head = t->next;
536 1544f909 2004-12-25 devnull 	if(t->next)
537 1544f909 2004-12-25 devnull 		t->next->prev = t->prev;
538 1544f909 2004-12-25 devnull 	else
539 1544f909 2004-12-25 devnull 		l->tail = t->prev;
540 1544f909 2004-12-25 devnull }
541 1544f909 2004-12-25 devnull 
542 1544f909 2004-12-25 devnull static void
543 1544f909 2004-12-25 devnull addthreadinproc(Proc *p, _Thread *t)
544 1544f909 2004-12-25 devnull {
545 1544f909 2004-12-25 devnull 	_Threadlist *l;
546 1544f909 2004-12-25 devnull 
547 1544f909 2004-12-25 devnull 	l = &p->allthreads;
548 1544f909 2004-12-25 devnull 	if(l->tail){
549 1544f909 2004-12-25 devnull 		l->tail->allnext = t;
550 1544f909 2004-12-25 devnull 		t->allprev = l->tail;
551 1544f909 2004-12-25 devnull 	}else{
552 1544f909 2004-12-25 devnull 		l->head = t;
553 1544f909 2004-12-25 devnull 		t->allprev = nil;
554 1544f909 2004-12-25 devnull 	}
555 1544f909 2004-12-25 devnull 	l->tail = t;
556 1544f909 2004-12-25 devnull 	t->allnext = nil;
557 1544f909 2004-12-25 devnull }
558 1544f909 2004-12-25 devnull 
559 1544f909 2004-12-25 devnull static void
560 1544f909 2004-12-25 devnull delthreadinproc(Proc *p, _Thread *t)
561 1544f909 2004-12-25 devnull {
562 1544f909 2004-12-25 devnull 	_Threadlist *l;
563 1544f909 2004-12-25 devnull 
564 1544f909 2004-12-25 devnull 	l = &p->allthreads;
565 1544f909 2004-12-25 devnull 	if(t->allprev)
566 1544f909 2004-12-25 devnull 		t->allprev->allnext = t->allnext;
567 1544f909 2004-12-25 devnull 	else
568 1544f909 2004-12-25 devnull 		l->head = t->allnext;
569 1544f909 2004-12-25 devnull 	if(t->allnext)
570 1544f909 2004-12-25 devnull 		t->allnext->allprev = t->allprev;
571 1544f909 2004-12-25 devnull 	else
572 1544f909 2004-12-25 devnull 		l->tail = t->allprev;
573 1544f909 2004-12-25 devnull }
574 1544f909 2004-12-25 devnull 
575 4dbefdd4 2004-12-27 devnull Proc *_threadprocs;
576 4dbefdd4 2004-12-27 devnull Lock _threadprocslock;
577 4dbefdd4 2004-12-27 devnull static Proc *_threadprocstail;
578 1544f909 2004-12-25 devnull 
579 1544f909 2004-12-25 devnull static void
580 4dbefdd4 2004-12-27 devnull addproc(Proc *p)
581 1544f909 2004-12-25 devnull {
582 4dbefdd4 2004-12-27 devnull 	lock(&_threadprocslock);
583 4dbefdd4 2004-12-27 devnull 	if(_threadprocstail){
584 4dbefdd4 2004-12-27 devnull 		_threadprocstail->next = p;
585 4dbefdd4 2004-12-27 devnull 		p->prev = _threadprocstail;
586 4dbefdd4 2004-12-27 devnull 	}else{
587 4dbefdd4 2004-12-27 devnull 		_threadprocs = p;
588 4dbefdd4 2004-12-27 devnull 		p->prev = nil;
589 4dbefdd4 2004-12-27 devnull 	}
590 4dbefdd4 2004-12-27 devnull 	_threadprocstail = p;
591 4dbefdd4 2004-12-27 devnull 	p->next = nil;
592 4dbefdd4 2004-12-27 devnull 	unlock(&_threadprocslock);
593 e1dc7e45 2004-12-27 devnull }
594 e1dc7e45 2004-12-27 devnull 
595 4dbefdd4 2004-12-27 devnull static void
596 4dbefdd4 2004-12-27 devnull delproc(Proc *p)
597 e1dc7e45 2004-12-27 devnull {
598 4dbefdd4 2004-12-27 devnull 	lock(&_threadprocslock);
599 4dbefdd4 2004-12-27 devnull 	if(p->prev)
600 4dbefdd4 2004-12-27 devnull 		p->prev->next = p->next;
601 4dbefdd4 2004-12-27 devnull 	else
602 4dbefdd4 2004-12-27 devnull 		_threadprocs = p->next;
603 4dbefdd4 2004-12-27 devnull 	if(p->next)
604 4dbefdd4 2004-12-27 devnull 		p->next->prev = p->prev;
605 4dbefdd4 2004-12-27 devnull 	else
606 4dbefdd4 2004-12-27 devnull 		_threadprocstail = p->prev;
607 4dbefdd4 2004-12-27 devnull 	unlock(&_threadprocslock);
608 1544f909 2004-12-25 devnull }