diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/lowlevelil.py | 14 | ||||
| -rw-r--r-- | python/mediumlevelil.py | 4 |
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 |