1 3d7e9092 2003-10-14 devnull #include <u.h>
2 3d7e9092 2003-10-14 devnull #include <libc.h>
3 3d7e9092 2003-10-14 devnull #include "plumb.h"
5 3d7e9092 2003-10-14 devnull static char attrbuf[4096];
7 b7e6f415 2003-11-23 devnull char *home;
10 3d7e9092 2003-10-14 devnull plumbopen(char *name, int omode)
12 3d7e9092 2003-10-14 devnull int fd, f;
14 b7e6f415 2003-11-23 devnull char buf[256];
16 3d7e9092 2003-10-14 devnull if(name[0] == '/')
17 3d7e9092 2003-10-14 devnull return open(name, omode);
18 b7e6f415 2003-11-23 devnull if(home == nil){
19 b7e6f415 2003-11-23 devnull home = getenv("HOME");
20 b7e6f415 2003-11-23 devnull if(home == nil)
21 b7e6f415 2003-11-23 devnull return -1;
23 b7e6f415 2003-11-23 devnull snprint(buf, sizeof buf, "%s/mnt/plumb", home);
24 b7e6f415 2003-11-23 devnull /* fd = open(buf, omode);
25 3d7e9092 2003-10-14 devnull if(fd >= 0)
26 3d7e9092 2003-10-14 devnull return fd;
27 3d7e9092 2003-10-14 devnull snprint(buf, sizeof buf, "/mnt/term/mnt/plumb/%s", name);
28 3d7e9092 2003-10-14 devnull fd = open(buf, omode);
29 3d7e9092 2003-10-14 devnull if(fd >= 0)
30 3d7e9092 2003-10-14 devnull return fd;
31 b7e6f415 2003-11-23 devnull /* try mounting service * /
32 3d7e9092 2003-10-14 devnull s = getenv("plumbsrv");
33 3d7e9092 2003-10-14 devnull if(s == nil)
34 3d7e9092 2003-10-14 devnull return -1;
35 3d7e9092 2003-10-14 devnull snprint(buf, sizeof buf, "/mnt/plumb/%s", name);
37 3d7e9092 2003-10-14 devnull return open(buf, omode);
40 3d7e9092 2003-10-14 devnull static int
41 3d7e9092 2003-10-14 devnull Strlen(char *s)
43 3d7e9092 2003-10-14 devnull if(s == nil)
44 3d7e9092 2003-10-14 devnull return 0;
45 3d7e9092 2003-10-14 devnull return strlen(s);
48 3d7e9092 2003-10-14 devnull static char*
49 3d7e9092 2003-10-14 devnull Strcpy(char *s, char *t)
51 3d7e9092 2003-10-14 devnull if(t == nil)
52 3d7e9092 2003-10-14 devnull return s;
53 3d7e9092 2003-10-14 devnull return strcpy(s, t) + strlen(t);
56 3d7e9092 2003-10-14 devnull /* quote attribute value, if necessary */
57 3d7e9092 2003-10-14 devnull static char*
58 3d7e9092 2003-10-14 devnull quote(char *s)
63 3d7e9092 2003-10-14 devnull if(s == nil){
64 3d7e9092 2003-10-14 devnull attrbuf[0] = '\0';
65 3d7e9092 2003-10-14 devnull return attrbuf;
67 3d7e9092 2003-10-14 devnull if(strpbrk(s, " '=\t") == nil)
68 3d7e9092 2003-10-14 devnull return s;
69 3d7e9092 2003-10-14 devnull t = attrbuf;
70 3d7e9092 2003-10-14 devnull *t++ = '\'';
71 3d7e9092 2003-10-14 devnull while(t < attrbuf+sizeof attrbuf-2){
72 3d7e9092 2003-10-14 devnull c = *s++;
73 3d7e9092 2003-10-14 devnull if(c == '\0')
75 3d7e9092 2003-10-14 devnull *t++ = c;
76 3d7e9092 2003-10-14 devnull if(c == '\'')
77 3d7e9092 2003-10-14 devnull *t++ = c;
79 3d7e9092 2003-10-14 devnull *t++ = '\'';
80 3d7e9092 2003-10-14 devnull *t = '\0';
81 3d7e9092 2003-10-14 devnull return attrbuf;
85 3d7e9092 2003-10-14 devnull plumbpackattr(Plumbattr *attr)
88 3d7e9092 2003-10-14 devnull Plumbattr *a;
89 3d7e9092 2003-10-14 devnull char *s, *t;
91 3d7e9092 2003-10-14 devnull if(attr == nil)
92 3d7e9092 2003-10-14 devnull return nil;
94 3d7e9092 2003-10-14 devnull for(a=attr; a!=nil; a=a->next)
95 3d7e9092 2003-10-14 devnull n += Strlen(a->name) + 1 + Strlen(quote(a->value)) + 1;
96 3d7e9092 2003-10-14 devnull s = malloc(n);
97 3d7e9092 2003-10-14 devnull if(s == nil)
98 3d7e9092 2003-10-14 devnull return nil;
100 3d7e9092 2003-10-14 devnull *t = '\0';
101 3d7e9092 2003-10-14 devnull for(a=attr; a!=nil; a=a->next){
102 3d7e9092 2003-10-14 devnull if(t != s)
103 3d7e9092 2003-10-14 devnull *t++ = ' ';
104 3d7e9092 2003-10-14 devnull strcpy(t, a->name);
105 3d7e9092 2003-10-14 devnull strcat(t, "=");
106 3d7e9092 2003-10-14 devnull strcat(t, quote(a->value));
107 3d7e9092 2003-10-14 devnull t += strlen(t);
109 3d7e9092 2003-10-14 devnull if(t > s+n)
110 3d7e9092 2003-10-14 devnull abort();
111 3d7e9092 2003-10-14 devnull return s;
115 3d7e9092 2003-10-14 devnull plumblookup(Plumbattr *attr, char *name)
117 3d7e9092 2003-10-14 devnull while(attr){
118 3d7e9092 2003-10-14 devnull if(strcmp(attr->name, name) == 0)
119 3d7e9092 2003-10-14 devnull return attr->value;
120 3d7e9092 2003-10-14 devnull attr = attr->next;
122 3d7e9092 2003-10-14 devnull return nil;
126 3d7e9092 2003-10-14 devnull plumbpack(Plumbmsg *m, int *np)
128 3d7e9092 2003-10-14 devnull int n, ndata;
129 3d7e9092 2003-10-14 devnull char *buf, *p, *attr;
131 3d7e9092 2003-10-14 devnull ndata = m->ndata;
132 3d7e9092 2003-10-14 devnull if(ndata < 0)
133 3d7e9092 2003-10-14 devnull ndata = Strlen(m->data);
134 3d7e9092 2003-10-14 devnull attr = plumbpackattr(m->attr);
135 3d7e9092 2003-10-14 devnull n = Strlen(m->src)+1 + Strlen(m->dst)+1 + Strlen(m->wdir)+1 +
136 3d7e9092 2003-10-14 devnull Strlen(m->type)+1 + Strlen(attr)+1 + 16 + ndata;
137 3d7e9092 2003-10-14 devnull buf = malloc(n+1); /* +1 for '\0' */
138 3d7e9092 2003-10-14 devnull if(buf == nil){
139 3d7e9092 2003-10-14 devnull free(attr);
140 3d7e9092 2003-10-14 devnull return nil;
142 3d7e9092 2003-10-14 devnull p = Strcpy(buf, m->src);
143 3d7e9092 2003-10-14 devnull *p++ = '\n';
144 3d7e9092 2003-10-14 devnull p = Strcpy(p, m->dst);
145 3d7e9092 2003-10-14 devnull *p++ = '\n';
146 3d7e9092 2003-10-14 devnull p = Strcpy(p, m->wdir);
147 3d7e9092 2003-10-14 devnull *p++ = '\n';
148 3d7e9092 2003-10-14 devnull p = Strcpy(p, m->type);
149 3d7e9092 2003-10-14 devnull *p++ = '\n';
150 3d7e9092 2003-10-14 devnull p = Strcpy(p, attr);
151 3d7e9092 2003-10-14 devnull *p++ = '\n';
152 3d7e9092 2003-10-14 devnull p += sprint(p, "%d\n", ndata);
153 3d7e9092 2003-10-14 devnull memmove(p, m->data, ndata);
154 3d7e9092 2003-10-14 devnull *np = (p-buf)+ndata;
155 3d7e9092 2003-10-14 devnull buf[*np] = '\0'; /* null terminate just in case */
156 3d7e9092 2003-10-14 devnull if(*np >= n+1)
157 3d7e9092 2003-10-14 devnull abort();
158 3d7e9092 2003-10-14 devnull free(attr);
159 3d7e9092 2003-10-14 devnull return buf;
163 3d7e9092 2003-10-14 devnull plumbsend(int fd, Plumbmsg *m)
165 3d7e9092 2003-10-14 devnull char *buf;
168 3d7e9092 2003-10-14 devnull buf = plumbpack(m, &n);
169 3d7e9092 2003-10-14 devnull if(buf == nil)
170 3d7e9092 2003-10-14 devnull return -1;
171 3d7e9092 2003-10-14 devnull n = write(fd, buf, n);
172 3d7e9092 2003-10-14 devnull free(buf);
173 3d7e9092 2003-10-14 devnull return n;
176 3d7e9092 2003-10-14 devnull static int
177 3d7e9092 2003-10-14 devnull plumbline(char **linep, char *buf, int i, int n, int *bad)
179 3d7e9092 2003-10-14 devnull int starti;
180 3d7e9092 2003-10-14 devnull char *p;
182 3d7e9092 2003-10-14 devnull if(*bad)
183 3d7e9092 2003-10-14 devnull return i;
184 3d7e9092 2003-10-14 devnull starti = i;
185 3d7e9092 2003-10-14 devnull while(i<n && buf[i]!='\n')
187 3d7e9092 2003-10-14 devnull if(i == n)
188 3d7e9092 2003-10-14 devnull *bad = 1;
190 3d7e9092 2003-10-14 devnull p = malloc((i-starti) + 1);
191 3d7e9092 2003-10-14 devnull if(p == nil)
192 3d7e9092 2003-10-14 devnull *bad = 1;
194 3d7e9092 2003-10-14 devnull memmove(p, buf+starti, i-starti);
195 3d7e9092 2003-10-14 devnull p[i-starti] = '\0';
197 3d7e9092 2003-10-14 devnull *linep = p;
200 3d7e9092 2003-10-14 devnull return i;
204 3d7e9092 2003-10-14 devnull plumbfree(Plumbmsg *m)
206 3d7e9092 2003-10-14 devnull Plumbattr *a, *next;
208 3d7e9092 2003-10-14 devnull free(m->src);
209 3d7e9092 2003-10-14 devnull free(m->dst);
210 3d7e9092 2003-10-14 devnull free(m->wdir);
211 3d7e9092 2003-10-14 devnull free(m->type);
212 3d7e9092 2003-10-14 devnull for(a=m->attr; a!=nil; a=next){
213 3d7e9092 2003-10-14 devnull next = a->next;
214 3d7e9092 2003-10-14 devnull free(a->name);
215 3d7e9092 2003-10-14 devnull free(a->value);
216 3d7e9092 2003-10-14 devnull free(a);
218 3d7e9092 2003-10-14 devnull free(m->data);
219 3d7e9092 2003-10-14 devnull free(m);
222 3d7e9092 2003-10-14 devnull Plumbattr*
223 3d7e9092 2003-10-14 devnull plumbunpackattr(char *p)
225 3d7e9092 2003-10-14 devnull Plumbattr *attr, *prev, *a;
226 3d7e9092 2003-10-14 devnull char *q, *v;
227 3d7e9092 2003-10-14 devnull int c, quoting;
229 3d7e9092 2003-10-14 devnull attr = prev = nil;
230 3d7e9092 2003-10-14 devnull while(*p!='\0' && *p!='\n'){
231 3d7e9092 2003-10-14 devnull while(*p==' ' || *p=='\t')
233 3d7e9092 2003-10-14 devnull if(*p == '\0')
235 3d7e9092 2003-10-14 devnull for(q=p; *q!='\0' && *q!='\n' && *q!=' ' && *q!='\t'; q++)
236 3d7e9092 2003-10-14 devnull if(*q == '=')
238 3d7e9092 2003-10-14 devnull if(*q != '=')
239 3d7e9092 2003-10-14 devnull break; /* malformed attribute */
240 3d7e9092 2003-10-14 devnull a = malloc(sizeof(Plumbattr));
241 3d7e9092 2003-10-14 devnull if(a == nil)
243 3d7e9092 2003-10-14 devnull a->name = malloc(q-p+1);
244 3d7e9092 2003-10-14 devnull if(a->name == nil){
245 3d7e9092 2003-10-14 devnull free(a);
248 3d7e9092 2003-10-14 devnull memmove(a->name, p, q-p);
249 3d7e9092 2003-10-14 devnull a->name[q-p] = '\0';
250 3d7e9092 2003-10-14 devnull /* process quotes in value */
251 3d7e9092 2003-10-14 devnull q++; /* skip '=' */
252 3d7e9092 2003-10-14 devnull v = attrbuf;
253 3d7e9092 2003-10-14 devnull quoting = 0;
254 3d7e9092 2003-10-14 devnull while(*q!='\0' && *q!='\n'){
255 3d7e9092 2003-10-14 devnull if(v >= attrbuf+sizeof attrbuf)
257 3d7e9092 2003-10-14 devnull c = *q++;
258 3d7e9092 2003-10-14 devnull if(quoting){
259 3d7e9092 2003-10-14 devnull if(c == '\''){
260 3d7e9092 2003-10-14 devnull if(*q == '\'')
263 3d7e9092 2003-10-14 devnull quoting = 0;
264 3d7e9092 2003-10-14 devnull continue;
268 3d7e9092 2003-10-14 devnull if(c==' ' || c=='\t')
270 3d7e9092 2003-10-14 devnull if(c == '\''){
271 3d7e9092 2003-10-14 devnull quoting = 1;
272 3d7e9092 2003-10-14 devnull continue;
275 3d7e9092 2003-10-14 devnull *v++ = c;
277 3d7e9092 2003-10-14 devnull a->value = malloc(v-attrbuf+1);
278 3d7e9092 2003-10-14 devnull if(a->value == nil){
279 3d7e9092 2003-10-14 devnull free(a->name);
280 3d7e9092 2003-10-14 devnull free(a);
283 3d7e9092 2003-10-14 devnull memmove(a->value, attrbuf, v-attrbuf);
284 3d7e9092 2003-10-14 devnull a->value[v-attrbuf] = '\0';
285 3d7e9092 2003-10-14 devnull a->next = nil;
286 3d7e9092 2003-10-14 devnull if(prev == nil)
287 3d7e9092 2003-10-14 devnull attr = a;
289 3d7e9092 2003-10-14 devnull prev->next = a;
290 3d7e9092 2003-10-14 devnull prev = a;
293 3d7e9092 2003-10-14 devnull return attr;
296 3d7e9092 2003-10-14 devnull Plumbattr*
297 3d7e9092 2003-10-14 devnull plumbaddattr(Plumbattr *attr, Plumbattr *new)
299 3d7e9092 2003-10-14 devnull Plumbattr *l;
301 3d7e9092 2003-10-14 devnull l = attr;
302 3d7e9092 2003-10-14 devnull if(l == nil)
303 3d7e9092 2003-10-14 devnull return new;
304 3d7e9092 2003-10-14 devnull while(l->next != nil)
305 3d7e9092 2003-10-14 devnull l = l->next;
306 3d7e9092 2003-10-14 devnull l->next = new;
307 3d7e9092 2003-10-14 devnull return attr;
310 3d7e9092 2003-10-14 devnull Plumbattr*
311 3d7e9092 2003-10-14 devnull plumbdelattr(Plumbattr *attr, char *name)
313 3d7e9092 2003-10-14 devnull Plumbattr *l, *prev;
315 3d7e9092 2003-10-14 devnull prev = nil;
316 3d7e9092 2003-10-14 devnull for(l=attr; l!=nil; l=l->next){
317 3d7e9092 2003-10-14 devnull if(strcmp(name, l->name) == 0)
319 3d7e9092 2003-10-14 devnull prev = l;
321 3d7e9092 2003-10-14 devnull if(l == nil)
322 3d7e9092 2003-10-14 devnull return nil;
323 3d7e9092 2003-10-14 devnull if(prev)
324 3d7e9092 2003-10-14 devnull prev->next = l->next;
326 3d7e9092 2003-10-14 devnull attr = l->next;
327 3d7e9092 2003-10-14 devnull free(l->name);
328 3d7e9092 2003-10-14 devnull free(l->value);
329 3d7e9092 2003-10-14 devnull free(l);
330 3d7e9092 2003-10-14 devnull return attr;
333 3d7e9092 2003-10-14 devnull Plumbmsg*
334 3d7e9092 2003-10-14 devnull plumbunpackpartial(char *buf, int n, int *morep)
336 3d7e9092 2003-10-14 devnull Plumbmsg *m;
337 3d7e9092 2003-10-14 devnull int i, bad;
338 3d7e9092 2003-10-14 devnull char *ntext, *attr;
340 3d7e9092 2003-10-14 devnull m = malloc(sizeof(Plumbmsg));
341 3d7e9092 2003-10-14 devnull if(m == nil)
342 3d7e9092 2003-10-14 devnull return nil;
343 3d7e9092 2003-10-14 devnull memset(m, 0, sizeof(Plumbmsg));
344 3d7e9092 2003-10-14 devnull if(morep != nil)
345 3d7e9092 2003-10-14 devnull *morep = 0;
346 3d7e9092 2003-10-14 devnull bad = 0;
347 3d7e9092 2003-10-14 devnull i = plumbline(&m->src, buf, 0, n, &bad);
348 3d7e9092 2003-10-14 devnull i = plumbline(&m->dst, buf, i, n, &bad);
349 3d7e9092 2003-10-14 devnull i = plumbline(&m->wdir, buf, i, n, &bad);
350 3d7e9092 2003-10-14 devnull i = plumbline(&m->type, buf, i, n, &bad);
351 3d7e9092 2003-10-14 devnull i = plumbline(&attr, buf, i, n, &bad);
352 3d7e9092 2003-10-14 devnull m->attr = plumbunpackattr(attr);
353 3d7e9092 2003-10-14 devnull free(attr);
354 3d7e9092 2003-10-14 devnull i = plumbline(&ntext, buf, i, n, &bad);
355 3d7e9092 2003-10-14 devnull m->ndata = atoi(ntext);
356 3d7e9092 2003-10-14 devnull if(m->ndata != n-i){
357 3d7e9092 2003-10-14 devnull bad = 1;
358 3d7e9092 2003-10-14 devnull if(morep!=nil && m->ndata>n-i)
359 3d7e9092 2003-10-14 devnull *morep = m->ndata - (n-i);
361 3d7e9092 2003-10-14 devnull free(ntext);
362 3d7e9092 2003-10-14 devnull if(!bad){
363 3d7e9092 2003-10-14 devnull m->data = malloc(n-i+1); /* +1 for '\0' */
364 3d7e9092 2003-10-14 devnull if(m->data == nil)
365 3d7e9092 2003-10-14 devnull bad = 1;
367 3d7e9092 2003-10-14 devnull memmove(m->data, buf+i, m->ndata);
368 3d7e9092 2003-10-14 devnull m->ndata = n-i;
369 3d7e9092 2003-10-14 devnull /* null-terminate in case it's text */
370 3d7e9092 2003-10-14 devnull m->data[m->ndata] = '\0';
373 3d7e9092 2003-10-14 devnull if(bad){
374 3d7e9092 2003-10-14 devnull plumbfree(m);
375 3d7e9092 2003-10-14 devnull m = nil;
377 3d7e9092 2003-10-14 devnull return m;
380 3d7e9092 2003-10-14 devnull Plumbmsg*
381 3d7e9092 2003-10-14 devnull plumbunpack(char *buf, int n)
383 3d7e9092 2003-10-14 devnull return plumbunpackpartial(buf, n, nil);
386 3d7e9092 2003-10-14 devnull Plumbmsg*
387 3d7e9092 2003-10-14 devnull plumbrecv(int fd)
389 3d7e9092 2003-10-14 devnull char *buf;
390 3d7e9092 2003-10-14 devnull Plumbmsg *m;
391 3d7e9092 2003-10-14 devnull int n, more;
393 3d7e9092 2003-10-14 devnull buf = malloc(8192);
394 3d7e9092 2003-10-14 devnull if(buf == nil)
395 3d7e9092 2003-10-14 devnull return nil;
396 3d7e9092 2003-10-14 devnull n = read(fd, buf, 8192);
397 3d7e9092 2003-10-14 devnull m = nil;
398 3d7e9092 2003-10-14 devnull if(n > 0){
399 3d7e9092 2003-10-14 devnull m = plumbunpackpartial(buf, n, &more);
400 3d7e9092 2003-10-14 devnull if(m==nil && more>0){
401 3d7e9092 2003-10-14 devnull /* we now know how many more bytes to read for complete message */
402 3d7e9092 2003-10-14 devnull buf = realloc(buf, n+more);
403 3d7e9092 2003-10-14 devnull if(buf == nil)
404 3d7e9092 2003-10-14 devnull return nil;
405 3d7e9092 2003-10-14 devnull if(readn(fd, buf+n, more) == more)
406 3d7e9092 2003-10-14 devnull m = plumbunpackpartial(buf, n+more, nil);
409 3d7e9092 2003-10-14 devnull free(buf);
410 3d7e9092 2003-10-14 devnull return m;