ref: 04c4a9ca8b06e34a203f7ddacc2263cf2077e1aa
dir: /sys/src/cmd/acid/main.c/
#include <u.h> #include <libc.h> #include <bio.h> #include <mach.h> #define Extern #include "acid.h" #include "y.tab.h" extern int _ifmt(Fmt*); static Biobuf bioout; static char prog[128]; static char* lm[16]; static int nlm; static char* mtype; static int attachfiles(char*, int); int xfmt(Fmt*); int isnumeric(char*); void die(void); void loadmoduleobjtype(void); void usage(void) { fprint(2, "usage: acid [-kqw] [-l library] [-m machine] [pid] [file]\n"); exits("usage"); } void main(int argc, char *argv[]) { Lsym *l; Node *n; char *s; int pid, i; argv0 = argv[0]; pid = 0; aout = "8.out"; quiet = 1; mtype = 0; ARGBEGIN{ case 'm': mtype = ARGF(); break; case 'w': wtflag = 1; break; case 'l': s = ARGF(); if(s == 0) usage(); lm[nlm++] = s; break; case 'k': kernel++; break; case 'q': quiet = 0; break; default: usage(); }ARGEND if(argc > 0) { if(isnumeric(argv[0])) { pid = strtol(argv[0], 0, 0); snprint(prog, sizeof(prog), "/proc/%d/text", pid); aout = prog; if(argc > 1) aout = argv[1]; else if(kernel) aout = system(); } else { if(kernel) { fprint(2, "acid: -k requires a pid\n"); usage(); } aout = argv[0]; } } fmtinstall('x', xfmt); fmtinstall('L', Lfmt); Binit(&bioout, 1, OWRITE); bout = &bioout; kinit(); initialising = 1; pushfile(0); loadvars(); installbuiltin(); if(mtype && machbyname(mtype) == 0) print("unknown machine %s", mtype); if (attachfiles(aout, pid) < 0) varreg(); /* use default register set on error */ loadmodule("/sys/lib/acid/port"); loadmoduleobjtype(); for(i = 0; i < nlm; i++) { if(access(lm[i], AREAD) >= 0) loadmodule(lm[i]); else { s = smprint("/sys/lib/acid/%s", lm[i]); loadmodule(s); free(s); } } userinit(); varsym(); l = look("acidmap"); if(l && l->proc) { n = an(ONAME, ZN, ZN); n->sym = l; n = an(OCALL, n, ZN); execute(n); } interactive = 1; initialising = 0; line = 1; notify(catcher); for(;;) { if(setjmp(err)) { Binit(&bioout, 1, OWRITE); unwind(); } stacked = 0; Bprint(bout, "acid: "); if(yyparse() != 1) die(); restartio(); unwind(); } /* not reached */ } static int attachfiles(char *aout, int pid) { interactive = 0; if(setjmp(err)) return -1; if(aout) { /* executable given */ if(wtflag) text = open(aout, ORDWR); else text = open(aout, OREAD); if(text < 0) error("%s: can't open %s: %r", argv0, aout); readtext(aout); } if(pid) /* pid given */ sproc(pid); return 0; } void die(void) { Lsym *s; List *f; Bprint(bout, "\n"); if(setjmp(err) == 0) callhook("dying"); s = look("proclist"); if(s && s->v->type == TLIST) { for(f = s->v->l; f; f = f->next) Bprint(bout, "echo kill > /proc/%d/ctl\n", (int)f->ival); } exits(0); } void loadmoduleobjtype(void) { char *buf; buf = smprint("/sys/lib/acid/%s", mach->name); loadmodule(buf); free(buf); } int callhook(char *name) { Lsym *l; Node *n; l = look(name); if(l && l->proc) { n = an(ONAME, ZN, ZN); n->sym = l; n = an(OCALL, n, ZN); execute(n); return 0; } return -1; } void userinit(void) { char *buf, *p; p = getenv("home"); if(p != 0) { buf = smprint("%s/lib/acid", p); silent = 1; loadmodule(buf); free(buf); } interactive = 0; if(setjmp(err)) { unwind(); return; } callhook("acidinit"); } void loadmodule(char *s) { interactive = 0; if(setjmp(err)) { unwind(); return; } pushfile(s); silent = 0; yyparse(); popio(); return; } void readtext(char *s) { Dir *d; Lsym *l; Value *v; uvlong length; Symbol sym; if(mtype != 0){ symmap = newmap(0, 1); if(symmap == 0) print("%s: (error) loadmap: cannot make symbol map\n", argv0); length = 1<<24; d = dirfstat(text); if(d != nil){ length = d->length; free(d); } setmap(symmap, text, 0, length, 0, "binary"); return; } if(!crackhdr(text, &fhdr)) { print("can't decode file header\n"); return; } symmap = loadmap(0, text, &fhdr); if(symmap == 0) print("%s: (error) loadmap: cannot make symbol map\n", argv0); if(syminit(text, &fhdr) < 0) { print("%s: (error) syminit: %r", argv0); return; } print("%s:%s\n", s, fhdr.name); if(mach->sbreg && lookup(0, mach->sbreg, &sym)) { mach->sb = sym.value; l = enter("SB", Tid); l->v->fmt = 'A'; l->v->ival = mach->sb; l->v->type = TINT; l->v->set = 1; } l = mkvar("objtype"); v = l->v; v->fmt = 's'; v->set = 1; v->string = strnode(mach->name); v->type = TSTRING; l = mkvar("textfile"); v = l->v; v->fmt = 's'; v->set = 1; v->string = strnode(s); v->type = TSTRING; machbytype(fhdr.type); varreg(); } Node* an(int op, Node *l, Node *r) { Node *n; n = gmalloc(sizeof(Node)); memset(n, 0, sizeof(Node)); n->gclink = gcl; gcl = n; n->op = op; n->left = l; n->right = r; return n; } List* al(int t) { List *l; l = gmalloc(sizeof(List)); memset(l, 0, sizeof(List)); l->type = t; l->gclink = gcl; gcl = l; return l; } Node* con(vlong v) { Node *n; n = an(OCONST, ZN, ZN); n->ival = v; n->fmt = 'W'; n->type = TINT; return n; } void fatal(char *fmt, ...) { char buf[128]; va_list arg; va_start(arg, fmt); vseprint(buf, buf+sizeof(buf), fmt, arg); va_end(arg); fprint(2, "%s: %L (fatal problem) %s\n", argv0, buf); exits(buf); } void yyerror(char *fmt, ...) { char buf[128]; va_list arg; if(strcmp(fmt, "syntax error") == 0) { yyerror("syntax error, near symbol '%s'", symbol); return; } va_start(arg, fmt); vseprint(buf, buf+sizeof(buf), fmt, arg); va_end(arg); print("%L: %s\n", buf); } void marktree(Node *n) { if(n == 0) return; marktree(n->left); marktree(n->right); n->gcmark = 1; if(n->op != OCONST) return; switch(n->type) { case TSTRING: n->string->gcmark = 1; break; case TLIST: marklist(n->l); break; case TCODE: marktree(n->cc); break; } } void marklist(List *l) { while(l) { l->gcmark = 1; switch(l->type) { case TSTRING: l->string->gcmark = 1; break; case TLIST: marklist(l->l); break; case TCODE: marktree(l->cc); break; } l = l->next; } } void gc(void) { int i; Lsym *f; Value *v; Gc *m, **p, *next; if(dogc < Mempergc) return; dogc = 0; /* Mark */ for(m = gcl; m; m = m->gclink) m->gcmark = 0; /* Scan */ for(i = 0; i < Hashsize; i++) { for(f = hash[i]; f; f = f->hash) { marktree(f->proc); if(f->lexval != Tid) continue; for(v = f->v; v; v = v->pop) { switch(v->type) { case TSTRING: v->string->gcmark = 1; break; case TLIST: marklist(v->l); break; case TCODE: marktree(v->cc); break; } } } } /* Free */ p = &gcl; for(m = gcl; m; m = next) { next = m->gclink; if(m->gcmark == 0) { *p = next; free(m); /* Sleazy reliance on my malloc */ } else p = &m->gclink; } } void* gmalloc(long l) { void *p; dogc += l; p = malloc(l); if(p == 0) fatal("out of memory"); return p; } void checkqid(int f1, int pid) { int fd; Dir *d1, *d2; char buf[128]; if(kernel) return; d1 = dirfstat(f1); if(d1 == nil){ print("checkqid: (qid not checked) dirfstat: %r\n"); return; } snprint(buf, sizeof(buf), "/proc/%d/text", pid); fd = open(buf, OREAD); if(fd < 0 || (d2 = dirfstat(fd)) == nil){ print("checkqid: (qid not checked) dirstat %s: %r\n", buf); free(d1); if(fd >= 0) close(fd); return; } close(fd); if(d1->qid.path != d2->qid.path || d1->qid.vers != d2->qid.vers || d1->qid.type != d2->qid.type){ print("path %llux %llux vers %lud %lud type %d %d\n", d1->qid.path, d2->qid.path, d1->qid.vers, d2->qid.vers, d1->qid.type, d2->qid.type); print("warning: image does not match text for pid %d\n", pid); } free(d1); free(d2); } void catcher(void *junk, char *s) { USED(junk); if(strstr(s, "interrupt")) { gotint = 1; noted(NCONT); } noted(NDFLT); } char* system(void) { char *cpu, *p, *q; static char *kernel; cpu = getenv("cputype"); if(cpu == 0) sysfatal("$cputype not set"); p = getenv("terminal"); if(p == 0 || (p=strchr(p, ' ')) == 0 || p[1] == ' ' || p[1] == 0) sysfatal("$terminal not set"); else{ p++; q = strchr(p, ' '); if(q) *q = 0; q = strrchr(p, '/'); if(q) p = q + 1; } if(kernel != nil) free(kernel); kernel = smprint("/%s/9%s", cpu, p); return kernel; } int isnumeric(char *s) { while(*s) { if(*s < '0' || *s > '9') return 0; s++; } return 1; } int xfmt(Fmt *f) { f->flags ^= FmtSharp; return _ifmt(f); }