cj| z d dl Z d dlZd dlZ d dlZddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z eej ej fZ G d d ej Z G d
d ej Zd Zd
Z G d de j( e j* Z G d dej. Zy# e$ r dZY w xY w) N ) constants)
exceptions) protocols)
transports)loggerc e Zd ZdZdZdZdZdZy)SSLProtocolState UNWRAPPED DO_HANDSHAKEWRAPPEDFLUSHINGSHUTDOWNN)__name__
__module__ __qualname__r r r
r r )/usr/lib64/python3.12/asyncio/sslproto.pyr
r
s I!LGHHr r
c e Zd ZdZdZdZdZy)AppProtocolState
STATE_INITSTATE_CON_MADE STATE_EOFSTATE_CON_LOSTN)r r r r r r r r r r r r s J%NI%Nr r c Z | r t d t j }|sd|_ |S )Nz(Server side SSL needs a valid SSLContextF)
ValueErrorsslcreate_default_contextcheck_hostname) server_sideserver_hostname
sslcontexts r _create_transport_contextr$ / s2 CDD ++-J $)
! r c | ||dz }n
|}d|z }n| }||dz }n|}||cxk\ rdk\ sn t d|d|d ||fS )Ni r zhigh (z) must be >= low (z) must be >= 0)r )highlowkbhilos r add_flowcontrol_defaultsr, = sh | ;dBBRB
{
1W
=q= b" # # r6Mr c e Zd ZdZej
j Zd ZddZ d Z
d Z d Z d Z
efd Zd
Zd Zd Zd d
Zd Zd Zd dZd Zd Zed Zd Zd Z d Z d Z d Z d Z d Z!y) _SSLProtocolTransportTc . || _ || _ d| _ y NF)_loop
_ssl_protocol_closed)selfloop ssl_protocols r __init__z _SSLProtocolTransport.__init__X s
) r Nc < | j j || S N)r2 _get_extra_infor4 namedefaults r get_extra_infoz$_SSLProtocolTransport.get_extra_info] s 1!!11$@@r c : | j j | y r9 )r2 _set_app_protocol)r4 protocols r set_protocolz"_SSLProtocolTransport.set_protocola s ,,X6r c . | j j S r9 )r2
_app_protocolr4 s r get_protocolz"_SSLProtocolTransport.get_protocold s !!///r c R | j xs | j j S r9 )r3 r2 _is_transport_closingrE s r
is_closingz _SSLProtocolTransport.is_closingg s ||It 1 1 G G IIr c p | j s"d| _ | j j y d | _ y NT)r3 r2 _start_shutdownrE s r closez_SSLProtocolTransport.closej s1 || DL . . 0!%D r c X | j s d| _ |j dt y y )NTz9unclosed transport )r3 warnResourceWarning)r4 _warningss r __del__z _SSLProtocolTransport.__del__x s) || DL NN*
, r c 0 | j j S r9 )r2 _app_reading_pausedrE s r
is_readingz _SSLProtocolTransport.is_reading s %%9999r c : | j j y r9 )r2 _pause_readingrE s r
pause_readingz#_SSLProtocolTransport.pause_reading s
))+r c : | j j y r9 )r2 _resume_readingrE s r resume_readingz$_SSLProtocolTransport.resume_reading s
**,r c r | j j || | j j y r9 )r2 _set_write_buffer_limits_control_app_writingr4 r' r( s r set_write_buffer_limitsz-_SSLProtocolTransport.set_write_buffer_limits s1 $
33D#> //1r c Z | j j | j j fS r9 )r2 _outgoing_low_water_outgoing_high_waterrE s r get_write_buffer_limitsz-_SSLProtocolTransport.get_write_buffer_limits * ""66""779 9r c 8 | j j S r9 )r2 _get_write_buffer_sizerE s r get_write_buffer_sizez+_SSLProtocolTransport.get_write_buffer_size s ;!!88::r c r | j j || | j j y r9 )r2 _set_read_buffer_limits_control_ssl_readingr_ s r set_read_buffer_limitsz,_SSLProtocolTransport.set_read_buffer_limits s1 $
224= //1r c Z | j j | j j fS r9 )r2 _incoming_low_water_incoming_high_waterrE s r get_read_buffer_limitsz,_SSLProtocolTransport.get_read_buffer_limits re r c 8 | j j S r9 )r2 _get_read_buffer_sizerE s r get_read_buffer_sizez*_SSLProtocolTransport.get_read_buffer_size s 9!!7799r c . | j j S r9 )r2 _app_writing_pausedrE s r _protocol_pausedz&_SSLProtocolTransport._protocol_paused s !!555r c t |t t t f s!t dt | j |sy | j j |f y )Nz+data: expecting a bytes-like instance, got )
isinstancebytes bytearray
memoryview TypeErrortyper r2 _write_appdatar4 datas r writez_SSLProtocolTransport.write s]
$ : >? ##':#6#6"7 9 :
: ))4'2r c < | j j | y r9 )r2 r~ )r4 list_of_datas r
writelinesz _SSLProtocolTransport.writelines s
)),7r c t r9 )NotImplementedErrorrE s r write_eofz _SSLProtocolTransport.write_eof s "!r c yr0 r rE s r
can_write_eofz#_SSLProtocolTransport.can_write_eof s Or c ( | j d y r9 ) _force_closerE s r abortz_SSLProtocolTransport.abort s
$ r c b d| _ | j | j j | y y rK )r3 r2 _abortr4 excs r r z"_SSLProtocolTransport._force_close s. ) % %c * *r c | j j j | | j xj t | z
c_ y r9 )r2 _write_backlogappend_write_buffer_sizelenr s r _test__append_write_backlogz1_SSLProtocolTransport._test__append_write_backlog s7 ))006 --T:-r r9 NN)"r r r _start_tls_compatibler
_SendfileModeFALLBACK_sendfile_compatibler7 r> rB rF rI rM warningsrR rU rX r[ r` rd rh rl rp rs propertyrv r r r r r r r r r r r. r. R s !$22;;
A70J & !) ,:,-2,9;2,9: 6 6 38" +
;r r. c e Zd ZdZdZdZdZ d+dZd Zd,dZ d Z
d Z d Z d Z
d
Zd Zd Zd,d
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d Z!d Z"d Z#d Z$d! Z%d-d"Z&d# Z'd$ Z(d% Z)d-d&Z*d' Z+d( Z,d) Z-d.d*Z.y)/ SSLProtocoli Nc
t t d t | j | _ t | j | _ |t j }n|dk rt d| | t j } n| dk rt d| |s t || }|| _ |r
|s|| _
nd | _
|| _ t | | _ t# j$ | _ d| _ || _ || _ | j/ | d | _ d| _ d | _ || _ | | _ t j: | _ t j: | _ t@ jB | _" d| _# |rtH jJ | _& ntH jN | _& | j jQ | j< | j> | j | j | _) d| _* d| _+ d| _, d| _- d| _. | j_ d| _0 d| _1 d| _2 d| _3 | ji | jk y )Nz stdlib ssl module not availabler z7ssl_handshake_timeout should be a positive number, got z6ssl_shutdown_timeout should be a positive number, got )r# F)r! r" )6r RuntimeErrorrz max_size _ssl_bufferr{ _ssl_buffer_viewr SSL_HANDSHAKE_TIMEOUTr SSL_SHUTDOWN_TIMEOUTr$ _server_side_server_hostname _sslcontextdict_extra collectionsdequer r _waiterr1 r@ _app_transport_app_transport_created
_transport_ssl_handshake_timeout_ssl_shutdown_timeout MemoryBIO _incoming _outgoingr
r _state
_conn_lostr r
_app_stater wrap_bio_sslobj_ssl_writing_pausedrT _ssl_reading_pausedro rn rj
_eof_receivedru rc rb r] _get_app_transport)
r4 r5 app_protocolr# waiterr! r" call_connection_madessl_handshake_timeoutssl_shutdown_timeouts
r r7 zSSLProtocol.__init__ sE
; @A A$T]]3 *4+;+; < ($-$C$C !
"a
' ,-/0
0 '#,#A#A
!Q
& +,./
/ 2_.J ( ;$3D !$(D !% j1 * / / 1"#
|, "&+# &;#%9" &00 . 9 9DO . = =DO''00 NNDNN )) 1 1 1 3 $) #( #( $%!#$ $$& "#( $%!#$ %%' !r c || _ t |d rDt |t j r*|j
| _ |j | _ d| _ y d| _ y )N
get_bufferTF)
rD hasattrrx r BufferedProtocolr _app_protocol_get_bufferbuffer_updated _app_protocol_buffer_updated_app_protocol_is_buffer)r4 r s r r@ z SSLProtocol._set_app_protocola sP ) L, /<)C)CD,8,C,CD )0<0K0KD -+/D (+0D (r c | j y | j j s@|#| j j | d | _ y | j j d d | _ y r9 )r cancelled
set_exception
set_resultr s r _wakeup_waiterzSSLProtocol._wakeup_waiterl sZ << ||%%' **3/ ''- r c | j 9| j r t d t | j | | _ d| _ | j S )Nz$Creating _SSLProtocolTransport twiceT)r r r r. r1 rE s r r z SSLProtocol._get_app_transportv sJ &**"#IJJ"7
D"ID *.D '"""r c V | j d uxr | j j S r9 )r rI rE s r rH z!SSLProtocol._is_transport_closing~ s# d*Kt/I/I/KKr c 4 || _ | j y r9 )r _start_handshake)r4 transports r connection_madezSSLProtocol.connection_made s $ r c J | j j | j j | xj dz
c_ | j
d| j
_ | j t j k7 r| j t j k( s | j t j k( rEt j | _
| j j! | j" j$ | | j' t j( d | _ d | _ d | _ | j- | | j. r!| j. j1 d | _ | j2 r"| j2 j1 d | _ y y )Nr T)r clearr readr r r3 r r
r r r r r r r1 call_soonrD connection_lost
_set_stater r r _shutdown_timeout_handlecancel_handshake_timeout_handler s r r zSSLProtocol.connection_lost s>
!!# 1 **.D ' ;;*77 7#3#B#BB#3#=#=="2"A"A
$$T%7%7%G%GM (223 " ! C ( ( ) ) 0 0 2,0D ) ) ) * * 1 1 3-1D * *r c |}|dk s|| j kD r | j }t | j |k r*t | | _ t | j | _ | j
S Nr )r r r rz r{ r )r4 nwants r r zSSLProtocol.get_buffer s` 19t}},==D t 4 ' (D $.t/?/?$@D !$$$r c | j j | j d | | j t j
k( r| j
y | j t j k( r| j y | j t j k( r| j y | j t j k( r| j y y r9 )
r r r r r
r
_do_handshaker
_do_readr _do_flushr _do_shutdown)r4 nbytess r r zSSLProtocol.buffer_updated s T 2 27F ;< ;;*77 7
[[ , 4 4
4 MMO
[[ , 5 5
5 NN
[[ , 5 5
5 6r c d| _ | j j rt j d| | j
t j k( r| j t y | j
t j k( r=| j t j | j ry| j y | j
t j k( r@| j | j t j | j# y | j
t j k( r| j# y y # t$ $ r | j& j) w xY w)NTz%r received EOF)r r1 get_debugr debugr r
r _on_handshake_completeConnectionResetErrorr
r r rT r _do_writer r Exceptionr rM rE s r eof_receivedzSSLProtocol.eof_received s " zz##% .5{{ . ; ;;++,@A 0 8 88 0 9 9:++ NN$ 0 9 99 0 9 9:!!# 0 9 99!!# : OO ! ! # s& A"E -AE 6E A E $-E %E8c || j v r| j | S | j | j j || S |S r9 )r r r> r; s r r: zSSLProtocol._get_extra_info sC 4;; ;;t$ $
__
(??11$@ @Nr c & d}|t j k( rd}n| j t j k( r|t j k( rd}n| j t j k( r|t j k( rd}ne| j t j k( r|t j
k( rd}n2| j t j
k( r|t j k( rd}|r|| _ y t dj | j | )NFTz!cannot switch state from {} to {}) r
r r r r
r r r format)r4 new_statealloweds r r zSSLProtocol._set_state s (22 2G
KK+55 5 )66 6G
KK+88 8 )11 1G
KK+33 3 )22 2G
KK+44 4 )22 2G #DK 3::KK,-
-r c n j j r6t j d j j _ nd _ j
t j j j j fd _ j y )Nz%r starts SSL handshakec $ j S r9 )_check_handshake_timeoutrE s r z.SSLProtocol._start_handshake..$ s $*G*G*Ir )
r1 r r r time_handshake_start_timer r
r
call_laterr r r rE s `r r z SSLProtocol._start_handshake s :: ! LL2D 9)- ):D &)-D & (556
JJ ! !$"="="I
K
&
r c | j t j k( r+d| j d}| j t | y y )Nz$SSL handshake is taking longer than z! seconds: aborting the connection)r r
r r _fatal_errorConnectionAbortedError)r4 msgs r r z$SSLProtocol._check_handshake_timeout( sN ;;*77 76../ 0*+
4S 9 :
8r c | j j | j d y # t $ r | j Y y t
j $ r}| j | Y d }~y d }~ww xY wr9 )r do_handshaker SSLAgainErrors_process_outgoingr SSLErrorr s r r zSSLProtocol._do_handshake1 sb . LL % % '
' ' - % " " $|| - ' ' , , -s . A6 A6A11A6c | j !| j j d | _ | j } | | j t j
n||j
}| j j rA| j j! | j" z
}t% j& d| |dz | j( j+ ||j- |j/ | | j0 t2 j4 k( r>t2 j6 | _ | j8 j; | j= | j | j? y # t $ rm}d }| j t j t |t j rd}nd}| j || | j | Y d }~y d }~ww xY w)Nz1SSL handshake failed on verifying the certificatezSSL handshake failedz %r: SSL handshake took %.1f msg @@)peercertcipher compression
ssl_object) r r r r r
r
getpeercertr r rx r CertificateErrorr r r1 r r r r r r updater
r r r r r rD r r r )r4
handshake_excsslobjr r r dts r r z"SSLProtocol._on_handshake_complete; s ) ) 5 * * 1 1 3-1D * $ 0 8 89## ))+H :: ! "T%?%??B LL94c J
H"(--/'-'9'9';&, . ?? . 9 9 9 . = =DO . .t/F/F/H I
1 M OO , 6 6 7#s 3 34I, c3 ' $ s 4F G7
A#G22G7c j t j t j t j fv ry j
d j
_ j t j k( r j d y j t j j j j fd _
j y )NTc $ j S r9 )_check_shutdown_timeoutrE s r r z-SSLProtocol._start_shutdown..u s 446r )r r
r r r r r3 r r r r1 r r r r rE s `r rL zSSLProtocol._start_shutdownd s KK )) )) **
**.D ' ;;*77 7 KK OO , 5 5 6,0JJ,A,A**6-D )
NN r c | j t j t j fv r/| j j t
j d y y )NzSSL shutdown timed out)r r
r r r r r TimeoutErrorrE s r r z#SSLProtocol._check_shutdown_timeouty sN KK )) ))
OO ( (''(@A
C
r c | j | j t j | j y r9 )r r r
r r rE s r r zSSLProtocol._do_flush s*
(112 r c J | j s| j j | j | j | j d y # t $ r | j Y y t j $ r}| j | Y d }~y d }~ww xY wr9 ) r r unwrapr _call_eof_received_on_shutdown_completer r r r s r r zSSLProtocol._do_shutdown s
-%% ##%
" " $ # # % & &t , % " " $|| , & &s + + ,s &A B"5B"B B"c | j !| j j d | _ |r| j | y | j j | j
j y r9 )r r r r1 r r rM )r4 shutdown_excs r r z!SSLProtocol._on_shutdown_complete sU ( ( 4 ) ) 0 0 2,0D ) l + JJ !6!6 7r c | j t j | j | j j | y y r9 )r r
r r r r s r r zSSLProtocol._abort s6 (223 ?? & OO ( ( - 'r c 8 | j t j t j t j fv rH| j
t j k\ rt j d | xj
dz
c_ y |D ]; }| j j | | xj t | z
c_ = | j t j k( r| j y y # t $ r }| j# |d Y d }~y d }~ww xY w)NzSSL connection is closedr Fatal error on SSL protocol)r r
r r r r r !LOG_THRESHOLD_FOR_CONNLOST_WRITESr warningr r r r r
r r r )r4 r r exs r r~ zSSLProtocol._write_appdata s KK )) )) **
)"M"MM 9: OOq O D & &t , # #s4y 0 # ! A{{ . 6 66 7 A b"? @ @ As -C4 4 D=DDc ~ | j r| j d }| j j | }t | }||k r(||d | j d< | xj |z c_ n"| j d= | xj |z c_ | j r| j
y # t
$ r Y w xY wr )r r r r r r r )r4 r countdata_lens r r zSSLProtocol._do_write s %%**1- **40 t98#-1%&\D''*++u4+ ++A.++x7+ %%
s B B0 0 B<;B<c | j s@| j j }t | r| j j | | j
y r9 )r r r r r r r^ r s r r z SSLProtocol._process_outgoing sB ''>>&&(D4y%%d+ !!#r c | j t j t j fvry | j sZ| j
r| j
n| j | j r| j n| j | j y # t $ r }| j |d Y d }~y d }~ww xY w)Nr! )r r
r
r rT r _do_read__buffered_do_read__copiedr r r rk r r )r4 r$ s r r zSSLProtocol._do_read s KK (( ))
A++//++-))+&&NN$**, % % ' A b"? @ @ As A6B& & C /CC c d}d} j j }t | } j j || }|dkD rY|}||k r4 j j ||z
||d }|dkD r||z
}nn$||k r4 j
j
fd |dkD r j | |s! j j y y # t $ r Y Ew xY w)Nr r c $ j S r9 )r rE s r r z0SSLProtocol._do_read__buffered.. s r ) r rr r r r r1 r r r r rL )r4 offsetr& bufwantss ` r r* z SSLProtocol._do_read__buffered s ++D,F,F,HIC LL%%eS1Eqy un LL - -efnc&'l KE qy % un JJ(()@A A: - -f 5 # # % " s AC%
C% % C10C1c d}d}d} | j j | j }|sn$|rd}d}|}n|rd}|g}nj | L |r | j
j
n,|s*| j
j
dj |s!| j | j y y # t $ r Y yw xY w)N 1TFr )
r r r r r rD
data_receivedjoinr rL )r4 chunkzeroonefirstr s r r+ z SSLProtocol._do_read__copied
s ))$--8 D C !E C !5>DKK& , ,U 3 , ,SXXd^ < # # % "
s A
C CCc > | j t j k( rHt j | _ | j j }|rt
j d y y y # t t f$ r t $ r }| j |d Y d }~y d }~ww xY w)Nz?returning true from eof_received() has no effect when using sslz Error calling eof_received()) r r r r rD r r r# KeyboardInterrupt
SystemExit
BaseExceptionr )r4 keep_openr$ s r r z SSLProtocol._call_eof_received( s
B"2"A"AA"2"<"< . . ; ; = NN $B C B ":. B b"@ A A Bs A#A( (B BB c Z | j }|| j k\ r/| j s#d| _ | j j y || j k r0| j r#d| _ | j j y y y # t
t f$ r t $ r4}| j j d|| j | d Y d }~y d }~ww xY w# t
t f$ r t $ r4}| j j d|| j | d Y d }~y d }~ww xY w)NTz protocol.pause_writing() failedmessage exceptionr rA Fz protocol.resume_writing() failed)
rg rc ru rD
pause_writingr: r; r< r1 call_exception_handlerr rb resume_writing)r4 sizer s r r^ z SSLProtocol._control_app_writing7 s$ **, 4,, ,T5M5M'+D $
""002 T--
-$2J2J',D $
""113 3K
- &z2
11 @!$!%!4!4 $ 3
&z2
11 A!$!%!4!4 $ 3
s/ B 2C C'*CC D*6*D%%D*c H | j j | j z S r9 )r pendingr rE s r rg z"SSLProtocol._get_write_buffer_sizeT s ~~%%(?(???r c \ t ||t j \ }}|| _ || _ y r9 )r, r !FLOW_CONTROL_HIGH_WATER_SSL_WRITErc rb r_ s r r] z$SSLProtocol._set_write_buffer_limitsW s. , #yBBD c$(!#& r c d| _ y rK )rT rE s r rW zSSLProtocol._pause_reading_ s
#' r c n j r(d _ fd} j j | y y )NFc j t j k( r j y j t j k( r j y j t j k( r j y y r9 )r r
r
r r r r r rE s r resumez+SSLProtocol._resume_reading..resumef s` ;;"2":"::MMO[[$4$=$==NN$[[$4$=$==%%' >r )rT r1 r )r4 rM s ` r rZ zSSLProtocol._resume_readingb s2 # #',D $
(
JJ ( $r c | j }|| j k\ r.| j s"d| _ | j j y || j
k r/| j r"d| _ | j j
y y y )NTF)rr ro r r rX rn r[ )r4 rE s r rk z SSLProtocol._control_ssl_readingq su ))+ 4,, ,T5M5M'+D $ OO ) ) +
T--
-$2J2J',D $ OO * * , 3K
-r c \ t ||t j \ }}|| _ || _ y r9 )r, r FLOW_CONTROL_HIGH_WATER_SSL_READro rn r_ s r rj z#SSLProtocol._set_read_buffer_limitsz s. , #yAAC c$(!#& r c . | j j S r9 )r rG rE s r rr z!SSLProtocol._get_read_buffer_size s ~~%%%r c d| _ y rK )r rE s r rB zSSLProtocol.pause_writing s $( r c 4 d| _ | j y r0 )r r rE s r rD zSSLProtocol.resume_writing s $) r c f | j r| j j | t |t r5| j j rt
j d| |d y y t |t j s+| j j ||| j | d y y )Nz%r: %sT)exc_infor? ) r r rx OSErrorr1 r r r r CancelledErrorrC )r4 r r@ s r r zSSLProtocol._fatal_error s ?? OO ( ( - c7 #zz##% XtWtD &C!:!:; JJ - -" !__ /
ra s
?**C,?,?@Ntyy
&tyy
& *r;J 8 8 & 0 0r;jZ
),, Z
{
Cs B0 0B:9B: