From 905cd6113e643e6d04cc5eafba055a61aff30c84 Mon Sep 17 00:00:00 2001
From: Anairkoen Schno <aaron.cirr.com@gmail.com>
Date: Thu, 5 Dec 2019 15:20:47 -0600
Subject: [PATCH] Added safer virtualization and anti-yeet support

---
 IPA.Injector/Injector.cs | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/IPA.Injector/Injector.cs b/IPA.Injector/Injector.cs
index 128390a7..83f6a616 100644
--- a/IPA.Injector/Injector.cs
+++ b/IPA.Injector/Injector.cs
@@ -237,15 +237,23 @@ namespace IPA.Injector
             
             {
                 var ascPath = Path.Combine(managedPath,
-                    "Assembly-CSharp.dll");
+                    "MainAssembly.dll"); // TODO: change to config option for other games
 
                 #region Virtualize Assembly-CSharp.dll
 
                 {
                     CriticalSection.EnterExecuteSection();
 
-                    var ascModule = VirtualizedModule.Load(ascPath);
-                    ascModule.Virtualize(cAsmName, () => bkp?.Add(ascPath));
+                    try
+                    {
+                        var ascModule = VirtualizedModule.Load(ascPath);
+                        ascModule.Virtualize(cAsmName, () => bkp?.Add(ascPath));
+                    }
+                    catch (Exception e) 
+                    {
+                        loader.Error($"Could not virtualize {ascPath}");
+                        loader.Error(e);
+                    }
 
                     CriticalSection.ExitExecuteSection();
                 }