1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153 |
- /*
- * cpu.c - Make a connection to a cpu server
- *
- * Invoked by listen as 'cpu -R | -N service net netdir'
- * by users as 'cpu [-h system] [-c cmd args ...]'
- */
- #include <u.h>
- #include <libc.h>
- #include <bio.h>
- #include <auth.h>
- #include <fcall.h>
- #include <libsec.h>
- #define Maxfdata 8192
- #define MaxStr 128
- void remoteside(int);
- void fatal(int, char*, ...);
- void lclnoteproc(int);
- void rmtnoteproc(void);
- void catcher(void*, char*);
- void usage(void);
- void writestr(int, char*, char*, int);
- int readstr(int, char*, int);
- char *rexcall(int*, char*, char*);
- int setamalg(char*);
- char *keyspec = "";
- int notechan;
- int exportpid;
- char *system;
- int cflag;
- int dbg;
- char *user;
- char *patternfile;
- char *srvname = "ncpu";
- char *exportfs = "/bin/exportfs";
- char *ealgs = "rc4_256 sha1";
- /* message size for exportfs; may be larger so we can do big graphics in CPU window */
- int msgsize = Maxfdata+IOHDRSZ;
- /* authentication mechanisms */
- static int netkeyauth(int);
- static int netkeysrvauth(int, char*);
- static int p9auth(int);
- static int srvp9auth(int, char*);
- static int noauth(int);
- static int srvnoauth(int, char*);
- typedef struct AuthMethod AuthMethod;
- struct AuthMethod {
- char *name; /* name of method */
- int (*cf)(int); /* client side authentication */
- int (*sf)(int, char*); /* server side authentication */
- } authmethod[] =
- {
- { "p9", p9auth, srvp9auth,},
- { "netkey", netkeyauth, netkeysrvauth,},
- // { "none", noauth, srvnoauth,},
- { nil, nil}
- };
- AuthMethod *am = authmethod; /* default is p9 */
- char *p9authproto = "p9any";
- int setam(char*);
- void
- usage(void)
- {
- fprint(2, "usage: cpu [-h system] [-u user] [-a authmethod] [-e 'crypt hash'] [-k keypattern] [-P patternfile] [-c cmd args ...]\n");
- exits("usage");
- }
- void
- main(int argc, char **argv)
- {
- char dat[MaxStr], buf[MaxStr], cmd[MaxStr], *p, *err;
- int ac, fd, ms, data;
- char *av[10];
- /* see if we should use a larger message size */
- fd = open("/dev/draw", OREAD);
- if(fd > 0){
- ms = iounit(fd);
- if(msgsize < ms+IOHDRSZ)
- msgsize = ms+IOHDRSZ;
- close(fd);
- }
- user = getuser();
- if(user == nil)
- fatal(1, "can't read user name");
- ARGBEGIN{
- case 'a':
- p = EARGF(usage());
- if(setam(p) < 0)
- fatal(0, "unknown auth method %s", p);
- break;
- case 'e':
- ealgs = EARGF(usage());
- if(*ealgs == 0 || strcmp(ealgs, "clear") == 0)
- ealgs = nil;
- break;
- case 'd':
- dbg++;
- break;
- case 'f':
- /* ignored but accepted for compatibility */
- break;
- case 'O':
- p9authproto = "p9sk2";
- remoteside(1); /* From listen */
- break;
- case 'R': /* From listen */
- remoteside(0);
- break;
- case 'h':
- system = EARGF(usage());
- break;
- case 'c':
- cflag++;
- cmd[0] = '!';
- cmd[1] = '\0';
- while(p = ARGF()) {
- strcat(cmd, " ");
- strcat(cmd, p);
- }
- break;
- case 'k':
- keyspec = EARGF(usage());
- break;
- case 'P':
- patternfile = EARGF(usage());
- break;
- case 'u':
- user = EARGF(usage());
- keyspec = smprint("user=%s", user);
- break;
- default:
- usage();
- }ARGEND;
- if(argc != 0)
- usage();
- if(system == nil) {
- p = getenv("cpu");
- if(p == 0)
- fatal(0, "set $cpu");
- system = p;
- }
- if(err = rexcall(&data, system, srvname))
- fatal(1, "%s: %s", err, system);
- /* Tell the remote side the command to execute and where our working directory is */
- if(cflag)
- writestr(data, cmd, "command", 0);
- if(getwd(dat, sizeof(dat)) == 0)
- writestr(data, "NO", "dir", 0);
- else
- writestr(data, dat, "dir", 0);
- /* start up a process to pass along notes */
- lclnoteproc(data);
- /*
- * Wait for the other end to execute and start our file service
- * of /mnt/term
- */
- if(readstr(data, buf, sizeof(buf)) < 0)
- fatal(1, "waiting for FS: %r");
- if(strncmp("FS", buf, 2) != 0) {
- print("remote cpu: %s", buf);
- exits(buf);
- }
- /* Begin serving the gnot namespace */
- close(0);
- dup(data, 0);
- close(data);
- sprint(buf, "%d", msgsize);
- ac = 0;
- av[ac++] = exportfs;
- av[ac++] = "-m";
- av[ac++] = buf;
- if(dbg)
- av[ac++] = "-d";
- if(patternfile != nil){
- av[ac++] = "-P";
- av[ac++] = patternfile;
- }
- av[ac] = nil;
- exec(exportfs, av);
- fatal(1, "starting exportfs");
- }
- void
- fatal(int syserr, char *fmt, ...)
- {
- Fmt f;
- char *str;
- va_list arg;
- fmtstrinit(&f);
- fmtprint(&f, "cpu: ");
- va_start(arg, fmt);
- fmtvprint(&f, fmt, arg);
- va_end(arg);
- if(syserr)
- fmtprint(&f, ": %r");
- fmtprint(&f, "\n");
- str = fmtstrflush(&f);
- write(2, str, strlen(str));
- exits(str);
- }
- char *negstr = "negotiating authentication method";
- char bug[256];
- int
- old9p(int fd)
- {
- int p[2];
- if(pipe(p) < 0)
- fatal(1, "pipe");
- switch(rfork(RFPROC|RFFDG|RFNAMEG)) {
- case -1:
- fatal(1, "rfork srvold9p");
- case 0:
- if(fd != 1){
- dup(fd, 1);
- close(fd);
- }
- if(p[0] != 0){
- dup(p[0], 0);
- close(p[0]);
- }
- close(p[1]);
- if(0){
- fd = open("/sys/log/cpu", OWRITE);
- if(fd != 2){
- dup(fd, 2);
- close(fd);
- }
- execl("/bin/srvold9p", "srvold9p", "-ds", nil);
- } else
- execl("/bin/srvold9p", "srvold9p", "-s", nil);
- fatal(1, "exec srvold9p");
- default:
- close(fd);
- close(p[0]);
- }
- return p[1];
- }
- /* Invoked with stdin, stdout and stderr connected to the network connection */
- void
- remoteside(int old)
- {
- char user[MaxStr], home[MaxStr], buf[MaxStr], xdir[MaxStr], cmd[MaxStr];
- int i, n, fd, badchdir, gotcmd;
- rfork(RFENVG);
- putenv("service", "cpu");
- fd = 0;
- /* negotiate authentication mechanism */
- n = readstr(fd, cmd, sizeof(cmd));
- if(n < 0)
- fatal(1, "authenticating");
- if(setamalg(cmd) < 0){
- writestr(fd, "unsupported auth method", nil, 0);
- fatal(1, "bad auth method %s", cmd);
- } else
- writestr(fd, "", "", 1);
- fd = (*am->sf)(fd, user);
- if(fd < 0)
- fatal(1, "srvauth");
- /* Set environment values for the user */
- putenv("user", user);
- sprint(home, "/usr/%s", user);
- putenv("home", home);
- /* Now collect invoking cpu's current directory or possibly a command */
- gotcmd = 0;
- if(readstr(fd, xdir, sizeof(xdir)) < 0)
- fatal(1, "dir/cmd");
- if(xdir[0] == '!') {
- strcpy(cmd, &xdir[1]);
- gotcmd = 1;
- if(readstr(fd, xdir, sizeof(xdir)) < 0)
- fatal(1, "dir");
- }
- /* Establish the new process at the current working directory of the
- * gnot */
- badchdir = 0;
- if(strcmp(xdir, "NO") == 0)
- chdir(home);
- else if(chdir(xdir) < 0) {
- badchdir = 1;
- chdir(home);
- }
- /* Start the gnot serving its namespace */
- writestr(fd, "FS", "FS", 0);
- writestr(fd, "/", "exportfs dir", 0);
- n = read(fd, buf, sizeof(buf));
- if(n != 2 || buf[0] != 'O' || buf[1] != 'K')
- exits("remote tree");
- if(old)
- fd = old9p(fd);
- /* make sure buffers are big by doing fversion explicitly; pick a huge number; other side will trim */
- strcpy(buf, VERSION9P);
- if(fversion(fd, 64*1024, buf, sizeof buf) < 0)
- exits("fversion failed");
- if(mount(fd, -1, "/mnt/term", MCREATE|MREPL, "") < 0)
- exits("mount failed");
- close(fd);
- /* the remote noteproc uses the mount so it must follow it */
- rmtnoteproc();
- for(i = 0; i < 3; i++)
- close(i);
- if(open("/mnt/term/dev/cons", OREAD) != 0)
- exits("open stdin");
- if(open("/mnt/term/dev/cons", OWRITE) != 1)
- exits("open stdout");
- dup(1, 2);
- if(badchdir)
- print("cpu: failed to chdir to '%s'\n", xdir);
- if(gotcmd)
- execl("/bin/rc", "rc", "-lc", cmd, nil);
- else
- execl("/bin/rc", "rc", "-li", nil);
- fatal(1, "exec shell");
- }
- char*
- rexcall(int *fd, char *host, char *service)
- {
- char *na;
- char dir[MaxStr];
- char err[ERRMAX];
- char msg[MaxStr];
- int n;
- na = netmkaddr(host, 0, service);
- if((*fd = dial(na, 0, dir, 0)) < 0)
- return "can't dial";
- /* negotiate authentication mechanism */
- if(ealgs != nil)
- snprint(msg, sizeof(msg), "%s %s", am->name, ealgs);
- else
- snprint(msg, sizeof(msg), "%s", am->name);
- writestr(*fd, msg, negstr, 0);
- n = readstr(*fd, err, sizeof err);
- if(n < 0)
- return negstr;
- if(*err){
- werrstr(err);
- return negstr;
- }
- /* authenticate */
- *fd = (*am->cf)(*fd);
- if(*fd < 0)
- return "can't authenticate";
- return 0;
- }
- void
- writestr(int fd, char *str, char *thing, int ignore)
- {
- int l, n;
- l = strlen(str);
- n = write(fd, str, l+1);
- if(!ignore && n < 0)
- fatal(1, "writing network: %s", thing);
- }
- int
- readstr(int fd, char *str, int len)
- {
- int n;
- while(len) {
- n = read(fd, str, 1);
- if(n < 0)
- return -1;
- if(*str == '\0')
- return 0;
- str++;
- len--;
- }
- return -1;
- }
- static int
- readln(char *buf, int n)
- {
- int i;
- char *p;
- n--; /* room for \0 */
- p = buf;
- for(i=0; i<n; i++){
- if(read(0, p, 1) != 1)
- break;
- if(*p == '\n' || *p == '\r')
- break;
- p++;
- }
- *p = '\0';
- return p-buf;
- }
- /*
- * user level challenge/response
- */
- static int
- netkeyauth(int fd)
- {
- char chall[32];
- char resp[32];
- strecpy(chall, chall+sizeof chall, getuser());
- print("user[%s]: ", chall);
- if(readln(resp, sizeof(resp)) < 0)
- return -1;
- if(*resp != 0)
- strcpy(chall, resp);
- writestr(fd, chall, "challenge/response", 1);
- for(;;){
- if(readstr(fd, chall, sizeof chall) < 0)
- break;
- if(*chall == 0)
- return fd;
- print("challenge: %s\nresponse: ", chall);
- if(readln(resp, sizeof(resp)) < 0)
- break;
- writestr(fd, resp, "challenge/response", 1);
- }
- return -1;
- }
- static int
- netkeysrvauth(int fd, char *user)
- {
- char response[32];
- Chalstate *ch;
- int tries;
- AuthInfo *ai;
- if(readstr(fd, user, 32) < 0)
- return -1;
- ai = nil;
- ch = nil;
- for(tries = 0; tries < 10; tries++){
- if((ch = auth_challenge("proto=p9cr role=server user=%q", user)) == nil)
- return -1;
- writestr(fd, ch->chal, "challenge", 1);
- if(readstr(fd, response, sizeof response) < 0)
- return -1;
- ch->resp = response;
- ch->nresp = strlen(response);
- if((ai = auth_response(ch)) != nil)
- break;
- }
- auth_freechal(ch);
- if(ai == nil)
- return -1;
- writestr(fd, "", "challenge", 1);
- if(auth_chuid(ai, 0) < 0)
- fatal(1, "newns");
- auth_freeAI(ai);
- return fd;
- }
- static void
- mksecret(char *t, uchar *f)
- {
- sprint(t, "%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux%2.2ux",
- f[0], f[1], f[2], f[3], f[4], f[5], f[6], f[7], f[8], f[9]);
- }
- /*
- * plan9 authentication followed by rc4 encryption
- */
- static int
- p9auth(int fd)
- {
- uchar key[16];
- uchar digest[SHA1dlen];
- char fromclientsecret[21];
- char fromserversecret[21];
- int i;
- AuthInfo *ai;
- ai = auth_proxy(fd, auth_getkey, "proto=%q role=client %s", p9authproto, keyspec);
- if(ai == nil)
- return -1;
- memmove(key+4, ai->secret, ai->nsecret);
- if(ealgs == nil)
- return fd;
- /* exchange random numbers */
- srand(truerand());
- for(i = 0; i < 4; i++)
- key[i] = rand();
- if(write(fd, key, 4) != 4)
- return -1;
- if(readn(fd, key+12, 4) != 4)
- return -1;
- /* scramble into two secrets */
- sha1(key, sizeof(key), digest, nil);
- mksecret(fromclientsecret, digest);
- mksecret(fromserversecret, digest+10);
- /* set up encryption */
- i = pushssl(fd, ealgs, fromclientsecret, fromserversecret, nil);
- if(i < 0)
- werrstr("can't establish ssl connection: %r");
- return i;
- }
- static int
- noauth(int fd)
- {
- ealgs = nil;
- return fd;
- }
- static int
- srvnoauth(int fd, char *user)
- {
- strecpy(user, user+MaxStr, getuser());
- ealgs = nil;
- return fd;
- }
- void
- loghex(uchar *p, int n)
- {
- char buf[100];
- int i;
- for(i = 0; i < n; i++)
- sprint(buf+2*i, "%2.2ux", p[i]);
- syslog(0, "cpu", buf);
- }
- static int
- srvp9auth(int fd, char *user)
- {
- uchar key[16];
- uchar digest[SHA1dlen];
- char fromclientsecret[21];
- char fromserversecret[21];
- int i;
- AuthInfo *ai;
- ai = auth_proxy(0, nil, "proto=%q role=server %s", p9authproto, keyspec);
- if(ai == nil)
- return -1;
- if(auth_chuid(ai, nil) < 0)
- return -1;
- strecpy(user, user+MaxStr, ai->cuid);
- memmove(key+4, ai->secret, ai->nsecret);
- if(ealgs == nil)
- return fd;
- /* exchange random numbers */
- srand(truerand());
- for(i = 0; i < 4; i++)
- key[i+12] = rand();
- if(readn(fd, key, 4) != 4)
- return -1;
- if(write(fd, key+12, 4) != 4)
- return -1;
- /* scramble into two secrets */
- sha1(key, sizeof(key), digest, nil);
- mksecret(fromclientsecret, digest);
- mksecret(fromserversecret, digest+10);
- /* set up encryption */
- i = pushssl(fd, ealgs, fromserversecret, fromclientsecret, nil);
- if(i < 0)
- werrstr("can't establish ssl connection: %r");
- return i;
- }
- /*
- * set authentication mechanism
- */
- int
- setam(char *name)
- {
- for(am = authmethod; am->name != nil; am++)
- if(strcmp(am->name, name) == 0)
- return 0;
- am = authmethod;
- return -1;
- }
- /*
- * set authentication mechanism and encryption/hash algs
- */
- int
- setamalg(char *s)
- {
- ealgs = strchr(s, ' ');
- if(ealgs != nil)
- *ealgs++ = 0;
- return setam(s);
- }
- char *rmtnotefile = "/mnt/term/dev/cpunote";
- /*
- * loop reading /mnt/term/dev/note looking for notes.
- * The child returns to start the shell.
- */
- void
- rmtnoteproc(void)
- {
- int n, fd, pid, notepid;
- char buf[256];
- /* new proc returns to start shell */
- pid = rfork(RFPROC|RFFDG|RFNOTEG|RFNAMEG|RFMEM);
- switch(pid){
- case -1:
- syslog(0, "cpu", "cpu -R: can't start noteproc: %r");
- return;
- case 0:
- return;
- }
- /* new proc reads notes from other side and posts them to shell */
- switch(notepid = rfork(RFPROC|RFFDG|RFMEM)){
- case -1:
- syslog(0, "cpu", "cpu -R: can't start wait proc: %r");
- _exits(0);
- case 0:
- fd = open(rmtnotefile, OREAD);
- if(fd < 0){
- syslog(0, "cpu", "cpu -R: can't open %s", rmtnotefile);
- _exits(0);
- }
-
- for(;;){
- n = read(fd, buf, sizeof(buf)-1);
- if(n <= 0){
- postnote(PNGROUP, pid, "hangup");
- _exits(0);
- }
- buf[n] = 0;
- postnote(PNGROUP, pid, buf);
- }
- }
- /* original proc waits for shell proc to die and kills note proc */
- for(;;){
- n = waitpid();
- if(n < 0 || n == pid)
- break;
- }
- postnote(PNPROC, notepid, "kill");
- _exits(0);
- }
- enum
- {
- Qdir,
- Qcpunote,
- Nfid = 32,
- };
- struct {
- char *name;
- Qid qid;
- ulong perm;
- } fstab[] =
- {
- [Qdir] { ".", {Qdir, 0, QTDIR}, DMDIR|0555 },
- [Qcpunote] { "cpunote", {Qcpunote, 0}, 0444 },
- };
- typedef struct Note Note;
- struct Note
- {
- Note *next;
- char msg[ERRMAX];
- };
- typedef struct Request Request;
- struct Request
- {
- Request *next;
- Fcall f;
- };
- typedef struct Fid Fid;
- struct Fid
- {
- int fid;
- int file;
- int omode;
- };
- Fid fids[Nfid];
- struct {
- Lock;
- Note *nfirst, *nlast;
- Request *rfirst, *rlast;
- } nfs;
- int
- fsreply(int fd, Fcall *f)
- {
- uchar buf[IOHDRSZ+Maxfdata];
- int n;
- if(dbg)
- fprint(2, "<-%F\n", f);
- n = convS2M(f, buf, sizeof buf);
- if(n > 0){
- if(write(fd, buf, n) != n){
- close(fd);
- return -1;
- }
- }
- return 0;
- }
- /* match a note read request with a note, reply to the request */
- int
- kick(int fd)
- {
- Request *rp;
- Note *np;
- int rv;
- for(;;){
- lock(&nfs);
- rp = nfs.rfirst;
- np = nfs.nfirst;
- if(rp == nil || np == nil){
- unlock(&nfs);
- break;
- }
- nfs.rfirst = rp->next;
- nfs.nfirst = np->next;
- unlock(&nfs);
- rp->f.type = Rread;
- rp->f.count = strlen(np->msg);
- rp->f.data = np->msg;
- rv = fsreply(fd, &rp->f);
- free(rp);
- free(np);
- if(rv < 0)
- return -1;
- }
- return 0;
- }
- void
- flushreq(int tag)
- {
- Request **l, *rp;
- lock(&nfs);
- for(l = &nfs.rfirst; *l != nil; l = &(*l)->next){
- rp = *l;
- if(rp->f.tag == tag){
- *l = rp->next;
- unlock(&nfs);
- free(rp);
- return;
- }
- }
- unlock(&nfs);
- }
- Fid*
- getfid(int fid)
- {
- int i, freefid;
- freefid = -1;
- for(i = 0; i < Nfid; i++){
- if(freefid < 0 && fids[i].file < 0)
- freefid = i;
- if(fids[i].fid == fid)
- return &fids[i];
- }
- if(freefid >= 0){
- fids[freefid].fid = fid;
- return &fids[freefid];
- }
- return nil;
- }
- int
- fsstat(int fd, Fid *fid, Fcall *f)
- {
- Dir d;
- uchar statbuf[256];
- memset(&d, 0, sizeof(d));
- d.name = fstab[fid->file].name;
- d.uid = user;
- d.gid = user;
- d.muid = user;
- d.qid = fstab[fid->file].qid;
- d.mode = fstab[fid->file].perm;
- d.atime = d.mtime = time(0);
- f->stat = statbuf;
- f->nstat = convD2M(&d, statbuf, sizeof statbuf);
- return fsreply(fd, f);
- }
- int
- fsread(int fd, Fid *fid, Fcall *f)
- {
- Dir d;
- uchar buf[256];
- Request *rp;
- switch(fid->file){
- default:
- return -1;
- case Qdir:
- if(f->offset == 0 && f->count >0){
- memset(&d, 0, sizeof(d));
- d.name = fstab[Qcpunote].name;
- d.uid = user;
- d.gid = user;
- d.muid = user;
- d.qid = fstab[Qcpunote].qid;
- d.mode = fstab[Qcpunote].perm;
- d.atime = d.mtime = time(0);
- f->count = convD2M(&d, buf, sizeof buf);
- f->data = (char*)buf;
- } else
- f->count = 0;
- return fsreply(fd, f);
- case Qcpunote:
- rp = mallocz(sizeof(*rp), 1);
- if(rp == nil)
- return -1;
- rp->f = *f;
- lock(&nfs);
- if(nfs.rfirst == nil)
- nfs.rfirst = rp;
- else
- nfs.rlast->next = rp;
- nfs.rlast = rp;
- unlock(&nfs);
- return kick(fd);;
- }
- }
- char Eperm[] = "permission denied";
- char Enofile[] = "out of files";
- char Enotdir[] = "not a directory";
- void
- notefs(int fd)
- {
- uchar buf[IOHDRSZ+Maxfdata];
- int i, j, n, ncpunote;
- char err[ERRMAX];
- Fcall f;
- Fid *fid, *nfid;
- int doreply;
- rfork(RFNOTEG);
- fmtinstall('F', fcallfmt);
- for(n = 0; n < Nfid; n++){
- fids[n].file = -1;
- fids[n].omode = -1;
- }
- ncpunote = 0;
- for(;;){
- n = read9pmsg(fd, buf, sizeof(buf));
- if(n <= 0){
- if(dbg)
- fprint(2, "read9pmsg(%d) returns %d: %r\n", fd, n);
- break;
- }
- if(convM2S(buf, n, &f) <= BIT16SZ)
- break;
- if(dbg)
- fprint(2, "->%F\n", &f);
- doreply = 1;
- fid = getfid(f.fid);
- if(fid == nil){
- nofids:
- f.type = Rerror;
- f.ename = Enofile;
- fsreply(fd, &f);
- continue;
- }
- switch(f.type++){
- default:
- f.type = Rerror;
- f.ename = "unknown type";
- break;
- case Tflush:
- flushreq(f.oldtag);
- break;
- case Tversion:
- if(f.msize > IOHDRSZ+Maxfdata)
- f.msize = IOHDRSZ+Maxfdata;
- break;
- case Tauth:
- f.type = Rerror;
- f.ename = "cpu: authentication not required";
- break;
- case Tattach:
- f.qid = fstab[Qdir].qid;
- fid->file = Qdir;
- break;
- case Twalk:
- nfid = nil;
- if(f.newfid != f.fid){
- nfid = getfid(f.newfid);
- if(nfid == nil)
- goto nofids;
- nfid->file = fid->file;
- fid = nfid;
- }
- f.ename = nil;
- for(i=0; i<f.nwname; i++){
- if(i > MAXWELEM){
- f.type = Rerror;
- f.ename = "too many name elements";
- break;
- }
- if(fid->file != Qdir){
- f.type = Rerror;
- f.ename = Enotdir;
- break;
- }
- if(strcmp(f.wname[i], "cpunote") == 0){
- fid->file = Qcpunote;
- f.wqid[i] = fstab[Qcpunote].qid;
- continue;
- }
- f.type = Rerror;
- f.ename = err;
- strcpy(err, "cpu: file \"");
- for(j=0; j<=i; j++){
- if(strlen(err)+1+strlen(f.wname[j])+32 > sizeof err)
- break;
- if(j != 0)
- strcat(err, "/");
- strcat(err, f.wname[j]);
- }
- strcat(err, "\" does not exist");
- break;
- }
- if(nfid != nil && (f.ename != nil || i < f.nwname))
- nfid ->file = -1;
- if(f.type != Rerror)
- f.nwqid = i;
- break;
- case Topen:
- if(f.mode != OREAD){
- f.type = Rerror;
- f.ename = Eperm;
- }
- fid->omode = f.mode;
- if(fid->file == Qcpunote)
- ncpunote++;
- f.qid = fstab[fid->file].qid;
- break;
- case Tcreate:
- f.type = Rerror;
- f.ename = Eperm;
- break;
- case Tread:
- if(fsread(fd, fid, &f) < 0)
- goto err;
- doreply = 0;
- break;
- case Twrite:
- f.type = Rerror;
- f.ename = Eperm;
- break;
- case Tclunk:
- if(fid->omode != -1 && fid->file == Qcpunote){
- ncpunote--;
- if(ncpunote == 0) /* remote side is done */
- goto err;
- }
- fid->file = -1;
- fid->omode = -1;
- break;
- case Tremove:
- f.type = Rerror;
- f.ename = Eperm;
- break;
- case Tstat:
- if(fsstat(fd, fid, &f) < 0)
- goto err;
- doreply = 0;
- break;
- case Twstat:
- f.type = Rerror;
- f.ename = Eperm;
- break;
- }
- if(doreply)
- if(fsreply(fd, &f) < 0)
- break;
- }
- err:
- if(dbg)
- fprint(2, "notefs exiting: %r\n");
- werrstr("success");
- postnote(PNGROUP, exportpid, "kill");
- if(dbg)
- fprint(2, "postnote PNGROUP %d: %r\n", exportpid);
- close(fd);
- }
- char notebuf[ERRMAX];
- void
- catcher(void*, char *text)
- {
- int n;
- n = strlen(text);
- if(n >= sizeof(notebuf))
- n = sizeof(notebuf)-1;
- memmove(notebuf, text, n);
- notebuf[n] = '\0';
- noted(NCONT);
- }
- /*
- * mount in /dev a note file for the remote side to read.
- */
- void
- lclnoteproc(int netfd)
- {
- Waitmsg *w;
- Note *np;
- int pfd[2];
- int pid;
- if(pipe(pfd) < 0){
- fprint(2, "cpu: can't start note proc: pipe: %r\n");
- return;
- }
- /* new proc mounts and returns to start exportfs */
- switch(pid = rfork(RFPROC|RFNAMEG|RFFDG|RFMEM)){
- default:
- exportpid = pid;
- break;
- case -1:
- fprint(2, "cpu: can't start note proc: rfork: %r\n");
- return;
- case 0:
- close(pfd[0]);
- if(mount(pfd[1], -1, "/dev", MBEFORE, "") < 0)
- fprint(2, "cpu: can't mount note proc: %r\n");
- close(pfd[1]);
- return;
- }
- close(netfd);
- close(pfd[1]);
- /* new proc listens for note file system rpc's */
- switch(rfork(RFPROC|RFNAMEG|RFMEM)){
- case -1:
- fprint(2, "cpu: can't start note proc: rfork1: %r\n");
- _exits(0);
- case 0:
- notefs(pfd[0]);
- _exits(0);
- }
- /* original proc waits for notes */
- notify(catcher);
- w = nil;
- for(;;) {
- *notebuf = 0;
- free(w);
- w = wait();
- if(w == nil) {
- if(*notebuf == 0)
- break;
- np = mallocz(sizeof(Note), 1);
- if(np != nil){
- strcpy(np->msg, notebuf);
- lock(&nfs);
- if(nfs.nfirst == nil)
- nfs.nfirst = np;
- else
- nfs.nlast->next = np;
- nfs.nlast = np;
- unlock(&nfs);
- kick(pfd[0]);
- }
- unlock(&nfs);
- } else if(w->pid == exportpid)
- break;
- }
- if(w == nil)
- exits(nil);
- exits(0);
- /* exits(w->msg); */
- }
|