diff options
Diffstat (limited to 'utils/acid/proc.c')
| -rw-r--r-- | utils/acid/proc.c | 274 |
1 files changed, 274 insertions, 0 deletions
diff --git a/utils/acid/proc.c b/utils/acid/proc.c new file mode 100644 index 00000000..f13dbf95 --- /dev/null +++ b/utils/acid/proc.c @@ -0,0 +1,274 @@ +#include <lib9.h> +#include <bio.h> +#include <ctype.h> +#include "mach.h" +#define Extern extern +#include "acid.h" +#include "y.tab.h" + +void +nocore(void) +{ + int i; + + if(cormap == 0) + return; + + for (i = 0; i < cormap->nsegs; i++) + if (cormap->seg[i].mget == 0 && cormap->seg[i].inuse && cormap->seg[i].fd >= 0) + close(cormap->seg[i].fd); + free(cormap); + cormap = 0; +} + +void +sproc(int pid) +{ + Lsym *s; + char buf[64]; + ulong proctab; + int fd, i, fcor; + + if(symmap == 0) + error("no map"); + + if(rdebug) { + fcor = -1; + proctab = 0; + i = remoteio(pid, "proc", buf, sizeof(buf)); + if(i >= 0) { + buf[i] = '\0'; + proctab = strtoul(buf, 0, 16); + } else + error("can't access pid %d: %r", pid); + s = look("proc"); + if(s != 0) + s->v->vstore.u0.sival = proctab; + } else { + sprint(buf, "/proc/%d/mem", pid); + fcor = open(buf, ORDWR); + if(fcor < 0) + error("setproc: open %s: %r", buf); + + checkqid(symmap->seg[0].fd, pid); + + if(kernel) { + proctab = 0; + sprint(buf, "/proc/%d/proc", pid); + fd = open(buf, OREAD); + if(fd >= 0) { + i = read(fd, buf, sizeof(buf)); + if(i >= 0) { + buf[i] = '\0'; + proctab = strtoul(buf, 0, 0); + } + close(fd); + } + s = look("proc"); + if(s != 0) + s->v->vstore.u0.sival = proctab; + } + } + + s = look("pid"); + s->v->vstore.u0.sival = pid; + + nocore(); + if(rdebug) { + cormap = attachremt(remfd, &fhdr); + for(i = 0; i < cormap->nsegs; i++) + setmapio(cormap, i, remget, remput); + } else + cormap = attachproc(pid, kernel, fcor, &fhdr); + if (cormap == 0) + error("setproc: cant make coremap"); + i = findseg(cormap, "text"); + if (i >= 0) + cormap->seg[i].name = "*text"; + i = findseg(cormap, "data"); + if (i >= 0) + cormap->seg[i].name = "*data"; + install(pid); +} + +void +notes(int pid) +{ + Lsym *s; + Value *v; + int i, fd; + char buf[128]; + List *l, **tail; + + s = look("notes"); + if(s == 0) + return; + v = s->v; + + if(!rdebug) { + sprint(buf, "/proc/%d/note", pid); + fd = open(buf, OREAD); + if(fd < 0) + error("pid=%d: open note: %r", pid); + } else + fd = -1; + + v->set = 1; + v->type = TLIST; + v->vstore.u0.sl = 0; + tail = &v->vstore.u0.sl; + for(;;) { + if(rdebug) + i = remoteio(pid, "note", buf, sizeof(buf)); + else + i = read(fd, buf, sizeof(buf)); + if(i <= 0) + break; + buf[i] = '\0'; + l = al(TSTRING); + l->lstore.u0.sstring = strnode(buf); + l->lstore.fmt = 's'; + *tail = l; + tail = &l->next; + } + if(fd >= 0) + close(fd); +} + +void +dostop(int pid) +{ + Lsym *s; + Node *np, *p; + + s = look("stopped"); + if(s && s->proc) { + np = an(ONAME, ZN, ZN); + np->sym = s; + np->nstore.fmt = 'D'; + np->type = TINT; + p = con(pid); + p->nstore.fmt = 'D'; + np = an(OCALL, np, p); + execute(np); + } +} + +void +install(int pid) +{ + Lsym *s; + List *l; + char buf[128]; + int i, fd, new, p; + + new = -1; + for(i = 0; i < Maxproc; i++) { + p = ptab[i].pid; + if(p == pid) + return; + if(p == 0 && new == -1) + new = i; + } + if(new == -1) + error("no free process slots"); + + if(!rdebug) { + sprint(buf, "/proc/%d/ctl", pid); + fd = open(buf, OWRITE); + if(fd < 0) + error("pid=%d: open ctl: %r", pid); + } else + fd = -1; + ptab[new].pid = pid; + ptab[new].ctl = fd; + + s = look("proclist"); + l = al(TINT); + l->lstore.fmt = 'D'; + l->lstore.u0.sival = pid; + l->next = s->v->vstore.u0.sl; + s->v->vstore.u0.sl = l; + s->v->set = 1; +} + +void +deinstall(int pid) +{ + int i; + Lsym *s; + List *f, **d; + + for(i = 0; i < Maxproc; i++) { + if(ptab[i].pid == pid) { + if(ptab[i].ctl >= 0) + close(ptab[i].ctl); + ptab[i].pid = 0; + s = look("proclist"); + d = &s->v->vstore.u0.sl; + for(f = *d; f; f = f->next) { + if(f->lstore.u0.sival == pid) { + *d = f->next; + break; + } + } + s = look("pid"); + if(s->v->vstore.u0.sival == pid) + s->v->vstore.u0.sival = 0; + return; + } + } +} + +void +msg(int pid, char *msg) +{ + int i; + int l; + int ok; + char err[ERRMAX]; + + for(i = 0; i < Maxproc; i++) { + if(ptab[i].pid == pid) { + l = strlen(msg); + if(rdebug) + ok = sendremote(pid, msg) >= 0; + else + ok = write(ptab[i].ctl, msg, l) == l; + if(!ok) { + errstr(err, sizeof err); + if(strcmp(err, "process exited") == 0) + deinstall(pid); + error("msg: pid=%d %s: %s", pid, msg, err); + } + return; + } + } + error("msg: pid=%d: not found for %s", pid, msg); +} + +char * +getstatus(int pid) +{ + int fd; + char *p; + + static char buf[128]; + + if(rdebug) { + if(remoteio(pid, "status", buf, sizeof(buf)) < 0) + error("remote status: pid %d: %r", pid); + return buf; + } + sprint(buf, "/proc/%d/status", pid); + fd = open(buf, OREAD); + if(fd < 0) + error("open %s: %r", buf); + read(fd, buf, sizeof(buf)); + close(fd); + p = buf+56+12; /* Do better! */ + while(*p == ' ') + p--; + p[1] = '\0'; + return buf+56; /* ditto */ +} |