support practice mode

pwncollege · Jul 12, 2024 · 3025393 · 3025393
1 parent ac0c45b
commit 3025393
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/program-misuse/babysuid b/program-misuse/babysuid
@@ -55,6 +55,8 @@ hostname = socket.gethostname()
 dojo_prefix = "program-misuse~level"
 if hostname.startswith(dojo_prefix):
     level = int(hostname[len(dojo_prefix):])
+elif hostname.startswith("practice-"+dojo_prefix):
+    level = int(hostname[len(9+dojo_prefix):])
 else:
     level = int(sys.argv[1])