kovidgoyal/kitty
1
1
Fork 0
mirror of https://github.com/kovidgoyal/kitty.git synced 2024-11-11 01:28:19 +03:00
Code Issues Projects Releases Wiki Activity

handle failures to send data over socket gracefully

Browse Source
This commit is contained in:
Kovid Goyal 2022-07-03 11:39:35 +05:30
parent 65a7ec3f85
commit 16e59784c6
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
kitty/prewarm.py
View File

@ -450,12 +450,20 @@ class SocketChild:
def handle_death(self, status: int) -> None:
if hasattr(os, 'waitstatus_to_exitcode'):
status = os.waitstatus_to_exitcode(status)
self.conn.sendall(f'{status}'.encode('ascii'))
try:
self.conn.sendall(f'{status}'.encode('ascii'))
except OSError as e:
print_error(f'Failed to send exit status of socket child with error: {e}')
self.conn.shutdown(socket.SHUT_RDWR)
self.conn.close()
def handle_creation(self) -> None:
self.conn.sendall(f'{self.pid}:'.encode('ascii'))
def handle_creation(self) -> bool:
try:
self.conn.sendall(f'{self.pid}:'.encode('ascii'))
except OSError as e:
print_error(f'Failed to send pid of socket child with error: {e}')
return False
return True
def main(stdin_fd: int, stdout_fd: int, notify_child_death_fd: int, unix_socket: socket.socket) -> None: