From 2c75c418ea472fca237df5b8f80e47c9cbe35467 Mon Sep 17 00:00:00 2001
From: KyleMiles <krm504@nyu.edu>
Date: Mon, 1 Nov 2021 23:17:40 -0400
Subject: Adds MediumLevelILFunction.aliased_vars,
 HighLevelILFunction.aliased_vars, and the respective C and C++ functions;
 Addressed the API side of #2662

---
 function.cpp | 44 ++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 40 insertions(+), 4 deletions(-)

(limited to 'function.cpp')

diff --git a/function.cpp b/function.cpp
index 45061a96..20c978ac 100644
--- a/function.cpp
+++ b/function.cpp
@@ -1123,8 +1123,26 @@ set<Variable> Function::GetMediumLevelILVariables()
 	BNVariable* vars = BNGetMediumLevelILVariables(mlil->GetObject(), &count);
 
 	set<Variable> result;
-	for (size_t i = 0; i < count; i++)
-	        result.insert({vars[i]});
+	for (size_t i = 0; i < count; ++i)
+		result.emplace(vars[i]);
+
+	BNFreeVariableList(vars);
+	return result;
+}
+
+
+set<Variable> Function::GetMediumLevelILAliasedVariables()
+{
+	Ref<MediumLevelILFunction> mlil = this->GetMediumLevelIL();
+	if (!mlil)
+		return {};
+
+	size_t count;
+	BNVariable* vars = BNGetMediumLevelILAliasedVariables(mlil->GetObject(), &count);
+
+	set<Variable> result;
+	for (size_t i = 0; i < count; ++i)
+		result.emplace(vars[i]);
 
 	BNFreeVariableList(vars);
 	return result;
@@ -1141,8 +1159,26 @@ set<Variable> Function::GetHighLevelILVariables()
 	BNVariable* vars = BNGetHighLevelILVariables(hlil->GetObject(), &count);
 
 	set<Variable> result;
-	for (size_t i = 0; i < count; i++)
-	        result.insert({vars[i]});
+	for (size_t i = 0; i < count; ++i)
+		result.emplace(vars[i]);
+
+	BNFreeVariableList(vars);
+	return result;
+}
+
+
+set<Variable> Function::GetHighLevelILAliasedVariables()
+{
+	Ref<HighLevelILFunction> hlil = this->GetHighLevelIL();
+	if (!hlil)
+		return {};
+
+	size_t count;
+	BNVariable* vars = BNGetHighLevelILAliasedVariables(hlil->GetObject(), &count);
+
+	set<Variable> result;
+	for (size_t i = 0; i < count; ++i)
+		result.emplace(vars[i]);
 
 	BNFreeVariableList(vars);
 	return result;
-- 
cgit v1.3.1