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