Fix registry permission set

PenguLoader · Aug 10, 2023 · 6e3d4fe · 6e3d4fe
1 parent 3770692
commit 6e3d4fe
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/loader/Main/IFEO.cs b/loader/Main/IFEO.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Security.AccessControl;
 using Microsoft.Win32;
 
 namespace PenguLoader.Main
@@ -25,10 +26,22 @@ public static bool SetDebugger(string target, string debugger)
             {
                 if (key == null) return false;
 
-                using (var image = key.CreateSubKey(target))
+                using (var image = key.CreateSubKey(target, RegistryKeyPermissionCheck.ReadWriteSubTree))
                 {
                     if (image == null) return false;
 
+                    try
+                    {
+                        var user = Environment.UserDomainName + "\\" + Environment.UserName;
+                        RegistryAccessRule rule = new RegistryAccessRule(user, RegistryRights.FullControl, AccessControlType.Allow);
+                        RegistrySecurity security = new RegistrySecurity();
+                        security.AddAccessRule(rule);
+                        image.SetAccessControl(security);
+                    }
+                    catch
+                    {
+                    }
+
                     image.SetValue(VALUE_NAME, debugger, RegistryValueKind.String);
                     return true;
                 }
@@ -49,6 +62,9 @@ public static void RemoveDebugger(string target)
             }
         }
 
-        private static RegistryKey OpenIFEOKey(bool writable = false) => Registry.LocalMachine.OpenSubKey(IFEO_PATH, writable);
+        static RegistryKey OpenIFEOKey(bool writable = false)
+        {
+            return Registry.LocalMachine.OpenSubKey(IFEO_PATH, writable);
+        }
     }
 }