diff options
Diffstat (limited to 'python/architecture.py')
| -rw-r--r-- | python/architecture.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/python/architecture.py b/python/architecture.py index b9001bf3..a6e1a083 100644 --- a/python/architecture.py +++ b/python/architecture.py @@ -2724,6 +2724,16 @@ class ReferenceSource(object): else: return "<ref: %#x>" % self._address + def __eq__(self, value): + if not isinstance(value, ReferenceSource): + return False + return self.function == value.function and self.arch == value.arch and self.address == value.address + + def __lt__(self, value): + if not isinstance(value, ReferenceSource): + raise TypeError("Can only compare to other ReferenceSource objects") + return self.address < value.address + @property def function(self): """ """ |