int mountflag = MREPL;

#include <u.h>
#include <libc.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);
int doauth = 1;
int asnone = 0;

void
usage(void)
{
fprint(2, "usage: %s [-abcCmnNq] [net!]host [srvname [mtpt]]\n", argv0);
fprint(2, " or %s -e [-abcCmnNq] command [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");
}
if(strstr(c, "write on closed pipe") == 0){
fprint(2, "write on closed pipe\n");
noted(NCONT);
}
noted(NDFLT);
}

int
connectcmd(char *cmd)
{
int p[2];

if(pipe(p) < 0)
return -1;
switch(fork()){
case -1:
fprint(2, "fork failed: %r\n");
_exits("exec");
case 0:
rfork(RFNOTEG);
dup(p[0], 0);
dup(p[0], 1);
close(p[0]);
close(p[1]);
execl("/bin/rc", "rc", "-c", cmd, nil);
fprint(2, "exec failed: %r\n");
_exits("exec");
default:
close(p[0]);
return p[1];
}
}

void
main(int argc, char *argv[])
{
int fd, doexec;
char *srv, *mtpt;
char dir[1024];
char err[ERRMAX];
char *p, *p2;
int domount, reallymount, try, sleeptime;

notify(ignore);

domount = 0;
reallymount = 0;
doexec = 0;
sleeptime = 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 'e':
doexec = 1;
break;
case 'm':
domount = 1;
reallymount = 1;
break;
case 'N':
asnone = 1;
/* no break */
case 'n':
doauth = 0;
break;
case 'q':
domount = 1;
reallymount = 0;
break;
case 'r':
/* deprecated -r flag; ignored for compatibility */
break;
case 's':
sleeptime = atoi(EARGF(usage()));
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];
srv = smprint("/srv/%s", p);
p2 = strchr(p, '!');
p2 = p2 ? p2+1 : p;
mtpt = smprint("/n/%s", p2);
break;
case 2: /* calculate mtpt from address, srv given */
srv = smprint("/srv/%s", argv[1]);
p = strrchr(argv[0], '/');
p = p ? p+1 : argv[0];
p2 = strchr(p, '!');
p2 = p2 ? p2+1 : p;
mtpt = smprint("/n/%s", p2);
break;
case 3: /* srv and mtpt given */
domount = 1;
reallymount = 1;
srv = smprint("/srv/%s", argv[1]);
mtpt = smprint("%s", argv[2]);
break;
default:
srv = mtpt = nil;
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);
if(doexec)
fd = connectcmd(dest);
else{
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);

if(sleeptime){
fprint(2, "sleep...");
sleep(sleeptime*1000);
}

post(srv, fd);

Mount:
if(domount == 0 || reallymount == 0)
exits(0);

if(asnone){
try = open("#c/user", OWRITE);
if(try < 0 || write(try, "none", 4) != 4){
fprint(2, "srv %s: can't become none: %r\n", dest);
exits("becomenone");
}
try = 0;
}

if((!doauth && mount(fd, -1, mtpt, mountflag, "") == -1)
|| (doauth && amount(fd, mtpt, mountflag, "") == -1)){
err[0] = 0;
errstr(err, sizeof err);
if(strstr(err, "Hangup") || 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");
}