Skip to content

G
gevent
Commit ad5dd7b9 authored by Denis Bilenko's avatar Denis Bilenko
Browse files
Options

do not mask EBADF in socket, ssl, fileobject 

- update test__socket_close.py
parent 55934de4
Hide whitespace changes
Inline Side-by-side
Showing with 33 additions and 111 deletions
gevent/fileobject.py
View file @ ad5dd7b9
......@@ -115,11 +115,7 @@ else:
data = _read(self.fileno(), size)
except (IOError, OSError):
code = sys.exc_info()[1].args[0]
if code in ignored_errors:
pass
elif code == EBADF:
return ''
else:
if code not in ignored_errors:
raise
sys.exc_clear()
else:
......@@ -134,13 +130,7 @@ else:
if data.endswith('\r'):
self._eat_newline = True
return self._translate_newlines(data)
try:
self.hub.wait(self._read_event)
except IOError:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return ''
raise
self.hub.wait(self._read_event)
def _translate_newlines(self, data):
data = data.replace("\r\n", "\n")
......
gevent/socket.py
View file @ ad5dd7b9
......@@ -385,20 +385,11 @@ class socket(object):
return sock.recv(*args)
except error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return ''
if ex.args[0] != EWOULDBLOCK or self.timeout == 0.0:
raise
# QQQ without clearing exc_info test__refcount.test_clean_exit fails
sys.exc_clear()
try:
self._wait(self._read_event)
except error:
# if another greenlet called this socket, _wait would fail with EBADF
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return ''
raise
self._wait(self._read_event)
def recvfrom(self, *args):
sock = self._sock
......@@ -431,18 +422,10 @@ class socket(object):
return sock.recv_into(*args)
except error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return 0
if ex.args[0] != EWOULDBLOCK or self.timeout == 0.0:
raise
sys.exc_clear()
try:
self._wait(self._read_event)
except error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return 0
raise
self._wait(self._read_event)
def send(self, data, flags=0, timeout=timeout_default):
sock = self._sock
......@@ -455,13 +438,7 @@ class socket(object):
if ex.args[0] != EWOULDBLOCK or timeout == 0.0:
raise
sys.exc_clear()
try:
self._wait(self._write_event)
except error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return 0
raise
self._wait(self._write_event)
try:
return sock.send(data, flags)
except error:
......
gevent/ssl.py
View file @ ad5dd7b9
......@@ -116,25 +116,13 @@ class SSLSocket(socket):
if self.timeout == 0.0:
raise
sys.exc_clear()
try:
self._wait(self._read_event, timeout_exc=_SSLErrorReadTimeout)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return ''
raise
self._wait(self._read_event, timeout_exc=_SSLErrorReadTimeout)
elif ex.args[0] == SSL_ERROR_WANT_WRITE:
if self.timeout == 0.0:
raise
sys.exc_clear()
try:
# note: using _SSLErrorReadTimeout rather than _SSLErrorWriteTimeout below is intentional
self._wait(self._write_event, timeout_exc=_SSLErrorReadTimeout)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return ''
raise
# note: using _SSLErrorReadTimeout rather than _SSLErrorWriteTimeout below is intentional
self._wait(self._write_event, timeout_exc=_SSLErrorReadTimeout)
else:
raise
......@@ -150,24 +138,12 @@ class SSLSocket(socket):
if self.timeout == 0.0:
raise
sys.exc_clear()
try:
self._wait(self._read_event, timeout_exc=_SSLErrorWriteTimeout)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return 0
raise
self._wait(self._read_event, timeout_exc=_SSLErrorWriteTimeout)
elif ex.args[0] == SSL_ERROR_WANT_WRITE:
if self.timeout == 0.0:
raise
sys.exc_clear()
try:
self._wait(self._write_event, timeout_exc=_SSLErrorWriteTimeout)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return 0
raise
self._wait(self._write_event, timeout_exc=_SSLErrorWriteTimeout)
else:
raise
......@@ -201,24 +177,12 @@ class SSLSocket(socket):
if self.timeout == 0.0:
return 0
sys.exc_clear()
try:
self._wait(self._read_event)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return 0
raise
self._wait(self._read_event)
elif x.args[0] == SSL_ERROR_WANT_WRITE:
if self.timeout == 0.0:
return 0
sys.exc_clear()
try:
self._wait(self._write_event)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return 0
raise
self._wait(self._write_event)
else:
raise
else:
......@@ -267,13 +231,7 @@ class SSLSocket(socket):
if self.timeout == 0.0:
raise
sys.exc_clear()
try:
self._wait(self._read_event)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return 0
raise
self._wait(self._read_event)
continue
else:
raise
......@@ -312,24 +270,12 @@ class SSLSocket(socket):
if self.timeout == 0.0:
raise
sys.exc_clear()
try:
self._wait(self._read_event, timeout_exc=_SSLErrorReadTimeout)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return ''
raise
self._wait(self._read_event, timeout_exc=_SSLErrorReadTimeout)
elif ex.args[0] == SSL_ERROR_WANT_WRITE:
if self.timeout == 0.0:
raise
sys.exc_clear()
try:
self._wait(self._write_event, timeout_exc=_SSLErrorWriteTimeout)
except socket_error:
ex = sys.exc_info()[1]
if ex.args[0] == EBADF:
return ''
raise
self._wait(self._write_event, timeout_exc=_SSLErrorWriteTimeout)
else:
raise
......
greentest/test__socket_close.py
View file @ ad5dd7b9
......@@ -13,6 +13,8 @@ def readall(socket, address):
class Test(greentest.TestCase):
error_fatal = False
def setUp(self):
self.server = server.StreamServer(('', 0), readall)
self.server.start()
......@@ -24,20 +26,27 @@ class Test(greentest.TestCase):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('127.0.0.1', self.server.server_port))
receiver = gevent.spawn(sock.recv, 25)
gevent.sleep(0.001)
sock.close()
receiver.join(timeout=0.1)
assert receiver.ready(), receiver
self.assertEqual(receiver.value, '')
try:
gevent.sleep(0.001)
sock.close()
receiver.join(timeout=0.1)
assert receiver.ready(), receiver
self.assertEqual(receiver.value, None)
assert isinstance(receiver.exception, socket.error)
self.assertEqual(receiver.exception.errno, socket.EBADF)
finally:
receiver.kill()
def test_recv_twice(self):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('127.0.0.1', self.server.server_port))
receiver = gevent.spawn(sock.recv, 25)
gevent.sleep(0.001)
self.assertRaises(AssertionError, sock.recv, 25)
self.assertRaises(AssertionError, sock.recv, 25)
receiver.kill()
try:
gevent.sleep(0.001)
self.assertRaises(AssertionError, sock.recv, 25)
self.assertRaises(AssertionError, sock.recv, 25)
finally:
receiver.kill()
if __name__ == '__main__':
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment
GitLab Nexedi Edition | About GitLab | About Nexedi | 沪ICP备2021021310号-2 | 沪ICP备2021021310号-7