op public repos

Blob

Date:: Thu Aug 11 17:33:43 2005 UTC
Message:: silence various warnings
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 	if(n < 0)
155 		sysfatal("read error: %r");
156 	threadexitsall(0);	
157 }
158 
159 void
160 xreadfd(int argc, char **argv)
161 {
162 	char buf[4096];
163 	int n;
164 	int fd;
165 
166 	ARGBEGIN{
167 	default:
168 		usage();
169 	}ARGEND
170 
171 	if(argc != 1)
172 		usage();
173 
174 	fd = xopenfd(argv[0], OREAD);
175 	while((n = read(fd, buf, sizeof buf)) > 0)
176 		write(1, buf, n);
177 	if(n < 0)
178 		sysfatal("read error: %r");
179 	threadexitsall(0);	
180 }
181 
182 void
183 xwrite(int argc, char **argv)
184 {
185 	char buf[4096];
186 	int n, did;
187 	CFid *fid;
188 	Biobuf *b;
189 	char *p;
190 	int byline;
191 
192 	byline = 0;
193 	ARGBEGIN{
194 	case 'l':
195 		byline = 1;
196 		break;
197 	default:
198 		usage();
199 	}ARGEND
200 
201 	if(argc != 1)
202 		usage();
203 
204 	did = 0;
205 	fid = xopen(argv[0], OWRITE|OTRUNC);
206 	if(byline){
207 		n = 0;
208 		b = malloc(sizeof *b);
209 		if(b == nil)
210 			sysfatal("out of memory");
211 		Binit(b, 0, OREAD);
212 		while((p = Brdstr(b, '\n', 0)) != nil){
213 			n = strlen(p);
214 			did = 1;
215 			if(fswrite(fid, p, n) != n)
216 				fprint(2, "write: %r\n");
217 		}
218 		free(b);
219 	}else{
220 		while((n = read(0, buf, sizeof buf)) > 0){
221 			did = 1;
222 			if(fswrite(fid, buf, n) != n)
223 				sysfatal("write error: %r");
224 		}
225 	}
226 	if(n == 0 && !did){
227 		if(fswrite(fid, buf, 0) != 0)
228 			sysfatal("write error: %r");
229 	}
230 	if(n < 0)
231 		sysfatal("read error: %r");
232 	threadexitsall(0);	
233 }
234 
235 void
236 xwritefd(int argc, char **argv)
237 {
238 	char buf[4096];
239 	int n;
240 	int fd;
241 
242 	ARGBEGIN{
243 	default:
244 		usage();
245 	}ARGEND
246 
247 	if(argc != 1)
248 		usage();
249 
250 	fd = xopenfd(argv[0], OWRITE|OTRUNC);
251 	while((n = read(0, buf, sizeof buf)) > 0)
252 		if(write(fd, buf, n) != n)
253 			sysfatal("write error: %r");
254 	if(n < 0)
255 		sysfatal("read error: %r");
256 	threadexitsall(0);	
257 }
258 
259 void
260 xstat(int argc, char **argv)
261 {
262 	Dir *d;
263 	CFsys *fs;
264 	char *name;
265 
266 	ARGBEGIN{
267 	default:
268 		usage();
269 	}ARGEND
270 
271 	if(argc != 1)
272 		usage();
273 
274 	name = argv[0];
275 	fs = xparse(name, &name);
276 	if((d = fsdirstat(fs, name)) == 0)
277 		sysfatal("dirstat: %r");
278 	fmtinstall('D', dirfmt);
279 	fmtinstall('M', dirmodefmt);
280 	print("%D\n", d);
281 	threadexitsall(0);
282 }
283 
284 void
285 xrdwr(int argc, char **argv)
286 {
287 	char buf[4096];
288 	int n;
289 	CFid *fid;
290 
291 	ARGBEGIN{
292 	default:
293 		usage();
294 	}ARGEND
295 
296 	if(argc != 1)
297 		usage();
298 
299 	fid = xopen(argv[0], ORDWR);
300 	for(;;){
301 		if((n = fsread(fid, buf, sizeof buf)) < 0)
302 			fprint(2, "read: %r\n");
303 		else{
304 			write(1, buf, n);
305 			write(1, "\n", 1);
306 		}
307 		n = read(0, buf, sizeof buf);
308 		if(n <= 0)
309 			break;
310 		if(buf[n-1] == '\n')
311 			n--;
312 		if(fswrite(fid, buf, n) != n)
313 			fprint(2, "write: %r\n");
314 	}
315 	threadexitsall(0);	
316 }
317 
318 static char *mon[] = 
319 {
320 	"Jan", "Feb", "Mar", "Apr", "May", "Jun", 
321 	"Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
322 };
323 
324 
325 int
326 timefmt(Fmt *fmt)
327 {
328 	ulong u;
329 	static ulong time0;
330 	Tm *tm;
331 	
332 	if(time0 == 0)
333 		time0 = time(0);
334 	u = va_arg(fmt->args, ulong);
335 	tm = localtime(u);
336 	if((long)(time0-u) < 6*30*86400)
337 		return fmtprint(fmt, "%s %2d %02d:%02d",
338 			mon[tm->mon], tm->mday, tm->hour, tm->min);
339 	return fmtprint(fmt, "%s %2d %5d",
340 		mon[tm->mon], tm->mday, tm->year+1900);
341 }
342 
343 static int
344 dircmp(const void *va, const void *vb)
345 {
346 	Dir *a, *b;
347 	
348 	a = (Dir*)va;
349 	b = (Dir*)vb;
350 	return strcmp(a->name, b->name);
351 }
352 
353 void
354 xls(int argc, char **argv)
355 {
356 	char *err, *name, *xname, *f[4], buf[4096];
357 	int nf, i, j, l;
358 	int lflag, dflag, n, len[4];
359 	Dir *d;
360 	CFid *fid;
361 	CFsys *fs;
362 
363 	err = nil;
364 	lflag = dflag = 0;
365 	ARGBEGIN{
366 	case 'l':
367 		lflag = 1;
368 		break;
369 	case 'd':
370 		dflag = 1;
371 		break;
372 	}ARGEND
373 	
374 	fmtinstall('D', dirfmt);
375 	fmtinstall('M', dirmodefmt);
376 	quotefmtinstall();
377 	fmtinstall('T', timefmt);
378 	
379 	for(i=0; i<argc; i++){
380 		name = argv[i];
381 		fs = xparse(name, &xname);
382 		if((d = fsdirstat(fs, xname)) == nil){
383 			fprint(2, "dirstat %s: %r\n", name);
384 			fsunmount(fs);
385 			err = "errors";
386 			continue;
387 		}
388 		if((d->mode&DMDIR) && !dflag){
389 			if((fid = fsopen(fs, xname, OREAD)) == nil){
390 				fprint(2, "open %s: %r\n", name);
391 				fsunmount(fs);
392 				free(d);
393 				err = "errors";
394 				continue;
395 			}
396 			free(d);
397 			n = fsdirreadall(fid, &d);
398 			fsclose(fid);
399 			if(n < 0){
400 				fprint(2, "dirreadall %s: %r\n", name);
401 				fsunmount(fs);
402 				err = "errors";
403 				continue;
404 			}
405 			qsort(d, n, sizeof d[0], dircmp);
406 			for(j=0; j<5; j++)
407 				len[j] = 0;
408 			for(i=0; i<n; i++){
409 				d[i].type = 'M';
410 				d[i].dev = 0;
411 				snprint(buf, sizeof buf, "%d %s %s %lld",
412 					d[i].dev, d[i].uid, d[i].gid, d[i].length);
413 				nf = getfields(buf, f, 4, 0, " ");
414 				for(j=0; j<4; j++){
415 					l = strlen(f[j]);
416 					if(l > len[j])
417 						len[j] = l;
418 				}
419 			}
420 			for(i=0; i<n; i++)
421 				print("%M %C %*d %*s %*s %*lld %T %q\n",
422 					d[i].mode, d[i].type, len[0], d[i].dev,
423 					-len[1], d[i].uid, -len[2], d[i].gid,
424 					len[3], d[i].length, d[i].mtime, d[i].name);
425 			
426 		}else{
427 			d->type = 'M';
428 			d->dev = 0;
429 			print("%M %C %d %s %s %lld %T %q\n",
430 				d->mode, d->type, d->dev,
431 				d->uid, d->gid, d->length, d->mtime, d->name);
432 		}
433 		free(d);
434 	}
435 	threadexitsall(err);
436 }
437