Blame


1 d3df3087 2003-12-06 devnull #include <u.h>
2 d3df3087 2003-12-06 devnull #include <libc.h>
3 d3df3087 2003-12-06 devnull #include <fcall.h>
4 d3df3087 2003-12-06 devnull #include <thread.h>
5 32f69c36 2003-12-11 devnull #include <errno.h>
6 d9ed7243 2006-04-21 devnull
7 d9ed7243 2006-04-21 devnull #define err err9pserve /* Darwin x86 */
8 d3df3087 2003-12-06 devnull
9 d3df3087 2003-12-06 devnull enum
10 d3df3087 2003-12-06 devnull {
11 d3df3087 2003-12-06 devnull STACK = 32768,
12 d3df3087 2003-12-06 devnull NHASH = 31,
13 f2466297 2006-05-04 devnull MAXMSG = 64, /* per connection */
14 f2466297 2006-05-04 devnull MAXMSGSIZE = 4*1024*1024
15 d3df3087 2003-12-06 devnull };
16 d3df3087 2003-12-06 devnull
17 d3df3087 2003-12-06 devnull typedef struct Hash Hash;
18 d3df3087 2003-12-06 devnull typedef struct Fid Fid;
19 d3df3087 2003-12-06 devnull typedef struct Msg Msg;
20 d3df3087 2003-12-06 devnull typedef struct Conn Conn;
21 d3df3087 2003-12-06 devnull typedef struct Queue Queue;
22 d3df3087 2003-12-06 devnull
23 d3df3087 2003-12-06 devnull struct Hash
24 d3df3087 2003-12-06 devnull {
25 d3df3087 2003-12-06 devnull Hash *next;
26 d3df3087 2003-12-06 devnull uint n;
27 d3df3087 2003-12-06 devnull void *v;
28 d3df3087 2003-12-06 devnull };
29 d3df3087 2003-12-06 devnull
30 d3df3087 2003-12-06 devnull struct Fid
31 d3df3087 2003-12-06 devnull {
32 d3df3087 2003-12-06 devnull int fid;
33 d3df3087 2003-12-06 devnull int ref;
34 d3df3087 2003-12-06 devnull int cfid;
35 21e626de 2004-12-28 devnull int openfd;
36 9777187b 2006-02-03 devnull int offset;
37 9777187b 2006-02-03 devnull int coffset;
38 215993f8 2005-09-13 devnull int isdir;
39 d3df3087 2003-12-06 devnull Fid *next;
40 d3df3087 2003-12-06 devnull };
41 d3df3087 2003-12-06 devnull
42 d3df3087 2003-12-06 devnull struct Msg
43 d3df3087 2003-12-06 devnull {
44 d3df3087 2003-12-06 devnull Conn *c;
45 d3df3087 2003-12-06 devnull int internal;
46 841d71b5 2007-08-22 rsc int sync;
47 d3df3087 2003-12-06 devnull int ref;
48 d3df3087 2003-12-06 devnull int ctag;
49 d3df3087 2003-12-06 devnull int tag;
50 32f69c36 2003-12-11 devnull int isopenfd;
51 d3df3087 2003-12-06 devnull Fcall tx;
52 d3df3087 2003-12-06 devnull Fcall rx;
53 d3df3087 2003-12-06 devnull Fid *fid;
54 d3df3087 2003-12-06 devnull Fid *newfid;
55 d3df3087 2003-12-06 devnull Fid *afid;
56 d3df3087 2003-12-06 devnull Msg *oldm;
57 d3df3087 2003-12-06 devnull Msg *next;
58 d3df3087 2003-12-06 devnull uchar *tpkt;
59 d3df3087 2003-12-06 devnull uchar *rpkt;
60 d3df3087 2003-12-06 devnull };
61 d3df3087 2003-12-06 devnull
62 d3df3087 2003-12-06 devnull struct Conn
63 d3df3087 2003-12-06 devnull {
64 d3df3087 2003-12-06 devnull int fd;
65 32f69c36 2003-12-11 devnull int fdmode;
66 32f69c36 2003-12-11 devnull Fid *fdfid;
67 d3df3087 2003-12-06 devnull int nmsg;
68 d3df3087 2003-12-06 devnull int nfid;
69 d3df3087 2003-12-06 devnull Channel *inc;
70 d3df3087 2003-12-06 devnull Channel *internal;
71 d3df3087 2003-12-06 devnull int inputstalled;
72 d3df3087 2003-12-06 devnull char dir[40];
73 d3df3087 2003-12-06 devnull Hash *tag[NHASH];
74 d3df3087 2003-12-06 devnull Hash *fid[NHASH];
75 d3df3087 2003-12-06 devnull Queue *outq;
76 d3df3087 2003-12-06 devnull Queue *inq;
77 841d71b5 2007-08-22 rsc Channel *outqdead;
78 d3df3087 2003-12-06 devnull };
79 d3df3087 2003-12-06 devnull
80 5c84c448 2005-03-21 devnull char *xaname;
81 d3df3087 2003-12-06 devnull char *addr;
82 d3df3087 2003-12-06 devnull int afd;
83 d3df3087 2003-12-06 devnull char adir[40];
84 d3df3087 2003-12-06 devnull int isunix;
85 d3df3087 2003-12-06 devnull Queue *outq;
86 d3df3087 2003-12-06 devnull Queue *inq;
87 05b7f431 2004-03-02 devnull int verbose = 0;
88 e2a1725d 2005-01-04 devnull int logging = 0;
89 5a8e63b2 2004-02-29 devnull int msize = 8192;
90 34e8254f 2005-07-27 devnull u32int xafid = NOFID;
91 5c84c448 2005-03-21 devnull int attached;
92 5c84c448 2005-03-21 devnull int versioned;
93 9b253297 2007-10-12 rsc int noauth;
94 d3df3087 2003-12-06 devnull
95 d3df3087 2003-12-06 devnull void *gethash(Hash**, uint);
96 d3df3087 2003-12-06 devnull int puthash(Hash**, uint, void*);
97 d3df3087 2003-12-06 devnull int delhash(Hash**, uint, void*);
98 74dd0321 2010-12-07 rsc Msg *mread9p(Ioproc*, int);
99 ceb04770 2003-12-09 devnull int mwrite9p(Ioproc*, int, uchar*);
100 ceb04770 2003-12-09 devnull uchar *read9ppkt(Ioproc*, int);
101 d3df3087 2003-12-06 devnull int write9ppkt(int, uchar*);
102 e2a1725d 2005-01-04 devnull Msg *msgnew(int);
103 d3df3087 2003-12-06 devnull void msgput(Msg*);
104 e2a1725d 2005-01-04 devnull void msgclear(Msg*);
105 d3df3087 2003-12-06 devnull Msg *msgget(int);
106 e2a1725d 2005-01-04 devnull void msgincref(Msg*);
107 d3df3087 2003-12-06 devnull Fid *fidnew(int);
108 d3df3087 2003-12-06 devnull void fidput(Fid*);
109 d3df3087 2003-12-06 devnull void *emalloc(int);
110 d3df3087 2003-12-06 devnull void *erealloc(void*, int);
111 ceb04770 2003-12-09 devnull Queue *qalloc(void);
112 d3df3087 2003-12-06 devnull int sendq(Queue*, void*);
113 d3df3087 2003-12-06 devnull void *recvq(Queue*);
114 d3df3087 2003-12-06 devnull void connthread(void*);
115 ceb04770 2003-12-09 devnull void connoutthread(void*);
116 d3df3087 2003-12-06 devnull void listenthread(void*);
117 ceb04770 2003-12-09 devnull void outputthread(void*);
118 ceb04770 2003-12-09 devnull void inputthread(void*);
119 d3df3087 2003-12-06 devnull void rewritehdr(Fcall*, uchar*);
120 74dd0321 2010-12-07 rsc void repack(Fcall*, uchar**);
121 d3df3087 2003-12-06 devnull int tlisten(char*, char*);
122 d3df3087 2003-12-06 devnull int taccept(int, char*);
123 ceb04770 2003-12-09 devnull int iolisten(Ioproc*, char*, char*);
124 ceb04770 2003-12-09 devnull int ioaccept(Ioproc*, int, char*);
125 32f69c36 2003-12-11 devnull int iorecvfd(Ioproc*, int);
126 32f69c36 2003-12-11 devnull int iosendfd(Ioproc*, int, int);
127 32f69c36 2003-12-11 devnull void mainproc(void*);
128 32f69c36 2003-12-11 devnull int ignorepipe(void*, char*);
129 e2a1725d 2005-01-04 devnull int timefmt(Fmt*);
130 5c84c448 2005-03-21 devnull void dorootstat(void);
131 d3df3087 2003-12-06 devnull
132 d3df3087 2003-12-06 devnull void
133 d3df3087 2003-12-06 devnull usage(void)
134 d3df3087 2003-12-06 devnull {
135 ef6606c7 2008-06-11 rsc fprint(2, "usage: 9pserve [-lnv] [-A aname afid] [-c addr] [-M msize] address\n");
136 d3df3087 2003-12-06 devnull fprint(2, "\treads/writes 9P messages on stdin/stdout\n");
137 38c10d1a 2005-01-17 devnull threadexitsall("usage");
138 d3df3087 2003-12-06 devnull }
139 d3df3087 2003-12-06 devnull
140 ceb04770 2003-12-09 devnull uchar vbuf[128];
141 32f69c36 2003-12-11 devnull extern int _threaddebuglevel;
142 d3df3087 2003-12-06 devnull void
143 d3df3087 2003-12-06 devnull threadmain(int argc, char **argv)
144 d3df3087 2003-12-06 devnull {
145 ef6606c7 2008-06-11 rsc char *file, *x, *addr;
146 e2a1725d 2005-01-04 devnull int fd;
147 ceb04770 2003-12-09 devnull
148 bf486dfc 2009-12-10 rsc rfork(RFNOTEG);
149 3a6f92ee 2005-03-18 devnull x = getenv("verbose9pserve");
150 5c84c448 2005-03-21 devnull if(x){
151 3a6f92ee 2005-03-18 devnull verbose = atoi(x);
152 5c84c448 2005-03-21 devnull fprint(2, "verbose9pserve %s => %d\n", x, verbose);
153 5c84c448 2005-03-21 devnull }
154 d3df3087 2003-12-06 devnull ARGBEGIN{
155 d3df3087 2003-12-06 devnull default:
156 d3df3087 2003-12-06 devnull usage();
157 5c84c448 2005-03-21 devnull case 'A':
158 5c84c448 2005-03-21 devnull attached = 1;
159 5c84c448 2005-03-21 devnull xaname = EARGF(usage());
160 5c84c448 2005-03-21 devnull xafid = atoi(EARGF(usage()));
161 5c84c448 2005-03-21 devnull break;
162 5c84c448 2005-03-21 devnull case 'M':
163 5c84c448 2005-03-21 devnull versioned = 1;
164 5c84c448 2005-03-21 devnull msize = atoi(EARGF(usage()));
165 9b253297 2007-10-12 rsc break;
166 ef6606c7 2008-06-11 rsc case 'c':
167 ef6606c7 2008-06-11 rsc addr = netmkaddr(EARGF(usage()), "net", "9fs");
168 ef6606c7 2008-06-11 rsc if((fd = dial(addr, nil, nil, nil)) < 0)
169 ef6606c7 2008-06-11 rsc sysfatal("dial %s: %r", addr);
170 ef6606c7 2008-06-11 rsc dup(fd, 0);
171 ef6606c7 2008-06-11 rsc dup(fd, 1);
172 ef6606c7 2008-06-11 rsc if(fd > 1)
173 ef6606c7 2008-06-11 rsc close(fd);
174 ef6606c7 2008-06-11 rsc break;
175 9b253297 2007-10-12 rsc case 'n':
176 9b253297 2007-10-12 rsc noauth = 1;
177 5c84c448 2005-03-21 devnull break;
178 ceb04770 2003-12-09 devnull case 'v':
179 ceb04770 2003-12-09 devnull verbose++;
180 ceb04770 2003-12-09 devnull break;
181 d3df3087 2003-12-06 devnull case 'u':
182 5c84c448 2005-03-21 devnull isunix++;
183 d3df3087 2003-12-06 devnull break;
184 e2a1725d 2005-01-04 devnull case 'l':
185 e2a1725d 2005-01-04 devnull logging++;
186 e2a1725d 2005-01-04 devnull break;
187 d3df3087 2003-12-06 devnull }ARGEND
188 5c84c448 2005-03-21 devnull
189 5c84c448 2005-03-21 devnull if(attached && !versioned){
190 5c84c448 2005-03-21 devnull fprint(2, "-A must be used with -M\n");
191 5c84c448 2005-03-21 devnull usage();
192 5c84c448 2005-03-21 devnull }
193 d3df3087 2003-12-06 devnull
194 d3df3087 2003-12-06 devnull if(argc != 1)
195 d3df3087 2003-12-06 devnull usage();
196 ceb04770 2003-12-09 devnull addr = argv[0];
197 d3df3087 2003-12-06 devnull
198 e2a1725d 2005-01-04 devnull fmtinstall('T', timefmt);
199 e2a1725d 2005-01-04 devnull
200 d3df3087 2003-12-06 devnull if((afd = announce(addr, adir)) < 0)
201 d3df3087 2003-12-06 devnull sysfatal("announce %s: %r", addr);
202 e2a1725d 2005-01-04 devnull if(logging){
203 e2a1725d 2005-01-04 devnull if(strncmp(addr, "unix!", 5) == 0)
204 e2a1725d 2005-01-04 devnull addr += 5;
205 e2a1725d 2005-01-04 devnull file = smprint("%s.log", addr);
206 e2a1725d 2005-01-04 devnull if(file == nil)
207 e2a1725d 2005-01-04 devnull sysfatal("smprint log: %r");
208 e2a1725d 2005-01-04 devnull if((fd = create(file, OWRITE, 0666)) < 0)
209 e2a1725d 2005-01-04 devnull sysfatal("create %s: %r", file);
210 e2a1725d 2005-01-04 devnull dup(fd, 2);
211 e2a1725d 2005-01-04 devnull if(fd > 2)
212 e2a1725d 2005-01-04 devnull close(fd);
213 e2a1725d 2005-01-04 devnull }
214 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T 9pserve running\n");
215 e2a1725d 2005-01-04 devnull proccreate(mainproc, nil, STACK);
216 32f69c36 2003-12-11 devnull }
217 32f69c36 2003-12-11 devnull
218 32f69c36 2003-12-11 devnull void
219 32f69c36 2003-12-11 devnull mainproc(void *v)
220 32f69c36 2003-12-11 devnull {
221 c91bd328 2004-03-05 devnull int n, nn;
222 32f69c36 2003-12-11 devnull Fcall f;
223 32f69c36 2003-12-11 devnull USED(v);
224 32f69c36 2003-12-11 devnull
225 32f69c36 2003-12-11 devnull atnotify(ignorepipe, 1);
226 ceb04770 2003-12-09 devnull fmtinstall('D', dirfmt);
227 ceb04770 2003-12-09 devnull fmtinstall('M', dirmodefmt);
228 ceb04770 2003-12-09 devnull fmtinstall('F', fcallfmt);
229 ceb04770 2003-12-09 devnull fmtinstall('H', encodefmt);
230 ceb04770 2003-12-09 devnull
231 ceb04770 2003-12-09 devnull outq = qalloc();
232 ceb04770 2003-12-09 devnull inq = qalloc();
233 ceb04770 2003-12-09 devnull
234 5c84c448 2005-03-21 devnull if(!versioned){
235 5c84c448 2005-03-21 devnull f.type = Tversion;
236 74dd0321 2010-12-07 rsc f.version = "9P2000";
237 5c84c448 2005-03-21 devnull f.msize = msize;
238 5c84c448 2005-03-21 devnull f.tag = NOTAG;
239 5c84c448 2005-03-21 devnull n = convS2M(&f, vbuf, sizeof vbuf);
240 9e2f1d9b 2005-11-21 devnull if(n <= BIT16SZ)
241 9e2f1d9b 2005-11-21 devnull sysfatal("convS2M conversion error");
242 5c84c448 2005-03-21 devnull if(verbose > 1) fprint(2, "%T * <- %F\n", &f);
243 5c84c448 2005-03-21 devnull nn = write(1, vbuf, n);
244 5c84c448 2005-03-21 devnull if(n != nn)
245 5c84c448 2005-03-21 devnull sysfatal("error writing Tversion: %r\n");
246 5c84c448 2005-03-21 devnull n = read9pmsg(0, vbuf, sizeof vbuf);
247 f2466297 2006-05-04 devnull if(n < 0)
248 fdd0f841 2006-05-04 devnull sysfatal("read9pmsg failure");
249 5c84c448 2005-03-21 devnull if(convM2S(vbuf, n, &f) != n)
250 5c84c448 2005-03-21 devnull sysfatal("convM2S failure");
251 5c84c448 2005-03-21 devnull if(f.msize < msize)
252 5c84c448 2005-03-21 devnull msize = f.msize;
253 5c84c448 2005-03-21 devnull if(verbose > 1) fprint(2, "%T * -> %F\n", &f);
254 5c84c448 2005-03-21 devnull }
255 32f69c36 2003-12-11 devnull
256 32f69c36 2003-12-11 devnull threadcreate(inputthread, nil, STACK);
257 32f69c36 2003-12-11 devnull threadcreate(outputthread, nil, STACK);
258 5c84c448 2005-03-21 devnull
259 cbeb0b26 2006-04-01 devnull /* if(rootfid) */
260 cbeb0b26 2006-04-01 devnull /* dorootstat(); */
261 5c84c448 2005-03-21 devnull
262 ceb04770 2003-12-09 devnull threadcreate(listenthread, nil, STACK);
263 32f69c36 2003-12-11 devnull threadexits(0);
264 d3df3087 2003-12-06 devnull }
265 d3df3087 2003-12-06 devnull
266 32f69c36 2003-12-11 devnull int
267 32f69c36 2003-12-11 devnull ignorepipe(void *v, char *s)
268 32f69c36 2003-12-11 devnull {
269 32f69c36 2003-12-11 devnull USED(v);
270 32f69c36 2003-12-11 devnull if(strcmp(s, "sys: write on closed pipe") == 0)
271 32f69c36 2003-12-11 devnull return 1;
272 c042e61e 2005-03-18 devnull if(strcmp(s, "sys: tstp") == 0)
273 6322b295 2007-06-18 devnull return 1;
274 6322b295 2007-06-18 devnull if(strcmp(s, "sys: window size change") == 0)
275 c042e61e 2005-03-18 devnull return 1;
276 c042e61e 2005-03-18 devnull fprint(2, "9pserve %s: %T note: %s\n", addr, s);
277 32f69c36 2003-12-11 devnull return 0;
278 32f69c36 2003-12-11 devnull }
279 32f69c36 2003-12-11 devnull
280 d3df3087 2003-12-06 devnull void
281 d3df3087 2003-12-06 devnull listenthread(void *arg)
282 d3df3087 2003-12-06 devnull {
283 d3df3087 2003-12-06 devnull Conn *c;
284 ceb04770 2003-12-09 devnull Ioproc *io;
285 d3df3087 2003-12-06 devnull
286 ceb04770 2003-12-09 devnull io = ioproc();
287 d3df3087 2003-12-06 devnull USED(arg);
288 7ffc5208 2004-12-28 devnull threadsetname("listen %s", adir);
289 d3df3087 2003-12-06 devnull for(;;){
290 ceb04770 2003-12-09 devnull c = emalloc(sizeof(Conn));
291 ceb04770 2003-12-09 devnull c->fd = iolisten(io, adir, c->dir);
292 d3df3087 2003-12-06 devnull if(c->fd < 0){
293 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T listen: %r\n");
294 d3df3087 2003-12-06 devnull close(afd);
295 d3df3087 2003-12-06 devnull free(c);
296 d3df3087 2003-12-06 devnull return;
297 d3df3087 2003-12-06 devnull }
298 32f69c36 2003-12-11 devnull c->inc = chancreate(sizeof(void*), 0);
299 32f69c36 2003-12-11 devnull c->internal = chancreate(sizeof(void*), 0);
300 32f69c36 2003-12-11 devnull c->inq = qalloc();
301 32f69c36 2003-12-11 devnull c->outq = qalloc();
302 841d71b5 2007-08-22 rsc c->outqdead = chancreate(sizeof(void*), 0);
303 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T incoming call on %s\n", c->dir);
304 d3df3087 2003-12-06 devnull threadcreate(connthread, c, STACK);
305 d3df3087 2003-12-06 devnull }
306 d3df3087 2003-12-06 devnull }
307 d3df3087 2003-12-06 devnull
308 d3df3087 2003-12-06 devnull void
309 32f69c36 2003-12-11 devnull send9pmsg(Msg *m)
310 d3df3087 2003-12-06 devnull {
311 d3df3087 2003-12-06 devnull int n, nn;
312 d3df3087 2003-12-06 devnull
313 74dd0321 2010-12-07 rsc n = sizeS2M(&m->rx);
314 d3df3087 2003-12-06 devnull m->rpkt = emalloc(n);
315 74dd0321 2010-12-07 rsc nn = convS2M(&m->rx, m->rpkt, n);
316 9e2f1d9b 2005-11-21 devnull if(nn <= BIT16SZ)
317 74dd0321 2010-12-07 rsc sysfatal("convS2M conversion error");
318 d3df3087 2003-12-06 devnull if(nn != n)
319 74dd0321 2010-12-07 rsc sysfatal("sizeS2M and convS2M disagree");
320 d3df3087 2003-12-06 devnull sendq(m->c->outq, m);
321 d3df3087 2003-12-06 devnull }
322 d3df3087 2003-12-06 devnull
323 d3df3087 2003-12-06 devnull void
324 ceb04770 2003-12-09 devnull sendomsg(Msg *m)
325 ceb04770 2003-12-09 devnull {
326 ceb04770 2003-12-09 devnull int n, nn;
327 ceb04770 2003-12-09 devnull
328 74dd0321 2010-12-07 rsc n = sizeS2M(&m->tx);
329 ceb04770 2003-12-09 devnull m->tpkt = emalloc(n);
330 74dd0321 2010-12-07 rsc nn = convS2M(&m->tx, m->tpkt, n);
331 9e2f1d9b 2005-11-21 devnull if(nn <= BIT16SZ)
332 74dd0321 2010-12-07 rsc sysfatal("convS2M conversion error");
333 ceb04770 2003-12-09 devnull if(nn != n)
334 74dd0321 2010-12-07 rsc sysfatal("sizeS2M and convS2M disagree");
335 ceb04770 2003-12-09 devnull sendq(outq, m);
336 ceb04770 2003-12-09 devnull }
337 ceb04770 2003-12-09 devnull
338 ceb04770 2003-12-09 devnull void
339 ceb04770 2003-12-09 devnull err(Msg *m, char *ename)
340 ceb04770 2003-12-09 devnull {
341 ceb04770 2003-12-09 devnull m->rx.type = Rerror;
342 ceb04770 2003-12-09 devnull m->rx.ename = ename;
343 ceb04770 2003-12-09 devnull m->rx.tag = m->tx.tag;
344 32f69c36 2003-12-11 devnull send9pmsg(m);
345 ceb04770 2003-12-09 devnull }
346 ceb04770 2003-12-09 devnull
347 5c84c448 2005-03-21 devnull char*
348 5c84c448 2005-03-21 devnull estrdup(char *s)
349 5c84c448 2005-03-21 devnull {
350 5c84c448 2005-03-21 devnull char *t;
351 5c84c448 2005-03-21 devnull
352 5c84c448 2005-03-21 devnull t = emalloc(strlen(s)+1);
353 5c84c448 2005-03-21 devnull strcpy(t, s);
354 5c84c448 2005-03-21 devnull return t;
355 5c84c448 2005-03-21 devnull }
356 5c84c448 2005-03-21 devnull
357 ceb04770 2003-12-09 devnull void
358 d3df3087 2003-12-06 devnull connthread(void *arg)
359 d3df3087 2003-12-06 devnull {
360 d3df3087 2003-12-06 devnull int i, fd;
361 d3df3087 2003-12-06 devnull Conn *c;
362 05b7f431 2004-03-02 devnull Hash *h, *hnext;
363 841d71b5 2007-08-22 rsc Msg *m, *om, *mm, sync;
364 d3df3087 2003-12-06 devnull Fid *f;
365 ceb04770 2003-12-09 devnull Ioproc *io;
366 d3df3087 2003-12-06 devnull
367 d3df3087 2003-12-06 devnull c = arg;
368 7ffc5208 2004-12-28 devnull threadsetname("conn %s", c->dir);
369 ceb04770 2003-12-09 devnull io = ioproc();
370 ceb04770 2003-12-09 devnull fd = ioaccept(io, c->fd, c->dir);
371 d3df3087 2003-12-06 devnull if(fd < 0){
372 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T accept %s: %r\n", c->dir);
373 d3df3087 2003-12-06 devnull goto out;
374 d3df3087 2003-12-06 devnull }
375 d3df3087 2003-12-06 devnull close(c->fd);
376 d3df3087 2003-12-06 devnull c->fd = fd;
377 ceb04770 2003-12-09 devnull threadcreate(connoutthread, c, STACK);
378 74dd0321 2010-12-07 rsc while((m = mread9p(io, c->fd)) != nil){
379 e2a1725d 2005-01-04 devnull if(verbose > 1) fprint(2, "%T fd#%d -> %F\n", c->fd, &m->tx);
380 d3df3087 2003-12-06 devnull m->c = c;
381 ceb04770 2003-12-09 devnull m->ctag = m->tx.tag;
382 d3df3087 2003-12-06 devnull c->nmsg++;
383 e2a1725d 2005-01-04 devnull if(verbose > 1) fprint(2, "%T fd#%d: new msg %p\n", c->fd, m);
384 d3df3087 2003-12-06 devnull if(puthash(c->tag, m->tx.tag, m) < 0){
385 d3df3087 2003-12-06 devnull err(m, "duplicate tag");
386 d3df3087 2003-12-06 devnull continue;
387 d3df3087 2003-12-06 devnull }
388 e2a1725d 2005-01-04 devnull msgincref(m);
389 d3df3087 2003-12-06 devnull switch(m->tx.type){
390 ceb04770 2003-12-09 devnull case Tversion:
391 ceb04770 2003-12-09 devnull m->rx.tag = m->tx.tag;
392 ceb04770 2003-12-09 devnull m->rx.msize = m->tx.msize;
393 5a8e63b2 2004-02-29 devnull if(m->rx.msize > msize)
394 5a8e63b2 2004-02-29 devnull m->rx.msize = msize;
395 ceb04770 2003-12-09 devnull m->rx.version = "9P2000";
396 ceb04770 2003-12-09 devnull m->rx.type = Rversion;
397 32f69c36 2003-12-11 devnull send9pmsg(m);
398 ceb04770 2003-12-09 devnull continue;
399 d3df3087 2003-12-06 devnull case Tflush:
400 d3df3087 2003-12-06 devnull if((m->oldm = gethash(c->tag, m->tx.oldtag)) == nil){
401 ceb04770 2003-12-09 devnull m->rx.tag = m->tx.tag;
402 ceb04770 2003-12-09 devnull m->rx.type = Rflush;
403 32f69c36 2003-12-11 devnull send9pmsg(m);
404 d3df3087 2003-12-06 devnull continue;
405 d3df3087 2003-12-06 devnull }
406 e2a1725d 2005-01-04 devnull msgincref(m->oldm);
407 d3df3087 2003-12-06 devnull break;
408 d3df3087 2003-12-06 devnull case Tattach:
409 ceb04770 2003-12-09 devnull m->afid = nil;
410 ceb04770 2003-12-09 devnull if(m->tx.afid != NOFID
411 ceb04770 2003-12-09 devnull && (m->afid = gethash(c->fid, m->tx.afid)) == nil){
412 ceb04770 2003-12-09 devnull err(m, "unknown fid");
413 ceb04770 2003-12-09 devnull continue;
414 ceb04770 2003-12-09 devnull }
415 3a6f92ee 2005-03-18 devnull if(m->afid)
416 3a6f92ee 2005-03-18 devnull m->afid->ref++;
417 d3df3087 2003-12-06 devnull m->fid = fidnew(m->tx.fid);
418 d3df3087 2003-12-06 devnull if(puthash(c->fid, m->tx.fid, m->fid) < 0){
419 d3df3087 2003-12-06 devnull err(m, "duplicate fid");
420 d3df3087 2003-12-06 devnull continue;
421 d3df3087 2003-12-06 devnull }
422 d3df3087 2003-12-06 devnull m->fid->ref++;
423 5c84c448 2005-03-21 devnull if(attached && m->afid==nil){
424 5c84c448 2005-03-21 devnull if(m->tx.aname[0] && strcmp(xaname, m->tx.aname) != 0){
425 5c84c448 2005-03-21 devnull err(m, "invalid attach name");
426 5c84c448 2005-03-21 devnull continue;
427 5c84c448 2005-03-21 devnull }
428 5c84c448 2005-03-21 devnull m->tx.afid = xafid;
429 5c84c448 2005-03-21 devnull m->tx.aname = xaname;
430 7cb82e15 2006-05-27 devnull m->tx.uname = getuser(); /* what srv.c used */
431 74dd0321 2010-12-07 rsc repack(&m->tx, &m->tpkt);
432 5c84c448 2005-03-21 devnull }
433 d3df3087 2003-12-06 devnull break;
434 d3df3087 2003-12-06 devnull case Twalk:
435 d3df3087 2003-12-06 devnull if((m->fid = gethash(c->fid, m->tx.fid)) == nil){
436 d3df3087 2003-12-06 devnull err(m, "unknown fid");
437 d3df3087 2003-12-06 devnull continue;
438 d3df3087 2003-12-06 devnull }
439 ceb04770 2003-12-09 devnull m->fid->ref++;
440 d3df3087 2003-12-06 devnull if(m->tx.newfid == m->tx.fid){
441 d3df3087 2003-12-06 devnull m->fid->ref++;
442 d3df3087 2003-12-06 devnull m->newfid = m->fid;
443 d3df3087 2003-12-06 devnull }else{
444 d3df3087 2003-12-06 devnull m->newfid = fidnew(m->tx.newfid);
445 d3df3087 2003-12-06 devnull if(puthash(c->fid, m->tx.newfid, m->newfid) < 0){
446 d3df3087 2003-12-06 devnull err(m, "duplicate fid");
447 d3df3087 2003-12-06 devnull continue;
448 d3df3087 2003-12-06 devnull }
449 d3df3087 2003-12-06 devnull m->newfid->ref++;
450 d3df3087 2003-12-06 devnull }
451 d3df3087 2003-12-06 devnull break;
452 d3df3087 2003-12-06 devnull case Tauth:
453 5c84c448 2005-03-21 devnull if(attached){
454 5c84c448 2005-03-21 devnull err(m, "authentication not required");
455 9b253297 2007-10-12 rsc continue;
456 9b253297 2007-10-12 rsc }
457 9b253297 2007-10-12 rsc if(noauth){
458 9b253297 2007-10-12 rsc err(m, "authentication rejected");
459 5c84c448 2005-03-21 devnull continue;
460 5c84c448 2005-03-21 devnull }
461 ceb04770 2003-12-09 devnull m->afid = fidnew(m->tx.afid);
462 ceb04770 2003-12-09 devnull if(puthash(c->fid, m->tx.afid, m->afid) < 0){
463 d3df3087 2003-12-06 devnull err(m, "duplicate fid");
464 d3df3087 2003-12-06 devnull continue;
465 d3df3087 2003-12-06 devnull }
466 ceb04770 2003-12-09 devnull m->afid->ref++;
467 d3df3087 2003-12-06 devnull break;
468 b5c7be2f 2006-03-03 devnull case Tcreate:
469 74dd0321 2010-12-07 rsc if(m->tx.perm&(DMSYMLINK|DMDEVICE|DMNAMEDPIPE|DMSOCKET)){
470 b5c7be2f 2006-03-03 devnull err(m, "unsupported file type");
471 b5c7be2f 2006-03-03 devnull continue;
472 b5c7be2f 2006-03-03 devnull }
473 b5c7be2f 2006-03-03 devnull goto caseTopen;
474 32f69c36 2003-12-11 devnull case Topenfd:
475 49588d5d 2003-12-17 devnull if(m->tx.mode&~(OTRUNC|3)){
476 49588d5d 2003-12-17 devnull err(m, "bad openfd mode");
477 32f69c36 2003-12-11 devnull continue;
478 32f69c36 2003-12-11 devnull }
479 32f69c36 2003-12-11 devnull m->isopenfd = 1;
480 32f69c36 2003-12-11 devnull m->tx.type = Topen;
481 32f69c36 2003-12-11 devnull m->tpkt[4] = Topen;
482 32f69c36 2003-12-11 devnull /* fall through */
483 b5c7be2f 2006-03-03 devnull caseTopen:
484 d3df3087 2003-12-06 devnull case Topen:
485 d3df3087 2003-12-06 devnull case Tclunk:
486 d3df3087 2003-12-06 devnull case Tread:
487 d3df3087 2003-12-06 devnull case Twrite:
488 ceb04770 2003-12-09 devnull case Tremove:
489 d3df3087 2003-12-06 devnull case Tstat:
490 d3df3087 2003-12-06 devnull case Twstat:
491 d3df3087 2003-12-06 devnull if((m->fid = gethash(c->fid, m->tx.fid)) == nil){
492 d3df3087 2003-12-06 devnull err(m, "unknown fid");
493 d3df3087 2003-12-06 devnull continue;
494 d3df3087 2003-12-06 devnull }
495 d3df3087 2003-12-06 devnull m->fid->ref++;
496 d3df3087 2003-12-06 devnull break;
497 d3df3087 2003-12-06 devnull }
498 d3df3087 2003-12-06 devnull
499 d3df3087 2003-12-06 devnull /* have everything - translate and send */
500 d3df3087 2003-12-06 devnull m->c = c;
501 d3df3087 2003-12-06 devnull m->ctag = m->tx.tag;
502 d3df3087 2003-12-06 devnull m->tx.tag = m->tag;
503 d3df3087 2003-12-06 devnull if(m->fid)
504 d3df3087 2003-12-06 devnull m->tx.fid = m->fid->fid;
505 d3df3087 2003-12-06 devnull if(m->newfid)
506 d3df3087 2003-12-06 devnull m->tx.newfid = m->newfid->fid;
507 d3df3087 2003-12-06 devnull if(m->afid)
508 d3df3087 2003-12-06 devnull m->tx.afid = m->afid->fid;
509 d3df3087 2003-12-06 devnull if(m->oldm)
510 d3df3087 2003-12-06 devnull m->tx.oldtag = m->oldm->tag;
511 ceb04770 2003-12-09 devnull /* reference passes to outq */
512 d3df3087 2003-12-06 devnull sendq(outq, m);
513 d3df3087 2003-12-06 devnull while(c->nmsg >= MAXMSG){
514 d3df3087 2003-12-06 devnull c->inputstalled = 1;
515 d3df3087 2003-12-06 devnull recvp(c->inc);
516 d3df3087 2003-12-06 devnull }
517 d3df3087 2003-12-06 devnull }
518 d3df3087 2003-12-06 devnull
519 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T fd#%d eof; flushing conn\n", c->fd);
520 ceb04770 2003-12-09 devnull
521 d3df3087 2003-12-06 devnull /* flush all outstanding messages */
522 d3df3087 2003-12-06 devnull for(i=0; i<NHASH; i++){
523 841d71b5 2007-08-22 rsc while((h = c->tag[i]) != nil){
524 d3df3087 2003-12-06 devnull om = h->v;
525 841d71b5 2007-08-22 rsc msgincref(om); /* for us */
526 e2a1725d 2005-01-04 devnull m = msgnew(0);
527 d3df3087 2003-12-06 devnull m->internal = 1;
528 d3df3087 2003-12-06 devnull m->c = c;
529 32f69c36 2003-12-11 devnull c->nmsg++;
530 d3df3087 2003-12-06 devnull m->tx.type = Tflush;
531 d3df3087 2003-12-06 devnull m->tx.tag = m->tag;
532 d3df3087 2003-12-06 devnull m->tx.oldtag = om->tag;
533 d3df3087 2003-12-06 devnull m->oldm = om;
534 e2a1725d 2005-01-04 devnull msgincref(om);
535 e2a1725d 2005-01-04 devnull msgincref(m); /* for outq */
536 ceb04770 2003-12-09 devnull sendomsg(m);
537 05b7f431 2004-03-02 devnull mm = recvp(c->internal);
538 05b7f431 2004-03-02 devnull assert(mm == m);
539 32f69c36 2003-12-11 devnull msgput(m); /* got from recvp */
540 32f69c36 2003-12-11 devnull msgput(m); /* got from msgnew */
541 ec924708 2007-08-28 rsc if(delhash(c->tag, om->ctag, om) == 0)
542 841d71b5 2007-08-22 rsc msgput(om); /* got from hash table */
543 841d71b5 2007-08-22 rsc msgput(om); /* got from msgincref */
544 d3df3087 2003-12-06 devnull }
545 d3df3087 2003-12-06 devnull }
546 841d71b5 2007-08-22 rsc
547 841d71b5 2007-08-22 rsc /*
548 841d71b5 2007-08-22 rsc * outputthread has written all its messages
549 841d71b5 2007-08-22 rsc * to the remote connection (because we've gotten all the replies!),
550 841d71b5 2007-08-22 rsc * but it might not have gotten a chance to msgput
551 841d71b5 2007-08-22 rsc * the very last one. sync up to make sure.
552 841d71b5 2007-08-22 rsc */
553 841d71b5 2007-08-22 rsc memset(&sync, 0, sizeof sync);
554 841d71b5 2007-08-22 rsc sync.sync = 1;
555 841d71b5 2007-08-22 rsc sync.c = c;
556 841d71b5 2007-08-22 rsc sendq(outq, &sync);
557 841d71b5 2007-08-22 rsc recvp(c->outqdead);
558 d3df3087 2003-12-06 devnull
559 841d71b5 2007-08-22 rsc /* everything is quiet; can close the local output queue. */
560 841d71b5 2007-08-22 rsc sendq(c->outq, nil);
561 841d71b5 2007-08-22 rsc recvp(c->outqdead);
562 841d71b5 2007-08-22 rsc
563 4a6f0510 2007-08-22 rsc /* should be no messages left anywhere. */
564 4a6f0510 2007-08-22 rsc assert(c->nmsg == 0);
565 4a6f0510 2007-08-22 rsc
566 d3df3087 2003-12-06 devnull /* clunk all outstanding fids */
567 d3df3087 2003-12-06 devnull for(i=0; i<NHASH; i++){
568 05b7f431 2004-03-02 devnull for(h=c->fid[i]; h; h=hnext){
569 d3df3087 2003-12-06 devnull f = h->v;
570 e2a1725d 2005-01-04 devnull m = msgnew(0);
571 d3df3087 2003-12-06 devnull m->internal = 1;
572 d3df3087 2003-12-06 devnull m->c = c;
573 32f69c36 2003-12-11 devnull c->nmsg++;
574 d3df3087 2003-12-06 devnull m->tx.type = Tclunk;
575 d3df3087 2003-12-06 devnull m->tx.tag = m->tag;
576 d3df3087 2003-12-06 devnull m->tx.fid = f->fid;
577 d3df3087 2003-12-06 devnull m->fid = f;
578 d3df3087 2003-12-06 devnull f->ref++;
579 e2a1725d 2005-01-04 devnull msgincref(m);
580 ceb04770 2003-12-09 devnull sendomsg(m);
581 05b7f431 2004-03-02 devnull mm = recvp(c->internal);
582 05b7f431 2004-03-02 devnull assert(mm == m);
583 e2a1725d 2005-01-04 devnull msgclear(m);
584 32f69c36 2003-12-11 devnull msgput(m); /* got from recvp */
585 32f69c36 2003-12-11 devnull msgput(m); /* got from msgnew */
586 32f69c36 2003-12-11 devnull fidput(f); /* got from hash table */
587 05b7f431 2004-03-02 devnull hnext = h->next;
588 05b7f431 2004-03-02 devnull free(h);
589 d3df3087 2003-12-06 devnull }
590 d3df3087 2003-12-06 devnull }
591 d3df3087 2003-12-06 devnull
592 d3df3087 2003-12-06 devnull out:
593 e2a1725d 2005-01-04 devnull closeioproc(io);
594 d3df3087 2003-12-06 devnull assert(c->nmsg == 0);
595 d3df3087 2003-12-06 devnull assert(c->nfid == 0);
596 d3df3087 2003-12-06 devnull close(c->fd);
597 32f69c36 2003-12-11 devnull chanfree(c->internal);
598 32f69c36 2003-12-11 devnull c->internal = 0;
599 32f69c36 2003-12-11 devnull chanfree(c->inc);
600 32f69c36 2003-12-11 devnull c->inc = 0;
601 32f69c36 2003-12-11 devnull free(c->inq);
602 32f69c36 2003-12-11 devnull c->inq = 0;
603 d3df3087 2003-12-06 devnull free(c);
604 d3df3087 2003-12-06 devnull }
605 32f69c36 2003-12-11 devnull
606 32f69c36 2003-12-11 devnull static void
607 32f69c36 2003-12-11 devnull openfdthread(void *v)
608 32f69c36 2003-12-11 devnull {
609 32f69c36 2003-12-11 devnull Conn *c;
610 32f69c36 2003-12-11 devnull Fid *fid;
611 32f69c36 2003-12-11 devnull Msg *m;
612 32f69c36 2003-12-11 devnull int n;
613 32f69c36 2003-12-11 devnull vlong tot;
614 32f69c36 2003-12-11 devnull Ioproc *io;
615 32f69c36 2003-12-11 devnull char buf[1024];
616 32f69c36 2003-12-11 devnull
617 32f69c36 2003-12-11 devnull c = v;
618 32f69c36 2003-12-11 devnull fid = c->fdfid;
619 32f69c36 2003-12-11 devnull io = ioproc();
620 7ffc5208 2004-12-28 devnull threadsetname("openfd %s", c->fdfid);
621 32f69c36 2003-12-11 devnull tot = 0;
622 05b7f431 2004-03-02 devnull m = nil;
623 32f69c36 2003-12-11 devnull if(c->fdmode == OREAD){
624 32f69c36 2003-12-11 devnull for(;;){
625 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T tread...");
626 e2a1725d 2005-01-04 devnull m = msgnew(0);
627 32f69c36 2003-12-11 devnull m->internal = 1;
628 32f69c36 2003-12-11 devnull m->c = c;
629 32f69c36 2003-12-11 devnull m->tx.type = Tread;
630 5a8e63b2 2004-02-29 devnull m->tx.count = msize - IOHDRSZ;
631 32f69c36 2003-12-11 devnull m->tx.fid = fid->fid;
632 32f69c36 2003-12-11 devnull m->tx.tag = m->tag;
633 32f69c36 2003-12-11 devnull m->tx.offset = tot;
634 32f69c36 2003-12-11 devnull m->fid = fid;
635 32f69c36 2003-12-11 devnull fid->ref++;
636 e2a1725d 2005-01-04 devnull msgincref(m);
637 32f69c36 2003-12-11 devnull sendomsg(m);
638 32f69c36 2003-12-11 devnull recvp(c->internal);
639 49588d5d 2003-12-17 devnull if(m->rx.type == Rerror){
640 cbeb0b26 2006-04-01 devnull /* fprint(2, "%T read error: %s\n", m->rx.ename); */
641 32f69c36 2003-12-11 devnull break;
642 49588d5d 2003-12-17 devnull }
643 32f69c36 2003-12-11 devnull if(m->rx.count == 0)
644 32f69c36 2003-12-11 devnull break;
645 32f69c36 2003-12-11 devnull tot += m->rx.count;
646 49588d5d 2003-12-17 devnull if(iowrite(io, c->fd, m->rx.data, m->rx.count) != m->rx.count){
647 cbeb0b26 2006-04-01 devnull /* fprint(2, "%T pipe write error: %r\n"); */
648 32f69c36 2003-12-11 devnull break;
649 49588d5d 2003-12-17 devnull }
650 32f69c36 2003-12-11 devnull msgput(m);
651 32f69c36 2003-12-11 devnull msgput(m);
652 05b7f431 2004-03-02 devnull m = nil;
653 32f69c36 2003-12-11 devnull }
654 32f69c36 2003-12-11 devnull }else{
655 32f69c36 2003-12-11 devnull for(;;){
656 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T twrite...");
657 5a8e63b2 2004-02-29 devnull n = sizeof buf;
658 5a8e63b2 2004-02-29 devnull if(n > msize)
659 5a8e63b2 2004-02-29 devnull n = msize;
660 5a8e63b2 2004-02-29 devnull if((n=ioread(io, c->fd, buf, n)) <= 0){
661 49588d5d 2003-12-17 devnull if(n < 0)
662 e2a1725d 2005-01-04 devnull fprint(2, "%T pipe read error: %r\n");
663 32f69c36 2003-12-11 devnull break;
664 32f69c36 2003-12-11 devnull }
665 e2a1725d 2005-01-04 devnull m = msgnew(0);
666 32f69c36 2003-12-11 devnull m->internal = 1;
667 32f69c36 2003-12-11 devnull m->c = c;
668 32f69c36 2003-12-11 devnull m->tx.type = Twrite;
669 32f69c36 2003-12-11 devnull m->tx.fid = fid->fid;
670 32f69c36 2003-12-11 devnull m->tx.data = buf;
671 32f69c36 2003-12-11 devnull m->tx.count = n;
672 32f69c36 2003-12-11 devnull m->tx.tag = m->tag;
673 32f69c36 2003-12-11 devnull m->tx.offset = tot;
674 32f69c36 2003-12-11 devnull m->fid = fid;
675 32f69c36 2003-12-11 devnull fid->ref++;
676 e2a1725d 2005-01-04 devnull msgincref(m);
677 32f69c36 2003-12-11 devnull sendomsg(m);
678 32f69c36 2003-12-11 devnull recvp(c->internal);
679 49588d5d 2003-12-17 devnull if(m->rx.type == Rerror){
680 cbeb0b26 2006-04-01 devnull /* fprint(2, "%T write error: %s\n", m->rx.ename); */
681 49588d5d 2003-12-17 devnull }
682 05b7f431 2004-03-02 devnull tot += n;
683 32f69c36 2003-12-11 devnull msgput(m);
684 32f69c36 2003-12-11 devnull msgput(m);
685 05b7f431 2004-03-02 devnull m = nil;
686 32f69c36 2003-12-11 devnull }
687 32f69c36 2003-12-11 devnull }
688 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T eof on %d fid %d\n", c->fd, fid->fid);
689 32f69c36 2003-12-11 devnull close(c->fd);
690 32f69c36 2003-12-11 devnull closeioproc(io);
691 32f69c36 2003-12-11 devnull if(m){
692 32f69c36 2003-12-11 devnull msgput(m);
693 32f69c36 2003-12-11 devnull msgput(m);
694 32f69c36 2003-12-11 devnull }
695 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T eof on %d fid %d ref %d\n", c->fd, fid->fid, fid->ref);
696 21e626de 2004-12-28 devnull if(--fid->openfd == 0){
697 e2a1725d 2005-01-04 devnull m = msgnew(0);
698 49588d5d 2003-12-17 devnull m->internal = 1;
699 49588d5d 2003-12-17 devnull m->c = c;
700 49588d5d 2003-12-17 devnull m->tx.type = Tclunk;
701 05b7f431 2004-03-02 devnull m->tx.tag = m->tag;
702 49588d5d 2003-12-17 devnull m->tx.fid = fid->fid;
703 49588d5d 2003-12-17 devnull m->fid = fid;
704 49588d5d 2003-12-17 devnull fid->ref++;
705 e2a1725d 2005-01-04 devnull msgincref(m);
706 49588d5d 2003-12-17 devnull sendomsg(m);
707 49588d5d 2003-12-17 devnull recvp(c->internal);
708 49588d5d 2003-12-17 devnull msgput(m);
709 49588d5d 2003-12-17 devnull msgput(m);
710 49588d5d 2003-12-17 devnull }
711 32f69c36 2003-12-11 devnull fidput(fid);
712 32f69c36 2003-12-11 devnull c->fdfid = nil;
713 32f69c36 2003-12-11 devnull chanfree(c->internal);
714 32f69c36 2003-12-11 devnull c->internal = 0;
715 32f69c36 2003-12-11 devnull free(c);
716 32f69c36 2003-12-11 devnull }
717 d3df3087 2003-12-06 devnull
718 32f69c36 2003-12-11 devnull int
719 32f69c36 2003-12-11 devnull xopenfd(Msg *m)
720 32f69c36 2003-12-11 devnull {
721 32f69c36 2003-12-11 devnull char errs[ERRMAX];
722 32f69c36 2003-12-11 devnull int n, p[2];
723 32f69c36 2003-12-11 devnull Conn *nc;
724 32f69c36 2003-12-11 devnull
725 32f69c36 2003-12-11 devnull if(pipe(p) < 0){
726 32f69c36 2003-12-11 devnull rerrstr(errs, sizeof errs);
727 32f69c36 2003-12-11 devnull err(m, errs);
728 21e626de 2004-12-28 devnull /* XXX return here? */
729 32f69c36 2003-12-11 devnull }
730 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T xopen pipe %d %d...", p[0], p[1]);
731 32f69c36 2003-12-11 devnull
732 32f69c36 2003-12-11 devnull /* now we're committed. */
733 32f69c36 2003-12-11 devnull
734 32f69c36 2003-12-11 devnull /* a new connection for this fid */
735 32f69c36 2003-12-11 devnull nc = emalloc(sizeof(Conn));
736 32f69c36 2003-12-11 devnull nc->internal = chancreate(sizeof(void*), 0);
737 32f69c36 2003-12-11 devnull
738 32f69c36 2003-12-11 devnull /* a ref for us */
739 32f69c36 2003-12-11 devnull nc->fdfid = m->fid;
740 32f69c36 2003-12-11 devnull m->fid->ref++;
741 21e626de 2004-12-28 devnull nc->fdfid->openfd++;
742 32f69c36 2003-12-11 devnull nc->fdmode = m->tx.mode;
743 32f69c36 2003-12-11 devnull nc->fd = p[0];
744 32f69c36 2003-12-11 devnull
745 32f69c36 2003-12-11 devnull /* a thread to tend the pipe */
746 32f69c36 2003-12-11 devnull threadcreate(openfdthread, nc, STACK);
747 32f69c36 2003-12-11 devnull
748 49588d5d 2003-12-17 devnull /* if mode is ORDWR, that openfdthread will write; start a reader */
749 49588d5d 2003-12-17 devnull if((m->tx.mode&3) == ORDWR){
750 49588d5d 2003-12-17 devnull nc = emalloc(sizeof(Conn));
751 49588d5d 2003-12-17 devnull nc->internal = chancreate(sizeof(void*), 0);
752 49588d5d 2003-12-17 devnull nc->fdfid = m->fid;
753 49588d5d 2003-12-17 devnull m->fid->ref++;
754 21e626de 2004-12-28 devnull nc->fdfid->openfd++;
755 49588d5d 2003-12-17 devnull nc->fdmode = OREAD;
756 49588d5d 2003-12-17 devnull nc->fd = dup(p[0], -1);
757 49588d5d 2003-12-17 devnull threadcreate(openfdthread, nc, STACK);
758 49588d5d 2003-12-17 devnull }
759 49588d5d 2003-12-17 devnull
760 49588d5d 2003-12-17 devnull /* steal fid from other connection */
761 49588d5d 2003-12-17 devnull if(delhash(m->c->fid, m->fid->cfid, m->fid) == 0)
762 49588d5d 2003-12-17 devnull fidput(m->fid);
763 49588d5d 2003-12-17 devnull
764 32f69c36 2003-12-11 devnull /* rewrite as Ropenfd */
765 32f69c36 2003-12-11 devnull m->rx.type = Ropenfd;
766 32f69c36 2003-12-11 devnull n = GBIT32(m->rpkt);
767 32f69c36 2003-12-11 devnull m->rpkt = erealloc(m->rpkt, n+4);
768 32f69c36 2003-12-11 devnull PBIT32(m->rpkt+n, p[1]);
769 32f69c36 2003-12-11 devnull n += 4;
770 32f69c36 2003-12-11 devnull PBIT32(m->rpkt, n);
771 32f69c36 2003-12-11 devnull m->rpkt[4] = Ropenfd;
772 32f69c36 2003-12-11 devnull m->rx.unixfd = p[1];
773 32f69c36 2003-12-11 devnull return 0;
774 32f69c36 2003-12-11 devnull }
775 32f69c36 2003-12-11 devnull
776 d3df3087 2003-12-06 devnull void
777 d3df3087 2003-12-06 devnull connoutthread(void *arg)
778 d3df3087 2003-12-06 devnull {
779 d3df3087 2003-12-06 devnull int err;
780 d3df3087 2003-12-06 devnull Conn *c;
781 d3df3087 2003-12-06 devnull Msg *m, *om;
782 ceb04770 2003-12-09 devnull Ioproc *io;
783 d3df3087 2003-12-06 devnull
784 d3df3087 2003-12-06 devnull c = arg;
785 ceb04770 2003-12-09 devnull io = ioproc();
786 7ffc5208 2004-12-28 devnull threadsetname("connout %s", c->dir);
787 841d71b5 2007-08-22 rsc while((m = recvq(c->outq)) != nil){
788 d3df3087 2003-12-06 devnull err = m->tx.type+1 != m->rx.type;
789 32f69c36 2003-12-11 devnull if(!err && m->isopenfd)
790 32f69c36 2003-12-11 devnull if(xopenfd(m) < 0)
791 32f69c36 2003-12-11 devnull continue;
792 d3df3087 2003-12-06 devnull switch(m->tx.type){
793 d3df3087 2003-12-06 devnull case Tflush:
794 d3df3087 2003-12-06 devnull om = m->oldm;
795 ceb04770 2003-12-09 devnull if(om)
796 ceb04770 2003-12-09 devnull if(delhash(om->c->tag, om->ctag, om) == 0)
797 ceb04770 2003-12-09 devnull msgput(om);
798 d3df3087 2003-12-06 devnull break;
799 d3df3087 2003-12-06 devnull case Tclunk:
800 ceb04770 2003-12-09 devnull case Tremove:
801 ceb04770 2003-12-09 devnull if(m->fid)
802 ceb04770 2003-12-09 devnull if(delhash(m->c->fid, m->fid->cfid, m->fid) == 0)
803 ceb04770 2003-12-09 devnull fidput(m->fid);
804 d3df3087 2003-12-06 devnull break;
805 d3df3087 2003-12-06 devnull case Tauth:
806 ceb04770 2003-12-09 devnull if(err && m->afid){
807 80568494 2005-02-11 devnull if(verbose) fprint(2, "%T auth error\n");
808 ceb04770 2003-12-09 devnull if(delhash(m->c->fid, m->afid->cfid, m->afid) == 0)
809 80568494 2005-02-11 devnull fidput(m->afid);
810 ceb04770 2003-12-09 devnull }
811 ceb04770 2003-12-09 devnull break;
812 d3df3087 2003-12-06 devnull case Tattach:
813 ceb04770 2003-12-09 devnull if(err && m->fid)
814 d3df3087 2003-12-06 devnull if(delhash(m->c->fid, m->fid->cfid, m->fid) == 0)
815 d3df3087 2003-12-06 devnull fidput(m->fid);
816 d3df3087 2003-12-06 devnull break;
817 d3df3087 2003-12-06 devnull case Twalk:
818 5c84c448 2005-03-21 devnull if(err || m->rx.nwqid < m->tx.nwname)
819 5c84c448 2005-03-21 devnull if(m->tx.fid != m->tx.newfid && m->newfid)
820 d3df3087 2003-12-06 devnull if(delhash(m->c->fid, m->newfid->cfid, m->newfid) == 0)
821 d3df3087 2003-12-06 devnull fidput(m->newfid);
822 d3df3087 2003-12-06 devnull break;
823 215993f8 2005-09-13 devnull case Tread:
824 215993f8 2005-09-13 devnull break;
825 215993f8 2005-09-13 devnull case Tstat:
826 215993f8 2005-09-13 devnull break;
827 215993f8 2005-09-13 devnull case Topen:
828 215993f8 2005-09-13 devnull case Tcreate:
829 215993f8 2005-09-13 devnull m->fid->isdir = (m->rx.qid.type & QTDIR);
830 215993f8 2005-09-13 devnull break;
831 d3df3087 2003-12-06 devnull }
832 d3df3087 2003-12-06 devnull if(delhash(m->c->tag, m->ctag, m) == 0)
833 d3df3087 2003-12-06 devnull msgput(m);
834 e2a1725d 2005-01-04 devnull if(verbose > 1) fprint(2, "%T fd#%d <- %F\n", c->fd, &m->rx);
835 ceb04770 2003-12-09 devnull rewritehdr(&m->rx, m->rpkt);
836 ceb04770 2003-12-09 devnull if(mwrite9p(io, c->fd, m->rpkt) < 0)
837 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T write error: %r\n");
838 d3df3087 2003-12-06 devnull msgput(m);
839 d3df3087 2003-12-06 devnull if(c->inputstalled && c->nmsg < MAXMSG)
840 d3df3087 2003-12-06 devnull nbsendp(c->inc, 0);
841 d3df3087 2003-12-06 devnull }
842 ceb04770 2003-12-09 devnull closeioproc(io);
843 841d71b5 2007-08-22 rsc free(c->outq);
844 05b7f431 2004-03-02 devnull c->outq = nil;
845 841d71b5 2007-08-22 rsc sendp(c->outqdead, nil);
846 d3df3087 2003-12-06 devnull }
847 d3df3087 2003-12-06 devnull
848 d3df3087 2003-12-06 devnull void
849 d3df3087 2003-12-06 devnull outputthread(void *arg)
850 d3df3087 2003-12-06 devnull {
851 d3df3087 2003-12-06 devnull Msg *m;
852 ceb04770 2003-12-09 devnull Ioproc *io;
853 d3df3087 2003-12-06 devnull
854 d3df3087 2003-12-06 devnull USED(arg);
855 ceb04770 2003-12-09 devnull io = ioproc();
856 7ffc5208 2004-12-28 devnull threadsetname("output");
857 d3df3087 2003-12-06 devnull while((m = recvq(outq)) != nil){
858 841d71b5 2007-08-22 rsc if(m->sync){
859 841d71b5 2007-08-22 rsc sendp(m->c->outqdead, nil);
860 841d71b5 2007-08-22 rsc continue;
861 841d71b5 2007-08-22 rsc }
862 e2a1725d 2005-01-04 devnull if(verbose > 1) fprint(2, "%T * <- %F\n", &m->tx);
863 ceb04770 2003-12-09 devnull rewritehdr(&m->tx, m->tpkt);
864 ceb04770 2003-12-09 devnull if(mwrite9p(io, 1, m->tpkt) < 0)
865 d3df3087 2003-12-06 devnull sysfatal("output error: %r");
866 d3df3087 2003-12-06 devnull msgput(m);
867 d3df3087 2003-12-06 devnull }
868 ceb04770 2003-12-09 devnull closeioproc(io);
869 e2a1725d 2005-01-04 devnull fprint(2, "%T output eof\n");
870 32f69c36 2003-12-11 devnull threadexitsall(0);
871 d3df3087 2003-12-06 devnull }
872 d3df3087 2003-12-06 devnull
873 d3df3087 2003-12-06 devnull void
874 d3df3087 2003-12-06 devnull inputthread(void *arg)
875 d3df3087 2003-12-06 devnull {
876 d3df3087 2003-12-06 devnull uchar *pkt;
877 d3df3087 2003-12-06 devnull int n, nn, tag;
878 d3df3087 2003-12-06 devnull Msg *m;
879 ceb04770 2003-12-09 devnull Ioproc *io;
880 d3df3087 2003-12-06 devnull
881 7ffc5208 2004-12-28 devnull threadsetname("input");
882 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T input thread\n");
883 ceb04770 2003-12-09 devnull io = ioproc();
884 ceb04770 2003-12-09 devnull USED(arg);
885 ceb04770 2003-12-09 devnull while((pkt = read9ppkt(io, 0)) != nil){
886 d3df3087 2003-12-06 devnull n = GBIT32(pkt);
887 d3df3087 2003-12-06 devnull if(n < 7){
888 e2a1725d 2005-01-04 devnull fprint(2, "%T short 9P packet from server\n");
889 d3df3087 2003-12-06 devnull free(pkt);
890 d3df3087 2003-12-06 devnull continue;
891 d3df3087 2003-12-06 devnull }
892 e2a1725d 2005-01-04 devnull if(verbose > 2) fprint(2, "%T read %.*H\n", n, pkt);
893 d3df3087 2003-12-06 devnull tag = GBIT16(pkt+5);
894 d3df3087 2003-12-06 devnull if((m = msgget(tag)) == nil){
895 e2a1725d 2005-01-04 devnull fprint(2, "%T unexpected 9P response tag %d\n", tag);
896 d3df3087 2003-12-06 devnull free(pkt);
897 d3df3087 2003-12-06 devnull continue;
898 d3df3087 2003-12-06 devnull }
899 74dd0321 2010-12-07 rsc if((nn = convM2S(pkt, n, &m->rx)) != n){
900 e2a1725d 2005-01-04 devnull fprint(2, "%T bad packet - convM2S %d but %d\n", nn, n);
901 d3df3087 2003-12-06 devnull free(pkt);
902 d3df3087 2003-12-06 devnull msgput(m);
903 d3df3087 2003-12-06 devnull continue;
904 d3df3087 2003-12-06 devnull }
905 e2a1725d 2005-01-04 devnull if(verbose > 1) fprint(2, "%T * -> %F%s\n", &m->rx,
906 05b7f431 2004-03-02 devnull m->internal ? " (internal)" : "");
907 d3df3087 2003-12-06 devnull m->rpkt = pkt;
908 d3df3087 2003-12-06 devnull m->rx.tag = m->ctag;
909 ceb04770 2003-12-09 devnull if(m->internal)
910 05b7f431 2004-03-02 devnull sendp(m->c->internal, m);
911 05b7f431 2004-03-02 devnull else if(m->c->outq)
912 ceb04770 2003-12-09 devnull sendq(m->c->outq, m);
913 05b7f431 2004-03-02 devnull else
914 05b7f431 2004-03-02 devnull msgput(m);
915 d3df3087 2003-12-06 devnull }
916 ceb04770 2003-12-09 devnull closeioproc(io);
917 cbeb0b26 2006-04-01 devnull /*fprint(2, "%T input eof\n"); */
918 32f69c36 2003-12-11 devnull threadexitsall(0);
919 d3df3087 2003-12-06 devnull }
920 d3df3087 2003-12-06 devnull
921 d3df3087 2003-12-06 devnull void*
922 d3df3087 2003-12-06 devnull gethash(Hash **ht, uint n)
923 d3df3087 2003-12-06 devnull {
924 d3df3087 2003-12-06 devnull Hash *h;
925 d3df3087 2003-12-06 devnull
926 d3df3087 2003-12-06 devnull for(h=ht[n%NHASH]; h; h=h->next)
927 d3df3087 2003-12-06 devnull if(h->n == n)
928 d3df3087 2003-12-06 devnull return h->v;
929 d3df3087 2003-12-06 devnull return nil;
930 d3df3087 2003-12-06 devnull }
931 d3df3087 2003-12-06 devnull
932 d3df3087 2003-12-06 devnull int
933 d3df3087 2003-12-06 devnull delhash(Hash **ht, uint n, void *v)
934 d3df3087 2003-12-06 devnull {
935 d3df3087 2003-12-06 devnull Hash *h, **l;
936 d3df3087 2003-12-06 devnull
937 d3df3087 2003-12-06 devnull for(l=&ht[n%NHASH]; h=*l; l=&h->next)
938 d3df3087 2003-12-06 devnull if(h->n == n){
939 ceb04770 2003-12-09 devnull if(h->v != v){
940 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T delhash %d got %p want %p\n", n, h->v, v);
941 ceb04770 2003-12-09 devnull return -1;
942 ceb04770 2003-12-09 devnull }
943 d3df3087 2003-12-06 devnull *l = h->next;
944 d3df3087 2003-12-06 devnull free(h);
945 d3df3087 2003-12-06 devnull return 0;
946 d3df3087 2003-12-06 devnull }
947 d3df3087 2003-12-06 devnull return -1;
948 d3df3087 2003-12-06 devnull }
949 d3df3087 2003-12-06 devnull
950 d3df3087 2003-12-06 devnull int
951 d3df3087 2003-12-06 devnull puthash(Hash **ht, uint n, void *v)
952 d3df3087 2003-12-06 devnull {
953 d3df3087 2003-12-06 devnull Hash *h;
954 d3df3087 2003-12-06 devnull
955 d3df3087 2003-12-06 devnull if(gethash(ht, n))
956 d3df3087 2003-12-06 devnull return -1;
957 d3df3087 2003-12-06 devnull h = emalloc(sizeof(Hash));
958 d3df3087 2003-12-06 devnull h->next = ht[n%NHASH];
959 d3df3087 2003-12-06 devnull h->n = n;
960 d3df3087 2003-12-06 devnull h->v = v;
961 d3df3087 2003-12-06 devnull ht[n%NHASH] = h;
962 d3df3087 2003-12-06 devnull return 0;
963 d3df3087 2003-12-06 devnull }
964 d3df3087 2003-12-06 devnull
965 d3df3087 2003-12-06 devnull Fid **fidtab;
966 d3df3087 2003-12-06 devnull int nfidtab;
967 d3df3087 2003-12-06 devnull Fid *freefid;
968 d3df3087 2003-12-06 devnull
969 d3df3087 2003-12-06 devnull Fid*
970 d3df3087 2003-12-06 devnull fidnew(int cfid)
971 d3df3087 2003-12-06 devnull {
972 d3df3087 2003-12-06 devnull Fid *f;
973 d3df3087 2003-12-06 devnull
974 d3df3087 2003-12-06 devnull if(freefid == nil){
975 ceb04770 2003-12-09 devnull fidtab = erealloc(fidtab, (nfidtab+1)*sizeof(fidtab[0]));
976 5c84c448 2005-03-21 devnull if(nfidtab == xafid){
977 5c84c448 2005-03-21 devnull fidtab[nfidtab++] = nil;
978 5c84c448 2005-03-21 devnull fidtab = erealloc(fidtab, (nfidtab+1)*sizeof(fidtab[0]));
979 5c84c448 2005-03-21 devnull }
980 d3df3087 2003-12-06 devnull fidtab[nfidtab] = emalloc(sizeof(Fid));
981 ceb04770 2003-12-09 devnull freefid = fidtab[nfidtab];
982 ceb04770 2003-12-09 devnull freefid->fid = nfidtab++;
983 d3df3087 2003-12-06 devnull }
984 d3df3087 2003-12-06 devnull f = freefid;
985 d3df3087 2003-12-06 devnull freefid = f->next;
986 ceb04770 2003-12-09 devnull f->cfid = cfid;
987 d3df3087 2003-12-06 devnull f->ref = 1;
988 9777187b 2006-02-03 devnull f->offset = 0;
989 9777187b 2006-02-03 devnull f->coffset = 0;
990 215993f8 2005-09-13 devnull f->isdir = -1;
991 d3df3087 2003-12-06 devnull return f;
992 d3df3087 2003-12-06 devnull }
993 d3df3087 2003-12-06 devnull
994 d3df3087 2003-12-06 devnull void
995 d3df3087 2003-12-06 devnull fidput(Fid *f)
996 d3df3087 2003-12-06 devnull {
997 ceb04770 2003-12-09 devnull if(f == nil)
998 ceb04770 2003-12-09 devnull return;
999 d3df3087 2003-12-06 devnull assert(f->ref > 0);
1000 d3df3087 2003-12-06 devnull if(--f->ref > 0)
1001 d3df3087 2003-12-06 devnull return;
1002 d3df3087 2003-12-06 devnull f->next = freefid;
1003 d3df3087 2003-12-06 devnull f->cfid = -1;
1004 d3df3087 2003-12-06 devnull freefid = f;
1005 d3df3087 2003-12-06 devnull }
1006 d3df3087 2003-12-06 devnull
1007 d3df3087 2003-12-06 devnull Msg **msgtab;
1008 d3df3087 2003-12-06 devnull int nmsgtab;
1009 16c1defa 2005-11-12 devnull int nmsg;
1010 d3df3087 2003-12-06 devnull Msg *freemsg;
1011 d3df3087 2003-12-06 devnull
1012 e2a1725d 2005-01-04 devnull void
1013 e2a1725d 2005-01-04 devnull msgincref(Msg *m)
1014 e2a1725d 2005-01-04 devnull {
1015 e2a1725d 2005-01-04 devnull if(verbose > 1) fprint(2, "%T msgincref @0x%lux %p tag %d/%d ref %d=>%d\n",
1016 e2a1725d 2005-01-04 devnull getcallerpc(&m), m, m->tag, m->ctag, m->ref, m->ref+1);
1017 e2a1725d 2005-01-04 devnull m->ref++;
1018 e2a1725d 2005-01-04 devnull }
1019 e2a1725d 2005-01-04 devnull
1020 d3df3087 2003-12-06 devnull Msg*
1021 e2a1725d 2005-01-04 devnull msgnew(int x)
1022 d3df3087 2003-12-06 devnull {
1023 d3df3087 2003-12-06 devnull Msg *m;
1024 d3df3087 2003-12-06 devnull
1025 d3df3087 2003-12-06 devnull if(freemsg == nil){
1026 ceb04770 2003-12-09 devnull msgtab = erealloc(msgtab, (nmsgtab+1)*sizeof(msgtab[0]));
1027 d3df3087 2003-12-06 devnull msgtab[nmsgtab] = emalloc(sizeof(Msg));
1028 ceb04770 2003-12-09 devnull freemsg = msgtab[nmsgtab];
1029 ceb04770 2003-12-09 devnull freemsg->tag = nmsgtab++;
1030 d3df3087 2003-12-06 devnull }
1031 d3df3087 2003-12-06 devnull m = freemsg;
1032 d3df3087 2003-12-06 devnull freemsg = m->next;
1033 d3df3087 2003-12-06 devnull m->ref = 1;
1034 e2a1725d 2005-01-04 devnull if(verbose > 1) fprint(2, "%T msgnew @0x%lux %p tag %d ref %d\n",
1035 e2a1725d 2005-01-04 devnull getcallerpc(&x), m, m->tag, m->ref);
1036 16c1defa 2005-11-12 devnull nmsg++;
1037 d3df3087 2003-12-06 devnull return m;
1038 e2a1725d 2005-01-04 devnull }
1039 e2a1725d 2005-01-04 devnull
1040 e2a1725d 2005-01-04 devnull /*
1041 e2a1725d 2005-01-04 devnull * Clear data associated with connections, so that
1042 e2a1725d 2005-01-04 devnull * if all msgs have been msgcleared, the connection
1043 e2a1725d 2005-01-04 devnull * can be freed. Note that this does *not* free the tpkt
1044 e2a1725d 2005-01-04 devnull * and rpkt; they are freed in msgput with the msg itself.
1045 e2a1725d 2005-01-04 devnull * The io write thread might still be holding a ref to msg
1046 e2a1725d 2005-01-04 devnull * even once the connection has finished with it.
1047 e2a1725d 2005-01-04 devnull */
1048 e2a1725d 2005-01-04 devnull void
1049 e2a1725d 2005-01-04 devnull msgclear(Msg *m)
1050 e2a1725d 2005-01-04 devnull {
1051 e2a1725d 2005-01-04 devnull if(m->c){
1052 e2a1725d 2005-01-04 devnull m->c->nmsg--;
1053 e2a1725d 2005-01-04 devnull m->c = nil;
1054 e2a1725d 2005-01-04 devnull }
1055 e2a1725d 2005-01-04 devnull if(m->oldm){
1056 e2a1725d 2005-01-04 devnull msgput(m->oldm);
1057 e2a1725d 2005-01-04 devnull m->oldm = nil;
1058 e2a1725d 2005-01-04 devnull }
1059 e2a1725d 2005-01-04 devnull if(m->fid){
1060 e2a1725d 2005-01-04 devnull fidput(m->fid);
1061 e2a1725d 2005-01-04 devnull m->fid = nil;
1062 e2a1725d 2005-01-04 devnull }
1063 e2a1725d 2005-01-04 devnull if(m->afid){
1064 e2a1725d 2005-01-04 devnull fidput(m->afid);
1065 e2a1725d 2005-01-04 devnull m->afid = nil;
1066 e2a1725d 2005-01-04 devnull }
1067 e2a1725d 2005-01-04 devnull if(m->newfid){
1068 e2a1725d 2005-01-04 devnull fidput(m->newfid);
1069 e2a1725d 2005-01-04 devnull m->newfid = nil;
1070 e2a1725d 2005-01-04 devnull }
1071 e2a1725d 2005-01-04 devnull if(m->rx.type == Ropenfd && m->rx.unixfd >= 0){
1072 e2a1725d 2005-01-04 devnull close(m->rx.unixfd);
1073 e2a1725d 2005-01-04 devnull m->rx.unixfd = -1;
1074 e2a1725d 2005-01-04 devnull }
1075 d3df3087 2003-12-06 devnull }
1076 d3df3087 2003-12-06 devnull
1077 d3df3087 2003-12-06 devnull void
1078 d3df3087 2003-12-06 devnull msgput(Msg *m)
1079 d3df3087 2003-12-06 devnull {
1080 05b7f431 2004-03-02 devnull if(m == nil)
1081 05b7f431 2004-03-02 devnull return;
1082 05b7f431 2004-03-02 devnull
1083 e2a1725d 2005-01-04 devnull if(verbose > 1) fprint(2, "%T msgput 0x%lux %p tag %d/%d ref %d\n",
1084 e2a1725d 2005-01-04 devnull getcallerpc(&m), m, m->tag, m->ctag, m->ref);
1085 d3df3087 2003-12-06 devnull assert(m->ref > 0);
1086 d3df3087 2003-12-06 devnull if(--m->ref > 0)
1087 d3df3087 2003-12-06 devnull return;
1088 16c1defa 2005-11-12 devnull nmsg--;
1089 e2a1725d 2005-01-04 devnull msgclear(m);
1090 e2a1725d 2005-01-04 devnull if(m->tpkt){
1091 e2a1725d 2005-01-04 devnull free(m->tpkt);
1092 e2a1725d 2005-01-04 devnull m->tpkt = nil;
1093 e2a1725d 2005-01-04 devnull }
1094 e2a1725d 2005-01-04 devnull if(m->rpkt){
1095 e2a1725d 2005-01-04 devnull free(m->rpkt);
1096 e2a1725d 2005-01-04 devnull m->rpkt = nil;
1097 e2a1725d 2005-01-04 devnull }
1098 32f69c36 2003-12-11 devnull m->isopenfd = 0;
1099 32f69c36 2003-12-11 devnull m->internal = 0;
1100 d3df3087 2003-12-06 devnull m->next = freemsg;
1101 d3df3087 2003-12-06 devnull freemsg = m;
1102 d3df3087 2003-12-06 devnull }
1103 d3df3087 2003-12-06 devnull
1104 ceb04770 2003-12-09 devnull Msg*
1105 ceb04770 2003-12-09 devnull msgget(int n)
1106 ceb04770 2003-12-09 devnull {
1107 ceb04770 2003-12-09 devnull Msg *m;
1108 ceb04770 2003-12-09 devnull
1109 ceb04770 2003-12-09 devnull if(n < 0 || n >= nmsgtab)
1110 ceb04770 2003-12-09 devnull return nil;
1111 ceb04770 2003-12-09 devnull m = msgtab[n];
1112 ceb04770 2003-12-09 devnull if(m->ref == 0)
1113 ceb04770 2003-12-09 devnull return nil;
1114 e2a1725d 2005-01-04 devnull if(verbose) fprint(2, "%T msgget %d = %p\n", n, m);
1115 e2a1725d 2005-01-04 devnull msgincref(m);
1116 ceb04770 2003-12-09 devnull return m;
1117 ceb04770 2003-12-09 devnull }
1118 ceb04770 2003-12-09 devnull
1119 ceb04770 2003-12-09 devnull
1120 d3df3087 2003-12-06 devnull void*
1121 d3df3087 2003-12-06 devnull emalloc(int n)
1122 d3df3087 2003-12-06 devnull {
1123 d3df3087 2003-12-06 devnull void *v;
1124 d3df3087 2003-12-06 devnull
1125 d3df3087 2003-12-06 devnull v = mallocz(n, 1);
1126 ceb04770 2003-12-09 devnull if(v == nil){
1127 ceb04770 2003-12-09 devnull abort();
1128 ceb04770 2003-12-09 devnull sysfatal("out of memory allocating %d", n);
1129 ceb04770 2003-12-09 devnull }
1130 d3df3087 2003-12-06 devnull return v;
1131 d3df3087 2003-12-06 devnull }
1132 d3df3087 2003-12-06 devnull
1133 d3df3087 2003-12-06 devnull void*
1134 d3df3087 2003-12-06 devnull erealloc(void *v, int n)
1135 d3df3087 2003-12-06 devnull {
1136 d3df3087 2003-12-06 devnull v = realloc(v, n);
1137 ceb04770 2003-12-09 devnull if(v == nil){
1138 ceb04770 2003-12-09 devnull abort();
1139 ceb04770 2003-12-09 devnull sysfatal("out of memory reallocating %d", n);
1140 ceb04770 2003-12-09 devnull }
1141 d3df3087 2003-12-06 devnull return v;
1142 d3df3087 2003-12-06 devnull }
1143 d3df3087 2003-12-06 devnull
1144 d3df3087 2003-12-06 devnull typedef struct Qel Qel;
1145 d3df3087 2003-12-06 devnull struct Qel
1146 d3df3087 2003-12-06 devnull {
1147 d3df3087 2003-12-06 devnull Qel *next;
1148 d3df3087 2003-12-06 devnull void *p;
1149 d3df3087 2003-12-06 devnull };
1150 d3df3087 2003-12-06 devnull
1151 d3df3087 2003-12-06 devnull struct Queue
1152 d3df3087 2003-12-06 devnull {
1153 d3df3087 2003-12-06 devnull QLock lk;
1154 d3df3087 2003-12-06 devnull Rendez r;
1155 d3df3087 2003-12-06 devnull Qel *head;
1156 d3df3087 2003-12-06 devnull Qel *tail;
1157 d3df3087 2003-12-06 devnull };
1158 d3df3087 2003-12-06 devnull
1159 d3df3087 2003-12-06 devnull Queue*
1160 d3df3087 2003-12-06 devnull qalloc(void)
1161 d3df3087 2003-12-06 devnull {
1162 d3df3087 2003-12-06 devnull Queue *q;
1163 d3df3087 2003-12-06 devnull
1164 d3df3087 2003-12-06 devnull q = mallocz(sizeof(Queue), 1);
1165 d3df3087 2003-12-06 devnull if(q == nil)
1166 d3df3087 2003-12-06 devnull return nil;
1167 d3df3087 2003-12-06 devnull q->r.l = &q->lk;
1168 d3df3087 2003-12-06 devnull return q;
1169 d3df3087 2003-12-06 devnull }
1170 d3df3087 2003-12-06 devnull
1171 d3df3087 2003-12-06 devnull int
1172 d3df3087 2003-12-06 devnull sendq(Queue *q, void *p)
1173 d3df3087 2003-12-06 devnull {
1174 d3df3087 2003-12-06 devnull Qel *e;
1175 d3df3087 2003-12-06 devnull
1176 d3df3087 2003-12-06 devnull e = emalloc(sizeof(Qel));
1177 d3df3087 2003-12-06 devnull qlock(&q->lk);
1178 d3df3087 2003-12-06 devnull e->p = p;
1179 d3df3087 2003-12-06 devnull e->next = nil;
1180 d3df3087 2003-12-06 devnull if(q->head == nil)
1181 d3df3087 2003-12-06 devnull q->head = e;
1182 d3df3087 2003-12-06 devnull else
1183 d3df3087 2003-12-06 devnull q->tail->next = e;
1184 d3df3087 2003-12-06 devnull q->tail = e;
1185 d3df3087 2003-12-06 devnull rwakeup(&q->r);
1186 d3df3087 2003-12-06 devnull qunlock(&q->lk);
1187 d3df3087 2003-12-06 devnull return 0;
1188 d3df3087 2003-12-06 devnull }
1189 d3df3087 2003-12-06 devnull
1190 d3df3087 2003-12-06 devnull void*
1191 d3df3087 2003-12-06 devnull recvq(Queue *q)
1192 d3df3087 2003-12-06 devnull {
1193 d3df3087 2003-12-06 devnull void *p;
1194 d3df3087 2003-12-06 devnull Qel *e;
1195 d3df3087 2003-12-06 devnull
1196 d3df3087 2003-12-06 devnull qlock(&q->lk);
1197 841d71b5 2007-08-22 rsc while(q->head == nil)
1198 d3df3087 2003-12-06 devnull rsleep(&q->r);
1199 d3df3087 2003-12-06 devnull e = q->head;
1200 d3df3087 2003-12-06 devnull q->head = e->next;
1201 d3df3087 2003-12-06 devnull qunlock(&q->lk);
1202 d3df3087 2003-12-06 devnull p = e->p;
1203 d3df3087 2003-12-06 devnull free(e);
1204 d3df3087 2003-12-06 devnull return p;
1205 d3df3087 2003-12-06 devnull }
1206 ceb04770 2003-12-09 devnull
1207 ceb04770 2003-12-09 devnull uchar*
1208 ceb04770 2003-12-09 devnull read9ppkt(Ioproc *io, int fd)
1209 ceb04770 2003-12-09 devnull {
1210 ceb04770 2003-12-09 devnull uchar buf[4], *pkt;
1211 ceb04770 2003-12-09 devnull int n, nn;
1212 ceb04770 2003-12-09 devnull
1213 ceb04770 2003-12-09 devnull n = ioreadn(io, fd, buf, 4);
1214 ceb04770 2003-12-09 devnull if(n != 4)
1215 ceb04770 2003-12-09 devnull return nil;
1216 ceb04770 2003-12-09 devnull n = GBIT32(buf);
1217 f2466297 2006-05-04 devnull if(n > MAXMSGSIZE)
1218 f2466297 2006-05-04 devnull return nil;
1219 ceb04770 2003-12-09 devnull pkt = emalloc(n);
1220 ceb04770 2003-12-09 devnull PBIT32(pkt, n);
1221 ceb04770 2003-12-09 devnull nn = ioreadn(io, fd, pkt+4, n-4);
1222 ceb04770 2003-12-09 devnull if(nn != n-4){
1223 ceb04770 2003-12-09 devnull free(pkt);
1224 ceb04770 2003-12-09 devnull return nil;
1225 ceb04770 2003-12-09 devnull }
1226 32f69c36 2003-12-11 devnull /* would do this if we ever got one of these, but we only generate them
1227 32f69c36 2003-12-11 devnull if(pkt[4] == Ropenfd){
1228 32f69c36 2003-12-11 devnull newfd = iorecvfd(io, fd);
1229 32f69c36 2003-12-11 devnull PBIT32(pkt+n-4, newfd);
1230 32f69c36 2003-12-11 devnull }
1231 32f69c36 2003-12-11 devnull */
1232 ceb04770 2003-12-09 devnull return pkt;
1233 ceb04770 2003-12-09 devnull }
1234 ceb04770 2003-12-09 devnull
1235 ceb04770 2003-12-09 devnull Msg*
1236 74dd0321 2010-12-07 rsc mread9p(Ioproc *io, int fd)
1237 ceb04770 2003-12-09 devnull {
1238 ceb04770 2003-12-09 devnull int n, nn;
1239 ceb04770 2003-12-09 devnull uchar *pkt;
1240 ceb04770 2003-12-09 devnull Msg *m;
1241 ceb04770 2003-12-09 devnull
1242 ceb04770 2003-12-09 devnull if((pkt = read9ppkt(io, fd)) == nil)
1243 ceb04770 2003-12-09 devnull return nil;
1244 ceb04770 2003-12-09 devnull
1245 e2a1725d 2005-01-04 devnull m = msgnew(0);
1246 ceb04770 2003-12-09 devnull m->tpkt = pkt;
1247 ceb04770 2003-12-09 devnull n = GBIT32(pkt);
1248 74dd0321 2010-12-07 rsc nn = convM2S(pkt, n, &m->tx);
1249 ceb04770 2003-12-09 devnull if(nn != n){
1250 e2a1725d 2005-01-04 devnull fprint(2, "%T read bad packet from %d\n", fd);
1251 d5791246 2018-03-23 0intro free(m->tpkt);
1252 d5791246 2018-03-23 0intro free(m);
1253 ceb04770 2003-12-09 devnull return nil;
1254 ceb04770 2003-12-09 devnull }
1255 ceb04770 2003-12-09 devnull return m;
1256 ceb04770 2003-12-09 devnull }
1257 ceb04770 2003-12-09 devnull
1258 ceb04770 2003-12-09 devnull int
1259 ceb04770 2003-12-09 devnull mwrite9p(Ioproc *io, int fd, uchar *pkt)
1260 ceb04770 2003-12-09 devnull {
1261 32f69c36 2003-12-11 devnull int n, nfd;
1262 ceb04770 2003-12-09 devnull
1263 ceb04770 2003-12-09 devnull n = GBIT32(pkt);
1264 e2a1725d 2005-01-04 devnull if(verbose > 2) fprint(2, "%T write %d %d %.*H\n", fd, n, n, pkt);
1265 bb426e77 2005-01-06 devnull if(verbose > 1) fprint(2, "%T before iowrite\n");
1266 ceb04770 2003-12-09 devnull if(iowrite(io, fd, pkt, n) != n){
1267 e2a1725d 2005-01-04 devnull fprint(2, "%T write error: %r\n");
1268 ceb04770 2003-12-09 devnull return -1;
1269 ceb04770 2003-12-09 devnull }
1270 bb426e77 2005-01-06 devnull if(verbose > 1) fprint(2, "%T after iowrite\n");
1271 32f69c36 2003-12-11 devnull if(pkt[4] == Ropenfd){
1272 32f69c36 2003-12-11 devnull nfd = GBIT32(pkt+n-4);
1273 32f69c36 2003-12-11 devnull if(iosendfd(io, fd, nfd) < 0){
1274 e2a1725d 2005-01-04 devnull fprint(2, "%T send fd error: %r\n");
1275 32f69c36 2003-12-11 devnull return -1;
1276 32f69c36 2003-12-11 devnull }
1277 32f69c36 2003-12-11 devnull }
1278 ceb04770 2003-12-09 devnull return 0;
1279 ceb04770 2003-12-09 devnull }
1280 ceb04770 2003-12-09 devnull
1281 ceb04770 2003-12-09 devnull void
1282 ceb04770 2003-12-09 devnull restring(uchar *pkt, int pn, char *s)
1283 ceb04770 2003-12-09 devnull {
1284 ceb04770 2003-12-09 devnull int n;
1285 ceb04770 2003-12-09 devnull
1286 ceb04770 2003-12-09 devnull if(s < (char*)pkt || s >= (char*)pkt+pn)
1287 ceb04770 2003-12-09 devnull return;
1288 ceb04770 2003-12-09 devnull
1289 ceb04770 2003-12-09 devnull n = strlen(s);
1290 ceb04770 2003-12-09 devnull memmove(s+1, s, n);
1291 ceb04770 2003-12-09 devnull PBIT16((uchar*)s-1, n);
1292 5c84c448 2005-03-21 devnull }
1293 5c84c448 2005-03-21 devnull
1294 5c84c448 2005-03-21 devnull void
1295 74dd0321 2010-12-07 rsc repack(Fcall *f, uchar **ppkt)
1296 5c84c448 2005-03-21 devnull {
1297 5c84c448 2005-03-21 devnull uint n, nn;
1298 5c84c448 2005-03-21 devnull uchar *pkt;
1299 5c84c448 2005-03-21 devnull
1300 5c84c448 2005-03-21 devnull pkt = *ppkt;
1301 5c84c448 2005-03-21 devnull n = GBIT32(pkt);
1302 74dd0321 2010-12-07 rsc nn = sizeS2M(f);
1303 5c84c448 2005-03-21 devnull if(nn > n){
1304 5c84c448 2005-03-21 devnull free(pkt);
1305 5c84c448 2005-03-21 devnull pkt = emalloc(nn);
1306 5c84c448 2005-03-21 devnull *ppkt = pkt;
1307 5c84c448 2005-03-21 devnull }
1308 74dd0321 2010-12-07 rsc n = convS2M(f, pkt, nn);
1309 9e2f1d9b 2005-11-21 devnull if(n <= BIT16SZ)
1310 9e2f1d9b 2005-11-21 devnull sysfatal("convS2M conversion error");
1311 9e2f1d9b 2005-11-21 devnull if(n != nn)
1312 74dd0321 2010-12-07 rsc sysfatal("convS2M and sizeS2M disagree");
1313 ceb04770 2003-12-09 devnull }
1314 ceb04770 2003-12-09 devnull
1315 ceb04770 2003-12-09 devnull void
1316 ceb04770 2003-12-09 devnull rewritehdr(Fcall *f, uchar *pkt)
1317 ceb04770 2003-12-09 devnull {
1318 ceb04770 2003-12-09 devnull int i, n;
1319 ceb04770 2003-12-09 devnull
1320 ceb04770 2003-12-09 devnull n = GBIT32(pkt);
1321 ceb04770 2003-12-09 devnull PBIT16(pkt+5, f->tag);
1322 ceb04770 2003-12-09 devnull switch(f->type){
1323 ceb04770 2003-12-09 devnull case Tversion:
1324 ceb04770 2003-12-09 devnull case Rversion:
1325 ceb04770 2003-12-09 devnull restring(pkt, n, f->version);
1326 ceb04770 2003-12-09 devnull break;
1327 ceb04770 2003-12-09 devnull case Tauth:
1328 ceb04770 2003-12-09 devnull PBIT32(pkt+7, f->afid);
1329 ceb04770 2003-12-09 devnull restring(pkt, n, f->uname);
1330 ceb04770 2003-12-09 devnull restring(pkt, n, f->aname);
1331 ceb04770 2003-12-09 devnull break;
1332 ceb04770 2003-12-09 devnull case Tflush:
1333 ceb04770 2003-12-09 devnull PBIT16(pkt+7, f->oldtag);
1334 ceb04770 2003-12-09 devnull break;
1335 ceb04770 2003-12-09 devnull case Tattach:
1336 ceb04770 2003-12-09 devnull restring(pkt, n, f->uname);
1337 ceb04770 2003-12-09 devnull restring(pkt, n, f->aname);
1338 ceb04770 2003-12-09 devnull PBIT32(pkt+7, f->fid);
1339 ceb04770 2003-12-09 devnull PBIT32(pkt+11, f->afid);
1340 ceb04770 2003-12-09 devnull break;
1341 ceb04770 2003-12-09 devnull case Twalk:
1342 ceb04770 2003-12-09 devnull PBIT32(pkt+7, f->fid);
1343 ceb04770 2003-12-09 devnull PBIT32(pkt+11, f->newfid);
1344 ceb04770 2003-12-09 devnull for(i=0; i<f->nwname; i++)
1345 ceb04770 2003-12-09 devnull restring(pkt, n, f->wname[i]);
1346 ceb04770 2003-12-09 devnull break;
1347 ceb04770 2003-12-09 devnull case Tcreate:
1348 ceb04770 2003-12-09 devnull restring(pkt, n, f->name);
1349 ceb04770 2003-12-09 devnull /* fall through */
1350 ceb04770 2003-12-09 devnull case Topen:
1351 ceb04770 2003-12-09 devnull case Tclunk:
1352 ceb04770 2003-12-09 devnull case Tremove:
1353 ceb04770 2003-12-09 devnull case Tstat:
1354 ceb04770 2003-12-09 devnull case Twstat:
1355 9777187b 2006-02-03 devnull case Twrite:
1356 ceb04770 2003-12-09 devnull PBIT32(pkt+7, f->fid);
1357 ceb04770 2003-12-09 devnull break;
1358 9777187b 2006-02-03 devnull case Tread:
1359 9777187b 2006-02-03 devnull PBIT32(pkt+7, f->fid);
1360 9777187b 2006-02-03 devnull PBIT64(pkt+11, f->offset);
1361 9777187b 2006-02-03 devnull break;
1362 ceb04770 2003-12-09 devnull case Rerror:
1363 ceb04770 2003-12-09 devnull restring(pkt, n, f->ename);
1364 5a8e63b2 2004-02-29 devnull break;
1365 5a8e63b2 2004-02-29 devnull }
1366 32f69c36 2003-12-11 devnull }
1367 32f69c36 2003-12-11 devnull
1368 32f69c36 2003-12-11 devnull static long
1369 ceb04770 2003-12-09 devnull _iolisten(va_list *arg)
1370 ceb04770 2003-12-09 devnull {
1371 ceb04770 2003-12-09 devnull char *a, *b;
1372 ceb04770 2003-12-09 devnull
1373 ceb04770 2003-12-09 devnull a = va_arg(*arg, char*);
1374 ceb04770 2003-12-09 devnull b = va_arg(*arg, char*);
1375 ceb04770 2003-12-09 devnull return listen(a, b);
1376 ceb04770 2003-12-09 devnull }
1377 ceb04770 2003-12-09 devnull
1378 ceb04770 2003-12-09 devnull int
1379 ceb04770 2003-12-09 devnull iolisten(Ioproc *io, char *a, char *b)
1380 ceb04770 2003-12-09 devnull {
1381 ceb04770 2003-12-09 devnull return iocall(io, _iolisten, a, b);
1382 ceb04770 2003-12-09 devnull }
1383 ceb04770 2003-12-09 devnull
1384 ceb04770 2003-12-09 devnull static long
1385 ceb04770 2003-12-09 devnull _ioaccept(va_list *arg)
1386 ceb04770 2003-12-09 devnull {
1387 ceb04770 2003-12-09 devnull int fd;
1388 ceb04770 2003-12-09 devnull char *dir;
1389 ceb04770 2003-12-09 devnull
1390 ceb04770 2003-12-09 devnull fd = va_arg(*arg, int);
1391 ceb04770 2003-12-09 devnull dir = va_arg(*arg, char*);
1392 ceb04770 2003-12-09 devnull return accept(fd, dir);
1393 ceb04770 2003-12-09 devnull }
1394 ceb04770 2003-12-09 devnull
1395 ceb04770 2003-12-09 devnull int
1396 ceb04770 2003-12-09 devnull ioaccept(Ioproc *io, int fd, char *dir)
1397 ceb04770 2003-12-09 devnull {
1398 ceb04770 2003-12-09 devnull return iocall(io, _ioaccept, fd, dir);
1399 ceb04770 2003-12-09 devnull }
1400 e2a1725d 2005-01-04 devnull
1401 e2a1725d 2005-01-04 devnull int
1402 e2a1725d 2005-01-04 devnull timefmt(Fmt *fmt)
1403 e2a1725d 2005-01-04 devnull {
1404 e2a1725d 2005-01-04 devnull static char *mon[] = { "Jan", "Feb", "Mar", "Apr", "May", "Jun",
1405 e2a1725d 2005-01-04 devnull "Jul", "Aug", "Sep", "Oct", "Nov", "Dec" };
1406 bb426e77 2005-01-06 devnull vlong ns;
1407 e2a1725d 2005-01-04 devnull Tm tm;
1408 bb426e77 2005-01-06 devnull ns = nsec();
1409 e2a1725d 2005-01-04 devnull tm = *localtime(time(0));
1410 bb426e77 2005-01-06 devnull return fmtprint(fmt, "%s %2d %02d:%02d:%02d.%03d",
1411 bb426e77 2005-01-06 devnull mon[tm.mon], tm.mday, tm.hour, tm.min, tm.sec,
1412 bb426e77 2005-01-06 devnull (int)(ns%1000000000)/1000000);
1413 e2a1725d 2005-01-04 devnull }