WinAPI Fix

BUTR · Sep 2, 2023 · 6d4b3cd · 6d4b3cd
1 parent e63dcab
commit 6d4b3cd
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/src/BUTR.Site.NexusMods.Server/Utils/MethodReplacer.cs b/src/BUTR.Site.NexusMods.Server/Utils/MethodReplacer.cs
@@ -1,4 +1,6 @@
-using System;
+using Mono.Unix.Native;
+
+using System;
 using System.Linq;
 using System.Reflection;
 using System.Runtime.CompilerServices;
@@ -44,8 +46,17 @@ public static void Replace(MethodBase source, MethodBase target)
                     0xC3  // ret
                 }).ToArray();
 
-        VirtualProtect(sourceAddress, (uint) instruction.Length, Protection.PAGE_EXECUTE_READWRITE, out var old);
+        var oldWindows = default(Protection);
+        if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
+            _ = Syscall.mprotect(sourceAddress, (ulong) instruction.Length, MmapProts.PROT_WRITE | MmapProts.PROT_READ);
+        if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
+            _ = VirtualProtect(sourceAddress, (uint) instruction.Length, Protection.PAGE_EXECUTE_READWRITE, out oldWindows);
+
         Marshal.Copy(instruction, 0, sourceAddress, instruction.Length);
-        VirtualProtect(sourceAddress, (uint) instruction.Length, old, out _);
+
+        if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
+            _ = Syscall.mprotect(sourceAddress, (ulong) instruction.Length, MmapProts.PROT_EXEC | MmapProts.PROT_READ);
+        if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
+            _ = VirtualProtect(sourceAddress, (uint) instruction.Length, oldWindows, out _);
     }
 }