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