KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Better error message when socket is closed without response

Browse Source
This commit is contained in:
Kovid Goyal 2022-08-10 18:53:15 +05:30
parent fe598bac8a
commit 963d1b58f8
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
kitty/remote_control.py
View File

@ -261,11 +261,14 @@ class SocketIO:
def simple_recv(self, timeout: float) -> bytes:
dcs = re.compile(br'\x1bP@kitty-cmd([^\x1b]+)\x1b\\')
self.socket.settimeout(timeout)
st = monotonic()
with self.socket.makefile('rb') as src:
data = src.read()
m = dcs.search(data)
if m is None:
raise TimeoutError('Timed out while waiting to read cmd response')
if monotonic() - st > timeout:
raise TimeoutError('Timed out while waiting to read cmd response')
raise EOFError('Remote control connection was closed by kitty without any response being received')
return bytes(m.group(1))