diff options
| -rw-r--r-- | binaryninjaapi.h | 6 | ||||
| -rw-r--r-- | function.cpp | 43 |
2 files changed, 49 insertions, 0 deletions
diff --git a/binaryninjaapi.h b/binaryninjaapi.h index 0895d831..6ddeaa17 100644 --- a/binaryninjaapi.h +++ b/binaryninjaapi.h @@ -1273,6 +1273,12 @@ namespace BinaryNinja void ApplyAutoDiscoveredType(Type* type); Ref<FunctionGraph> CreateFunctionGraph(); + + std::map<int64_t, NameAndType> GetStackLayout(); + void CreateAutoStackVariable(int64_t offset, Type* type, const std::string& name); + void CreateUserStackVariable(int64_t offset, Type* type, const std::string& name); + void DeleteAutoStackVariable(int64_t offset); + void DeleteUserStackVariable(int64_t offset); }; struct FunctionGraphTextLine 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); +} |