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