From 167ced17956bbb79aee3a6e6ef88079d822427cb Mon Sep 17 00:00:00 2001 From: Chinmay Date: Wed, 9 Sep 2020 22:17:15 -0700 Subject: Update API name for ParsePossibleValueSet --- python/function.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'python/function.py') diff --git a/python/function.py b/python/function.py index 44c2f383..bf5afb87 100644 --- a/python/function.py +++ b/python/function.py @@ -382,13 +382,15 @@ class PossibleValueSet(object): def __eq__(self, other): if self.type in [RegisterValueType.ConstantValue, RegisterValueType.ConstantPointerValue] and isinstance(other, numbers.Integral): return self.value == other - if self.type in [RegisterValueType.ConstantValue, RegisterValueType.ConstantPointerValue] and hasattr(other, 'type') and other.type == self.type: + if not isinstance(other, self.__class__): + return NotImplemented + if self.type in [RegisterValueType.ConstantValue, RegisterValueType.ConstantPointerValue]: return self.value == other.value - elif self.type == RegisterValueType.StackFrameOffset and hasattr(other, 'type') and other.type == self.type: + elif self.type == RegisterValueType.StackFrameOffset: return self.offset == other.offset - elif self.type in [RegisterValueType.SignedRangeValue, RegisterValueType.UnsignedRangeValue] and hasattr(other, 'type') and other.type == self.type: + elif self.type in [RegisterValueType.SignedRangeValue, RegisterValueType.UnsignedRangeValue]: return self.ranges == other.ranges - elif self.type in [RegisterValueType.InSetOfValues, RegisterValueType.NotInSetOfValues] and hasattr(other, 'type') and other.type == self.type: + elif self.type in [RegisterValueType.InSetOfValues, RegisterValueType.NotInSetOfValues]: return self.values == other.values elif self.type == RegisterValueType.UndeterminedValue and hasattr(other, 'type'): return self.type == other.type -- cgit v1.3.1