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