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 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 eb423b5c 2005-01-04 devnull CFsys*
 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 eb423b5c 2005-01-04 devnull 	CFsys *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 eb423b5c 2005-01-04 devnull CFid*
104 d3df3087 2003-12-06 devnull xopen(char *name, int mode)
105 d3df3087 2003-12-06 devnull {
106 eb423b5c 2005-01-04 devnull 	CFid *fid;
107 eb423b5c 2005-01-04 devnull 	CFsys *fs;
108 d3df3087 2003-12-06 devnull 
109 d3df3087 2003-12-06 devnull 	fs = xparse(name, &name);
110 d3df3087 2003-12-06 devnull 	fid = fsopen(fs, name, mode);
111 d3df3087 2003-12-06 devnull 	if(fid == nil)
112 d3df3087 2003-12-06 devnull 		sysfatal("fsopen %s: %r", name);
113 d3df3087 2003-12-06 devnull 	return fid;
114 d3df3087 2003-12-06 devnull }
115 d3df3087 2003-12-06 devnull 
116 32f69c36 2003-12-11 devnull int
117 32f69c36 2003-12-11 devnull xopenfd(char *name, int mode)
118 32f69c36 2003-12-11 devnull {
119 eb423b5c 2005-01-04 devnull 	CFsys *fs;
120 32f69c36 2003-12-11 devnull 
121 32f69c36 2003-12-11 devnull 	fs = xparse(name, &name);
122 32f69c36 2003-12-11 devnull 	return fsopenfd(fs, name, mode);
123 32f69c36 2003-12-11 devnull }
124 32f69c36 2003-12-11 devnull 
125 d3df3087 2003-12-06 devnull void
126 d3df3087 2003-12-06 devnull xread(int argc, char **argv)
127 d3df3087 2003-12-06 devnull {
128 d3df3087 2003-12-06 devnull 	char buf[1024];
129 d3df3087 2003-12-06 devnull 	int n;
130 eb423b5c 2005-01-04 devnull 	CFid *fid;
131 d3df3087 2003-12-06 devnull 
132 d3df3087 2003-12-06 devnull 	ARGBEGIN{
133 d3df3087 2003-12-06 devnull 	default:
134 d3df3087 2003-12-06 devnull 		usage();
135 d3df3087 2003-12-06 devnull 	}ARGEND
136 d3df3087 2003-12-06 devnull 
137 d3df3087 2003-12-06 devnull 	if(argc != 1)
138 d3df3087 2003-12-06 devnull 		usage();
139 d3df3087 2003-12-06 devnull 
140 d3df3087 2003-12-06 devnull 	fid = xopen(argv[0], OREAD);
141 d3df3087 2003-12-06 devnull 	while((n = fsread(fid, buf, sizeof buf)) > 0)
142 d3df3087 2003-12-06 devnull 		write(1, buf, n);
143 d3df3087 2003-12-06 devnull 	if(n < 0)
144 d3df3087 2003-12-06 devnull 		sysfatal("read error: %r");
145 d3df3087 2003-12-06 devnull 	exits(0);	
146 d3df3087 2003-12-06 devnull }
147 d3df3087 2003-12-06 devnull 
148 d3df3087 2003-12-06 devnull void
149 32f69c36 2003-12-11 devnull xreadfd(int argc, char **argv)
150 32f69c36 2003-12-11 devnull {
151 32f69c36 2003-12-11 devnull 	char buf[1024];
152 32f69c36 2003-12-11 devnull 	int n;
153 32f69c36 2003-12-11 devnull 	int fd;
154 32f69c36 2003-12-11 devnull 
155 32f69c36 2003-12-11 devnull 	ARGBEGIN{
156 32f69c36 2003-12-11 devnull 	default:
157 32f69c36 2003-12-11 devnull 		usage();
158 32f69c36 2003-12-11 devnull 	}ARGEND
159 32f69c36 2003-12-11 devnull 
160 32f69c36 2003-12-11 devnull 	if(argc != 1)
161 32f69c36 2003-12-11 devnull 		usage();
162 32f69c36 2003-12-11 devnull 
163 32f69c36 2003-12-11 devnull 	fd = xopenfd(argv[0], OREAD);
164 32f69c36 2003-12-11 devnull 	while((n = read(fd, buf, sizeof buf)) > 0)
165 32f69c36 2003-12-11 devnull 		write(1, buf, n);
166 32f69c36 2003-12-11 devnull 	if(n < 0)
167 32f69c36 2003-12-11 devnull 		sysfatal("read error: %r");
168 32f69c36 2003-12-11 devnull 	exits(0);	
169 32f69c36 2003-12-11 devnull }
170 32f69c36 2003-12-11 devnull 
171 32f69c36 2003-12-11 devnull void
172 d3df3087 2003-12-06 devnull xwrite(int argc, char **argv)
173 d3df3087 2003-12-06 devnull {
174 d3df3087 2003-12-06 devnull 	char buf[1024];
175 3e610791 2004-10-22 devnull 	int n, did;
176 eb423b5c 2005-01-04 devnull 	CFid *fid;
177 d3df3087 2003-12-06 devnull 
178 d3df3087 2003-12-06 devnull 	ARGBEGIN{
179 d3df3087 2003-12-06 devnull 	default:
180 d3df3087 2003-12-06 devnull 		usage();
181 d3df3087 2003-12-06 devnull 	}ARGEND
182 d3df3087 2003-12-06 devnull 
183 d3df3087 2003-12-06 devnull 	if(argc != 1)
184 d3df3087 2003-12-06 devnull 		usage();
185 d3df3087 2003-12-06 devnull 
186 3e610791 2004-10-22 devnull 	did = 0;
187 d3df3087 2003-12-06 devnull 	fid = xopen(argv[0], OWRITE|OTRUNC);
188 3e610791 2004-10-22 devnull 	while((n = read(0, buf, sizeof buf)) > 0){
189 3e610791 2004-10-22 devnull 		did = 1;
190 d3df3087 2003-12-06 devnull 		if(fswrite(fid, buf, n) != n)
191 d3df3087 2003-12-06 devnull 			sysfatal("write error: %r");
192 3e610791 2004-10-22 devnull 	}
193 3e610791 2004-10-22 devnull 	if(n == 0 && !did){
194 3e610791 2004-10-22 devnull 		if(fswrite(fid, buf, 0) != 0)
195 3e610791 2004-10-22 devnull 			sysfatal("write error: %r");
196 3e610791 2004-10-22 devnull 	}
197 d3df3087 2003-12-06 devnull 	if(n < 0)
198 d3df3087 2003-12-06 devnull 		sysfatal("read error: %r");
199 d3df3087 2003-12-06 devnull 	exits(0);	
200 d3df3087 2003-12-06 devnull }
201 d3df3087 2003-12-06 devnull 
202 d3df3087 2003-12-06 devnull void
203 32f69c36 2003-12-11 devnull xwritefd(int argc, char **argv)
204 32f69c36 2003-12-11 devnull {
205 32f69c36 2003-12-11 devnull 	char buf[1024];
206 32f69c36 2003-12-11 devnull 	int n;
207 32f69c36 2003-12-11 devnull 	int fd;
208 32f69c36 2003-12-11 devnull 
209 32f69c36 2003-12-11 devnull 	ARGBEGIN{
210 32f69c36 2003-12-11 devnull 	default:
211 32f69c36 2003-12-11 devnull 		usage();
212 32f69c36 2003-12-11 devnull 	}ARGEND
213 32f69c36 2003-12-11 devnull 
214 32f69c36 2003-12-11 devnull 	if(argc != 1)
215 32f69c36 2003-12-11 devnull 		usage();
216 32f69c36 2003-12-11 devnull 
217 32f69c36 2003-12-11 devnull 	fd = xopenfd(argv[0], OWRITE|OTRUNC);
218 32f69c36 2003-12-11 devnull 	while((n = read(0, buf, sizeof buf)) > 0)
219 32f69c36 2003-12-11 devnull 		if(write(fd, buf, n) != n)
220 32f69c36 2003-12-11 devnull 			sysfatal("write error: %r");
221 32f69c36 2003-12-11 devnull 	if(n < 0)
222 32f69c36 2003-12-11 devnull 		sysfatal("read error: %r");
223 32f69c36 2003-12-11 devnull 	exits(0);	
224 32f69c36 2003-12-11 devnull }
225 32f69c36 2003-12-11 devnull 
226 32f69c36 2003-12-11 devnull void
227 d3df3087 2003-12-06 devnull xstat(int argc, char **argv)
228 d3df3087 2003-12-06 devnull {
229 d3df3087 2003-12-06 devnull 	Dir *d;
230 eb423b5c 2005-01-04 devnull 	CFsys *fs;
231 eb423b5c 2005-01-04 devnull 	char *name;
232 d3df3087 2003-12-06 devnull 
233 d3df3087 2003-12-06 devnull 	ARGBEGIN{
234 d3df3087 2003-12-06 devnull 	default:
235 d3df3087 2003-12-06 devnull 		usage();
236 d3df3087 2003-12-06 devnull 	}ARGEND
237 d3df3087 2003-12-06 devnull 
238 d3df3087 2003-12-06 devnull 	if(argc != 1)
239 d3df3087 2003-12-06 devnull 		usage();
240 d3df3087 2003-12-06 devnull 
241 eb423b5c 2005-01-04 devnull 	name = argv[0];
242 eb423b5c 2005-01-04 devnull 	fs = xparse(name, &name);
243 eb423b5c 2005-01-04 devnull 	if((d = fsdirstat(fs, name)) == 0)
244 eb423b5c 2005-01-04 devnull 		sysfatal("dirstat: %r");
245 d3df3087 2003-12-06 devnull 	fmtinstall('D', dirfmt);
246 d3df3087 2003-12-06 devnull 	fmtinstall('M', dirmodefmt);
247 d3df3087 2003-12-06 devnull 	print("%D\n", d);
248 d3df3087 2003-12-06 devnull 	exits(0);
249 d3df3087 2003-12-06 devnull }