add type checks for various comparators

2 files changed, 17 insertions, 1 deletions

diff --git a/python/lowlevelil.py b/python/lowlevelil.py
index 90ecbd74..efa92a7a 100644
--- a/python/lowlevelil.py
+++ b/python/lowlevelil.py
@@ -61,6 +61,8 @@ class ILRegister(object):
 		return self.name
 
 	def __eq__(self, other):
+		if not isinstance(other, type(self)):
+			return False
 		return self.info == other.info
 
 
@@ -81,6 +83,8 @@ class ILRegisterStack(object):
 		return self.name
 
 	def __eq__(self, other):
+		if not isinstance(other, type(self)):
+			return False
 		return self.info == other.info
 
 
@@ -114,6 +118,8 @@ class ILSemanticFlagClass(object):
 		return self.name
 
 	def __eq__(self, other):
+		if not isinstance(other, type(self)):
+			return False
 		return self.index == other.index
 
 
@@ -130,6 +136,8 @@ class ILSemanticFlagGroup(object):
 		return self.name
 
 	def __eq__(self, other):
+		if not isinstance(other, type(self)):
+			return False
 		return self.index == other.index
 
 
@@ -149,6 +157,8 @@ class ILIntrinsic(object):
 		return self.name
 
 	def __eq__(self, other):
+		if not isinstance(other, type(self)):
+			return False
 		return self.index == other.index
 
 
@@ -512,6 +522,8 @@ class LowLevelILInstruction(object):
 		return "<il: %s>" % str(self)
 
 	def __eq__(self, value):
+		if not isinstance(value, type(self)):
+			return False
 		return self.function == value.function and self.expr_index == value.expr_index
 
 	@property
@@ -763,7 +775,7 @@ class LowLevelILFunction(object):
 			core.BNFreeLowLevelILFunction(self.handle)
 
 	def __eq__(self, value):
-		if not isinstance(value, LowLevelILFunction):
+		if not isinstance(value, type(self)):
 			return False
 		return ctypes.addressof(self.handle.contents) == ctypes.addressof(value.handle.contents)
 
diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index e8d58f96..b6b7dbfc 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -43,6 +43,8 @@ class SSAVariable(object):
 		return "<ssa %s version %d>" % (repr(self.var), self.version)
 
 	def __eq__(self, other):
+		if not isinstance(other, type(self)):
+			return False
 		return isinstance(other, SSAVariable) and (
 			(self.var, self.version) ==
 			(other.var, other.version)
@@ -318,6 +320,8 @@ class MediumLevelILInstruction(object):
 		return "<il: %s>" % str(self)
 
 	def __eq__(self, value):
+		if not isinstance(value, type(self)):
+			return False
 		return self.function == value.function and self.expr_index == value.expr_index
 
 	@property