op public repos

Blob

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