op public repos

Blob

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