expose overriding call types at individual call sites

1 files changed, 20 insertions, 0 deletions

diff --git a/python/function.py b/python/function.py
index 4707aa45..4ecece2a 100644
--- a/python/function.py
+++ b/python/function.py
@@ -2225,6 +2225,17 @@ class Function(object):
 		core.BNSetAutoCallRegisterStackAdjustmentForRegisterStack(self.handle, arch.handle, addr, reg_stack,
 			adjust.value, adjust.confidence)
 
+	def set_call_type_adjustment(self, addr, adjust_type, arch=None):
+		if arch is None:
+			arch = self.arch
+		if adjust_type is None:
+			tc = None
+		else:
+			tc = core.BNTypeWithConfidence()
+			tc.type = adjust_type.handle
+			tc.confidence = adjust_type.confidence
+		core.BNSetUserCallTypeAdjustment(self.handle, arch.handle, addr, tc)
+
 	def set_call_stack_adjustment(self, addr, adjust, arch=None):
 		if arch is None:
 			arch = self.arch
@@ -2256,6 +2267,15 @@ class Function(object):
 		core.BNSetUserCallRegisterStackAdjustmentForRegisterStack(self.handle, arch.handle, addr, reg_stack,
 			adjust.value, adjust.confidence)
 
+	def get_call_type_adjustment(self, addr, arch=None):
+		if arch is None:
+			arch = self.arch
+		result = core.BNGetCallTypeAdjustment(self.handle, arch.handle, addr)
+		if result.type:
+			platform = self.platform
+			return types.Type(result.type, platform = platform, confidence = result.confidence)
+		return None
+
 	def get_call_stack_adjustment(self, addr, arch=None):
 		if arch is None:
 			arch = self.arch