From 8a0c912bc565cf4683c722a06ee4d54922dd0f09 Mon Sep 17 00:00:00 2001
From: Katy Coe <djkaty@start.no>
Date: Sun, 2 Feb 2020 23:28:45 +0100
Subject: [PATCH] IDA: Add custom attribute generator function names and
 addresses

---
 Il2CppDumper/Il2CppIDAScriptDumper.cs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/Il2CppDumper/Il2CppIDAScriptDumper.cs b/Il2CppDumper/Il2CppIDAScriptDumper.cs
index 36be921..05e6037 100644
--- a/Il2CppDumper/Il2CppIDAScriptDumper.cs
+++ b/Il2CppDumper/Il2CppIDAScriptDumper.cs
@@ -72,6 +72,11 @@ def SetName(addr, name):
                 writeLines($"SetName({method.VirtualAddress.Value.Start.ToAddressString()}," +
                            $"'{method.DeclaringType.Name}$${method.Name}{method.GetFullTypeParametersString()}')");
             }
+
+            foreach (var method in model.AttributesByIndices.Values.Where(m => m.VirtualAddress.HasValue)) {
+                writeLines($"SetName({method.VirtualAddress.Value.Start.ToAddressString()}," +
+                           $"'CustomAttributesGenerator${method.AttributeType.FullName}')");
+            }
         }
 
         private void writeMethods(string typeName, IEnumerable<MethodBase> methods) {