From b6682e134a670e7bfc8cc6171d3d81d890007b12 Mon Sep 17 00:00:00 2001
From: Xusheng <xusheng@vector35.com>
Date: Mon, 22 Aug 2022 12:38:39 +0800
Subject: let .vars of an SSA MLIL/HLIL function return a list of SSA variables

---
 python/mediumlevelil.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'python/mediumlevelil.py')

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)
-- 
cgit v1.3.1