op public repos

Blob

Date:: Fri Mar 7 15:45:59 2008 UTC
Message:: acme: sync against Plan 9, remove -$ option
Actions:: History | Blame | Raw File
1 #include <u.h>
2 #include <libc.h>
3 #include <draw.h>
4 #include <thread.h>
5 #include <cursor.h>
6 #include <mouse.h>
7 #include <keyboard.h>
8 #include <frame.h>
9 #include <fcall.h>
10 #include <plumb.h>
11 #include "dat.h"
12 #include "fns.h"
13 
14 static	int	sfd;
15 
16 enum
17 {
18 	Nhash	= 16,
19 	DEBUG	= 0
20 };
21 
22 static	Fid	*fids[Nhash];
23 
24 Fid	*newfid(int);
25 
26 static	Xfid*	fsysflush(Xfid*, Fid*);
27 static	Xfid*	fsysauth(Xfid*, Fid*);
28 static	Xfid*	fsysversion(Xfid*, Fid*);
29 static	Xfid*	fsysattach(Xfid*, Fid*);
30 static	Xfid*	fsyswalk(Xfid*, Fid*);
31 static	Xfid*	fsysopen(Xfid*, Fid*);
32 static	Xfid*	fsyscreate(Xfid*, Fid*);
33 static	Xfid*	fsysread(Xfid*, Fid*);
34 static	Xfid*	fsyswrite(Xfid*, Fid*);
35 static	Xfid*	fsysclunk(Xfid*, Fid*);
36 static	Xfid*	fsysremove(Xfid*, Fid*);
37 static	Xfid*	fsysstat(Xfid*, Fid*);
38 static	Xfid*	fsyswstat(Xfid*, Fid*);
39 
40 Xfid* 	(*fcall[Tmax])(Xfid*, Fid*);
41 
42 static void
43 initfcall(void)
44 {
45 	fcall[Tflush]	= fsysflush;
46 	fcall[Tversion]	= fsysversion;
47 	fcall[Tauth]	= fsysauth;
48 	fcall[Tattach]	= fsysattach;
49 	fcall[Twalk]	= fsyswalk;
50 	fcall[Topen]	= fsysopen;
51 	fcall[Tcreate]	= fsyscreate;
52 	fcall[Tread]	= fsysread;
53 	fcall[Twrite]	= fsyswrite;
54 	fcall[Tclunk]	= fsysclunk;
55 	fcall[Tremove]= fsysremove;
56 	fcall[Tstat]	= fsysstat;
57 	fcall[Twstat]	= fsyswstat;
58 }
59 
60 char Eperm[] = "permission denied";
61 char Eexist[] = "file does not exist";
62 char Enotdir[] = "not a directory";
63 
64 Dirtab dirtab[]=
65 {
66 	{ ".",			QTDIR,	Qdir,		0500|DMDIR },
67 	{ "acme",		QTDIR,	Qacme,	0500|DMDIR },
68 	{ "cons",		QTFILE,	Qcons,	0600 },
69 	{ "consctl",	QTFILE,	Qconsctl,	0000 },
70 	{ "draw",		QTDIR,	Qdraw,	0000|DMDIR },	/* to suppress graphics progs started in acme */
71 	{ "editout",	QTFILE,	Qeditout,	0200 },
72 	{ "index",		QTFILE,	Qindex,	0400 },
73 	{ "label",		QTFILE,	Qlabel,	0600 },
74 	{ "new",		QTDIR,	Qnew,	0500|DMDIR },
75 	{ nil, }
76 };
77 
78 Dirtab dirtabw[]=
79 {
80 	{ ".",			QTDIR,		Qdir,			0500|DMDIR },
81 	{ "addr",		QTFILE,		QWaddr,		0600 },
82 	{ "body",		QTAPPEND,	QWbody,		0600|DMAPPEND },
83 	{ "ctl",		QTFILE,		QWctl,		0600 },
84 	{ "data",		QTFILE,		QWdata,		0600 },
85 	{ "editout",	QTFILE,		QWeditout,	0200 },
86 	{ "errors",		QTFILE,		QWerrors,		0200 },
87 	{ "event",		QTFILE,		QWevent,		0600 },
88 	{ "rdsel",		QTFILE,		QWrdsel,		0400 },
89 	{ "wrsel",		QTFILE,		QWwrsel,		0200 },
90 	{ "tag",		QTAPPEND,	QWtag,		0600|DMAPPEND },
91 	{ "xdata",		QTFILE,		QWxdata,		0600 },
92 	{ nil, }
93 };
94 
95 typedef struct Mnt Mnt;
96 struct Mnt
97 {
98 	QLock	lk;
99 	int		id;
100 	Mntdir	*md;
101 };
102 
103 Mnt	mnt;
104 
105 Xfid*	respond(Xfid*, Fcall*, char*);
106 int		dostat(int, Dirtab*, uchar*, int, uint);
107 uint	getclock(void);
108 
109 char	*user = "Wile E. Coyote";
110 static int closing = 0;
111 int	messagesize = Maxblock+IOHDRSZ;	/* good start */
112 
113 void	fsysproc(void *);
114 
115 void
116 fsysinit(void)
117 {
118 	int p[2];
119 	char *u;
120 
121 	initfcall();
122 	if(pipe(p) < 0)
123 		error("can't create pipe");
124 	if(post9pservice(p[0], "acme", mtpt) < 0)
125 		error("can't post service");
126 	sfd = p[1];
127 	fmtinstall('F', fcallfmt);
128 	if((u = getuser()) != nil)
129 		user = estrdup(u);
130 	proccreate(fsysproc, nil, STACK);
131 }
132 
133 void
134 fsysproc(void *v)
135 {
136 	int n;
137 	Xfid *x;
138 	Fid *f;
139 	Fcall t;
140 	uchar *buf;
141 
142 	threadsetname("fsysproc");
143 
144 	USED(v);
145 	x = nil;
146 	for(;;){
147 		buf = emalloc(messagesize+UTFmax);	/* overflow for appending partial rune in xfidwrite */
148 		n = read9pmsg(sfd, buf, messagesize);
149 		if(n <= 0){
150 			if(closing)
151 				break;
152 			error("i/o error on server channel");
153 		}
154 		if(x == nil){
155 			sendp(cxfidalloc, nil);
156 			x = recvp(cxfidalloc);
157 		}
158 		x->buf = buf;
159 		if(convM2S(buf, n, &x->fcall) != n)
160 			error("convert error in convM2S");
161 		if(DEBUG)
162 			fprint(2, "%F\n", &x->fcall);
163 		if(fcall[x->fcall.type] == nil)
164 			x = respond(x, &t, "bad fcall type");
165 		else{
166 			switch(x->fcall.type){
167 			case Tversion:
168 			case Tauth:
169 			case Tflush:
170 				f = nil;
171 				break;
172 			case Tattach:
173 				f = newfid(x->fcall.fid);
174 				break;
175 			default:
176 				f = newfid(x->fcall.fid);
177 				if(!f->busy){
178 					x->f = f;
179 					x = respond(x, &t, "fid not in use");
180 					continue;
181 				}
182 				break;
183 			}
184 			x->f = f;
185 			x  = (*fcall[x->fcall.type])(x, f);
186 		}
187 	}
188 }
189 
190 Mntdir*
191 fsysaddid(Rune *dir, int ndir, Rune **incl, int nincl)
192 {
193 	Mntdir *m;
194 	int id;
195 
196 	qlock(&mnt.lk);
197 	id = ++mnt.id;
198 	m = emalloc(sizeof *m);
199 	m->id = id;
200 	m->dir =  dir;
201 	m->ref = 1;	/* one for Command, one will be incremented in attach */
202 	m->ndir = ndir;
203 	m->next = mnt.md;
204 	m->incl = incl;
205 	m->nincl = nincl;
206 	mnt.md = m;
207 	qunlock(&mnt.lk);
208 	return m;
209 }
210 
211 void
212 fsysincid(Mntdir *m)
213 {
214 	qlock(&mnt.lk);
215 	m->ref++;
216 	qunlock(&mnt.lk);
217 }
218 
219 void
220 fsysdelid(Mntdir *idm)
221 {
222 	Mntdir *m, *prev;
223 	int i;
224 	char buf[64];
225 
226 	if(idm == nil)
227 		return;
228 	qlock(&mnt.lk);
229 	if(--idm->ref > 0){
230 		qunlock(&mnt.lk);
231 		return;
232 	}
233 	prev = nil;
234 	for(m=mnt.md; m; m=m->next){
235 		if(m == idm){
236 			if(prev)
237 				prev->next = m->next;
238 			else
239 				mnt.md = m->next;
240 			for(i=0; i<m->nincl; i++)
241 				free(m->incl[i]);
242 			free(m->incl);
243 			free(m->dir);
244 			free(m);
245 			qunlock(&mnt.lk);
246 			return;
247 		}
248 		prev = m;
249 	}
250 	qunlock(&mnt.lk);
251 	sprint(buf, "fsysdelid: can't find id %d\n", idm->id);
252 	sendp(cerr, estrdup(buf));
253 }
254 
255 /*
256  * Called only in exec.c:/^run(), from a different FD group
257  */
258 Mntdir*
259 fsysmount(Rune *dir, int ndir, Rune **incl, int nincl)
260 {
261 	return fsysaddid(dir, ndir, incl, nincl);
262 }
263 
264 void
265 fsysclose(void)
266 {
267 	closing = 1;
268 	/*
269 	 * apparently this is not kosher on openbsd.
270 	 * perhaps because fsysproc is reading from sfd right now,
271 	 * the close hangs indefinitely.
272 	close(sfd);
273 	 */
274 }
275 
276 Xfid*
277 respond(Xfid *x, Fcall *t, char *err)
278 {
279 	int n;
280 
281 	if(err){
282 		t->type = Rerror;
283 		t->ename = err;
284 	}else
285 		t->type = x->fcall.type+1;
286 	t->fid = x->fcall.fid;
287 	t->tag = x->fcall.tag;
288 	if(x->buf == nil)
289 		x->buf = emalloc(messagesize);
290 	n = convS2M(t, x->buf, messagesize);
291 	if(n <= 0)
292 		error("convert error in convS2M");
293 	if(write(sfd, x->buf, n) != n)
294 		error("write error in respond");
295 	free(x->buf);
296 	x->buf = nil;
297 	if(DEBUG)
298 		fprint(2, "r: %F\n", t);
299 	return x;
300 }
301 
302 static
303 Xfid*
304 fsysversion(Xfid *x, Fid *f)
305 {
306 	Fcall t;
307 
308 	USED(f);
309 	if(x->fcall.msize < 256)
310 		return respond(x, &t, "version: message size too small");
311 	messagesize = x->fcall.msize;
312 	t.msize = messagesize;
313 	if(strncmp(x->fcall.version, "9P2000", 6) != 0)
314 		return respond(x, &t, "unrecognized 9P version");
315 	t.version = "9P2000";
316 	return respond(x, &t, nil);
317 }
318 
319 static
320 Xfid*
321 fsysauth(Xfid *x, Fid *f)
322 {
323 	Fcall t;
324 
325 	USED(f);
326 	return respond(x, &t, "acme: authentication not required");
327 }
328 
329 static
330 Xfid*
331 fsysflush(Xfid *x, Fid *f)
332 {
333 	USED(f);
334 	sendp(x->c, (void*)xfidflush);
335 	return nil;
336 }
337 
338 static
339 Xfid*
340 fsysattach(Xfid *x, Fid *f)
341 {
342 	Fcall t;
343 	int id;
344 	Mntdir *m;
345 	char buf[128];
346 
347 	if(strcmp(x->fcall.uname, user) != 0)
348 		return respond(x, &t, Eperm);
349 	f->busy = TRUE;
350 	f->open = FALSE;
351 	f->qid.path = Qdir;
352 	f->qid.type = QTDIR;
353 	f->qid.vers = 0;
354 	f->dir = dirtab;
355 	f->nrpart = 0;
356 	f->w = nil;
357 	t.qid = f->qid;
358 	f->mntdir = nil;
359 	id = atoi(x->fcall.aname);
360 	qlock(&mnt.lk);
361 	for(m=mnt.md; m; m=m->next)
362 		if(m->id == id){
363 			f->mntdir = m;
364 			m->ref++;
365 			break;
366 		}
367 	if(m == nil && x->fcall.aname[0]){
368 		snprint(buf, sizeof buf, "unknown id '%s' in attach", x->fcall.aname);
369 		sendp(cerr, estrdup(buf));
370 	}
371 	qunlock(&mnt.lk);
372 	return respond(x, &t, nil);
373 }
374 
375 static
376 Xfid*
377 fsyswalk(Xfid *x, Fid *f)
378 {
379 	Fcall t;
380 	int c, i, j, id;
381 	Qid q;
382 	uchar type;
383 	ulong path;
384 	Fid *nf;
385 	Dirtab *d, *dir;
386 	Window *w;
387 	char *err;
388 
389 	nf = nil;
390 	w = nil;
391 	if(f->open)
392 		return respond(x, &t, "walk of open file");
393 	if(x->fcall.fid != x->fcall.newfid){
394 		nf = newfid(x->fcall.newfid);
395 		if(nf->busy)
396 			return respond(x, &t, "newfid already in use");
397 		nf->busy = TRUE;
398 		nf->open = FALSE;
399 		nf->mntdir = f->mntdir;
400 		if(f->mntdir)
401 			f->mntdir->ref++;
402 		nf->dir = f->dir;
403 		nf->qid = f->qid;
404 		nf->w = f->w;
405 		nf->nrpart = 0;	/* not open, so must be zero */
406 		if(nf->w)
407 			incref(&nf->w->ref);
408 		f = nf;	/* walk f */
409 	}
410 
411 	t.nwqid = 0;
412 	err = nil;
413 	dir = nil;
414 	id = WIN(f->qid);
415 	q = f->qid;
416 
417 	if(x->fcall.nwname > 0){
418 		for(i=0; i<x->fcall.nwname; i++){
419 			if((q.type & QTDIR) == 0){
420 				err = Enotdir;
421 				break;
422 			}
423 
424 			if(strcmp(x->fcall.wname[i], "..") == 0){
425 				type = QTDIR;
426 				path = Qdir;
427 				id = 0;
428 				if(w){
429 					winclose(w);
430 					w = nil;
431 				}
432     Accept:
433 				if(i == MAXWELEM){
434 					err = "name too long";
435 					break;
436 				}
437 				q.type = type;
438 				q.vers = 0;
439 				q.path = QID(id, path);
440 				t.wqid[t.nwqid++] = q;
441 				continue;
442 			}
443 
444 			/* is it a numeric name? */
445 			for(j=0; (c=x->fcall.wname[i][j]); j++)
446 				if(c<'0' || '9'<c)
447 					goto Regular;
448 			/* yes: it's a directory */
449 			if(w)	/* name has form 27/23; get out before losing w */
450 				break;
451 			id = atoi(x->fcall.wname[i]);
452 			qlock(&row.lk);
453 			w = lookid(id, FALSE);
454 			if(w == nil){
455 				qunlock(&row.lk);
456 				break;
457 			}
458 			incref(&w->ref);	/* we'll drop reference at end if there's an error */
459 			path = Qdir;
460 			type = QTDIR;
461 			qunlock(&row.lk);
462 			dir = dirtabw;
463 			goto Accept;
464 	
465     Regular:
466 			if(strcmp(x->fcall.wname[i], "new") == 0){
467 				if(w)
468 					error("w set in walk to new");
469 				sendp(cnewwindow, nil);	/* signal newwindowthread */
470 				w = recvp(cnewwindow);	/* receive new window */
471 				incref(&w->ref);
472 				type = QTDIR;
473 				path = QID(w->id, Qdir);
474 				id = w->id;
475 				dir = dirtabw;
476 				goto Accept;
477 			}
478 
479 			if(id == 0)
480 				d = dirtab;
481 			else
482 				d = dirtabw;
483 			d++;	/* skip '.' */
484 			for(; d->name; d++)
485 				if(strcmp(x->fcall.wname[i], d->name) == 0){
486 					path = d->qid;
487 					type = d->type;
488 					dir = d;
489 					goto Accept;
490 				}
491 
492 			break;	/* file not found */
493 		}
494 
495 		if(i==0 && err == nil)
496 			err = Eexist;
497 	}
498 
499 	if(err!=nil || t.nwqid<x->fcall.nwname){
500 		if(nf){
501 			nf->busy = FALSE;
502 			fsysdelid(nf->mntdir);
503 		}
504 	}else if(t.nwqid  == x->fcall.nwname){
505 		if(w){
506 			f->w = w;
507 			w = nil;	/* don't drop the reference */
508 		}
509 		if(dir)
510 			f->dir = dir;
511 		f->qid = q;
512 	}
513 
514 	if(w != nil)
515 		winclose(w);
516 
517 	return respond(x, &t, err);
518 }
519 
520 static
521 Xfid*
522 fsysopen(Xfid *x, Fid *f)
523 {
524 	Fcall t;
525 	int m;
526 
527 	/* can't truncate anything, so just disregard */
528 	x->fcall.mode &= ~(OTRUNC|OCEXEC);
529 	/* can't execute or remove anything */
530 	if(x->fcall.mode==OEXEC || (x->fcall.mode&ORCLOSE))
531 		goto Deny;
532 	switch(x->fcall.mode){
533 	default:
534 		goto Deny;
535 	case OREAD:
536 		m = 0400;
537 		break;
538 	case OWRITE:
539 		m = 0200;
540 		break;
541 	case ORDWR:
542 		m = 0600;
543 		break;
544 	}
545 	if(((f->dir->perm&~(DMDIR|DMAPPEND))&m) != m)
546 		goto Deny;
547 
548 	sendp(x->c, (void*)xfidopen);
549 	return nil;
550 
551     Deny:
552 	return respond(x, &t, Eperm);
553 }
554 
555 static
556 Xfid*
557 fsyscreate(Xfid *x, Fid *f)
558 {
559 	Fcall t;
560 
561 	USED(f);
562 	return respond(x, &t, Eperm);
563 }
564 
565 static
566 int
567 idcmp(const void *a, const void *b)
568 {
569 	return *(int*)a - *(int*)b;
570 }
571 
572 static
573 Xfid*
574 fsysread(Xfid *x, Fid *f)
575 {
576 	Fcall t;
577 	uchar *b;
578 	int i, id, n, o, e, j, k, *ids, nids;
579 	Dirtab *d, dt;
580 	Column *c;
581 	uint clock, len;
582 	char buf[16];
583 
584 	if(f->qid.type & QTDIR){
585 		if(FILE(f->qid) == Qacme){	/* empty dir */
586 			t.data = nil;
587 			t.count = 0;
588 			respond(x, &t, nil);
589 			return x;
590 		}
591 		o = x->fcall.offset;
592 		e = x->fcall.offset+x->fcall.count;
593 		clock = getclock();
594 		b = emalloc(messagesize);
595 		id = WIN(f->qid);
596 		n = 0;
597 		if(id > 0)
598 			d = dirtabw;
599 		else
600 			d = dirtab;
601 		d++;	/* first entry is '.' */
602 		for(i=0; d->name!=nil && i<e; i+=len){
603 			len = dostat(WIN(x->f->qid), d, b+n, x->fcall.count-n, clock);
604 			if(len <= BIT16SZ)
605 				break;
606 			if(i >= o)
607 				n += len;
608 			d++;
609 		}
610 		if(id == 0){
611 			qlock(&row.lk);
612 			nids = 0;
613 			ids = nil;
614 			for(j=0; j<row.ncol; j++){
615 				c = row.col[j];
616 				for(k=0; k<c->nw; k++){
617 					ids = realloc(ids, (nids+1)*sizeof(int));
618 					ids[nids++] = c->w[k]->id;
619 				}
620 			}
621 			qunlock(&row.lk);
622 			qsort(ids, nids, sizeof ids[0], idcmp);
623 			j = 0;
624 			dt.name = buf;
625 			for(; j<nids && i<e; i+=len){
626 				k = ids[j];
627 				sprint(dt.name, "%d", k);
628 				dt.qid = QID(k, Qdir);
629 				dt.type = QTDIR;
630 				dt.perm = DMDIR|0700;
631 				len = dostat(k, &dt, b+n, x->fcall.count-n, clock);
632 				if(len == 0)
633 					break;
634 				if(i >= o)
635 					n += len;
636 				j++;
637 			}
638 			free(ids);
639 		}
640 		t.data = (char*)b;
641 		t.count = n;
642 		respond(x, &t, nil);
643 		free(b);
644 		return x;
645 	}
646 	sendp(x->c, (void*)xfidread);
647 	return nil;
648 }
649 
650 static
651 Xfid*
652 fsyswrite(Xfid *x, Fid *f)
653 {
654 	USED(f);
655 	sendp(x->c, (void*)xfidwrite);
656 	return nil;
657 }
658 
659 static
660 Xfid*
661 fsysclunk(Xfid *x, Fid *f)
662 {
663 	fsysdelid(f->mntdir);
664 	sendp(x->c, (void*)xfidclose);
665 	return nil;
666 }
667 
668 static
669 Xfid*
670 fsysremove(Xfid *x, Fid *f)
671 {
672 	Fcall t;
673 
674 	USED(f);
675 	return respond(x, &t, Eperm);
676 }
677 
678 static
679 Xfid*
680 fsysstat(Xfid *x, Fid *f)
681 {
682 	Fcall t;
683 
684 	t.stat = emalloc(messagesize-IOHDRSZ);
685 	t.nstat = dostat(WIN(x->f->qid), f->dir, t.stat, messagesize-IOHDRSZ, getclock());
686 	x = respond(x, &t, nil);
687 	free(t.stat);
688 	return x;
689 }
690 
691 static
692 Xfid*
693 fsyswstat(Xfid *x, Fid *f)
694 {
695 	Fcall t;
696 
697 	USED(f);
698 	return respond(x, &t, Eperm);
699 }
700 
701 Fid*
702 newfid(int fid)
703 {
704 	Fid *f, *ff, **fh;
705 
706 	ff = nil;
707 	fh = &fids[fid&(Nhash-1)];
708 	for(f=*fh; f; f=f->next)
709 		if(f->fid == fid)
710 			return f;
711 		else if(ff==nil && f->busy==FALSE)
712 			ff = f;
713 	if(ff){
714 		ff->fid = fid;
715 		return ff;
716 	}
717 	f = emalloc(sizeof *f);
718 	f->fid = fid;
719 	f->next = *fh;
720 	*fh = f;
721 	return f;
722 }
723 
724 uint
725 getclock(void)
726 {
727 	return time(0);
728 }
729 
730 int
731 dostat(int id, Dirtab *dir, uchar *buf, int nbuf, uint clock)
732 {
733 	Dir d;
734 
735 	d.qid.path = QID(id, dir->qid);
736 	d.qid.vers = 0;
737 	d.qid.type = dir->type;
738 	d.mode = dir->perm;
739 	d.length = 0;	/* would be nice to do better */
740 	d.name = dir->name;
741 	d.uid = user;
742 	d.gid = user;
743 	d.muid = user;
744 	d.atime = clock;
745 	d.mtime = clock;
746 	return convD2M(&d, buf, nbuf);
747 }