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 32f69c36 2003-12-11 devnull #include <fs.h>
  5 3d7e9092 2003-10-14 devnull #include "plumb.h"
  6 3d7e9092 2003-10-14 devnull 
  7 3d7e9092 2003-10-14 devnull static char attrbuf[4096];
  8 3d7e9092 2003-10-14 devnull 
  9 b7e6f415 2003-11-23 devnull char *home;
 10 49588d5d 2003-12-17 devnull static Fsys *fsplumb;
 11 49588d5d 2003-12-17 devnull static int pfd = -1;
 12 49588d5d 2003-12-17 devnull static Fid *pfid;
 13 3d7e9092 2003-10-14 devnull int
 14 3d7e9092 2003-10-14 devnull plumbopen(char *name, int omode)
 15 3d7e9092 2003-10-14 devnull {
 16 32f69c36 2003-12-11 devnull 	int fd;
 17 3d7e9092 2003-10-14 devnull 
 18 49588d5d 2003-12-17 devnull 	if(fsplumb == nil)
 19 49588d5d 2003-12-17 devnull 		fsplumb = nsmount("plumb", "");
 20 49588d5d 2003-12-17 devnull 	if(fsplumb == nil)
 21 3d7e9092 2003-10-14 devnull 		return -1;
 22 49588d5d 2003-12-17 devnull 	/*
 23 49588d5d 2003-12-17 devnull 	 * It's important that when we send something,
 24 49588d5d 2003-12-17 devnull 	 * we find out whether it was a valid plumb write.
 25 49588d5d 2003-12-17 devnull 	 * (If it isn't, the client might fall back to some
 26 49588d5d 2003-12-17 devnull 	 * other mechanism or indicate to the user what happened.)
 27 49588d5d 2003-12-17 devnull 	 * We can't use a pipe for this, so we have to use the
 28 49588d5d 2003-12-17 devnull 	 * fid interface.  But we need to return a fd. 
 29 49588d5d 2003-12-17 devnull 	 * Return a fd for /dev/null so that we return a unique
 30 49588d5d 2003-12-17 devnull 	 * file descriptor.  In plumbsend we'll look for pfd
 31 49588d5d 2003-12-17 devnull 	 * and use the recorded fid instead.
 32 49588d5d 2003-12-17 devnull 	 */
 33 49588d5d 2003-12-17 devnull 	if((omode&3) == OWRITE){
 34 49588d5d 2003-12-17 devnull 		if(pfd != -1){
 35 49588d5d 2003-12-17 devnull 			werrstr("already have plumb send open");
 36 49588d5d 2003-12-17 devnull 			return -1;
 37 49588d5d 2003-12-17 devnull 		}
 38 49588d5d 2003-12-17 devnull 		pfd = open("/dev/null", OWRITE);
 39 49588d5d 2003-12-17 devnull 		if(pfd < 0)
 40 49588d5d 2003-12-17 devnull 			return -1;
 41 49588d5d 2003-12-17 devnull 		pfid = fsopen(fsplumb, name, omode);
 42 49588d5d 2003-12-17 devnull 		if(pfid == nil){
 43 49588d5d 2003-12-17 devnull 			close(pfd);
 44 49588d5d 2003-12-17 devnull 			pfd = -1;
 45 49588d5d 2003-12-17 devnull 			return -1;
 46 49588d5d 2003-12-17 devnull 		}
 47 49588d5d 2003-12-17 devnull 		return pfd;
 48 49588d5d 2003-12-17 devnull 	}
 49 49588d5d 2003-12-17 devnull 
 50 49588d5d 2003-12-17 devnull 	fd = fsopenfd(fsplumb, name, omode);
 51 32f69c36 2003-12-11 devnull 	return fd;
 52 49588d5d 2003-12-17 devnull }
 53 49588d5d 2003-12-17 devnull 
 54 49588d5d 2003-12-17 devnull int
 55 49588d5d 2003-12-17 devnull plumbsend(int fd, Plumbmsg *m)
 56 49588d5d 2003-12-17 devnull {
 57 49588d5d 2003-12-17 devnull 	char *buf;
 58 49588d5d 2003-12-17 devnull 	int n;
 59 49588d5d 2003-12-17 devnull 
 60 49588d5d 2003-12-17 devnull 	if(fd != pfd){
 61 49588d5d 2003-12-17 devnull 		werrstr("fd is not the plumber");
 62 49588d5d 2003-12-17 devnull 		return -1;
 63 49588d5d 2003-12-17 devnull 	}
 64 49588d5d 2003-12-17 devnull 	buf = plumbpack(m, &n);
 65 49588d5d 2003-12-17 devnull 	if(buf == nil)
 66 49588d5d 2003-12-17 devnull 		return -1;
 67 49588d5d 2003-12-17 devnull 	n = fswrite(pfid, buf, n);
 68 49588d5d 2003-12-17 devnull 	free(buf);
 69 49588d5d 2003-12-17 devnull 	return n;
 70 3d7e9092 2003-10-14 devnull }
 71 3d7e9092 2003-10-14 devnull 
 72 3d7e9092 2003-10-14 devnull static int
 73 3d7e9092 2003-10-14 devnull Strlen(char *s)
 74 3d7e9092 2003-10-14 devnull {
 75 3d7e9092 2003-10-14 devnull 	if(s == nil)
 76 3d7e9092 2003-10-14 devnull 		return 0;
 77 3d7e9092 2003-10-14 devnull 	return strlen(s);
 78 3d7e9092 2003-10-14 devnull }
 79 3d7e9092 2003-10-14 devnull 
 80 3d7e9092 2003-10-14 devnull static char*
 81 3d7e9092 2003-10-14 devnull Strcpy(char *s, char *t)
 82 3d7e9092 2003-10-14 devnull {
 83 3d7e9092 2003-10-14 devnull 	if(t == nil)
 84 3d7e9092 2003-10-14 devnull 		return s;
 85 3d7e9092 2003-10-14 devnull 	return strcpy(s, t) + strlen(t);
 86 3d7e9092 2003-10-14 devnull }
 87 3d7e9092 2003-10-14 devnull 
 88 3d7e9092 2003-10-14 devnull /* quote attribute value, if necessary */
 89 3d7e9092 2003-10-14 devnull static char*
 90 3d7e9092 2003-10-14 devnull quote(char *s)
 91 3d7e9092 2003-10-14 devnull {
 92 3d7e9092 2003-10-14 devnull 	char *t;
 93 3d7e9092 2003-10-14 devnull 	int c;
 94 3d7e9092 2003-10-14 devnull 
 95 3d7e9092 2003-10-14 devnull 	if(s == nil){
 96 3d7e9092 2003-10-14 devnull 		attrbuf[0] = '\0';
 97 3d7e9092 2003-10-14 devnull 		return attrbuf;
 98 3d7e9092 2003-10-14 devnull 	}
 99 3d7e9092 2003-10-14 devnull 	if(strpbrk(s, " '=\t") == nil)
100 3d7e9092 2003-10-14 devnull 		return s;
101 3d7e9092 2003-10-14 devnull 	t = attrbuf;
102 3d7e9092 2003-10-14 devnull 	*t++ = '\'';
103 3d7e9092 2003-10-14 devnull 	while(t < attrbuf+sizeof attrbuf-2){
104 3d7e9092 2003-10-14 devnull 		c = *s++;
105 3d7e9092 2003-10-14 devnull 		if(c == '\0')
106 3d7e9092 2003-10-14 devnull 			break;
107 3d7e9092 2003-10-14 devnull 		*t++ = c;
108 3d7e9092 2003-10-14 devnull 		if(c == '\'')
109 3d7e9092 2003-10-14 devnull 			*t++ = c;
110 3d7e9092 2003-10-14 devnull 	}
111 3d7e9092 2003-10-14 devnull 	*t++ = '\'';
112 3d7e9092 2003-10-14 devnull 	*t = '\0';
113 3d7e9092 2003-10-14 devnull 	return attrbuf;
114 3d7e9092 2003-10-14 devnull }
115 3d7e9092 2003-10-14 devnull 
116 3d7e9092 2003-10-14 devnull char*
117 3d7e9092 2003-10-14 devnull plumbpackattr(Plumbattr *attr)
118 3d7e9092 2003-10-14 devnull {
119 3d7e9092 2003-10-14 devnull 	int n;
120 3d7e9092 2003-10-14 devnull 	Plumbattr *a;
121 3d7e9092 2003-10-14 devnull 	char *s, *t;
122 3d7e9092 2003-10-14 devnull 
123 3d7e9092 2003-10-14 devnull 	if(attr == nil)
124 3d7e9092 2003-10-14 devnull 		return nil;
125 3d7e9092 2003-10-14 devnull 	n = 0;
126 3d7e9092 2003-10-14 devnull 	for(a=attr; a!=nil; a=a->next)
127 3d7e9092 2003-10-14 devnull 		n += Strlen(a->name) + 1 + Strlen(quote(a->value)) + 1;
128 3d7e9092 2003-10-14 devnull 	s = malloc(n);
129 3d7e9092 2003-10-14 devnull 	if(s == nil)
130 3d7e9092 2003-10-14 devnull 		return nil;
131 3d7e9092 2003-10-14 devnull 	t = s;
132 3d7e9092 2003-10-14 devnull 	*t = '\0';
133 3d7e9092 2003-10-14 devnull 	for(a=attr; a!=nil; a=a->next){
134 3d7e9092 2003-10-14 devnull 		if(t != s)
135 3d7e9092 2003-10-14 devnull 			*t++ = ' ';
136 3d7e9092 2003-10-14 devnull 		strcpy(t, a->name);
137 3d7e9092 2003-10-14 devnull 		strcat(t, "=");
138 3d7e9092 2003-10-14 devnull 		strcat(t, quote(a->value));
139 3d7e9092 2003-10-14 devnull 		t += strlen(t);
140 3d7e9092 2003-10-14 devnull 	}
141 3d7e9092 2003-10-14 devnull 	if(t > s+n)
142 3d7e9092 2003-10-14 devnull 		abort();
143 3d7e9092 2003-10-14 devnull 	return s;
144 3d7e9092 2003-10-14 devnull }
145 3d7e9092 2003-10-14 devnull 
146 3d7e9092 2003-10-14 devnull char*
147 3d7e9092 2003-10-14 devnull plumblookup(Plumbattr *attr, char *name)
148 3d7e9092 2003-10-14 devnull {
149 3d7e9092 2003-10-14 devnull 	while(attr){
150 3d7e9092 2003-10-14 devnull 		if(strcmp(attr->name, name) == 0)
151 3d7e9092 2003-10-14 devnull 			return attr->value;
152 3d7e9092 2003-10-14 devnull 		attr = attr->next;
153 3d7e9092 2003-10-14 devnull 	}
154 3d7e9092 2003-10-14 devnull 	return nil;
155 3d7e9092 2003-10-14 devnull }
156 3d7e9092 2003-10-14 devnull 
157 3d7e9092 2003-10-14 devnull char*
158 3d7e9092 2003-10-14 devnull plumbpack(Plumbmsg *m, int *np)
159 3d7e9092 2003-10-14 devnull {
160 3d7e9092 2003-10-14 devnull 	int n, ndata;
161 3d7e9092 2003-10-14 devnull 	char *buf, *p, *attr;
162 3d7e9092 2003-10-14 devnull 
163 3d7e9092 2003-10-14 devnull 	ndata = m->ndata;
164 3d7e9092 2003-10-14 devnull 	if(ndata < 0)
165 3d7e9092 2003-10-14 devnull 		ndata = Strlen(m->data);
166 3d7e9092 2003-10-14 devnull 	attr = plumbpackattr(m->attr);
167 3d7e9092 2003-10-14 devnull 	n = Strlen(m->src)+1 + Strlen(m->dst)+1 + Strlen(m->wdir)+1 +
168 3d7e9092 2003-10-14 devnull 		Strlen(m->type)+1 + Strlen(attr)+1 + 16 + ndata;
169 3d7e9092 2003-10-14 devnull 	buf = malloc(n+1);	/* +1 for '\0' */
170 3d7e9092 2003-10-14 devnull 	if(buf == nil){
171 3d7e9092 2003-10-14 devnull 		free(attr);
172 3d7e9092 2003-10-14 devnull 		return nil;
173 3d7e9092 2003-10-14 devnull 	}
174 3d7e9092 2003-10-14 devnull 	p = Strcpy(buf, m->src);
175 3d7e9092 2003-10-14 devnull 	*p++ = '\n';
176 3d7e9092 2003-10-14 devnull 	p = Strcpy(p, m->dst);
177 3d7e9092 2003-10-14 devnull 	*p++ = '\n';
178 3d7e9092 2003-10-14 devnull 	p = Strcpy(p, m->wdir);
179 3d7e9092 2003-10-14 devnull 	*p++ = '\n';
180 3d7e9092 2003-10-14 devnull 	p = Strcpy(p, m->type);
181 3d7e9092 2003-10-14 devnull 	*p++ = '\n';
182 3d7e9092 2003-10-14 devnull 	p = Strcpy(p, attr);
183 3d7e9092 2003-10-14 devnull 	*p++ = '\n';
184 3d7e9092 2003-10-14 devnull 	p += sprint(p, "%d\n", ndata);
185 3d7e9092 2003-10-14 devnull 	memmove(p, m->data, ndata);
186 3d7e9092 2003-10-14 devnull 	*np = (p-buf)+ndata;
187 3d7e9092 2003-10-14 devnull 	buf[*np] = '\0';	/* null terminate just in case */
188 3d7e9092 2003-10-14 devnull 	if(*np >= n+1)
189 3d7e9092 2003-10-14 devnull 		abort();
190 3d7e9092 2003-10-14 devnull 	free(attr);
191 3d7e9092 2003-10-14 devnull 	return buf;
192 3d7e9092 2003-10-14 devnull }
193 3d7e9092 2003-10-14 devnull 
194 3d7e9092 2003-10-14 devnull static int
195 3d7e9092 2003-10-14 devnull plumbline(char **linep, char *buf, int i, int n, int *bad)
196 3d7e9092 2003-10-14 devnull {
197 3d7e9092 2003-10-14 devnull 	int starti;
198 3d7e9092 2003-10-14 devnull 	char *p;
199 3d7e9092 2003-10-14 devnull 
200 3d7e9092 2003-10-14 devnull 	if(*bad)
201 3d7e9092 2003-10-14 devnull 		return i;
202 3d7e9092 2003-10-14 devnull 	starti = i;
203 3d7e9092 2003-10-14 devnull 	while(i<n && buf[i]!='\n')
204 3d7e9092 2003-10-14 devnull 		i++;
205 3d7e9092 2003-10-14 devnull 	if(i == n)
206 3d7e9092 2003-10-14 devnull 		*bad = 1;
207 3d7e9092 2003-10-14 devnull 	else{
208 3d7e9092 2003-10-14 devnull 		p = malloc((i-starti) + 1);
209 3d7e9092 2003-10-14 devnull 		if(p == nil)
210 3d7e9092 2003-10-14 devnull 			*bad = 1;
211 3d7e9092 2003-10-14 devnull 		else{
212 3d7e9092 2003-10-14 devnull 			memmove(p, buf+starti, i-starti);
213 3d7e9092 2003-10-14 devnull 			p[i-starti] = '\0';
214 3d7e9092 2003-10-14 devnull 		}
215 3d7e9092 2003-10-14 devnull 		*linep = p;
216 3d7e9092 2003-10-14 devnull 		i++;
217 3d7e9092 2003-10-14 devnull 	}
218 3d7e9092 2003-10-14 devnull 	return i;
219 3d7e9092 2003-10-14 devnull }
220 3d7e9092 2003-10-14 devnull 
221 3d7e9092 2003-10-14 devnull void
222 3d7e9092 2003-10-14 devnull plumbfree(Plumbmsg *m)
223 3d7e9092 2003-10-14 devnull {
224 3d7e9092 2003-10-14 devnull 	Plumbattr *a, *next;
225 3d7e9092 2003-10-14 devnull 
226 3d7e9092 2003-10-14 devnull 	free(m->src);
227 3d7e9092 2003-10-14 devnull 	free(m->dst);
228 3d7e9092 2003-10-14 devnull 	free(m->wdir);
229 3d7e9092 2003-10-14 devnull 	free(m->type);
230 3d7e9092 2003-10-14 devnull 	for(a=m->attr; a!=nil; a=next){
231 3d7e9092 2003-10-14 devnull 		next = a->next;
232 3d7e9092 2003-10-14 devnull 		free(a->name);
233 3d7e9092 2003-10-14 devnull 		free(a->value);
234 3d7e9092 2003-10-14 devnull 		free(a);
235 3d7e9092 2003-10-14 devnull 	}
236 3d7e9092 2003-10-14 devnull 	free(m->data);
237 3d7e9092 2003-10-14 devnull 	free(m);
238 3d7e9092 2003-10-14 devnull }
239 3d7e9092 2003-10-14 devnull 
240 3d7e9092 2003-10-14 devnull Plumbattr*
241 3d7e9092 2003-10-14 devnull plumbunpackattr(char *p)
242 3d7e9092 2003-10-14 devnull {
243 3d7e9092 2003-10-14 devnull 	Plumbattr *attr, *prev, *a;
244 3d7e9092 2003-10-14 devnull 	char *q, *v;
245 3d7e9092 2003-10-14 devnull 	int c, quoting;
246 3d7e9092 2003-10-14 devnull 
247 3d7e9092 2003-10-14 devnull 	attr = prev = nil;
248 3d7e9092 2003-10-14 devnull 	while(*p!='\0' && *p!='\n'){
249 3d7e9092 2003-10-14 devnull 		while(*p==' ' || *p=='\t')
250 3d7e9092 2003-10-14 devnull 			p++;
251 3d7e9092 2003-10-14 devnull 		if(*p == '\0')
252 3d7e9092 2003-10-14 devnull 			break;
253 3d7e9092 2003-10-14 devnull 		for(q=p; *q!='\0' && *q!='\n' && *q!=' ' && *q!='\t'; q++)
254 3d7e9092 2003-10-14 devnull 			if(*q == '=')
255 3d7e9092 2003-10-14 devnull 				break;
256 3d7e9092 2003-10-14 devnull 		if(*q != '=')
257 3d7e9092 2003-10-14 devnull 			break;	/* malformed attribute */
258 3d7e9092 2003-10-14 devnull 		a = malloc(sizeof(Plumbattr));
259 3d7e9092 2003-10-14 devnull 		if(a == nil)
260 3d7e9092 2003-10-14 devnull 			break;
261 3d7e9092 2003-10-14 devnull 		a->name = malloc(q-p+1);
262 3d7e9092 2003-10-14 devnull 		if(a->name == nil){
263 3d7e9092 2003-10-14 devnull 			free(a);
264 3d7e9092 2003-10-14 devnull 			break;
265 3d7e9092 2003-10-14 devnull 		}
266 3d7e9092 2003-10-14 devnull 		memmove(a->name, p, q-p);
267 3d7e9092 2003-10-14 devnull 		a->name[q-p] = '\0';
268 3d7e9092 2003-10-14 devnull 		/* process quotes in value */
269 3d7e9092 2003-10-14 devnull 		q++;	/* skip '=' */
270 3d7e9092 2003-10-14 devnull 		v = attrbuf;
271 3d7e9092 2003-10-14 devnull 		quoting = 0;
272 3d7e9092 2003-10-14 devnull 		while(*q!='\0' && *q!='\n'){
273 3d7e9092 2003-10-14 devnull 			if(v >= attrbuf+sizeof attrbuf)
274 3d7e9092 2003-10-14 devnull 				break;
275 3d7e9092 2003-10-14 devnull 			c = *q++;
276 3d7e9092 2003-10-14 devnull 			if(quoting){
277 3d7e9092 2003-10-14 devnull 				if(c == '\''){
278 3d7e9092 2003-10-14 devnull 					if(*q == '\'')
279 3d7e9092 2003-10-14 devnull 						q++;
280 3d7e9092 2003-10-14 devnull 					else{
281 3d7e9092 2003-10-14 devnull 						quoting = 0;
282 3d7e9092 2003-10-14 devnull 						continue;
283 3d7e9092 2003-10-14 devnull 					}
284 3d7e9092 2003-10-14 devnull 				}
285 3d7e9092 2003-10-14 devnull 			}else{
286 3d7e9092 2003-10-14 devnull 				if(c==' ' || c=='\t')
287 3d7e9092 2003-10-14 devnull 					break;
288 3d7e9092 2003-10-14 devnull 				if(c == '\''){
289 3d7e9092 2003-10-14 devnull 					quoting = 1;
290 3d7e9092 2003-10-14 devnull 					continue;
291 3d7e9092 2003-10-14 devnull 				}
292 3d7e9092 2003-10-14 devnull 			}
293 3d7e9092 2003-10-14 devnull 			*v++ = c;
294 3d7e9092 2003-10-14 devnull 		}
295 3d7e9092 2003-10-14 devnull 		a->value = malloc(v-attrbuf+1);
296 3d7e9092 2003-10-14 devnull 		if(a->value == nil){
297 3d7e9092 2003-10-14 devnull 			free(a->name);
298 3d7e9092 2003-10-14 devnull 			free(a);
299 3d7e9092 2003-10-14 devnull 			break;
300 3d7e9092 2003-10-14 devnull 		}
301 3d7e9092 2003-10-14 devnull 		memmove(a->value, attrbuf, v-attrbuf);
302 3d7e9092 2003-10-14 devnull 		a->value[v-attrbuf] = '\0';
303 3d7e9092 2003-10-14 devnull 		a->next = nil;
304 3d7e9092 2003-10-14 devnull 		if(prev == nil)
305 3d7e9092 2003-10-14 devnull 			attr = a;
306 3d7e9092 2003-10-14 devnull 		else
307 3d7e9092 2003-10-14 devnull 			prev->next = a;
308 3d7e9092 2003-10-14 devnull 		prev = a;
309 3d7e9092 2003-10-14 devnull 		p = q;
310 3d7e9092 2003-10-14 devnull 	}
311 3d7e9092 2003-10-14 devnull 	return attr;
312 3d7e9092 2003-10-14 devnull }
313 3d7e9092 2003-10-14 devnull 
314 3d7e9092 2003-10-14 devnull Plumbattr*
315 3d7e9092 2003-10-14 devnull plumbaddattr(Plumbattr *attr, Plumbattr *new)
316 3d7e9092 2003-10-14 devnull {
317 3d7e9092 2003-10-14 devnull 	Plumbattr *l;
318 3d7e9092 2003-10-14 devnull 
319 3d7e9092 2003-10-14 devnull 	l = attr;
320 3d7e9092 2003-10-14 devnull 	if(l == nil)
321 3d7e9092 2003-10-14 devnull 		return new;
322 3d7e9092 2003-10-14 devnull 	while(l->next != nil)
323 3d7e9092 2003-10-14 devnull 		l = l->next;
324 3d7e9092 2003-10-14 devnull 	l->next = new;
325 3d7e9092 2003-10-14 devnull 	return attr;
326 3d7e9092 2003-10-14 devnull }
327 3d7e9092 2003-10-14 devnull 
328 3d7e9092 2003-10-14 devnull Plumbattr*
329 3d7e9092 2003-10-14 devnull plumbdelattr(Plumbattr *attr, char *name)
330 3d7e9092 2003-10-14 devnull {
331 3d7e9092 2003-10-14 devnull 	Plumbattr *l, *prev;
332 3d7e9092 2003-10-14 devnull 
333 3d7e9092 2003-10-14 devnull 	prev = nil;
334 3d7e9092 2003-10-14 devnull 	for(l=attr; l!=nil; l=l->next){
335 3d7e9092 2003-10-14 devnull 		if(strcmp(name, l->name) == 0)
336 3d7e9092 2003-10-14 devnull 			break;
337 3d7e9092 2003-10-14 devnull 		prev = l;
338 3d7e9092 2003-10-14 devnull 	}
339 3d7e9092 2003-10-14 devnull 	if(l == nil)
340 3d7e9092 2003-10-14 devnull 		return nil;
341 3d7e9092 2003-10-14 devnull 	if(prev)
342 3d7e9092 2003-10-14 devnull 		prev->next = l->next;
343 3d7e9092 2003-10-14 devnull 	else
344 3d7e9092 2003-10-14 devnull 		attr = l->next;
345 3d7e9092 2003-10-14 devnull 	free(l->name);
346 3d7e9092 2003-10-14 devnull 	free(l->value);
347 3d7e9092 2003-10-14 devnull 	free(l);
348 3d7e9092 2003-10-14 devnull 	return attr;
349 3d7e9092 2003-10-14 devnull }
350 3d7e9092 2003-10-14 devnull 
351 3d7e9092 2003-10-14 devnull Plumbmsg*
352 3d7e9092 2003-10-14 devnull plumbunpackpartial(char *buf, int n, int *morep)
353 3d7e9092 2003-10-14 devnull {
354 3d7e9092 2003-10-14 devnull 	Plumbmsg *m;
355 3d7e9092 2003-10-14 devnull 	int i, bad;
356 3d7e9092 2003-10-14 devnull 	char *ntext, *attr;
357 3d7e9092 2003-10-14 devnull 
358 3d7e9092 2003-10-14 devnull 	m = malloc(sizeof(Plumbmsg));
359 3d7e9092 2003-10-14 devnull 	if(m == nil)
360 3d7e9092 2003-10-14 devnull 		return nil;
361 3d7e9092 2003-10-14 devnull 	memset(m, 0, sizeof(Plumbmsg));
362 3d7e9092 2003-10-14 devnull 	if(morep != nil)
363 3d7e9092 2003-10-14 devnull 		*morep = 0;
364 3d7e9092 2003-10-14 devnull 	bad = 0;
365 3d7e9092 2003-10-14 devnull 	i = plumbline(&m->src, buf, 0, n, &bad);
366 3d7e9092 2003-10-14 devnull 	i = plumbline(&m->dst, buf, i, n, &bad);
367 3d7e9092 2003-10-14 devnull 	i = plumbline(&m->wdir, buf, i, n, &bad);
368 3d7e9092 2003-10-14 devnull 	i = plumbline(&m->type, buf, i, n, &bad);
369 3d7e9092 2003-10-14 devnull 	i = plumbline(&attr, buf, i, n, &bad);
370 3d7e9092 2003-10-14 devnull 	m->attr = plumbunpackattr(attr);
371 3d7e9092 2003-10-14 devnull 	free(attr);
372 3d7e9092 2003-10-14 devnull 	i = plumbline(&ntext, buf, i, n, &bad);
373 3d7e9092 2003-10-14 devnull 	m->ndata = atoi(ntext);
374 3d7e9092 2003-10-14 devnull 	if(m->ndata != n-i){
375 3d7e9092 2003-10-14 devnull 		bad = 1;
376 3d7e9092 2003-10-14 devnull 		if(morep!=nil && m->ndata>n-i)
377 3d7e9092 2003-10-14 devnull 			*morep = m->ndata - (n-i);
378 3d7e9092 2003-10-14 devnull 	}
379 3d7e9092 2003-10-14 devnull 	free(ntext);
380 3d7e9092 2003-10-14 devnull 	if(!bad){
381 3d7e9092 2003-10-14 devnull 		m->data = malloc(n-i+1);	/* +1 for '\0' */
382 3d7e9092 2003-10-14 devnull 		if(m->data == nil)
383 3d7e9092 2003-10-14 devnull 			bad = 1;
384 3d7e9092 2003-10-14 devnull 		else{
385 3d7e9092 2003-10-14 devnull 			memmove(m->data, buf+i, m->ndata);
386 3d7e9092 2003-10-14 devnull 			m->ndata = n-i;
387 3d7e9092 2003-10-14 devnull 			/* null-terminate in case it's text */
388 3d7e9092 2003-10-14 devnull 			m->data[m->ndata] = '\0';
389 3d7e9092 2003-10-14 devnull 		}
390 3d7e9092 2003-10-14 devnull 	}
391 3d7e9092 2003-10-14 devnull 	if(bad){
392 3d7e9092 2003-10-14 devnull 		plumbfree(m);
393 3d7e9092 2003-10-14 devnull 		m = nil;
394 3d7e9092 2003-10-14 devnull 	}
395 3d7e9092 2003-10-14 devnull 	return m;
396 3d7e9092 2003-10-14 devnull }
397 3d7e9092 2003-10-14 devnull 
398 3d7e9092 2003-10-14 devnull Plumbmsg*
399 3d7e9092 2003-10-14 devnull plumbunpack(char *buf, int n)
400 3d7e9092 2003-10-14 devnull {
401 3d7e9092 2003-10-14 devnull 	return plumbunpackpartial(buf, n, nil);
402 3d7e9092 2003-10-14 devnull }
403 3d7e9092 2003-10-14 devnull 
404 3d7e9092 2003-10-14 devnull Plumbmsg*
405 3d7e9092 2003-10-14 devnull plumbrecv(int fd)
406 3d7e9092 2003-10-14 devnull {
407 3d7e9092 2003-10-14 devnull 	char *buf;
408 3d7e9092 2003-10-14 devnull 	Plumbmsg *m;
409 3d7e9092 2003-10-14 devnull 	int n, more;
410 3d7e9092 2003-10-14 devnull 
411 3d7e9092 2003-10-14 devnull 	buf = malloc(8192);
412 3d7e9092 2003-10-14 devnull 	if(buf == nil)
413 3d7e9092 2003-10-14 devnull 		return nil;
414 3d7e9092 2003-10-14 devnull 	n = read(fd, buf, 8192);
415 3d7e9092 2003-10-14 devnull 	m = nil;
416 3d7e9092 2003-10-14 devnull 	if(n > 0){
417 3d7e9092 2003-10-14 devnull 		m = plumbunpackpartial(buf, n, &more);
418 3d7e9092 2003-10-14 devnull 		if(m==nil && more>0){
419 3d7e9092 2003-10-14 devnull 			/* we now know how many more bytes to read for complete message */
420 3d7e9092 2003-10-14 devnull 			buf = realloc(buf, n+more);
421 3d7e9092 2003-10-14 devnull 			if(buf == nil)
422 3d7e9092 2003-10-14 devnull 				return nil;
423 3d7e9092 2003-10-14 devnull 			if(readn(fd, buf+n, more) == more)
424 3d7e9092 2003-10-14 devnull 				m = plumbunpackpartial(buf, n+more, nil);
425 3d7e9092 2003-10-14 devnull 		}
426 3d7e9092 2003-10-14 devnull 	}
427 3d7e9092 2003-10-14 devnull 	free(buf);
428 3d7e9092 2003-10-14 devnull 	return m;
429 3d7e9092 2003-10-14 devnull }