Assign method metadata tokens before writing (#180)

BepInEx · Sep 29, 2024 · db5837f · db5837f
1 parent d2cd4f7
commit db5837f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Il2CppInterop.Generator/Passes/Pass90WriteToDisk.cs b/Il2CppInterop.Generator/Passes/Pass90WriteToDisk.cs
@@ -49,12 +49,13 @@ public static void DoPass(RewriteGlobalContext context, GeneratorOptions options
                 asmContext.NewAssembly.CustomAttributes.Add(targetFrameworkAttribute);
             }
 
-            // Optimize macros in all methods.
+            // Optimize macros in all methods and assign tokens.
             foreach (var type in module.GetAllTypes())
             {
                 foreach (var method in type.Methods)
                 {
                     method.CilMethodBody?.Instructions.OptimizeMacros();
+                    module.TokenAllocator.AssignNextAvailableToken(method);
                 }
             }
         }