Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-12-08 06:10:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

add Overlapped.getresultex to implement timeout in sendfile

Browse source
This commit is contained in:
AN Long 2023-11-23 22:21:22 +08:00
parent 490df91637
commit f94935df16
3 changed files with 116 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

13
Lib/socket.py
View file

@ -461,7 +461,8 @@ def _sendfile_use_send(self, file, offset=0, count=None):
if _overlapped:
def _sendfile_use_transmitfile(self, file, offset=0, count=None):
self._check_sendfile_params(file, offset, count)
if self.gettimeout() == 0:
timeout = self.gettimeout()
if timeout == 0:
raise ValueError("non-blocking sockets are not supported")
ov = _overlapped.Overlapped()
offset_low = offset & 0xffff_ffff
@ -469,7 +470,15 @@ def _sendfile_use_transmitfile(self, file, offset=0, count=None):
count = count or 0
ov.TransmitFile(self.fileno(), msvcrt.get_osfhandle(file.fileno()),
offset_low, offset_high, count, 0, 0)
sent = ov.getresult(True)
timeout_ms = _overlapped.INFINITE
if timeout is not None:
timeout_ms = int(timeout * 1000)
try:
sent = ov.getresultex(timeout_ms, False)
except WindowsError as e:
if e.winerror == 258:
raise TimeoutError('timed out')
raise
if sent > 0 and hasattr(file, 'seek'):
file.seek(offset + sent)
return sent