1 #include <u.h>
2 #include <libc.h>
3 #include <fcall.h>
4 #include <fs.h>
5 
6 char *addr;
7 
8 void
9 usage(void)
10 {
11 	fprint(2, "usage: 9p [-a address] cmd args...\n");
12 	fprint(2, "possible cmds:\n");
13 	fprint(2, "	read name\n");
14 	fprint(2, "	write name\n");
15 	fprint(2, "	stat name\n");
16 //	fprint(2, "	ls name\n");
17 	fprint(2, "without -a, name elem/path means /path on server unix!$ns/elem\n");
18 	exits("usage");
19 }
20 
21 void xread(int, char**);
22 void xwrite(int, char**);
23 void xstat(int, char**);
24 void xls(int, char**);
25 
26 struct {
27 	char *s;
28 	void (*f)(int, char**);
29 } cmds[] = {
30 	"read", xread,
31 	"write", xwrite,
32 	"stat", xstat,
33 //	"ls", xls,
34 };
35 
36 void
37 main(int argc, char **argv)
38 {
39 	char *cmd;
40 	int i;
41 
42 	ARGBEGIN{
43 	case 'a':
44 		addr = EARGF(usage());
45 		break;
46 	default:
47 		usage();
48 	}ARGEND
49 
50 	if(argc < 1)
51 		usage();
52 
53 	cmd = argv[0];
54 	for(i=0; i<nelem(cmds); i++){
55 		if(strcmp(cmds[i].s, cmd) == 0){
56 			cmds[i].f(argc, argv);
57 			exits(0);
58 		}
59 	}
60 	usage();	
61 }
62 
63 Fsys*
64 xparse(char *name, char **path)
65 {
66 	int fd;
67 	char *ns;
68 	char *p;
69 	Fsys *fs;
70 
71 	if(addr == nil){
72 		p = strchr(name, '/');
73 		if(p == nil)
74 			p = name+strlen(name);
75 		else
76 			*p++ = 0;
77 		*path = p;
78 		if(*name == 0)
79 			usage();
80 		ns = getenv("ns");
81 		if(ns == nil)
82 			sysfatal("ns not set");
83 		addr = smprint("unix!%s/%s", ns, name);
84 		if(addr == nil)
85 			sysfatal("out of memory");
86 	}else
87 		*path = name;
88 
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 	return fs;
95 }
96 
97 Fid*
98 xwalk(char *name)
99 {
100 	Fid *fid;
101 	Fsys *fs;
102 
103 	fs = xparse(name, &name);
104 	fid = fswalk(fsroot(fs), name);
105 	if(fid == nil)
106 		sysfatal("fswalk %s: %r", name);
107 	return fid;
108 }
109 
110 Fid*
111 xopen(char *name, int mode)
112 {
113 	Fid *fid;
114 	Fsys *fs;
115 
116 	fs = xparse(name, &name);
117 	fid = fsopen(fs, name, mode);
118 	if(fid == nil)
119 		sysfatal("fsopen %s: %r", name);
120 	return fid;
121 }
122 
123 void
124 xread(int argc, char **argv)
125 {
126 	char buf[1024];
127 	int n;
128 	Fid *fid;
129 
130 	ARGBEGIN{
131 	default:
132 		usage();
133 	}ARGEND
134 
135 	if(argc != 1)
136 		usage();
137 
138 	fid = xopen(argv[0], OREAD);
139 	while((n = fsread(fid, buf, sizeof buf)) > 0)
140 		write(1, buf, n);
141 	if(n < 0)
142 		sysfatal("read error: %r");
143 	exits(0);	
144 }
145 
146 void
147 xwrite(int argc, char **argv)
148 {
149 	char buf[1024];
150 	int n;
151 	Fid *fid;
152 
153 	ARGBEGIN{
154 	default:
155 		usage();
156 	}ARGEND
157 
158 	if(argc != 1)
159 		usage();
160 
161 	fid = xopen(argv[0], OWRITE|OTRUNC);
162 	while((n = read(0, buf, sizeof buf)) > 0)
163 		if(fswrite(fid, buf, n) != n)
164 			sysfatal("write error: %r");
165 	if(n < 0)
166 		sysfatal("read error: %r");
167 	exits(0);	
168 }
169 
170 void
171 xstat(int argc, char **argv)
172 {
173 	Dir *d;
174 	Fid *fid;
175 
176 	ARGBEGIN{
177 	default:
178 		usage();
179 	}ARGEND
180 
181 	if(argc != 1)
182 		usage();
183 
184 	fid = xwalk(argv[0]);
185 	if((d = fsdirfstat(fid)) < 0)
186 		sysfatal("dirfstat: %r");
187 	fmtinstall('D', dirfmt);
188 	fmtinstall('M', dirmodefmt);
189 	print("%D\n", d);
190 	exits(0);
191 }