From c10c712c71bccb9eb54c30d134f9cc89a9887e6a Mon Sep 17 00:00:00 2001
From: Sam Byass <SamboyCoding@users.noreply.github.com>
Date: Sat, 11 Jul 2020 00:55:23 +0100
Subject: [PATCH] Fix key error in KFA location

---
 Cpp2IL/KeyFunctionAddresses.cs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/Cpp2IL/KeyFunctionAddresses.cs b/Cpp2IL/KeyFunctionAddresses.cs
index e78d9993..5fbfd332 100644
--- a/Cpp2IL/KeyFunctionAddresses.cs
+++ b/Cpp2IL/KeyFunctionAddresses.cs
@@ -344,9 +344,13 @@ private static void TryUseArrayEnumerator(List<(TypeDefinition type, List<CppMet
                     var potentialRME = callTargets[6];
 
                     //Firstly, sanity checks on the three known addresses we have
-                    var shouldBeO_GT = SharedState.MethodsByAddress[shouldPointToObjectGetType];
-                    var shouldBeT_GIP = SharedState.MethodsByAddress[shouldPointToTypeGetIsPointer];
-                    var shouldBeIOE_C = SharedState.MethodsByAddress[shouldPointToIOECtor];
+                    SharedState.MethodsByAddress.TryGetValue(shouldPointToObjectGetType, out var shouldBeO_GT);
+                    SharedState.MethodsByAddress.TryGetValue(shouldPointToTypeGetIsPointer, out var shouldBeT_GIP);
+                    SharedState.MethodsByAddress.TryGetValue(shouldPointToIOECtor, out var shouldBeIOE_C);
+                    
+//                     var shouldBeO_GT = SharedState.MethodsByAddress[shouldPointToObjectGetType];
+//                     var shouldBeT_GIP = SharedState.MethodsByAddress[shouldPointToTypeGetIsPointer];
+//                     var shouldBeIOE_C = SharedState.MethodsByAddress[shouldPointToIOECtor];
 
                     if (shouldBeO_GT?.Name == "GetType" && shouldBeT_GIP?.Name == "get_IsPointer" && shouldBeIOE_C?.Name == ".ctor")
                     {
@@ -368,4 +372,4 @@ private static void TryUseArrayEnumerator(List<(TypeDefinition type, List<CppMet
                 Console.WriteLine("\t\t\t\tType not present, not using.");
         }
     }
-}
\ No newline at end of file
+}