idapython: add idc.GetType for function types

only supports stdcall calling convention atm

Author: williballenthin

idb/idapython.py

@@ -649,6 +649,31 @@ def RptCmt(self, ea):
     def GetCommentEx(self, ea, repeatable):
         return self.api.ida_bytes.get_cmt(ea, repeatable)
 
+    def GetType(self, ea):
+        try:
+            f = idb.analysis.Function(self.idb, ea)
+        except Exception as e:
+            logger.warning('failed to fetch function for GetType: %s', e)
+            return None
+
+        try:
+            name = f.get_name()
+            sig = f.get_signature()
+        except KeyError:
+            return None
+
+        params = []
+        for param in sig.parameters:
+            params.append('%s %s' % (param.type, param.name))
+
+        return '{rtype:s} {cc:s} {name:s}({params:s})'.format(
+            rtype=sig.rtype,
+            cc=sig.calling_convention,
+            name=name,
+            params=', '.join(params),
+        )
+
+
     @staticmethod
     def hasValue(flags):
         return flags & FLAGS.FF_IVL > 0

tests/test_idaapi.py

@@ -679,3 +679,16 @@ def test_exports(kernel32_idb, version, bitness, expected):
 
     assert api.ida_entry.get_entry_ordinal(1572) == 0x68901695
     assert api.ida_entry.get_entry_name(0x68901695) == 'DllEntryPoint'
+
+
+@kern32_test()
+def test_GetType(kernel32_idb, version, bitness, expected):
+    api = idb.IDAPython(kernel32_idb)
+    assert api.idc.GetType(0x68901695) == 'BOOL __stdcall DllEntryPoint(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpReserved)'
+
+    # valid function, but no type data associated...
+
+    #     .text:6899AE01                         ; Attributes: bp-based frame
+    #     .text:6899AE01
+    #     .text:6899AE01                         sub_6899AE01 proc near
+    assert api.idc.GetType(0x6899AE01) == None