Import sources from 2011-03-30 iso image

1 files changed, 224 insertions, 0 deletions

diff --git a/sys/src/libthread/main.c b/sys/src/libthread/main.c
new file mode 100755
index 000000000..72e3a8701
--- /dev/null
+++ b/sys/src/libthread/main.c
@@ -0,0 +1,224 @@
+#include <u.h>
+#include <libc.h>
+#include <thread.h>
+#include "threadimpl.h"
+
+typedef struct Mainarg Mainarg;
+struct Mainarg
+{
+	int	argc;
+	char	**argv;
+};
+
+int	mainstacksize;
+int	_threadnotefd;
+int	_threadpasserpid;
+static jmp_buf _mainjmp;
+static void mainlauncher(void*);
+extern void (*_sysfatal)(char*, va_list);
+extern void (*__assert)(char*);
+extern int (*_dial)(char*, char*, char*, int*);
+
+extern int _threaddial(char*, char*, char*, int*);
+
+static Proc **mainp;
+
+void
+main(int argc, char **argv)
+{
+	Mainarg *a;
+	Proc *p;
+
+	rfork(RFREND);
+	mainp = &p;
+	if(setjmp(_mainjmp))
+		_schedinit(p);
+
+//_threaddebuglevel = (DBGSCHED|DBGCHAN|DBGREND)^~0;
+	_systhreadinit();
+	_qlockinit(_threadrendezvous);
+	_sysfatal = _threadsysfatal;
+	_dial = _threaddial;
+	__assert = _threadassert;
+	notify(_threadnote);
+	if(mainstacksize == 0)
+		mainstacksize = 8*1024;
+
+	a = _threadmalloc(sizeof *a, 1);
+	a->argc = argc;
+	a->argv = argv;
+
+	p = _newproc(mainlauncher, a, mainstacksize, "threadmain", 0, 0);
+	_schedinit(p);
+	abort();	/* not reached */
+}
+
+static void
+mainlauncher(void *arg)
+{
+	Mainarg *a;
+
+	a = arg;
+	threadmain(a->argc, a->argv);
+	threadexits("threadmain");
+}
+
+static char*
+skip(char *p)
+{
+	while(*p == ' ')
+		p++;
+	while(*p != ' ' && *p != 0)
+		p++;
+	return p;
+}
+
+static long
+_times(long *t)
+{
+	char b[200], *p;
+	int f;
+	ulong r;
+
+	memset(b, 0, sizeof(b));
+	f = open("/dev/cputime", OREAD|OCEXEC);
+	if(f < 0)
+		return 0;
+	if(read(f, b, sizeof(b)) <= 0){
+		close(f);
+		return 0;
+	}
+	p = b;
+	if(t)
+		t[0] = atol(p);
+	p = skip(p);
+	if(t)
+		t[1] = atol(p);
+	p = skip(p);
+	r = atol(p);
+	if(t){
+		p = skip(p);
+		t[2] = atol(p);
+		p = skip(p);
+		t[3] = atol(p);
+	}
+	return r;
+}
+
+static void
+efork(Execargs *e)
+{
+	char buf[ERRMAX];
+
+	_threaddebug(DBGEXEC, "_schedexec %s", e->prog);
+	close(e->fd[0]);
+	exec(e->prog, e->args);
+	_threaddebug(DBGEXEC, "_schedexec failed: %r");
+	rerrstr(buf, sizeof buf);
+	if(buf[0]=='\0')
+		strcpy(buf, "exec failed");
+	write(e->fd[1], buf, strlen(buf));
+	close(e->fd[1]);
+	_exits(buf);
+}
+
+int
+_schedexec(Execargs *e)
+{
+	int pid;
+
+	switch(pid = rfork(RFREND|RFNOTEG|RFFDG|RFMEM|RFPROC)){
+	case 0:
+		efork(e);
+	default:
+		return pid;
+	}
+}
+
+int
+_schedfork(Proc *p)
+{
+	int pid;
+
+	switch(pid = rfork(RFPROC|RFMEM|RFNOWAIT|p->rforkflag)){
+	case 0:
+		*mainp = p;	/* write to stack, so local to proc */
+		longjmp(_mainjmp, 1);
+	default:
+		return pid;
+	}
+}
+
+void
+_schedexit(Proc *p)
+{
+	char ex[ERRMAX];
+	Proc **l;
+
+	lock(&_threadpq.lock);
+	for(l=&_threadpq.head; *l; l=&(*l)->next){
+		if(*l == p){
+			*l = p->next;
+			if(*l == nil)
+				_threadpq.tail = l;
+			break;
+		}
+	}
+	unlock(&_threadpq.lock);
+
+	utfecpy(ex, ex+sizeof ex, p->exitstr);
+	free(p);
+	_exits(ex);
+}
+
+void
+_schedexecwait(void)
+{
+	int pid;
+	Channel *c;
+	Proc *p;
+	Thread *t;
+	Waitmsg *w;
+
+	p = _threadgetproc();
+	t = p->thread;
+	pid = t->ret;
+	_threaddebug(DBGEXEC, "_schedexecwait %d", t->ret);
+
+	rfork(RFCFDG);
+	for(;;){
+		w = wait();
+		if(w == nil)
+			break;
+		if(w->pid == pid)
+			break;
+		free(w);
+	}
+	if(w != nil){
+		if((c = _threadwaitchan) != nil)
+			sendp(c, w);
+		else
+			free(w);
+	}
+	threadexits("procexec");
+}
+
+static Proc **procp;
+
+void
+_systhreadinit(void)
+{
+	procp = privalloc();
+}
+
+Proc*
+_threadgetproc(void)
+{
+	return *procp;
+}
+
+void
+_threadsetproc(Proc *p)
+{
+	*procp = p;
+}