1 3d7e9092 2003-10-14 devnull #include <u.h>
2 3d7e9092 2003-10-14 devnull #include <libc.h>
3 32f69c36 2003-12-11 devnull #include <fcall.h>
4 3d7e9092 2003-10-14 devnull #include "plumb.h"
6 3d7e9092 2003-10-14 devnull static char attrbuf[4096];
8 b7e6f415 2003-11-23 devnull char *home;
10 3d7e9092 2003-10-14 devnull static int
11 3d7e9092 2003-10-14 devnull Strlen(char *s)
13 3d7e9092 2003-10-14 devnull if(s == nil)
14 3d7e9092 2003-10-14 devnull return 0;
15 3d7e9092 2003-10-14 devnull return strlen(s);
18 3d7e9092 2003-10-14 devnull static char*
19 3d7e9092 2003-10-14 devnull Strcpy(char *s, char *t)
21 3d7e9092 2003-10-14 devnull if(t == nil)
22 3d7e9092 2003-10-14 devnull return s;
23 3d7e9092 2003-10-14 devnull return strcpy(s, t) + strlen(t);
26 3d7e9092 2003-10-14 devnull /* quote attribute value, if necessary */
27 3d7e9092 2003-10-14 devnull static char*
28 3d7e9092 2003-10-14 devnull quote(char *s)
33 3d7e9092 2003-10-14 devnull if(s == nil){
34 3d7e9092 2003-10-14 devnull attrbuf[0] = '\0';
35 3d7e9092 2003-10-14 devnull return attrbuf;
37 3d7e9092 2003-10-14 devnull if(strpbrk(s, " '=\t") == nil)
38 3d7e9092 2003-10-14 devnull return s;
39 3d7e9092 2003-10-14 devnull t = attrbuf;
40 3d7e9092 2003-10-14 devnull *t++ = '\'';
41 3d7e9092 2003-10-14 devnull while(t < attrbuf+sizeof attrbuf-2){
42 3d7e9092 2003-10-14 devnull c = *s++;
43 3d7e9092 2003-10-14 devnull if(c == '\0')
45 3d7e9092 2003-10-14 devnull *t++ = c;
46 3d7e9092 2003-10-14 devnull if(c == '\'')
47 3d7e9092 2003-10-14 devnull *t++ = c;
49 3d7e9092 2003-10-14 devnull *t++ = '\'';
50 3d7e9092 2003-10-14 devnull *t = '\0';
51 3d7e9092 2003-10-14 devnull return attrbuf;
55 3d7e9092 2003-10-14 devnull plumbpackattr(Plumbattr *attr)
58 3d7e9092 2003-10-14 devnull Plumbattr *a;
59 3d7e9092 2003-10-14 devnull char *s, *t;
61 3d7e9092 2003-10-14 devnull if(attr == nil)
62 3d7e9092 2003-10-14 devnull return nil;
64 3d7e9092 2003-10-14 devnull for(a=attr; a!=nil; a=a->next)
65 3d7e9092 2003-10-14 devnull n += Strlen(a->name) + 1 + Strlen(quote(a->value)) + 1;
66 3d7e9092 2003-10-14 devnull s = malloc(n);
67 3d7e9092 2003-10-14 devnull if(s == nil)
68 3d7e9092 2003-10-14 devnull return nil;
70 3d7e9092 2003-10-14 devnull *t = '\0';
71 3d7e9092 2003-10-14 devnull for(a=attr; a!=nil; a=a->next){
72 3d7e9092 2003-10-14 devnull if(t != s)
73 3d7e9092 2003-10-14 devnull *t++ = ' ';
74 3d7e9092 2003-10-14 devnull strcpy(t, a->name);
75 3d7e9092 2003-10-14 devnull strcat(t, "=");
76 3d7e9092 2003-10-14 devnull strcat(t, quote(a->value));
77 3d7e9092 2003-10-14 devnull t += strlen(t);
79 3d7e9092 2003-10-14 devnull if(t > s+n)
81 3d7e9092 2003-10-14 devnull return s;
85 3d7e9092 2003-10-14 devnull plumblookup(Plumbattr *attr, char *name)
87 3d7e9092 2003-10-14 devnull while(attr){
88 3d7e9092 2003-10-14 devnull if(strcmp(attr->name, name) == 0)
89 3d7e9092 2003-10-14 devnull return attr->value;
90 3d7e9092 2003-10-14 devnull attr = attr->next;
92 3d7e9092 2003-10-14 devnull return nil;
96 3d7e9092 2003-10-14 devnull plumbpack(Plumbmsg *m, int *np)
98 3d7e9092 2003-10-14 devnull int n, ndata;
99 3d7e9092 2003-10-14 devnull char *buf, *p, *attr;
101 3d7e9092 2003-10-14 devnull ndata = m->ndata;
102 3d7e9092 2003-10-14 devnull if(ndata < 0)
103 3d7e9092 2003-10-14 devnull ndata = Strlen(m->data);
104 3d7e9092 2003-10-14 devnull attr = plumbpackattr(m->attr);
105 3d7e9092 2003-10-14 devnull n = Strlen(m->src)+1 + Strlen(m->dst)+1 + Strlen(m->wdir)+1 +
106 3d7e9092 2003-10-14 devnull Strlen(m->type)+1 + Strlen(attr)+1 + 16 + ndata;
107 3d7e9092 2003-10-14 devnull buf = malloc(n+1); /* +1 for '\0' */
108 3d7e9092 2003-10-14 devnull if(buf == nil){
109 3d7e9092 2003-10-14 devnull free(attr);
110 3d7e9092 2003-10-14 devnull return nil;
112 3d7e9092 2003-10-14 devnull p = Strcpy(buf, m->src);
113 3d7e9092 2003-10-14 devnull *p++ = '\n';
114 3d7e9092 2003-10-14 devnull p = Strcpy(p, m->dst);
115 3d7e9092 2003-10-14 devnull *p++ = '\n';
116 3d7e9092 2003-10-14 devnull p = Strcpy(p, m->wdir);
117 3d7e9092 2003-10-14 devnull *p++ = '\n';
118 3d7e9092 2003-10-14 devnull p = Strcpy(p, m->type);
119 3d7e9092 2003-10-14 devnull *p++ = '\n';
120 3d7e9092 2003-10-14 devnull p = Strcpy(p, attr);
121 3d7e9092 2003-10-14 devnull *p++ = '\n';
122 3d7e9092 2003-10-14 devnull p += sprint(p, "%d\n", ndata);
123 3d7e9092 2003-10-14 devnull memmove(p, m->data, ndata);
124 3d7e9092 2003-10-14 devnull *np = (p-buf)+ndata;
125 3d7e9092 2003-10-14 devnull buf[*np] = '\0'; /* null terminate just in case */
126 3d7e9092 2003-10-14 devnull if(*np >= n+1)
127 3d7e9092 2003-10-14 devnull abort();
128 3d7e9092 2003-10-14 devnull free(attr);
129 3d7e9092 2003-10-14 devnull return buf;
132 3d7e9092 2003-10-14 devnull static int
133 3d7e9092 2003-10-14 devnull plumbline(char **linep, char *buf, int i, int n, int *bad)
135 3d7e9092 2003-10-14 devnull int starti;
136 3d7e9092 2003-10-14 devnull char *p;
138 3d7e9092 2003-10-14 devnull if(*bad)
139 3d7e9092 2003-10-14 devnull return i;
140 3d7e9092 2003-10-14 devnull starti = i;
141 3d7e9092 2003-10-14 devnull while(i<n && buf[i]!='\n')
143 3d7e9092 2003-10-14 devnull if(i == n)
144 3d7e9092 2003-10-14 devnull *bad = 1;
146 3d7e9092 2003-10-14 devnull p = malloc((i-starti) + 1);
147 3d7e9092 2003-10-14 devnull if(p == nil)
148 3d7e9092 2003-10-14 devnull *bad = 1;
150 3d7e9092 2003-10-14 devnull memmove(p, buf+starti, i-starti);
151 3d7e9092 2003-10-14 devnull p[i-starti] = '\0';
153 3d7e9092 2003-10-14 devnull *linep = p;
156 3d7e9092 2003-10-14 devnull return i;
160 3d7e9092 2003-10-14 devnull plumbfree(Plumbmsg *m)
162 3d7e9092 2003-10-14 devnull Plumbattr *a, *next;
164 3d7e9092 2003-10-14 devnull free(m->src);
165 3d7e9092 2003-10-14 devnull free(m->dst);
166 3d7e9092 2003-10-14 devnull free(m->wdir);
167 3d7e9092 2003-10-14 devnull free(m->type);
168 3d7e9092 2003-10-14 devnull for(a=m->attr; a!=nil; a=next){
169 3d7e9092 2003-10-14 devnull next = a->next;
170 3d7e9092 2003-10-14 devnull free(a->name);
171 3d7e9092 2003-10-14 devnull free(a->value);
172 3d7e9092 2003-10-14 devnull free(a);
174 3d7e9092 2003-10-14 devnull free(m->data);
175 3d7e9092 2003-10-14 devnull free(m);
178 3d7e9092 2003-10-14 devnull Plumbattr*
179 3d7e9092 2003-10-14 devnull plumbunpackattr(char *p)
181 3d7e9092 2003-10-14 devnull Plumbattr *attr, *prev, *a;
182 3d7e9092 2003-10-14 devnull char *q, *v;
183 3d7e9092 2003-10-14 devnull int c, quoting;
185 3d7e9092 2003-10-14 devnull attr = prev = nil;
186 3d7e9092 2003-10-14 devnull while(*p!='\0' && *p!='\n'){
187 3d7e9092 2003-10-14 devnull while(*p==' ' || *p=='\t')
189 3d7e9092 2003-10-14 devnull if(*p == '\0')
191 3d7e9092 2003-10-14 devnull for(q=p; *q!='\0' && *q!='\n' && *q!=' ' && *q!='\t'; q++)
192 3d7e9092 2003-10-14 devnull if(*q == '=')
194 3d7e9092 2003-10-14 devnull if(*q != '=')
195 3d7e9092 2003-10-14 devnull break; /* malformed attribute */
196 3d7e9092 2003-10-14 devnull a = malloc(sizeof(Plumbattr));
197 3d7e9092 2003-10-14 devnull if(a == nil)
199 3d7e9092 2003-10-14 devnull a->name = malloc(q-p+1);
200 3d7e9092 2003-10-14 devnull if(a->name == nil){
201 3d7e9092 2003-10-14 devnull free(a);
204 3d7e9092 2003-10-14 devnull memmove(a->name, p, q-p);
205 3d7e9092 2003-10-14 devnull a->name[q-p] = '\0';
206 3d7e9092 2003-10-14 devnull /* process quotes in value */
207 3d7e9092 2003-10-14 devnull q++; /* skip '=' */
208 3d7e9092 2003-10-14 devnull v = attrbuf;
209 3d7e9092 2003-10-14 devnull quoting = 0;
210 3d7e9092 2003-10-14 devnull while(*q!='\0' && *q!='\n'){
211 3d7e9092 2003-10-14 devnull if(v >= attrbuf+sizeof attrbuf)
213 3d7e9092 2003-10-14 devnull c = *q++;
214 3d7e9092 2003-10-14 devnull if(quoting){
215 3d7e9092 2003-10-14 devnull if(c == '\''){
216 3d7e9092 2003-10-14 devnull if(*q == '\'')
219 3d7e9092 2003-10-14 devnull quoting = 0;
220 3d7e9092 2003-10-14 devnull continue;
224 3d7e9092 2003-10-14 devnull if(c==' ' || c=='\t')
226 3d7e9092 2003-10-14 devnull if(c == '\''){
227 3d7e9092 2003-10-14 devnull quoting = 1;
228 3d7e9092 2003-10-14 devnull continue;
231 3d7e9092 2003-10-14 devnull *v++ = c;
233 3d7e9092 2003-10-14 devnull a->value = malloc(v-attrbuf+1);
234 3d7e9092 2003-10-14 devnull if(a->value == nil){
235 3d7e9092 2003-10-14 devnull free(a->name);
236 3d7e9092 2003-10-14 devnull free(a);
239 3d7e9092 2003-10-14 devnull memmove(a->value, attrbuf, v-attrbuf);
240 3d7e9092 2003-10-14 devnull a->value[v-attrbuf] = '\0';
241 3d7e9092 2003-10-14 devnull a->next = nil;
242 3d7e9092 2003-10-14 devnull if(prev == nil)
243 3d7e9092 2003-10-14 devnull attr = a;
245 3d7e9092 2003-10-14 devnull prev->next = a;
246 3d7e9092 2003-10-14 devnull prev = a;
249 3d7e9092 2003-10-14 devnull return attr;
252 3d7e9092 2003-10-14 devnull Plumbattr*
253 3d7e9092 2003-10-14 devnull plumbaddattr(Plumbattr *attr, Plumbattr *new)
255 3d7e9092 2003-10-14 devnull Plumbattr *l;
257 3d7e9092 2003-10-14 devnull l = attr;
258 3d7e9092 2003-10-14 devnull if(l == nil)
259 3d7e9092 2003-10-14 devnull return new;
260 3d7e9092 2003-10-14 devnull while(l->next != nil)
261 3d7e9092 2003-10-14 devnull l = l->next;
262 3d7e9092 2003-10-14 devnull l->next = new;
263 3d7e9092 2003-10-14 devnull return attr;
266 3d7e9092 2003-10-14 devnull Plumbattr*
267 3d7e9092 2003-10-14 devnull plumbdelattr(Plumbattr *attr, char *name)
269 3d7e9092 2003-10-14 devnull Plumbattr *l, *prev;
271 3d7e9092 2003-10-14 devnull prev = nil;
272 3d7e9092 2003-10-14 devnull for(l=attr; l!=nil; l=l->next){
273 3d7e9092 2003-10-14 devnull if(strcmp(name, l->name) == 0)
275 3d7e9092 2003-10-14 devnull prev = l;
277 3d7e9092 2003-10-14 devnull if(l == nil)
278 3d7e9092 2003-10-14 devnull return nil;
279 3d7e9092 2003-10-14 devnull if(prev)
280 3d7e9092 2003-10-14 devnull prev->next = l->next;
282 3d7e9092 2003-10-14 devnull attr = l->next;
283 3d7e9092 2003-10-14 devnull free(l->name);
284 3d7e9092 2003-10-14 devnull free(l->value);
285 3d7e9092 2003-10-14 devnull free(l);
286 3d7e9092 2003-10-14 devnull return attr;
289 3d7e9092 2003-10-14 devnull Plumbmsg*
290 3d7e9092 2003-10-14 devnull plumbunpackpartial(char *buf, int n, int *morep)
292 3d7e9092 2003-10-14 devnull Plumbmsg *m;
293 3d7e9092 2003-10-14 devnull int i, bad;
294 3d7e9092 2003-10-14 devnull char *ntext, *attr;
296 3d7e9092 2003-10-14 devnull m = malloc(sizeof(Plumbmsg));
297 3d7e9092 2003-10-14 devnull if(m == nil)
298 3d7e9092 2003-10-14 devnull return nil;
299 3d7e9092 2003-10-14 devnull memset(m, 0, sizeof(Plumbmsg));
300 3d7e9092 2003-10-14 devnull if(morep != nil)
301 3d7e9092 2003-10-14 devnull *morep = 0;
302 3d7e9092 2003-10-14 devnull bad = 0;
303 3d7e9092 2003-10-14 devnull i = plumbline(&m->src, buf, 0, n, &bad);
304 3d7e9092 2003-10-14 devnull i = plumbline(&m->dst, buf, i, n, &bad);
305 3d7e9092 2003-10-14 devnull i = plumbline(&m->wdir, buf, i, n, &bad);
306 3d7e9092 2003-10-14 devnull i = plumbline(&m->type, buf, i, n, &bad);
307 3d7e9092 2003-10-14 devnull i = plumbline(&attr, buf, i, n, &bad);
308 3d7e9092 2003-10-14 devnull m->attr = plumbunpackattr(attr);
309 3d7e9092 2003-10-14 devnull free(attr);
310 3d7e9092 2003-10-14 devnull i = plumbline(&ntext, buf, i, n, &bad);
311 3d7e9092 2003-10-14 devnull m->ndata = atoi(ntext);
312 3d7e9092 2003-10-14 devnull if(m->ndata != n-i){
313 3d7e9092 2003-10-14 devnull bad = 1;
314 3d7e9092 2003-10-14 devnull if(morep!=nil && m->ndata>n-i)
315 3d7e9092 2003-10-14 devnull *morep = m->ndata - (n-i);
317 3d7e9092 2003-10-14 devnull free(ntext);
318 3d7e9092 2003-10-14 devnull if(!bad){
319 3d7e9092 2003-10-14 devnull m->data = malloc(n-i+1); /* +1 for '\0' */
320 3d7e9092 2003-10-14 devnull if(m->data == nil)
321 3d7e9092 2003-10-14 devnull bad = 1;
323 3d7e9092 2003-10-14 devnull memmove(m->data, buf+i, m->ndata);
324 3d7e9092 2003-10-14 devnull m->ndata = n-i;
325 3d7e9092 2003-10-14 devnull /* null-terminate in case it's text */
326 3d7e9092 2003-10-14 devnull m->data[m->ndata] = '\0';
329 3d7e9092 2003-10-14 devnull if(bad){
330 3d7e9092 2003-10-14 devnull plumbfree(m);
331 3d7e9092 2003-10-14 devnull m = nil;
333 3d7e9092 2003-10-14 devnull return m;
336 3d7e9092 2003-10-14 devnull Plumbmsg*
337 3d7e9092 2003-10-14 devnull plumbunpack(char *buf, int n)
339 3d7e9092 2003-10-14 devnull return plumbunpackpartial(buf, n, nil);