9pcon.c 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367
  1. #include <u.h>
  2. #include <libc.h>
  3. #include <auth.h>
  4. #include <fcall.h>
  5. #include <bio.h>
  6. uint messagesize = 65536; /* just a buffer size */
  7. void
  8. usage(void)
  9. {
  10. fprint(2, "usage: aux/9pcon [-m messagesize] /srv/service | -c command | -n networkaddress\n");
  11. exits("usage");
  12. }
  13. int
  14. connectcmd(char *cmd)
  15. {
  16. int p[2];
  17. if(pipe(p) < 0)
  18. return -1;
  19. switch(fork()){
  20. case -1:
  21. fprint(2, "fork failed: %r\n");
  22. _exits("exec");
  23. case 0:
  24. dup(p[0], 0);
  25. dup(p[0], 1);
  26. close(p[1]);
  27. execl("/bin/rc", "rc", "-c", cmd, nil);
  28. fprint(2, "exec failed: %r\n");
  29. _exits("exec");
  30. default:
  31. close(p[0]);
  32. return p[1];
  33. }
  34. }
  35. void
  36. watch(int fd)
  37. {
  38. int n;
  39. uchar *buf;
  40. Fcall f;
  41. buf = malloc(messagesize);
  42. if(buf == nil)
  43. sysfatal("out of memory");
  44. while((n = read9pmsg(fd, buf, messagesize)) > 0){
  45. if(convM2S(buf, n, &f) == 0){
  46. print("convM2S: %r\n");
  47. continue;
  48. }
  49. print("\t<- %F\n", &f);
  50. }
  51. if(n == 0)
  52. print("server eof\n");
  53. else
  54. print("read9pmsg from server: %r\n");
  55. }
  56. char*
  57. tversion(Fcall *f, int, char **argv)
  58. {
  59. f->msize = atoi(argv[0]);
  60. if(f->msize > messagesize)
  61. return "message size too big; use -m option on command line";
  62. f->version = argv[1];
  63. return nil;
  64. }
  65. char*
  66. tauth(Fcall *f, int, char **argv)
  67. {
  68. f->afid = atoi(argv[0]);
  69. f->uname = argv[1];
  70. f->aname = argv[2];
  71. return nil;
  72. }
  73. char*
  74. tflush(Fcall *f, int, char **argv)
  75. {
  76. f->oldtag = atoi(argv[0]);
  77. return nil;
  78. }
  79. char*
  80. tattach(Fcall *f, int, char **argv)
  81. {
  82. f->fid = atoi(argv[0]);
  83. f->afid = atoi(argv[1]);
  84. f->uname = argv[2];
  85. f->aname = argv[3];
  86. return nil;
  87. }
  88. char*
  89. twalk(Fcall *f, int argc, char **argv)
  90. {
  91. int i;
  92. if(argc < 2)
  93. return "usage: Twalk tag fid newfid [name...]";
  94. f->fid = atoi(argv[0]);
  95. f->newfid = atoi(argv[1]);
  96. f->nwname = argc-2;
  97. if(f->nwname > MAXWELEM)
  98. return "too many names";
  99. for(i=0; i<argc-2; i++)
  100. f->wname[i] = argv[2+i];
  101. return nil;
  102. }
  103. char*
  104. topen(Fcall *f, int, char **argv)
  105. {
  106. f->fid = atoi(argv[0]);
  107. f->mode = atoi(argv[1]);
  108. return nil;
  109. }
  110. char*
  111. tcreate(Fcall *f, int, char **argv)
  112. {
  113. f->fid = atoi(argv[0]);
  114. f->name = argv[1];
  115. f->perm = strtol(argv[2], 0, 8);
  116. f->mode = atoi(argv[3]);
  117. return nil;
  118. }
  119. char*
  120. tread(Fcall *f, int, char **argv)
  121. {
  122. f->fid = atoi(argv[0]);
  123. f->offset = strtoll(argv[1], 0, 0);
  124. f->count = strtol(argv[2], 0, 0);
  125. return nil;
  126. }
  127. char*
  128. twrite(Fcall *f, int, char **argv)
  129. {
  130. f->fid = atoi(argv[0]);
  131. f->offset = strtoll(argv[1], 0, 0);
  132. f->data = argv[2];
  133. f->count = strlen(argv[2]);
  134. return nil;
  135. }
  136. char*
  137. tclunk(Fcall *f, int, char **argv)
  138. {
  139. f->fid = atoi(argv[0]);
  140. return nil;
  141. }
  142. char*
  143. tremove(Fcall *f, int, char **argv)
  144. {
  145. f->fid = atoi(argv[0]);
  146. return nil;
  147. }
  148. char*
  149. tstat(Fcall *f, int, char **argv)
  150. {
  151. f->fid = atoi(argv[0]);
  152. return nil;
  153. }
  154. ulong
  155. xstrtoul(char *s)
  156. {
  157. if(strcmp(s, "~0") == 0)
  158. return ~0UL;
  159. return strtoul(s, 0, 0);
  160. }
  161. uvlong
  162. xstrtoull(char *s)
  163. {
  164. if(strcmp(s, "~0") == 0)
  165. return ~0ULL;
  166. return strtoull(s, 0, 0);
  167. }
  168. char*
  169. twstat(Fcall *f, int, char **argv)
  170. {
  171. static uchar buf[DIRMAX];
  172. Dir d;
  173. memset(&d, 0, sizeof d);
  174. nulldir(&d);
  175. d.name = argv[1];
  176. d.uid = argv[2];
  177. d.gid = argv[3];
  178. d.mode = xstrtoul(argv[4]);
  179. d.mtime = xstrtoul(argv[5]);
  180. d.length = xstrtoull(argv[6]);
  181. f->fid = atoi(argv[0]);
  182. f->stat = buf;
  183. f->nstat = convD2M(&d, buf, sizeof buf);
  184. if(f->nstat < BIT16SZ)
  185. return "convD2M failed (internal error)";
  186. return nil;
  187. }
  188. int taggen;
  189. char*
  190. settag(Fcall*, int, char **argv)
  191. {
  192. static char buf[120];
  193. taggen = atoi(argv[0])-1;
  194. snprint(buf, sizeof buf, "next tag is %d", taggen+1);
  195. return buf;
  196. }
  197. typedef struct Cmd Cmd;
  198. struct Cmd {
  199. char *name;
  200. int type;
  201. int argc;
  202. char *usage;
  203. char *(*fn)(Fcall *f, int, char**);
  204. };
  205. Cmd msg9p[] = {
  206. "Tversion", Tversion, 2, "messagesize version", tversion,
  207. "Tauth", Tauth, 3, "afid uname aname", tauth,
  208. "Tflush", Tflush, 1, "oldtag", tflush,
  209. "Tattach", Tattach, 4, "fid afid uname aname", tattach,
  210. "Twalk", Twalk, 0, "fid newfid [name...]", twalk,
  211. "Topen", Topen, 2, "fid mode", topen,
  212. "Tcreate", Tcreate, 4, "fid name perm mode", tcreate,
  213. "Tread", Tread, 3, "fid offset count", tread,
  214. "Twrite", Twrite, 3, "fid offset data", twrite,
  215. "Tclunk", Tclunk, 1, "fid", tclunk,
  216. "Tremove", Tremove, 1, "fid", tremove,
  217. "Tstat", Tstat, 1, "fid", tstat,
  218. "Twstat", Twstat, 7, "fid name uid gid mode mtime length", twstat,
  219. "nexttag", 0, 0, "", settag,
  220. };
  221. void
  222. shell9p(int fd)
  223. {
  224. char *e, *f[10], *p;
  225. uchar *buf;
  226. int i, n, nf;
  227. Biobuf b;
  228. Fcall t;
  229. buf = malloc(messagesize);
  230. if(buf == nil){
  231. fprint(2, "out of memory\n");
  232. return;
  233. }
  234. taggen = 0;
  235. Binit(&b, 0, OREAD);
  236. while(p = Brdline(&b, '\n')){
  237. p[Blinelen(&b)-1] = '\0';
  238. if(p[0] == '#')
  239. continue;
  240. if((nf = tokenize(p, f, nelem(f))) == 0)
  241. continue;
  242. for(i=0; i<nelem(msg9p); i++)
  243. if(strcmp(f[0], msg9p[i].name) == 0)
  244. break;
  245. if(i == nelem(msg9p)){
  246. fprint(2, "?unknown message\n");
  247. continue;
  248. }
  249. memset(&t, 0, sizeof t);
  250. t.type = msg9p[i].type;
  251. if(t.type == Tversion)
  252. t.tag = NOTAG;
  253. else
  254. t.tag = ++taggen;
  255. if(nf < 1 || (msg9p[i].argc && nf != 1+msg9p[i].argc)){
  256. fprint(2, "?usage: %s %s\n", msg9p[i].name, msg9p[i].usage);
  257. continue;
  258. }
  259. if(e = msg9p[i].fn(&t, nf-1, f+1)){
  260. fprint(2, "?%s\n", e);
  261. continue;
  262. }
  263. n = convS2M(&t, buf, messagesize);
  264. if(n <= BIT16SZ){
  265. fprint(2, "?message too large for buffer\n");
  266. continue;
  267. }
  268. if(write(fd, buf, n) != n){
  269. fprint(2, "?write fails: %r\n");
  270. break;
  271. }
  272. print("\t-> %F\n", &t);
  273. }
  274. }
  275. void
  276. main(int argc, char **argv)
  277. {
  278. int fd, pid, cmd, net;
  279. cmd = 0;
  280. net = 0;
  281. ARGBEGIN{
  282. case 'c':
  283. cmd = 1;
  284. break;
  285. case 'm':
  286. messagesize = atoi(EARGF(usage()));
  287. break;
  288. case 'n':
  289. net = 1;
  290. break;
  291. default:
  292. usage();
  293. }ARGEND
  294. fmtinstall('F', fcallfmt);
  295. fmtinstall('D', dirfmt);
  296. fmtinstall('M', dirmodefmt);
  297. if(argc != 1)
  298. usage();
  299. if(cmd && net)
  300. usage();
  301. if(cmd)
  302. fd = connectcmd(argv[0]);
  303. else if(net){
  304. fd = dial(netmkaddr(argv[0], "net", "9fs"), 0, 0, 0);
  305. if(fd < 0)
  306. sysfatal("dial: %r");
  307. }else{
  308. fd = open(argv[0], ORDWR);
  309. if(fd < 0)
  310. sysfatal("open: %r");
  311. }
  312. switch(pid = rfork(RFPROC|RFMEM)){
  313. case -1:
  314. sysfatal("rfork: %r");
  315. break;
  316. case 0:
  317. watch(fd);
  318. postnote(PNPROC, getppid(), "kill");
  319. break;
  320. default:
  321. shell9p(fd);
  322. postnote(PNPROC, pid, "kill");
  323. break;
  324. }
  325. exits(nil);
  326. }