mirror of
https://github.com/python/cpython.git
synced 2025-09-27 10:50:04 +00:00
[3.12] gh-113892: Add a extra check to ProactorEventLoop.sock_connect
to ensure that the given socket is in non-blocking mode (GH-119519) (#119913)
(cherry picked from commit cf3bba3f06
)
Co-authored-by: Kirill Podoprigora <kirill.bast9@mail.ru>
This commit is contained in:
parent
34a0e7a824
commit
55f65156cb
3 changed files with 12 additions and 2 deletions
|
@ -724,6 +724,8 @@ class BaseProactorEventLoop(base_events.BaseEventLoop):
|
|||
return await self._proactor.sendto(sock, data, 0, address)
|
||||
|
||||
async def sock_connect(self, sock, address):
|
||||
if self._debug and sock.gettimeout() != 0:
|
||||
raise ValueError("the socket must be non-blocking")
|
||||
return await self._proactor.connect(sock, address)
|
||||
|
||||
async def sock_accept(self, sock):
|
||||
|
|
|
@ -1006,9 +1006,9 @@ class ProactorEventLoopUnixSockSendfileTests(test_utils.TestCase):
|
|||
self.addCleanup(self.file.close)
|
||||
super().setUp()
|
||||
|
||||
def make_socket(self, cleanup=True):
|
||||
def make_socket(self, cleanup=True, blocking=False):
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
sock.setblocking(False)
|
||||
sock.setblocking(blocking)
|
||||
sock.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, 1024)
|
||||
sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 1024)
|
||||
if cleanup:
|
||||
|
@ -1070,6 +1070,11 @@ class ProactorEventLoopUnixSockSendfileTests(test_utils.TestCase):
|
|||
0, None))
|
||||
self.assertEqual(self.file.tell(), 0)
|
||||
|
||||
def test_blocking_socket(self):
|
||||
self.loop.set_debug(True)
|
||||
sock = self.make_socket(blocking=True)
|
||||
with self.assertRaisesRegex(ValueError, "must be non-blocking"):
|
||||
self.run_loop(self.loop.sock_sendfile(sock, self.file))
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
|
|
@ -0,0 +1,3 @@
|
|||
Now, the method ``sock_connect`` of :class:`asyncio.ProactorEventLoop`
|
||||
raises a :exc:`ValueError` if given socket is not in
|
||||
non-blocking mode, as well as in other loop implementations.
|
Loading…
Add table
Add a link
Reference in a new issue