Blame

Date:
Message:
do threading right
Actions:
History | Blob | Raw File
  1 b8c14089 2003-11-23 devnull #include <u.h>

  2 b8c14089 2003-11-23 devnull #include <libc.h>

  3 b8c14089 2003-11-23 devnull #include <regexp.h>

  4 b8c14089 2003-11-23 devnull #include <thread.h>

  5 b8c14089 2003-11-23 devnull #include <plumb.h>

  6 b8c14089 2003-11-23 devnull #include <fcall.h>

  7 b8c14089 2003-11-23 devnull #include "plumber.h"

  8 b8c14089 2003-11-23 devnull 

  9 32f69c36 2003-12-11 devnull int debug;

 10 b8c14089 2003-11-23 devnull char	*plumbfile;

 11 b8c14089 2003-11-23 devnull char *user;

 12 b8c14089 2003-11-23 devnull char *home;

 13 b8c14089 2003-11-23 devnull char *progname;

 14 b8c14089 2003-11-23 devnull Ruleset **rules;

 15 b8c14089 2003-11-23 devnull int	printerrors=1;

 16 b8c14089 2003-11-23 devnull jmp_buf	parsejmp;

 17 b8c14089 2003-11-23 devnull char	*lasterror;

 18 b8c14089 2003-11-23 devnull 

 19 b8c14089 2003-11-23 devnull void

 20 b8c14089 2003-11-23 devnull makeports(Ruleset *rules[])

 21 b8c14089 2003-11-23 devnull {

 22 b8c14089 2003-11-23 devnull 	int i;

 23 b8c14089 2003-11-23 devnull 

 24 b8c14089 2003-11-23 devnull 	for(i=0; rules[i]; i++)

 25 b8c14089 2003-11-23 devnull 		addport(rules[i]->port);

 26 b8c14089 2003-11-23 devnull }

 27 b8c14089 2003-11-23 devnull 

 28 b8c14089 2003-11-23 devnull void

 29 b8c14089 2003-11-23 devnull threadmain(int argc, char *argv[])

 30 b8c14089 2003-11-23 devnull {

 31 b8c14089 2003-11-23 devnull 	char buf[512];

 32 f7b74c17 2004-12-28 devnull 	int fd;

 33 b8c14089 2003-11-23 devnull 

 34 b8c14089 2003-11-23 devnull 	progname = "plumber";

 35 b8c14089 2003-11-23 devnull 

 36 b8c14089 2003-11-23 devnull 	ARGBEGIN{

 37 32f69c36 2003-12-11 devnull 	case 'd':

 38 32f69c36 2003-12-11 devnull 		debug = 1;

 39 32f69c36 2003-12-11 devnull 		break;

 40 b8c14089 2003-11-23 devnull 	case 'p':

 41 b8c14089 2003-11-23 devnull 		plumbfile = ARGF();

 42 b8c14089 2003-11-23 devnull 		break;

 43 b8c14089 2003-11-23 devnull 	}ARGEND

 44 b8c14089 2003-11-23 devnull 

 45 32f69c36 2003-12-11 devnull 	user = getuser();

 46 b8c14089 2003-11-23 devnull 	home = getenv("home");

 47 32f69c36 2003-12-11 devnull 	if(home == nil)

 48 32f69c36 2003-12-11 devnull 		home = getenv("HOME");

 49 b8c14089 2003-11-23 devnull 	if(user==nil || home==nil)

 50 b8c14089 2003-11-23 devnull 		error("can't initialize $user or $home: %r");

 51 b8c14089 2003-11-23 devnull 	if(plumbfile == nil){

 52 b8c14089 2003-11-23 devnull 		sprint(buf, "%s/lib/plumbing", home);

 53 8ad51794 2004-03-25 devnull 		if(access(buf, 0) >= 0)

 54 8ad51794 2004-03-25 devnull 			plumbfile = estrdup(buf);

 55 8ad51794 2004-03-25 devnull 		else

 56 8ad51794 2004-03-25 devnull 			plumbfile = unsharp("#9/plumb/initial.plumbing");

 57 b8c14089 2003-11-23 devnull 	}

 58 b8c14089 2003-11-23 devnull 

 59 b8c14089 2003-11-23 devnull 	fd = open(plumbfile, OREAD);

 60 b8c14089 2003-11-23 devnull 	if(fd < 0)

 61 b8c14089 2003-11-23 devnull 		error("can't open rules file %s: %r", plumbfile);

 62 b8c14089 2003-11-23 devnull 	if(setjmp(parsejmp))

 63 b8c14089 2003-11-23 devnull 		error("parse error");

 64 b8c14089 2003-11-23 devnull 

 65 b8c14089 2003-11-23 devnull 	rules = readrules(plumbfile, fd);

 66 b8c14089 2003-11-23 devnull 	close(fd);

 67 b8c14089 2003-11-23 devnull 

 68 df49b303 2004-12-28 devnull 	/*

 69 df49b303 2004-12-28 devnull 	 * Start all processes and threads from other proc

 70 df49b303 2004-12-28 devnull 	 * so we (main pid) can return to user.

 71 df49b303 2004-12-28 devnull 	 */

 72 f4b26b64 2005-01-04 devnull 	printerrors = 0;

 73 f4b26b64 2005-01-04 devnull 	makeports(rules);

 74 f4b26b64 2005-01-04 devnull 	startfsys();

 75 df49b303 2004-12-28 devnull 	threadexits(nil);

 76 b8c14089 2003-11-23 devnull }

 77 b8c14089 2003-11-23 devnull 

 78 b8c14089 2003-11-23 devnull void

 79 b8c14089 2003-11-23 devnull error(char *fmt, ...)

 80 b8c14089 2003-11-23 devnull {

 81 b8c14089 2003-11-23 devnull 	char buf[512];

 82 b8c14089 2003-11-23 devnull 	va_list args;

 83 b8c14089 2003-11-23 devnull 

 84 b8c14089 2003-11-23 devnull 	va_start(args, fmt);

 85 b8c14089 2003-11-23 devnull 	vseprint(buf, buf+sizeof buf, fmt, args);

 86 b8c14089 2003-11-23 devnull 	va_end(args);

 87 b8c14089 2003-11-23 devnull 

 88 b8c14089 2003-11-23 devnull 	fprint(2, "%s: %s\n", progname, buf);

 89 b8c14089 2003-11-23 devnull 	threadexitsall("error");

 90 b8c14089 2003-11-23 devnull }

 91 b8c14089 2003-11-23 devnull 

 92 b8c14089 2003-11-23 devnull void

 93 b8c14089 2003-11-23 devnull parseerror(char *fmt, ...)

 94 b8c14089 2003-11-23 devnull {

 95 b8c14089 2003-11-23 devnull 	char buf[512];

 96 b8c14089 2003-11-23 devnull 	va_list args;

 97 b8c14089 2003-11-23 devnull 

 98 b8c14089 2003-11-23 devnull 	va_start(args, fmt);

 99 b8c14089 2003-11-23 devnull 	vseprint(buf, buf+sizeof buf, fmt, args);

100 b8c14089 2003-11-23 devnull 	va_end(args);

101 b8c14089 2003-11-23 devnull 

102 b8c14089 2003-11-23 devnull 	if(printerrors){

103 b8c14089 2003-11-23 devnull 		printinputstack();

104 b8c14089 2003-11-23 devnull 		fprint(2, "%s\n", buf);

105 b8c14089 2003-11-23 devnull 	}

106 b8c14089 2003-11-23 devnull 	do; while(popinput());

107 b8c14089 2003-11-23 devnull 	lasterror = estrdup(buf);

108 b8c14089 2003-11-23 devnull 	longjmp(parsejmp, 1);

109 b8c14089 2003-11-23 devnull }

