Blame


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 c8b6342d 2005-01-13 devnull home = getenv("HOME");
47 b8c14089 2003-11-23 devnull if(user==nil || home==nil)
48 b8c14089 2003-11-23 devnull error("can't initialize $user or $home: %r");
49 b8c14089 2003-11-23 devnull if(plumbfile == nil){
50 b8c14089 2003-11-23 devnull sprint(buf, "%s/lib/plumbing", home);
51 8ad51794 2004-03-25 devnull if(access(buf, 0) >= 0)
52 8ad51794 2004-03-25 devnull plumbfile = estrdup(buf);
53 8ad51794 2004-03-25 devnull else
54 8ad51794 2004-03-25 devnull plumbfile = unsharp("#9/plumb/initial.plumbing");
55 b8c14089 2003-11-23 devnull }
56 b8c14089 2003-11-23 devnull
57 b8c14089 2003-11-23 devnull fd = open(plumbfile, OREAD);
58 b8c14089 2003-11-23 devnull if(fd < 0)
59 b8c14089 2003-11-23 devnull error("can't open rules file %s: %r", plumbfile);
60 b8c14089 2003-11-23 devnull if(setjmp(parsejmp))
61 b8c14089 2003-11-23 devnull error("parse error");
62 b8c14089 2003-11-23 devnull
63 b8c14089 2003-11-23 devnull rules = readrules(plumbfile, fd);
64 b8c14089 2003-11-23 devnull close(fd);
65 b8c14089 2003-11-23 devnull
66 df49b303 2004-12-28 devnull /*
67 df49b303 2004-12-28 devnull * Start all processes and threads from other proc
68 df49b303 2004-12-28 devnull * so we (main pid) can return to user.
69 df49b303 2004-12-28 devnull */
70 f4b26b64 2005-01-04 devnull printerrors = 0;
71 f4b26b64 2005-01-04 devnull makeports(rules);
72 f4b26b64 2005-01-04 devnull startfsys();
73 df49b303 2004-12-28 devnull threadexits(nil);
74 b8c14089 2003-11-23 devnull }
75 b8c14089 2003-11-23 devnull
76 b8c14089 2003-11-23 devnull void
77 b8c14089 2003-11-23 devnull error(char *fmt, ...)
78 b8c14089 2003-11-23 devnull {
79 b8c14089 2003-11-23 devnull char buf[512];
80 b8c14089 2003-11-23 devnull va_list args;
81 b8c14089 2003-11-23 devnull
82 b8c14089 2003-11-23 devnull va_start(args, fmt);
83 b8c14089 2003-11-23 devnull vseprint(buf, buf+sizeof buf, fmt, args);
84 b8c14089 2003-11-23 devnull va_end(args);
85 b8c14089 2003-11-23 devnull
86 b8c14089 2003-11-23 devnull fprint(2, "%s: %s\n", progname, buf);
87 b8c14089 2003-11-23 devnull threadexitsall("error");
88 b8c14089 2003-11-23 devnull }
89 b8c14089 2003-11-23 devnull
90 b8c14089 2003-11-23 devnull void
91 b8c14089 2003-11-23 devnull parseerror(char *fmt, ...)
92 b8c14089 2003-11-23 devnull {
93 b8c14089 2003-11-23 devnull char buf[512];
94 b8c14089 2003-11-23 devnull va_list args;
95 b8c14089 2003-11-23 devnull
96 b8c14089 2003-11-23 devnull va_start(args, fmt);
97 b8c14089 2003-11-23 devnull vseprint(buf, buf+sizeof buf, fmt, args);
98 b8c14089 2003-11-23 devnull va_end(args);
99 b8c14089 2003-11-23 devnull
100 b8c14089 2003-11-23 devnull if(printerrors){
101 b8c14089 2003-11-23 devnull printinputstack();
102 b8c14089 2003-11-23 devnull fprint(2, "%s\n", buf);
103 b8c14089 2003-11-23 devnull }
104 b8c14089 2003-11-23 devnull do; while(popinput());
105 b8c14089 2003-11-23 devnull lasterror = estrdup(buf);
106 b8c14089 2003-11-23 devnull longjmp(parsejmp, 1);
107 b8c14089 2003-11-23 devnull }
108 b8c14089 2003-11-23 devnull
109 b8c14089 2003-11-23 devnull void*
110 b8c14089 2003-11-23 devnull emalloc(long n)
111 b8c14089 2003-11-23 devnull {
112 b8c14089 2003-11-23 devnull void *p;
113 b8c14089 2003-11-23 devnull
114 b8c14089 2003-11-23 devnull p = malloc(n);
115 b8c14089 2003-11-23 devnull if(p == nil)
116 b8c14089 2003-11-23 devnull error("malloc failed: %r");
117 b8c14089 2003-11-23 devnull memset(p, 0, n);
118 b8c14089 2003-11-23 devnull return p;
119 b8c14089 2003-11-23 devnull }
120 b8c14089 2003-11-23 devnull
121 b8c14089 2003-11-23 devnull void*
122 b8c14089 2003-11-23 devnull erealloc(void *p, long n)
123 b8c14089 2003-11-23 devnull {
124 b8c14089 2003-11-23 devnull p = realloc(p, n);
125 b8c14089 2003-11-23 devnull if(p == nil)
126 b8c14089 2003-11-23 devnull error("realloc failed: %r");
127 b8c14089 2003-11-23 devnull return p;
128 b8c14089 2003-11-23 devnull }
129 b8c14089 2003-11-23 devnull
130 b8c14089 2003-11-23 devnull char*
131 b8c14089 2003-11-23 devnull estrdup(char *s)
132 b8c14089 2003-11-23 devnull {
133 b8c14089 2003-11-23 devnull char *t;
134 b8c14089 2003-11-23 devnull
135 b8c14089 2003-11-23 devnull t = strdup(s);
136 b8c14089 2003-11-23 devnull if(t == nil)
137 b8c14089 2003-11-23 devnull error("estrdup failed: %r");
138 b8c14089 2003-11-23 devnull return t;
139 b8c14089 2003-11-23 devnull }