#include <u.h>#include <libc.h>#include <bio.h>#include <auth.h> char *dest = "system";int mountflag = MREPL; void error(char *);void rpc(int, int);void post(char*, int);void mountfs(char*, int); voidusage(void){ fprint(2, "usage: %s [-abcCm] [net!]host [srvname [mtpt]]\n", argv0); exits("usage");} voidignore(void *a, char *c){ USED(a); if(strcmp(c, "alarm") == 0){ fprint(2, "srv: timeout establishing connection to %s\n", dest); exits("timeout"); } noted(NDFLT);} voidmain(int argc, char *argv[]){ int fd; char srv[64], mtpt[64]; char dir[1024]; char err[ERRMAX]; char *p, *p2; int domount, reallymount, try; notify(ignore); domount = 0; reallymount = 0; ARGBEGIN{ case 'a': mountflag |= MAFTER; domount = 1; reallymount = 1; break; case 'b': mountflag |= MBEFORE; domount = 1; reallymount = 1; break; case 'c': mountflag |= MCREATE; domount = 1; reallymount = 1; break; case 'C': mountflag |= MCACHE; domount = 1; reallymount = 1; break; case 'm': domount = 1; reallymount = 1; break; case 'q': domount = 1; reallymount = 0; break; case 'r': /* deprecated -r flag; ignored for compatibility */ break; default: usage(); break; }ARGEND if((mountflag&MAFTER)&&(mountflag&MBEFORE)) usage(); switch(argc){ case 1: /* calculate srv and mtpt from address */ p = strrchr(argv[0], '/'); p = p ? p+1 : argv[0]; snprint(srv, sizeof(srv), "/srv/%.28s", p); p2 = strchr(p, '!'); p2 = p2 ? p2+1 : p; snprint(mtpt, sizeof(mtpt), "/n/%.28s", p2); break; case 2: /* calculate mtpt from address, srv given */ snprint(srv, sizeof(srv), "/srv/%.28s", argv[1]); p = strrchr(argv[0], '/'); p = p ? p+1 : argv[0]; p2 = strchr(p, '!'); p2 = p2 ? p2+1 : p; snprint(mtpt, sizeof(mtpt), "/n/%.28s", p2); break; case 3: /* srv and mtpt given */ domount = 1; reallymount = 1; snprint(srv, sizeof(srv), "/srv/%.28s", argv[1]); snprint(mtpt, sizeof(mtpt), "%.28s", argv[2]); break; default: usage(); } try = 0; dest = *argv;Again: try++; if(access(srv, 0) == 0){ if(domount){ fd = open(srv, ORDWR); if(fd >= 0) goto Mount; remove(srv); } else{ fprint(2, "srv: %s already exists\n", srv); exits(0); } } alarm(10000); dest = netmkaddr(dest, 0, "9fs"); fd = dial(dest, 0, dir, 0); if(fd < 0) { fprint(2, "srv: dial %s: %r\n", dest); exits("dial"); } alarm(0); post(srv, fd); Mount: if(domount == 0 || reallymount == 0) exits(0); if(amount(fd, mtpt, mountflag, "") < 0){ err[0] = 0; errstr(err, sizeof err); if(strstr(err, "hungup") || strstr(err, "timed out")){ remove(srv); if(try == 1) goto Again; } fprint(2, "srv %s: mount failed: %s\n", dest, err); exits("mount"); } exits(0);} voidpost(char *srv, int fd){ int f; char buf[128]; fprint(2, "post...\n"); f = create(srv, OWRITE, 0666); if(f < 0){ sprint(buf, "create(%s)", srv); error(buf); } sprint(buf, "%d", fd); if(write(f, buf, strlen(buf)) != strlen(buf)) error("write");} voiderror(char *s){ fprint(2, "srv %s: %s: %r\n", dest, s); exits("srv: error");}
#include <u.h>
#include <libc.h>
#include <bio.h>
#include <auth.h>
char *dest = "system";
int mountflag = MREPL;
void error(char *);
void rpc(int, int);
void post(char*, int);
void mountfs(char*, int);
void
usage(void)
{
fprint(2, "usage: %s [-abcCm] [net!]host [srvname [mtpt]]\n", argv0);
exits("usage");
}
void
ignore(void *a, char *c)
{
USED(a);
if(strcmp(c, "alarm") == 0){
fprint(2, "srv: timeout establishing connection to %s\n", dest);
exits("timeout");
}
noted(NDFLT);
}
void
main(int argc, char *argv[])
{
int fd;
char srv[64], mtpt[64];
char dir[1024];
char err[ERRMAX];
char *p, *p2;
int domount, reallymount, try;
notify(ignore);
domount = 0;
reallymount = 0;
ARGBEGIN{
case 'a':
mountflag |= MAFTER;
domount = 1;
reallymount = 1;
break;
case 'b':
mountflag |= MBEFORE;
domount = 1;
reallymount = 1;
break;
case 'c':
mountflag |= MCREATE;
domount = 1;
reallymount = 1;
break;
case 'C':
mountflag |= MCACHE;
domount = 1;
reallymount = 1;
break;
case 'm':
domount = 1;
reallymount = 1;
break;
case 'q':
domount = 1;
reallymount = 0;
break;
case 'r':
/* deprecated -r flag; ignored for compatibility */
break;
default:
usage();
break;
}ARGEND
if((mountflag&MAFTER)&&(mountflag&MBEFORE))
usage();
switch(argc){
case 1: /* calculate srv and mtpt from address */
p = strrchr(argv[0], '/');
p = p ? p+1 : argv[0];
snprint(srv, sizeof(srv), "/srv/%.28s", p);
p2 = strchr(p, '!');
p2 = p2 ? p2+1 : p;
snprint(mtpt, sizeof(mtpt), "/n/%.28s", p2);
break;
case 2: /* calculate mtpt from address, srv given */
snprint(srv, sizeof(srv), "/srv/%.28s", argv[1]);
p = strrchr(argv[0], '/');
p = p ? p+1 : argv[0];
p2 = strchr(p, '!');
p2 = p2 ? p2+1 : p;
snprint(mtpt, sizeof(mtpt), "/n/%.28s", p2);
break;
case 3: /* srv and mtpt given */
domount = 1;
reallymount = 1;
snprint(srv, sizeof(srv), "/srv/%.28s", argv[1]);
snprint(mtpt, sizeof(mtpt), "%.28s", argv[2]);
break;
default:
usage();
}
try = 0;
dest = *argv;
Again:
try++;
if(access(srv, 0) == 0){
if(domount){
fd = open(srv, ORDWR);
if(fd >= 0)
goto Mount;
remove(srv);
}
else{
fprint(2, "srv: %s already exists\n", srv);
exits(0);
}
}
alarm(10000);
dest = netmkaddr(dest, 0, "9fs");
fd = dial(dest, 0, dir, 0);
if(fd < 0) {
fprint(2, "srv: dial %s: %r\n", dest);
exits("dial");
}
alarm(0);
post(srv, fd);
Mount:
if(domount == 0 || reallymount == 0)
exits(0);
if(amount(fd, mtpt, mountflag, "") < 0){
err[0] = 0;
errstr(err, sizeof err);
if(strstr(err, "hungup") || strstr(err, "timed out")){
remove(srv);
if(try == 1)
goto Again;
}
fprint(2, "srv %s: mount failed: %s\n", dest, err);
exits("mount");
}
exits(0);
}
void
post(char *srv, int fd)
{
int f;
char buf[128];
fprint(2, "post...\n");
f = create(srv, OWRITE, 0666);
if(f < 0){
sprint(buf, "create(%s)", srv);
error(buf);
}
sprint(buf, "%d", fd);
if(write(f, buf, strlen(buf)) != strlen(buf))
error("write");
}
void
error(char *s)
{
fprint(2, "srv %s: %s: %r\n", dest, s);
exits("srv: error");
}