1 46f79934 2005-01-04 devnull /* Copyright (C) 2003 Russ Cox, Massachusetts Institute of Technology */
  2 46f79934 2005-01-04 devnull /* See COPYRIGHT */
  3 46f79934 2005-01-04 devnull 
  4 46f79934 2005-01-04 devnull #include <u.h>
  5 46f79934 2005-01-04 devnull #include <libc.h>
  6 46f79934 2005-01-04 devnull #include <fcall.h>
  7 46f79934 2005-01-04 devnull #include <9pclient.h>
  8 46f79934 2005-01-04 devnull #include <thread.h>
  9 46f79934 2005-01-04 devnull #include "fsimpl.h"
 10 46f79934 2005-01-04 devnull 
 11 46f79934 2005-01-04 devnull static int _fssend(Mux*, void*);
 12 46f79934 2005-01-04 devnull static void *_fsrecv(Mux*);
 13 46f79934 2005-01-04 devnull static int _fsgettag(Mux*, void*);
 14 46f79934 2005-01-04 devnull static int _fssettag(Mux*, void*, uint);
 15 46f79934 2005-01-04 devnull 
 16 59518849 2005-02-11 devnull int chatty9pclient;
 17 59518849 2005-02-11 devnull 
 18 46f79934 2005-01-04 devnull enum
 19 46f79934 2005-01-04 devnull {
 20 46f79934 2005-01-04 devnull 	CFidchunk = 32
 21 46f79934 2005-01-04 devnull };
 22 46f79934 2005-01-04 devnull 
 23 46f79934 2005-01-04 devnull CFsys*
 24 46f79934 2005-01-04 devnull fsinit(int fd)
 25 46f79934 2005-01-04 devnull {
 26 46f79934 2005-01-04 devnull 	CFsys *fs;
 27 59518849 2005-02-11 devnull 	int n;
 28 59518849 2005-02-11 devnull 	
 29 46f79934 2005-01-04 devnull 	fmtinstall('F', fcallfmt);
 30 46f79934 2005-01-04 devnull 	fmtinstall('D', dirfmt);
 31 46f79934 2005-01-04 devnull 	fmtinstall('M', dirmodefmt);
 32 46f79934 2005-01-04 devnull 
 33 46f79934 2005-01-04 devnull 	fs = mallocz(sizeof(CFsys), 1);
 34 2d70c581 2005-07-13 devnull 	if(fs == nil){
 35 2d70c581 2005-07-13 devnull 		werrstr("mallocz: %r");
 36 46f79934 2005-01-04 devnull 		return nil;
 37 2d70c581 2005-07-13 devnull 	}
 38 46f79934 2005-01-04 devnull 	fs->fd = fd;
 39 46f79934 2005-01-04 devnull 	fs->ref = 1;
 40 46f79934 2005-01-04 devnull 	fs->mux.aux = fs;
 41 46f79934 2005-01-04 devnull 	fs->mux.mintag = 0;
 42 46f79934 2005-01-04 devnull 	fs->mux.maxtag = 256;
 43 46f79934 2005-01-04 devnull 	fs->mux.send = _fssend;
 44 46f79934 2005-01-04 devnull 	fs->mux.recv = _fsrecv;
 45 46f79934 2005-01-04 devnull 	fs->mux.gettag = _fsgettag;
 46 46f79934 2005-01-04 devnull 	fs->mux.settag = _fssettag;
 47 46f79934 2005-01-04 devnull 	fs->iorecv = ioproc();
 48 46f79934 2005-01-04 devnull 	fs->iosend = ioproc();
 49 46f79934 2005-01-04 devnull 	muxinit(&fs->mux);
 50 59518849 2005-02-11 devnull 	
 51 59518849 2005-02-11 devnull 	strcpy(fs->version, "9P2000");
 52 59518849 2005-02-11 devnull 	if((n = fsversion(fs, 8192, fs->version, sizeof fs->version)) < 0){
 53 2d70c581 2005-07-13 devnull 		werrstr("fsversion: %r");
 54 59518849 2005-02-11 devnull 		_fsunmount(fs);
 55 59518849 2005-02-11 devnull 		return nil;
 56 59518849 2005-02-11 devnull 	}
 57 59518849 2005-02-11 devnull 	fs->msize = n;
 58 46f79934 2005-01-04 devnull 	return fs;
 59 46f79934 2005-01-04 devnull }
 60 46f79934 2005-01-04 devnull 
 61 46f79934 2005-01-04 devnull CFid*
 62 46f79934 2005-01-04 devnull fsroot(CFsys *fs)
 63 46f79934 2005-01-04 devnull {
 64 46f79934 2005-01-04 devnull 	/* N.B. no incref */
 65 46f79934 2005-01-04 devnull 	return fs->root;
 66 46f79934 2005-01-04 devnull }
 67 46f79934 2005-01-04 devnull 
 68 46f79934 2005-01-04 devnull CFsys*
 69 46f79934 2005-01-04 devnull fsmount(int fd, char *aname)
 70 46f79934 2005-01-04 devnull {
 71 46f79934 2005-01-04 devnull 	CFsys *fs;
 72 46f79934 2005-01-04 devnull 	CFid *fid;
 73 46f79934 2005-01-04 devnull 
 74 46f79934 2005-01-04 devnull 	fs = fsinit(fd);
 75 46f79934 2005-01-04 devnull 	if(fs == nil)
 76 46f79934 2005-01-04 devnull 		return nil;
 77 46f79934 2005-01-04 devnull 
 78 59518849 2005-02-11 devnull 	if((fid = fsattach(fs, nil, getuser(), aname)) == nil){
 79 59518849 2005-02-11 devnull 		_fsunmount(fs);
 80 59518849 2005-02-11 devnull 		return nil;
 81 59518849 2005-02-11 devnull 	}
 82 46f79934 2005-01-04 devnull 	fssetroot(fs, fid);
 83 46f79934 2005-01-04 devnull 	return fs;
 84 59518849 2005-02-11 devnull }
 85 59518849 2005-02-11 devnull 
 86 59518849 2005-02-11 devnull void
 87 59518849 2005-02-11 devnull _fsunmount(CFsys *fs)
 88 59518849 2005-02-11 devnull {
 89 59518849 2005-02-11 devnull 	fs->fd = -1;
 90 59518849 2005-02-11 devnull 	fsunmount(fs);
 91 46f79934 2005-01-04 devnull }
 92 46f79934 2005-01-04 devnull 
 93 46f79934 2005-01-04 devnull void
 94 46f79934 2005-01-04 devnull fsunmount(CFsys *fs)
 95 46f79934 2005-01-04 devnull {
 96 46f79934 2005-01-04 devnull 	fsclose(fs->root);
 97 46f79934 2005-01-04 devnull 	fs->root = nil;
 98 46f79934 2005-01-04 devnull 	_fsdecref(fs);
 99 46f79934 2005-01-04 devnull }
