op public repos

Blob

Date:: Sun Jan 31 16:52:08 2021 UTC
Message:: 9p: parse lines in rdwr command Use bio(3) to read at most one line of input per iteration, even if there is more than one line available in the input buffer. This makes it easier to interact with line-oriented ctl files like that of factotum(4) from shell scripts, without the need to control when and how much data is flushed to a pipe.
Actions:: History | Blame | Raw File
1 #include <u.h>
2 #include <signal.h>
3 #include <libc.h>
4 #include <bio.h>
5 #include <fcall.h>
6 #include <9pclient.h>
7 #include <auth.h>
8 #include <thread.h>
9 
10 char *addr;
11 
12 void
13 usage(void)
14 {
15 	fprint(2, "usage: 9p [-n] [-a address] [-A aname] cmd args...\n");
16 	fprint(2, "possible cmds:\n");
17 	fprint(2, "	read name\n");
18 	fprint(2, "	readfd name\n");
19 	fprint(2, "	write [-l] name\n");
20 	fprint(2, "	writefd name\n");
21 	fprint(2, "	stat name\n");
22 	fprint(2, "	rdwr name\n");
23 	fprint(2, "	ls [-ldn] name\n");
24 	fprint(2, "without -a, name elem/path means /path on server unix!$ns/elem\n");
25 	threadexitsall("usage");
26 }
27 
28 CFsys *(*nsmnt)(char*, char*) = nsamount;
29 CFsys *(*fsmnt)(int, char*) = fsamount;
30 
31 char *aname;
32 void xread(int, char**);
33 void xwrite(int, char**);
34 void xreadfd(int, char**);
35 void xwritefd(int, char**);
36 void xstat(int, char**);
37 void xls(int, char**);
38 void xrdwr(int, char**);
39 void xrm(int, char**);
40 void xcreate(int, char**);
41 void xcon(int, char**);
42 
43 struct {
44 	char *s;
45 	void (*f)(int, char**);
46 } cmds[] = {
47 	"con", xcon,
48 	"read", xread,
49 	"write", xwrite,
50 	"readfd", xreadfd,
51 	"writefd", xwritefd,
52 	"stat", xstat,
53 	"rdwr", xrdwr,
54 	"ls", xls,
55 	"rm", xrm,
56 	"create", xcreate,
57 };
58 
59 void
60 threadmain(int argc, char **argv)
61 {
62 	char *cmd;
63 	int i;
64 
65 	ARGBEGIN{
66 	case 'A':
67 		aname = EARGF(usage());
68 		break;
69 	case 'a':
70 		addr = EARGF(usage());
71 		if(strchr(addr, '!') == nil)
72 			addr = netmkaddr(addr, "tcp", "9fs");
73 		break;
74 	case 'n':
75 		nsmnt = nsmount;
76 		fsmnt = fsmount;
77 		break;
78 	case 'D':
79 		chatty9pclient = 1;
80 		break;
81 	default:
82 		usage();
83 	}ARGEND
84 
85 	signal(SIGINT, SIG_DFL);
86 
87 	if(argc < 1)
88 		usage();
89 
90 	cmd = argv[0];
91 	for(i=0; i<nelem(cmds); i++){
92 		if(strcmp(cmds[i].s, cmd) == 0){
93 			cmds[i].f(argc, argv);
94 			threadexitsall(0);
95 		}
96 	}
97 	usage();
98 }
99 
100 CFsys*
101 xparse(char *name, char **path)
102 {
103 	int fd;
104 	char *p;
105 	CFsys *fs;
106 
107 	if(addr == nil){
108 		p = strchr(name, '/');
109 		if(p == nil)
110 			p = name+strlen(name);
111 		else
112 			*p++ = 0;
113 		*path = p;
114 		fs = nsmnt(name, aname);
115 		if(fs == nil)
116 			sysfatal("mount: %r");
117 	}else{
118 		*path = name;
119 		if((fd = dial(addr, nil, nil, nil)) < 0)
120 			sysfatal("dial: %r");
121 		if((fs = fsmnt(fd, aname)) == nil)
122 			sysfatal("mount: %r");
123 	}
124 	return fs;
125 }
126 
127 CFid*
128 xopen(char *name, int mode)
129 {
130 	CFid *fid;
131 	CFsys *fs;
132 
133 	fs = xparse(name, &name);
134 	fid = fsopen(fs, name, mode);
135 	if(fid == nil)
136 		sysfatal("fsopen %s: %r", name);
137 	return fid;
138 }
139 
140 int
141 xopenfd(char *name, int mode)
142 {
143 	CFsys *fs;
144 
145 	fs = xparse(name, &name);
146 	return fsopenfd(fs, name, mode);
147 }
148 
149 void
150 xread(int argc, char **argv)
151 {
152 	char buf[4096];
153 	int n;
154 	CFid *fid;
155 
156 	ARGBEGIN{
157 	default:
158 		usage();
159 	}ARGEND
160 
161 	if(argc != 1)
162 		usage();
163 
164 	fid = xopen(argv[0], OREAD);
165 	while((n = fsread(fid, buf, sizeof buf)) > 0)
166 		if(write(1, buf, n) < 0)
167 			sysfatal("write error: %r");
168 	fsclose(fid);
169 	if(n < 0)
170 		sysfatal("read error: %r");
171 	threadexitsall(0);
172 }
173 
174 void
175 xreadfd(int argc, char **argv)
176 {
177 	char buf[4096];
178 	int n;
179 	int fd;
180 
181 	ARGBEGIN{
182 	default:
183 		usage();
184 	}ARGEND
185 
186 	if(argc != 1)
187 		usage();
188 
189 	fd = xopenfd(argv[0], OREAD);
190 	while((n = read(fd, buf, sizeof buf)) > 0)
191 		if(write(1, buf, n) < 0)
192 			sysfatal("write error: %r");
193 	if(n < 0)
194 		sysfatal("read error: %r");
195 	threadexitsall(0);
196 }
197 
198 void
199 xwrite(int argc, char **argv)
200 {
201 	char buf[4096];
202 	int n, did;
203 	CFid *fid;
204 	Biobuf *b;
205 	char *p;
206 	int byline;
207 
208 	byline = 0;
209 	ARGBEGIN{
210 	case 'l':
211 		byline = 1;
212 		break;
213 	default:
214 		usage();
215 	}ARGEND
216 
217 	if(argc != 1)
218 		usage();
219 
220 	did = 0;
221 	fid = xopen(argv[0], OWRITE|OTRUNC);
222 	if(byline){
223 		n = 0;
224 		b = malloc(sizeof *b);
225 		if(b == nil)
226 			sysfatal("out of memory");
227 		Binit(b, 0, OREAD);
228 		while((p = Brdstr(b, '\n', 0)) != nil){
229 			n = strlen(p);
230 			did = 1;
231 			if(fswrite(fid, p, n) != n)
232 				fprint(2, "write: %r\n");
233 		}
234 		free(b);
235 	}else{
236 		while((n = read(0, buf, sizeof buf)) > 0){
237 			did = 1;
238 			if(fswrite(fid, buf, n) != n)
239 				sysfatal("write error: %r");
240 		}
241 	}
242 	if(n == 0 && !did){
243 		if(fswrite(fid, buf, 0) != 0)
244 			sysfatal("write error: %r");
245 	}
246 	if(n < 0)
247 		sysfatal("read error: %r");
248 	fsclose(fid);
249 	threadexitsall(0);
250 }
251 
252 void
253 xwritefd(int argc, char **argv)
254 {
255 	char buf[4096];
256 	int n;
257 	int fd;
258 
259 	ARGBEGIN{
260 	default:
261 		usage();
262 	}ARGEND
263 
264 	if(argc != 1)
265 		usage();
266 
267 	fd = xopenfd(argv[0], OWRITE|OTRUNC);
268 	while((n = read(0, buf, sizeof buf)) > 0)
269 		if(write(fd, buf, n) != n)
270 			sysfatal("write error: %r");
271 	if(n < 0)
272 		sysfatal("read error: %r");
273 	threadexitsall(0);
274 }
275 
276 void
277 xstat(int argc, char **argv)
278 {
279 	Dir *d;
280 	CFsys *fs;
281 	char *name;
282 
283 	ARGBEGIN{
284 	default:
285 		usage();
286 	}ARGEND
287 
288 	if(argc != 1)
289 		usage();
290 
291 	name = argv[0];
292 	fs = xparse(name, &name);
293 	if((d = fsdirstat(fs, name)) == 0)
294 		sysfatal("dirstat: %r");
295 	fmtinstall('D', dirfmt);
296 	fmtinstall('M', dirmodefmt);
297 	print("%D\n", d);
298 	threadexitsall(0);
299 }
300 
301 void
302 xrdwr(int argc, char **argv)
303 {
304 	char buf[4096];
305 	char *p;
306 	int n;
307 	CFid *fid;
308 	Biobuf *b;
309 
310 	ARGBEGIN{
311 	default:
312 		usage();
313 	}ARGEND
314 
315 	if(argc != 1)
316 		usage();
317 
318 	if((b = Bfdopen(0, OREAD)) == nil)
319 		sysfatal("out of memory");
320 	fid = xopen(argv[0], ORDWR);
321 	for(;;){
322 		fsseek(fid, 0, 0);
323 		if((n = fsread(fid, buf, sizeof buf)) < 0)
324 			fprint(2, "read: %r\n");
325 		else{
326 			if(write(1, buf, n) < 0 || write(1, "\n", 1) < 0)
327 				sysfatal("write error: %r");
328 		}
329 		if((p = Brdstr(b, '\n', 1)) == nil)
330 			break;
331 		n = strlen(p);
332 		if(fswrite(fid, p, n) != n)
333 			fprint(2, "write: %r\n");
334 		free(p);
335 	}
336 	fsclose(fid);
337 	Bterm(b);
338 	threadexitsall(0);
339 }
340 
341 void
342 xcreate(int argc, char **argv)
343 {
344 	int i;
345 	CFsys *fs;
346 	CFid *fid;
347 	char *p;
348 
349 	ARGBEGIN{
350 	default:
351 		usage();
352 	}ARGEND
353 
354 	if(argc == 0)
355 		usage();
356 
357 	for(i=0; i<argc; i++){
358 		fs = xparse(argv[i], &p);
359 		if((fid=fscreate(fs, p, OREAD, 0666)) == nil)
360 			fprint(2, "create %s: %r\n", argv[i]);
361 		else
362 			fsclose(fid);
363 		fsunmount(fs);
364 	}
365 }
366 
367 void
368 xrm(int argc, char **argv)
369 {
370 	int i;
371 	CFsys *fs;
372 	char *p;
373 
374 	ARGBEGIN{
375 	default:
376 		usage();
377 	}ARGEND
378 
379 	if(argc == 0)
380 		usage();
381 
382 	for(i=0; i<argc; i++){
383 		fs = xparse(argv[i], &p);
384 		if(fsremove(fs, p) < 0)
385 			fprint(2, "remove %s: %r\n", argv[i]);
386 		fsunmount(fs);
387 	}
388 }
389 
390 void
391 rdcon(void *v)
392 {
393 	int n;
394 	char buf[4096];
395 	CFid *fid;
396 
397 	fid = v;
398 	for(;;){
399 		n = read(0, buf, sizeof buf);
400 		if(n <= 0)
401 			threadexitsall(0);
402 		if(buf[0] == 'R'-'A'+1)
403 			threadexitsall(0);
404 		if(fswrite(fid, buf, n) != n)
405 			fprint(2, "write: %r\n");
406 	}
407 }
408 
409 void
410 xcon(int argc, char **argv)
411 {
412 	char buf[4096], *r, *w, *e;
413 	int n, nocr;
414 	CFid *fid;
415 
416 	nocr = 1;
417 
418 	ARGBEGIN{
419 	case 'r':
420 		nocr = 0;
421 		break;
422 	default:
423 		usage();
424 	}ARGEND
425 
426 	if(argc != 1)
427 		usage();
428 
429 	fid = xopen(argv[0], ORDWR);
430 	proccreate(rdcon, fid, 32768);
431 	for(;;){
432 		n = fsread(fid, buf, sizeof buf);
433 		if(n <= 0)
434 			threadexitsall(0);
435 		if(nocr){
436 			for(r=w=buf, e=buf+n; r<e; r++)
437 				if(*r != '\r')
438 					*w++ = *r;
439 			n = w-buf;
440 		}
441 		if(write(1, buf, n) != n)
442 			threadexitsall(0);
443 	}
444 	fsclose(fid);
445 	threadexitsall(0);
446 }
447 
448 static char *mon[] =
449 {
450 	"Jan", "Feb", "Mar", "Apr", "May", "Jun",
451 	"Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
452 };
453 
454 
455 int
456 timefmt(Fmt *fmt)
457 {
458 	ulong u;
459 	static ulong time0;
460 	Tm *tm;
461 
462 	if(time0 == 0)
463 		time0 = time(0);
464 	u = va_arg(fmt->args, ulong);
465 	tm = localtime(u);
466 	if((long)(time0-u) < 6*30*86400)
467 		return fmtprint(fmt, "%s %2d %02d:%02d",
468 			mon[tm->mon], tm->mday, tm->hour, tm->min);
469 	return fmtprint(fmt, "%s %2d %5d",
470 		mon[tm->mon], tm->mday, tm->year+1900);
471 }
472 
473 static int
474 dircmp(const void *va, const void *vb)
475 {
476 	Dir *a, *b;
477 
478 	a = (Dir*)va;
479 	b = (Dir*)vb;
480 	return strcmp(a->name, b->name);
481 }
482 
483 static int
484 timecmp(const void *va, const void *vb)
485 {
486 	Dir *a, *b;
487 
488 	a = (Dir*)va;
489 	b = (Dir*)vb;
490 	if(a->mtime < b->mtime)
491 		return -1;
492 	else if(a->mtime > b->mtime)
493 		return 1;
494 	else
495 		return 0;
496 }
497 
498 char *dot[] = { "." };
499 
500 void
501 xls(int argc, char **argv)
502 {
503 	char *err, *name, *xname, *f[4], buf[4096];
504 	int i, j, l, sort;
505 	int lflag, dflag, tflag, n, len[4];
506 	Dir *d;
507 	CFid *fid;
508 	CFsys *fs;
509 
510 	err = nil;
511 	sort = 1;
512 	lflag = dflag = tflag = 0;
513 	ARGBEGIN{
514 	case 'n':
515 		sort = 0;
516 		break;
517 	case 'l':
518 		lflag = 1;
519 		break;
520 	case 'd':
521 		dflag = 1;
522 		break;
523 	case 't':
524 		tflag = 1;
525 		break;
526 	}ARGEND
527 
528 	fmtinstall('D', dirfmt);
529 	fmtinstall('M', dirmodefmt);
530 	quotefmtinstall();
531 	fmtinstall('T', timefmt);
532 
533 	if(argc == 0){
534 		argv = dot;
535 		argc = 1;
536 	}
537 	for(i=0; i<argc; i++){
538 		name = argv[i];
539 		fs = xparse(name, &xname);
540 		if((d = fsdirstat(fs, xname)) == nil){
541 			fprint(2, "dirstat %s: %r\n", name);
542 			fsunmount(fs);
543 			err = "errors";
544 			continue;
545 		}
546 		if((d->mode&DMDIR) && !dflag){
547 			if((fid = fsopen(fs, xname, OREAD)) == nil){
548 				fprint(2, "open %s: %r\n", name);
549 				fsunmount(fs);
550 				free(d);
551 				err = "errors";
552 				continue;
553 			}
554 			free(d);
555 			n = fsdirreadall(fid, &d);
556 			fsclose(fid);
557 			if(n < 0){
558 				fprint(2, "dirreadall %s: %r\n", name);
559 				fsunmount(fs);
560 				err = "errors";
561 				continue;
562 			}
563 			if(sort){
564 				if(tflag)
565 					qsort(d, n, sizeof d[0], timecmp);
566 				else
567 					qsort(d, n, sizeof d[0], dircmp);
568 			}
569 			for(j=0; j<4; j++)
570 				len[j] = 0;
571 			for(i=0; i<n; i++){
572 				d[i].type = 'M';
573 				d[i].dev = 0;
574 				snprint(buf, sizeof buf, "%d %s %s %lld",
575 					d[i].dev, d[i].uid, d[i].gid, d[i].length);
576 				getfields(buf, f, 4, 0, " ");
577 				for(j=0; j<4; j++){
578 					l = strlen(f[j]);
579 					if(l > len[j])
580 						len[j] = l;
581 				}
582 			}
583 			for(i=0; i<n; i++){
584 				if(!lflag){
585 					print("%q\n", d[i].name);
586 					continue;
587 				}
588 				print("%M %C %*d %*s %*s %*lld %T %q\n",
589 					d[i].mode, d[i].type, len[0], d[i].dev,
590 					-len[1], d[i].uid, -len[2], d[i].gid,
591 					len[3], d[i].length, d[i].mtime, d[i].name);
592 			}
593 		}else{
594 			d->type = 'M';
595 			d->dev = 0;
596 			if(lflag)
597 				print("%M %C %d %s %s %lld %T %q\n",
598 					d->mode, d->type, d->dev,
599 					d->uid, d->gid, d->length, d->mtime, d->name);
600 			else
601 				print("%q\n", d->name);
602 		}
603 		free(d);
604 	}
605 	threadexitsall(err);
606 }