Support Windows process

Author: ptr-yudai

ptrlib/connection/sock.py

@@ -300,8 +300,9 @@ def _is_alive_impl(self) -> bool:
             try:
                 self._sock.setblocking(False)
                 return self._sock.recv(1, socket.MSG_PEEK) == 1
-            except (BlockingIOError, ValueError):
+            except (BlockingIOError, ValueError, BrokenPipeError):
                 # SSLSocket may raise ValueError but we treat it as alive
+                # BrokenPipeError may happen when recv connection is closed
                 return True
             except (ConnectionResetError, socket.timeout):
                 return False

ptrlib/connection/ssh.py

@@ -2,7 +2,6 @@
 """
 import shlex
 import os
-from typing import Optional
 from ptrlib.arch.common import which
 from .proc import Process
 

ptrlib/connection/tube.py

@@ -13,7 +13,6 @@
 - Abstract methods for transport-specific implementations.
 """
 import abc
-import code
 import contextlib
 import io
 import os
@@ -105,8 +104,8 @@ def _log_error(self, message: str, *args, stacklevel: int = 2):
     # --- Properties -------------------------------------------------------
 
     @property
-    def newline(self) -> list[bytes]:
-        """List of byte sequences considered as newline terminators.
+    def newline(self) -> bytes:
+        """A byte sequence considered as newline terminators.
 
         Examples:
             ```
@@ -116,6 +115,20 @@ def newline(self) -> list[bytes]:
             sock.newline = "\\r\\n"
             ```
         """
+        return self._newline[0]
+
+    @property
+    def newlines(self) -> list[bytes]:
+        """List of byte sequences considered as newline terminators.
+
+        Examples:
+            ```
+            p = Process(["wine", "a.exe"])
+            p.newline = [b"\\n", "\\r\\n"]
+            p.newline  # b"\n"
+            p.newlines # b"\r\n"
+            ```
+        """
         return self._newline
 
     @newline.setter
@@ -302,7 +315,7 @@ def recvline(self,
             OSError: If a system error occurred.
         """
         return self.recvuntil(
-            delim=self._newline,
+            delim=self.newlines,
             blocksize=blocksize,
             regex=None,
             timeout=timeout,
@@ -583,7 +596,7 @@ def sendline(self,
             data = str(data)
 
         data = str2bytes(data)
-        return self.sendall(data + self.newline[0])
+        return self.sendall(data + self.newline)
 
     def sendkey(self):
         pass
@@ -1098,7 +1111,7 @@ class TubeTimeout(TimeoutError):
     """
     def __init__(self, message: str, *, buffered: bytes = b''):
         super().__init__(message)
-        self.buffered = buffered
+        self.buffered = bytes(buffered)
 
     def __bytes__(self) -> bytes:
         return self.buffered

ptrlib/connection/unixproc.py

@@ -90,9 +90,6 @@ def __str__(self) -> str:
     @property
     def returncode(self) -> int:
         """Get the return code of the spawned process.
-
-        Raises:
-            ChildProcessError: If the process is still running.
         """
         return self.wait()
 
@@ -286,8 +283,10 @@ def wait(self, timeout: float | None = None) -> int:
         if self._proc is None:
             return 0
         code = self._proc.wait(timeout)
-        self._is_alive = False
-        self._log_info(f"Process {str(self)} stopped with exit code {code}")
+
+        if self._is_alive:
+            self._is_alive = False
+            self._log_info(f"Process {str(self)} stopped with exit code {code}")
         return code
 
     def kill(self,