From 9eb11731806fdd15ee808207039c463e9a4e617d Mon Sep 17 00:00:00 2001
From: Zero <zero@cock.li>
Date: Tue, 18 Jul 2023 19:12:05 -0400
Subject: [PATCH] Add -restore switch

When Windows updates, this program tries to restore the original
uDWM_win11drc.bak to uDWM.dll even though it should't, users have
to manually delete this file to get rid of rounded corners again.

See: https://github.com/valinet/Win11DisableRoundedCorners/issues/35
---
 Win11DisableRoundedCorners/Win11DisableRoundedCorners.c | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/Win11DisableRoundedCorners/Win11DisableRoundedCorners.c b/Win11DisableRoundedCorners/Win11DisableRoundedCorners.c
index a2aa088..ded8753 100644
--- a/Win11DisableRoundedCorners/Win11DisableRoundedCorners.c
+++ b/Win11DisableRoundedCorners/Win11DisableRoundedCorners.c
@@ -8,6 +8,11 @@ int main(int argc, char** argv)
 {
     BOOL bRestore = FALSE;
 
+    if (argc == 2 && strcmp(argv[1], "-restore") == 0)
+    {
+        bRestore = TRUE;
+    }
+
     char szTaskkill[MAX_PATH];
     ZeroMemory(
         szTaskkill,
@@ -50,7 +55,6 @@ int main(int argc, char** argv)
         MAX_PATH,
         "\\uDWM_win11drc.bak"
     );
-    bRestore = fileExists(szOriginalDWM);
 
     char szModifiedDWM[_MAX_PATH];
     ZeroMemory(
@@ -290,4 +294,4 @@ int main(int argc, char** argv)
     }
     printf("Operation successful.\n");
 	return 0;
-}
\ No newline at end of file
+}