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