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] 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 name\n");
24 	fprint(2, "without -a, name elem/path means /path on server unix!$ns/elem\n");
25 	threadexitsall("usage");
26 }
27 
28 void xread(int, char**);
29 void xwrite(int, char**);
30 void xreadfd(int, char**);
31 void xwritefd(int, char**);
32 void xstat(int, char**);
33 void xls(int, char**);
34 void xrdwr(int, char**);
35 
36 struct {
37 	char *s;
38 	void (*f)(int, char**);
39 } cmds[] = {
40 	"read", xread,
41 	"write", xwrite,
42 	"readfd", xreadfd,
43 	"writefd", xwritefd,
44 	"stat", xstat,
45 	"rdwr", xrdwr,
46 //	"ls", xls,
47 };
48 
49 void
50 threadmain(int argc, char **argv)
51 {
52 	char *cmd;
53 	int i;
54 
55 	ARGBEGIN{
56 	case 'a':
57 		addr = EARGF(usage());
58 		break;
59 	case 'D':
60 		chatty9pclient = 1;
61 		break;
62 	default:
63 		usage();
64 	}ARGEND
65 
66 	signal(SIGINT, SIG_DFL);
67 
68 	if(argc < 1)
69 		usage();
70 
71 	cmd = argv[0];
72 	for(i=0; i<nelem(cmds); i++){
73 		if(strcmp(cmds[i].s, cmd) == 0){
74 			cmds[i].f(argc, argv);
75 			threadexitsall(0);
76 		}
77 	}
78 	usage();	
79 }
80 
81 CFsys*
82 xparse(char *name, char **path)
83 {
84 	int fd;
85 	char *p;
86 	CFsys *fs;
87 
88 	if(addr == nil){
89 		p = strchr(name, '/');
90 		if(p == nil)
91 			p = name+strlen(name);
92 		else
93 			*p++ = 0;
94 		*path = p;
95 		fs = nsamount(name, "");
96 		if(fs == nil)
97 			sysfatal("mount: %r");
98 	}else{
99 		*path = name;
100 		if((fd = dial(addr, nil, nil, nil)) < 0)
101 			sysfatal("dial: %r");
102 		if((fs = fsamount(fd, "")) == nil)
103 			sysfatal("fsmount: %r");
104 	}
105 	return fs;
106 }
107 
108 CFid*
109 xopen(char *name, int mode)
110 {
111 	CFid *fid;
112 	CFsys *fs;
113 
114 	fs = xparse(name, &name);
115 	fid = fsopen(fs, name, mode);
116 	if(fid == nil)
117 		sysfatal("fsopen %s: %r", name);
118 	return fid;
119 }
120 
121 int
122 xopenfd(char *name, int mode)
123 {
124 	CFsys *fs;
125 
126 	fs = xparse(name, &name);
127 	return fsopenfd(fs, name, mode);
128 }
129 
130 void
131 xread(int argc, char **argv)
132 {
133 	char buf[4096];
134 	int n;
135 	CFid *fid;
136 
137 	ARGBEGIN{
138 	default:
139 		usage();
140 	}ARGEND
141 
142 	if(argc != 1)
143 		usage();
144 
145 	fid = xopen(argv[0], OREAD);
146 	while((n = fsread(fid, buf, sizeof buf)) > 0)
147 		write(1, buf, n);
148 	if(n < 0)
149 		sysfatal("read error: %r");
150 	threadexitsall(0);	
151 }
152 
153 void
154 xreadfd(int argc, char **argv)
155 {
156 	char buf[4096];
157 	int n;
158 	int fd;
159 
160 	ARGBEGIN{
161 	default:
162 		usage();
163 	}ARGEND
164 
165 	if(argc != 1)
166 		usage();
167 
168 	fd = xopenfd(argv[0], OREAD);
169 	while((n = read(fd, buf, sizeof buf)) > 0)
170 		write(1, buf, n);
171 	if(n < 0)
172 		sysfatal("read error: %r");
173 	threadexitsall(0);	
174 }
175 
176 void
177 xwrite(int argc, char **argv)
178 {
179 	char buf[4096];
180 	int n, did;
181 	CFid *fid;
182 	Biobuf *b;
183 	char *p;
184 	int byline;
185 
186 	byline = 0;
187 	ARGBEGIN{
188 	case 'l':
189 		byline = 1;
190 		break;
191 	default:
192 		usage();
193 	}ARGEND
194 
195 	if(argc != 1)
196 		usage();
197 
198 	did = 0;
199 	fid = xopen(argv[0], OWRITE|OTRUNC);
200 	if(byline){
201 		n = 0;
202 		b = malloc(sizeof *b);
203 		if(b == nil)
204 			sysfatal("out of memory");
205 		Binit(b, 0, OREAD);
206 		while((p = Brdstr(b, '\n', 0)) != nil){
207 			n = strlen(p);
208 			did = 1;
209 			if(fswrite(fid, p, n) != n)
210 				fprint(2, "write: %r\n");
211 		}
212 		free(b);
213 	}else{
214 		while((n = read(0, buf, sizeof buf)) > 0){
215 			did = 1;
216 			if(fswrite(fid, buf, n) != n)
217 				sysfatal("write error: %r");
218 		}
219 	}
220 	if(n == 0 && !did){
221 		if(fswrite(fid, buf, 0) != 0)
222 			sysfatal("write error: %r");
223 	}
224 	if(n < 0)
225 		sysfatal("read error: %r");
226 	threadexitsall(0);	
227 }
228 
229 void
230 xwritefd(int argc, char **argv)
231 {
232 	char buf[4096];
233 	int n;
234 	int fd;
235 
236 	ARGBEGIN{
237 	default:
238 		usage();
239 	}ARGEND
240 
241 	if(argc != 1)
242 		usage();
243 
244 	fd = xopenfd(argv[0], OWRITE|OTRUNC);
245 	while((n = read(0, buf, sizeof buf)) > 0)
246 		if(write(fd, buf, n) != n)
247 			sysfatal("write error: %r");
248 	if(n < 0)
249 		sysfatal("read error: %r");
250 	threadexitsall(0);	
251 }
252 
253 void
254 xstat(int argc, char **argv)
255 {
256 	Dir *d;
257 	CFsys *fs;
258 	char *name;
259 
260 	ARGBEGIN{
261 	default:
262 		usage();
263 	}ARGEND
264 
265 	if(argc != 1)
266 		usage();
267 
268 	name = argv[0];
269 	fs = xparse(name, &name);
270 	if((d = fsdirstat(fs, name)) == 0)
271 		sysfatal("dirstat: %r");
272 	fmtinstall('D', dirfmt);
273 	fmtinstall('M', dirmodefmt);
274 	print("%D\n", d);
275 	threadexitsall(0);
276 }
277 
278 void
279 xrdwr(int argc, char **argv)
280 {
281 	char buf[4096];
282 	int n;
283 	CFid *fid;
284 
285 	ARGBEGIN{
286 	default:
287 		usage();
288 	}ARGEND
289 
290 	if(argc != 1)
291 		usage();
292 
293 	fid = xopen(argv[0], ORDWR);
294 	for(;;){
295 		if((n = fsread(fid, buf, sizeof buf)) < 0)
296 			fprint(2, "read: %r\n");
297 		else{
298 			write(1, buf, n);
299 			write(1, "\n", 1);
300 		}
301 		n = read(0, buf, sizeof buf);
302 		if(n <= 0)
303 			break;
304 		if(buf[n-1] == '\n')
305 			n--;
306 		if(fswrite(fid, buf, n) != n)
307 			fprint(2, "write: %r\n");
308 	}
309 	threadexitsall(0);	
310 }