U
e5dJj @ s d dl Z d dlZz d dlZW n ek
r4 dZY nX ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
dd Z d
Z d Z
d Zd
ZG dd deZG dd dejejZG dd dejZdS ) N ) base_events) constants) protocols)
transports)loggerc C s" | r t dt }|s d|_|S )Nz(Server side SSL needs a valid SSLContextF)
ValueErrorsslZcreate_default_contextZcheck_hostname) server_sideserver_hostname
sslcontext r
(/usr/lib64/python3.8/asyncio/sslproto.py_create_transport_context s r Z UNWRAPPEDZ DO_HANDSHAKEZWRAPPEDZSHUTDOWNc @ sz e Zd ZdZdddZedd Zedd Zed d
Zed d Z dd
dZ
dddZ dd Z d ddZ
d ddZdS ) _SSLPipei Nc C sH || _ || _|| _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__8 s
z_SSLPipe.__init__c C s | j S N)r r r
r
r r N s z_SSLPipe.contextc C s | j S r )r r! r
r
r
ssl_objectS s z_SSLPipe.ssl_objectc C s | j S r )r r! r
r
r need_ssldata[ s z_SSLPipe.need_ssldatac C s
| j tkS r )r _WRAPPEDr! r
r
r wrappeda s z_SSLPipe.wrappedc C sR | j tkrtd| 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_handshakej s
z_SSLPipe.do_handshakec C sB | j tkrtd| j tkr$tdt| _ || _| 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}ng }g |fS d| _|r8| j| g }g }z| j tkrz| j t| _ | j rl| d |rz||fW S | j tkr| j
| j }| | |sqqnJ| j t
kr| j d | _t| _ | jr| n| j tkr| | j
W nz tjtjfk
rl } zRt|dd }|tjtjtjfkrP| j tkrN| j rN| | |tjk| _W 5 d }~X Y nX | jjr| | j
||fS )NFerrno)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* sZ
z_SSLPipe.feed_ssldatar c
C s | j tkr6|t|k r&||d g}ng }|t|fS g }t|}d| _z(|t|k rn|| j||d 7 }W nh tjk
r } zHt |dd }|j
dkrtj }|_ |tj tj
tjfkr |tj k| _W 5 d }~X Y nX | jjr|| j |t|ks | jrBq qB||fS )NFr3 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 s4
z_SSLPipe.feed_appdata)N)N)N)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 Zedd 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_closing1 s z _SSLProtocolTransport.is_closingc C s d| _ | j d S NT)rO rN _start_shutdownr! r
r
r close4 s z_SSLProtocolTransport.closec C s& | j s"|d| t| d | d S )Nzunclosed transport )source)rO ResourceWarningr^ )r Z_warnr
r
r __del__? s z _SSLProtocolTransport.__del__c C s | j j}|d krtd| S )Nz*SSL transport has not been initialized yet)rN
_transportr(
is_reading)r Ztrr
r
r rc D s z _SSLProtocolTransport.is_readingc C s | j j d S r )rN rb
pause_readingr! r
r
r rd J s z#_SSLProtocolTransport.pause_readingc C s | j j d S r )rN rb resume_readingr! r
r
r re R s z$_SSLProtocolTransport.resume_readingc C s | j j|| d S r )rN rb set_write_buffer_limits)r ZhighZlowr
r
r rf Z s z-_SSLProtocolTransport.set_write_buffer_limitsc C s | j j S r )rN rb get_write_buffer_sizer! r
r
r rg o s z+_SSLProtocolTransport.get_write_buffer_sizec C s
| j jjS r )rN rb _protocol_pausedr! r
r
r rh s s z&_SSLProtocolTransport._protocol_pausedc C s< t |tttfs$tdt|j |s,d S | j| d S )Nz+data: expecting a bytes-like instance, got )
isinstancebytes bytearrayrD TypeErrortyperH rN _write_appdatar r? r
r
r r4 x 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.abort)N)NN)rH rI rJ r Z
_SendfileModeZFALLBACKZ_sendfile_compatibler rU rX rZ r[ r^ warningswarnra rc rd re rf rg rK rh r4 rp rr r
r
r
r rL s$
rL c @ s e Zd Zd+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 krtd|d kr tj}n|dkr6td| |sDt||}|| _|rZ|sZ|| _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 s@
zSSLProtocol.__init__c C s || _ t|tj| _d S r )rY ri r ZBufferedProtocol_app_protocol_is_buffer)r r r
r
r rV s
z SSLProtocol._set_app_protocolc C sD | j d krd S | j s:|d k r.| 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 rv r r r _start_handshake)r transportr
r
r connection_made s zSSLProtocol.connection_madec C sn | j r d| _ | j| jj| n| jd k r2d| j_d | _d | _t| dd rT| 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 s | j d S r )rY
pause_writingr! r
r
r r s zSSLProtocol.pause_writingc C s | j d S r )rY resume_writingr! r
r
r r s zSSLProtocol.resume_writingc C s" | j d krd S z| j |\}}W nL ttfk
r< Y n4 tk
rn } z| |d W Y d S d }~X Y nX |D ]}| j| qt|D ]}|rz&| jrt
| j | n | j | W nP ttfk
r Y n8 tk
r
} z| |d W Y d S d }~X Y nX q|
q qd 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 s<
zSSLProtocol.data_receivedc C sT zB| j rtd| | t | js@| j
}|r@t d W 5 | j X d S )Nz%r received EOFz?returning true from eof_received() has no effect when using ssl) rb r^ rM get_debugr debugr ConnectionResetErrorr rY eof_receivedZwarning)r Z keep_openr
r
r r - s
zSSLProtocol.eof_receivedc C s4 || j kr| j | S | jd k r,| j||S |S d S r )rx rb rU rR r
r
r rQ C s
zSSLProtocol._get_extra_infoc C s. | j r
d S | jr| nd| _ | d d S )NTr& )r r rq rn r! r
r
r r] K s
zSSLProtocol._start_shutdownc C s. | j |df | jt|7 _| d S )Nr )r{ r7 r| rC _process_write_backlogro r
r
r rn T s zSSLProtocol._write_appdatac C s\ | j r$td| | 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 Y s
z SSLProtocol._start_handshakec C s* | j dkr&d| j d}| t| d S )NTz$SSL handshake is taking longer than z! seconds: aborting the connection)r r r ConnectionAbortedError)r msgr
r
r r h s
z$SSLProtocol._check_handshake_timeoutc
C s d| _ | j | jj}z|d k r&|| }W nb ttfk
rJ Y nJ tk
r } z,t |t
j rld}nd}| || W Y d S d }~X Y nX | j
r| j
| j }td| |d | jj|| | |d | jr| 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 ri r r9 r rM r r r r r rx updater r r rY r r~ r r r r )r Z
handshake_excZsslobjr rA r Zdtr
r
r _on_handshake_completeq s8
z"SSLProtocol._on_handshake_completec
C sB | j d ks| jd krd S ztt| jD ]}| jd \}}|rR| j||\}}n*|rj| j| j}d}n| j| j }d}|D ]}| j
| q|t|k r||f| jd< | j j r| j q| jd= | j
t|8 _
q(W n\ ttfk
r Y nD tk
r< } z$| jr | | n | |d W 5 d }~X Y nX d S )Nr r z Fatal error on SSL transport)rb r rangerC r{ rG r/ r r1 _finalizer4 Z_pausedre r| r r r r r )r ir? rF r- r@ rA r
r
r r s:
z"SSLProtocol._process_write_backlogFatal error on transportc C sV t |tr(| j r@tjd| |dd n| j||| j| d | jrR| j| d S )Nz%r: %sT)exc_info)messageZ exceptionr rW ) ri 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 k r| j d S r )r rb r^ r! r
r
r r s
zSSLProtocol._finalizec C s( z| jd k r| j W 5 | X d S r )r rb rr r! r
r
r rq s
zSSLProtocol._abort)FNTN)N)N)r )rH rI rJ r rV r r r r r r r rQ r] rn r r r r r r rq r
r
r
r ru s.
.
&
)+
ru )ry rs r ImportError r r r r logr r r r) r$ r0 objectr Z_FlowControlMixinZ TransportrL ZProtocolru r
r
r
r s*
yx