Challenge: Fix VM build

pwncollege · Sep 9, 2024 · 9e4d2f7 · 9e4d2f7
1 parent b06852c
commit 9e4d2f7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/challenge/vm/vm b/challenge/vm/vm
@@ -204,7 +204,8 @@ def debug():
 
 
 def build(path):
-    os.setuid(os.geteuid())
+    ruid, euid, suid = os.getresuid()
+    os.seteuid(ruid)
 
     with open(path, "r") as f:
         src = f.read()
@@ -229,6 +230,7 @@ def build(path):
 
         subprocess.run(["make", "-C", workdir], stdout=sys.stderr, check=True)
 
+        os.seteuid(euid)
         shutil.copy(f"{workdir}/debug.ko", "/challenge/debug.ko")