add contains for LLIL and MLIL basic blocks

1 files changed, 8 insertions, 0 deletions

diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index 624f77a2..8d9768ec 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -1167,6 +1167,14 @@ class MediumLevelILBasicBlock(basicblock.BasicBlock):
 	def __hash__(self):
 		return hash((self.start, self.end, self.il_function))
 
+	def __contains__(self, instruction):
+		if type(instruction) != MediumLevelILInstruction or instruction.il_basic_block != self:
+			return False
+		if instruction.instr_index >= self.start and instruction.instr_index <= self.end:
+			return True
+		else:
+			return False
+
 	@property
 	def il_function(self):
 		""" """