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