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