mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
""" It demonstrates how to iterate a cblock_t object.
|
|
|
|
Author: EiNSTeiN_ <einstein@g3nius.org>
|
|
|
|
This is a rewrite in Python of the vds7 example that comes with hexrays sdk.
|
|
"""
|
|
|
|
import idautils
|
|
import idaapi
|
|
import idc
|
|
|
|
import traceback
|
|
|
|
class cblock_visitor_t(idaapi.ctree_visitor_t):
|
|
|
|
def __init__(self):
|
|
idaapi.ctree_visitor_t.__init__(self, idaapi.CV_FAST)
|
|
return
|
|
|
|
def visit_insn(self, ins):
|
|
|
|
try:
|
|
if ins.op == idaapi.cit_block:
|
|
self.dump_block(ins.ea, ins.cblock)
|
|
except:
|
|
traceback.print_exc()
|
|
|
|
return 0
|
|
|
|
def dump_block(self, ea, b):
|
|
# iterate over all block instructions
|
|
print "dumping block %x" % (ea, )
|
|
for ins in b:
|
|
print " %x: insn %s" % (ins.ea, ins.opname)
|
|
|
|
return
|
|
|
|
|
|
class vds7_hooks_t(idaapi.Hexrays_Hooks):
|
|
def maturity(self, cfunc, maturity):
|
|
if maturity == idaapi.CMAT_BUILT:
|
|
cbv = cblock_visitor_t()
|
|
cbv.apply_to(cfunc.body, None)
|
|
return 0
|
|
|
|
|
|
if idaapi.init_hexrays_plugin():
|
|
vds7_hooks = vds7_hooks_t()
|
|
vds7_hooks.hook()
|
|
else:
|
|
print 'cblock visitor: hexrays is not available.'
|