def __main__(): p = Parser(sys.argv[1]) cfg = p.parse() #print("Labels:", p.labels) cfg.parser = p remove_trailing_jumps(cfg) cfg.number_postorder() Inst.trail = ";" #print("Basic blocks:") #dump_bblocks(cfg) #with open(sys.argv[1] + ".0.dot", "w") as f: dot.dot(cfg, f) cprinter.dump_c(cfg)
#!/usr/bin/env python3 import sys from parser import * import dot from core import * from xform import * import cprinter p = Parser(sys.argv[1]) cfg = p.parse() #print("Labels:", p.labels) cfg.parser = p foreach_bblock(cfg, remove_trailing_jumps) cfg.number_postorder() Inst.trail = ";" #print("Basic blocks:") #dump_bblocks(cfg) #with open(sys.argv[1] + ".0.dot", "w") as f: dot.dot(cfg, f) cprinter.dump_c(cfg)
#!/usr/bin/env python3 import sys from parser import * import dot from core import * from xform import * import cprinter p = Parser(sys.argv[1]) cfg = p.parse() #print("Labels:", p.labels) cfg.parser = p remove_trailing_jumps(cfg) cfg.number_postorder() Inst.trail = ";" #print("Basic blocks:") #dump_bblocks(cfg) #with open(sys.argv[1] + ".0.dot", "w") as f: dot.dot(cfg, f) cprinter.dump_c(cfg)
def handle_file_unprotected(args): p = Parser(args.file) cfg = p.parse() cfg.parser = p # If we want to get asm back, i.e. stay close to the input, don't remove # trailing jumps. This will work OK for data flow algos, but will produce # broken or confusing output for control flow algos (for which asm output # shouldn't be used of course). # Update: it's unsafe to use this during dataflow analysis #if args.format != "asm": # foreach_bblock(cfg, remove_trailing_jumps) if args.debug: with open(args.file + ".0.bb", "w") as f: dump_bblocks(cfg, f, no_graph_header=args.no_graph_header) with open(args.file + ".0.dot", "w") as f: dot.dot(cfg, f) if args.script: for s in args.script: mod = __import__(s) mod.apply(cfg) elif hasattr(p, "script"): for op_type, op_name in p.script: if op_type == "xform:": func = globals()[op_name] func(cfg) elif op_type == "xform_bblock:": func = globals()[op_name] foreach_bblock(cfg, func) elif op_type == "xform_inst:": func = globals()[op_name] foreach_inst(cfg, func) elif op_type == "script:": mod = __import__(op_name) mod.apply(cfg) else: assert 0 if args.debug: with open(args.file + ".out.bb", "w") as f: dump_bblocks(cfg, f, no_graph_header=args.no_graph_header) with open(args.file + ".out.dot", "w") as f: dot.dot(cfg, f) if args.output and args.format != "none": out = open(args.output, "w") else: out = sys.stdout if args.no_comments: Inst.show_comments = False if args.format == "bblocks": p = CFGPrinter(cfg, out) if args.no_graph_header: p.print_graph_header = lambda: None p.inst_printer = repr if args.repr else str p.no_dead = args.no_dead p.print() elif args.format == "asm": p = AsmPrinter(cfg, out) p.no_dead = args.no_dead p.print() elif args.format == "c": #foreach_bblock(cfg, remove_trailing_jumps) cfg.number_postorder() Inst.trail = ";" cprinter.no_dead = args.no_dead cprinter.dump_c(cfg, out) if out is not sys.stdout: out.close() progdb.update_funcdb(cfg) return cfg
def handle_file_unprotected(args): p = Parser(args.file) cfg = p.parse() cfg.parser = p # If we want to get asm back, i.e. stay close to the input, don't remove # trailing jumps. This will work OK for data flow algos, but will produce # broken or confusing output for control flow algos (for which asm output # shouldn't be used of course). # Update: it's unsafe to use this during dataflow analysis #if args.format != "asm": # foreach_bblock(cfg, remove_trailing_jumps) if args.debug: with open(args.file + ".0.bb", "w") as f: dump_bblocks(cfg, f, no_graph_header=args.no_graph_header) with open(args.file + ".0.dot", "w") as f: dot.dot(cfg, f) if args.script: for s in args.script: mod = __import__(s) mod.apply(cfg) elif hasattr(p, "script"): for op_type, op_name in p.script: if op_type == "xform:": func = globals()[op_name] func(cfg) elif op_type == "xform_bblock:": func = globals()[op_name] foreach_bblock(cfg, func) elif op_type == "xform_inst:": func = globals()[op_name] foreach_inst(cfg, func) elif op_type == "script:": mod = __import__(op_name) mod.apply(cfg) else: assert 0 if args.debug: with open(args.file + ".out.bb", "w") as f: dump_bblocks(cfg, f, no_graph_header=args.no_graph_header) with open(args.file + ".out.dot", "w") as f: dot.dot(cfg, f) if args.output and args.format != "none": out = open(args.output, "w") else: out = sys.stdout if args.no_comments: Inst.show_comments = False if args.format == "bblocks": p = CFGPrinter(cfg, out) if args.no_graph_header: p.print_graph_header = lambda: None p.inst_printer = repr if args.repr else str p.no_dead = args.no_dead p.print() elif args.format == "asm": p = AsmPrinter(cfg, out) p.no_dead = args.no_dead p.print() elif args.format == "c": #foreach_bblock(cfg, remove_trailing_jumps) cfg.number_postorder() Inst.trail = ";" cprinter.no_dead = args.no_dead cprinter.dump_c(cfg, out) if out is not sys.stdout: out.close() progdb.update_funcdb(cfg) return cfg
def handle_file(args): p = Parser(args.file) cfg = p.parse() cfg.parser = p # If we want to get asm back, i.e. stay close to the input, don't remove # trailing jumps. This will work OK for data flow algos, but will produce # broken or confusing output for control flow algos (for which asm output # shouldn't be used of course). # Update: it's unsafe to use this during dataflow analysis #if args.format != "asm": # foreach_bblock(cfg, remove_trailing_jumps) if args.debug: with open(args.file + ".0.bb", "w") as f: dump_bblocks(cfg, f) with open(args.file + ".0.dot", "w") as f: dot.dot(cfg, f) if args.script: mod = __import__(args.script) mod.apply(cfg) elif hasattr(p, "script"): for (type, xform) in p.script: func = globals()[xform] if type == "xform:": func(cfg) elif type == "xform_bblock:": foreach_bblock(cfg, func) else: assert 0 if args.debug: with open(args.file + ".out.bb", "w") as f: dump_bblocks(cfg, f) with open(args.file + ".out.dot", "w") as f: dot.dot(cfg, f) if args.output: out = open(args.output, "w") else: out = sys.stdout if args.format == "bblocks": p = CFGPrinter(cfg, out) p.inst_printer = repr if args.repr else str p.no_dead = args.no_dead p.print() elif args.format == "asm": p = AsmPrinter(cfg, out) p.no_dead = args.no_dead p.print() elif args.format == "c": #foreach_bblock(cfg, remove_trailing_jumps) cfg.number_postorder() Inst.trail = ";" cprinter.no_dead = args.no_dead cprinter.dump_c(cfg, out) if args.output: out.close() return cfg