rsa: rename getkey() to getrsakey(), document rsa2csr in rsa(8)

[plan9front.git] / sys / src / cmd / ratrace.c

diff --git a/sys/src/cmd/ratrace.c b/sys/src/cmd/ratrace.c
index ff5084491784d52de23b13cde1ad4c6b6a6c542c..95c50933c193f2da534b78ad69bc2e16d9e37b82 100644 (file)
--- a/sys/src/cmd/ratrace.c
+++ b/sys/src/cmd/ratrace.c
@@ -1,62 +1,77 @@
 #include <u.h>
 #include <libc.h>
-#include <bio.h>
 #include <thread.h>
 
 enum {
        Stacksize       = 8*1024,
-       Bufsize         = 8*1024,
 };
 
 Channel *out;
 Channel *quit;
 Channel *forkc;
-int nread = 0;
+int readers = 1;
 
-typedef struct Str Str;
-struct Str {
-       char    *buf;
-       int     len;
+typedef struct Msg Msg;
+struct Msg {
+       int     pid;
+       char    buf[8*1024];
+};
+
+typedef struct Reader Reader;
+struct Reader {
+       int     pid;
+
+       int     tfd;
+       int     cfd;
+
+       Msg*    msg;
 };
 
 void
-die(char *s)
+die(Reader *r)
 {
-       fprint(2, "%s\n", s);
-       exits(s);
+       Msg *s;
+
+       s = r->msg;
+       snprint(s->buf, sizeof(s->buf), " = %r\n");
+       s->pid = r->pid;
+       sendp(quit, s);
+       if(r->tfd >= 0)
+               close(r->tfd);
+       if(r->cfd >= 0)
+               close(r->cfd);
+       threadexits(nil);
 }
 
 void
-cwrite(int fd, char *path, char *cmd, int len)
+cwrite(Reader *r, char *cmd)
 {
-       if (write(fd, cmd, len) < len) {
-               fprint(2, "cwrite: %s: failed %d bytes: %r\n", path, len);
-               sendp(quit, nil);
-               threadexits(nil);
-       }
+       if (write(r->cfd, cmd, strlen(cmd)) < 0)
+               die(r);
 }
 
 void
 reader(void *v)
 {
-       int cfd, tfd, forking = 0, pid, newpid;
-       char *ctl, *truss;
-       Str *s;
-
-       pid = (int)(uintptr)v;
-       ctl = smprint("/proc/%d/ctl", pid);
-       if ((cfd = open(ctl, OWRITE)) < 0)
-               die(smprint("%s: %r", ctl));
-       truss = smprint("/proc/%d/syscall", pid);
-       if ((tfd = open(truss, OREAD)) < 0)
-               die(smprint("%s: %r", truss));
-
-       cwrite(cfd, ctl, "stop", 4);
-       cwrite(cfd, truss, "startsyscall", 12);
-
-       s = mallocz(sizeof(Str) + Bufsize, 1);
-       s->buf = (char *)&s[1];
-       while((s->len = pread(tfd, s->buf, Bufsize - 1, 0)) > 0){
+       int forking = 0, newpid;
+       Reader r;
+       Msg *s;
+
+       r.pid = (int)(uintptr)v;
+       r.tfd = r.cfd = -1;
+
+       r.msg = s = mallocz(sizeof(Msg), 1);
+       snprint(s->buf, sizeof(s->buf), "/proc/%d/ctl", r.pid);
+       if ((r.cfd = open(s->buf, OWRITE)) < 0)
+               die(&r);
+       snprint(s->buf, sizeof(s->buf), "/proc/%d/syscall", r.pid);
+       if ((r.tfd = open(s->buf, OREAD)) < 0)
+               die(&r);
+
+       cwrite(&r, "stop");
+       cwrite(&r, "startsyscall");
+
+       while(pread(r.tfd, s->buf, sizeof(s->buf)-1, 0) > 0){
                if (forking && s->buf[1] == '=' && s->buf[3] != '-') {
                        forking = 0;
                        newpid = strtol(&s->buf[3], 0, 0);
@@ -82,59 +97,67 @@ reader(void *v)
                        }
                        free(rf);
                }
+               s->pid = r.pid;
                sendp(out, s);
-               cwrite(cfd, truss, "startsyscall", 12);
-               s = mallocz(sizeof(Str) + Bufsize, 1);
-               s->buf = (char *)&s[1];
+
+               r.msg = s = mallocz(sizeof(Msg), 1);
+               cwrite(&r, "startsyscall");
        }
-       sendp(quit, nil);
-       threadexitsall(nil);
+       die(&r);
 }
 
 void
-writer(void *)
+writer(int lastpid)
 {
-       int newpid;
+       char lastc = -1;
        Alt a[4];
-       Str *s;
+       Msg *s;
+       int n;
 
        a[0].op = CHANRCV;
        a[0].c = quit;
-       a[0].v = nil;
+       a[0].v = &s;
        a[1].op = CHANRCV;
        a[1].c = out;
        a[1].v = &s;
        a[2].op = CHANRCV;
        a[2].c = forkc;
-       a[2].v = &newpid;
+       a[2].v = nil;
        a[3].op = CHANEND;
 
-       for(;;)
+       while(readers > 0){
                switch(alt(a)){
                case 0:
-                       nread--;
-                       if(nread <= 0)
-                               goto done;
-                       break;
+                       readers--;
                case 1:
-                       /* it's a nice null terminated thing */
-                       fprint(2, "%s", s->buf);
+                       if(s->pid != lastpid){
+                               lastpid = s->pid;
+                               if(lastc != '\n'){
+                                       lastc = '\n';
+                                       write(2, &lastc, 1);
+                               }
+                               if(s->buf[1] == '=')
+                                       fprint(2, "%d ...", lastpid);
+                       }
+                       n = strlen(s->buf);
+                       if(n > 0){
+                               write(2, s->buf, n);
+                               lastc = s->buf[n-1];
+                       }
                        free(s);
                        break;
                case 2:
-                       // procrfork(reader, (void*)newpid, Stacksize, 0);
-                       nread++;
+                       readers++;
                        break;
                }
-done:
-       exits(nil);
+       }
 }
 
 void
 usage(void)
 {
        fprint(2, "Usage: ratrace [-c cmd [arg...]] | [pid]\n");
-       exits("usage");
+       threadexits("usage");
 }
 
 void
@@ -180,10 +203,11 @@ threadmain(int argc, char **argv)
                pid = atoi(argv[1]);
        }
 
-       out   = chancreate(sizeof(char*), 0);
-       quit  = chancreate(sizeof(char*), 0);
-       forkc = chancreate(sizeof(ulong *), 0);
-       nread++;
-       procrfork(writer, nil, Stacksize, 0);
-       reader((void*)pid);
+       out   = chancreate(sizeof(Msg*), 0);
+       quit  = chancreate(sizeof(Msg*), 0);
+       forkc = chancreate(sizeof(void*), 0);
+       procrfork(reader, (void*)pid, Stacksize, 0);
+
+       writer(pid);
+       threadexits(nil);
 }