handle timeouts during dockerized verification

pwncollege · Nov 13, 2024 · fd3dceb · fd3dceb
1 parent 5c47170
commit fd3dceb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pwnshop/challenge.py b/pwnshop/challenge.py
@@ -283,11 +283,12 @@ def run_challenge(
 
         if self._verify_container:
             docker_cmd = "docker exec -u root -i".split()
-            for k,v in kwargs.get("env", {}).items():
+            for k,v in kwargs.pop("env", {}).items():
                 docker_cmd += [ "-e", f"{k}={v.decode() if type(v) is bytes else v}" ]
-            kwargs.pop("env", None)
             docker_cmd += [ "-w", self.work_dir ]
             docker_cmd.append(self._verify_container.name)
+            if alarm := kwargs.pop("alarm", None):
+                docker_cmd += [ "timeout", "-sALRM", str(alarm) ]
             argv = docker_cmd + argv
 
         with pwnlib.tubes.process.process(