From 8ffff2407948d4a3508d6f1c98444eb42e1e9d9e Mon Sep 17 00:00:00 2001
From: Katy Coe <djkaty@users.noreply.github.com>
Date: Sun, 10 Jan 2021 08:26:12 +0100
Subject: [PATCH] DLL: Add assembly and module tokens

---
 Il2CppInspector.Common/Outputs/AssemblyShims.cs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Il2CppInspector.Common/Outputs/AssemblyShims.cs b/Il2CppInspector.Common/Outputs/AssemblyShims.cs
index ca67ddf..3bd8d51 100644
--- a/Il2CppInspector.Common/Outputs/AssemblyShims.cs
+++ b/Il2CppInspector.Common/Outputs/AssemblyShims.cs
@@ -486,6 +486,10 @@ namespace Il2CppInspector.Outputs
                 // Create assembly and add primary module to list
                 var module = CreateAssembly(asm.ShortName);
                 modules.Add(asm, module);
+
+                // Add token attributes
+                module.AddAttribute(module, tokenAttribute, ("Token", $"0x{asm.ImageDefinition.token:X8}"));
+                module.Assembly.AddAttribute(module, tokenAttribute, ("Token", $"0x{asm.AssemblyDefinition.token:X8}"));
             }
 
             // Add custom attribute attributes (must do this after all assemblies are created due to type referencing)