Fix key error in KFA location

SamboyCoding · Jul 10, 2020 · c10c712 · c10c712
1 parent 8a50c56
commit c10c712
Showing 1 changed file with 8 additions and 4 deletions.
diff --git 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.");
         }
     }
-}
+}