Added __hash__ and __eq__ to Variable and SSAVariable (#725)

2 files changed, 15 insertions, 0 deletions

diff --git a/python/function.py b/python/function.py
index d06d604d..b8db6875 100644
--- a/python/function.py
+++ b/python/function.py
@@ -203,6 +203,12 @@ class Variable(object):
 	def __str__(self):
 		return self.name
 
+	def __eq__(self, other):
+		return self.identifier == other.identifier
+
+	def __hash__(self):
+		return hash(self.identifier)
+
 
 class ConstantReference(object):
 	def __init__(self, val, size, ptr, intermediate):
diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index 1274bd9b..1fdfab4b 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -36,6 +36,15 @@ class SSAVariable(object):
 	def __repr__(self):
 		return "<ssa %s version %d>" % (repr(self.var), self.version)
 
+	def __eq__(self, other):
+		return (
+			(self.var.identifier, self.version) == 
+			(other.var.identifier, other.version)
+		)
+
+	def __hash__(self):
+		return hash(self.var.identifier, self.version)
+
 
 class MediumLevelILLabel(object):
 	def __init__(self, handle = None):