op public repos

Blob

Date:: Sun Oct 21 16:04:56 2012 UTC
Message:: silence more warnings R=rsc http://codereview.appspot.com/6744056
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 	int n;
306 	CFid *fid;
307 
308 	ARGBEGIN{
309 	default:
310 		usage();
311 	}ARGEND
312 
313 	if(argc != 1)
314 		usage();
315 
316 	fid = xopen(argv[0], ORDWR);
317 	for(;;){
318 		fsseek(fid, 0, 0);
319 		if((n = fsread(fid, buf, sizeof buf)) < 0)
320 			fprint(2, "read: %r\n");
321 		else{
322 			if(write(1, buf, n) < 0 || write(1, "\n", 1) < 0)
323 				sysfatal("write error: %r");
324 		}
325 		n = read(0, buf, sizeof buf);
326 		if(n <= 0)
327 			break;
328 		if(buf[n-1] == '\n')
329 			n--;
330 		if(fswrite(fid, buf, n) != n)
331 			fprint(2, "write: %r\n");
332 	}
333 	fsclose(fid);
334 	threadexitsall(0);	
335 }
336 
337 void
338 xcreate(int argc, char **argv)
339 {
340 	int i;
341 	CFsys *fs;
342 	CFid *fid;
343 	char *p;
344 
345 	ARGBEGIN{
346 	default:
347 		usage();
348 	}ARGEND
349 	
350 	if(argc == 0)
351 		usage();
352 	
353 	for(i=0; i<argc; i++){
354 		fs = xparse(argv[i], &p);
355 		if((fid=fscreate(fs, p, OREAD, 0666)) == nil)
356 			fprint(2, "create %s: %r\n", argv[i]);
357 		else
358 			fsclose(fid);
359 		fsunmount(fs);
360 	}
361 }
362 
363 void
364 xrm(int argc, char **argv)
365 {
366 	int i;
367 	CFsys *fs;
368 	char *p;
369 
370 	ARGBEGIN{
371 	default:
372 		usage();
373 	}ARGEND
374 	
375 	if(argc == 0)
376 		usage();
377 	
378 	for(i=0; i<argc; i++){
379 		fs = xparse(argv[i], &p);
380 		if(fsremove(fs, p) < 0)
381 			fprint(2, "remove %s: %r\n", argv[i]);
382 		fsunmount(fs);
383 	}
384 }
385 
386 void
387 rdcon(void *v)
388 {
389 	int n;
390 	char buf[4096];
391 	CFid *fid;
392 	
393 	fid = v;
394 	for(;;){
395 		n = read(0, buf, sizeof buf);
396 		if(n <= 0)
397 			threadexitsall(0);
398 		if(buf[0] == 'R'-'A'+1)
399 			threadexitsall(0);
400 		if(fswrite(fid, buf, n) != n)
401 			fprint(2, "write: %r\n");
402 	}
403 }
404 
405 void
406 xcon(int argc, char **argv)
407 {
408 	char buf[4096], *r, *w, *e;
409 	int n, nocr;
410 	CFid *fid;
411 	
412 	nocr = 1;
413 
414 	ARGBEGIN{
415 	case 'r':
416 		nocr = 0;
417 		break;
418 	default:
419 		usage();
420 	}ARGEND
421 
422 	if(argc != 1)
423 		usage();
424 
425 	fid = xopen(argv[0], ORDWR);
426 	proccreate(rdcon, fid, 32768);
427 	for(;;){
428 		n = fsread(fid, buf, sizeof buf);
429 		if(n <= 0)
430 			threadexitsall(0);
431 		if(nocr){
432 			for(r=w=buf, e=buf+n; r<e; r++)
433 				if(*r != '\r')
434 					*w++ = *r;
435 			n = w-buf;
436 		}
437 		if(write(1, buf, n) != n)
438 			threadexitsall(0);
439 	}
440 	fsclose(fid);
441 	threadexitsall(0);	
442 }
443 
444 static char *mon[] = 
445 {
446 	"Jan", "Feb", "Mar", "Apr", "May", "Jun", 
447 	"Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
448 };
449 
450 
451 int
452 timefmt(Fmt *fmt)
453 {
454 	ulong u;
455 	static ulong time0;
456 	Tm *tm;
457 	
458 	if(time0 == 0)
459 		time0 = time(0);
460 	u = va_arg(fmt->args, ulong);
461 	tm = localtime(u);
462 	if((long)(time0-u) < 6*30*86400)
463 		return fmtprint(fmt, "%s %2d %02d:%02d",
464 			mon[tm->mon], tm->mday, tm->hour, tm->min);
465 	return fmtprint(fmt, "%s %2d %5d",
466 		mon[tm->mon], tm->mday, tm->year+1900);
467 }
468 
469 static int
470 dircmp(const void *va, const void *vb)
471 {
472 	Dir *a, *b;
473 	
474 	a = (Dir*)va;
475 	b = (Dir*)vb;
476 	return strcmp(a->name, b->name);
477 }
478 
479 static int
480 timecmp(const void *va, const void *vb)
481 {
482 	Dir *a, *b;
483 	
484 	a = (Dir*)va;
485 	b = (Dir*)vb;
486 	if(a->mtime < b->mtime)
487 		return -1;
488 	else if(a->mtime > b->mtime)
489 		return 1;
490 	else
491 		return 0;
492 }
493 
494 char *dot[] = { "." };
495 
496 void
497 xls(int argc, char **argv)
498 {
499 	char *err, *name, *xname, *f[4], buf[4096];
500 	int i, j, l, sort;
501 	int lflag, dflag, tflag, n, len[4];
502 	Dir *d;
503 	CFid *fid;
504 	CFsys *fs;
505 
506 	err = nil;
507 	sort = 1;
508 	lflag = dflag = tflag = 0;
509 	ARGBEGIN{
510 	case 'n':
511 		sort = 0;
512 		break;
513 	case 'l':
514 		lflag = 1;
515 		break;
516 	case 'd':
517 		dflag = 1;
518 		break;
519 	case 't':
520 		tflag = 1;
521 		break;
522 	}ARGEND
523 	
524 	fmtinstall('D', dirfmt);
525 	fmtinstall('M', dirmodefmt);
526 	quotefmtinstall();
527 	fmtinstall('T', timefmt);
528 	
529 	if(argc == 0){
530 		argv = dot;
531 		argc = 1;
532 	}
533 	for(i=0; i<argc; i++){
534 		name = argv[i];
535 		fs = xparse(name, &xname);
536 		if((d = fsdirstat(fs, xname)) == nil){
537 			fprint(2, "dirstat %s: %r\n", name);
538 			fsunmount(fs);
539 			err = "errors";
540 			continue;
541 		}
542 		if((d->mode&DMDIR) && !dflag){
543 			if((fid = fsopen(fs, xname, OREAD)) == nil){
544 				fprint(2, "open %s: %r\n", name);
545 				fsunmount(fs);
546 				free(d);
547 				err = "errors";
548 				continue;
549 			}
550 			free(d);
551 			n = fsdirreadall(fid, &d);
552 			fsclose(fid);
553 			if(n < 0){
554 				fprint(2, "dirreadall %s: %r\n", name);
555 				fsunmount(fs);
556 				err = "errors";
557 				continue;
558 			}
559 			if(sort){
560 				if(tflag)
561 					qsort(d, n, sizeof d[0], timecmp);
562 				else
563 					qsort(d, n, sizeof d[0], dircmp);
564 			}
565 			for(j=0; j<4; j++)
566 				len[j] = 0;
567 			for(i=0; i<n; i++){
568 				d[i].type = 'M';
569 				d[i].dev = 0;
570 				snprint(buf, sizeof buf, "%d %s %s %lld",
571 					d[i].dev, d[i].uid, d[i].gid, d[i].length);
572 				getfields(buf, f, 4, 0, " ");
573 				for(j=0; j<4; j++){
574 					l = strlen(f[j]);
575 					if(l > len[j])
576 						len[j] = l;
577 				}
578 			}
579 			for(i=0; i<n; i++){
580 				if(!lflag){
581 					print("%q\n", d[i].name);
582 					continue;
583 				}
584 				print("%M %C %*d %*s %*s %*lld %T %q\n",
585 					d[i].mode, d[i].type, len[0], d[i].dev,
586 					-len[1], d[i].uid, -len[2], d[i].gid,
587 					len[3], d[i].length, d[i].mtime, d[i].name);
588 			}
589 		}else{
590 			d->type = 'M';
591 			d->dev = 0;
592 			if(lflag)
593 				print("%M %C %d %s %s %lld %T %q\n",
594 					d->mode, d->type, d->dev,
595 					d->uid, d->gid, d->length, d->mtime, d->name);
596 			else
597 				print("%q\n", d->name);
598 		}
599 		free(d);
600 	}
601 	threadexitsall(err);
602 }
603