summaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
Diffstat (limited to 'python')
-rw-r--r--python/function.py4
-rw-r--r--python/mediumlevelil.py6
2 files changed, 5 insertions, 5 deletions
diff --git a/python/function.py b/python/function.py
index cbbdf9bd..f8aaae44 100644
--- a/python/function.py
+++ b/python/function.py
@@ -241,10 +241,10 @@ class Variable(object):
return self.name
def __eq__(self, other):
- return self.identifier == other.identifier
+ return (self.identifier, self.function) == (other.identifier, other.function)
def __hash__(self):
- return hash(self.identifier)
+ return hash((self.identifier, self.function))
class ConstantReference(object):
diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index 333a6c4d..6f5515bb 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -39,12 +39,12 @@ class SSAVariable(object):
def __eq__(self, other):
return (
- (self.var.identifier, self.var.function, self.version) ==
- (other.var.identifier, other.var.function, other.version)
+ (self.var, self.version) ==
+ (other.var, other.version)
)
def __hash__(self):
- return hash((self.var.identifier, self.var.function, self.version))
+ return hash((self.var, self.version))
class MediumLevelILLabel(object):