110 b8c14089 2003-11-23 devnull 

111 b8c14089 2003-11-23 devnull void*

112 b8c14089 2003-11-23 devnull emalloc(long n)

113 b8c14089 2003-11-23 devnull {

114 b8c14089 2003-11-23 devnull 	void *p;

115 b8c14089 2003-11-23 devnull 

116 b8c14089 2003-11-23 devnull 	p = malloc(n);

117 b8c14089 2003-11-23 devnull 	if(p == nil)

118 b8c14089 2003-11-23 devnull 		error("malloc failed: %r");

119 b8c14089 2003-11-23 devnull 	memset(p, 0, n);

120 b8c14089 2003-11-23 devnull 	return p;

121 b8c14089 2003-11-23 devnull }

122 b8c14089 2003-11-23 devnull 

123 b8c14089 2003-11-23 devnull void*

124 b8c14089 2003-11-23 devnull erealloc(void *p, long n)

125 b8c14089 2003-11-23 devnull {

126 b8c14089 2003-11-23 devnull 	p = realloc(p, n);

127 b8c14089 2003-11-23 devnull 	if(p == nil)

128 b8c14089 2003-11-23 devnull 		error("realloc failed: %r");

129 b8c14089 2003-11-23 devnull 	return p;

130 b8c14089 2003-11-23 devnull }

131 b8c14089 2003-11-23 devnull 

132 b8c14089 2003-11-23 devnull char*

133 b8c14089 2003-11-23 devnull estrdup(char *s)

134 b8c14089 2003-11-23 devnull {

135 b8c14089 2003-11-23 devnull 	char *t;

136 b8c14089 2003-11-23 devnull 

137 b8c14089 2003-11-23 devnull 	t = strdup(s);

138 b8c14089 2003-11-23 devnull 	if(t == nil)

139 b8c14089 2003-11-23 devnull 		error("estrdup failed: %r");

140 b8c14089 2003-11-23 devnull 	return t;

141 b8c14089 2003-11-23 devnull }