100 46f79934 2005-01-04 devnull 
101 46f79934 2005-01-04 devnull void
102 46f79934 2005-01-04 devnull _fsdecref(CFsys *fs)
103 46f79934 2005-01-04 devnull {
104 46f79934 2005-01-04 devnull 	CFid *f, **l, *next;
105 46f79934 2005-01-04 devnull 
106 46f79934 2005-01-04 devnull 	qlock(&fs->lk);
107 46f79934 2005-01-04 devnull 	--fs->ref;
108 cbeb0b26 2006-04-01 devnull 	/*fprint(2, "fsdecref %p to %d\n", fs, fs->ref); */
109 46f79934 2005-01-04 devnull 	if(fs->ref == 0){
110 6135dd69 2005-01-19 devnull 		if(fs->fd >= 0)
111 6135dd69 2005-01-19 devnull 			close(fs->fd);
112 46f79934 2005-01-04 devnull 		/* trim the list down to just the first in each chunk */
113 46f79934 2005-01-04 devnull 		for(l=&fs->freefid; *l; ){
114 46f79934 2005-01-04 devnull 			if((*l)->fid%CFidchunk == 0)
115 46f79934 2005-01-04 devnull 				l = &(*l)->next;
116 46f79934 2005-01-04 devnull 			else
117 46f79934 2005-01-04 devnull 				*l = (*l)->next;
118 46f79934 2005-01-04 devnull 		}
119 46f79934 2005-01-04 devnull 		/* now free the list */
120 46f79934 2005-01-04 devnull 		for(f=fs->freefid; f; f=next){
121 46f79934 2005-01-04 devnull 			next = f->next;
122 46f79934 2005-01-04 devnull 			free(f);
123 46f79934 2005-01-04 devnull 		}
124 46f79934 2005-01-04 devnull 		closeioproc(fs->iorecv);
125 46f79934 2005-01-04 devnull 		closeioproc(fs->iosend);
126 46f79934 2005-01-04 devnull 		free(fs);
127 46f79934 2005-01-04 devnull 		return;
128 46f79934 2005-01-04 devnull 	}
129 46f79934 2005-01-04 devnull 	qunlock(&fs->lk);
130 46f79934 2005-01-04 devnull }
131 46f79934 2005-01-04 devnull 
132 46f79934 2005-01-04 devnull int
133 46f79934 2005-01-04 devnull fsversion(CFsys *fs, int msize, char *version, int nversion)
134 46f79934 2005-01-04 devnull {
135 46f79934 2005-01-04 devnull 	void *freep;
136 46f79934 2005-01-04 devnull 	int r, oldmintag, oldmaxtag;
137 46f79934 2005-01-04 devnull 	Fcall tx, rx;
138 46f79934 2005-01-04 devnull 
139 46f79934 2005-01-04 devnull 	tx.tag = 0;
140 46f79934 2005-01-04 devnull 	tx.type = Tversion;
141 46f79934 2005-01-04 devnull 	tx.version = version;
142 46f79934 2005-01-04 devnull 	tx.msize = msize;
143 46f79934 2005-01-04 devnull 
144 46f79934 2005-01-04 devnull 	/*
145 46f79934 2005-01-04 devnull 	 * bit of a clumsy hack -- force libmux to use NOTAG as tag.
146 46f79934 2005-01-04 devnull 	 * version can only be sent when there are no other messages
147 46f79934 2005-01-04 devnull 	 * outstanding on the wire, so this is more reasonable than it looks.
148 46f79934 2005-01-04 devnull 	 */
149 46f79934 2005-01-04 devnull 	oldmintag = fs->mux.mintag;
150 46f79934 2005-01-04 devnull 	oldmaxtag = fs->mux.maxtag;
151 46f79934 2005-01-04 devnull 	fs->mux.mintag = NOTAG;
152 46f79934 2005-01-04 devnull 	fs->mux.maxtag = NOTAG+1;
153 46f79934 2005-01-04 devnull 	r = _fsrpc(fs, &tx, &rx, &freep);
154 46f79934 2005-01-04 devnull 	fs->mux.mintag = oldmintag;
155 46f79934 2005-01-04 devnull 	fs->mux.maxtag = oldmaxtag;
156 2d70c581 2005-07-13 devnull 	if(r < 0){
157 2d70c581 2005-07-13 devnull 		werrstr("fsrpc: %r");
158 46f79934 2005-01-04 devnull 		return -1;
159 2d70c581 2005-07-13 devnull 	}
160 46f79934 2005-01-04 devnull 
161 46f79934 2005-01-04 devnull 	strecpy(version, version+nversion, rx.version);
162 46f79934 2005-01-04 devnull 	free(freep);
163 80ecfd24 2005-02-08 devnull 	fs->msize = rx.msize;
164 46f79934 2005-01-04 devnull 	return rx.msize;
165 46f79934 2005-01-04 devnull }
166 46f79934 2005-01-04 devnull 
167 46f79934 2005-01-04 devnull CFid*
168 46f79934 2005-01-04 devnull fsattach(CFsys *fs, CFid *afid, char *user, char *aname)
169 46f79934 2005-01-04 devnull {
170 46f79934 2005-01-04 devnull 	Fcall tx, rx;
171 46f79934 2005-01-04 devnull 	CFid *fid;
172 46f79934 2005-01-04 devnull 
173 46f79934 2005-01-04 devnull 	if(aname == nil)
174 46f79934 2005-01-04 devnull 		aname = "";
175 46f79934 2005-01-04 devnull 
176 46f79934 2005-01-04 devnull 	if((fid = _fsgetfid(fs)) == nil)
177 46f79934 2005-01-04 devnull 		return nil;
178 46f79934 2005-01-04 devnull 
179 46f79934 2005-01-04 devnull 	tx.tag = 0;
180 46f79934 2005-01-04 devnull 	tx.type = Tattach;
181 46f79934 2005-01-04 devnull 	tx.afid = afid ? afid->fid : NOFID;
182 46f79934 2005-01-04 devnull 	tx.fid = fid->fid;
183 46f79934 2005-01-04 devnull 	tx.uname = user;
184 46f79934 2005-01-04 devnull 	tx.aname = aname;
185 46f79934 2005-01-04 devnull 
186 46f79934 2005-01-04 devnull 	if(_fsrpc(fs, &tx, &rx, 0) < 0){
187 46f79934 2005-01-04 devnull 		_fsputfid(fid);
188 46f79934 2005-01-04 devnull 		return nil;
189 46f79934 2005-01-04 devnull 	}
190 46f79934 2005-01-04 devnull 	fid->qid = rx.qid;
191 46f79934 2005-01-04 devnull 	return fid;
192 46f79934 2005-01-04 devnull }
193 46f79934 2005-01-04 devnull 
194 46f79934 2005-01-04 devnull void
195 46f79934 2005-01-04 devnull fssetroot(CFsys *fs, CFid *fid)
196 46f79934 2005-01-04 devnull {
197 46f79934 2005-01-04 devnull 	if(fs->root)
198 46f79934 2005-01-04 devnull 		_fsputfid(fs->root);
199 46f79934 2005-01-04 devnull 	fs->root = fid;
200 46f79934 2005-01-04 devnull }
201 46f79934 2005-01-04 devnull 
202 46f79934 2005-01-04 devnull int
203 46f79934 2005-01-04 devnull _fsrpc(CFsys *fs, Fcall *tx, Fcall *rx, void **freep)
204 46f79934 2005-01-04 devnull {
205 46f79934 2005-01-04 devnull 	int n, nn;
206 46f79934 2005-01-04 devnull 	void *tpkt, *rpkt;
207 46f79934 2005-01-04 devnull 
208 46f79934 2005-01-04 devnull 	n = sizeS2M(tx);
209 46f79934 2005-01-04 devnull 	tpkt = malloc(n);
210 46f79934 2005-01-04 devnull 	if(freep)
211 46f79934 2005-01-04 devnull 		*freep = nil;
212 46f79934 2005-01-04 devnull 	if(tpkt == nil)
213 46f79934 2005-01-04 devnull 		return -1;
214 59518849 2005-02-11 devnull 	tx->tag = 0;
215 59518849 2005-02-11 devnull 	if(chatty9pclient)
216 59518849 2005-02-11 devnull 		fprint(2, "<- %F\n", tx);
217 46f79934 2005-01-04 devnull 	nn = convS2M(tx, tpkt, n);
218 46f79934 2005-01-04 devnull 	if(nn != n){
219 46f79934 2005-01-04 devnull 		free(tpkt);
220 2d70c581 2005-07-13 devnull 		werrstr("lib9pclient: sizeS2M convS2M mismatch");
221 46f79934 2005-01-04 devnull 		fprint(2, "%r\n");
222 46f79934 2005-01-04 devnull 		return -1;
223 46f79934 2005-01-04 devnull 	}
224 46f79934 2005-01-04 devnull 	rpkt = muxrpc(&fs->mux, tpkt);
225 46f79934 2005-01-04 devnull 	free(tpkt);
226 2d70c581 2005-07-13 devnull 	if(rpkt == nil){
227 2d70c581 2005-07-13 devnull 		werrstr("muxrpc: %r");
228 46f79934 2005-01-04 devnull 		return -1;
229 2d70c581 2005-07-13 devnull 	}
230 46f79934 2005-01-04 devnull 	n = GBIT32((uchar*)rpkt);
231 46f79934 2005-01-04 devnull 	nn = convM2S(rpkt, n, rx);
232 46f79934 2005-01-04 devnull 	if(nn != n){
233 46f79934 2005-01-04 devnull 		free(rpkt);
234 2d70c581 2005-07-13 devnull 		werrstr("lib9pclient: convM2S packet size mismatch %d %d", n, nn);
235 46f79934 2005-01-04 devnull 		fprint(2, "%r\n");
236 46f79934 2005-01-04 devnull 		return -1;
237 46f79934 2005-01-04 devnull 	}
238 59518849 2005-02-11 devnull 	if(chatty9pclient)
239 59518849 2005-02-11 devnull 		fprint(2, "-> %F\n", rx);
240 46f79934 2005-01-04 devnull 	if(rx->type == Rerror){
241 46f79934 2005-01-04 devnull 		werrstr("%s", rx->ename);
242 46f79934 2005-01-04 devnull 		free(rpkt);
243 46f79934 2005-01-04 devnull 		return -1;
244 46f79934 2005-01-04 devnull 	}
245 46f79934 2005-01-04 devnull 	if(rx->type != tx->type+1){
246 46f79934 2005-01-04 devnull 		werrstr("packet type mismatch -- tx %d rx %d",
247 46f79934 2005-01-04 devnull 			tx->type, rx->type);
248 46f79934 2005-01-04 devnull 		free(rpkt);
249 46f79934 2005-01-04 devnull 		return -1;
250 46f79934 2005-01-04 devnull 	}
251 46f79934 2005-01-04 devnull 	if(freep)
252 46f79934 2005-01-04 devnull 		*freep = rpkt;
253 46f79934 2005-01-04 devnull 	else
254 46f79934 2005-01-04 devnull 		free(rpkt);
255 46f79934 2005-01-04 devnull 	return 0;
256 46f79934 2005-01-04 devnull }
257 46f79934 2005-01-04 devnull 
258 46f79934 2005-01-04 devnull CFid*
259 46f79934 2005-01-04 devnull _fsgetfid(CFsys *fs)
260 46f79934 2005-01-04 devnull {
261 46f79934 2005-01-04 devnull 	int i;
262 46f79934 2005-01-04 devnull 	CFid *f;
263 46f79934 2005-01-04 devnull 
264 46f79934 2005-01-04 devnull 	qlock(&fs->lk);
265 46f79934 2005-01-04 devnull 	if(fs->freefid == nil){
266 46f79934 2005-01-04 devnull 		f = mallocz(sizeof(CFid)*CFidchunk, 1);
267 46f79934 2005-01-04 devnull 		if(f == nil){
268 46f79934 2005-01-04 devnull 			qunlock(&fs->lk);
269 46f79934 2005-01-04 devnull 			return nil;
270 46f79934 2005-01-04 devnull 		}
271 46f79934 2005-01-04 devnull 		for(i=0; i<CFidchunk; i++){
272 46f79934 2005-01-04 devnull 			f[i].fid = fs->nextfid++;
273 46f79934 2005-01-04 devnull 			f[i].next = &f[i+1];
274 46f79934 2005-01-04 devnull 			f[i].fs = fs;
275 46f79934 2005-01-04 devnull 		}
276 46f79934 2005-01-04 devnull 		f[i-1].next = nil;
277 46f79934 2005-01-04 devnull 		fs->freefid = f;
278 46f79934 2005-01-04 devnull 	}
279 46f79934 2005-01-04 devnull 	f = fs->freefid;
280 46f79934 2005-01-04 devnull 	fs->freefid = f->next;
281 46f79934 2005-01-04 devnull 	fs->ref++;
282 46f79934 2005-01-04 devnull 	qunlock(&fs->lk);
283 33baa59e 2005-01-28 devnull 	f->offset = 0;
284 33baa59e 2005-01-28 devnull 	f->mode = -1;
285 33baa59e 2005-01-28 devnull 	f->qid.path = 0;
286 33baa59e 2005-01-28 devnull 	f->qid.vers = 0;
287 33baa59e 2005-01-28 devnull 	f->qid.type = 0;
288 46f79934 2005-01-04 devnull 	return f;
289 46f79934 2005-01-04 devnull }
290 46f79934 2005-01-04 devnull 
291 46f79934 2005-01-04 devnull void
292 46f79934 2005-01-04 devnull _fsputfid(CFid *f)
293 46f79934 2005-01-04 devnull {
294 46f79934 2005-01-04 devnull 	CFsys *fs;
295 46f79934 2005-01-04 devnull 
296 46f79934 2005-01-04 devnull 	fs = f->fs;
297 46f79934 2005-01-04 devnull 	qlock(&fs->lk);
298 46f79934 2005-01-04 devnull 	f->next = fs->freefid;
299 46f79934 2005-01-04 devnull 	fs->freefid = f;
300 46f79934 2005-01-04 devnull 	qunlock(&fs->lk);
301 46f79934 2005-01-04 devnull 	_fsdecref(fs);
302 46f79934 2005-01-04 devnull }
303 46f79934 2005-01-04 devnull 
304 46f79934 2005-01-04 devnull static int
305 46f79934 2005-01-04 devnull _fsgettag(Mux *mux, void *pkt)
306 46f79934 2005-01-04 devnull {
307 46f79934 2005-01-04 devnull 	return GBIT16((uchar*)pkt+5);
308 46f79934 2005-01-04 devnull }
309 46f79934 2005-01-04 devnull 
310 46f79934 2005-01-04 devnull static int
311 46f79934 2005-01-04 devnull _fssettag(Mux *mux, void *pkt, uint tag)
312 46f79934 2005-01-04 devnull {
313 46f79934 2005-01-04 devnull 	PBIT16((uchar*)pkt+5, tag);
314 46f79934 2005-01-04 devnull 	return 0;
315 46f79934 2005-01-04 devnull }
316 46f79934 2005-01-04 devnull 
317 46f79934 2005-01-04 devnull static int
318 46f79934 2005-01-04 devnull _fssend(Mux *mux, void *pkt)
319 46f79934 2005-01-04 devnull {
320 46f79934 2005-01-04 devnull 	CFsys *fs;
321 46f79934 2005-01-04 devnull 
322 46f79934 2005-01-04 devnull 	fs = mux->aux;
323 46f79934 2005-01-04 devnull 	return iowrite(fs->iosend, fs->fd, pkt, GBIT32((uchar*)pkt));
324 46f79934 2005-01-04 devnull }
325 46f79934 2005-01-04 devnull 
326 46f79934 2005-01-04 devnull static void*
327 46f79934 2005-01-04 devnull _fsrecv(Mux *mux)
328 46f79934 2005-01-04 devnull {
329 46f79934 2005-01-04 devnull 	uchar *pkt;
330 46f79934 2005-01-04 devnull 	uchar buf[4];
331 46f79934 2005-01-04 devnull 	int n, nfd;
332 46f79934 2005-01-04 devnull 	CFsys *fs;
333 46f79934 2005-01-04 devnull 
334 46f79934 2005-01-04 devnull 	fs = mux->aux;
335 46f79934 2005-01-04 devnull 	n = ioreadn(fs->iorecv, fs->fd, buf, 4);
336 46f79934 2005-01-04 devnull 	if(n != 4)
337 46f79934 2005-01-04 devnull 		return nil;
338 46f79934 2005-01-04 devnull 	n = GBIT32(buf);
339 46f79934 2005-01-04 devnull 	pkt = malloc(n+4);
340 46f79934 2005-01-04 devnull 	if(pkt == nil){
341 2d70c581 2005-07-13 devnull 		fprint(2, "lib9pclient out of memory reading 9p packet; here comes trouble\n");
342 46f79934 2005-01-04 devnull 		return nil;
343 46f79934 2005-01-04 devnull 	}
344 46f79934 2005-01-04 devnull 	PBIT32(pkt, n);
345 46f79934 2005-01-04 devnull 	if(ioreadn(fs->iorecv, fs->fd, pkt+4, n-4) != n-4){
346 46f79934 2005-01-04 devnull 		free(pkt);
347 46f79934 2005-01-04 devnull 		return nil;
348 46f79934 2005-01-04 devnull 	}
349 46f79934 2005-01-04 devnull 	if(pkt[4] == Ropenfd){
350 46f79934 2005-01-04 devnull 		if((nfd=iorecvfd(fs->iorecv, fs->fd)) < 0){
351 46f79934 2005-01-04 devnull 			fprint(2, "recv fd error: %r\n");
352 46f79934 2005-01-04 devnull 			free(pkt);
353 46f79934 2005-01-04 devnull 			return nil;
354 46f79934 2005-01-04 devnull 		}
355 46f79934 2005-01-04 devnull 		PBIT32(pkt+n-4, nfd);
356 46f79934 2005-01-04 devnull 	}
357 46f79934 2005-01-04 devnull 	return pkt;
358 46f79934 2005-01-04 devnull }