12 static char *encprotos[] = {
21 char *ealgs = "rc4_256 sha1";
22 int encproto = Encnone;
23 char *aan = "/bin/aan";
24 char *anstring = "tcp!*!0";
31 int connect(char*, char*);
33 void catcher(void*, char*);
34 void sysfatal(char*, ...);
36 int filter(int, char *, char *);
38 static void mksecret(char *, uchar *);
41 * based on libthread's threadsetname, but drags in less library code.
42 * actually just sets the arguments displayed.
45 procsetname(char *fmt, ...)
53 cmdname = vsmprint(fmt, arg);
57 snprint(buf, sizeof buf, "#p/%d/args", getpid());
58 if((fd = open(buf, OWRITE)) >= 0){
59 write(fd, cmdname, strlen(cmdname)+1);
66 post(char *name, char *envname, int srvfd)
71 fd = create(name, OWRITE, 0600);
74 snprint(buf, sizeof(buf), "%d", srvfd);
75 if(write(fd, buf, strlen(buf)) != strlen(buf))
76 sysfatal("srv write: %r");
78 putenv(envname, name);
82 lookup(char *s, char *l[])
86 for (i = 0; l[i] != 0; i++)
87 if (strcmp(l[i], s) == 0)
93 main(int argc, char **argv)
95 char *mntpt, *srvpost, srvfile[64];
96 int backwards = 0, fd, mntflags;
121 /* ignored but allowed for compatibility */
124 if ((encproto = lookup(EARGF(usage()), encprotos)) < 0)
128 ealgs = EARGF(usage());
129 if(*ealgs == 0 || strcmp(ealgs, "clear") == 0)
133 keyspec = EARGF(usage());
139 anstring = EARGF(usage());
142 srvpost = EARGF(usage());
154 mntpt = 0; /* to shut up compiler */
176 if (encproto == Enctls)
177 sysfatal("%s: tls has not yet been implemented", argv[0]);
185 fd = connect(argv[0], argv[1]);
187 fprint(fd, "impo %s %s\n", filterp? "aan": "nofilter", encprotos[encproto]);
189 if (encproto != Encnone && ealgs && ai) {
190 uchar key[16], digest[SHA1dlen];
191 char fromclientsecret[21];
192 char fromserversecret[21];
196 sysfatal("secret too small to ssl");
197 memmove(key+4, ai->secret, 8);
199 /* exchange random numbers */
201 for(i = 0; i < 4; i++)
203 if(write(fd, key, 4) != 4)
204 sysfatal("can't write key part: %r");
205 if(readn(fd, key+12, 4) != 4)
206 sysfatal("can't read key part: %r");
208 /* scramble into two secrets */
209 sha1(key, sizeof(key), digest, nil);
210 mksecret(fromclientsecret, digest);
211 mksecret(fromserversecret, digest+10);
214 fd = filter(fd, filterp, backwards ? nil : argv[0]);
216 /* set up encryption */
217 procsetname("pushssl");
218 fd = pushssl(fd, ealgs, fromclientsecret, fromserversecret, nil);
220 sysfatal("can't establish ssl connection: %r");
223 fd = filter(fd, filterp, backwards ? nil : argv[0]);
229 snprint(srvfile, sizeof(srvfile), "/srv/%s", srvpost);
231 post(srvfile, srvpost, fd);
233 procsetname("mount on %s", mntpt);
234 if(mount(fd, -1, mntpt, mntflags, "") < 0)
235 sysfatal("can't mount %s: %r", argv[1]);
238 if(backwards && argc > 1){
239 exec(argv[1], &argv[1]);
240 sysfatal("exec: %r");
246 catcher(void*, char *msg)
249 if(strcmp(msg, "alarm") == 0)
255 connect(char *system, char *tree)
257 char buf[ERRMAX], dir[128], *na;
260 na = netmkaddr(system, 0, "exportfs");
261 procsetname("dial %s", na);
262 if((fd = dial(na, 0, dir, 0)) < 0)
263 sysfatal("can't dial %s: %r", system);
266 procsetname("auth_proxy auth_getkey proto=p9any role=client %s", keyspec);
267 ai = auth_proxy(fd, auth_getkey, "proto=p9any role=client %s", keyspec);
269 sysfatal("%r: %s", system);
273 procsetname("writing tree name %s", tree);
274 n = write(fd, tree, strlen(tree));
276 sysfatal("can't write tree: %r");
278 strcpy(buf, "can't read tree");
280 procsetname("awaiting OK for %s", tree);
281 n = read(fd, buf, sizeof buf - 1);
282 if(n!=2 || buf[0]!='O' || buf[1]!='K'){
284 sysfatal("timed out connecting to %s", na);
285 buf[sizeof buf - 1] = '\0';
286 sysfatal("bad remote tree: %s", buf);
298 * Ignore doauth==0 on purpose. Is it useful here?
301 procsetname("auth_proxy auth_getkey proto=p9any role=server");
302 ai = auth_proxy(0, auth_getkey, "proto=p9any role=server");
304 sysfatal("auth_proxy: %r");
305 if(auth_chuid(ai, nil) < 0)
306 sysfatal("auth_chuid: %r");
307 putenv("service", "import");
311 open("/dev/null", ORDWR);
313 open("/dev/null", ORDWR);
321 fprint(2, "usage: import [-abcC] [-A] [-E clear|ssl|tls] "
322 "[-e 'crypt auth'|clear] [-k keypattern] [-p] [-n address ] [-z] host remotefs [mountpoint]\n");
327 filter(int fd, char *cmd, char *host)
329 char addr[128], buf[256], *s, *file, *argv[16];
330 int lfd, p[2], len, argc;
333 /* Get a free port and post it to the client. */
334 if (announce(anstring, addr) < 0)
335 sysfatal("filter: Cannot announce %s: %r", anstring);
337 snprint(buf, sizeof(buf), "%s/local", addr);
338 if ((lfd = open(buf, OREAD)) < 0)
339 sysfatal("filter: Cannot open %s: %r", buf);
340 if ((len = read(lfd, buf, sizeof buf - 1)) < 0)
341 sysfatal("filter: Cannot read %s: %r", buf);
344 if ((s = strchr(buf, '\n')) != nil)
346 if (write(fd, buf, len) != len)
347 sysfatal("filter: cannot write port; %r");
349 /* Read address string from connection */
350 if ((len = read(fd, buf, sizeof buf - 1)) < 0)
351 sysfatal("filter: cannot write port; %r");
354 if ((s = strrchr(buf, '!')) == nil)
355 sysfatal("filter: illegally formatted port %s", buf);
356 strecpy(addr, addr+sizeof(addr), netmkaddr(host, "tcp", s+1));
357 strecpy(strrchr(addr, '!'), addr+sizeof(addr), s);
361 fprint(2, "filter: %s\n", addr);
363 snprint(buf, sizeof(buf), "%s", cmd);
364 argc = tokenize(buf, argv, nelem(argv)-3);
366 sysfatal("filter: empty command");
374 if((s = strrchr(argv[0], '/')) != nil)
378 sysfatal("pipe: %r");
380 switch(rfork(RFNOWAIT|RFPROC|RFMEM|RFFDG|RFREND)) {
382 sysfatal("filter: rfork; %r\n");
385 if (dup(p[0], 1) < 0)
386 sysfatal("filter: Cannot dup to 1; %r");
387 if (dup(p[0], 0) < 0)
388 sysfatal("filter: Cannot dup to 0; %r");
392 sysfatal("filter: exec; %r");
402 mksecret(char *t, uchar *f)
404 sprint(t, "%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux",
405 f[0], f[1], f[2], f[3], f[4], f[5], f[6], f[7], f[8], f[9]);