wip

Author: neek78

Lib/process_handle.py

@@ -19,6 +19,7 @@
     _mswindows = False
 else:
     _mswindows = True
+    import _winapi
 
 _USE_KQUEUE_POLLER = hasattr(select, "kqueue")
 _USE_PIDFD = hasattr(os, "pidfd_open")
@@ -192,9 +193,17 @@ class _BusywaitPoller(_Poller):
 
 
 class ProcessHandle:
+    def __init__(self, pid: int = None):
+        #FIXME:
+        #self._pid = pid
+        self._pid = None
+
     def alloc_waiter(self) -> None:
         pass
 
+    def close(self) -> none:
+        pass
+
     @property
     def pid(self) -> int:
         # if not cached, go out to the system to get pid
@@ -205,35 +214,52 @@ def pid(self) -> int:
 
 class WindowsProcessHandle(ProcessHandle):
     def __init__(self, handle: _process_support.Handle, pid: int = None):
+        super().__init__(pid)
         self._handle = handle
-        self._pid = pid
 
     def close(self) -> None:
         self._handle = None
 
     @property
     def is_closed(self):
-        return self._fd is None
+        return self._handle is None
 
     @property
     def is_definitive(self):
         return True
 
     def _get_pid(self) -> int:
-        assert False
-        pass
+        assert not self.is_closed
+        return _winapi.GetProcessId(self._handle)
 
-    def process_exists(self) -> bool:
-        return False
+    def _poll_process(self, timeout: float = None) -> None:
+        if timeout is None:
+            timeout_millis = _winapi.INFINITE
+        else:
+            assert timeout >= 0.0
+            timeout_millis = int(timeout * 1000)
 
-class PosixProcessHandle(ProcessHandle):
-    # def block_until_completed(timeout: float = None) -> None:
-    def __init__(self, pid=None):
-        self._pid = pid
+        result = _winapi.WaitForSingleObject(self._handle, timeout_millis)
+        if result == _winapi.WAIT_TIMEOUT:
+            raise TimeoutExpired(timeout)
+        elif _winapi.WAIT_OBJECT_0:
+            return 
+        else:
+            # winapi.WaitForSingleObject() throws on WAIT_FAILED, so 
+            # we shouldn't get here
+            assert False, "unexpected return from WaitForSingleObject()"
 
-    def close(self) -> none:
-        pass
+    def process_exists(self) -> bool:
+        try:
+            self._poll_process(0)
+            return False
+        except TimeoutExpired:
+            return True
 
+    def block_while_alive(self, timeout: float = None):
+        self._poll_process(timeout)
+
+class PosixProcessHandle(ProcessHandle):
 
 #FIXME: this method is redundant now
     def _do_waitid(self, id_type, identifier, wait_flags):