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