op public repos

Blob

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