WIP

Author: moreati

mitogen/parent.py

@@ -1439,6 +1439,7 @@ def _first_stage():
             if os.uname()[0]=='Darwin'and os.uname()[2][:2]in'2021'and sys.version[:3]=='2.7':os.environ['PYTHON_LAUNCHED_FROM_WRAPPER']='1'
             os.environ['ARGV0']=sys.executable
             os.execl(sys.executable,sys.executable+'(mitogen:CONTEXT_NAME)')
+        signal.alarm(5)
         os.write(1,'MITO000\n'.encode())
         C=''.encode()
         while PREAMBLE_COMPRESSED_LEN-len(C)and select.select([0],[],[]):C+=os.read(0,PREAMBLE_COMPRESSED_LEN-len(C))
@@ -1479,11 +1480,11 @@ def get_boot_command(self):
         # Just enough to decode, decompress, and exec the first stage.
         # Priorities: wider compatibility, faster startup, shorter length.
         # `sys.path=...` for https://github.com/python/cpython/issues/115911.
-        # `import os,select` here (not stage 1) to save a few bytes overall.
+        # `import os,...` here (not stage 1) saves a few bytes overall.
         return self.get_python_argv() + [
             '-c',
             'import sys;sys.path=[p for p in sys.path if p];'
-            'import binascii,os,select,zlib;'
+            'import binascii,os,select,signal,zlib;'
             'exec(zlib.decompress(binascii.a2b_base64("%s")))' % (encoded.decode(),),
         ]