let .vars of an SSA MLIL/HLIL function return a list of SSA variables

2 files changed, 12 insertions, 6 deletions

diff --git a/python/highlevelil.py b/python/highlevelil.py
index 2acaea74..451942fc 100644
--- a/python/highlevelil.py
+++ b/python/highlevelil.py
@@ -2435,9 +2435,10 @@ class HighLevelILFunction:
 		if self.source_function is None:
 			return []
 
-		if self.il_form in [
-		    FunctionGraphType.HighLevelILFunctionGraph, FunctionGraphType.HighLevelILSSAFormFunctionGraph
-		]:
+		if self.il_form == FunctionGraphType.HighLevelILSSAFormFunctionGraph:
+			return self.ssa_vars
+
+		if self.il_form == FunctionGraphType.HighLevelILFunctionGraph:
 			count = ctypes.c_ulonglong()
 			core_variables = core.BNGetHighLevelILVariables(self.handle, count)
 			assert core_variables is not None, "core.BNGetHighLevelILVariables returned None"
diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index bdc9b954..06298e6e 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -3150,9 +3150,14 @@ class MediumLevelILFunction:
 			return []
 
 		if self.il_form in [
-		    FunctionGraphType.MediumLevelILFunctionGraph, FunctionGraphType.MediumLevelILSSAFormFunctionGraph,
-		    FunctionGraphType.MappedMediumLevelILFunctionGraph,
-		    FunctionGraphType.MappedMediumLevelILSSAFormFunctionGraph
+			FunctionGraphType.MediumLevelILSSAFormFunctionGraph,
+			FunctionGraphType.MappedMediumLevelILSSAFormFunctionGraph
+		]:
+			return self.ssa_vars
+
+		if self.il_form in [
+		    FunctionGraphType.MediumLevelILFunctionGraph,
+		    FunctionGraphType.MappedMediumLevelILFunctionGraph
 		]:
 			count = ctypes.c_ulonglong()
 			core_variables = core.BNGetMediumLevelILVariables(self.handle, count)