From 0077803164e0bf4313ade5946544263cccd11482 Mon Sep 17 00:00:00 2001
From: Anairkoen Schno <aaron.cirr.com@gmail.com>
Date: Sun, 4 Nov 2018 22:35:32 -0600
Subject: [PATCH] Added self-protection to updater

---
 IPA.Injector/Updates.cs | 64 ++++++++++++++++++++++++-----------------
 1 file changed, 38 insertions(+), 26 deletions(-)

diff --git a/IPA.Injector/Updates.cs b/IPA.Injector/Updates.cs
index ab7bc6c1..bde6f240 100644
--- a/IPA.Injector/Updates.cs
+++ b/IPA.Injector/Updates.cs
@@ -11,43 +11,55 @@ namespace IPA.Injector
         public static void InstallPendingUpdates()
         {
             var pendingDir = Path.Combine(BeatSaber.InstallPath, "IPA", "Pending");
-            if (Directory.Exists(pendingDir))
-            { // there are pending updates, install
-                updater.Info("Installing pending updates");
+            if (!Directory.Exists(pendingDir)) return; 
+            
+            // there are pending updates, install
+            updater.Info("Installing pending updates");
 
-                var toDelete = new string[0];
-                var delFn = Path.Combine(pendingDir, DeleteFileName);
-                if (File.Exists(delFn))
-                {
-                    toDelete = File.ReadAllLines(delFn);
-                    File.Delete(delFn);
-                }
-
-                foreach (var file in toDelete)
-                {
-                    try
-                    {
-                        File.Delete(Path.Combine(BeatSaber.InstallPath, file));
-                    }
-                    catch (Exception e)
-                    {
-                        updater.Error("While trying to install pending updates: Error deleting file marked for deletion");
-                        updater.Error(e);
-                    }
-                }
+            var toDelete = new string[0];
+            var delFn = Path.Combine(pendingDir, DeleteFileName);
+            if (File.Exists(delFn))
+            {
+                toDelete = File.ReadAllLines(delFn);
+                File.Delete(delFn);
+            }
 
+            foreach (var file in toDelete)
+            {
                 try
                 {
-                    LoneFunctions.CopyAll(new DirectoryInfo(pendingDir), new DirectoryInfo(BeatSaber.InstallPath));
+                    File.Delete(Path.Combine(BeatSaber.InstallPath, file));
                 }
                 catch (Exception e)
                 {
-                    updater.Error("While trying to install pending updates: Error copying files in");
+                    updater.Error("While trying to install pending updates: Error deleting file marked for deletion");
                     updater.Error(e);
                 }
+            }
+
+            #region Self Protection
+            if (Directory.Exists(Path.Combine(pendingDir, "IPA")))
+                Directory.Delete(Path.Combine(pendingDir, "IPA"), true);
+            if (File.Exists(Path.Combine(pendingDir, "IPA.exe")))
+            {
+                File.Delete(Path.Combine(pendingDir, "IPA.exe"));
+                if (File.Exists(Path.Combine(pendingDir, "Mono.Cecil.dll")))
+                    File.Delete(Path.Combine(pendingDir, "Mono.Cecil.dll"));
+            }
 
-                Directory.Delete(pendingDir, true);
+            #endregion
+
+            try
+            {
+                LoneFunctions.CopyAll(new DirectoryInfo(pendingDir), new DirectoryInfo(BeatSaber.InstallPath));
+            }
+            catch (Exception e)
+            {
+                updater.Error("While trying to install pending updates: Error copying files in");
+                updater.Error(e);
             }
+
+            Directory.Delete(pendingDir, true);
         }
     }
 }