op public repos

Blob

Date:: Tue Nov 27 20:39:06 2007 UTC
Message:: merge
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 		write(1, buf, n);
167 	fsclose(fid);
168 	if(n < 0)
169 		sysfatal("read error: %r");
170 	threadexitsall(0);	
171 }
172 
173 void
174 xreadfd(int argc, char **argv)
175 {
176 	char buf[4096];
177 	int n;
178 	int fd;
179 
180 	ARGBEGIN{
181 	default:
182 		usage();
183 	}ARGEND
184 
185 	if(argc != 1)
186 		usage();
187 
188 	fd = xopenfd(argv[0], OREAD);
189 	while((n = read(fd, buf, sizeof buf)) > 0)
190 		write(1, buf, n);
191 	if(n < 0)
192 		sysfatal("read error: %r");
193 	threadexitsall(0);	
194 }
195 
196 void
197 xwrite(int argc, char **argv)
198 {
199 	char buf[4096];
200 	int n, did;
201 	CFid *fid;
202 	Biobuf *b;
203 	char *p;
204 	int byline;
205 
206 	byline = 0;
207 	ARGBEGIN{
208 	case 'l':
209 		byline = 1;
210 		break;
211 	default:
212 		usage();
213 	}ARGEND
214 
215 	if(argc != 1)
216 		usage();
217 
218 	did = 0;
219 	fid = xopen(argv[0], OWRITE|OTRUNC);
220 	if(byline){
221 		n = 0;
222 		b = malloc(sizeof *b);
223 		if(b == nil)
224 			sysfatal("out of memory");
225 		Binit(b, 0, OREAD);
226 		while((p = Brdstr(b, '\n', 0)) != nil){
227 			n = strlen(p);
228 			did = 1;
229 			if(fswrite(fid, p, n) != n)
230 				fprint(2, "write: %r\n");
231 		}
232 		free(b);
233 	}else{
234 		while((n = read(0, buf, sizeof buf)) > 0){
235 			did = 1;
236 			if(fswrite(fid, buf, n) != n)
237 				sysfatal("write error: %r");
238 		}
239 	}
240 	if(n == 0 && !did){
241 		if(fswrite(fid, buf, 0) != 0)
242 			sysfatal("write error: %r");
243 	}
244 	if(n < 0)
245 		sysfatal("read error: %r");
246 	fsclose(fid);
247 	threadexitsall(0);	
248 }
249 
250 void
251 xwritefd(int argc, char **argv)
252 {
253 	char buf[4096];
254 	int n;
255 	int fd;
256 
257 	ARGBEGIN{
258 	default:
259 		usage();
260 	}ARGEND
261 
262 	if(argc != 1)
263 		usage();
264 
265 	fd = xopenfd(argv[0], OWRITE|OTRUNC);
266 	while((n = read(0, buf, sizeof buf)) > 0)
267 		if(write(fd, buf, n) != n)
268 			sysfatal("write error: %r");
269 	if(n < 0)
270 		sysfatal("read error: %r");
271 	threadexitsall(0);	
272 }
273 
274 void
275 xstat(int argc, char **argv)
276 {
277 	Dir *d;
278 	CFsys *fs;
279 	char *name;
280 
281 	ARGBEGIN{
282 	default:
283 		usage();
284 	}ARGEND
285 
286 	if(argc != 1)
287 		usage();
288 
289 	name = argv[0];
290 	fs = xparse(name, &name);
291 	if((d = fsdirstat(fs, name)) == 0)
292 		sysfatal("dirstat: %r");
293 	fmtinstall('D', dirfmt);
294 	fmtinstall('M', dirmodefmt);
295 	print("%D\n", d);
296 	threadexitsall(0);
297 }
298 
299 void
300 xrdwr(int argc, char **argv)
301 {
302 	char buf[4096];
303 	int n;
304 	CFid *fid;
305 
306 	ARGBEGIN{
307 	default:
308 		usage();
309 	}ARGEND
310 
311 	if(argc != 1)
312 		usage();
313 
314 	fid = xopen(argv[0], ORDWR);
315 	for(;;){
316 		fsseek(fid, 0, 0);
317 		if((n = fsread(fid, buf, sizeof buf)) < 0)
318 			fprint(2, "read: %r\n");
319 		else{
320 			write(1, buf, n);
321 			write(1, "\n", 1);
322 		}
323 		n = read(0, buf, sizeof buf);
324 		if(n <= 0)
325 			break;
326 		if(buf[n-1] == '\n')
327 			n--;
328 		if(fswrite(fid, buf, n) != n)
329 			fprint(2, "write: %r\n");
330 	}
331 	fsclose(fid);
332 	threadexitsall(0);	
333 }
334 
335 void
336 xcreate(int argc, char **argv)
337 {
338 	int i;
339 	CFsys *fs;
340 	CFid *fid;
341 	char *p;
342 
343 	ARGBEGIN{
344 	default:
345 		usage();
346 	}ARGEND
347 	
348 	if(argc == 0)
349 		usage();
350 	
351 	for(i=0; i<argc; i++){
352 		fs = xparse(argv[i], &p);
353 		if((fid=fscreate(fs, p, OREAD, 0666)) == nil)
354 			fprint(2, "create %s: %r\n", argv[i]);
355 		else
356 			fsclose(fid);
357 		fsunmount(fs);
358 	}
359 }
360 
361 void
362 xrm(int argc, char **argv)
363 {
364 	int i;
365 	CFsys *fs;
366 	char *p;
367 
368 	ARGBEGIN{
369 	default:
370 		usage();
371 	}ARGEND
372 	
373 	if(argc == 0)
374 		usage();
375 	
376 	for(i=0; i<argc; i++){
377 		fs = xparse(argv[i], &p);
378 		if(fsremove(fs, p) < 0)
379 			fprint(2, "remove %s: %r\n", argv[i]);
380 		fsunmount(fs);
381 	}
382 }
383 
384 void
385 rdcon(void *v)
386 {
387 	int n;
388 	char buf[4096];
389 	CFid *fid;
390 	
391 	fid = v;
392 	for(;;){
393 		n = read(0, buf, sizeof buf);
394 		if(n <= 0)
395 			threadexitsall(0);
396 		if(buf[0] == 'R'-'A'+1)
397 			threadexitsall(0);
398 		if(fswrite(fid, buf, n) != n)
399 			fprint(2, "write: %r\n");
400 	}
401 }
402 
403 void
404 xcon(int argc, char **argv)
405 {
406 	char buf[4096], *r, *w, *e;
407 	int n, nocr;
408 	CFid *fid;
409 	
410 	nocr = 1;
411 
412 	ARGBEGIN{
413 	case 'r':
414 		nocr = 0;
415 		break;
416 	default:
417 		usage();
418 	}ARGEND
419 
420 	if(argc != 1)
421 		usage();
422 
423 	fid = xopen(argv[0], ORDWR);
424 	proccreate(rdcon, fid, 32768);
425 	for(;;){
426 		n = fsread(fid, buf, sizeof buf);
427 		if(n <= 0)
428 			threadexitsall(0);
429 		if(nocr){
430 			for(r=w=buf, e=buf+n; r<e; r++)
431 				if(*r != '\r')
432 					*w++ = *r;
433 			n = w-buf;
434 		}
435 		if(write(1, buf, n) != n)
436 			threadexitsall(0);
437 	}
438 	fsclose(fid);
439 	threadexitsall(0);	
440 }
441 
442 static char *mon[] = 
443 {
444 	"Jan", "Feb", "Mar", "Apr", "May", "Jun", 
445 	"Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
446 };
447 
448 
449 int
450 timefmt(Fmt *fmt)
451 {
452 	ulong u;
453 	static ulong time0;
454 	Tm *tm;
455 	
456 	if(time0 == 0)
457 		time0 = time(0);
458 	u = va_arg(fmt->args, ulong);
459 	tm = localtime(u);
460 	if((long)(time0-u) < 6*30*86400)
461 		return fmtprint(fmt, "%s %2d %02d:%02d",
462 			mon[tm->mon], tm->mday, tm->hour, tm->min);
463 	return fmtprint(fmt, "%s %2d %5d",
464 		mon[tm->mon], tm->mday, tm->year+1900);
465 }
466 
467 static int
468 dircmp(const void *va, const void *vb)
469 {
470 	Dir *a, *b;
471 	
472 	a = (Dir*)va;
473 	b = (Dir*)vb;
474 	return strcmp(a->name, b->name);
475 }
476 
477 char *dot[] = { "." };
478 
479 void
480 xls(int argc, char **argv)
481 {
482 	char *err, *name, *xname, *f[4], buf[4096];
483 	int nf, i, j, l, sort;
484 	int lflag, dflag, n, len[4];
485 	Dir *d;
486 	CFid *fid;
487 	CFsys *fs;
488 
489 	err = nil;
490 	sort = 0;
491 	lflag = dflag = 0;
492 	ARGBEGIN{
493 	case 'n':
494 		sort = 0;
495 		break;
496 	case 'l':
497 		lflag = 1;
498 		break;
499 	case 'd':
500 		dflag = 1;
501 		break;
502 	}ARGEND
503 	
504 	fmtinstall('D', dirfmt);
505 	fmtinstall('M', dirmodefmt);
506 	quotefmtinstall();
507 	fmtinstall('T', timefmt);
508 	
509 	if(argc == 0){
510 		argv = dot;
511 		argc = 1;
512 	}
513 	for(i=0; i<argc; i++){
514 		name = argv[i];
515 		fs = xparse(name, &xname);
516 		if((d = fsdirstat(fs, xname)) == nil){
517 			fprint(2, "dirstat %s: %r\n", name);
518 			fsunmount(fs);
519 			err = "errors";
520 			continue;
521 		}
522 		if((d->mode&DMDIR) && !dflag){
523 			if((fid = fsopen(fs, xname, OREAD)) == nil){
524 				fprint(2, "open %s: %r\n", name);
525 				fsunmount(fs);
526 				free(d);
527 				err = "errors";
528 				continue;
529 			}
530 			free(d);
531 			n = fsdirreadall(fid, &d);
532 			fsclose(fid);
533 			if(n < 0){
534 				fprint(2, "dirreadall %s: %r\n", name);
535 				fsunmount(fs);
536 				err = "errors";
537 				continue;
538 			}
539 			if(sort)
540 				qsort(d, n, sizeof d[0], dircmp);
541 			for(j=0; j<4; j++)
542 				len[j] = 0;
543 			for(i=0; i<n; i++){
544 				d[i].type = 'M';
545 				d[i].dev = 0;
546 				snprint(buf, sizeof buf, "%d %s %s %lld",
547 					d[i].dev, d[i].uid, d[i].gid, d[i].length);
548 				nf = getfields(buf, f, 4, 0, " ");
549 				for(j=0; j<4; j++){
550 					l = strlen(f[j]);
551 					if(l > len[j])
552 						len[j] = l;
553 				}
554 			}
555 			for(i=0; i<n; i++)
556 				print("%M %C %*d %*s %*s %*lld %T %q\n",
557 					d[i].mode, d[i].type, len[0], d[i].dev,
558 					-len[1], d[i].uid, -len[2], d[i].gid,
559 					len[3], d[i].length, d[i].mtime, d[i].name);
560 			
561 		}else{
562 			d->type = 'M';
563 			d->dev = 0;
564 			print("%M %C %d %s %s %lld %T %q\n",
565 				d->mode, d->type, d->dev,
566 				d->uid, d->gid, d->length, d->mtime, d->name);
567 		}
568 		free(d);
569 	}
570 	threadexitsall(err);
571 }
572