#include <u.h> #include <signal.h> #include <libc.h> #include <fcall.h> #include <fs.h> #include <thread.h> char *addr; void usage(void) { fprint(2, "usage: 9p [-a address] cmd args...\n"); fprint(2, "possible cmds:\n"); fprint(2, " read name\n"); fprint(2, " readfd name\n"); fprint(2, " write name\n"); fprint(2, " writefd name\n"); fprint(2, " stat name\n"); // fprint(2, " ls name\n"); fprint(2, "without -a, name elem/path means /path on server unix!$ns/elem\n"); exits("usage"); } void xread(int, char**); void xwrite(int, char**); void xreadfd(int, char**); void xwritefd(int, char**); void xstat(int, char**); void xls(int, char**); struct { char *s; void (*f)(int, char**); } cmds[] = { "read", xread, "write", xwrite, "readfd", xreadfd, "writefd", xwritefd, "stat", xstat, // "ls", xls, }; void threadmain(int argc, char **argv) { char *cmd; int i; ARGBEGIN{ case 'a': addr = EARGF(usage()); break; default: usage(); }ARGEND signal(SIGINT, SIG_DFL); if(argc < 1) usage(); cmd = argv[0]; for(i=0; i<nelem(cmds); i++){ if(strcmp(cmds[i].s, cmd) == 0){ cmds[i].f(argc, argv); threadexitsall(0); } } usage(); } Fsys* xparse(char *name, char **path) { int fd; char *p; Fsys *fs; if(addr == nil){ p = strchr(name, '/'); if(p == nil) p = name+strlen(name); else *p++ = 0; *path = p; fs = nsmount(name, ""); if(fs == nil) sysfatal("mount: %r"); }else{ *path = name; fprint(2, "dial %s...", addr); if((fd = dial(addr, nil, nil, nil)) < 0) sysfatal("dial: %r"); if((fs = fsmount(fd, "")) == nil) sysfatal("fsmount: %r"); } return fs; } Fid* xwalk(char *name) { Fid *fid; Fsys *fs; fs = xparse(name, &name); fid = fswalk(fsroot(fs), name); if(fid == nil) sysfatal("fswalk %s: %r", name); return fid; } Fid* xopen(char *name, int mode) { Fid *fid; Fsys *fs; fs = xparse(name, &name); fid = fsopen(fs, name, mode); if(fid == nil) sysfatal("fsopen %s: %r", name); return fid; } int xopenfd(char *name, int mode) { Fsys *fs; fs = xparse(name, &name); return fsopenfd(fs, name, mode); } void xread(int argc, char **argv) { char buf[1024]; int n; Fid *fid; ARGBEGIN{ default: usage(); }ARGEND if(argc != 1) usage(); fid = xopen(argv[0], OREAD); while((n = fsread(fid, buf, sizeof buf)) > 0) write(1, buf, n); if(n < 0) sysfatal("read error: %r"); exits(0); } void xreadfd(int argc, char **argv) { char buf[1024]; int n; int fd; ARGBEGIN{ default: usage(); }ARGEND if(argc != 1) usage(); fd = xopenfd(argv[0], OREAD); while((n = read(fd, buf, sizeof buf)) > 0) write(1, buf, n); if(n < 0) sysfatal("read error: %r"); exits(0); } void xwrite(int argc, char **argv) { char buf[1024]; int n, did; Fid *fid; ARGBEGIN{ default: usage(); }ARGEND if(argc != 1) usage(); did = 0; fid = xopen(argv[0], OWRITE|OTRUNC); while((n = read(0, buf, sizeof buf)) > 0){ did = 1; if(fswrite(fid, buf, n) != n) sysfatal("write error: %r"); } if(n == 0 && !did){ if(fswrite(fid, buf, 0) != 0) sysfatal("write error: %r"); } if(n < 0) sysfatal("read error: %r"); exits(0); } void xwritefd(int argc, char **argv) { char buf[1024]; int n; int fd; ARGBEGIN{ default: usage(); }ARGEND if(argc != 1) usage(); fd = xopenfd(argv[0], OWRITE|OTRUNC); while((n = read(0, buf, sizeof buf)) > 0) if(write(fd, buf, n) != n) sysfatal("write error: %r"); if(n < 0) sysfatal("read error: %r"); exits(0); } void xstat(int argc, char **argv) { Dir *d; Fid *fid; ARGBEGIN{ default: usage(); }ARGEND if(argc != 1) usage(); fid = xwalk(argv[0]); if((d = fsdirfstat(fid)) == 0) sysfatal("dirfstat: %r"); fmtinstall('D', dirfmt); fmtinstall('M', dirmodefmt); print("%D\n", d); exits(0); }