֦iK d Z ddlZej dk7 r ed ddlZddlZddlZddlmZ ddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ dZej6 Zej8 Z dZ dZ dZ dZ G d dejB Z" G d dejB Z# G d de# Z$ G d d e# Z% G d d e& Z' G d d ejP Z) G d! d"ejT Z+ G d# d$ Z, G d% d&ejZ Z.e)Z/ G d' d(ej` Z1 G d) d*ej` Z2e2Z3y)+z.Selector and proactor event loops for Windows. Nwin32z
win32 only)partial )events)base_subprocess)futures)
exceptions)proactor_events)selector_events)tasks)
windows_utils)logger)SelectorEventLoopProactorEventLoop IocpProactorDefaultEventLoopPolicy WindowsSelectorEventLoopPolicy WindowsProactorEventLoopPolicyi i gMbP?g?c X e Zd ZdZdd fd
Z fdZd Zd
fd Z fdZ fd Z xZ
S ) _OverlappedFuturezSubclass of Future which represents an overlapped operation.
Cancelling it will immediately cancel the overlapped operation.
Nloopc f t | | | j r
| j d= || _ y Nr )super__init___source_traceback_ov)selfovr __class__s //usr/lib64/python3.12/asyncio/windows_events.pyr z_OverlappedFuture.__init__7 s1
d# ! !&&r* c t | }| j H| j j rdnd}|j dd| d| j j
dd |S )Npending completedr z overlapped=)r
_repr_infor r& insertaddressr infostater" s r# r* z _OverlappedFuture._repr_info= s\ w!# 88 !%!1!1I{E KK\%4883C3CB2GqI J r$ c | j y | j j d | _ y # t $ rM}d|| d}| j r| j |d< | j j | Y d }~d | _ y d }~ww xY w)Nz&Cancelling an overlapped future failedmessage exceptionfuturesource_traceback)r cancelOSErrorr _loopcall_exception_handler)r exccontexts r# _cancel_overlappedz$_OverlappedFuture._cancel_overlappedD s 88
7 HHOO 7C G
%%.2.D.D*+ JJ - -g 6 6 7s 1 B r$ c d | _ y rB )r )r futs r# _unregister_wait_cbz)_BaseWaitHandleFuture._unregister_wait_cb s r$ c | j sy d| _ | j }d | _ t j | | j d y # t $ rh}|j
t j k7 rAd|| d}| j r| j |d< | j j | Y d }~y Y d }~~d }~ww xY wNFz$Failed to unregister the wait handler1 r5 ) rT rS _overlappedUnregisterWaitr7 winerrorERROR_IO_PENDINGr r8 r9 rd r rV r: r; s r# _unregister_waitz&_BaseWaitHandleFuture._unregister_wait s '' & &{ 3
&
||{ ; ;; E!$ "
))262H2HG ./
11': <
s A CAB<<Cc D | j t | | S r> )rl r r6 r@ s r# r6 z _BaseWaitHandleFuture.cancel s w~#~&&r$ c D | j t | | y rB )rl r rC rD s r# rC z#_BaseWaitHandleFuture.set_exception s
i(r$ c D | j t | | y rB )rl r rF rG s r# rF z _BaseWaitHandleFuture.set_result s
6"r$ rB )
rI rJ rK rL r r] r* rd rl r6 rC rF rM rN s @r# rP rP a s6 <8< '
'0')# #r$ rP c B e Zd ZdZdd fd
Zd Z fdZ fdZ xZS )_WaitCancelFuturezoSubclass of Future which represents a wait for the cancellation of a
_WaitHandleFuture using an event.
Nr c : t | |||| d | _ y )Nr )r r _done_callback)r r! eventrV r r" s r# r z_WaitCancelFuture.__init__ s!
UKd; "r$ c t d )Nz'_WaitCancelFuture must not be cancelled) RuntimeErrorr\ s r# r6 z_WaitCancelFuture.cancel s DEEr$ c ` t | | | j | j | y y rB )r rF rs rG s r# rF z _WaitCancelFuture.set_result s/
6" * % +r$ c ` t | | | j | j | y y rB )r rC rs rD s r# rC z _WaitCancelFuture.set_exception s/
i( * % +r$ ) rI rJ rK rL r r6 rF rC rM rN s @r# rq rq s' 8< #
F&
& &r$ rq c 4 e Zd Zdd fd
Z fdZd Z xZS )_WaitHandleFutureNr c t | |||| || _ d| _ t j
d ddd | _ d | _ y )Nr TF)r r _proactor_unregister_proactorrg CreateEvent_event
_event_fut)r r! rU rV proactorr r" s r# r z_WaitHandleFuture.__init__ sG
V[t<!$(!!--dD%F r$ c | j -t j | j d | _ d | _ | j j | j d | _ t | ! | y rB ) r rY CloseHandler r| _unregisterr r rd )r rc r" s r# rd z%_WaitHandleFuture._unregister_wait_cb sY ;; " , DK "DO
""488,
#C(r$ c | j sy d| _ | j }d | _ t j || j | j j | j | j | _ y # t
$ rh}|j t j k7 rAd|| d}| j r| j |d< | j j | Y d }~y Y d }~d }~ww xY wrf )rT rS rg UnregisterWaitExr r7 ri rj r r8 r9 r| _wait_cancelrd r rk s r# rl z"_WaitHandleFuture._unregister_wait s '' ( (dkk B ..55dkk6:6N6NP
||{ ; ;; E!$ "
))262H2HG ./
11': <
s A? ? C0AC++C0)rI rJ rK r rd rl rM rN s @r# rz rz s BF )$Pr$ rz c 2 e Zd ZdZd Zd Zd Zd Zd ZeZ y)
PipeServerzXClass representing a pipe server.
This is much like a bound, listening socket.
c || _ t j | _ d | _ d | _ | j
d | _ y NT)_addressweakrefWeakSet_free_instances_pipe_accept_pipe_future_server_pipe_handle)r r, s r# r zPipeServer.__init__ s;
& 0
#' --d3
r$ c L | j | j d c}| _ |S )NF)r r )r tmps r# _get_unconnected_pipez PipeServer._get_unconnected_pipe s%
**d&>&>u&ETZ
r$ c
, | j ry t j t j z }|r|t j z }t j
| j |t j t j z t j z t j t j t j t j t j }t j | }| j j# | |S rB )closedrY PIPE_ACCESS_DUPLEXFILE_FLAG_OVERLAPPED FILE_FLAG_FIRST_PIPE_INSTANCECreateNamedPiper PIPE_TYPE_MESSAGEPIPE_READMODE_MESSAGE PIPE_WAITPIPE_UNLIMITED_INSTANCESr
BUFSIZENMPWAIT_WAIT_FOREVERNULL
PipeHandler add)r firstflagshpipes r# r z PipeServer._server_pipe_handle s ;;=**W-I-II W:: :E # # MM5 % %(E(E E
, , ! !=#8#8 ( (',,
8 ''* & r$ c | j d u S rB )r r\ s r# r zPipeServer.closed s
%&r$ c
| j !| j j d | _ | j J| j D ] }|j d | _ d | _ | j j
y y rB )r r6 r r closer clear)r r s r# r zPipeServer.close# sp # # / $ $ + + -'+D $ == $ ,,
- DJ DM & & ( %r$ N)
rI rJ rK rL r r r r r __del__ r$ r# r r s' 4$'
) Gr$ r c e Zd ZdZy)_WindowsSelectorEventLoopz'Windows version of selector event loop.N)rI rJ rK rL r r$ r# r r 2 s 1r$ r c D e Zd ZdZd fd Z fdZd Zd Z ddZ xZ S )r z2Windows version of proactor event loop using IOCP.c < |
t }t | | y rB )r r r )r r r" s r# r zProactorEventLoop.__init__9 s #~H
"r$ c 4 | j J | j | j t | | j a| j j
}| j j
|'|j s| j j | d | _ y y # | j a| j j
}| j j
|'|j s| j j | d | _ w w xY wrB )
_self_reading_future call_soon_loop_self_readingr run_foreverr r6 r& r| r )r r! r" s r# r z ProactorEventLoop.run_forever> s 1,,4 44 NN422 3 G !((4..22))002 >"**NN..r2,0) 5t((4..22))002 >"**NN..r2,0) 5s 7B( (A/Dc K | j j | }| d { } | }| j ||d|i }||fS 7 %w)Naddrextra)r| connect_pipe_make_duplex_pipe_transport)r protocol_factoryr, fr protocoltranss r# create_pipe_connectionz(ProactorEventLoop.create_pipe_connectionQ sZ NN ' ' 0w#%00x8>7H 1 Jh s !A A &A c f K t d fd j gS w)Nc J d } | ri| j }j j | j r|j y }j ||di j
}|y j j | } | _ | j y # t $ r9 |r#|j dk7 r|j j Y y t $ rz}|r9|j dk7 r&j d||d |j n$j rt# j$ d|d j Y d }~y d }~wt& j( $ r |r|j Y y Y y w xY w) Nr r r zPipe accept failed)r2 r3 r z Accept pipe failed on pipe %rT)exc_info)rH r discardr r r r r| accept_piper add_done_callbackBrokenPipeErrorfilenor r7 r9 _debugr warningr CancelledError) r r r r: r, loop_accept_piper r servers r# r z>ProactorEventLoop.start_serving_pipe..loop_accept_pipe\ st D)
6 88:D**2248 }}
/ 1H44 hvw.? 5 A 335<NN..t4* ./*##$45+ #
1DKKMR /JJL /0
1DKKMR ///#7%( $1
JJL[[NN#B#'$8 /00 ,,
!JJL
!s1 A
B7 /B7 B7 7?F"8F" A0E55(F"!F"rB )r r )r r r, r r s ```@@r# start_serving_pipez$ProactorEventLoop.start_serving_pipeY s2 G$+ 6 + 6Z
'(xs *1c
K | j }
t | |||||||f|
|d| } |
d { | S 7 # t t f$ r t $ r+ | j | j
d { 7 w xY ww)N)waiterr )
create_future_WindowsSubprocessTransport
SystemExitKeyboardInterrupt
BaseExceptionr _wait) r r argsshellstdinstdoutstderrbufsizer kwargsr transps r# _make_subprocess_transportz,ProactorEventLoop._make_subprocess_transport s ##%,T8T5-2FFG74:%7 067 LL
-. LLN,,. s1 'A>8 68 A>8 ;A;3A64A;;A>rB )
rI rJ rK rL r r r r r rM rN s @r# r r 6 s% <#
1& 1j 04r$ r c e Zd ZdZefdZd Zd Zd Zd!dZ d Z
e d Z e
d
Zd"d Zd"d Zd"d
Zd"dZd#dZd"dZd Zd Zd Zd Zd Zd!dZd Zd Z d Z d Z d Z d Z d!d Z!d Z"d Z#d Z$y)$r z#Proactor implementation using IOCP.c d | _ g | _ t j t j t
d| | _ i | _ t j | _
g | _ t j | _ y rX )
r8 _resultsrg CreateIoCompletionPortINVALID_HANDLE_VALUEr _iocp_cacher r rT
_unregistered_stopped_serving)r concurrencys r# r zIocpProactor.__init__ s_
77 , ,dA{D
"??, ' 1 r$ c 2 | j t d y )NzIocpProactor is closed)r rv r\ s r#
_check_closedzIocpProactor._check_closed s :: 78 8 r$ c dt | j z dt | j z g}| j |j d d| j
j ddj | dS )Nzoverlapped#=%sz
result#=%sr < r) )lenr r r ra r" rI join)r r. s r# __repr__zIocpProactor.__repr__ s_ 3t{{#33 s4== 113 :: KK ! NN 3 3SXXd^DDr$ c || _ y rB )r8 )r r s r# set_loopzIocpProactor.set_loop s
r$ Nc z | j s| j | | j }g | _ |d }S # d }w xY wrB )r r] )r timeoutr s r# selectzIocpProactor.select s: }} JJw mm
C$Cs 6 :c \ | j j }|j | |S rB )r8 r rF )r valuerc s r# _resultzIocpProactor._result s% jj&&( u
r$ c |j S # t $ rD}|j t j t j
fv rt
|j d }~ww xY wrB ) getresultr7 ri rg ERROR_NETNAME_DELETEDERROR_OPERATION_ABORTEDConnectionResetErrorr )r keyr! r: s r# finish_socket_funcz IocpProactor.finish_socket_func sY <<> ! || A A + C C E E*CHH55 s A ?AA c | j ||| S # t $ r,}|j t j k( r |d fcY d }~S d }~ww xY wrB )r r7 ri rg ERROR_PORT_UNREACHABLE)clsr r r! empty_resultr: s r# _finish_recvfromz IocpProactor._finish_recvfrom sN ))%b9 9 ||{ A AA#T))
s A
AA
AA
c | j | t j t } t |t
j
r"|j
|j || n |j |j | | j ||| j S # t $ r | j d cY S w xY w)Nr$ )
_register_with_iocprg
Overlappedr
isinstancesocketWSARecvr ReadFiler r _registerr r connnbytesr r! s r# recvzIocpProactor.recv s &
#
#D
) %$
.
4;;=&%8 DKKM62 ~~b$(?(?@@ %<<$ $ % A B% %CCc | j | t j t } t |t
j
r"|j
|j || n |j |j | | j ||| j S # t $ r | j d cY S w xY wrX )
r rg r
r r r WSARecvIntor ReadFileIntor r r r r r bufr r! s r# recv_intozIocpProactor.recv_into s &
#
#D
) #$
.t{{}c59
s3 ~~b$(?(?@@ #<<? " #r c * | j | t j t } |j |j || | j ||t | j d S # t $ r | j d cY S w xY w)N)r$ Nr$ r ) r rg r
r WSARecvFromr r r r r r r s r# recvfromzIocpProactor.recvfrom s &
#
#D
) - NN4;;=&% 8 ~~b$0E0E=@)B C C -<< , , - !A5 5BBc * | j | t j t } |j |j || | j ||t | j d S # t $ r | j d cY S w xY w)Nr Nr r ) r rg r
r WSARecvFromIntor r r r r r r s r#
recvfrom_intozIocpProactor.recvfrom_into s &
#
#D
) + t{{}c5 9 ~~b$0E0E=>)@ A A +<< * * +r c | j | t j t }|j |j ||| | j
||| j S rB )r rg r
r WSASendTor r r )r r r r r r! s r# sendtozIocpProactor.sendto sQ &
#
#D
)
T[[]C5~~b$(?(?@@r$ c H | j | t j t }t |t
j
r"|j
|j || n |j |j | | j ||| j S rB ) r rg r
r r r WSASendr WriteFiler r r s r# sendzIocpProactor.send sq &
#
#D
) dFMM * JJt{{}c5 1 LL ,~~b$(?(?@@r$ c | | j | j j t j t
}|j
j j fd}d }| j || } || }t j || j |S )Nc , |j t j dj }j t
j t j | j j j fS )Nz@P) r structpackr
setsockoptr
SOL_SOCKETrg SO_UPDATE_ACCEPT_CONTEXT
settimeout
gettimeout getpeername)r r r! r r listeners r#
finish_acceptz*IocpProactor.accept..finish_accept* sl LLN++dHOO$56C OOF - - ' @ @#
G OOH / / 1 2))++ +r$ c v K | d { y 7 # t j $ r |j w xY wwrB )r r r )r4 r s r# accept_coroz(IocpProactor.accept..accept_coro3 s2
,,
s 9 9 %69r ) r _get_accept_socketfamilyrg r
r AcceptExr r r
ensure_futurer8 )r r5 r! r6 r8 r4 coror s ` @r# acceptzIocpProactor.accept$ s *&&x7
#
#D
)
HOO%t{{}5 , Hm<64(
Dtzz2
r$ c j t j k( rQt j j | | j j }|j d |S | j t j j j t j" t$ }|j' j | fd}| j) || S # t $ r?}|j t j k7 r j! d dk( r Y d }~d }~ww xY w)Nr r c |j j t j t j
d S rX )r r/ r r0 rg SO_UPDATE_CONNECT_CONTEXT)r r r! r s r# finish_connectz,IocpProactor.connect..finish_connectV s1 LLN OOF - - ' A A1
FKr$ )typer
SOCK_DGRAMrg
WSAConnectr r8 r rF r BindLocalr: r7 ri errno WSAEINVAL getsocknamer
r ConnectExr )r r r, rc er! rB s ` r# connectzIocpProactor.connect@ s 99)) )
" "4;;=' :****,C NN4 J & ! !$++- =
#
#D
)
T[[]G, ~~b$77! zzU__, !!$) * s .D E
5EE c 6 | j | t j t }|dz }|dz dz }|j |j t
j |j |||dd | j ||| j S )Nl r )
r rg r
r TransmitFiler msvcrt
get_osfhandler r )r sockfileoffsetcountr!
offset_low offset_highs r# sendfilezIocpProactor.sendfile_ s &
#
#D
)k)
|{2
,,T[[];"K q! %
~~b$(?(?@@r$ c | j t j t }|j j }|r| j
S fd}| j || S )Nc ( |j S rB )r )r r r! r s r# finish_accept_pipez4IocpProactor.accept_pipe..finish_accept_pipev s LLNKr$ )r rg r
r ConnectNamedPiper r r )r r r! connectedr[ s ` r# r zIocpProactor.accept_pipek sf &
#
#D
)''
6 <<% % ~~b$(:;;r$ c < K t } t j | } t j | S # t $ r(}|j t j
k7 r Y d }~nd }~ww xY wt
|dz t }t j | d { 7 w)N ) CONNECT_PIPE_INIT_DELAYrg ConnectPiper7 ri ERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr sleepr
r )r r, delayrU r: s r# r zIocpProactor.connect_pipe| s '
$009 ''//
<<;#>#>> ?
#9:E++e$ $ $ s6 B 6 B A' A" B "A''.B BB c ( | j ||d S )zWait for a handle.
Return a Future object. The result of the future is True if the wait
completed, or False if the wait did not complete (on timeout).
F)_wait_for_handle)r rU r s r# wait_for_handlez IocpProactor.wait_for_handle s $$VWe<.finish_wait_for_handle s 779 r$ r )r rY INFINITEmathceilrg r
r RegisterWaitWithQueuer r, rq r8 rz r r ) r rU r
_is_cancelmsr! rV ro r s @r# rh z IocpProactor._wait_for_handle s ?!!B 7S=)B
#
#D
)!77 DJJ
B0 !"fk
KA!"fk4'+zz3A ##B' $%b!-C"D BJJ r$ c || j vrL| j j | t j |j | j
dd y y rX )rT r rg r r r r objs r# r z IocpProactor._register_with_iocp sI d&& & % . .szz|TZZA N 'r$ c ^ | j t || j }|j r
|j d= |j s |d d | }|j | ||||f| j |j < |S # t $ r}|j | Y d }~>d }~ww xY wr )
r r r8 r r& rF r7 rC r r, )r r! rx callbackr r rK s r# r zIocpProactor._register s
btzz 2 ##B'zz
$ tR0 U# $%b#x"8 BJJ
#""
#s
B B,B''B,c Z | j | j j | y)a
Unregister an overlapped object.
Call this method when its future has been cancelled. The event can
already be signalled (pending in the proactor event queue). It is also
safe if the event is never signalled (because it was cancelled).
N)r r ra )r r! s r# r zIocpProactor._unregister s$
!!"%r$ c R t j | }|j d |S rX )r r2 )r r: ss r# r9 z IocpProactor._get_accept_socket s MM& ! Qr$ c " |t }n<|dk r t d t j |dz }|t k\ r t d t j
| j | }|nd}|\ }}}} | j j | \ }} }
} |
| j v r|j# nI|j% s9 | ||| } |j' | | j( j+ | d }| j0 D ]( } | j j | j2 d * | j0 j5 y # t $ rl | j j r%| j j dd||||fz d |dt j fvrt j | Y }w xY w# t, $ r7}
|j/ |
| j( j+ | Y d }
~
d }
~
ww xY w# d }w xY w)Nr znegative timeoutrm ztimeout too bigz8GetQueuedCompletionStatus() returned an unexpected eventz)err=%s transferred=%s key=%#x address=%#x)r2 status)rp
ValueErrorrq rr rg GetQueuedCompletionStatusr r popKeyErrorr8 get_debugr9 r rY r r r6 donerF r ra r7 rC r r, r )r r ru r err transferredr r, r r! rx rz r rK s r# r] zIocpProactor._poll s ?B
q[ /0 0 7S=)BX~ !233 ::4::rJF~B-3 *Cc7
'+{{w'?$2sH d+++
VVX $[#r :E
LL'MM((+ AM R $$B KKOOBJJ - % "E
::'')JJ55%7#N&);W%E$F7 q+"B"B CC'',
, ,OOA&MM((++, AsC 4 E
G ,H
A1GG H,H<H
HH
Hc : | j j | y rB )r r rw s r#
_stop_servingzIocpProactor._stop_serving2 s
!!#&r$ c 4 | j y t | j j D ]: \ }}}}|j rt |t r* |j < d}t j }||z } | j rx| t j k rCt j d| t j |z
t j |z } | j! | | j rxg | _ t% j& | j d | _ y # t $ rS}| j >++ K !4>>#3j#@B >>+j8
JJz " kk
DJJ'
; Czz-'C),&)#
00:=:O:OG$6 7
99'B Cs D;; FAFFc $ | j y rB )r r\ s r# r zIocpProactor.__del__g s
r$ rB )r r! )%rI rJ rK rL rp r r r r r r staticmethodr classmethodr r r r r# r& r* r> rL rX r r ri r rh r r r r9 r] r r r r r$ r# r r s -#+ 29E
A A C AAA8 8>
A<"0&= DO @&
7#r' -^r$ r c e Zd Zd Zy)r c t j |f|||||d| _ fd} j j j t
j j } | j | y )N)r r r r r c \ j j }j | y rB )_procpoll_process_exited)r
returncoder s r# rz z4_WindowsSubprocessTransport._start..callbackr s! *J ,r$ ) r
Popenr r8 r| ri intrR r )
r r r r r r r r rz r s
` r# _startz"_WindowsSubprocessTransport._startm so "(( ' U6&' %'
-
JJ 0 0TZZ5G5G1H I H%r$ N)rI rJ rK r r r$ r# r r k s
&r$ r c e Zd ZeZy)r N)rI rJ rK r
_loop_factoryr r$ r# r r } %Mr$ r c e Zd ZeZy)r N)rI rJ rK r r r r$ r# r r r r$ r )4rL sysplatform ImportErrorrg rY rG functoolsr rq rP r r- r r r r r r r
r r r
logr __all__r rp ERROR_CONNECTION_REFUSEDERROR_CONNECTION_ABORTEDr` rd Futurer rP rq rz objectr BaseSelectorEventLoopr BaseProactorEventLoopr r BaseSubprocessTransportr r BaseDefaultEventLoopPolicyr r r r r$ r# r s\ 4
<<7
l
##
|| - -`G#GNN G#T&- &01P- 1Ph8 8v2 E E 2g== gTH HV &/"I"I & . &V%F%F &&V%F%F & 8 r$