add type checks for various comparators

1 files changed, 4 insertions, 0 deletions

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