ref: 21947578303cdb0fcb4b0cdfe80aa9dab5f6b447
dir: /appl/cmd/mprof.b/
implement Prof; include "sys.m"; sys: Sys; include "draw.m"; include "arg.m"; arg: Arg; include "profile.m"; profile: Profile; include "sh.m"; stderr: ref Sys->FD; Prof: module { init: fn(nil: ref Draw->Context, argv: list of string); init0: fn(nil: ref Draw->Context, argv: list of string): Profile->Prof; }; ignored(s: string) { sys->fprint(stderr, "mprof: warning: %s ignored\n", s); } exits(e: string) { if(profile != nil) profile->end(); raise "fail:" + e; } pfatal(s: string) { sys->fprint(stderr, "mprof: %s: %s\n", s, profile->lasterror()); exits("error"); } badmodule(p: string) { sys->fprint(stderr, "mprof: cannot load %s: %r\n", p); exits("bad module"); } usage(s: string) { sys->fprint(stderr, "mprof: %s\n", s); sys->fprint(stderr, "usage: mprof [-bcMflnve] [-m modname]... [cmd arg ...]\n"); exits("usage"); } init(ctxt: ref Draw->Context, argv: list of string) { init0(ctxt, argv); } init0(ctxt: ref Draw->Context, argv: list of string): Profile->Prof { sys = load Sys Sys->PATH; stderr = sys->fildes(2); arg = load Arg Arg->PATH; if(arg == nil) badmodule(Arg->PATH); arg->init(argv); profile = load Profile Profile->PATH; if(profile == nil) badmodule(Profile->PATH); if(profile->init() < 0) pfatal("cannot initialize profile device"); v := 0; begin := end := 0; ep := 0; wm := 0; mem := 0; exec, mods: list of string; while((c := arg->opt()) != 0){ case c { 'b' => begin = 1; 'c' => end = 1; 'M' => v |= profile->MODULE; 'f' => v |= profile->FUNCTION; 'l' => v |= profile->LINE; 'n' => v |= profile->FULLHDR; 'v' => v |= profile->VERBOSE; 'm' => if((s := arg->arg()) == nil) usage("missing module name"); mods = s :: mods; 'e' => ep = 1; 'g' => wm = 1; '1' => mem |= Profile->MAIN; '2' => mem |= Profile->HEAP; '3' => mem |= Profile->IMAGE; * => usage(sys->sprint("unknown option -%c", c)); } } exec = arg->argv(); if(begin && end) ignored("-e option"); if((begin || end) && v != 0) ignored("output format"); if(begin && exec != nil) begin = 0; if(begin == 0 && exec == nil){ if(mods != nil) ignored("-m option"); mods = nil; } if(end){ if(mods != nil) ignored("-m option"); if(ep || exec != nil) ignored("command"); profile->end(); exit; } for( ; mods != nil; mods = tl mods) profile->profile(hd mods); if(begin){ if(profile->memstart(mem) < 0) pfatal("cannot start profiling"); exit; } r := 0; if(exec != nil){ if(ep) profile->profile(disname(hd exec)); if(profile->memstart(mem) < 0) pfatal("cannot start profiling"); # r = run(ctxt, hd exec, exec); wfd := openwait(sys->pctl(0, nil)); ci := chan of int; spawn execute(ctxt, hd exec, exec, ci); epid := <- ci; wait(wfd, epid); } if(profile->stop() < 0) pfatal("cannot stop profiling"); if(exec == nil || r >= 0){ modl := profile->memstats(); if(modl.mods == nil) pfatal("no profile information"); if(wm){ if(exec == nil){ if(profile->memstart(mem) < 0) pfatal("cannot restart profiling"); } else profile->end(); return modl; } if(!(v&(profile->MODULE|profile->FUNCTION|profile->LINE))) v |= profile->MODULE|profile->LINE; if(profile->memshow(modl, v) < 0) pfatal("cannot show profile"); if(exec == nil){ if(profile->memstart(mem) < 0) pfatal("cannot restart profiling"); exit; } } profile->end(); return (nil, 0, nil); } disname(cmd: string): string { file := cmd; if(len file<4 || file[len file-4:]!=".dis") file += ".dis"; if(exists(file)) return file; if(file[0]!='/' && file[0:2]!="./") file = "/dis/"+file; # if(exists(file)) # return file; return file; } execute(ctxt: ref Draw->Context, cmd : string, argl : list of string, ci: chan of int) { ci <-= sys->pctl(Sys->FORKNS|Sys->NEWFD|Sys->NEWPGRP, 0 :: 1 :: 2 :: stderr.fd :: nil); file := cmd; if(len file<4 || file[len file-4:]!=".dis") file += ".dis"; c := load Command file; if(c == nil) { err := sys->sprint("%r"); if(file[0]!='/' && file[0:2]!="./"){ c = load Command "/dis/"+file; if(c == nil) err = sys->sprint("%r"); } if(c == nil){ sys->fprint(stderr, "mprof: %s: %s\n", cmd, err); return; } } c->init(ctxt, argl); } # run(ctxt: ref Draw->Context, cmd : string, argl : list of string): int # { # file := cmd; # if(len file<4 || file[len file-4:]!=".dis") # file += ".dis"; # c := load Command file; # if(c == nil) { # err := sys->sprint("%r"); # if(file[0]!='/' && file[0:2]!="./"){ # c = load Command "/dis/"+file; # if(c == nil) # err = sys->sprint("%r"); # } # if(c == nil){ # sys->fprint(stderr, "mprof: %s: %s\n", cmd, err); # return -1; # } # } # c->init(ctxt, argl); # return 0; # } openwait(pid : int) : ref Sys->FD { w := sys->sprint("#p/%d/wait", pid); fd := sys->open(w, Sys->OREAD); if (fd == nil) pfatal("fd == nil in wait"); return fd; } wait(wfd : ref Sys->FD, wpid : int) { n : int; buf := array[Sys->WAITLEN] of byte; status := ""; for(;;) { if ((n = sys->read(wfd, buf, len buf)) < 0) pfatal("bad read in wait"); status = string buf[0:n]; if (int status == wpid) break; } } exists(f: string): int { return sys->open(f, Sys->OREAD) != nil; }