Fix exception handling in ClientHost to include OSError

pwncollege · Feb 22, 2025 · 2b850f4 · 2b850f4
1 parent b1c137f
commit 2b850f4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/intercepting-communication/level-7/run b/intercepting-communication/level-7/run
@@ -1,8 +1,6 @@
 #!/usr/bin/exec-suid --real -- /usr/bin/python -I
 
-import os
 import socket
-import subprocess
 import time
 from dojjail import Host, Network
 
@@ -17,7 +15,7 @@ class ClientHost(Host):
                 client_socket.sendall(flag.encode())
                 client_socket.close()
                 time.sleep(1)
-            except (ConnectionError, TimeoutError):
+            except (OSError, ConnectionError, TimeoutError):
                 continue
 
 user_host = Host("ip-10-0-0-1")