runall.py

@@ -1,31 +1,26 @@
 from itertools import product
-import signal
+import subprocess
-from subprocess import DEVNULL, PIPE, Popen, TimeoutExpired
 
 chars = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!"#$%&\'()*+,-./:;<=>?@[\\]^_`{|}~ '
 
-# Prevent zombie processes
-signal.signal(signal.SIGCHLD, signal.SIG_IGN)
-
 
 def check_pair(script, instr, outstr):
-    process = Popen(
+    try:
-        ["/usr/bin/env", "bash", "-c", "--", script],
+        result = subprocess.run(
-        stdin=PIPE,
+            [
-        stdout=PIPE,
+                "/usr/bin/env", "bash", "-c",
-        stderr=DEVNULL,
+                ";\n".join((
+                    # Cleanup child processes on exit
+                    "trap 'kill -9 $(jobs -p)' SIGINT SIGTERM EXIT",
+                    script,
+                ))
+            ],
+            timeout=5,
+            input=instr.encode(),
+            capture_output=True,
         )
-
+        return result.returncode == 0 and result.stdout == outstr.encode()
-    try:
+    except subprocess.TimeoutExpired:
-        stdout, stderr = process.communicate(instr.encode(), timeout=5)
-        return process.returncode == 0 and stdout == outstr.encode()
-    except TimeoutExpired:
-        try:
-            process.kill()
-            stdout, stderr = process.communicate()
-        except ProcessLookupError:
-            pass
-
         return False
 
 
@@ -63,8 +58,8 @@ if __name__ == '__main__':
     print("\nSearching for identity")
     find_script((
         ("1", "1"),
-        ("42", "42"),
+        ("2", "2"),
-        ("1984", "1984"),
+        ("3", "3"),
     ), max_length=3)
 
     # Successor