AssemblyLoader fix (#6)

aelurum · Sep 8, 2023 · 2e10e62 · 2e10e62
1 parent e216abd
commit 2e10e62
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/AssetStudioUtility/AssemblyLoader.cs b/AssetStudioUtility/AssemblyLoader.cs
@@ -33,7 +33,12 @@ public void Load(string path)
 
         public TypeDefinition GetTypeDefinition(string assemblyName, string fullName)
         {
-            if (moduleDic.TryGetValue(assemblyName, out var module))
+            moduleDic.TryGetValue(assemblyName, out var module);
+            if (module == null && !assemblyName.Contains(".dll"))
+            {
+                moduleDic.TryGetValue(assemblyName + ".dll", out module);
+            }
+            if (module != null)
             {
                 var typeDef = module.GetType(fullName);
                 if (typeDef == null && assemblyName == "UnityEngine.dll")