diff options
Diffstat (limited to 'python/flowgraph.py')
| -rw-r--r-- | python/flowgraph.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/python/flowgraph.py b/python/flowgraph.py index 1d3d9504..725af442 100644 --- a/python/flowgraph.py +++ b/python/flowgraph.py @@ -168,6 +168,9 @@ class FlowGraphNode: elif core.BNIsMediumLevelILBasicBlock(block): 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) + block = highlevelil.HighLevelILBasicBlock(block, hlil_func, view) else: block = basicblock.BasicBlock(block, view) return block |