o
0iCk @ s d dl Z d dlZzd dlZW n ey dZY nw ddlmZ ddlmZ ddlmZ ddlm Z dd Z
d Z d
Z d Z
d ZG d
d deZG dd dejejZG dd dejZdS ) N ) constants) protocols)
transports)loggerc C s" | rt dt }|sd|_|S )Nz(Server side SSL needs a valid SSLContextF)
ValueErrorsslZcreate_default_contextZcheck_hostname) server_sideserver_hostname
sslcontext r 7/opt/alt/python310/lib64/python3.10/asyncio/sslproto.py_create_transport_context s r Z UNWRAPPEDZ DO_HANDSHAKEZWRAPPEDZSHUTDOWNc @ s| e Zd Z dZdddZedd Zedd Zed d
Zed d Z dd
dZ
dddZ dd Z dddZ
dddZdS ) _SSLPipei Nc C sJ || _ || _|| _t| _t | _t | _d | _ d| _
d | _ d | _ d S NF)
_context _server_side_server_hostname
_UNWRAPPED_stater Z MemoryBIO _incoming _outgoing_sslobj
_need_ssldata
_handshake_cb _shutdown_cb)selfcontextr r
r r r
__init__7 s
z_SSLPipe.__init__c C | j S N)r r r r r
r M s z_SSLPipe.contextc C r r )r r! r r r
ssl_objectR s z_SSLPipe.ssl_objectc C r r )r r! r r r
need_ssldataZ s z_SSLPipe.need_ssldatac C s | j tkS r )r _WRAPPEDr! r r r
wrapped` s
z_SSLPipe.wrappedc C sT | j tkr
td| jj| j| j| j| jd| _ t
| _ || _ | j ddd\}}|S )Nz"handshake in progress or completed)r r
T)only_handshake)
r r RuntimeErrorr Zwrap_bior r r r r
_DO_HANDSHAKEr feed_ssldatar callbackssldataappdatar r r
do_handshakei s
z_SSLPipe.do_handshakec C sD | j tkr
td| j tkrtdt| _ || _| d\}}|S )Nzno security layer presentzshutdown in progressr& )r r r( _SHUTDOWNr r* r+ r r r
shutdown~ s
z_SSLPipe.shutdownc C s | j | d\}}d S )Nr& )r Z write_eofr* )r r- r. r r r
feed_eof s
z_SSLPipe.feed_eofFc
C s | j tkr|r|g}g |fS g }g |fS d| _|r | j| g }g }zZ| j tkrA| j t| _ | j r:| d |rA||fW S | j tkrX | j
| j }| | |sVnqGn%| j t
krp| j d | _t| _ | jro| n
| j tkr}| | j
W n: tjtjfy } z*t|dd }|tjtjtjfvr| j tkr| j r| | |tjk| _W Y d }~nd }~ww | jjr| | j
||fS )NFTerrno)r r r r writer) r r/ r$ r readmax_sizeappendr0 Zunwrapr r SSLErrorCertificateErrorgetattrSSL_ERROR_WANT_READSSL_ERROR_WANT_WRITESSL_ERROR_SYSCALLr pending)r datar' r. r- chunkexc exc_errnor r r
r* sj
z_SSLPipe.feed_ssldatar c
C s | j tkr |t|k r||d g}ng }|t|fS g }t|} d| _z|t|k r9|| j||d 7 }W n5 tjyo } z(t |dd }|j
dkrTtj }|_ |tj tj
tjfvr_ |tj k| _W Y d }~nd }~ww | jjr||| j |t|ks| jr ||fS q#)NTFr3 ZPROTOCOL_IS_SHUTDOWN)r r len
memoryviewr r r4 r r8 r: reasonr; r3 r< r= r r> r7 r5 )r r? offsetr- ZviewrA rB r r r
feed_appdata s@
z_SSLPipe.feed_appdatar )F)r )__name__
__module__ __qualname__r6 r propertyr r" r# r% r/ r1 r2 r* rG r r r r
r # s"
Kr c @ s e Zd ZejjZdd Zd$ddZdd Z dd Z
d
d Z d d
Z e
jfddZdd Zdd Zdd Zd%ddZdd Zdd Zed d Zd d Zd d! Zd"d# ZdS )&_SSLProtocolTransportc C s || _ || _d| _d S r )_loop
_ssl_protocol_closed)r loopZ ssl_protocolr r r
r s
z _SSLProtocolTransport.__init__Nc C s | j ||S r )rN _get_extra_infor namedefaultr r r
get_extra_info& s z$_SSLProtocolTransport.get_extra_infoc C s | j | d S r )rN _set_app_protocol)r protocolr r r
set_protocol* s z"_SSLProtocolTransport.set_protocolc C s | j jS r )rN
_app_protocolr! r r r
get_protocol- s z"_SSLProtocolTransport.get_protocolc C s | j S r )rO r! r r r
is_closing0 s z _SSLProtocolTransport.is_closingc C s d| _ | j d S NT)rO rN _start_shutdownr! r r r
close3 s z_SSLProtocolTransport.closec C s* | j s|d| t| d | d S d S )Nzunclosed transport )source)rO ResourceWarningr^ )r Z_warnr r r
__del__> s z _SSLProtocolTransport.__del__c C s | j j}|d u r td| S )Nz*SSL transport has not been initialized yet)rN
_transportr(
is_reading)r Ztrr r r
rc C s z _SSLProtocolTransport.is_readingc C | j j d S r )rN rb
pause_readingr! r r r
re I z#_SSLProtocolTransport.pause_readingc C rd r )rN rb resume_readingr! r r r
rg Q rf z$_SSLProtocolTransport.resume_readingc C s | j j|| d S r )rN rb set_write_buffer_limits)r ZhighZlowr r r
rh Y s z-_SSLProtocolTransport.set_write_buffer_limitsc C | j j S r )rN rb get_write_buffer_sizer! r r r
rj n s z+_SSLProtocolTransport.get_write_buffer_sizec C ri r )rN rb get_write_buffer_limitsr! r r r
rk r s z-_SSLProtocolTransport.get_write_buffer_limitsc C s
| j jjS r )rN rb _protocol_pausedr! r r r
rl x s
z&_SSLProtocolTransport._protocol_pausedc C s> t |tttfstdt|j |sd S | j| d S )Nz+data: expecting a bytes-like instance, got )
isinstancebytes bytearrayrD TypeErrortyperH rN _write_appdatar r? r r r
r4 } s z_SSLProtocolTransport.writec C s dS r r r! r r r
can_write_eof s z#_SSLProtocolTransport.can_write_eofc C s | j d| _d S r\ )rN _abortrO r! r r r
abort s
z_SSLProtocolTransport.abortr )NN)rH rI rJ r Z
_SendfileModeZFALLBACKZ_sendfile_compatibler rU rX rZ r[ r^ warningswarnra rc re rg rh rj rk rK rl r4 rt rv r r r r
rL s(
rL c @ s e Zd Z d+ddZdd Zd,dd Zd
d Zd d
Zdd Zdd Z dd Z
dd Z d,ddZ dd Z
dd Zd d Zd d Zd d! Zd"d# Zd-d%d&Zd'd( Zd)d* ZdS ). SSLProtocolFNTc C s t d u rtd|d u rtj}n |dkrtd| |s"t||}|| _|r-|s-|| _nd | _|| _t |d| _
t | _
d| _|| _|| _| | t| j| | _d | _d| _d| _d| _d | _|| _|| _d S )Nz stdlib ssl module not availabler z7ssl_handshake_timeout should be a positive number, got )r F)r r( r ZSSL_HANDSHAKE_TIMEOUTr r r r _sslcontextdict_extra collectionsdeque_write_backlog_write_buffer_size_waiterrM rV rL _app_transport_sslpipe_session_established
_in_handshake _in_shutdownrb _call_connection_made_ssl_handshake_timeout) r rP app_protocolr Zwaiterr r
Zcall_connection_madeZssl_handshake_timeoutr r r
r sB
zSSLProtocol.__init__c C s || _ t|tj| _d S r )rY rm r ZBufferedProtocol_app_protocol_is_buffer)r r r r r
rV s
z SSLProtocol._set_app_protocolc C sD | j d u rd S | j s |d ur| j | n| j d d | _ d S r )r Z cancelledZ
set_exceptionZ
set_resultr rA r r r
_wakeup_waiter s
zSSLProtocol._wakeup_waiterc C s( || _ t| j| j| j| _| d S r )rb r rz r r r _start_handshake)r transportr r r
connection_made s zSSLProtocol.connection_madec C sp | j rd| _ | j| jj| n | jd urd| j_d | _d | _t| dd r+| j
| | d | _d | _ d S )NFT_handshake_timeout_handle)
r rM call_soonrY connection_lostr rO rb r: r cancelr r r r r r
r s
zSSLProtocol.connection_lostc C | j d S r )rY
pause_writingr! r r r
r zSSLProtocol.pause_writingc C r r )rY resume_writingr! r r r
r r zSSLProtocol.resume_writingc C s | j d u rd S z
| j |\}}W n" ttfy ty4 } z
| |d W Y d }~d S d }~ww |D ]}| j| q7|D ]A}|r}z| jrRt
| j | n| j | W qB ttfyc ty| } z| |d W Y d }~ d S d }~ww |
d S d S )NzSSL error in data receivedz/application protocol failed to receive SSL data)r r*
SystemExitKeyboardInterrupt
BaseException _fatal_errorrb r4 r r Z _feed_data_to_buffered_protorY
data_receivedr] )r r? r- r. er@ Zexr r r
r sF
zSSLProtocol.data_receivedc C s~ z7| j r
td| | t | js)| j }|r1t d W | j
d S W | j
d S W | j
d S | j
w )Nz%r received EOFz?returning true from eof_received() has no effect when using ssl) rM get_debugr debugr ConnectionResetErrorr rY eof_receivedZwarningrb r^ )r Z keep_openr r r
r 2 s
zSSLProtocol.eof_receivedc C s0 || j v r
| j | S | jd ur| j||S |S r )r| rb rU rR r r r
rQ H s
zSSLProtocol._get_extra_infoc C s0 | j rd S | jr| d S d| _ | d d S )NTr& )r r ru rr r! r r r
r] P s zSSLProtocol._start_shutdownc C s. | j |df | jt|7 _| d S )Nr )r r7 r rC _process_write_backlogrs r r r
rr Y s zSSLProtocol._write_appdatac C s\ | j rtd| | j | _nd | _d| _| jd | j | j
| j | _ |
d S )Nz%r starts SSL handshakeT)r& r )rM r r r time_handshake_start_timer r r7 Z
call_laterr _check_handshake_timeoutr r r! r r r
r ^ s
z SSLProtocol._start_handshakec C s. | j du rd| j d}| t| d S d S )NTz$SSL handshake is taking longer than z! seconds: aborting the connection)r r r ConnectionAbortedError)r msgr r r
r m s
z$SSLProtocol._check_handshake_timeoutc
C s d| _ | j | jj}z |d ur|| }W n- ttfy" tyE } zt |t
j r2d}nd}| || W Y d }~d S d }~ww | j
r\| j
| j }td| |d | jj|| | |d | jrt| j| j | d| _ | j
| j d S )NFz1SSL handshake failed on verifying the certificatezSSL handshake failedz %r: SSL handshake took %.1f msg @@)peercertcipher compressionr" T) r r r r r" Z getpeercertr r r rm r r9 r rM r r r r r r| updater r r rY r r r r r r )r Z
handshake_excZsslobjr rA r Zdtr r r
_on_handshake_completev s<
z"SSLProtocol._on_handshake_completec
C sL | j d u s
| jd u r d S zgtt| jD ]\}| jd \}}|r)| j||\}}n|r5| j| j}d}n | j| j }d}|D ]}| j
| q@|t|k rc||f| jd< | j j r_| j W d S | jd= | j
t|8 _
qW d S ttfy} ty } z | jr| | n
| |d W Y d }~d S W Y d }~d S d }~ww )Nr r z Fatal error on SSL transport)rb r rangerC r rG r/ r r1 _finalizer4 Z_pausedrg r r r r r r )r ir? rF r- r@ rA r r r
r sB
z"SSLProtocol._process_write_backlogFatal error on transportc C sZ t |tr| j rtjd| |dd n | j||| j| d | jr+| j| d S d S )Nz%r: %sT)exc_info)messageZ exceptionr rW ) rm OSErrorrM r r r Zcall_exception_handlerrb Z _force_close)r rA r r r r
r s
zSSLProtocol._fatal_errorc C s" d | _ | jd ur| j d S d S r )r rb r^ r! r r r
r s
zSSLProtocol._finalizec C s< z| j d ur| j W | d S W | d S | w r )rb rv r r! r r r
ru s
zSSLProtocol._abort)FNTNr )r )rH rI rJ r rV r r r r r r r rQ r] rr r r r r r r ru r r r r
ry s0
.
&
)
+ ry )r} rw r ImportError r r r logr r r r) r$ r0 objectr Z_FlowControlMixinZ TransportrL ZProtocolry r r r r
s, y~