From e9a35074bb068e4d71d0f60c520128b245edac48 Mon Sep 17 00:00:00 2001 From: Peter LaFosse Date: Wed, 22 Sep 2021 11:30:45 -0400 Subject: Add __contains__ for Function, Section, Segment, BasicBlock and BinaryView Addresses issue binaryninja-api#1113 --- python/function.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'python/function.py') 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) -- cgit v1.3.1