clean up containers

pwncollege · Nov 14, 2024 · 255eb3d · 255eb3d
1 parent 6007e00
commit 255eb3d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pwnshop/challenge.py b/pwnshop/challenge.py
@@ -110,6 +110,12 @@ def __init_subclass__(cls, register=True):
         if register and getattr(cls_module, "PWNSHOP_AUTOREGISTER", True):
             register_challenge(cls)
 
+    def __del__(self):
+        if self._build_container:
+            self._build_container.kill()
+        if self._verify_container:
+            self._verify_container.kill()
+
     @property
     def TEMPLATE_PATH(self):
         raise NotImplementedError()