diff options
Diffstat (limited to 'python/function.py')
| -rw-r--r-- | python/function.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/python/function.py b/python/function.py index 42e88fa3..cedfb9e7 100644 --- a/python/function.py +++ b/python/function.py @@ -264,6 +264,11 @@ class Function: result += token.text return result + def __contains__(self, value:Union[basicblock.BasicBlock, int]): + if isinstance(value, basicblock.BasicBlock): + return value.function == self + return self in [block.function for block in self.view.get_basic_blocks_at(int(value))] + @classmethod def _unregister(cls, func:'core.BNFunction') -> None: handle = ctypes.cast(func, ctypes.c_void_p) |