cjk l d Z ddlZddlZddlZddlZddlZ eed re dz
Z ddlmZ ddlm Z ddlm
Z
dd lm Z dd
lm Z dd l
mZ dd lmZ d
ZdeddZdeddZ eed rd eddZd eddZ G d de j, Z G d dee j, Z G d d Z G d d Zy) ) StreamReader StreamWriterStreamReaderProtocolopen_connection start_server NAF_UNIX)open_unix_connectionstart_unix_server )
coroutines)events)
exceptions)format_helpers) protocols)logger)sleepi )limitc K t j }t || }t || |j fd| |fi | d { \ }}t | || }||fS 7 w)Nr loopr c S N protocols (/usr/lib64/python3.12/asyncio/streams.pyz!open_connection..1 s )r
get_running_loopr r create_connectionr )
hostportr kwdsr reader transport_writerr s
@r r r s} " " " $D
D
1F#F6H///$.(,. .LIq
)Xvt
.s A A*A(A*c K t j fd} j |||fi | d { S 7 w)Nc > t } t | }|S Nr r r r r% r client_connected_cbr r s r factoryz start_server..factoryN s& E5'0C-13r )r
r
create_server)r. r" r# r r$ r/ r s ` ` @r r r 6 sE ( " " $D $##GT4@4@ @@ @s 5A?Ac K t j }t || }t || |j fd| fi | d { \ }}t ||| }||fS 7 w)Nr r c S r r r s r r z&open_unix_connection..b s Hr )r
r r r create_unix_connectionr ) pathr r$ r r% r& r' r( r s @r r r Z sw N&&( E5'T: 8T 8 8 d ,&* , , 1 i64@v~ ,s A
A)
A'A)c K t j fd} j ||fi | d { S 7 w)Nc > t } t | }|S r+ r, r- s r r/ z"start_unix_server..factoryk s& !D9F+F4G157HOr )r
r create_unix_server)r. r4 r r$ r/ r s ` ` @r r
r
f s? K&&( -T,,WdCdCCCCs 4A >A c 4 e Zd Z ddZd Zd Zd Zd Zd Zy) FlowControlMixinNc |t j | _ n|| _ d| _ t j
| _ d| _ y NF)r
get_event_loop_loop_paused collectionsdeque_drain_waiters_connection_lost)selfr s r __init__zFlowControlMixin.__init__~ s> < ..0DJ DJ ) / / 1 % r c t d| _ | j j rt j d| y y )NTz%r pauses writing)r> r= get_debugr debugrC s r
pause_writingz FlowControlMixin.pause_writing s- :: ! LL,d 3 "r c d| _ | j j rt j d| | j
D ]$ }|j
r|j d & y )NFz%r resumes writing)r> r= rF r rG rA done
set_resultrC waiters r resume_writingz FlowControlMixin.resume_writing sO :: ! LL-t 4))F;;=!!$' *r c d| _ | j sy | j D ]8 }|j r||j d (|j | : y NT)rB r> rA rK rL
set_exceptionrC excrN s r connection_lostz FlowControlMixin.connection_lost sN $ || ))F;;=;%%d+((- *r c N K | j r t d | j sy | j j }| j
j
| | d { | j
j | y 7 # | j
j | w xY ww)NzConnection lost)rB ConnectionResetErrorr> r=
create_futurerA appendremoverM s r
_drain_helperz FlowControlMixin._drain_helper s &'89 9|| ))+ ""6* /LL & &v .
& &v .s0 AB% B "B#B ' B%B B""B%c t r )NotImplementedErrorrC streams r _get_close_waiterz"FlowControlMixin._get_close_waiter s !!r r ) __name__
__module__ __qualname__rD rI rO rU r[ r` r r r r9 r9 t s% &4 ( .
/"r r9 c d e Zd Z dZd fd Zed Zd Zd Z fdZ d Z
d Z d Z d
Z
xZS ) r Nc 4 t | | |,t j | | _ |j
| _ nd | _ ||| _ d| _ d | _ d | _ d | _
|| _ d| _ | j j | _ y )Nr F)superrD weakrefref_stream_reader_wr_source_traceback_strong_reader_reject_connection_stream_writer_task
_transport_client_connected_cb _over_sslr= rX _closed)rC
stream_readerr. r __class__s r rD z StreamReaderProtocol.__init__ s
d# $%,[[ %?D "%2%D%DD "%)D " * #0D "' "
$7! zz//1 r c < | j y | j S r )ri rH s r _stream_readerz#StreamReaderProtocol._stream_reader s ! ! )%%''r c | j }|j }|| _ || _ |j d d u| _ y )N
sslcontext)r= r& rm ro get_extra_inforq )rC r( r r& s r _replace_writerz$StreamReaderProtocol._replace_writer s< zz$$ $#"11,?tKr c x j rKddi} j r j |d< j j | j y _ j }||j j d d u _ j t | j _ j | j }t j | rA fd} j j | _ j j# | d _ y y )NmessagezpAn open stream was garbage collected prior to establishing network connection; call "stream.close()" explicitly.source_tracebackrx c | j rj y | j }|0j j d|d j y y )Nz*Unhandled exception in client_connected_cb)r| exceptionr& ) cancelledcloser r= call_exception_handler)taskrT rC r& s r callbackz6StreamReaderProtocol.connection_made..callback s\ ~~'!) ..*C
99'S),)2;
")
'r )rl rj r= r abortro rv
set_transportry rq rp r rm r iscoroutine create_taskrn add_done_callbackrk )rC r& contextr% resr s `` r connection_madez$StreamReaderProtocol.connection_made s# " " @G
%%.2.D.D*+ JJ - -g 6 OO #$$ +"11,?tK $ $ 0".y$/5/3zz#;D ++F,0,?,?AC%%c* * "ZZ 3 3C 8
,,X6"&D / 1r c f | j }|$||j n|j | | j j s9| | j j d n| j j | t | | d | _ d | _ d | _
d | _ y r ) rv feed_eofrR rr rK rL rf rU ri rm rn ro )rC rT r% rt s r rU z$StreamReaderProtocol.connection_lost
s $$ {!$$S)|| "{ ''- **3/
$!% "
r c D | j }||j | y y r )rv feed_data)rC datar% s r
data_receivedz"StreamReaderProtocol.data_received s& $$ T " r c Z | j }||j | j ryy)NFT)rv r rq )rC r% s r eof_receivedz!StreamReaderProtocol.eof_received! s, $$ OO >> r c | j S r )rr r^ s r r` z&StreamReaderProtocol._get_close_waiter, s ||r c | j }|j r"|j s|j y y y # t $ r Y y w xY wr )rr rK r r AttributeError)rC closeds r __del__z StreamReaderProtocol.__del__/ sM #\\F {{}V%5%5%7 " &8} s A A
A
NN)ra rb rc rj rD propertyrv rz r rU r r r` r
__classcell__)rt s @r r r sN 2( ( (
L('T $#
#r r c x e Zd Z d Zd Zed Zd Zd Zd Z d Z
d Z d Z d
Z
dd Zd
Zd d d ddZd Zy )r c || _ || _ || _ || _ | j j | _ | j
j
d y r )ro _protocol_readerr= rX
_complete_futrL )rC r& r r% r s r rD zStreamWriter.__init__E sI #!
!ZZ 5 5 7 %%d+r c | j j d| j g}| j |j d| j dj dj
| S )N
transport=zreader=<{}> )rt ra ro r rY formatjoinrC infos r __repr__zStreamWriter.__repr__O s[ '':doo5H)IJ << # KK'$,,!12 3}}SXXd^,,r c | j S r ro rH s r r& zStreamWriter.transportU s r c : | j j | y r )ro writerC r s r r zStreamWriter.writeY s d#r c : | j j | y r )ro
writelinesr s r r zStreamWriter.writelines\ s ""4(r c 6 | j j S r )ro write_eofrH s r r zStreamWriter.write_eof_ s ((**r c 6 | j j S r )ro
can_write_eofrH s r r zStreamWriter.can_write_eofb s ,,..r c 6 | j j S r )ro r rH s r r zStreamWriter.closee s $$&&r c 6 | j j S r )ro
is_closingrH s r r zStreamWriter.is_closingh s ))++r c V K | j j | d { y 7 wr )r r` rH s r wait_closedzStreamWriter.wait_closedk s nn..t444s )')Nc : | j j || S r )ro ry )rC namedefaults r ry zStreamWriter.get_extra_infon s --dG<!B8B 9B B)server_hostnamessl_handshake_timeoutssl_shutdown_timeoutc & K | j j d u}| j }| j d { | j j | j
|||||| d { }|| _ |j
| y 7 Q7 w)N) server_sider r r )r rp r r= start_tlsro rz )rC rx r r r r r
new_transports r r zStreamWriter.start_tls s Bnn99E >>jjl "jj 2 2 OOXz#_"7!5 3 7 7
( & 7s! 9BB
3B/B0 BBc | j j sc| j j rt j
dt y | j t j
d| t y y )Nzloop is closedz unclosed )ro r r= is_closedwarningswarnResourceWarningr rH s r r zStreamWriter.__del__ sT ))+zz##%
.@
$ 2OD ,r r )ra rb rc rD r r r& r r r r r r r ry r r r r r r r r ; sh ,- $)+/',5=-4 )-.2-1' Er r c e Zd ZdZedfdZd Zd Zd Zd Z d Z
d Z d Z d
Z
d Zd Zd
ZddZddZd Zd Zd Zy)r Nc l |dk r t d || _ |t j | _ n|| _ t | _ d| _ d | _ d | _ d | _
d| _ | j j r.t j t j d | _ y y )Nr zLimit cannot be <= 0Fr )
ValueError_limitr
r< r= bytearray_buffer_eof_waiter
_exceptionro r> rF r
extract_stacksys _getframerj )rC r r s r rD zStreamReader.__init__ s A: 34 4 < ..0DJ DJ { :: !%3%A%A
a &"D " "r c dg}| j r'|j t | j d | j r|j d | j t
k7 r |j d| j | j r |j d| j | j r |j d| j | j r |j d| j | j r|j d d j d
j | S ) Nr z byteseofzlimit=zwaiter=z
exception=r pausedr r ) r rY lenr r _DEFAULT_LIMITr r ro r> r r r s r r zStreamReader.__repr__ s << KK3t||,-V4 5 99 KK ;;. ( KK&
. / << KK'$,,!12 3 ?? KK*T__$78 9 ?? KK*T__$78 9 << KK !}}SXXd^,,r c | j S r )r rH s r r zStreamReader.exception s r c || _ | j }|*d | _ |j s|j | y y y r )r r r rR rS s r rR zStreamReader.set_exception sC DL##%$$S) & r c v | j }|*d | _ |j s|j d y y y r )r r rL rM s r _wakeup_waiterzStreamReader._wakeup_waiter s? ? DL##%!!$' & r c || _ y r r )rC r& s r r zStreamReader.set_transport s #r c | j rEt | j | j k r"d| _ | j j y y y r; )r> r r r ro resume_readingrH s r _maybe_resume_transportz$StreamReader._maybe_resume_transport s; < r r
pause_readingr] r s r r zStreamReader.feed_data s D! OO 'LLDLL!A O3
$--/ $ 4 !
(
'
' #'
's -B B%$B%c . K | j t | d | j r!d| _ | j j | j
j
| _ | j d { d | _ y 7 # d | _ w xY ww)NzF() called while another coroutine is already waiting for incoming dataF)r RuntimeErrorr> ro r r= rX )rC func_names r _wait_for_datazStreamReader._wait_for_data s << # + 4 56
6 << DL OO * * ,zz//1 ,, DL
DLs0 A(B+B :B;B ?BB BBc K d}t | } | j | d { }|S 7 # t j $ r}|j cY d }~S d }~wt j
$ r}| j j ||j r | j d |j |z = n| j j | j t |j d d }~ww xY ww)N
r )
r readuntilr IncompleteReadErrorpartialLimitOverrunErrorr
startswithconsumedclearr r args)rC sepseplenlinees r readlinezStreamReader.readline% s S
( ,,D --- 99 ++ (||&&sAJJ7LL!5!**v"5!56 ""$ ( ( * QVVAY' '
(sJ C6/ -/ C6/ C3 A
C3C6C3)BC..C33C6c K t | }|dk( r t d | j | j d} t | j }||z
|k\ rO| j j || }|dk7 rn|dz |z
}|| j
kD rt
j d| | j rEt | j }| j j t
j |d | j d d { || j
kD rt
j d| | j d ||z }| j d ||z = | j t | S 7 iw)Nr z,Separator should be at least one-byte stringr z2Separator is not found, and chunk exceed the limitr z2Separator is found, but chunk is longer than limit)r r r r findr r r r bytesr r r r )rC separatorr offsetbuflenisepchunks r r zStreamReader.readuntilD s & Y Q; KL L ?? &// !* &F &(||((F;2: !f,DKK'$66L yy dll+ ""$ 44UDAA %%k2 2 2= @ $++ ..DdL
L ^dVm, LL $- ( $$&U|
3s D E7 E5 A*E7c K | j | j |dk( ry|dk rLg } | j | j d { }|sn|j | 8dj | S | j
s%| j s| j d d { t t | j
d | }| j
d |= | j |S 7 7 Hw)Nr r read) r r r rY r r r r r
memoryviewr )rC nblocksblockr s r r zStreamReader.read s * ?? &// ! 6 q5
F "ii 44
e$
88F# #||DII%%f- - - Z -bq12 LL! $$& 5
.s& AC*C& AC* C( AC*(C*c K |dk r t d | j | j |dk( ryt | j |k r| j rEt | j }| j j
t j || | j d d { t | j |k rt | j |k( r0t | j }| j j
n0t t | j d | }| j d |= | j |S 7 w)Nr z*readexactly size can not be less than zeror readexactly) r r r r r r r r r r r
r )rC r
incompleter s r r zStreamReader.readexactly s
q5 IJ J ?? &// ! 6$,, !#yy "4<< 0
""$ 44ZCC%%m4 4 4
$,, !# t|| ! &D LL DLL1"156D RaR $$&
5s B-E/E0 E
B Ec | S r r rH s r __aiter__zStreamReader.__aiter__ s r c X K | j d { }|dk( rt |S 7 w)Nr )r StopAsyncIteration)rC vals r __anext__zStreamReader.__anext__ s+ MMO# #:$ $
$s *(*)r )r )ra rb rc rj r rD r r rR r r r r r r r r r r r r r r r r r r sf +$ ",-$ *($-
.$, 8 >Y v1f'Rr r r r )__all__r? socketr r rg hasattr r r
r r r logr tasksr r r r r r
Protocolr9 r r r r r r r s '
69 <