Upgrade for nix PATH changes

pwncollege · Jul 11, 2024 · f31c6b7 · f31c6b7
1 parent d43ae74
commit f31c6b7
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/access-control/.init b/access-control/.init
@@ -6,3 +6,8 @@ for i in $NEED_SETUID
 do
     chmod u+s "$i"
 done
+
+if [ -e /challenge/.path ]; then
+    mkdir -p /challenge/bin
+    ln -sf "$(cat /challenge/.path)" /challenge/bin/
+fi
diff --git a/access-control/level-4/.path b/access-control/level-4/.path
@@ -0,0 +1 @@
+/bin/cat
diff --git a/access-control/level-5/.path b/access-control/level-5/.path
@@ -0,0 +1 @@
+/bin/cp
diff --git a/access-control/run b/access-control/run
@@ -236,10 +236,7 @@ def level4():
     print("Before:")
     os.system(f"/bin/ls -la /bin/cat")
 
-    # create a symlink
     os.chmod("/bin/cat", 0o4755)
-    os.mkdir("/challenge/bin")
-    os.symlink("/bin/cat", "/challenge/bin/cat")
 
     print("After:")
     os.system(f"/bin/ls -la /bin/cat")
@@ -259,8 +256,6 @@ def level5():
     os.system(f"/bin/ls -la /bin/cp")
 
     os.chmod("/bin/cp", 0o4755)
-    os.mkdir("/challenge/bin")
-    os.symlink("/bin/cp", "/challenge/bin/cp")
 
     print("After:")
     os.system(f"/bin/ls -la /bin/cp")