diff options
Diffstat (limited to 'appl/cmd/stackv.b')
| -rw-r--r-- | appl/cmd/stackv.b | 445 |
1 files changed, 445 insertions, 0 deletions
diff --git a/appl/cmd/stackv.b b/appl/cmd/stackv.b new file mode 100644 index 00000000..173f8b30 --- /dev/null +++ b/appl/cmd/stackv.b @@ -0,0 +1,445 @@ +implement Stackv; + +include "sys.m"; + sys: Sys; +include "draw.m"; +include "debug.m"; + debug: Debug; + Prog, Module, Exp: import debug; + Tadt, Tarray, Tbig, Tbyte, Treal, + Tfn, Tint, Tlist, + Tref, Tstring, Tslice: import Debug; +include "arg.m"; +include "bufio.m"; + bufio: Bufio; + Iobuf: import bufio; + +stderr: ref Sys->FD; +stdout: ref Iobuf; +hasht := array[97] of list of string; + +Stackv: module { + init: fn(ctxt: ref Draw->Context, argv: list of string); +}; + +maxrecur := 16r7ffffffe; + +badmodule(p: string) +{ + sys->fprint(stderr, "stackv: cannot load %q: %r\n", p); + raise "fail:bad module"; +} + +currp: ref Prog; +showtypes := 1; +showsource := 0; +sep := "\t"; + +init(nil: ref Draw->Context, argv: list of string) +{ + + sys = load Sys Sys->PATH; + stderr = sys->fildes(2); + debug = load Debug Debug->PATH; + if(debug == nil) + badmodule(Debug->PATH); + bufio = load Bufio Bufio->PATH; + if (bufio == nil) + badmodule(Bufio->PATH); + arg := load Arg Arg->PATH; + if (arg == nil) + badmodule(Arg->PATH); + stdout = bufio->fopen(sys->fildes(1), Sys->OWRITE); + + arg->init(argv); + arg->setusage("stackv [-Tl] [-i indent] [-r maxdepth] [-s dis sbl]... [pid[.sym...] ...]"); + sblfile := ""; + while((opt := arg->opt()) != 0){ + case opt { + 's' => + arg->earg(); # XXX make it a list of maps from dis to sbl later + sblfile = arg->earg(); + 'l' => + showsource = 1; + 'r' => + maxrecur = int arg->earg(); + 'T' => + showtypes = 0; + 'i' => + sep = arg->earg(); + * => + arg->usage(); + } + } + debug->init(); + argv = arg->argv(); + printpids := len argv > 1; + if(printpids) + maxrecur++; + for(; argv != nil; argv = tl argv) + db(sys->tokenize(hd argv, ".").t1, printpids); +} + +db(toks: list of string, printpid: int): int +{ + if(toks == nil){ + sys->fprint(stderr, "stackv: bad pid\n"); + return -1; + } + if((pid := int hd toks) <= 0){ + sys->fprint(stderr, "stackv: bad pid %q\n", hd toks); + return -1; + } + err: string; + p: ref Prog; + + # reuse process if possible + if(currp == nil || currp.id != pid){ + (currp, err) = debug->prog(pid); + if(err != nil){ + sys->fprint(stderr, "stackv: %s\n", err); + return -1; + } + if(currp == nil){ + sys->fprint(stderr, "stackv: nil prog from pid %d\n", pid); + return -1; + } + } + p = currp; + stk: array of ref Exp; + (stk, err) = p.stack(); + if(err != nil){ + sys->fprint(stderr, "stackv: %s\n", err); + return -1; + } + for (i := 0; i < len stk; i++) { + stk[i].m.stdsym(); + stk[i].findsym(); + } + depth := 0; + if(printpid){ + stdout.puts(sys->sprint("prog %d {\n", pid)); # } + depth++; + } + pexp(stk, tl toks, depth); + if(printpid) + stdout.puts("}\n"); + stdout.flush(); + return 0; +} + +pexp(stk: array of ref Exp, toks: list of string, depth: int) +{ + if(toks == nil){ + for (i := 0; i < len stk; i++) + pfn(stk[i], depth); + }else{ + exp := stackfindsym(stk, toks, depth); + if(exp == nil) + return; + pname(exp, depth); + stdout.putc('\n'); + } +} + +stackfindsym(stk: array of ref Exp, toks: list of string, depth: int): ref Exp +{ + fname := hd toks; + toks = tl toks; + for(i := 0; i < len stk; i++){ + s := stk[i].name; + if(s == fname) + break; + if(hasdot(s) && toks != nil && s == fname+"."+hd toks){ + fname += "."+hd toks; + toks = tl toks; + break; + } + } + if(i == len stk){ + indent(depth); + stdout.puts("function not found\n"); + return nil; + } + if(toks == nil) + return stk[i]; + stk = stk[i].expand(); + if(hd toks == "module"){ + if((e := getname(stk, "module")) == nil){ + indent(depth); + stdout.puts(sys->sprint("no module declarations in function %q\n", fname)); + }else if((e = symfindsym(e, tl toks, depth)) != nil) + return e; + return nil; + } + for(t := "locals" :: "args" :: "module" :: nil; t != nil; t = tl t){ + if((e := getname(stk, hd t)) == nil) + continue; + if((e = symfindsym(e, toks, depth)) != nil) + return e; + } + indent(depth); + stdout.puts(sys->sprint("symbol %q not found in function %q\n", hd toks, fname)); + return nil; +} + +hasdot(s: string): int +{ + for(i := 0; i < len s; i++) + if(s[i] == '.') + return 1; + return 0; +} + +symfindsym(e: ref Exp, toks: list of string, depth: int): ref Exp +{ + if(toks == nil) + return e; + exps := e.expand(); + for(i := 0; i < len exps; i++) + if(exps[i].name == hd toks) + return symfindsym(exps[i], tl toks, depth); + return nil; +} + +pfn(exp: ref Exp, depth: int) +{ + (v, w) := exp.val(); + if(!w || v == nil){ + indent(depth); + stdout.puts(sys->sprint("no value for fn %q\n", exp.name)); + return; + } + exps := exp.expand(); + indent(depth); + stdout.puts("["+exp.srcstr()+"]\n"); + indent(depth); + stdout.puts(symname(exp)+"("); + if((e := getname(exps, "args")) != nil){ + args := e.expand(); + for(i := 0; i < len args; i++){ + pname(args[i], depth+1); + if(i != len args - 1) + stdout.puts(", "); + } + } + stdout.puts(")\n"); + indent(depth); + stdout.puts("{\n"); # } + if((e = getname(exps, "locals")) != nil){ + locals := e.expand(); + for(i := 0; i < len locals; i++){ + indent(depth+1); + pname(locals[i], depth+1); + stdout.puts("\n"); + } + } + indent(depth); + stdout.puts("}\n"); +} + +getname(exps: array of ref Exp, name: string): ref Exp +{ + for(i := 0; i < len exps; i++) + if(exps[i].name == name) + return exps[i]; + return nil; +} + +strval(v: string): string +{ + for(i := 0; i < len v; i++) + if(v[i] == '"') + break; + if(i < len v) + v = v[i:]; + return v; +} + +pname(exp: ref Exp, depth: int) +{ + (v, w) := exp.val(); + if (!w && v == nil) { + stdout.puts(sys->sprint("%s: %s = novalue", symname(exp), exp.typename())); + return; + } + case exp.kind() { + Tfn => + pfn(exp, depth); + Tint => + stdout.puts(sys->sprint("%s := %s", symname(exp), v)); + Tstring => + stdout.puts(sys->sprint("%s := %s", symname(exp), strval(v))); + Tbyte or + Tbig or + Treal => + stdout.puts(sys->sprint("%s := %s %s", symname(exp), exp.typename(), v)); + * => + if(showtypes) + stdout.puts(sys->sprint("%s: %s = ", symname(exp), exp.typename())); + else + stdout.puts(sys->sprint("%s := ", symname(exp))); + pval(exp, v, w, depth); + } +} + +srcstr(src: ref Debug->Src): string +{ + if(src == nil) + return nil; + if(src.start.file != src.stop.file) + return sys->sprint("%q:%d.%d,%q:%d.%d", src.start.file, src.start.line, src.start.pos, src.stop.file, src.stop.line, src.stop.pos); + if(src.start.line != src.stop.line) + return sys->sprint("%q:%d.%d,%d.%d", src.start.file, src.start.line, src.start.pos, src.stop.line, src.stop.pos); + return sys->sprint("%q:%d.%d,%d", src.start.file, src.start.line, src.start.pos, src.stop.pos); +} + +pval(exp: ref Exp, v: string, w: int, depth: int) +{ + if(depth >= maxrecur){ + stdout.puts(v); + return; + } + case exp.kind() { + Tarray => + if(pref(v)){ + if(depth+1 >= maxrecur) + stdout.puts(v+"{...}"); + else{ + stdout.puts(v+"{\n"); + indent(depth+1); + parray(exp, depth+1); + stdout.puts("\n"); + indent(depth); + stdout.puts("}"); + } + } + Tlist => + if(v == "nil") + stdout.puts("nil"); + else + if(depth+1 >= maxrecur) + stdout.puts(v+"{...}"); + else{ + stdout.puts("{\n"); + indent(depth+1); + plist(exp, v, w, depth+1); + stdout.puts("\n"); + indent(depth); + stdout.puts("}"); + } + Tadt => + pgenval(exp, nil, w, depth); + Tref => + if(pref(v)) + pgenval(exp, v, w, depth); + Tstring => + stdout.puts(strval(v)); + * => + pgenval(exp, v, w, depth); + } +} + +parray(exp: ref Exp, depth: int) +{ + exps := exp.expand(); + for(i := 0; i < len exps; i++){ + e := exps[i]; + (v, w) := e.val(); + if(e.kind() == Tslice) + parray(e, depth); + else{ + pval(e, v, w, depth); + stdout.puts(", "); + } + } +} + +plist(exp: ref Exp, v: string, w: int, depth: int) +{ + while(w && v != "nil"){ + exps := exp.expand(); + h := getname(exps, "hd"); + (hv, vw) := h.val(); + if(pref(v) == 0) + return; + stdout.puts(v+"("); + pval(h, hv, vw, depth); + stdout.puts(") :: "); + h = nil; + exp = getname(exps, "tl"); + (v, w) = exp.val(); + } + stdout.puts("nil"); +} + +pgenval(exp: ref Exp, v: string, w: int, depth: int) +{ + if(w){ + exps := exp.expand(); + if(len exps == 0) + stdout.puts(v); + else{ + stdout.puts(v+"{\n"); # } + if (len exps > 0){ + if(depth >= maxrecur){ + indent(depth); + stdout.puts(sys->sprint("...[%d]\n", len exps)); + }else{ + for (i := 0; i < len exps; i++){ + indent(depth+1); + pname(exps[i], depth+1); + stdout.puts("\n"); + } + } + } + indent(depth); # { + stdout.puts("}"); + } + }else + stdout.puts(v); +} + +symname(exp: ref Exp): string +{ + if(showsource == 0) + return exp.name; + return exp.name+"["+srcstr(exp.src())+"]"; +} + +indent(n: int) +{ + while(n-- > 0) + stdout.puts(sep); +} + +pref(v: string): int +{ + if(addref(v) == 0){ + stdout.puts(v); + if(v != "nil") + stdout.puts("(qv)"); + return 0; + } + return 1; +} + +addref(v: string): int +{ + slot := hashfn(v, len hasht); + for(l := hasht[slot]; l != nil; l = tl l) + if((hd l) == v) + return 0; + hasht[slot] = v :: hasht[slot]; + return 1; +} + +hashfn(s: string, n: int): int +{ + h := 0; + m := len s; + for(i:=0; i<m; i++){ + h = 65599*h+s[i]; + } + return (h & 16r7fffffff) % n; +} |