֦iK d dl Z e j dk7 r ed d dlZd dlZd dlZd dlmZ d dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ dZej4 Zej6 ZdZ dZ dZ dZ G d dej@ Z! G d dej@ Z" G d de" Z# G d de" Z$ G d d e% Z& G d d ejN Z( G d d!ejR Z* G d" d# Z+ G d$ d%ejX Z-e(Z. G d& d'ej^ Z0 G d( d)ej^ Z1e1Z2y)* 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 V e Zd Z dd fd
Z fdZd Zd fd Z fdZ fdZ xZ S )
_OverlappedFutureNloopc 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 ) rS rR _overlappedUnregisterWaitr7 winerrorERROR_IO_PENDINGr r8 r9 rc r rU r: r; s r# _unregister_waitz&_BaseWaitHandleFuture._unregister_wait s '' & &{ 3
&
||{ ; ;; E!$ "
))262H2HG ./
11': <
s A CAB<<Cc D | j t | | S r> )rk r r6 r@ s r# r6 z _BaseWaitHandleFuture.cancel s w~#~&&r$ c D | j t | | y rB )rk r rC rD s r# rC z#_BaseWaitHandleFuture.set_exception s
i(r$ c D | j t | | y rB )rk r rF rG s r# rF z _BaseWaitHandleFuture.set_result s
6"r$ rB ) rI rJ rK r r\ r* rc rk r6 rC rF rL rM s @r# rO rO a s6 <8< '
'0')# #r$ rO c @ e Zd Z dd fd
Zd Z fdZ fdZ xZS )_WaitCancelFutureNr c : t | |||| d | _ y )Nr )r r _done_callback)r r! eventrU 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 rr rG s r# rF z _WaitCancelFuture.set_result s/
6" * % +r$ c ` t | | | j | j | y y rB )r rC rr rD s r# rC z _WaitCancelFuture.set_exception s/
i( * % +r$ )rI rJ rK r r6 rF rC rL rM s @r# rp rp s' 8< #
F&
& &r$ rp 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_proactorrf CreateEvent_event
_event_fut)r r! rT rU 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~ rX CloseHandler r{ _unregisterr r rc )r rb r" s r# rc 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 wre )rS rR rf UnregisterWaitExr~ r7 rh ri r r8 r9 r{ _wait_cancelrc r rj s r# rk z"_WaitHandleFuture._unregister_wait s '' ( (dkk B ..55dkk6:6N6NP
||{ ; ;; E!$ "
))262H2HG ./
11': <
s A? ? C0AC++C0)rI rJ rK r rc rk rL rM s @r# ry ry s BF )$Pr$ ry c 0 e Zd Z d Zd Zd Zd Zd ZeZy)
PipeServerc || _ 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 )closedrX 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 r r r r r __del__ r$ r# r r s' 4$'
) Gr$ r c e Zd Zy)_WindowsSelectorEventLoopN)rI rJ rK r r$ r# r r 2 s 1r$ r c B e Zd Z d fd Z fdZd Zd Z ddZ xZS )r c < |
t }t | | y rB )r r r )r r r" s r# r zProactorEventLoop.__init__9 s #~H
"r$ c | 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 )
call_soon_loop_self_readingr run_forever_self_reading_futurer r6 r& r{ r )r r! r" s r# r z ProactorEventLoop.run_forever> s 1 NN422 3 G !((4..22))002 >"**NN..r2,0) 5t((4..22))002 >"**NN..r2,0) 5s )B A/D c 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 r r r r r rL rM s @r# r r 6 s% <#
1& 1j 04r$ r c e Zd Z efdZd Zd Zd Zd dZd Z e
d Z e d Z
d!d
Zd!d Zd!d Zd!d
Zd"dZd!dZd Zd Zd Zd Zd Zd dZd Zd Zd Z d Z d Z d Z d d Z d Z!d Z"d Z#y)#r c d | _ g | _ t j t j t
d| | _ i | _ t j | _
g | _ t j | _ y rW )
r8 _resultsrf CreateIoCompletionPortINVALID_HANDLE_VALUEr _iocp_cacher r rS
_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 ru 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 r` 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 valuerb 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 rh rf 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 rh rf 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_iocprf
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 wrW )
r rf 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 rf 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 rf 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 rf 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 rf 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_SOCKETrf 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_socketfamilyrf 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 rW )r r/ r r0 rf SO_UPDATE_CONNECT_CONTEXT)r r r! r s r# finish_connectz,IocpProactor.connect..finish_connectV s1 LLN OOF - - ' A A1
FKr$ )typer
SOCK_DGRAMrf
WSAConnectr r8 r rF r BindLocalr: r7 rh errno WSAEINVAL getsocknamer
r ConnectExr )r r r, rb 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 rf 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 rf 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_DELAYrf ConnectPiper7 rh ERROR_PIPE_BUSYminCONNECT_PIPE_MAX_DELAYr sleepr
r )r r, delayrT 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 r )_wait_for_handle)r rT r s r# wait_for_handlez IocpProactor.wait_for_handle s
$$VWe<.finish_wait_for_handle s 779 r$ r )r rX INFINITEmathceilrf r
r RegisterWaitWithQueuer r, rp r8 ry r r ) r rT r
_is_cancelmsr! rU 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 rW )rS r rf 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 \ | j | j j | y rB )r r r` )r r! s r# r zIocpProactor._unregister s)
!!"%r$ c R t j | }|j d |S rW )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 rf GetQueuedCompletionStatusr r popKeyErrorr8 get_debugr9 r rX r r r6 donerF r r` 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 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 intrQ 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 )3sysplatform ImportErrorrf rX 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 rO rp ry 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$