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