/* * Framework for USB devices that provide a file tree. * The main process (usbd or the driver's main proc) * calls fsinit() to start FS operation. * * One or more drivers call fsstart/fsend to register * or unregister their operations for their subtrees. * * root dir has qids with 0 in high 32 bits. * for other files we keep the device id in there. * The low 32 bits for directories at / must be 0. */#include <u.h>#include <libc.h>#include <thread.h>#include <fcall.h>#include "usb.h"#include "usbfs.h" #undef dprint#define dprint if(usbfsdebug)fprint typedef struct Rpc Rpc; enum{	Nproc = 3,		/* max nb. of cached FS procs */ 	Nofid = ~0,		/* null value for fid number */	Notag = ~0,		/* null value for tags */	Dietag = 0xdead,		/* fake tag to ask outproc to die */ 	Stack = 16 * 1024, 	/* Fsproc requests */	Run = 0,		/* call f(r) */	Exit,			/* terminate */ }; struct Rpc{	Fcall	t;	Fcall	r;	Fid*	fid;	int	flushed;	Rpc*	next;	char	data[Bufsize];}; int usbfsdebug; char Enotfound[] = "file not found";char Etoosmall[] = "parameter too small";char Eio[] = "i/o error";char Eperm[] = "permission denied";char Ebadcall[] = "unknown fs call";char Ebadfid[] = "fid not found";char Einuse[] = "fid already in use";char Eisopen[] = "it is already open";char Ebadctl[] = "unknown control request"; static char *user;static ulong epoch;static ulong msgsize = Msgsize;static int fsfd = -1;static Channel *outc;	/* of Rpc* */ static QLock rpclck;	/* protect vars in this block */static Fid *freefids;static Fid *fids;static Rpc *freerpcs;static Rpc *rpcs; static Channel*procc;static Channel*endc; static Usbfs* fsops; static void fsioproc(void*); static voidschedproc(void*){	Channel *proc[Nproc];	int nproc;	Channel *p; 	Alt a[] =	{		{procc, &proc[0], CHANSND},		{endc, &p, CHANRCV},		{nil, nil, CHANEND}	};	memset(proc, 0, sizeof(proc));	nproc = 0;	for(;;){		if(nproc == 0){			proc[0] = chancreate(sizeof(Rpc*), 0);			proccreate(fsioproc, proc[0], Stack);			nproc++;		}		switch(alt(a)){		case 0:			proc[0] = nil;			if(nproc > 1){				proc[0] = proc[nproc-1];				proc[nproc-1] = nil;			}			nproc--;			break;		case 1:			if(nproc < nelem(proc))				proc[nproc++] = p;			else				sendp(p, nil);			break;		default:			sysfatal("alt");		}	}} static voiddump(void){	Rpc *rpc;	Fid *fid; 	qlock(&rpclck);	fprint(2, "dump:\n");	for(rpc = rpcs; rpc != nil; rpc = rpc->next)		fprint(2, "rpc %#p %F next %#p\n", rpc, &rpc->t, rpc->next);	for(fid = fids; fid != nil; fid = fid->next)		fprint(2, "fid %d qid %#llux omode %d aux %#p\n",			fid->fid, fid->qid.path, fid->omode, fid->aux);	fprint(2, "\n");	qunlock(&rpclck);} static Rpc*newrpc(void){	Rpc *r; 	qlock(&rpclck);	r = freerpcs;	if(r != nil)		freerpcs = r->next;	else		r = emallocz(sizeof(Rpc), 0);	r->next = rpcs;	rpcs = r;	r->t.tag = r->r.tag = Notag;	r->t.fid = r->r.fid = Nofid;	r->t.type = r->r.type = 0;	r->flushed = 0;	r->fid = nil;	r->r.data = (char*)r->data;	qunlock(&rpclck);	return r;} static voidfreerpc(Rpc *r){	Rpc **l;	if(r == nil)		return;	qlock(&rpclck);	for(l = &rpcs; *l != nil && *l != r; l = &(*l)->next)		;	assert(*l == r);	*l = r->next;	r->next = freerpcs;	freerpcs = r;	r->t.type = 0;	r->t.tag = 0x77777777;	qunlock(&rpclck);} static voidflushrpc(int tag){	Rpc *r; 	qlock(&rpclck);	for(r = rpcs; r != nil; r = r->next)		if(r->t.tag == tag){			r->flushed = 1;			break;		}	qunlock(&rpclck);} static Fid*getfid(int fid, int alloc){	Fid *f; 	qlock(&rpclck);	for(f = fids; f != nil && f->fid != fid; f = f->next)		;	if(f != nil && alloc != 0){	/* fid in use */		qunlock(&rpclck);		return nil;	}	if(f == nil && alloc != 0){		if(freefids != nil){			f = freefids;			freefids = freefids->next;		}else			f = emallocz(sizeof(Fid), 1);		f->fid = fid;		f->aux = nil;		f->omode = ONONE;		f->next = fids;		fids = f;	}	qunlock(&rpclck);	return f;} static voidfreefid(Fid *f){	Fid **l; 	if(f == nil)		return;	if(fsops->clunk != nil)		fsops->clunk(fsops, f);	qlock(&rpclck);	for(l = &fids; *l != nil && *l != f; l = &(*l)->next)		;	assert(*l == f);	*l = f->next;	f->next = freefids;	freefids = f;	qunlock(&rpclck);} static Rpc*fserror(Rpc *rpc, char* fmt, ...){	va_list arg;	char *c; 	va_start(arg, fmt);	c = (char*)rpc->data;	vseprint(c, c+sizeof(rpc->data), fmt, arg);	va_end(arg);	rpc->r.type = Rerror;	rpc->r.ename = (char*)rpc->data;	return rpc;} static Rpc*fsversion(Rpc *r){	if(r->t.msize < 256)		return fserror(r, Etoosmall);	if(strncmp(r->t.version, "9P2000", 6) != 0)		return fserror(r, "wrong version");	if(r->t.msize < msgsize)		msgsize = r->t.msize;	r->r.msize = msgsize;	r->r.version = "9P2000";	return r;} static Rpc*fsattach(Rpc *r){	static int already; 	/* Reload user because at boot it could be still none */	user=getuser();	if(already++ > 0 && strcmp(r->t.uname, user) != 0)		return fserror(r, Eperm);	if(r->fid == nil)		return fserror(r, Einuse); 	r->r.qid.type = QTDIR;	r->r.qid.path = fsops->qid;	r->r.qid.vers = 0;	r->fid->qid = r->r.qid;	return r;} static Rpc*fswalk(Rpc *r){	int i;	Fid *nfid, *ofid; 	if(r->fid->omode != ONONE)		return fserror(r, Eisopen); 	nfid = nil;	ofid = r->fid;	if(r->t.newfid != r->t.fid){		nfid = getfid(r->t.newfid, 1);		if(nfid == nil)			return fserror(r, Einuse);		nfid->qid = r->fid->qid;		if(fsops->clone != nil)			fsops->clone(fsops, ofid, nfid);		else			nfid->aux = r->fid->aux;		r->fid = nfid;	}	r->r.nwqid = 0;	for(i = 0; i < r->t.nwname; i++)		if(fsops->walk(fsops, r->fid, r->t.wname[i]) < 0)			break;		else			r->r.wqid[i] = r->fid->qid;	r->r.nwqid = i;	if(i != r->t.nwname && r->t.nwname > 0){		if(nfid != nil)			freefid(nfid);		r->fid = ofid;	}	if(i == 0 && r->t.nwname > 0)		return fserror(r, "%r");	return r;} static voidfsioproc(void* a){	long rc;	Channel *p = a;	Rpc *rpc;	Fcall *t, *r;	Fid *fid; 	threadsetname("fsioproc");	dprint(2, "%s: fsioproc pid %d\n", argv0, getpid());	while((rpc = recvp(p)) != nil){		t = &rpc->t;		r = &rpc->r;		fid = rpc->fid;		rc = -1;		dprint(2, "%s: fsioproc pid %d: req %d\n", argv0, getpid(), t->type);		switch(t->type){		case Topen:			rc = fsops->open(fsops, fid, t->mode);			if(rc >= 0){				r->iounit = 0;				r->qid = fid->qid;				fid->omode = t->mode & 3;			}			break;		case Tread:			rc = fsops->read(fsops, fid, r->data, t->count, t->offset);			if(rc >= 0){				if(rc > t->count)					print("%s: bug: read %ld bytes > %ud wanted\n",						argv0, rc, t->count);				r->count = rc;			}			/*			 * TODO: if we encounter a long run of continuous read			 * errors, we should do something more drastic so that			 * our caller doesn't just spin its wheels forever.			 */			break;		case Twrite:			rc = fsops->write(fsops, fid, t->data, t->count, t->offset);			r->count = rc;			break;		default:			sysfatal("fsioproc: bad type");		}		if(rc < 0)			sendp(outc, fserror(rpc, "%r"));		else			sendp(outc, rpc);		sendp(endc, p);	}	chanfree(p);	dprint(2, "%s: fsioproc %d exiting\n", argv0, getpid());	threadexits(nil);} static Rpc*fsopen(Rpc *r){	Channel *p; 	if(r->fid->omode != ONONE)		return fserror(r, Eisopen);	if((r->t.mode & 3) != OREAD && (r->fid->qid.type & QTDIR) != 0)		return fserror(r, Eperm);	p = recvp(procc);	sendp(p, r);	return nil;} intusbdirread(Usbfs*f, Qid q, char *data, long cnt, vlong off, Dirgen gen, void *arg){	int i, n, nd;	char name[Namesz];	Dir d; 	memset(&d, 0, sizeof(d));	d.name = name;	d.uid = d.gid = d.muid = user;	d.atime = time(nil);	d.mtime = epoch;	d.length = 0;	for(i = n = 0; gen(f, q, i, &d, arg) >= 0; i++){		if(usbfsdebug > 1)			fprint(2, "%s: dir %d q %#llux: %D\n", argv0, i, q.path, &d);		nd = convD2M(&d, (uchar*)data+n, cnt-n);		if(nd <= BIT16SZ)			break;		if(off > 0)			off -= nd;		else			n += nd;		d.name = name;		d.uid = d.gid = d.muid = user;		d.atime = time(nil);		d.mtime = epoch;		d.length = 0;	}	return n;} longusbreadbuf(void *data, long count, vlong offset, void *buf, long n){	if(offset >= n)		return 0;	if(offset + count > n)		count = n - offset;	memmove(data, (char*)buf + offset, count);	return count;} static Rpc*fsread(Rpc *r){	Channel *p; 	if(r->fid->omode != OREAD && r->fid->omode != ORDWR)		return fserror(r, Eperm);	p = recvp(procc);	sendp(p, r);	return nil;} static Rpc*fswrite(Rpc *r){	Channel *p; 	if(r->fid->omode != OWRITE && r->fid->omode != ORDWR)		return fserror(r, Eperm);	p = recvp(procc);	sendp(p, r);	return nil;} static Rpc*fsclunk(Rpc *r){	freefid(r->fid);	return r;} static Rpc*fsno(Rpc *r){	return fserror(r, Eperm);} static Rpc*fsstat(Rpc *r){	Dir d;	char name[Namesz]; 	memset(&d, 0, sizeof(d));	d.name = name;	d.uid = d.gid = d.muid = user;	d.atime = time(nil);	d.mtime = epoch;	d.length = 0;	if(fsops->stat(fsops, r->fid->qid, &d) < 0)		return fserror(r, "%r");	r->r.stat = (uchar*)r->data;	r->r.nstat = convD2M(&d, (uchar*)r->data, msgsize);	return r;} 

Lines 1–500 of 676. Load more