diff options
| author | Alex Grönholm <alex.gronholm@nextday.fi> | 2022-03-13 18:42:29 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-03-13 18:42:29 +0200 |
| commit | 9f04ee569cebb8b4c6f04bea95d91a19c5403806 (patch) | |
| tree | abd4626ee19d394f35109b4cd2830d50a4a4f9d9 /Lib/asyncio/windows_events.py | |
| parent | 7e473e94a52024ac821dd2f206290423e4987ead (diff) | |
| download | cpython-9f04ee569cebb8b4c6f04bea95d91a19c5403806.tar.gz cpython-9f04ee569cebb8b4c6f04bea95d91a19c5403806.zip | |
bpo-46805: Add low level UDP socket functions to asyncio (GH-31455)
Diffstat (limited to 'Lib/asyncio/windows_events.py')
| -rw-r--r-- | Lib/asyncio/windows_events.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/Lib/asyncio/windows_events.py b/Lib/asyncio/windows_events.py index 0d9a07ef477..90b259cbafe 100644 --- a/Lib/asyncio/windows_events.py +++ b/Lib/asyncio/windows_events.py @@ -512,6 +512,26 @@ class IocpProactor: return self._register(ov, conn, finish_recv) + def recvfrom_into(self, conn, buf, flags=0): + self._register_with_iocp(conn) + ov = _overlapped.Overlapped(NULL) + try: + ov.WSARecvFromInto(conn.fileno(), buf, flags) + except BrokenPipeError: + return self._result((0, None)) + + def finish_recv(trans, key, ov): + try: + return ov.getresult() + except OSError as exc: + if exc.winerror in (_overlapped.ERROR_NETNAME_DELETED, + _overlapped.ERROR_OPERATION_ABORTED): + raise ConnectionResetError(*exc.args) + else: + raise + + return self._register(ov, conn, finish_recv) + def sendto(self, conn, buf, flags=0, addr=None): self._register_with_iocp(conn) ov = _overlapped.Overlapped(NULL) |