1 3880f020 2005-03-18 devnull #include <u.h>
  2 3880f020 2005-03-18 devnull #include <libc.h>
  3 3880f020 2005-03-18 devnull #include <auth.h>
  4 3880f020 2005-03-18 devnull #include <fcall.h>
  5 3880f020 2005-03-18 devnull #include <thread.h>
  6 3880f020 2005-03-18 devnull 
  7 a2660887 2008-05-07 rsc int post9p(int, char*);
  8 5c84c448 2005-03-21 devnull int debug;
  9 5c84c448 2005-03-21 devnull char *aname = "";
 10 5c84c448 2005-03-21 devnull char *keypattern = "";
 11 5c84c448 2005-03-21 devnull int fd;
 12 5c84c448 2005-03-21 devnull int msize;
 13 5c84c448 2005-03-21 devnull int doauth;
 14 34e8254f 2005-07-27 devnull u32int afid = NOFID;
 15 5c84c448 2005-03-21 devnull extern char *post9parg;	/* clumsy hack */
 16 5c84c448 2005-03-21 devnull void xauth(void);
 17 5c84c448 2005-03-21 devnull AuthInfo* xauth_proxy(AuthGetkey *getkey, char *fmt, ...);
 18 5c84c448 2005-03-21 devnull 
 19 3880f020 2005-03-18 devnull void
 20 3880f020 2005-03-18 devnull usage(void)
 21 3880f020 2005-03-18 devnull {
 22 5c84c448 2005-03-21 devnull 	fprint(2, "usage: srv [-a] [-A aname] [-k keypattern] addr [srvname]\n");
 23 3880f020 2005-03-18 devnull 	threadexitsall("usage");
 24 3880f020 2005-03-18 devnull }
 25 3880f020 2005-03-18 devnull 
 26 3880f020 2005-03-18 devnull void
 27 3880f020 2005-03-18 devnull threadmain(int argc, char **argv)
 28 3880f020 2005-03-18 devnull {
 29 3880f020 2005-03-18 devnull 	char *addr, *service;
 30 5c84c448 2005-03-21 devnull 
 31 5c84c448 2005-03-21 devnull 	fmtinstall('F', fcallfmt);
 32 5c84c448 2005-03-21 devnull 	fmtinstall('M', dirmodefmt);
 33 fa325e9b 2020-01-10 cross 
 34 3880f020 2005-03-18 devnull 	ARGBEGIN{
 35 5c84c448 2005-03-21 devnull 	case 'D':
 36 5c84c448 2005-03-21 devnull 		debug = 1;
 37 5c84c448 2005-03-21 devnull 		break;
 38 5c84c448 2005-03-21 devnull 	case 'A':
 39 5c84c448 2005-03-21 devnull 		/* BUG: should be able to repeat this and establish multiple afids */
 40 5c84c448 2005-03-21 devnull 		aname = EARGF(usage());
 41 5c84c448 2005-03-21 devnull 		break;
 42 5c84c448 2005-03-21 devnull 	case 'a':
 43 5c84c448 2005-03-21 devnull 		doauth = 1;
 44 5c84c448 2005-03-21 devnull 		break;
 45 151b86a0 2007-10-12 rsc 	case 'n':
 46 151b86a0 2007-10-12 rsc 		doauth = -1;
 47 151b86a0 2007-10-12 rsc 		break;
 48 5c84c448 2005-03-21 devnull 	case 'k':
 49 5c84c448 2005-03-21 devnull 		keypattern = EARGF(usage());
 50 5c84c448 2005-03-21 devnull 		break;
 51 3880f020 2005-03-18 devnull 	default:
 52 3880f020 2005-03-18 devnull 		usage();
 53 3880f020 2005-03-18 devnull 	}ARGEND
 54 fa325e9b 2020-01-10 cross 
 55 3880f020 2005-03-18 devnull 	if(argc != 1 && argc != 2)
 56 3880f020 2005-03-18 devnull 		usage();
 57 3880f020 2005-03-18 devnull 
 58 3880f020 2005-03-18 devnull 	addr = netmkaddr(argv[0], "tcp", "9fs");
 59 3880f020 2005-03-18 devnull 	if((fd = dial(addr, nil, nil, nil)) < 0)
 60 3880f020 2005-03-18 devnull 		sysfatal("dial %s: %r", addr);
 61 5c84c448 2005-03-21 devnull 
 62 151b86a0 2007-10-12 rsc 	if(doauth > 0)
 63 5c84c448 2005-03-21 devnull 		xauth();
 64 5c84c448 2005-03-21 devnull 
 65 3880f020 2005-03-18 devnull 	if(argc == 2)
 66 3880f020 2005-03-18 devnull 		service = argv[1];
 67 3880f020 2005-03-18 devnull 	else
 68 3880f020 2005-03-18 devnull 		service = argv[0];
 69 5c84c448 2005-03-21 devnull 
 70 c48193be 2006-02-04 devnull 	rfork(RFNOTEG);
 71 a2660887 2008-05-07 rsc 	if(post9p(fd, service) < 0)
 72 a2660887 2008-05-07 rsc 		sysfatal("post9p: %r");
 73 3880f020 2005-03-18 devnull 
 74 3880f020 2005-03-18 devnull 	threadexitsall(0);
 75 3880f020 2005-03-18 devnull }
 76 5c84c448 2005-03-21 devnull 
 77 5c84c448 2005-03-21 devnull void
 78 5c84c448 2005-03-21 devnull do9p(Fcall *tx, Fcall *rx)
 79 5c84c448 2005-03-21 devnull {
 80 5c84c448 2005-03-21 devnull 	static uchar buf[9000];
 81 5c84c448 2005-03-21 devnull 	static char ebuf[200];
 82 5c84c448 2005-03-21 devnull 	int n;
 83 fa325e9b 2020-01-10 cross 
 84 5c84c448 2005-03-21 devnull 	n = convS2M(tx, buf, sizeof buf);
 85 5c84c448 2005-03-21 devnull 	if(n == BIT16SZ){
 86 5c84c448 2005-03-21 devnull 		werrstr("convS2M failed");
 87 5c84c448 2005-03-21 devnull 		goto err;
 88 5c84c448 2005-03-21 devnull 	}
 89 5c84c448 2005-03-21 devnull 	if(debug)
 90 5c84c448 2005-03-21 devnull 		fprint(2, "<- %F\n", tx);
 91 5c84c448 2005-03-21 devnull 	if(write(fd, buf, n) != n)
 92 5c84c448 2005-03-21 devnull 		goto err;
 93 5c84c448 2005-03-21 devnull 	if((n = read9pmsg(fd, buf, sizeof buf)) < 0)
 94 5c84c448 2005-03-21 devnull 		goto err;
 95 5c84c448 2005-03-21 devnull 	if(n == 0){
 96 5c84c448 2005-03-21 devnull 		werrstr("unexpected eof");
 97 5c84c448 2005-03-21 devnull 		goto err;
 98 5c84c448 2005-03-21 devnull 	}
 99 5c84c448 2005-03-21 devnull 	if(convM2S(buf, n, rx) != n){
100 5c84c448 2005-03-21 devnull 		werrstr("convM2S failed");
101 5c84c448 2005-03-21 devnull 		goto err;
102 5c84c448 2005-03-21 devnull 	}
103 5c84c448 2005-03-21 devnull 	if(debug)
104 5c84c448 2005-03-21 devnull 		fprint(2, "-> %F\n", rx);
105 5c84c448 2005-03-21 devnull 	if(rx->type != Rerror && rx->type != tx->type+1){
106 5c84c448 2005-03-21 devnull 		werrstr("unexpected type");
107 5c84c448 2005-03-21 devnull 		goto err;
108 5c84c448 2005-03-21 devnull 	}
109 5c84c448 2005-03-21 devnull 	if(rx->tag != tx->tag){
110 5c84c448 2005-03-21 devnull 		werrstr("unexpected tag");
111 5c84c448 2005-03-21 devnull 		goto err;
112 5c84c448 2005-03-21 devnull 	}
113 5c84c448 2005-03-21 devnull 	return;
114 5c84c448 2005-03-21 devnull 
115 5c84c448 2005-03-21 devnull err:
116 5c84c448 2005-03-21 devnull 	rerrstr(ebuf, sizeof ebuf);
117 5c84c448 2005-03-21 devnull 	rx->ename = ebuf;
118 5c84c448 2005-03-21 devnull 	rx->type = Rerror;
119 5c84c448 2005-03-21 devnull 	return;
120 5c84c448 2005-03-21 devnull }
121 5c84c448 2005-03-21 devnull 
122 5c84c448 2005-03-21 devnull void
123 5c84c448 2005-03-21 devnull xauth(void)
124 5c84c448 2005-03-21 devnull {
125 5c84c448 2005-03-21 devnull 	Fcall tx, rx;
126 5c84c448 2005-03-21 devnull 
127 5c84c448 2005-03-21 devnull 	afid = 0;
128 5c84c448 2005-03-21 devnull 	tx.type = Tversion;
129 5c84c448 2005-03-21 devnull 	tx.tag = NOTAG;
130 5c84c448 2005-03-21 devnull 	tx.version = "9P2000";
131 5c84c448 2005-03-21 devnull 	tx.msize = 8192;
132 5c84c448 2005-03-21 devnull 	do9p(&tx, &rx);
133 5c84c448 2005-03-21 devnull 	if(rx.type == Rerror)
134 5c84c448 2005-03-21 devnull 		sysfatal("Tversion: %s", rx.ename);
135 5c84c448 2005-03-21 devnull 	msize = rx.msize;
136 fa325e9b 2020-01-10 cross 
137 5c84c448 2005-03-21 devnull 	tx.type = Tauth;
138 5c84c448 2005-03-21 devnull 	tx.tag = 1;
139 5c84c448 2005-03-21 devnull 	tx.afid = afid;
140 5c84c448 2005-03-21 devnull 	tx.uname = getuser();
141 5c84c448 2005-03-21 devnull 	tx.aname = aname;
142 5c84c448 2005-03-21 devnull 	do9p(&tx, &rx);
143 5c84c448 2005-03-21 devnull 	if(rx.type == Rerror){
144 5c84c448 2005-03-21 devnull 		fprint(2, "rx: %s\n", rx.ename);
145 5c84c448 2005-03-21 devnull 		afid = NOFID;
146 5c84c448 2005-03-21 devnull 		return;
147 5c84c448 2005-03-21 devnull 	}
148 5c84c448 2005-03-21 devnull 
149 78378021 2005-05-07 devnull 	if(xauth_proxy(auth_getkey, "proto=p9any role=client %s", keypattern) == nil)
150 5c84c448 2005-03-21 devnull 		sysfatal("authproxy: %r");
151 5c84c448 2005-03-21 devnull }
152 5c84c448 2005-03-21 devnull 
153 5c84c448 2005-03-21 devnull int
154 5c84c448 2005-03-21 devnull xread(void *buf, int n)
155 5c84c448 2005-03-21 devnull {
156 5c84c448 2005-03-21 devnull 	Fcall tx, rx;
157 fa325e9b 2020-01-10 cross 
158 5c84c448 2005-03-21 devnull 	tx.type = Tread;
159 5c84c448 2005-03-21 devnull 	tx.tag = 1;
160 5c84c448 2005-03-21 devnull 	tx.fid = 0;	/* afid above */
161 5c84c448 2005-03-21 devnull 	tx.count = n;
162 5c84c448 2005-03-21 devnull 	tx.offset = 0;
163 5c84c448 2005-03-21 devnull 	do9p(&tx, &rx);
164 5c84c448 2005-03-21 devnull 	if(rx.type == Rerror){
165 5c84c448 2005-03-21 devnull 		werrstr("%s", rx.ename);
166 5c84c448 2005-03-21 devnull 		return -1;
167 5c84c448 2005-03-21 devnull 	}
168 fa325e9b 2020-01-10 cross 
169 5c84c448 2005-03-21 devnull 	if(rx.count > n){
170 5c84c448 2005-03-21 devnull 		werrstr("too much data returned");
171 5c84c448 2005-03-21 devnull 		return -1;
172 5c84c448 2005-03-21 devnull 	}
173 5c84c448 2005-03-21 devnull 	memmove(buf, rx.data, rx.count);
174 5c84c448 2005-03-21 devnull 	return rx.count;
175 5c84c448 2005-03-21 devnull }
176 5c84c448 2005-03-21 devnull 
177 5c84c448 2005-03-21 devnull int
178 5c84c448 2005-03-21 devnull xwrite(void *buf, int n)
179 5c84c448 2005-03-21 devnull {
180 5c84c448 2005-03-21 devnull 	Fcall tx, rx;
181 fa325e9b 2020-01-10 cross 
182 5c84c448 2005-03-21 devnull 	tx.type = Twrite;
183 5c84c448 2005-03-21 devnull 	tx.tag = 1;
184 5c84c448 2005-03-21 devnull 	tx.fid = 0;	/* afid above */
185 5c84c448 2005-03-21 devnull 	tx.data = buf;
186 5c84c448 2005-03-21 devnull 	tx.count = n;
187 5c84c448 2005-03-21 devnull 	tx.offset = 0;
188 5c84c448 2005-03-21 devnull 	do9p(&tx, &rx);
189 5c84c448 2005-03-21 devnull 	if(rx.type == Rerror){
190 5c84c448 2005-03-21 devnull 		werrstr("%s", rx.ename);
191 5c84c448 2005-03-21 devnull 		return -1;
192 5c84c448 2005-03-21 devnull 	}
193 5c84c448 2005-03-21 devnull 	return n;
194 5c84c448 2005-03-21 devnull }
195 5c84c448 2005-03-21 devnull 
196 5c84c448 2005-03-21 devnull 
197 5c84c448 2005-03-21 devnull /*
198 5c84c448 2005-03-21 devnull  * changed to add -A below
199 5c84c448 2005-03-21 devnull  */
200 5c84c448 2005-03-21 devnull #undef _exits
201 5c84c448 2005-03-21 devnull int
202 a2660887 2008-05-07 rsc post9p(int fd, char *name)
203 5c84c448 2005-03-21 devnull {
204 5c84c448 2005-03-21 devnull 	int i;
205 5c84c448 2005-03-21 devnull 	char *ns, *s;
206 5c84c448 2005-03-21 devnull 	Waitmsg *w;
207 5c84c448 2005-03-21 devnull 
208 5c84c448 2005-03-21 devnull 	if((ns = getns()) == nil)
209 5c84c448 2005-03-21 devnull 		return -1;
210 5c84c448 2005-03-21 devnull 
211 5c84c448 2005-03-21 devnull 	s = smprint("unix!%s/%s", ns, name);
212 5c84c448 2005-03-21 devnull 	free(ns);
213 5c84c448 2005-03-21 devnull 	if(s == nil)
214 5c84c448 2005-03-21 devnull 		return -1;
215 5c84c448 2005-03-21 devnull 	switch(fork()){
216 5c84c448 2005-03-21 devnull 	case -1:
217 5c84c448 2005-03-21 devnull 		return -1;
218 5c84c448 2005-03-21 devnull 	case 0:
219 5c84c448 2005-03-21 devnull 		dup(fd, 0);
220 5c84c448 2005-03-21 devnull 		dup(fd, 1);
221 5c84c448 2005-03-21 devnull 		for(i=3; i<20; i++)
222 5c84c448 2005-03-21 devnull 			close(i);
223 151b86a0 2007-10-12 rsc 		if(doauth > 0)
224 5c84c448 2005-03-21 devnull 			execlp("9pserve", "9pserve", "-u",
225 5c84c448 2005-03-21 devnull 				"-M",
226 5c84c448 2005-03-21 devnull 					smprint("%d", msize),
227 5c84c448 2005-03-21 devnull 				"-A",
228 5c84c448 2005-03-21 devnull 					aname,
229 5c84c448 2005-03-21 devnull 					smprint("%d", afid),
230 5c84c448 2005-03-21 devnull 				s, (char*)0);
231 151b86a0 2007-10-12 rsc 		else
232 151b86a0 2007-10-12 rsc 			execlp("9pserve", "9pserve",
233 151b86a0 2007-10-12 rsc 				doauth < 0 ? "-nu" : "-u", s, (char*)0);
234 5c84c448 2005-03-21 devnull 		fprint(2, "exec 9pserve: %r\n");
235 5c84c448 2005-03-21 devnull 		_exits("exec");
236 5c84c448 2005-03-21 devnull 	default:
237 5c84c448 2005-03-21 devnull 		w = wait();
238 5c84c448 2005-03-21 devnull 		if(w == nil)
239 5c84c448 2005-03-21 devnull 			return -1;
240 5c84c448 2005-03-21 devnull 		close(fd);
241 5c84c448 2005-03-21 devnull 		free(s);
242 5c84c448 2005-03-21 devnull 		if(w->msg && w->msg[0]){
243 5c84c448 2005-03-21 devnull 			free(w);
244 5c84c448 2005-03-21 devnull 			werrstr("9pserve failed");
245 5c84c448 2005-03-21 devnull 			return -1;
246 5c84c448 2005-03-21 devnull 		}
247 5c84c448 2005-03-21 devnull 		free(w);
248 5c84c448 2005-03-21 devnull 		return 0;
249 5c84c448 2005-03-21 devnull 	}
250 5c84c448 2005-03-21 devnull }
251 5c84c448 2005-03-21 devnull 
252 5c84c448 2005-03-21 devnull enum { ARgiveup = 100 };
253 5c84c448 2005-03-21 devnull static int
254 5c84c448 2005-03-21 devnull dorpc(AuthRpc *rpc, char *verb, char *val, int len, AuthGetkey *getkey)
255 5c84c448 2005-03-21 devnull {
256 5c84c448 2005-03-21 devnull 	int ret;
257 5c84c448 2005-03-21 devnull 
258 5c84c448 2005-03-21 devnull 	for(;;){
259 5c84c448 2005-03-21 devnull 		if((ret = auth_rpc(rpc, verb, val, len)) != ARneedkey && ret != ARbadkey)
260 5c84c448 2005-03-21 devnull 			return ret;
261 5c84c448 2005-03-21 devnull 		if(getkey == nil)
262 5c84c448 2005-03-21 devnull 			return ARgiveup;	/* don't know how */
263 5c84c448 2005-03-21 devnull 		if((*getkey)(rpc->arg) < 0)
264 5c84c448 2005-03-21 devnull 			return ARgiveup;	/* user punted */
265 5c84c448 2005-03-21 devnull 	}
266 5c84c448 2005-03-21 devnull }
267 5c84c448 2005-03-21 devnull 
268 5c84c448 2005-03-21 devnull 
269 5c84c448 2005-03-21 devnull /*
270 5c84c448 2005-03-21 devnull  *  this just proxies what the factotum tells it to.
271 5c84c448 2005-03-21 devnull  */
272 5c84c448 2005-03-21 devnull AuthInfo*
273 5c84c448 2005-03-21 devnull xfauth_proxy(AuthRpc *rpc, AuthGetkey *getkey, char *params)
274 5c84c448 2005-03-21 devnull {
275 5c84c448 2005-03-21 devnull 	char *buf;
276 5c84c448 2005-03-21 devnull 	int m, n, ret;
277 5c84c448 2005-03-21 devnull 	AuthInfo *a;
278 5c84c448 2005-03-21 devnull 	char oerr[ERRMAX];
279 5c84c448 2005-03-21 devnull 
280 5c84c448 2005-03-21 devnull 	rerrstr(oerr, sizeof oerr);
281 5c84c448 2005-03-21 devnull 	werrstr("UNKNOWN AUTH ERROR");
282 5c84c448 2005-03-21 devnull 
283 5c84c448 2005-03-21 devnull 	if(dorpc(rpc, "start", params, strlen(params), getkey) != ARok){
284 5c84c448 2005-03-21 devnull 		werrstr("fauth_proxy start: %r");
285 5c84c448 2005-03-21 devnull 		return nil;
286 5c84c448 2005-03-21 devnull 	}
287 5c84c448 2005-03-21 devnull 
288 5c84c448 2005-03-21 devnull 	buf = malloc(AuthRpcMax);
289 5c84c448 2005-03-21 devnull 	if(buf == nil)
290 5c84c448 2005-03-21 devnull 		return nil;
291 5c84c448 2005-03-21 devnull 	for(;;){
292 5c84c448 2005-03-21 devnull 		switch(dorpc(rpc, "read", nil, 0, getkey)){
293 5c84c448 2005-03-21 devnull 		case ARdone:
294 5c84c448 2005-03-21 devnull 			free(buf);
295 5c84c448 2005-03-21 devnull 			a = auth_getinfo(rpc);
296 5c84c448 2005-03-21 devnull 			errstr(oerr, sizeof oerr);	/* no error, restore whatever was there */
297 5c84c448 2005-03-21 devnull 			return a;
298 5c84c448 2005-03-21 devnull 		case ARok:
299 5c84c448 2005-03-21 devnull 			if(xwrite(rpc->arg, rpc->narg) != rpc->narg){
300 5c84c448 2005-03-21 devnull 				werrstr("auth_proxy write fid: %r");
301 5c84c448 2005-03-21 devnull 				goto Error;
302 5c84c448 2005-03-21 devnull 			}
303 5c84c448 2005-03-21 devnull 			break;
304 5c84c448 2005-03-21 devnull 		case ARphase:
305 5c84c448 2005-03-21 devnull 			n = 0;
306 5c84c448 2005-03-21 devnull 			memset(buf, 0, AuthRpcMax);
307 5c84c448 2005-03-21 devnull 			while((ret = dorpc(rpc, "write", buf, n, getkey)) == ARtoosmall){
308 5c84c448 2005-03-21 devnull 				if(atoi(rpc->arg) > AuthRpcMax)
309 5c84c448 2005-03-21 devnull 					break;
310 5c84c448 2005-03-21 devnull 				m = xread(buf+n, atoi(rpc->arg)-n);
311 5c84c448 2005-03-21 devnull 				if(m <= 0){
312 5c84c448 2005-03-21 devnull 					if(m == 0)
313 5c84c448 2005-03-21 devnull 						werrstr("auth_proxy short read: %s", buf);
314 5c84c448 2005-03-21 devnull 					goto Error;
315 5c84c448 2005-03-21 devnull 				}
316 5c84c448 2005-03-21 devnull 				n += m;
317 5c84c448 2005-03-21 devnull 			}
318 5c84c448 2005-03-21 devnull 			if(ret != ARok){
319 5c84c448 2005-03-21 devnull 				werrstr("auth_proxy rpc write: %s: %r", buf);
320 5c84c448 2005-03-21 devnull 				goto Error;
321 5c84c448 2005-03-21 devnull 			}
322 5c84c448 2005-03-21 devnull 			break;
323 5c84c448 2005-03-21 devnull 		default:
324 5c84c448 2005-03-21 devnull 			werrstr("auth_proxy rpc: %r");
325 5c84c448 2005-03-21 devnull 			goto Error;
326 5c84c448 2005-03-21 devnull 		}
327 5c84c448 2005-03-21 devnull 	}
328 5c84c448 2005-03-21 devnull Error:
329 5c84c448 2005-03-21 devnull 	free(buf);
330 5c84c448 2005-03-21 devnull 	return nil;
331 5c84c448 2005-03-21 devnull }
332 5c84c448 2005-03-21 devnull 
333 5c84c448 2005-03-21 devnull AuthInfo*
334 5c84c448 2005-03-21 devnull xauth_proxy(AuthGetkey *getkey, char *fmt, ...)
335 5c84c448 2005-03-21 devnull {
336 5c84c448 2005-03-21 devnull 	char *p;
337 5c84c448 2005-03-21 devnull 	va_list arg;
338 5c84c448 2005-03-21 devnull 	AuthInfo *ai;
339 5c84c448 2005-03-21 devnull 	AuthRpc *rpc;
340 5c84c448 2005-03-21 devnull 
341 5c84c448 2005-03-21 devnull 	quotefmtinstall();	/* just in case */
342 5c84c448 2005-03-21 devnull 	va_start(arg, fmt);
343 5c84c448 2005-03-21 devnull 	p = vsmprint(fmt, arg);
344 5c84c448 2005-03-21 devnull 	va_end(arg);
345 5c84c448 2005-03-21 devnull 
346 5c84c448 2005-03-21 devnull 	rpc = auth_allocrpc();
347 5c84c448 2005-03-21 devnull 	if(rpc == nil){
348 5c84c448 2005-03-21 devnull 		free(p);
349 5c84c448 2005-03-21 devnull 		return nil;
350 5c84c448 2005-03-21 devnull 	}
351 5c84c448 2005-03-21 devnull 
352 5c84c448 2005-03-21 devnull 	ai = xfauth_proxy(rpc, getkey, p);
353 5c84c448 2005-03-21 devnull 	free(p);
354 5c84c448 2005-03-21 devnull 	auth_freerpc(rpc);
355 5c84c448 2005-03-21 devnull 	return ai;
356 5c84c448 2005-03-21 devnull }