Fix subprocess.close() to let its processes die gracefully

Fixes #128.

Author: 1st1

tests/test_process.py

@@ -580,6 +580,32 @@ def cancel_make_transport():
             self.loop.run_until_complete(cancel_make_transport())
             tb.run_briefly(self.loop)
 
+    def test_close_gets_process_closed(self):
+
+        loop = self.loop
+
+        class Protocol(asyncio.SubprocessProtocol):
+
+            def __init__(self):
+                self.closed = loop.create_future()
+
+            def connection_lost(self, exc):
+                self.closed.set_result(1)
+
+        @asyncio.coroutine
+        def test_subprocess():
+            transport, protocol = yield from loop.subprocess_exec(
+                Protocol, *self.PROGRAM_BLOCKED)
+            pid = transport.get_pid()
+            transport.close()
+            self.assertIsNone(transport.get_returncode())
+            yield from protocol.closed
+            self.assertIsNotNone(transport.get_returncode())
+            with self.assertRaises(ProcessLookupError):
+                os.kill(pid, 0)
+
+        loop.run_until_complete(test_subprocess())
+
 
 class Test_UV_Process(_TestProcess, tb.UVTestCase):
 

uvloop/handles/process.pxd

@@ -20,6 +20,8 @@ cdef class UVProcess(UVHandle):
         char *uv_opt_file
         bytes __cwd
 
+        bint _kill_requested
+
     cdef _init(self, Loop loop, list args, dict env, cwd,
                start_new_session,
                _stdin, _stdout, _stderr, pass_fds,

uvloop/handles/process.pyx

@@ -10,6 +10,7 @@ cdef class UVProcess(UVHandle):
         self._fds_to_close = set()
         self._preexec_fn = None
         self._restore_signals = True
+        self._kill_requested = False
 
     cdef _init(self, Loop loop, list args, dict env,
                cwd, start_new_session,
@@ -303,6 +304,8 @@ cdef class UVProcess(UVHandle):
     cdef _kill(self, int signum):
         cdef int err
         self._ensure_alive()
+        if signum in {uv.SIGKILL, uv.SIGTERM}:
+            self._kill_requested = True
         err = uv.uv_process_kill(<uv.uv_process_t*>self._handle, signum)
         if err < 0:
             raise convert_error(err)
@@ -532,6 +535,11 @@ cdef class UVProcessTransport(UVProcess):
             else:
                 self._pending_calls.append((_CALL_CONNECTION_LOST, None, None))
 
+    cdef _warn_unclosed(self):
+        if self._kill_requested:
+            return
+        super()._warn_unclosed()
+
     def __stdio_inited(self, waiter, stdio_fut):
         exc = stdio_fut.exception()
         if exc is not None:
@@ -628,7 +636,11 @@ cdef class UVProcessTransport(UVProcess):
         if self._stderr is not None:
             self._stderr.close()
 
-        self._close()
+        if self._returncode is not None:
+            # The process is dead, just close the UV handle
+            # (otherwise, the process should have been killed already
+            # and we're just waiting for a SIGCHLD.)
+            self._close()
 
     def get_extra_info(self, name, default=None):
         return default