NativeAPI: Expose TypeTreeGenerator_getMonoBehaviorDefinitions

Author: mos9527

.gitignore

@@ -397,4 +397,9 @@ FodyWeavers.xsd
 *.msp
 
 # JetBrains Rider
-*.sln.iml
+*.sln.iml
+
+# Built binaries
+*.dll
+*.so 
+*.dylib

TypeTreeGeneratorAPI/NativeAPI.cs

@@ -144,7 +144,7 @@ unsafe public static int TypeTreeGenerator_generateTypeTreeNodesJson(IntPtr type
 
         [UnmanagedCallersOnly(EntryPoint = "TypeTreeGenerator_generateTreeNodesRaw")]
         unsafe public static int TypeTreeGenerator_generateTypeTreeNodesRaw(IntPtr typeTreeGeneratorPtr, IntPtr assemblyNamePtr, IntPtr fullNamePtr, IntPtr arrAddrPtr, IntPtr arrLengthPtr)
-        {
+        {            
             string? assemblyName = Marshal.PtrToStringUTF8(assemblyNamePtr);
             string? fullName = Marshal.PtrToStringUTF8(fullNamePtr);
 
@@ -154,19 +154,15 @@ unsafe public static int TypeTreeGenerator_generateTypeTreeNodesRaw(IntPtr typeT
             }
             try
             {
-                var typeTreeGenerator = (TypeTreeGenerator)GCHandle.FromIntPtr(typeTreeGeneratorPtr).Target!;
-
+                var typeTreeGenerator = (TypeTreeGenerator)GCHandle.FromIntPtr(typeTreeGeneratorPtr).Target!;                
                 var typeTreeNodes = typeTreeGenerator.GenerateTreeNodes(assemblyName, fullName);
                 if (typeTreeNodes == null)
                 {
                     return -1;
                 }
                 var (arrayPtr, arrayLength) = TypeTreeNodeSerializer.ToRaw(typeTreeNodes!);
 
-                // Write the JSON pointer to the address specified by `jsonAddr`
                 Marshal.WriteIntPtr(arrAddrPtr, arrayPtr);
-
-                // Write the JSON length to the address specified by `jsonLength`
                 Marshal.WriteInt32(arrLengthPtr, arrayLength);
 
                 return 0;
@@ -177,6 +173,39 @@ unsafe public static int TypeTreeGenerator_generateTypeTreeNodesRaw(IntPtr typeT
             }
         }
 
+        [UnmanagedCallersOnly(EntryPoint = "TypeTreeGenerator_getMonoBehaviorDefinitions")]
+        unsafe public static int TypeTreeGenerator_getMonoBehaviorDefinitions(IntPtr typeTreeGeneratorPtr, IntPtr arrAddrPtr, IntPtr arrLengthPtr)
+        {            
+            if (typeTreeGeneratorPtr == IntPtr.Zero )
+            {
+                return -1;
+            }
+            try
+            {
+                var typeTreeGenerator = (TypeTreeGenerator)GCHandle.FromIntPtr(typeTreeGeneratorPtr).Target!;
+
+                var typeNames = typeTreeGenerator.GetMonoBehaviourDefinitions();
+
+                var arrayLength = typeNames.Count;
+                var arrayPtr = Marshal.AllocCoTaskMem(Marshal.SizeOf<IntPtr>() * arrayLength * 2);                
+                for (int i = 0; i < arrayLength; i++)
+                {
+                    string module = typeNames[i].Module.Name;
+                    string fullName = typeNames[i].FullName;
+                    Marshal.WriteIntPtr(arrayPtr, (i * 2) * Marshal.SizeOf<IntPtr>(), Marshal.StringToCoTaskMemAnsi(module));
+                    Marshal.WriteIntPtr(arrayPtr, (i * 2 + 1) * Marshal.SizeOf<IntPtr>(), Marshal.StringToCoTaskMemAnsi(fullName));
+                }
+            
+                Marshal.WriteIntPtr(arrAddrPtr, arrayPtr);
+                Marshal.WriteInt32(arrLengthPtr, arrayLength);
+
+                return 0;
+            }
+            catch
+            {
+                return -1;
+            }
+        }
         [UnmanagedCallersOnly(EntryPoint = "FreeCoTaskMem")]
         public static void FreeCoTaskMem(IntPtr ptr)
         {

bindings/python/TypeTreeGeneratorAPI/TypeTreeGenerator.py

@@ -1,7 +1,7 @@
 import os
 import platform
 import ctypes
-from typing import List
+from typing import List, Tuple
 
 
 class TypeTreeNodeNative(ctypes.Structure):
@@ -78,6 +78,11 @@ def init_dll():
         ctypes.POINTER(ctypes.POINTER(TypeTreeNodeNative)),
         ctypes.POINTER(ctypes.c_int),
     ]
+    dll.TypeTreeGenerator_getMonoBehaviorDefinitions.argtypes = [
+        ctypes.c_void_p,
+        ctypes.POINTER(ctypes.POINTER(ctypes.c_char_p)),
+        ctypes.POINTER(ctypes.c_int),
+    ]
     dll.TypeTreeGenerator_del.argtypes = [ctypes.c_void_p]
     dll.FreeCoTaskMem.argtypes = [ctypes.c_void_p]
     DLL = dll  # type: ignore
@@ -146,5 +151,20 @@ def get_nodes(self, assembly: str, fullname: str) -> List[TypeTreeNode]:
         DLL.FreeCoTaskMem(nodes_ptr)
         return nodes
 
+    def get_monobehavior_definitions(self) -> List[Tuple[str, str]]:
+        names_ptr = ctypes.POINTER(ctypes.c_char_p)()
+        names_cnt = ctypes.c_int()
+        assert not DLL.TypeTreeGenerator_getMonoBehaviorDefinitions(
+            self.ptr,
+            ctypes.byref(names_ptr),
+            ctypes.byref(names_cnt),
+        ), "failed to get module exports"
+        names_array = ctypes.cast(
+            names_ptr, ctypes.POINTER(ctypes.c_char_p * names_cnt.value)
+        ).contents
+        names = [name.decode("ascii") for name in names_array]
+        DLL.FreeCoTaskMem(names_ptr)
+        return [(module, fullname) for module, fullname in zip(names[::2], names[1::2])]
+
 
 __all__ = ("TypeTreeGenerator", "TypeTreeNode", "TypeTreeNodeNative")