1 /* 
2  * Some notes on locking:
3  *
4  *	All the locking woes come from implementing
5  *	threadinterrupt (and threadkill).
6  *
7  *	_threadgetproc()->thread is always a live pointer.
8  *	p->threads, p->ready, and _threadrgrp also contain
9  * 	live thread pointers.  These may only be consulted
10  *	while holding p->lock or _threadrgrp.lock; in procs
11  *	other than p, the pointers are only guaranteed to be live
12  *	while the lock is still being held.
13  *
14  *	Thread structures can only be freed by the proc
15  *	they belong to.  Threads marked with t->inrendez
16  * 	need to be extracted from the _threadrgrp before
17  *	being freed.
18  *
19  *	_threadrgrp.lock cannot be acquired while holding p->lock.
20  */
21 
22 #include <assert.h>
23 #include <lib9.h>
24 #include <thread.h>
25 #include "label.h"
26 
27 typedef struct Thread	Thread;
28 typedef struct Proc	Proc;
29 typedef struct Tqueue	Tqueue;
30 typedef struct Pqueue	Pqueue;
31 typedef struct Rgrp		Rgrp;
32 typedef struct Execargs	Execargs;
33 
34 /* must match list in sched.c */
35 typedef enum
36 {
37 	Dead,
38 	Running,
39 	Ready,
40 	Rendezvous,
41 } State;
42 	
43 typedef enum
44 {
45 	Channone,
46 	Chanalt,
47 	Chansend,
48 	Chanrecv,
49 } Chanstate;
50 
51 enum
52 {
53 	RENDHASH = 10009,
54 	Printsize = 2048,
55 	NPRIV = 8,
56 };
57 
58 struct Rgrp
59 {
60 	Lock		lock;
61 	Thread	*hash[RENDHASH];
62 };
63 
64 struct Tqueue		/* Thread queue */
65 {
66 	int		asleep;
67 	Thread	*head;
68 	Thread	*tail;
69 };
70 
71 struct Thread
72 {
73 	Lock		lock;			/* protects thread data structure */
74 	Label	sched;		/* for context switches */
75 	int		id;			/* thread id */
76 	int 		grp;			/* thread group */
77 	int		moribund;	/* thread needs to die */
78 	State		state;		/* run state */
79 	State		nextstate;		/* next run state */
80 	uchar	*stk;			/* top of stack (lowest address of stack) */
81 	uint		stksize;		/* stack size */
82 	Thread	*next;		/* next on ready queue */
83 
84 	Proc		*proc;		/* proc of this thread */
85 	Thread	*nextt;		/* next on list of threads in this proc */
86 	Thread	*prevt;		/* prev on list of threads in this proc */
87 	int		ret;			/* return value for Exec, Fork */
88 
89 	char		*cmdname;	/* ptr to name of thread */
90 
91 	int		inrendez;
92 	Thread	*rendhash;	/* Trgrp linked list */
93 	ulong	rendtag;		/* rendezvous tag */
94 	ulong	rendval;		/* rendezvous value */
95 	int		rendbreak;	/* rendezvous has been taken */
96 
97 	Chanstate	chan;		/* which channel operation is current */
98 	Alt		*alt;			/* pointer to current alt structure (debugging) */
99 
100 	void*	udata[NPRIV];	/* User per-thread data pointer */
101 };
102 
103 struct Execargs
104 {
105 	char		*prog;
106 	char		**args;
107 	int		fd[2];
108 	int		*stdfd;
109 };
110 
111 struct Proc
112 {
113 	Lock		lock;
114 	Label	sched;		/* for context switches */
115 	Proc		*link;		/* in proctab */
116 	int		pid;			/* process id */
117 	int		splhi;		/* delay notes */
118 	Thread	*thread;		/* running thread */
119 	Thread	*idle;			/* idle thread */
120 
121 	int		needexec;
122 	Execargs	exec;		/* exec argument */
123 	Proc		*newproc;	/* fork argument */
124 	char		exitstr[ERRMAX];	/* exit status */
125 
126 	int		rforkflag;
127 	int		nthreads;
128 	Tqueue	threads;		/* All threads of this proc */
129 	Tqueue	ready;		/* Runnable threads */
130 	Lock		readylock;
131 
132 	char		printbuf[Printsize];
133 	int		blocked;		/* In a rendezvous */
134 	int		pending;		/* delayed note pending */
135 	int		nonotes;		/*  delay notes */
136 	uint		nextID;		/* ID of most recently created thread */
137 	Proc		*next;		/* linked list of Procs */
138 
139 	void		*arg;			/* passed between shared and unshared stk */
140 	char		str[ERRMAX];	/* used by threadexits to avoid malloc */
141 	char		errbuf[ERRMAX];	/* errstr */
142 
143 	void*	udata;		/* User per-proc data pointer */
144 };
145 
146 struct Pqueue {		/* Proc queue */
147 	Lock		lock;
148 	Proc		*head;
149 	Proc		**tail;
150 };
151 
152 struct Ioproc
153 {
154 	int tid;
155 	Channel *c, *creply;
156 	int inuse;
157 	long (*op)(va_list*);
158 	va_list arg;
159 	long ret;
160 	char err[ERRMAX];
161 	Ioproc *next;
162 };
163 
164 void		_gotolabel(Label*);
165 int		_setlabel(Label*);
166 void		_freeproc(Proc*);
167 Proc*	_newproc(void(*)(void*), void*, uint, char*, int, int);
168 int		_procsplhi(void);
169 void		_procsplx(int);
170 void		_sched(void);
171 int		_schedexec(Execargs*);
172 void		_schedexecwait(void);
173 void		_schedexit(Proc*);
174 int		_schedfork(Proc*);
175 void		_schedinit(void*);
176 void		_systhreadinit(void);
177 void		_threadassert(char*);
178 void		_threadbreakrendez(void);
179 void		__threaddebug(ulong, char*, ...);
180 #define _threaddebug if(!_threaddebuglevel){}else __threaddebug
181 void		_threadexitsall(char*);
182 void		_threadflagrendez(Thread*);
183 Proc*	_threadgetproc(void);
184 Proc*	_threaddelproc(void);
185 void		_threadsetproc(Proc*);
186 void		_threadinitstack(Thread*, void(*)(void*), void*);
187 void*	_threadmalloc(long, int);
188 void		_threadnote(void*, char*);
189 void		_threadready(Thread*);
190 void		_threadidle(void);
191 ulong	_threadrendezvous(ulong, ulong);
192 void		_threadsignal(void);
193 void		_threadsysfatal(char*, va_list);
194 long		_xdec(long*);
195 void		_xinc(long*);
196 void		_threadremove(Proc*, Thread*);
197 
198 extern int	_threadmultiproc;
199 extern int			_threaddebuglevel;
200 extern char*		_threadexitsallstatus;
201 extern Pqueue		_threadpq;
202 extern Channel*	_threadwaitchan;
203 extern Rgrp		_threadrgrp;
204 extern void _stackfree(void*);
205 
206 #define DBGAPPL	(1 << 0)
207 #define DBGSCHED	(1 << 16)
208 #define DBGCHAN	(1 << 17)
209 #define DBGREND	(1 << 18)
210 /* #define DBGKILL	(1 << 19) */
211 #define DBGNOTE	(1 << 20)
212 #define DBGEXEC	(1 << 21)
213 
214 #define ioproc_arg(io, type)	(va_arg((io)->arg, type))
215 extern int _threadgetpid(void);
216 extern void _threadmemset(void*, int, int);
217 extern void _threaddebugmemset(void*, int, int);
218 extern int _threadprocs;
219