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