diff options
Diffstat (limited to 'python/flowgraph.py')
| -rw-r--r-- | python/flowgraph.py | 58 |
1 files changed, 35 insertions, 23 deletions
diff --git a/python/flowgraph.py b/python/flowgraph.py index 7f4be6e2..165dd2b7 100644 --- a/python/flowgraph.py +++ b/python/flowgraph.py @@ -26,8 +26,9 @@ from typing import Optional # Binary Ninja components import binaryninja from . import _binaryninjacore as core -from .enums import (BranchType, InstructionTextTokenType, HighlightStandardColor, FlowGraphOption, - EdgePenStyle, ThemeColor) +from .enums import ( + BranchType, InstructionTextTokenType, HighlightStandardColor, FlowGraphOption, EdgePenStyle, ThemeColor +) from . import function from . import binaryview from . import lowlevelil @@ -52,8 +53,8 @@ class FlowGraphEdge: return "<%s: %s>" % (self.type.name, repr(self.target)) def __eq__(self, other): - return (self.type, self.source, self.target, self.points, self.back_edge, self.style) == \ - (other.type, other.source, other.target, other.points, other.back_edge, other.style) + return (self.type, self.source, self.target, self.points, self.back_edge, + self.style) == (other.type, other.source, other.target, other.points, other.back_edge, other.style) def __hash__(self): return hash((self.type, self.source, self.target, self.points, self.back_edge, self.style)) @@ -77,14 +78,14 @@ class EdgeStyle: return EdgeStyle(edge_style.style, edge_style.width, edge_style.color) def __eq__(self, other): - return (self.style, self.width, self.color) == \ - (other.style, other.width, other.color) + return (self.style, self.width, self.color) == (other.style, other.width, other.color) def __hash__(self): return hash((self.style, self.width, self.color)) + class FlowGraphNode: - def __init__(self, graph = None, handle = None): + def __init__(self, graph=None, handle=None): _handle = handle if _handle is None: if graph is None: @@ -94,7 +95,7 @@ class FlowGraphNode: self.handle = _handle self._graph = graph if self._graph is None: - self._graph = FlowGraph(handle = core.BNGetFlowGraphNodeOwner(self.handle)) + self._graph = FlowGraph(handle=core.BNGetFlowGraphNodeOwner(self.handle)) def __del__(self): if core is not None: @@ -131,7 +132,8 @@ class FlowGraphNode: try: for i in range(0, count.value): addr = lines[i].addr - if (lines[i].instrIndex != 0xffffffffffffffff) and (block is not None) and hasattr(block, 'il_function'): + if (lines[i].instrIndex != 0xffffffffffffffff) and (block + is not None) and hasattr(block, 'il_function'): il_instr = block.__dict__['il_function'][lines[i].instrIndex] else: il_instr = None @@ -142,7 +144,7 @@ class FlowGraphNode: @property def graph(self): - return self._graph + return self._graph @graph.setter def graph(self, value): @@ -159,14 +161,18 @@ class FlowGraphNode: core.BNFreeBasicBlock(block) return None - view = binaryview.BinaryView(handle = core.BNGetFunctionData(func_handle)) + view = binaryview.BinaryView(handle=core.BNGetFunctionData(func_handle)) func = function.Function(view, func_handle) if core.BNIsLowLevelILBasicBlock(block): - block = lowlevelil.LowLevelILBasicBlock(block, - lowlevelil.LowLevelILFunction(func.arch, core.BNGetBasicBlockLowLevelILFunction(block), func), view) + block = lowlevelil.LowLevelILBasicBlock( + block, lowlevelil.LowLevelILFunction(func.arch, core.BNGetBasicBlockLowLevelILFunction(block), func), + view + ) elif core.BNIsMediumLevelILBasicBlock(block): - mlil_func = mediumlevelil.MediumLevelILFunction(func.arch, core.BNGetBasicBlockMediumLevelILFunction(block), func) + mlil_func = mediumlevelil.MediumLevelILFunction( + func.arch, core.BNGetBasicBlockMediumLevelILFunction(block), func + ) block = mediumlevelil.MediumLevelILBasicBlock(block, mlil_func, view) elif core.BNIsHighLevelILBasicBlock(block): hlil_func = highlevelil.HighLevelILFunction(func.arch, core.BNGetBasicBlockHighLevelILFunction(block), func) @@ -230,7 +236,9 @@ class FlowGraphNode: for i in range(0, len(lines)): line = lines[i] if isinstance(line, str): - line = function.DisassemblyTextLine([function.InstructionTextToken(InstructionTextTokenType.TextToken, line)]) + line = function.DisassemblyTextLine([ + function.InstructionTextToken(InstructionTextTokenType.TextToken, line) + ]) if not isinstance(line, function.DisassemblyTextLine): line = function.DisassemblyTextLine(line) if line.address is None: @@ -269,7 +277,9 @@ class FlowGraphNode: points = [] for j in range(0, edges[i].pointCount): points.append((edges[i].points[j].x, edges[i].points[j].y)) - result.append(FlowGraphEdge(branch_type, self, target, points, edges[i].backEdge, EdgeStyle(edges[i].style))) + result.append( + FlowGraphEdge(branch_type, self, target, points, edges[i].backEdge, EdgeStyle(edges[i].style)) + ) core.BNFreeFlowGraphNodeEdgeList(edges, count.value) return result @@ -288,7 +298,9 @@ class FlowGraphNode: points = [] for j in range(0, edges[i].pointCount): points.append((edges[i].points[j].x, edges[i].points[j].y)) - result.append(FlowGraphEdge(branch_type, self, target, points, edges[i].backEdge, EdgeStyle(edges[i].style))) + result.append( + FlowGraphEdge(branch_type, self, target, points, edges[i].backEdge, EdgeStyle(edges[i].style)) + ) core.BNFreeFlowGraphNodeEdgeList(edges, count.value) return result @@ -336,7 +348,7 @@ class FlowGraphNode: class FlowGraphLayoutRequest: - def __init__(self, graph, callback = None): + def __init__(self, graph, callback=None): self.on_complete = callback self._cb = ctypes.CFUNCTYPE(None, ctypes.c_void_p)(self._complete) self.handle = core.BNStartFlowGraphLayout(graph.handle, None, self._cb) @@ -402,7 +414,7 @@ class FlowGraph: """ _registered_instances = [] - def __init__(self, handle:Optional[core.BNCustomFlowGraphHandle] = None): + def __init__(self, handle: Optional[core.BNCustomFlowGraphHandle] = None): _handle = handle if _handle is None: self._ext_cb = core.BNCustomFlowGraph() @@ -541,7 +553,7 @@ class FlowGraph: func = core.BNGetFunctionForFlowGraph(self.handle) if func is None: return None - return function.Function(handle = func) + return function.Function(handle=func) @function.setter def function(self, func): @@ -555,7 +567,7 @@ class FlowGraph: view = core.BNGetViewForFlowGraph(self.handle) if view is None: return None - return binaryview.BinaryView(handle = view) + return binaryview.BinaryView(handle=view) @view.setter def view(self, view): @@ -732,7 +744,7 @@ class FlowGraph: def shows_secondary_reg_highlighting(self, value): self.set_option(FlowGraphOption.FlowGraphShowsSecondaryRegisterHighlighting, value) - def layout(self, callback = None): + def layout(self, callback=None): """ ``layout`` starts rendering a graph for display. Once a layout is complete, each node will contain coordinates and extents that can be used to render a graph with minimum additional computation. @@ -807,7 +819,7 @@ class FlowGraph: """ return NotImplemented - def set_option(self, option, value = True): + def set_option(self, option, value=True): core.BNSetFlowGraphOption(self.handle, option, value) def is_option_set(self, option): |