Add SSAVariable.def_site and SSAVariable.use_sites

unknowntrojan · Nov 28, 2023 · 66fe34a · 66fe34a
1 parent 72a4548
commit 66fe34a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
@@ -85,6 +85,20 @@ def function(self) -> 'function.Function':
 	def dead_store_elimination(self) -> DeadStoreElimination:
 		return self.var.dead_store_elimination
 
+	@property
+	def def_site(self) -> Optional['MediumLevelILInstruction']:
+		"""
+		Gets the MediumLevelILInstruction where this SSAVariable is defined.
+		"""
+		return self.var.function.get_ssa_var_definition(self)
+
+	@property
+	def use_sites(self) -> List['MediumLevelILInstruction']:
+		"""
+		Gets the list of MediumLevelILInstructions where this SSAVariable is used inside of this function.
+		"""
+		return self.var.function.get_ssa_var_uses(self)
+
 
 class MediumLevelILLabel:
 	def __init__(self, handle: Optional[core.BNMediumLevelILLabel] = None):