B
C[%p @ s d dl Z d dlZy 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
5/opt/alt/python37/lib64/python3.7/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 )N)r )r r
r
r
ssl_objectS s z_SSLPipe.ssl_objectc C s | j S )N)r )r r
r
r need_ssldata[ s z_SSLPipe.need_ssldatac C s
| j tkS )N)r _WRAPPED)r 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_ssldata)r 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* 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 }y| j tkrx| j t| _ | j rl| d |rx||fS | j tkrxn| j
| j }| | |sP qW nJ| j t
kr| j d | _t| _ | jr| n| j tkr| | j
W nz tjtjfk
rn } zRt|dd }|tjtjtjfkrR| j tkrP| j rP| | |tjk| _W d d }~X Y nX | jjr| | j
||fS )NFerrno)r r r r writer& r r+ r! r readmax_sizeappendr, Zunwrapr r SSLErrorCertificateErrorgetattrSSL_ERROR_WANT_READSSL_ERROR_WANT_WRITESSL_ERROR_SYSCALLr pending)r datar$ r* r) chunkexc exc_errnor
r
r r' sX
z_SSLPipe.feed_ssldatar c
C s | j tkr6|t|k r&||d g}ng }|t|fS g }t|}xd| _y(|t|k rp|| j||d 7 }W nh tjk
r } zHt |dd }|j
dkrtj }|_ |tj tj
tjfkr |tj k| _W d d }~X Y nX | jjr|| j |t|ks| jrDP qDW ||fS )NFr/ ZPROTOCOL_IS_SHUTDOWN)r r len
memoryviewr r r0 r r4 r6 reasonr7 r/ r8 r9 r r: r3 r1 )r r; offsetr) Zviewr= r> r
r
r feed_appdata s4
z_SSLPipe.feed_appdata)N)N)N)F)r )__name__
__module__ __qualname__r2 r propertyr r r r" r+ r- r. r' rC r
r
r
r r $ s
Kr c @ s e Zd ZejjZdd Zd"ddZdd Z dd Z
d
d Z d d
Z dd Z
dd Zdd Zdd Zd#ddZdd Zedd Zd d Zd d Zd d! ZdS )$_SSLProtocolTransportc C s || _ || _d| _d S )NF)_loop
_ssl_protocol_closed)r loopZ ssl_protocolr
r
r r ! s z _SSLProtocolTransport.__init__Nc C s | j ||S )N)rJ _get_extra_info)r namedefaultr
r
r get_extra_info' s z$_SSLProtocolTransport.get_extra_infoc C s | j | d S )N)rJ _set_app_protocol)r protocolr
r
r set_protocol+ s z"_SSLProtocolTransport.set_protocolc C s | j jS )N)rJ
_app_protocol)r r
r
r get_protocol. s z"_SSLProtocolTransport.get_protocolc C s | j S )N)rK )r r
r
r
is_closing1 s z _SSLProtocolTransport.is_closingc C s d| _ | j d S )NT)rK rJ _start_shutdown)r r
r
r close4 s z_SSLProtocolTransport.closec C s( | j s$tjd| t| d | d S )Nzunclosed transport )source)rK warningswarnResourceWarningrX )r r
r
r __del__? s z _SSLProtocolTransport.__del__c C s | j j}|d krtd| S )Nz*SSL transport has not been initialized yet)rJ
_transportr%
is_reading)r trr
r
r r_ E s z _SSLProtocolTransport.is_readingc C s | j j d S )N)rJ r^
pause_reading)r r
r
r ra K s z#_SSLProtocolTransport.pause_readingc C s | j j d S )N)rJ r^ resume_reading)r r
r
r rb S s z$_SSLProtocolTransport.resume_readingc C s | j j|| d S )N)rJ r^ set_write_buffer_limits)r ZhighZlowr
r
r rc [ s z-_SSLProtocolTransport.set_write_buffer_limitsc C s | j j S )N)rJ r^ get_write_buffer_size)r r
r
r rd p s z+_SSLProtocolTransport.get_write_buffer_sizec C s
| j jjS )N)rJ r^ _protocol_paused)r r
r
r re t 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 bytearrayr@ TypeErrortyperD rJ _write_appdata)r r; r
r
r r0 y s
z_SSLProtocolTransport.writec C s dS )NFr
)r r
r
r
can_write_eof s z#_SSLProtocolTransport.can_write_eofc C s | j d| _d S )NT)rJ _abortrK )r r
r
r abort s
z_SSLProtocolTransport.abort)N)NN)rD rE rF r Z
_SendfileModeZFALLBACKZ_sendfile_compatibler rP rS rU rV rX r] r_ ra rb rc rd rG re r0 rl rn r
r
r
r rH s"
rH 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_waiterrI rQ rH _app_transport_sslpipe_session_established
_in_handshake _in_shutdownr^ _call_connection_made_ssl_handshake_timeout) r rL app_protocolr Zwaiterr
r Zcall_connection_madeZssl_handshake_timeoutr
r
r r s:
zSSLProtocol.__init__c C s || _ t|tj| _d S )N)rT rf r ZBufferedProtocol_app_protocol_is_buffer)r r r
r
r rQ s z SSLProtocol._set_app_protocolc C sD | j d krd S | j s:|d k r.| j | n | j d d | _ d S )N)rw Z cancelledZ
set_exceptionZ
set_result)r r= r
r
r _wakeup_waiter s
zSSLProtocol._wakeup_waiterc C s& || _ t| j| j| j| _| d S )N)r^ r rp r r ry _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)
rz rI call_soonrT connection_lostrx rK r^ r6 r cancelr ry )r r= r
r
r r s
zSSLProtocol.connection_lostc C s | j d S )N)rT
pause_writing)r r
r
r r s zSSLProtocol.pause_writingc C s | j d S )N)rT resume_writing)r r
r
r r s zSSLProtocol.resume_writingc C s | j d krd S y| j |\}}W n. tk
rP } z| |d d S d }~X Y nX x|D ]}| j| qXW xr|D ]j}|ry&| jrt| j | n | j
| W q tk
r } z| |d d S d }~X Y qX qr| P qrW d S )NzSSL error in data receivedz/application protocol failed to receive SSL data) ry r' Exception _fatal_errorr^ r0 r r Z _feed_data_to_buffered_protorT
data_receivedrW )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 d | j
X d S )Nz%r received EOFz?returning true from eof_received() has no effect when using ssl) rI get_debugr debugr ConnectionResetErrorr{ rT eof_receivedZwarningr^ rX )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 )N)rr r^ rP )r rN rO r
r
r rM @ s
zSSLProtocol._get_extra_infoc C s. | j r
d S | jr| nd| _ | d d S )NTr# )r| r{ rm rk )r r
r
r rW H s
zSSLProtocol._start_shutdownc C s. | j |df | jt|7 _| d S )Nr )ru r3 rv r? _process_write_backlog)r r; r
r
r rk Q 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 )rI r r r time_handshake_start_timer{ ru r3 Z
call_laterr~ _check_handshake_timeoutr r )r r
r
r r V 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 e s
z$SSLProtocol._check_handshake_timeoutc
C s d| _ | j | jj}y|d k r&|| }W nD tk
rt } z&t|tj rTd}nd}|
|| 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 ry r Z getpeercertr rf r r5 r rI r r r r r rr updater r r} rT r rx r rz r r )r Z
handshake_excZsslobjr r= r Zdtr
r
r _on_handshake_completen s2
z"SSLProtocol._on_handshake_completec
C s0 | j d ks| jd krd S yxtt| jD ]}| jd \}}|rT| j||\}}n*|rl| j| j}d}n| j| j }d}x|D ]}| j
| qW |t|k r||f| jd< | j j r| j P | jd= | j
t|8 _
q*W W nD tk
r* } z$| jr| | n | |d W d d }~X Y nX d S )Nr r z Fatal error on SSL transport)r^ ry ranger? ru rC r+ r r- _finalizer0 Z_pausedrb rv r r{ r )r ir; rB r) r< r= r
r
r r s4
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 rR ) rf OSErrorrI r r r Zcall_exception_handlerr^ Z _force_close)r r= r r
r
r r s
zSSLProtocol._fatal_errorc C s d | _ | jd k r| j d S )N)ry r^ rX )r r
r
r r s
zSSLProtocol._finalizec C s( z| j d k r| j W d | X d S )N)r^ rn r )r r
r
r rm s
zSSLProtocol._abort)FNTN)N)N)r )rD rE rF r rQ r r r r r r r rM rW rk r r r r r r rm r
r
r
r ro s*
+
"
')
ro )rs rZ r ImportError r r r r logr r r r& r! r, objectr Z_FlowControlMixinZ TransportrH ZProtocolro r
r
r
r s(
yx