ref: bdc1b187c45dbaeaa7dadce4d7999d9d550ec78f
dir: /sys/src/cmd/ns.c/
#include <u.h> #include <libc.h> #include <bio.h> #include <auth.h> #include <fcall.h> #pragma varargck type "P" char* int nsrv; Dir *srv; Biobuf stdout; typedef struct Mount Mount; struct Mount { char *cmd; char *flag; char *new; char *old; char *spec; }; void xlatemnt(Mount*); char *quote(char*); int rflag; void usage(void) { fprint(2, "usage: ns [-r] [pid]\n"); exits("usage"); } void main(int argc, char **argv) { Mount *m; int line, fd, n, pid; char buf[1024], *av[5]; ARGBEGIN{ case 'r': rflag++; break; default: usage(); }ARGEND if(argc > 1) usage(); if(argc == 1){ pid = atoi(argv[0]); if(pid == 0) usage(); }else pid = getpid(); Binit(&stdout, 1, OWRITE); sprint(buf, "/proc/%d/ns", pid); fd = open(buf, OREAD); if(fd < 0) { fprint(2, "ns: open %s: %r\n", buf); exits("open ns"); } for(line=1; ; line++) { n = read(fd, buf, sizeof(buf)); if(n == sizeof(buf)) { fprint(2, "ns: ns string too long\n"); exits("read ns"); } if(n < 0) { fprint(2, "ns: read %r\n"); exits("read ns"); } if(n == 0) break; buf[n] = '\0'; m = mallocz(sizeof(Mount), 1); if(m == nil) { fprint(2, "ns: no memory: %r\n"); exits("no memory"); } n = tokenize(buf, av, 5); switch(n){ case 2: if(strcmp(av[0], "cd") == 0){ Bprint(&stdout, "%s %s\n", av[0], av[1]); continue; } /* fall through */ default: fprint(2, "ns: unrecognized format of ns file: %d elements on line %d\n", n, line); exits("format"); case 5: m->cmd = strdup(av[0]); m->flag = strdup(av[1]); m->new = strdup(av[2]); m->old = strdup(av[3]); m->spec = strdup(av[4]); break; case 4: if(av[1][0] == '-'){ m->cmd = strdup(av[0]); m->flag = strdup(av[1]); m->new = strdup(av[2]); m->old = strdup(av[3]); m->spec = strdup(""); }else{ m->cmd = strdup(av[0]); m->flag = strdup(""); m->new = strdup(av[1]); m->old = strdup(av[2]); m->spec = strdup(av[3]); } break; case 3: m->cmd = strdup(av[0]); m->flag = strdup(""); m->new = strdup(av[1]); m->old = strdup(av[2]); m->spec = strdup(""); break; } if(!rflag && strcmp(m->cmd, "mount")==0) xlatemnt(m); Bprint(&stdout, "%s %s %s %s %s\n", m->cmd, m->flag, quote(m->new), quote(m->old), quote(m->spec)); free(m->cmd); free(m->flag); free(m->new); free(m->old); free(m->spec); free(m); } exits(nil); } void xlatemnt(Mount *m) { int n, fd; char *s, *t, *net, *port; char buf[256]; if(strncmp(m->new, "/net/", 5) != 0) return; s = strdup(m->new); net = s+5; for(t=net; *t!='/'; t++) if(*t == '\0') goto Return; *t = '\0'; port = t+1; for(t=port; *t!='/'; t++) if(*t == '\0') goto Return; *t = '\0'; if(strcmp(t+1, "data") != 0) goto Return; snprint(buf, sizeof buf, "/net/%s/%s/remote", net, port); fd = open(buf, OREAD); if(fd < 0) goto Return; n = read(fd, buf, sizeof buf); close(fd); if(n<=1 || n>sizeof buf) goto Return; if(buf[n-1] == '\n') --n; buf[n] = '\0'; t = malloc(strlen(net)+1+n+1); if(t == nil) goto Return; sprint(t, "%s!%s", net, buf); free(m->new); m->new = t; Return: free(s); } char* quote(char *s) { static char buf[3][1024]; static int i; char *p, *ep; if(strpbrk(s, " '\\\t#$") == nil) return s; i = (i+1)%3; p = &buf[i][0]; ep = &buf[i][1024]; *p++ = '\''; while(p < ep-5){ switch(*s){ case '\0': goto out; case '\'': *p++ = '\''; break; } *p++ = *s++; } out: *p++ = '\''; *p = '\0'; return buf[i]; }