From 47333ef2460edfa9b5ba5be26fd19f80c0d8d8b6 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Tue, 29 Aug 2017 20:13:57 -0400
Subject: Updating APIs to deal with stack adjustment

---
 type.cpp | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(limited to 'type.cpp')

diff --git a/type.cpp b/type.cpp
index aa891557..40a69ae2 100644
--- a/type.cpp
+++ b/type.cpp
@@ -425,6 +425,13 @@ uint64_t Type::GetOffset() const
 }
 
 
+Confidence<size_t> Type::GetStackAdjustment() const
+{
+	BNSizeWithConfidence result = BNGetTypeStackAdjustment(m_object);
+	return Confidence<size_t>(result.value, result.confidence);
+}
+
+
 string Type::GetString(Platform* platform) const
 {
 	char* str = BNGetTypeString(m_object, platform ? platform->GetObject() : nullptr);
@@ -663,7 +670,8 @@ Ref<Type> Type::ArrayType(const Confidence<Ref<Type>>& type, uint64_t elem)
 
 Ref<Type> Type::FunctionType(const Confidence<Ref<Type>>& returnValue,
 	const Confidence<Ref<CallingConvention>>& callingConvention,
-	const std::vector<FunctionParameter>& params, const Confidence<bool>& varArg)
+	const std::vector<FunctionParameter>& params, const Confidence<bool>& varArg,
+	const Confidence<size_t>& stackAdjust)
 {
 	BNTypeWithConfidence returnValueConf;
 	returnValueConf.type = returnValue->GetObject();
@@ -689,8 +697,12 @@ Ref<Type> Type::FunctionType(const Confidence<Ref<Type>>& returnValue,
 	varArgConf.value = varArg.GetValue();
 	varArgConf.confidence = varArg.GetConfidence();
 
+	BNSizeWithConfidence stackAdjustConf;
+	stackAdjustConf.value = stackAdjust.GetValue();
+	stackAdjustConf.confidence = stackAdjust.GetConfidence();
+
 	Type* type = new Type(BNCreateFunctionType(&returnValueConf, &callingConventionConf,
-		paramArray, params.size(), &varArgConf));
+		paramArray, params.size(), &varArgConf, &stackAdjustConf));
 	delete[] paramArray;
 	return type;
 }
-- 
cgit v1.3.1