Add API to set type of function

1 files changed, 12 insertions, 0 deletions

diff --git a/function.cpp b/function.cpp
index ea0fec86..55f5cffb 100644
--- a/function.cpp
+++ b/function.cpp
@@ -354,6 +354,18 @@ Ref<Type> Function::GetType() const
 }
 
 
+void Function::SetAutoType(Type* type)
+{
+	BNSetFunctionAutoType(m_object, type->GetObject());
+}
+
+
+void Function::SetUserType(Type* type)
+{
+	BNSetFunctionUserType(m_object, type->GetObject());
+}
+
+
 void Function::ApplyImportedTypes(Symbol* sym)
 {
 	BNApplyImportedTypes(m_object, sym->GetObject());