ref: 5e81cc48bdb00a3e7ae3334728659b8f014248da
dir: /sys/src/cmd/dtracy/dtracy.c/
#include <u.h> #include <libc.h> #include <dtracy.h> #include <bio.h> #include "dat.h" #include "fns.h" DTAgg noagg; char *dtracyroot = "#Δ"; int dtracyno; int ctlfd, buffd; int min(int a, int b) { return a < b ? a : b; } int max(int a, int b) { return a < b ? b : a; } void * emalloc(ulong n) { void *v; v = malloc(n); if(v == nil) sysfatal("malloc: %r"); memset(v, 0, n); setmalloctag(v, getcallerpc(&n)); return v; } void * erealloc(void *v, ulong n) { v = realloc(v, n); if(n != 0){ if(v == nil) sysfatal("realloc: %r"); setrealloctag(v, getcallerpc(&v)); } return v; } void * dtmalloc(ulong n) { return emalloc(n); } void dtfree(void *v) { free(v); } void defvar(char *name, int idx, Type *ty) { Symbol *s; s = getsym(name); s->type = SYMVAR; s->idx = idx; s->typ = ty; } void globvars(void) { defvar("arg0", DTV_ARG0, type(TYPINT, 8, 1)); defvar("arg1", DTV_ARG1, type(TYPINT, 8, 1)); defvar("arg2", DTV_ARG2, type(TYPINT, 8, 1)); defvar("arg3", DTV_ARG3, type(TYPINT, 8, 1)); defvar("arg4", DTV_ARG4, type(TYPINT, 8, 1)); defvar("arg5", DTV_ARG5, type(TYPINT, 8, 1)); defvar("arg6", DTV_ARG6, type(TYPINT, 8, 1)); defvar("arg7", DTV_ARG7, type(TYPINT, 8, 1)); defvar("arg8", DTV_ARG8, type(TYPINT, 8, 1)); defvar("arg9", DTV_ARG9, type(TYPINT, 8, 1)); defvar("pid", DTV_PID, type(TYPINT, 4, 1)); defvar("machno", DTV_MACHNO, type(TYPINT, 4, 1)); defvar("time", DTV_TIME, type(TYPINT, 8, 0)); defvar("probe", DTV_PROBE, type(TYPSTRING)); } int setup(void) { char buf[512]; char *p; int n; snprint(buf, sizeof(buf), "%s/clone", dtracyroot); ctlfd = open(buf, ORDWR); if(ctlfd < 0) return -1; n = read(ctlfd, buf, sizeof(buf) - 1); if(n < 0) return -1; buf[n] = 0; dtracyno = strtol(buf, &p, 10); if(p == buf || *p != 0){ werrstr("expected number reading from ctl"); return -1; } snprint(buf, sizeof(buf), "%s/%d/buf", dtracyroot, dtracyno); buffd = open(buf, OREAD); if(buffd < 0) return -1; return 0; } int progcopy(void) { char buf[512]; int fd; char *prog; Fmt f; fmtstrinit(&f); packclauses(&f); prog = fmtstrflush(&f); snprint(buf, sizeof(buf), "%s/%d/prog", dtracyroot, dtracyno); fd = open(buf, OWRITE); if(fd < 0) return -1; if(write(fd, prog, strlen(prog)) < 0){ close(fd); return -1; } close(fd); return 0; } int epidread(void) { char buf[512]; Biobuf *bp; char *s; char *f[5]; int a, b, c; snprint(buf, sizeof(buf), "%s/%d/epid", dtracyroot, dtracyno); bp = Bopen(buf, OREAD); if(bp == nil) return -1; for(; s = Brdstr(bp, '\n', 1), s != nil; free(s)){ if(tokenize(s, f, nelem(f)) != 4) goto err; a = atoi(f[0]); b = atoi(f[1]); c = atoi(f[2]); addepid(a, b, c, f[3]); } return 0; err: werrstr("epidread: invalid format"); free(s); return -1; } int bufread(Biobuf *bp) { static uchar buf[65536]; int n; n = read(buffd, buf, sizeof(buf)); if(n < 0) sysfatal("bufread: %r"); if(parsebuf(buf, n, bp) < 0) sysfatal("parsebuf: %r"); Bflush(bp); return 0; } void aggproc(void) { char buf[65536]; int buffd, n; extern int interrupted; switch(rfork(RFPROC|RFMEM)){ case -1: sysfatal("rfork: %r"); case 0: return; default: break; } snprint(buf, sizeof(buf), "%s/%d/aggbuf", dtracyroot, dtracyno); buffd = open(buf, OREAD); if(buffd < 0) sysfatal("open: %r"); agginit(); atnotify(aggnote, 1); while(!interrupted){ n = read(buffd, buf, sizeof(buf)); if(n < 0){ if(interrupted) break; sysfatal("aggbufread: %r"); } if(aggparsebuf((uchar *) buf, n) < 0) exits("error"); } aggdump(); exits(nil); } static void usage(void) { fprint(2, "usage: %s [ -d ] script\n", argv0); exits("usage"); } int dflag; void main(int argc, char **argv) { Biobuf *out; dflag = 0; ARGBEGIN { case 'd': dflag = 1; break; default: usage(); } ARGEND; if(argc != 1) usage(); fmtinstall(L'α', nodetfmt); fmtinstall('t', typetfmt); fmtinstall(L'I', dtefmt); fmtinstall(L'τ', typefmt); fmtinstall(L'ε', nodefmt); lexinit(); lexstring(argv[0]); globvars(); yyparse(); if(errors != 0) exits("errors"); if(dflag) dump(); else{ if(setup() < 0) sysfatal("setup: %r"); if(progcopy() < 0) sysfatal("progcopy: %r"); if(epidread() < 0) sysfatal("epidread: %r"); fprint(ctlfd, "go"); out = Bfdopen(1, OWRITE); if(out == nil) sysfatal("Bfdopen: %r"); if(aggid > 0) aggproc(); for(;;) bufread(out); } exits(nil); }