From 54b0dd360b7a014ba5c4d60afc52ae4c9324dad0 Mon Sep 17 00:00:00 2001
From: Meivyn <793322+Meivyn@users.noreply.github.com>
Date: Sun, 10 Sep 2023 20:51:24 -0400
Subject: [PATCH] Prevent file corruption

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

diff --git a/IPA.Injector/Injector.cs b/IPA.Injector/Injector.cs
index 75ba1d8d..e78b10c3 100644
--- a/IPA.Injector/Injector.cs
+++ b/IPA.Injector/Injector.cs
@@ -247,8 +247,11 @@ namespace IPA.Injector
 
                 if (modified)
                 {
+                    string tempFilePath = Path.GetTempFileName();
                     bkp?.Add(unityPath);
-                    unityAsmDef.Write(unityPath);
+                    unityAsmDef.Write(tempFilePath);
+                    File.Delete(unityPath);
+                    File.Move(tempFilePath, unityPath);
                 }
             }
             endPatchCoreModule:
@@ -279,7 +282,10 @@ namespace IPA.Injector
                         var deleter = ascModDef.GetType("IPAPluginsDirDeleter");
                         deleter.Methods.Clear(); // delete all methods
 
-                        ascAsmDef.Write(ascPath);
+                        string tempFilePath = Path.GetTempFileName();
+                        ascAsmDef.Write(tempFilePath);
+                        File.Delete(ascPath);
+                        File.Move(tempFilePath, ascPath);
 
                         isFirst = false;
                     }
@@ -333,4 +339,4 @@ namespace IPA.Injector
             _ = PluginComponent.Create();
         }
     }
-}
\ No newline at end of file
+}