1 #include <u.h> 2 #include <libc.h> 3 #include <bio.h> 4 #include <auth.h> 5 6 char *dest = "system"; 7 int mountflag = MREPL; 8 9 void error(char *); 10 void rpc(int, int); 11 void post(char*, int); 12 void mountfs(char*, int); 13 14 void 15 usage(void) 16 { 17 fprint(2, "usage: %s [-abcCm] [net!]host [srvname [mtpt]]\n", argv0); 18 fprint(2, " or %s -e [-abcCm] command [srvname [mtpt]]\n", argv0); 19 20 exits("usage"); 21 } 22 23 void 24 ignore(void *a, char *c) 25 { 26 USED(a); 27 if(strcmp(c, "alarm") == 0){ 28 fprint(2, "srv: timeout establishing connection to %s\n", dest); 29 exits("timeout"); 30 } 31 if(strstr(c, "write on closed pipe") == 0){ 32 fprint(2, "write on closed pipe\n"); 33 noted(NCONT); 34 } 35 noted(NDFLT); 36 } 37 38 int 39 connectcmd(char *cmd) 40 { 41 int p[2]; 42 43 if(pipe(p) < 0) 44 return -1; 45 switch(fork()){ 46 case -1: 47 fprint(2, "fork failed: %r\n"); 48 _exits("exec"); 49 case 0: 50 rfork(RFNOTEG); 51 dup(p[0], 0); 52 dup(p[0], 1); 53 close(p[1]); 54 execl("/bin/rc", "rc", "-c", cmd, nil); 55 fprint(2, "exec failed: %r\n"); 56 _exits("exec"); 57 default: 58 close(p[0]); 59 return p[1]; 60 } 61 } 62 63 void 64 main(int argc, char *argv[]) 65 { 66 int fd, doexec; 67 char *srv, *mtpt; 68 char dir[1024]; 69 char err[ERRMAX]; 70 char *p, *p2; 71 int domount, reallymount, try, sleeptime; 72 73 notify(ignore); 74 75 domount = 0; 76 reallymount = 0; 77 doexec = 0; 78 sleeptime = 0; 79 80 ARGBEGIN{ 81 case 'a': 82 mountflag |= MAFTER; 83 domount = 1; 84 reallymount = 1; 85 break; 86 case 'b': 87 mountflag |= MBEFORE; 88 domount = 1; 89 reallymount = 1; 90 break; 91 case 'c': 92 mountflag |= MCREATE; 93 domount = 1; 94 reallymount = 1; 95 break; 96 case 'C': 97 mountflag |= MCACHE; 98 domount = 1; 99 reallymount = 1; 100 break; 101 case 'e': 102 doexec = 1; 103 break; 104 case 'm': 105 domount = 1; 106 reallymount = 1; 107 break; 108 case 'q': 109 domount = 1; 110 reallymount = 0; 111 break; 112 case 'r': 113 /* deprecated -r flag; ignored for compatibility */ 114 break; 115 case 's': 116 sleeptime = atoi(EARGF(usage())); 117 break; 118 default: 119 usage(); 120 break; 121 }ARGEND 122 123 if((mountflag&MAFTER)&&(mountflag&MBEFORE)) 124 usage(); 125 126 switch(argc){ 127 case 1: /* calculate srv and mtpt from address */ 128 p = strrchr(argv[0], '/'); 129 p = p ? p+1 : argv[0]; 130 srv = smprint("/srv/%s", p); 131 p2 = strchr(p, '!'); 132 p2 = p2 ? p2+1 : p; 133 mtpt = smprint("/n/%s", p2); 134 break; 135 case 2: /* calculate mtpt from address, srv given */ 136 srv = smprint("/srv/%s", argv[1]); 137 p = strrchr(argv[0], '/'); 138 p = p ? p+1 : argv[0]; 139 p2 = strchr(p, '!'); 140 p2 = p2 ? p2+1 : p; 141 mtpt = smprint("/n/%s", p2); 142 break; 143 case 3: /* srv and mtpt given */ 144 domount = 1; 145 reallymount = 1; 146 srv = smprint("/srv/%s", argv[1]); 147 mtpt = smprint("%s", argv[2]); 148 break; 149 default: 150 srv = mtpt = nil; 151 usage(); 152 } 153 154 try = 0; 155 dest = *argv; 156 Again: 157 try++; 158 159 if(access(srv, 0) == 0){ 160 if(domount){ 161 fd = open(srv, ORDWR); 162 if(fd >= 0) 163 goto Mount; 164 remove(srv); 165 } 166 else{ 167 fprint(2, "srv: %s already exists\n", srv); 168 exits(0); 169 } 170 } 171 172 alarm(10000); 173 if(doexec) 174 fd = connectcmd(dest); 175 else{ 176 dest = netmkaddr(dest, 0, "9fs"); 177 fd = dial(dest, 0, dir, 0); 178 } 179 if(fd < 0) { 180 fprint(2, "srv: dial %s: %r\n", dest); 181 exits("dial"); 182 } 183 alarm(0); 184 185 if(sleeptime){ 186 fprint(2, "sleep..."); 187 sleep(sleeptime*1000); 188 } 189 190 post(srv, fd); 191 192 Mount: 193 if(domount == 0 || reallymount == 0) 194 exits(0); 195 196 if(amount(fd, mtpt, mountflag, "") < 0){ 197 err[0] = 0; 198 errstr(err, sizeof err); 199 if(strstr(err, "Hangup") || strstr(err, "hungup") || strstr(err, "timed out")){ 200 remove(srv); 201 if(try == 1) 202 goto Again; 203 } 204 fprint(2, "srv %s: mount failed: %s\n", dest, err); 205 exits("mount"); 206 } 207 exits(0); 208 } 209 210 void 211 post(char *srv, int fd) 212 { 213 int f; 214 char buf[128]; 215 216 fprint(2, "post...\n"); 217 f = create(srv, OWRITE, 0666); 218 if(f < 0){ 219 sprint(buf, "create(%s)", srv); 220 error(buf); 221 } 222 sprint(buf, "%d", fd); 223 if(write(f, buf, strlen(buf)) != strlen(buf)) 224 error("write"); 225 } 226 227 void 228 error(char *s) 229 { 230 fprint(2, "srv %s: %s: %r\n", dest, s); 231 exits("srv: error"); 232 } 233