2015-02-08 03:59:53 +01:00
|
|
|
|
|
|
|
|
import os, shutil, sys, optparse
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
|
|
|
|
|
p = optparse.OptionParser(description='Patch calling conventions for some functions, so it builds on windows')
|
|
|
|
|
p.add_option('-v', "--verbose", dest="verbose", action="store_true")
|
|
|
|
|
p.add_option('-f', "--file", dest="path", type="string", help="File name, without extension.")
|
|
|
|
|
opts, _ = p.parse_args(sys.argv[1:])
|
|
|
|
|
|
|
|
|
|
if not opts.path:
|
|
|
|
|
p.print_help()
|
|
|
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
|
|
patches = [
|
2015-04-20 14:26:30 +02:00
|
|
|
# misc
|
|
|
|
|
"virtual int idaapi print",
|
|
|
|
|
|
2015-02-08 03:59:53 +01:00
|
|
|
# user_lvar_visitor_t
|
|
|
|
|
"virtual int idaapi handle_retrieved_info",
|
|
|
|
|
"virtual int idaapi handle_retrieved_mapping",
|
|
|
|
|
"virtual int idaapi get_info_qty_for_saving",
|
|
|
|
|
"virtual bool idaapi get_info_for_saving",
|
|
|
|
|
"virtual lvar_mapping_t const *idaapi get_info_mapping_for_saving",
|
|
|
|
|
|
|
|
|
|
# ctree_visitor_t
|
|
|
|
|
"virtual int idaapi visit_insn",
|
|
|
|
|
"virtual int idaapi visit_expr",
|
|
|
|
|
"virtual int idaapi leave_insn",
|
|
|
|
|
"virtual int idaapi leave_expr",
|
|
|
|
|
|
|
|
|
|
# ctree_parentee_t
|
|
|
|
|
"virtual int idaapi visit_insn",
|
|
|
|
|
"virtual int idaapi visit_expr",
|
|
|
|
|
"virtual int idaapi leave_insn",
|
|
|
|
|
"virtual int idaapi leave_expr",
|
|
|
|
|
|
|
|
|
|
# cfunc_parentee_t
|
|
|
|
|
"virtual int idaapi visit_insn",
|
|
|
|
|
"virtual int idaapi visit_expr",
|
|
|
|
|
"virtual int idaapi leave_insn",
|
|
|
|
|
"virtual int idaapi leave_expr",
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
path = opts.path
|
|
|
|
|
outlines = []
|
|
|
|
|
outpath = "%s.cc" % path
|
|
|
|
|
with open(path, "r") as f:
|
|
|
|
|
lines = f.readlines()
|
|
|
|
|
for line in lines:
|
|
|
|
|
for patch in patches:
|
|
|
|
|
from_text = patch.replace("idaapi ", "")
|
|
|
|
|
if line.find(from_text) > -1:
|
|
|
|
|
line = line.replace(from_text, patch)
|
|
|
|
|
patches.remove(patch)
|
|
|
|
|
break
|
|
|
|
|
outlines.append(line)
|
|
|
|
|
with open(outpath, "w") as f:
|
|
|
|
|
f.writelines(outlines)
|
|
|
|
|
shutil.move(outpath, path)
|
