Added Function.type property for setting function type.

Closes tmr232#60.
shemesh999 · Jun 9, 2017 · 556743d · 556743d
1 parent 79061d3
commit 556743d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/sark/code/function.py b/sark/code/function.py
@@ -285,6 +285,16 @@ def has_name(self):
     def func_t(self):
         return self._func
 
+    @property
+    def type(self):
+        return idc.GetType(self.startEA)
+
+    @type.setter
+    def type(self, c_signature):
+        success = idc.SetType(self.startEA, c_signature)
+        if not success:
+            raise exceptions.SetTypeFailed(self.startEA, c_signature)
+
 
 def iter_function_lines(func_ea):
     """Iterate the lines of a function.

diff --git a/sark/exceptions.py b/sark/exceptions.py
@@ -13,13 +13,15 @@ class SarkNoSelection(SarkError):
 class SarkNoFunction(SarkError):
     pass
 
+
 class SarkAddFunctionFailed(SarkError):
     pass
 
 
 class SarkFunctionExists(SarkError):
     pass
 
+
 class SarkStructError(SarkError):
     pass
 
@@ -227,3 +229,8 @@ class PhraseNotSupported(PhraseError):
 class PhraseProcessorNotSupported(PhraseNotSupported):
     pass
 
+
+class SetTypeFailed(SarkError):
+    def __init__(self, ea, c_signature):
+        message = 'idc.SetType(ea={ea:08X}, "{c_signature}") failed'.format(ea=ea, c_signature=c_signature)
+        super(SetTypeFailed, self).__init__(message)