From b9389bc1cbc1fa54d522cf3cfb56a1aedfc01d6b Mon Sep 17 00:00:00 2001 From: Xusheng Date: Mon, 8 Aug 2022 11:39:18 +0800 Subject: Add missing get_basic_block_at() function for IL functions in Python API. Close #3339. --- python/mediumlevelil.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'python/mediumlevelil.py') diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py index d79e2984..07ca7a22 100644 --- a/python/mediumlevelil.py +++ b/python/mediumlevelil.py @@ -2758,6 +2758,25 @@ class MediumLevelILFunction: def basic_blocks(self) -> 'function.MediumLevelILBasicBlockList': return function.MediumLevelILBasicBlockList(self) + def get_basic_block_at(self, index: int) -> Optional['basicblock.BasicBlock']: + """ + ``get_basic_block_at`` returns the BasicBlock at the given MLIL instruction ``index``. + + :param int index: Index of the MLIL instruction of the BasicBlock to retrieve. + :Example: + >>> current_il_function.get_basic_block_at(current_il_index) + + """ + block = core.BNGetMediumLevelILBasicBlockForInstruction(self.handle, index) + if not block: + return None + + view = None + if self._source_function is not None: + view = self._source_function.view + + return MediumLevelILBasicBlock(block, self, view) + @property def instructions(self) -> Generator[MediumLevelILInstruction, None, None]: """A generator of mlil instructions of the current function""" -- cgit v1.3.1