update babysuid - only copy target if not exists

pwncollege · Jul 22, 2024 · 38dde7b · 38dde7b
1 parent 97cb32d
commit 38dde7b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/program-misuse/babysuid b/program-misuse/babysuid
@@ -71,5 +71,6 @@ restart this challenge container to make sure that I set the SUID bit!
 
 if not os.path.exists("/challenge/bin"):
     os.makedirs("/challenge/bin")
-assert shutil.copy2(shutil.which(utility), f"/challenge/bin/{utility}")
+if not os.path.isfile(f"/challenge/bin/{utility}"):
+    assert shutil.copy2(shutil.which(utility), f"/challenge/bin/{utility}")
 os.chmod(f"/challenge/bin/{utility}", 0o4755)