From b505e0464383284e7ce8fefeffc18713488c2c48 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Tue, 5 Apr 2016 23:01:17 -0400
Subject: Add backend support for discovering stack variables

---
 function.cpp | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

(limited to 'function.cpp')

diff --git a/function.cpp b/function.cpp
index ee2c5d36..df3b5090 100644
--- a/function.cpp
+++ b/function.cpp
@@ -210,3 +210,46 @@ Ref<FunctionGraph> Function::CreateFunctionGraph()
 	BNFunctionGraph* graph = BNCreateFunctionGraph(m_object);
 	return new FunctionGraph(graph);
 }
+
+
+map<int64_t, NameAndType> Function::GetStackLayout()
+{
+	size_t count;
+	BNStackVariable* vars = BNGetStackLayout(m_object, &count);
+
+	map<int64_t, NameAndType> result;
+	for (size_t i = 0; i < count; i++)
+	{
+		NameAndType nt;
+		nt.name = vars[i].name;
+		nt.type = new Type(BNNewTypeReference(vars[i].type));
+		result[vars[i].offset] = nt;
+	}
+
+	BNFreeStackLayout(vars, count);
+	return result;
+}
+
+
+void Function::CreateAutoStackVariable(int64_t offset, Type* type, const string& name)
+{
+	BNCreateAutoStackVariable(m_object, offset, type->GetObject(), name.c_str());
+}
+
+
+void Function::CreateUserStackVariable(int64_t offset, Type* type, const string& name)
+{
+	BNCreateUserStackVariable(m_object, offset, type->GetObject(), name.c_str());
+}
+
+
+void Function::DeleteAutoStackVariable(int64_t offset)
+{
+	BNDeleteAutoStackVariable(m_object, offset);
+}
+
+
+void Function::DeleteUserStackVariable(int64_t offset)
+{
+	BNDeleteUserStackVariable(m_object, offset);
+}
-- 
cgit v1.3.1