mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
103 lines
2.9 KiB
Python
103 lines
2.9 KiB
Python
#<pycode(py_gdl)>
|
|
import types
|
|
|
|
import _ida_idaapi
|
|
import ida_idaapi
|
|
# -----------------------------------------------------------------------
|
|
class BasicBlock(object):
|
|
"""Basic block class. It is returned by the Flowchart class"""
|
|
def __init__(self, id, bb, fc):
|
|
self._fc = fc
|
|
|
|
self.id = id
|
|
"""Basic block ID"""
|
|
|
|
self.start_ea = bb.start_ea
|
|
"""start_ea of basic block"""
|
|
|
|
self.end_ea = bb.end_ea
|
|
"""end_ea of basic block"""
|
|
|
|
self.type = self._fc._q.calc_block_type(self.id)
|
|
"""Block type (check fc_block_type_t enum)"""
|
|
|
|
|
|
def preds(self):
|
|
"""
|
|
Iterates the predecessors list
|
|
"""
|
|
q = self._fc._q
|
|
for i in range(0, self._fc._q.npred(self.id)):
|
|
yield self._fc[q.pred(self.id, i)]
|
|
|
|
|
|
def succs(self):
|
|
"""
|
|
Iterates the successors list
|
|
"""
|
|
q = self._fc._q
|
|
for i in range(0, q.nsucc(self.id)):
|
|
yield self._fc[q.succ(self.id, i)]
|
|
|
|
try:
|
|
if _BC695:
|
|
startEA = property(lambda self: self.start_ea, lambda self, ea: setattr(self, "start_ea", ea))
|
|
endEA = property(lambda self: self.end_ea, lambda self, ea: setattr(self, "end_ea", ea))
|
|
else:
|
|
startEA = ida_idaapi._make_badattr_property("startEA", "start_ea")
|
|
endEA = ida_idaapi._make_badattr_property("endEA", "end_ea")
|
|
except:
|
|
pass # BC695 not defined at compile-time
|
|
|
|
# -----------------------------------------------------------------------
|
|
class FlowChart(object):
|
|
"""
|
|
Flowchart class used to determine basic blocks.
|
|
Check ex_gdl_qflow_chart.py for sample usage.
|
|
"""
|
|
def __init__(self, f=None, bounds=None, flags=0):
|
|
"""
|
|
Constructor
|
|
@param f: A func_t type, use get_func(ea) to get a reference
|
|
@param bounds: A tuple of the form (start, end). Used if "f" is None
|
|
@param flags: one of the FC_xxxx flags. One interesting flag is FC_PREDS
|
|
"""
|
|
if (f is None) and (bounds is None or type(bounds) != tuple):
|
|
raise Exception("Please specifiy either a function or start/end pair")
|
|
|
|
if bounds is None:
|
|
bounds = (_ida_idaapi.BADADDR, _ida_idaapi.BADADDR)
|
|
|
|
# Create the flowchart
|
|
self._q = qflow_chart_t("", f, bounds[0], bounds[1], flags)
|
|
|
|
size = property(lambda self: self._q.size())
|
|
"""Number of blocks in the flow chart"""
|
|
|
|
|
|
def refresh(self):
|
|
"""Refreshes the flow chart"""
|
|
self._q.refresh()
|
|
|
|
|
|
def _getitem(self, index):
|
|
return BasicBlock(index, self._q[index], self)
|
|
|
|
|
|
def __iter__(self):
|
|
return (self._getitem(index) for index in range(0, self.size))
|
|
|
|
|
|
def __getitem__(self, index):
|
|
"""
|
|
Returns a basic block
|
|
|
|
@return: BasicBlock
|
|
"""
|
|
if index >= self.size:
|
|
raise KeyError
|
|
else:
|
|
return self._getitem(index)
|
|
|
|
#</pycode(py_gdl)>
|