mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
132 lines
4.5 KiB
Python
132 lines
4.5 KiB
Python
"""
|
|
Various hooks for Hexrays Decompiler
|
|
"""
|
|
from __future__ import print_function
|
|
|
|
import ida_idaapi
|
|
import ida_typeinf
|
|
import ida_hexrays
|
|
|
|
class vds_hooks_t(ida_hexrays.Hexrays_Hooks):
|
|
def _shorten(self, cfunc):
|
|
raw = str(cfunc)
|
|
if len(raw) > 20:
|
|
raw = raw[0:20] + "[...snipped...]"
|
|
return raw
|
|
|
|
def _format_lvar(self, v):
|
|
parts = []
|
|
if v:
|
|
if v.name:
|
|
parts.append("name=%s" % v.name)
|
|
if v.cmt:
|
|
parts.append("cmt=%s" % v.cmt)
|
|
parts.append("width=%s" % v.width)
|
|
parts.append("defblk=%s" % v.defblk)
|
|
parts.append("divisor=%s" % v.divisor)
|
|
return "{%s}" % ", ".join(parts)
|
|
|
|
def _log(self, msg):
|
|
print("### %s" % msg)
|
|
return 0
|
|
|
|
def flowchart(self, fc):
|
|
return self._log("flowchart: fc=%s" % fc)
|
|
|
|
def stkpnts(self, mba, stkpnts):
|
|
return self._log("stkpnts: mba=%s, stkpnts=%s" % (mba, stkpnts))
|
|
|
|
def prolog(self, mba, fc, reachable_blocks, decomp_flags):
|
|
return self._log("prolog: mba=%s, fc=%s, reachable_blocks=%s, decomp_flags=%x" % (mba, fc, reachable_blocks, decomp_flags))
|
|
|
|
def microcode(self, mba):
|
|
return self._log("microcode: mba=%s" % (mba,))
|
|
|
|
def preoptimized(self, mba):
|
|
return self._log("preoptimized: mba=%s" % (mba,))
|
|
|
|
def locopt(self, mba):
|
|
return self._log("locopt: mba=%s" % (mba,))
|
|
|
|
def prealloc(self, mba):
|
|
return self._log("prealloc: mba=%s" % (mba,))
|
|
|
|
def glbopt(self, mba):
|
|
return self._log("glbopt: mba=%s" % (mba,))
|
|
|
|
def structural(self, ctrl_graph):
|
|
return self._log("structural: ctrl_graph: %s" % (ctrl_graph,))
|
|
|
|
def maturity(self, cfunc, maturity):
|
|
return self._log("maturity: cfunc=%s, maturity=%s" % (self._shorten(cfunc), maturity))
|
|
|
|
def interr(self, code):
|
|
return self._log("interr: code=%s" % (code,))
|
|
|
|
def combine(self, blk, insn):
|
|
return self._log("combine: blk=%s, insn=%s" % (blk, insn))
|
|
|
|
def print_func(self, cfunc, printer):
|
|
# Note: we can't print/str()-ify 'cfunc' here,
|
|
# because that'll call print_func() us recursively.
|
|
return self._log("print_func: cfunc=..., printer=%s" % (printer,))
|
|
|
|
def func_printed(self, cfunc):
|
|
return self._log("func_printed: cfunc=%s" % (cfunc,))
|
|
|
|
def resolve_stkaddrs(self, mba):
|
|
return self._log("resolve_stkaddrs: mba=%s" % (mba,))
|
|
|
|
def open_pseudocode(self, vu):
|
|
return self._log("open_pseudocode: vu=%s" % (vu,))
|
|
|
|
def switch_pseudocode(self, vu):
|
|
return self._log("switch_pseudocode: vu=%s" % (vu,))
|
|
|
|
def refresh_pseudocode(self, vu):
|
|
return self._log("refresh_pseudocode: vu=%s" % (vu,))
|
|
|
|
def close_pseudocode(self, vu):
|
|
return self._log("close_pseudocode: vu=%s" % (vu,))
|
|
|
|
def keyboard(self, vu, key_code, shift_state):
|
|
return self._log("keyboard: vu=%s, key_code=%s, shift_state=%s" % (vu, key_code, shift_state))
|
|
|
|
def right_click(self, vu):
|
|
return self._log("right_click: vu=%s" % (vu,))
|
|
|
|
def double_click(self, vu, shift_state):
|
|
return self._log("double_click: vu=%s, shift_state=%s" % (vu, shift_state))
|
|
|
|
def curpos(self, vu):
|
|
return self._log("curpos: vu=%s (vu.cpos.lnnum=%d, vu.cpos.x=%d, vu.cpos.y=%d)" % (
|
|
vu, vu.cpos.lnnum, vu.cpos.x, vu.cpos.y))
|
|
|
|
def create_hint(self, vu):
|
|
return self._log("create_hint: vu=%s: " % (vu,))
|
|
|
|
def text_ready(self, vu):
|
|
return self._log("text_ready: vu=%s" % (vu,))
|
|
|
|
def populating_popup(self, widget, popup, vu):
|
|
return self._log("populating_popup: widget=%s, popup=%s, vu=%s" % (widget, popup, vu))
|
|
|
|
def lvar_name_changed(self, vu, v, name, is_user_name):
|
|
return self._log("lvar_name_changed: vu=%s, v=%s, name=%s, is_user_name=%s" % (vu, self._format_lvar(v), name, is_user_name))
|
|
|
|
def lvar_type_changed(self, vu, v, tif):
|
|
return self._log("lvar_type_changed: vu=%s, v=%s, tinfo=%s" % (vu, self._format_lvar(v), tif._print()))
|
|
|
|
def lvar_cmt_changed(self, vu, v, cmt):
|
|
return self._log("lvar_cmt_changed: vu=%s, v=%s, cmt=%s" % (vu, self._format_lvar(v), cmt))
|
|
|
|
def lvar_mapping_changed(self, vu, _from, to):
|
|
return self._log("lvar_mapping_changed: vu=%s, from=%s, to=%s" % (vu, _from, to))
|
|
|
|
def cmt_changed(self, cfunc, loc, cmt):
|
|
return self._log("cmt_changed: cfunc=%s, loc=%s, cmt=%s" % (self._shorten(cfunc),loc, cmt))
|
|
|
|
vds_hooks = vds_hooks_t()
|
|
vds_hooks.hook()
|
|
|