U
|
Z ±dj ã @ sÖ d dl Z d dlmZ d dlZd dlZd dlZd dlmZ d dlZd dl m
|
Z
|
ddl m Z ddl mZ dd lmZ dd
|
lmZmZmZmZmZmZmZmZmZmZm Z m Z m Z er¾ej d ks¾t G d d d ej!Z"dd Z#e j$dddZ%dd Z&dd Z'ej(ej)B ej*B ej+B ej,B Z-ej.ej/B ej+B ej,B Z0ej1dddG dd dZ2ej1dddd G d d d Z3d Z4ej1dddG d d! d!Z5ej1dddd G d"d# d#Z6d$Z7ej1dd%G d&d' d'Z8G d(d) d)Z9dS )*é N)Úcontextmanager)Ú TYPE_CHECKING)ÚValueé )Ú_coreé )Ú_public)Úwake_all) ÚffiÚkernel32ÚntdllÚws2_32ÚINVALID_HANDLE_VALUEÚraise_winerrorÚ_handleÚ
|
ErrorCodesÚ FileFlagsÚ AFDPollFlagsÚ WSAIoctlsÚCompletionModesÚIoControlCodesÚwin32c @ s e Zd ZdZdZdZdZdZdS )ÚCKeysr r r é é N)Ú__name__Ú
|
__module__Ú __qualname__ÚAFD_POLLÚWAIT_OVERLAPPEDÚ LATE_CANCELÚ FORCE_WAKEUPÚ USER_DEFINED© r# r# úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/_io_windows.pyr ¯ s
|
r c C s | s
|
t | S ©N)r )Úsuccessr# r# r$ Ú_check· s r' ©Úwhichc C sp t | dr| ¡ } t d¡}t d¡}t t d| ¡|tjd|t d¡|tjtj¡ }|rht ¡ }t
|
| |d S )NÚfilenozHANDLE *zDWORD *ZSOCKETr ÚHANDLE) Úhasattrr* r
|
Únewr ZWSAIoctlÚcastÚNULLÚsizeofZWSAGetLastErrorr )Úsockr) Zbase_ptrZout_sizeÚfailedÚcoder# r# r$ Ú_get_underlying_socket½ s&
|
|
|
|
÷ r4 c
|
C s~ z
|
t | W S tk
|
rv } zN|jtjkr* t| dr<| ¡ } t| } t | tj d}|| krbt
|
d|} W 5 d }~X Y q X q d S )Nr* r( zþUnexpected network configuration detected: SIO_BASE_HANDLE failed and SIO_BSP_HANDLE_POLL didn't return a different socket. Please file a bug at https://github.com/python-trio/trio/issues/new, and include the output of running: netsh winsock show catalog) r4 ÚOSErrorÚwinerrorr ZERROR_NOT_SOCKETr, r* r r ZSIO_BSP_HANDLE_POLLÚ RuntimeError)r1 ÚexZ next_sockr# r# r$ Ú_get_base_socketÓ s"
|
|
ÿÿ
|
r9 c C sZ d d¡d } t | ¡}t t d|¡tjtjtj B tj
|
tj tj tj
|
¡}|t krVt |S )Nz \\.\GLOBALROOT\Device\Afd\Triozutf-16les ÚLPCWSTR)Úencoder
|
Ú from_bufferr Z CreateFileWr. r Ú SYNCHRONIZEZFILE_SHARE_READZFILE_SHARE_WRITEr/ Ú OPEN_EXISTINGÚFILE_FLAG_OVERLAPPEDr r )ZrawnameZ rawname_bufÚhandler# r# r$ Ú_afd_helper_handle s
|
|
|
ù rA TF)ÚslotsÚeqc @ s0 e Zd ZejddZejddZejddZdS )Ú
|
AFDWaitersN©Údefault)r r r ÚattrÚibÚ read_taskÚ
|
write_taskÚ
|
current_opr# r# r# r$ rD L s rD )rB rC Úfrozenc @ s, e Zd Ze ¡ Ze ¡ Ze ¡ Ze ¡ ZdS )Ú AFDPollOpN) r r r rG rH Ú lpOverlappedÚ poll_infoÚwaitersÚ afd_groupr# r# r# r$ rM V s rM iô c @ s e Zd Ze ¡ Ze ¡ ZdS )ÚAFDGroupN)r r r rG rH Úsizer@ r# r# r# r$ rR i s rR c @ s8 e Zd Ze ¡ Ze ¡ Ze ¡ Ze ¡ ZejddZ dS )Ú_WindowsStatisticsZwindowsrE N)
|
r r r rG rH Útasks_waiting_readÚtasks_waiting_writeÚtasks_waiting_overlappedÚcompletion_key_monitorsÚbackendr# r# r# r$ rT o s
|
rT éè )rL c @ s e Zd Ze ¡ Ze ¡ ZdS )ÚCompletionKeyEventInfoN)r r r rG rH rN ÚdwNumberOfBytesTransferredr# r# r# r$ r[ s r[ c @ sØ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z e dd Ze dd Ze dd Ze dd Ze d d Zd d Ze d+d"d#Ze d,d$d%Ze d&d' Zee d(d) Zd*S )-ÚWindowsIOManagerc C sÎ d | _ g | _tt ttjdd¡| _ t dt ¡| _
|
t | _ i | _ i | _i | _t | _i | _t tj¡| _t ¡ T}t|tjd}zt|tjd}W n t k
|
r® t | Y nX ||krÀt dW 5 Q R X d S )Nr zOVERLAPPED_ENTRY[]r( zßUnexpected network configuration detected: SIO_BASE_HANDLE and SIO_BSP_HANDLE_SELECT differ. Please file a bug at https://github.com/python-trio/trio/issues/new, and include the output of running: netsh winsock show catalog) Ú_iocpÚ_all_afd_handlesr' r ÚCreateIoCompletionPortr r
|
r/ r- Ú
|
MAX_EVENTSÚ_eventsÚsetÚ_vacant_afd_groupsÚ_afd_opsÚ _afd_waitersÚ_overlapped_waitersÚ_posted_too_late_to_cancelÚ_completion_key_queuesÚ itertoolsÚcountr r" Ú_completion_key_counterÚsocketr4 r ZSIO_BSP_HANDLE_SELECTÚSIO_BASE_HANDLEr5 r9 r7 )ÚselfÚsZ select_handleÚ base_handler# r# r$ Ú__init__ s4 ÿ
|
ÿÿzWindowsIOManager.__init__c
|
C sP z(| jd k r&| j}d | _tt |¡ W 5 | j rJ| j ¡ }tt |¡ q*X d S r% )r_ Úpopr' r Ú CloseHandler^ )ro Z
|
afd_handleZiocpr# r# r$ ÚcloseÅ s
|
|
zWindowsIOManager.closec C s | ¡ d S r% )ru ©ro r# r# r$ Ú__del__Ð s zWindowsIOManager.__del__c C sX d}d}| j ¡ D ](}|jd k r(|d7 }|jd k r|d7 }qt||t| jt| jdS )Nr r )rU rV rW rX )rf ÚvaluesrI rJ rT Úlenrg ri )ro rU rV Úwaiterr# r# r$ Ú
|
statisticsÓ s
|
|
|
üzWindowsIOManager.statisticsc C s t t | jdtjtj¡ d S ©Nr )r' r ÚPostQueuedCompletionStatusr^ r r! r
|
r/ rv r# r# r$ Ú force_wakeupâ s ÿÿz WindowsIOManager.force_wakeupc
|
C s t d¡}td| }|dkr*|dkr*d}z tt | j| jt||d¡ W n6 t k
|
r } z|j
|
t j krj W Y ¢dS d }~X Y nX |d S )NÚPULONGrZ r r ) r
|
r- Úroundr' r ZGetQueuedCompletionStatusExr^ rb ra r5 r6 r Ú WAIT_TIMEOUT)ro ÚtimeoutÚreceivedZ millisecondsÚexcr# r# r$ Ú
|
get_eventsé s(
|
ÿÿ zWindowsIOManager.get_eventsc C sÄ t |D ]´}| j| }|jtjkrÐ|j}| j |¡}|j}|j |k rHnd |_ |j
|
dkrlt |j
|
¡}t | |jjd j}|jr|t@ rt |j¡ d |_|jrº|t@ rºt |j¡ d |_| |jjd j¡ q|jtjkr| j |j¡} |j}
|
|j} t |
|
| d} t | t | ¡ q|jtj krt| j |j¡ z| j |j¡} W n t!k
|
rZ Y nX t "d #| j$¡¡} | q|jtj%krq| j&|j }t't( )d|j¡}
|
|j} t |
|
| d} | *| ¡ qd S )Nr )rN r\ z|Failed to cancel overlapped I/O in {} and didn't receive the completion either. Did you forget to call register_with_iocp()?Ú uintptr_t)+Úrangerb ZlpCompletionKeyr r rN re rs rP rK ÚInternalr ÚRtlNtStatusToDosErrorr rO ÚHandlesÚEventsrI ÚREADABLE_FLAGSr Z
|
reschedulerJ ÚWRITABLE_FLAGSÚ _refresh_afdÚHandler rg r\ r[ r r rh ÚremoveÚKeyErrorÚTrioInternalErrorÚformatÚnamer! ri Úintr
|
r. Ú
|
put_nowait)ro r ÚiÚentryZlpoÚoprP r3 Úflagsrz Z
|
overlappedZ transferredÚinfor Úqueuer# r# r$ Úprocess_eventsú sf
|
|
|
ÿþÿ ÿz WindowsIOManager.process_eventsc C s4 t |}tt || j|d¡ tt |tj¡ d S r| )r r' r r` r^ Z"SetFileCompletionNotificationModesr Z FILE_SKIP_SET_EVENT_ON_HANDLE)ro r@ Zcompletion_keyr# r# r$ Ú_register_with_iocpJ s ÿÿz$WindowsIOManager._register_with_iocpc C s0 | j | }|jd k r|jj}ztt |j|jj¡ W n0 tk
|
rf } z|j t
|
j krV W 5 d }~X Y nX d |_| j d8 _ | j |¡ d}|jd k r|tO }|jd k r°|tO }|sÀ| j |= nlz| j ¡ }W n@ tk
|
r tdt }| |jtj¡ | j |j¡ Y nX | j |¡ t d¡}t d¡}d|_ d|_ d|_ ||j!d _"d|j!d _#||j!d _$z2tt %|jt&j'|t (d¡|t (d¡t j)|¡ W nN tk
|
ræ } z.|j t
|
j*krÖ| j |= t+|| W Y ¢d S W 5 d }~X Y nX t,||||}||_|| j-|< | j d7 _ |j t.kr,| j /|¡ d S )Nr r Ú LPOVERLAPPEDzAFD_POLL_INFO *l ÿÿÿÿ Z AFD_POLL_INFO)0rf rK rQ r' r Ú
|
CancelIoExr@ rN r5 r6 r ÚERROR_NOT_FOUNDrS rd ÚaddrI r rJ r rs r rR rA r r r r_ Úappendr
|
r- ÚTimeoutZNumberOfHandlesZ Exclusiver r ZStatusr ZDeviceIoControlr ZIOCTL_AFD_POLLr0 r/ ÚERROR_IO_PENDINGr rM re ÚMAX_AFD_GROUP_SIZEr ) ro rq rP rQ r r rN rO r r# r# r$ r [ s|
|
|
ÿÿ
|
|
|
|
øÿ
|
|
z WindowsIOManager._refresh_afdc s t | j ¡d kr,t j < td k r@tjtt ¡ ¡ fdd}t
|
|¡I d H d S )Nc s t d ¡ tjjS r% )Úsetattrr r ÚAbortZ SUCCEEDED)Ú_©rq Úmodero rP r# r$ Úabort_fn´ s
|
z,WindowsIOManager._afd_poll.<locals>.abort_fn) r9 rf ÚgetrD Úgetattrr ÚBusyResourceErrorr§ Ú current_taskr Úwait_task_rescheduled)ro r1 r« r¬ r# rª r$ Ú _afd_poll§ s
|
|
zWindowsIOManager._afd_pollc à s | |d¡I d H d S )NrI ©r² ©ro r1 r# r# r$ Ú wait_readable» s z WindowsIOManager.wait_readablec à s | |d¡I d H d S )NrJ r³ r´ r# r# r$ Ú wait_writable¿ s z WindowsIOManager.wait_writablec C s8 t |}| j |¡}|d k r4t|t ¡ | |¡ d S r% )r9 rf r r r ÚClosedResourceErrorr )ro r@ rP r# r# r$ Únotify_closingà s
|
z WindowsIOManager.notify_closingc C s | |tj¡ d S r% )r r r )ro r@ r# r# r$ Úregister_with_iocpÏ s z#WindowsIOManager.register_with_iocpc s² t ttr t d¡jkr2t d¡t ¡ }|j< d fdd}t |¡I d H }j
|
dkr®t j
|
¡}|t jkr¦d k r q®t d¡nt| |S )Nr z4another task is already waiting on that lpOverlappedc
|
s | zt t ¡ W n` tk
|
rx } zB|jtjkr\t t jdt j
|
¡ j ¡ n t d¡|W 5 d }~X Y nX t jjS )Nr z'CancelIoEx failed with unexpected error)r' r r r5 r6 r r¡ r} r^ r r rh r¢ r r r¨ ZFAILED)Z raise_cancel_r ©r@ rN Z raise_cancelro r# r$ Úabortà s( ÿÿ
|
ÿþz/WindowsIOManager.wait_overlapped.<locals>.abortr z!another task closed this resource)r Ú
|
isinstancer r
|
r. rg r r¯ r° r± r r r r ÚERROR_OPERATION_ABORTEDr· r )ro r@ rN Ztaskr» r r3 r# rº r$ Úwait_overlappedÓ s(
|
|
ÿ
|
#
|
|
z WindowsIOManager.wait_overlappedc
|
à sl t ¡ I d H t d¡}z || W n0 tk
|
rT } z|jtjkrD W 5 d }~X Y nX | ||¡I d H |S )Nr ) r Zcheckpoint_if_cancelledr
|
r- r5 r6 r r¥ r¾ )ro r@ Z submit_fnrN r r# r# r$ Ú_perform_overlapped s
|
z$WindowsIOManager._perform_overlappedr c
|
sJ t |¡6 fdd}| |¡I d H }|jW 5 Q R £ S Q R X d S )Nc sH | j j}d@ |_d? |_tt tt d ¡t
|
tj | ¡ d S )Nì ÿÿ é ÚLPCVOID) ÚDUMMYUNIONNAMEÚDUMMYSTRUCTNAMEÚOffsetÚ
|
OffsetHighr' r Ú WriteFiler r
|
r. ry r/ ©rN Z offset_fields©ZcbufÚ file_offsetr@ r# r$ Ú submit_write+ s
|
|
|
ûÿz7WindowsIOManager.write_overlapped.<locals>.submit_write©r
|
r< r¿ Z InternalHigh)ro r@ ÚdatarÊ rË rN r# rÉ r$ Úwrite_overlapped' s z!WindowsIOManager.write_overlappedc
|
sN t j|dd6 fdd}| |¡I d H }|jW 5 Q R £ S Q R X d S )NT)Zrequire_writablec sH | j j}d@ |_d? |_tt tt d ¡t
|
tj | ¡ d S )NrÀ rÁ ÚLPVOID) rà rÄ rÅ rÆ r' r ÚReadFiler r
|
r. ry r/ rÈ rÉ r# r$ Ú submit_readB s
|
|
|
ûÿz9WindowsIOManager.readinto_overlapped.<locals>.submit_readrÌ )ro r@ ÚbufferrÊ rÑ rN r# rÉ r$ Úreadinto_overlapped> s z$WindowsIOManager.readinto_overlappedc C s t t d| j¡S )Nr )r r
|
r. r^ rv r# r# r$ Ú current_iocpW s z WindowsIOManager.current_iocpc c s: t | j}t ¡ }|| j|< z||fV W 5 | j|= X d S r% )Únextrl r ZUnboundedQueueri )ro Úkeyr r# r# r$ Úmonitor_completion_key[ s
|
|
z'WindowsIOManager.monitor_completion_keyN)r )r )r r r rr ru rw r{ r~ r
r r r r² r rµ r¶ r¸ r¹ r¾ r¿ rÎ rÓ rÔ r r× r# r# r# r$ r]
s< ? PL
|
|
|
|
|
A
|
r] ):rj Ú
|
contextlibr Úenumrm ÚsysÚtypingr rG Zoutcomer Ú r Z_runr Z
|
_io_commonr Z _windows_cffir
|
r r r r r r r r r r r r ÚplatformÚAssertionErrorÚIntEnumr r' rn r4 r9 rA ZAFD_POLL_RECEIVEZAFD_POLL_ACCEPTZAFD_POLL_DISCONNECTZAFD_POLL_ABORTZAFD_POLL_LOCAL_CLOSEr Z AFD_POLL_SENDZAFD_POLL_CONNECT_FAILr rp rD rM r¦ rR rT ra r[ r] r# r# r# r$ Ú<module> sb < //ÿþýüÿ ÿþýÿ
|
|
