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 b0aecb4b 2019-10-29 crossd int foreground=0;
 11 b8c14089 2003-11-23 devnull char	*plumbfile;
 12 b8c14089 2003-11-23 devnull char *user;
 13 b8c14089 2003-11-23 devnull char *home;
 14 b8c14089 2003-11-23 devnull char *progname;
 15 b8c14089 2003-11-23 devnull Ruleset **rules;
 16 b8c14089 2003-11-23 devnull int	printerrors=1;
 17 b8c14089 2003-11-23 devnull jmp_buf	parsejmp;
 18 b8c14089 2003-11-23 devnull char	*lasterror;
 19 b8c14089 2003-11-23 devnull 
 20 b8c14089 2003-11-23 devnull void
 21 b8c14089 2003-11-23 devnull makeports(Ruleset *rules[])
 22 b8c14089 2003-11-23 devnull {
 23 b8c14089 2003-11-23 devnull 	int i;
 24 b8c14089 2003-11-23 devnull 
 25 b8c14089 2003-11-23 devnull 	for(i=0; rules[i]; i++)
 26 b8c14089 2003-11-23 devnull 		addport(rules[i]->port);
 27 b8c14089 2003-11-23 devnull }
 28 b8c14089 2003-11-23 devnull 
 29 b8c14089 2003-11-23 devnull void
 30 b8c14089 2003-11-23 devnull threadmain(int argc, char *argv[])
 31 b8c14089 2003-11-23 devnull {
 32 b8c14089 2003-11-23 devnull 	char buf[512];
 33 f7b74c17 2004-12-28 devnull 	int fd;
 34 b8c14089 2003-11-23 devnull 
 35 b8c14089 2003-11-23 devnull 	progname = "plumber";
 36 b8c14089 2003-11-23 devnull 
 37 b8c14089 2003-11-23 devnull 	ARGBEGIN{
 38 32f69c36 2003-12-11 devnull 	case 'd':
 39 32f69c36 2003-12-11 devnull 		debug = 1;
 40 32f69c36 2003-12-11 devnull 		break;
 41 b0aecb4b 2019-10-29 crossd 	case 'f':
 42 b0aecb4b 2019-10-29 crossd 		foreground = 1;
 43 b0aecb4b 2019-10-29 crossd 		break;
 44 b8c14089 2003-11-23 devnull 	case 'p':
 45 b8c14089 2003-11-23 devnull 		plumbfile = ARGF();
 46 b8c14089 2003-11-23 devnull 		break;
 47 b8c14089 2003-11-23 devnull 	}ARGEND
 48 b8c14089 2003-11-23 devnull 
 49 32f69c36 2003-12-11 devnull 	user = getuser();
 50 c8b6342d 2005-01-13 devnull 	home = getenv("HOME");
 51 b8c14089 2003-11-23 devnull 	if(user==nil || home==nil)
 52 b8c14089 2003-11-23 devnull 		error("can't initialize $user or $home: %r");
 53 b8c14089 2003-11-23 devnull 	if(plumbfile == nil){
 54 b8c14089 2003-11-23 devnull 		sprint(buf, "%s/lib/plumbing", home);
 55 8ad51794 2004-03-25 devnull 		if(access(buf, 0) >= 0)
 56 8ad51794 2004-03-25 devnull 			plumbfile = estrdup(buf);
 57 8ad51794 2004-03-25 devnull 		else
 58 8ad51794 2004-03-25 devnull 			plumbfile = unsharp("#9/plumb/initial.plumbing");
 59 b8c14089 2003-11-23 devnull 	}
 60 b8c14089 2003-11-23 devnull 
 61 b8c14089 2003-11-23 devnull 	fd = open(plumbfile, OREAD);
 62 b8c14089 2003-11-23 devnull 	if(fd < 0)
 63 b8c14089 2003-11-23 devnull 		error("can't open rules file %s: %r", plumbfile);
 64 b8c14089 2003-11-23 devnull 	if(setjmp(parsejmp))
 65 b8c14089 2003-11-23 devnull 		error("parse error");
 66 b8c14089 2003-11-23 devnull 
 67 b8c14089 2003-11-23 devnull 	rules = readrules(plumbfile, fd);
 68 b8c14089 2003-11-23 devnull 	close(fd);
 69 b8c14089 2003-11-23 devnull 
 70 df49b303 2004-12-28 devnull 	/*
 71 df49b303 2004-12-28 devnull 	 * Start all processes and threads from other proc
 72 df49b303 2004-12-28 devnull 	 * so we (main pid) can return to user.
 73 df49b303 2004-12-28 devnull 	 */
 74 f4b26b64 2005-01-04 devnull 	printerrors = 0;
 75 f4b26b64 2005-01-04 devnull 	makeports(rules);
 76 b0aecb4b 2019-10-29 crossd 	startfsys(foreground);
 77 df49b303 2004-12-28 devnull 	threadexits(nil);
 78 b8c14089 2003-11-23 devnull }
 79 b8c14089 2003-11-23 devnull 
 80 b8c14089 2003-11-23 devnull void
 81 b8c14089 2003-11-23 devnull error(char *fmt, ...)
 82 b8c14089 2003-11-23 devnull {
 83 b8c14089 2003-11-23 devnull 	char buf[512];
 84 b8c14089 2003-11-23 devnull 	va_list args;
 85 b8c14089 2003-11-23 devnull 
 86 b8c14089 2003-11-23 devnull 	va_start(args, fmt);
 87 b8c14089 2003-11-23 devnull 	vseprint(buf, buf+sizeof buf, fmt, args);
 88 b8c14089 2003-11-23 devnull 	va_end(args);
 89 b8c14089 2003-11-23 devnull 
 90 b8c14089 2003-11-23 devnull 	fprint(2, "%s: %s\n", progname, buf);
 91 b8c14089 2003-11-23 devnull 	threadexitsall("error");
 92 b8c14089 2003-11-23 devnull }
 93 b8c14089 2003-11-23 devnull 
 94 b8c14089 2003-11-23 devnull void
 95 b8c14089 2003-11-23 devnull parseerror(char *fmt, ...)
 96 b8c14089 2003-11-23 devnull {
 97 b8c14089 2003-11-23 devnull 	char buf[512];
 98 b8c14089 2003-11-23 devnull 	va_list args;
 99 b8c14089 2003-11-23 devnull 
100 b8c14089 2003-11-23 devnull 	va_start(args, fmt);
101 b8c14089 2003-11-23 devnull 	vseprint(buf, buf+sizeof buf, fmt, args);
102 b8c14089 2003-11-23 devnull 	va_end(args);
103 b8c14089 2003-11-23 devnull 
104 b8c14089 2003-11-23 devnull 	if(printerrors){
105 b8c14089 2003-11-23 devnull 		printinputstack();
106 b8c14089 2003-11-23 devnull 		fprint(2, "%s\n", buf);
107 b8c14089 2003-11-23 devnull 	}
108 b8c14089 2003-11-23 devnull 	do; while(popinput());
109 b8c14089 2003-11-23 devnull 	lasterror = estrdup(buf);
110 b8c14089 2003-11-23 devnull 	longjmp(parsejmp, 1);
111 b8c14089 2003-11-23 devnull }
112 b8c14089 2003-11-23 devnull 
113 b8c14089 2003-11-23 devnull void*
114 b8c14089 2003-11-23 devnull emalloc(long n)
115 b8c14089 2003-11-23 devnull {
116 b8c14089 2003-11-23 devnull 	void *p;
117 b8c14089 2003-11-23 devnull 
118 b8c14089 2003-11-23 devnull 	p = malloc(n);
119 b8c14089 2003-11-23 devnull 	if(p == nil)
120 b8c14089 2003-11-23 devnull 		error("malloc failed: %r");
121 b8c14089 2003-11-23 devnull 	memset(p, 0, n);
122 b8c14089 2003-11-23 devnull 	return p;
123 b8c14089 2003-11-23 devnull }
124 b8c14089 2003-11-23 devnull 
125 b8c14089 2003-11-23 devnull void*
126 b8c14089 2003-11-23 devnull erealloc(void *p, long n)
127 b8c14089 2003-11-23 devnull {
128 b8c14089 2003-11-23 devnull 	p = realloc(p, n);
129 b8c14089 2003-11-23 devnull 	if(p == nil)
130 b8c14089 2003-11-23 devnull 		error("realloc failed: %r");
131 b8c14089 2003-11-23 devnull 	return p;
132 b8c14089 2003-11-23 devnull }
133 b8c14089 2003-11-23 devnull 
134 b8c14089 2003-11-23 devnull char*
135 b8c14089 2003-11-23 devnull estrdup(char *s)
136 b8c14089 2003-11-23 devnull {
137 b8c14089 2003-11-23 devnull 	char *t;
138 b8c14089 2003-11-23 devnull 
139 b8c14089 2003-11-23 devnull 	t = strdup(s);
140 b8c14089 2003-11-23 devnull 	if(t == nil)
141 b8c14089 2003-11-23 devnull 		error("estrdup failed: %r");
142 b8c14089 2003-11-23 devnull 	return t;
143 b8c14089 2003-11-23 devnull }