ref: 12e226ac6798412271020d42645f69138715c519
dir: /utils/acid/proc.c/
#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 */ }