Add confidence levels to type objects

1 files changed, 8 insertions, 1 deletions

diff --git a/python/callingconvention.py b/python/callingconvention.py
index 4c87eef6..e6aa323c 100644
--- a/python/callingconvention.py
+++ b/python/callingconvention.py
@@ -25,6 +25,7 @@ import ctypes
 import _binaryninjacore as core
 import architecture
 import log
+import types
 
 
 class CallingConvention(object):
@@ -40,7 +41,7 @@ class CallingConvention(object):
 
 	_registered_calling_conventions = []
 
-	def __init__(self, arch, handle = None):
+	def __init__(self, arch, handle = None, confidence = types.Type.max_confidence):
 		if handle is None:
 			self.arch = arch
 			self._pending_reg_lists = {}
@@ -109,6 +110,8 @@ class CallingConvention(object):
 			else:
 				self.__dict__["float_return_reg"] = self.arch.get_reg_name(reg)
 
+		self.confidence = confidence
+
 	def __del__(self):
 		core.BNFreeCallingConvention(self.handle)
 
@@ -220,3 +223,7 @@ class CallingConvention(object):
 
 	def __str__(self):
 		return self.name
+
+	def with_confidence(self, confidence):
+		return CallingConvention(self.arch, handle = core.BNNewCallingConventionReference(self.handle),
+			confidence = confidence)