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 37cc72d3 2005-01-11 devnull #ifdef __sun__		/* 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 86502504 2005-01-18 devnull 	makecontext(&t->context.uc, (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 4dbefdd4 2004-12-27 devnull 	delproc(p);
285 1544f909 2004-12-25 devnull 	lock(&threadnproclock);
286 1544f909 2004-12-25 devnull 	if(p->sysproc)
287 1544f909 2004-12-25 devnull 		--threadnsysproc;
288 1544f909 2004-12-25 devnull 	if(--threadnproc == threadnsysproc)
289 4dbefdd4 2004-12-27 devnull 		threadexitsall(p->msg);
290 1544f909 2004-12-25 devnull 	unlock(&threadnproclock);
291 1544f909 2004-12-25 devnull 	unlock(&p->lock);
292 1544f909 2004-12-25 devnull 	free(p);
293 1544f909 2004-12-25 devnull 	setproc(0);
294 1544f909 2004-12-25 devnull }
295 1544f909 2004-12-25 devnull 
296 1544f909 2004-12-25 devnull void
297 1544f909 2004-12-25 devnull _threadsetsysproc(void)
298 1544f909 2004-12-25 devnull {
299 1544f909 2004-12-25 devnull 	lock(&threadnproclock);
300 1544f909 2004-12-25 devnull 	if(++threadnsysproc == threadnproc)
301 1544f909 2004-12-25 devnull 		exit(0);
302 1544f909 2004-12-25 devnull 	unlock(&threadnproclock);
303 1544f909 2004-12-25 devnull 	proc()->sysproc = 1;
304 1544f909 2004-12-25 devnull }
305 1544f909 2004-12-25 devnull 
306 4dbefdd4 2004-12-27 devnull void**
307 4dbefdd4 2004-12-27 devnull procdata(void)
308 4dbefdd4 2004-12-27 devnull {
309 4dbefdd4 2004-12-27 devnull 	return &proc()->udata;
310 4dbefdd4 2004-12-27 devnull }
311 4dbefdd4 2004-12-27 devnull 
312 4dbefdd4 2004-12-27 devnull extern Jmp *(*_notejmpbuf)(void);
313 4dbefdd4 2004-12-27 devnull static Jmp*
314 4dbefdd4 2004-12-27 devnull threadnotejmp(void)
315 4dbefdd4 2004-12-27 devnull {
316 4dbefdd4 2004-12-27 devnull 	return &proc()->sigjmp;
317 4dbefdd4 2004-12-27 devnull }
318 4dbefdd4 2004-12-27 devnull 
319 1544f909 2004-12-25 devnull /*
320 1544f909 2004-12-25 devnull  * debugging
321 1544f909 2004-12-25 devnull  */
322 1544f909 2004-12-25 devnull void
323 1544f909 2004-12-25 devnull threadsetname(char *fmt, ...)
324 1544f909 2004-12-25 devnull {
325 1544f909 2004-12-25 devnull 	va_list arg;
326 1544f909 2004-12-25 devnull 	_Thread *t;
327 1544f909 2004-12-25 devnull 
328 1544f909 2004-12-25 devnull 	t = proc()->thread;
329 1544f909 2004-12-25 devnull 	va_start(arg, fmt);
330 1544f909 2004-12-25 devnull 	vsnprint(t->name, sizeof t->name, fmt, arg);
331 1544f909 2004-12-25 devnull 	va_end(arg);
332 515b6e5b 2005-01-18 devnull }
333 515b6e5b 2005-01-18 devnull 
334 515b6e5b 2005-01-18 devnull char*
335 515b6e5b 2005-01-18 devnull threadgetname(void)
336 515b6e5b 2005-01-18 devnull {
337 515b6e5b 2005-01-18 devnull 	return proc()->thread->name;
338 1544f909 2004-12-25 devnull }
339 1544f909 2004-12-25 devnull 
340 1544f909 2004-12-25 devnull void
341 1544f909 2004-12-25 devnull threadsetstate(char *fmt, ...)
342 1544f909 2004-12-25 devnull {
343 1544f909 2004-12-25 devnull 	va_list arg;
344 1544f909 2004-12-25 devnull 	_Thread *t;
345 1544f909 2004-12-25 devnull 
346 1544f909 2004-12-25 devnull 	t = proc()->thread;
347 1544f909 2004-12-25 devnull 	va_start(arg, fmt);
348 1544f909 2004-12-25 devnull 	vsnprint(t->state, sizeof t->name, fmt, arg);
349 1544f909 2004-12-25 devnull 	va_end(arg);
350 c8b6342d 2005-01-13 devnull }
351 c8b6342d 2005-01-13 devnull 
352 c8b6342d 2005-01-13 devnull void
353 c8b6342d 2005-01-13 devnull needstack(int n)
354 c8b6342d 2005-01-13 devnull {
355 c8b6342d 2005-01-13 devnull 	_Thread *t;
356 c8b6342d 2005-01-13 devnull 
357 c8b6342d 2005-01-13 devnull 	t = proc()->thread;
358 c8b6342d 2005-01-13 devnull 
359 c8b6342d 2005-01-13 devnull 	if((char*)&t <= (char*)t->stk
360 c8b6342d 2005-01-13 devnull 	|| (char*)&t - (char*)t->stk < 256+n){
361 1fdb4a54 2005-01-23 devnull 		fprint(2, "thread stack overflow: &t=%p tstk=%p n=%d\n", &t, t->stk, 256+n);
362 c8b6342d 2005-01-13 devnull 		abort();
363 c8b6342d 2005-01-13 devnull 	}
364 1544f909 2004-12-25 devnull }
365 1544f909 2004-12-25 devnull 
366 1544f909 2004-12-25 devnull /*
367 1544f909 2004-12-25 devnull  * locking
368 1544f909 2004-12-25 devnull  */
369 1544f909 2004-12-25 devnull static int
370 1544f909 2004-12-25 devnull threadqlock(QLock *l, int block, ulong pc)
371 1544f909 2004-12-25 devnull {
372 b8f742db 2005-01-11 devnull //print("threadqlock %p\n", l);
373 1544f909 2004-12-25 devnull 	lock(&l->l);
374 1544f909 2004-12-25 devnull 	if(l->owner == nil){
375 1544f909 2004-12-25 devnull 		l->owner = (*threadnow)();
376 1544f909 2004-12-25 devnull //print("qlock %p @%#x by %p\n", l, pc, l->owner);
377 1544f909 2004-12-25 devnull 		unlock(&l->l);
378 1544f909 2004-12-25 devnull 		return 1;
379 1544f909 2004-12-25 devnull 	}
380 1544f909 2004-12-25 devnull 	if(!block){
381 1544f909 2004-12-25 devnull 		unlock(&l->l);
382 1544f909 2004-12-25 devnull 		return 0;
383 1544f909 2004-12-25 devnull 	}
384 1544f909 2004-12-25 devnull //print("qsleep %p @%#x by %p\n", l, pc, (*threadnow)());
385 1544f909 2004-12-25 devnull 	addthread(&l->waiting, (*threadnow)());
386 1544f909 2004-12-25 devnull 	unlock(&l->l);
387 1544f909 2004-12-25 devnull 
388 1544f909 2004-12-25 devnull 	_threadswitch();
389 1544f909 2004-12-25 devnull 
390 1544f909 2004-12-25 devnull 	if(l->owner != (*threadnow)()){
391 15cd8255 2005-01-04 devnull 		fprint(2, "%s: qlock pc=0x%lux owner=%p self=%p oops\n",
392 15cd8255 2005-01-04 devnull 			argv0, pc, l->owner, (*threadnow)());
393 1544f909 2004-12-25 devnull 		abort();
394 1544f909 2004-12-25 devnull 	}
395 1544f909 2004-12-25 devnull //print("qlock wakeup %p @%#x by %p\n", l, pc, (*threadnow)());
396 1544f909 2004-12-25 devnull 	return 1;
397 1544f909 2004-12-25 devnull }
398 1544f909 2004-12-25 devnull 
399 1544f909 2004-12-25 devnull static void
400 1544f909 2004-12-25 devnull threadqunlock(QLock *l, ulong pc)
401 1544f909 2004-12-25 devnull {
402 1544f909 2004-12-25 devnull 	lock(&l->l);
403 1544f909 2004-12-25 devnull //print("qlock unlock %p @%#x by %p (owner %p)\n", l, pc, (*threadnow)(), l->owner);
404 1aa9c533 2005-01-07 devnull 	if(l->owner == 0){
405 15cd8255 2005-01-04 devnull 		fprint(2, "%s: qunlock pc=0x%lux owner=%p self=%p oops\n",
406 15cd8255 2005-01-04 devnull 			argv0, pc, l->owner, (*threadnow)());
407 1aa9c533 2005-01-07 devnull 		abort();
408 1544f909 2004-12-25 devnull 	}
409 1544f909 2004-12-25 devnull 	if((l->owner = l->waiting.head) != nil){
410 1544f909 2004-12-25 devnull 		delthread(&l->waiting, l->owner);
411 1544f909 2004-12-25 devnull 		_threadready(l->owner);
412 1544f909 2004-12-25 devnull 	}
413 1544f909 2004-12-25 devnull 	unlock(&l->l);
414 1544f909 2004-12-25 devnull }
415 1544f909 2004-12-25 devnull 
416 1544f909 2004-12-25 devnull static int
417 1544f909 2004-12-25 devnull threadrlock(RWLock *l, int block, ulong pc)
418 1544f909 2004-12-25 devnull {
419 1544f909 2004-12-25 devnull 	USED(pc);
420 1544f909 2004-12-25 devnull 
421 1544f909 2004-12-25 devnull 	lock(&l->l);
422 1544f909 2004-12-25 devnull 	if(l->writer == nil && l->wwaiting.head == nil){
423 1544f909 2004-12-25 devnull 		l->readers++;
424 1544f909 2004-12-25 devnull 		unlock(&l->l);
425 1544f909 2004-12-25 devnull 		return 1;
426 1544f909 2004-12-25 devnull 	}
427 1544f909 2004-12-25 devnull 	if(!block){
428 1544f909 2004-12-25 devnull 		unlock(&l->l);
429 1544f909 2004-12-25 devnull 		return 0;
430 1544f909 2004-12-25 devnull 	}
431 1544f909 2004-12-25 devnull 	addthread(&l->rwaiting, (*threadnow)());
432 1544f909 2004-12-25 devnull 	unlock(&l->l);
433 1544f909 2004-12-25 devnull 	_threadswitch();
434 1544f909 2004-12-25 devnull 	return 1;	
435 1544f909 2004-12-25 devnull }
436 1544f909 2004-12-25 devnull 
437 1544f909 2004-12-25 devnull static int
438 1544f909 2004-12-25 devnull threadwlock(RWLock *l, int block, ulong pc)
439 1544f909 2004-12-25 devnull {
440 1544f909 2004-12-25 devnull 	USED(pc);
441 1544f909 2004-12-25 devnull 
442 1544f909 2004-12-25 devnull 	lock(&l->l);
443 1544f909 2004-12-25 devnull 	if(l->writer == nil && l->readers == 0){
444 1544f909 2004-12-25 devnull 		l->writer = (*threadnow)();
445 1544f909 2004-12-25 devnull 		unlock(&l->l);
446 1544f909 2004-12-25 devnull 		return 1;
447 1544f909 2004-12-25 devnull 	}
448 1544f909 2004-12-25 devnull 	if(!block){
449 1544f909 2004-12-25 devnull 		unlock(&l->l);
450 1544f909 2004-12-25 devnull 		return 0;
451 1544f909 2004-12-25 devnull 	}
452 1544f909 2004-12-25 devnull 	addthread(&l->wwaiting, (*threadnow)());
453 1544f909 2004-12-25 devnull 	unlock(&l->l);
454 1544f909 2004-12-25 devnull 	_threadswitch();
455 1544f909 2004-12-25 devnull 	return 1;
456 1544f909 2004-12-25 devnull }
457 1544f909 2004-12-25 devnull 
458 1544f909 2004-12-25 devnull static void
459 1544f909 2004-12-25 devnull threadrunlock(RWLock *l, ulong pc)
460 1544f909 2004-12-25 devnull {
461 1544f909 2004-12-25 devnull 	_Thread *t;
462 1544f909 2004-12-25 devnull 
463 1544f909 2004-12-25 devnull 	USED(pc);
464 1544f909 2004-12-25 devnull 	lock(&l->l);
465 1544f909 2004-12-25 devnull 	--l->readers;
466 1544f909 2004-12-25 devnull 	if(l->readers == 0 && (t = l->wwaiting.head) != nil){
467 1544f909 2004-12-25 devnull 		delthread(&l->wwaiting, t);
468 1544f909 2004-12-25 devnull 		l->writer = t;
469 1544f909 2004-12-25 devnull 		_threadready(t);
470 1544f909 2004-12-25 devnull 	}
471 1544f909 2004-12-25 devnull 	unlock(&l->l);
472 1544f909 2004-12-25 devnull }
473 1544f909 2004-12-25 devnull 
474 1544f909 2004-12-25 devnull static void
475 1544f909 2004-12-25 devnull threadwunlock(RWLock *l, ulong pc)
476 1544f909 2004-12-25 devnull {
477 1544f909 2004-12-25 devnull 	_Thread *t;
478 1544f909 2004-12-25 devnull 
479 1544f909 2004-12-25 devnull 	USED(pc);
480 1544f909 2004-12-25 devnull 	lock(&l->l);
481 1544f909 2004-12-25 devnull 	l->writer = nil;
482 1544f909 2004-12-25 devnull 	assert(l->readers == 0);
483 1544f909 2004-12-25 devnull 	while((t = l->rwaiting.head) != nil){
484 1544f909 2004-12-25 devnull 		delthread(&l->rwaiting, t);
485 1544f909 2004-12-25 devnull 		l->readers++;
486 1544f909 2004-12-25 devnull 		_threadready(t);
487 1544f909 2004-12-25 devnull 	}
488 1544f909 2004-12-25 devnull 	if(l->readers == 0 && (t = l->wwaiting.head) != nil){
489 1544f909 2004-12-25 devnull 		delthread(&l->wwaiting, t);
490 1544f909 2004-12-25 devnull 		l->writer = t;
491 1544f909 2004-12-25 devnull 		_threadready(t);
492 1544f909 2004-12-25 devnull 	}
493 1544f909 2004-12-25 devnull 	unlock(&l->l);
494 1544f909 2004-12-25 devnull }
495 1544f909 2004-12-25 devnull 
496 1544f909 2004-12-25 devnull /*
497 1544f909 2004-12-25 devnull  * sleep and wakeup
498 1544f909 2004-12-25 devnull  */
499 1544f909 2004-12-25 devnull static void
500 1544f909 2004-12-25 devnull threadrsleep(Rendez *r, ulong pc)
501 1544f909 2004-12-25 devnull {
502 1544f909 2004-12-25 devnull 	addthread(&r->waiting, proc()->thread);
503 1544f909 2004-12-25 devnull 	qunlock(r->l);
504 1544f909 2004-12-25 devnull 	_threadswitch();
505 1544f909 2004-12-25 devnull 	qlock(r->l);
506 1544f909 2004-12-25 devnull }
507 1544f909 2004-12-25 devnull 
508 1544f909 2004-12-25 devnull static int
509 1544f909 2004-12-25 devnull threadrwakeup(Rendez *r, int all, ulong pc)
510 1544f909 2004-12-25 devnull {
511 1544f909 2004-12-25 devnull 	int i;
512 1544f909 2004-12-25 devnull 	_Thread *t;
513 1544f909 2004-12-25 devnull 
514 1544f909 2004-12-25 devnull 	for(i=0;; i++){
515 1544f909 2004-12-25 devnull 		if(i==1 && !all)
516 1544f909 2004-12-25 devnull 			break;
517 1544f909 2004-12-25 devnull 		if((t = r->waiting.head) == nil)
518 1544f909 2004-12-25 devnull 			break;
519 1544f909 2004-12-25 devnull 		delthread(&r->waiting, t);
520 1544f909 2004-12-25 devnull 		_threadready(t);	
521 1544f909 2004-12-25 devnull 	}
522 1544f909 2004-12-25 devnull 	return i;
523 1544f909 2004-12-25 devnull }
524 1544f909 2004-12-25 devnull 
525 1544f909 2004-12-25 devnull /*
526 4dbefdd4 2004-12-27 devnull  * startup
527 4dbefdd4 2004-12-27 devnull  */
528 4dbefdd4 2004-12-27 devnull 
529 4dbefdd4 2004-12-27 devnull static int threadargc;
530 4dbefdd4 2004-12-27 devnull static char **threadargv;
531 4dbefdd4 2004-12-27 devnull int mainstacksize;
532 4dbefdd4 2004-12-27 devnull 
533 4dbefdd4 2004-12-27 devnull static void
534 4dbefdd4 2004-12-27 devnull threadmainstart(void *v)
535 4dbefdd4 2004-12-27 devnull {
536 4dbefdd4 2004-12-27 devnull 	USED(v);
537 0b561695 2005-01-16 devnull 
538 0b561695 2005-01-16 devnull 	/*
539 0b561695 2005-01-16 devnull 	 * N.B. This call to proc() is a program's first call (indirectly) to a
540 0b561695 2005-01-16 devnull 	 * pthreads function while executing on a non-pthreads-allocated
541 0b561695 2005-01-16 devnull 	 * stack.  If the pthreads implementation is using the stack pointer
542 0b561695 2005-01-16 devnull 	 * to locate the per-thread data, then this call will blow up.
543 0b561695 2005-01-16 devnull 	 * This means the pthread implementation is not suitable for
544 0b561695 2005-01-16 devnull 	 * running under libthread.  Time to write your own.  Sorry.
545 0b561695 2005-01-16 devnull 	 */
546 4dbefdd4 2004-12-27 devnull 	threadmainproc = proc();
547 4dbefdd4 2004-12-27 devnull 	threadmain(threadargc, threadargv);
548 2c87dda8 2004-12-28 devnull }
549 2c87dda8 2004-12-28 devnull 
550 4dbefdd4 2004-12-27 devnull int
551 4dbefdd4 2004-12-27 devnull main(int argc, char **argv)
552 4dbefdd4 2004-12-27 devnull {
553 4dbefdd4 2004-12-27 devnull 	Proc *p;
554 1d2533d0 2004-12-28 devnull 
555 1d2533d0 2004-12-28 devnull 	argv0 = argv[0];
556 4dbefdd4 2004-12-27 devnull 
557 2c87dda8 2004-12-28 devnull 	_threadsetupdaemonize();
558 2c87dda8 2004-12-28 devnull 
559 4dbefdd4 2004-12-27 devnull 	threadargc = argc;
560 4dbefdd4 2004-12-27 devnull 	threadargv = argv;
561 4dbefdd4 2004-12-27 devnull 
562 4dbefdd4 2004-12-27 devnull 	/*
563 4dbefdd4 2004-12-27 devnull 	 * Install locking routines into C library.
564 4dbefdd4 2004-12-27 devnull 	 */
565 4dbefdd4 2004-12-27 devnull 	_lock = _threadlock;
566 4dbefdd4 2004-12-27 devnull 	_unlock = _threadunlock;
567 4dbefdd4 2004-12-27 devnull 	_qlock = threadqlock;
568 4dbefdd4 2004-12-27 devnull 	_qunlock = threadqunlock;
569 4dbefdd4 2004-12-27 devnull 	_rlock = threadrlock;
570 4dbefdd4 2004-12-27 devnull 	_runlock = threadrunlock;
571 4dbefdd4 2004-12-27 devnull 	_wlock = threadwlock;
572 4dbefdd4 2004-12-27 devnull 	_wunlock = threadwunlock;
573 4dbefdd4 2004-12-27 devnull 	_rsleep = threadrsleep;
574 4dbefdd4 2004-12-27 devnull 	_rwakeup = threadrwakeup;
575 4dbefdd4 2004-12-27 devnull 	_notejmpbuf = threadnotejmp;
576 4dbefdd4 2004-12-27 devnull 
577 4dbefdd4 2004-12-27 devnull 	_pthreadinit();
578 4dbefdd4 2004-12-27 devnull 	p = procalloc();
579 4dbefdd4 2004-12-27 devnull 	_threadsetproc(p);
580 4dbefdd4 2004-12-27 devnull 	if(mainstacksize == 0)
581 a0e8d02d 2005-01-23 devnull 		mainstacksize = 256*1024;
582 4dbefdd4 2004-12-27 devnull 	_threadcreate(p, threadmainstart, nil, mainstacksize);
583 1fdb4a54 2005-01-23 devnull 	procscheduler(p);
584 15cd8255 2005-01-04 devnull 	_threaddaemonize();
585 8f4871fd 2005-02-13 devnull 	/*
586 8f4871fd 2005-02-13 devnull 	 * On Linux 2.6, if the main thread exits then the others
587 8f4871fd 2005-02-13 devnull 	 * keep running but the process shows up as a zombie in ps
588 8f4871fd 2005-02-13 devnull 	 * and is not attachable with ptrace.  We'll just sit around
589 8f4871fd 2005-02-13 devnull 	 * instead of exiting.
590 8f4871fd 2005-02-13 devnull 	 */
591 8f4871fd 2005-02-13 devnull 	for(;;)
592 8f4871fd 2005-02-13 devnull 		sleep(1000);
593 1d2533d0 2004-12-28 devnull 	_threadpexit();
594 15cd8255 2005-01-04 devnull 	return 0;
595 4dbefdd4 2004-12-27 devnull }
596 4dbefdd4 2004-12-27 devnull 
597 4dbefdd4 2004-12-27 devnull /*
598 1544f909 2004-12-25 devnull  * hooray for linked lists
599 1544f909 2004-12-25 devnull  */
600 1544f909 2004-12-25 devnull static void
601 1544f909 2004-12-25 devnull addthread(_Threadlist *l, _Thread *t)
602 1544f909 2004-12-25 devnull {
603 1544f909 2004-12-25 devnull 	if(l->tail){
604 1544f909 2004-12-25 devnull 		l->tail->next = t;
605 1544f909 2004-12-25 devnull 		t->prev = l->tail;
606 1544f909 2004-12-25 devnull 	}else{
607 1544f909 2004-12-25 devnull 		l->head = t;
608 1544f909 2004-12-25 devnull 		t->prev = nil;
609 1544f909 2004-12-25 devnull 	}
610 1544f909 2004-12-25 devnull 	l->tail = t;
611 1544f909 2004-12-25 devnull 	t->next = nil;
612 1544f909 2004-12-25 devnull }
613 1544f909 2004-12-25 devnull 
614 1544f909 2004-12-25 devnull static void
615 1544f909 2004-12-25 devnull delthread(_Threadlist *l, _Thread *t)
616 1544f909 2004-12-25 devnull {
617 1544f909 2004-12-25 devnull 	if(t->prev)
618 1544f909 2004-12-25 devnull 		t->prev->next = t->next;
619 1544f909 2004-12-25 devnull 	else
620 1544f909 2004-12-25 devnull 		l->head = t->next;
621 1544f909 2004-12-25 devnull 	if(t->next)
622 1544f909 2004-12-25 devnull 		t->next->prev = t->prev;
623 1544f909 2004-12-25 devnull 	else
624 1544f909 2004-12-25 devnull 		l->tail = t->prev;
625 1544f909 2004-12-25 devnull }
626 1544f909 2004-12-25 devnull 
627 1544f909 2004-12-25 devnull static void
628 1544f909 2004-12-25 devnull addthreadinproc(Proc *p, _Thread *t)
629 1544f909 2004-12-25 devnull {
630 1544f909 2004-12-25 devnull 	_Threadlist *l;
631 1544f909 2004-12-25 devnull 
632 1544f909 2004-12-25 devnull 	l = &p->allthreads;
633 1544f909 2004-12-25 devnull 	if(l->tail){
634 1544f909 2004-12-25 devnull 		l->tail->allnext = t;
635 1544f909 2004-12-25 devnull 		t->allprev = 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->allprev = nil;
639 1544f909 2004-12-25 devnull 	}
640 1544f909 2004-12-25 devnull 	l->tail = t;
641 1544f909 2004-12-25 devnull 	t->allnext = 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 delthreadinproc(Proc *p, _Thread *t)
646 1544f909 2004-12-25 devnull {
647 1544f909 2004-12-25 devnull 	_Threadlist *l;
648 1544f909 2004-12-25 devnull 
649 1544f909 2004-12-25 devnull 	l = &p->allthreads;
650 1544f909 2004-12-25 devnull 	if(t->allprev)
651 1544f909 2004-12-25 devnull 		t->allprev->allnext = t->allnext;
652 1544f909 2004-12-25 devnull 	else
653 1544f909 2004-12-25 devnull 		l->head = t->allnext;
654 1544f909 2004-12-25 devnull 	if(t->allnext)
655 1544f909 2004-12-25 devnull 		t->allnext->allprev = t->allprev;
656 1544f909 2004-12-25 devnull 	else
657 1544f909 2004-12-25 devnull 		l->tail = t->allprev;
658 1544f909 2004-12-25 devnull }
659 1544f909 2004-12-25 devnull 
660 4dbefdd4 2004-12-27 devnull Proc *_threadprocs;
661 4dbefdd4 2004-12-27 devnull Lock _threadprocslock;
662 4dbefdd4 2004-12-27 devnull static Proc *_threadprocstail;
663 1544f909 2004-12-25 devnull 
664 1544f909 2004-12-25 devnull static void
665 4dbefdd4 2004-12-27 devnull addproc(Proc *p)
666 1544f909 2004-12-25 devnull {
667 4dbefdd4 2004-12-27 devnull 	lock(&_threadprocslock);
668 4dbefdd4 2004-12-27 devnull 	if(_threadprocstail){
669 4dbefdd4 2004-12-27 devnull 		_threadprocstail->next = p;
670 4dbefdd4 2004-12-27 devnull 		p->prev = _threadprocstail;
671 4dbefdd4 2004-12-27 devnull 	}else{
672 4dbefdd4 2004-12-27 devnull 		_threadprocs = p;
673 4dbefdd4 2004-12-27 devnull 		p->prev = nil;
674 4dbefdd4 2004-12-27 devnull 	}
675 4dbefdd4 2004-12-27 devnull 	_threadprocstail = p;
676 4dbefdd4 2004-12-27 devnull 	p->next = nil;
677 4dbefdd4 2004-12-27 devnull 	unlock(&_threadprocslock);
678 e1dc7e45 2004-12-27 devnull }
679 e1dc7e45 2004-12-27 devnull 
680 4dbefdd4 2004-12-27 devnull static void
681 4dbefdd4 2004-12-27 devnull delproc(Proc *p)
682 e1dc7e45 2004-12-27 devnull {
683 4dbefdd4 2004-12-27 devnull 	lock(&_threadprocslock);
684 4dbefdd4 2004-12-27 devnull 	if(p->prev)
685 4dbefdd4 2004-12-27 devnull 		p->prev->next = p->next;
686 4dbefdd4 2004-12-27 devnull 	else
687 4dbefdd4 2004-12-27 devnull 		_threadprocs = p->next;
688 4dbefdd4 2004-12-27 devnull 	if(p->next)
689 4dbefdd4 2004-12-27 devnull 		p->next->prev = p->prev;
690 4dbefdd4 2004-12-27 devnull 	else
691 4dbefdd4 2004-12-27 devnull 		_threadprocstail = p->prev;
692 4dbefdd4 2004-12-27 devnull 	unlock(&_threadprocslock);
693 1544f909 2004-12-25 devnull }
694 15cd8255 2005-01-04 devnull 
695 15cd8255 2005-01-04 devnull /* 
696 15cd8255 2005-01-04 devnull  * notify - for now just use the usual mechanisms
697 15cd8255 2005-01-04 devnull  */
698 15cd8255 2005-01-04 devnull void
699 15cd8255 2005-01-04 devnull threadnotify(int (*f)(void*, char*), int in)
700 15cd8255 2005-01-04 devnull {
701 15cd8255 2005-01-04 devnull 	atnotify(f, in);
702 15cd8255 2005-01-04 devnull }