KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests 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

10
kitty/prewarm.py
View File

@ -450,12 +450,20 @@ class SocketChild:
def handle_death(self, status: int) -> None: def handle_death(self, status: int) -> None:
if hasattr(os, 'waitstatus_to_exitcode'): if hasattr(os, 'waitstatus_to_exitcode'):
status = os.waitstatus_to_exitcode(status) status = os.waitstatus_to_exitcode(status)
try:
self.conn.sendall(f'{status}'.encode('ascii')) 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.shutdown(socket.SHUT_RDWR)
self.conn.close() self.conn.close()
def handle_creation(self) -> None: def handle_creation(self) -> bool:
try:
self.conn.sendall(f'{self.pid}:'.encode('ascii')) 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: def main(stdin_fd: int, stdout_fd: int, notify_child_death_fd: int, unix_socket: socket.socket) -> None: