LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-23 01:46:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Treat OSError.errno == EBADF as a ConnectionResetError

Browse Source
This commit is contained in:
Lonami Exo 2017-09-03 11:54:26 +02:00
parent f8f389c850
commit 68cdc9eaf1
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
telethon/extensions/tcp_client.py
View File

@ -71,9 +71,13 @@ class TcpClient:
self._socket.sendall(data)
except socket.timeout as e:
raise TimeoutError() from e
except BrokenPipeError:
self.close()
except OSError as e:
if e.errno == errno.EBADF:
self._raise_connection_reset()
else:
raise
except BrokenPipeError:
self._raise_connection_reset()
def read(self, size):
"""Reads (receives) a whole block of 'size bytes
@ -92,11 +96,14 @@ class TcpClient:
partial = self._socket.recv(bytes_left)
except socket.timeout as e:
raise TimeoutError() from e
except OSError as e:
if e.errno == errno.EBADF:
self._raise_connection_reset()
else:
raise
if len(partial) == 0:
self.close()
raise ConnectionResetError(
'The server has closed the connection.')
self._raise_connection_reset()
buffer.write(partial)
bytes_left -= len(partial)
@ -104,3 +111,7 @@ class TcpClient:
# If everything went fine, return the read bytes
buffer.flush()
return buffer.raw.getvalue()
def _raise_connection_reset(self):
self.close() # Connection reset -> flag as socket closed
raise ConnectionResetError('The server has closed the connection.')