feat: implement GetAssemblyNames, print errors on native api, fix some warnings

Author: K0lb3

TypeTreeGeneratorAPI/NativeAPI.cs

@@ -1,4 +1,5 @@
-using System.Runtime.InteropServices;
+using System.Reflection;
+using System.Runtime.InteropServices;
 using TypeTreeGeneratorAPI.TypeTreeGenerator;
 
 namespace TypeTreeGeneratorAPI
@@ -21,11 +22,11 @@ public static IntPtr TypeTreeGenerator_create(IntPtr unityVersionPtr, IntPtr gen
             }
             catch (Exception ex)
             {
-                Console.WriteLine($"Error creating TypeTreeGenerator: {ex.Message}");
+                Console.WriteLine($"Error creating TypeTreeGenerator:\n{ex.Message}");
                 // If the generator type is not recognized, we return IntPtr.Zero to indicate failure.
                 // This should be handled by the caller to avoid further issues.
                 Console.WriteLine("Failed to create TypeTreeGenerator instance. Ensure the generator name is correct and supported.");
-                    return IntPtr.Zero;
+                return IntPtr.Zero;
             }
         }
 
@@ -49,8 +50,9 @@ unsafe public static int TypeTreeGenerator_loadDLL(IntPtr typeTreeGeneratorPtr,
                 }
                 return 0;
             }
-            catch
+            catch (Exception ex)
             {
+                Console.WriteLine($"Error loading dll:\n{ex.Message}");
                 return -1;
             }
         }
@@ -82,8 +84,9 @@ unsafe public static int TypeTreeGenerator_loadIL2CPP(IntPtr typeTreeGeneratorPt
                 handle.Instance.LoadIl2Cpp(assemblyData, metadataData);
                 return 0;
             }
-            catch
+            catch (Exception ex)
             {
+                Console.WriteLine($"Error loading il2cpp:\n{ex.Message}");
                 return -1;
             }
         }
@@ -102,12 +105,30 @@ public static int TypeTreeGenerator_delete(IntPtr typeTreeGeneratorPtr)
                 gch.Free();
                 return 0;
             }
-            catch
+            catch (Exception ex)
             {
+                Console.WriteLine($"Error deleting TypeTreeGenerator:\n{ex.Message}");
                 return -1;
             }
         }
 
+        [UnmanagedCallersOnly(EntryPoint = "TypeTreeGenerator_getLoadedDLLNames")]
+        public static IntPtr TypeTreeGenerator_generateTypeTreeNodesJson(IntPtr typeTreeGeneratorPtr)
+        {
+            if (typeTreeGeneratorPtr == IntPtr.Zero)
+            {
+                return 0;
+            }
+            var handle = (TypeTreeGeneratorHandle)GCHandle.FromIntPtr(typeTreeGeneratorPtr).Target!;
+            var names = handle.Instance.GetAssemblyNames();
+            if (names == null || names.Count == 0)
+            {
+                return 0;
+            }
+            var json = string.Join(",", names.Select(name => $"\"{name}\""));
+            return Marshal.StringToCoTaskMemUTF8($"[{json}]");
+        }
+
         [UnmanagedCallersOnly(EntryPoint = "TypeTreeGenerator_generateTreeNodesJson")]
         public static int TypeTreeGenerator_generateTypeTreeNodesJson(IntPtr typeTreeGeneratorPtr, IntPtr assemblyNamePtr, IntPtr fullNamePtr, IntPtr jsonAddr)
         {
@@ -131,8 +152,9 @@ public static int TypeTreeGenerator_generateTypeTreeNodesJson(IntPtr typeTreeGen
                 Marshal.WriteIntPtr(jsonAddr, Marshal.StringToCoTaskMemUTF8(json));
                 return 0;
             }
-            catch
+            catch (Exception ex)
             {
+                Console.WriteLine($"Error generating tree nodes:\n{ex.Message}");
                 return -1;
             }
         }
@@ -162,8 +184,9 @@ public static int TypeTreeGenerator_generateTypeTreeNodesRaw(IntPtr typeTreeGene
 
                 return 0;
             }
-            catch
+            catch (Exception ex)
             {
+                Console.WriteLine($"Error generating tree nodes:\n{ex.Message}");
                 return -1;
             }
         }
@@ -211,8 +234,9 @@ public static int TypeTreeGenerator_getMonoBehaviorDefinitions(IntPtr typeTreeGe
 
                 return 0;
             }
-            catch
+            catch (Exception ex)
             {
+                Console.WriteLine($"Error geting mono behaviour definitions:\n{ex.Message}");
                 return -1;
             }
         }

TypeTreeGeneratorAPI/TypeTreeGenerator/AssetRipper/AssetRipper.cs

@@ -28,9 +28,11 @@ public AssetRipperGenerator(string unityVersionString) : base(unityVersionString
 
 
             var type = FindType(assemblyName, nameSpace, className);
-            var monoType = default(SerializableType?);
-            string? failureReason = null;
-            if (!new FieldSerializer(new UnityVersion(6000)).TryCreateSerializableType(type, monoTypeCache, out monoType, out failureReason))
+            if (type == null)
+            {
+                return null;
+            }
+            if (!new FieldSerializer(new UnityVersion(6000)).TryCreateSerializableType(type, monoTypeCache, out SerializableType? monoType, out string? failureReason))
             {
                 throw new InvalidOperationException($"Failed to create SerializableType: {failureReason}");
             }
@@ -43,6 +45,10 @@ public AssetRipperGenerator(string unityVersionString) : base(unityVersionString
             var monoBehaviourDefinitions = new List<(string, string)>();
             foreach (var assembly in assemblyDefinitions.Values)
             {
+                if (assembly.Name is null)
+                {
+                    continue;
+                }
                 foreach (var module in assembly.Modules)
                 {
                     foreach (var type in module.TopLevelTypes)
@@ -119,5 +125,10 @@ private void loadAssembly(AssemblyDefinition assembly)
 
             return null;
         }
+
+        public override List<string> GetAssemblyNames()
+        {
+            return assemblyDefinitions.Keys.ToList();
+        }
     }
 }

TypeTreeGeneratorAPI/TypeTreeGenerator/AssetStudio/AssetStudioGenerator.cs

@@ -108,5 +108,10 @@ private List<TypeTreeNode> GenerateTreeNodes(TypeDefinition typeDef)
             }
             return null;
         }
+
+        public override List<string> GetAssemblyNames()
+        {
+            return [.. moduleDic.Keys];
+        }
     }
 }

TypeTreeGeneratorAPI/TypeTreeGenerator/AssetStudio/AssetStudioUtility/TypeDefinitionConverter.cs

@@ -104,7 +104,7 @@ private FieldReference ResolveGenericFieldReference(FieldReference fieldRef)
             return TypeDef.Module.ImportReference(field);
         }
 
-        private TypeReference ResolveDeclaringType(TypeReference declaringType)
+        private TypeReference? ResolveDeclaringType(TypeReference declaringType)
         {
             var typeDefinition = declaringType.Resolve();
             if (typeDefinition == null || !typeDefinition.HasGenericParameters)

TypeTreeGeneratorAPI/TypeTreeGenerator/AssetsTools/AssetsToolsGenerator.cs

@@ -139,10 +139,12 @@ public override void LoadIl2Cpp(byte[] assemblyData, byte[] metadataData)
             }
             foreach (var asmDef in LibCpp2IlMain.TheMetadata.AssemblyDefinitions)
             {
-                if (asmDef.Image.Types == null)
+                if (asmDef.Image.Types == null || asmDef.AssemblyName.Name is null)
                     continue;
                 foreach (var asmType in asmDef.Image.Types)
                 {
+                    if (asmType.FullName is null)
+                        continue;
                     var baseType = asmType.BaseType?.baseType;
                     while (baseType != null)
                     {
@@ -179,5 +181,23 @@ private static bool IsMonoBehaviour(TypeDefinition type)
             }
             return false;
         }
+
+        public override List<string> GetAssemblyNames()
+        {
+            if (monoLoaded && monoCecilGenerator != null)
+            {
+                return monoCecilGenerator.loadedAssemblies.Keys.ToList();
+            }
+            else if (LibCpp2IlMain.TheMetadata != null)
+            {
+                return LibCpp2IlMain.TheMetadata.AssemblyDefinitions
+                    .Select(asmDef => asmDef.AssemblyName.Name)
+                    .ToList();
+            }
+            else
+            {
+                return new List<string>();
+            }
+        }
     }
 }

TypeTreeGeneratorAPI/TypeTreeGenerator/TypeTreeGenerator.cs

@@ -17,6 +17,8 @@ public TypeTreeGenerator(string unityVersionString)
         abstract public List<(string, string)> GetMonoBehaviourDefinitions();
         abstract public List<TypeTreeNode>? GenerateTreeNodes(string assemblyName, string fullName);
 
+        abstract public List<string> GetAssemblyNames();
+
         abstract public void LoadDll(Stream dllStream);
 
         virtual public void LoadDll(byte[] dll)