ikan Uploader :
Directory : /opt/alt/python37/lib64/python3.7/asyncio/__pycache__/
Upload File : |
| Current File : //opt/alt/python37/lib64/python3.7/asyncio/__pycache__/sslproto.cpython-37.opt-2.pyc |
B
YI~d/i � @ 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� de�ZG dd� dejej�ZG dd� dej�ZdS )� N� )� base_events)� constants)� protocols)�
transports)�loggerc C s" | r t d��t�� }|s d|_|S )Nz(Server side SSL needs a valid SSLContextF)�
ValueError�sslZcreate_default_contextZcheck_hostname)� server_side�server_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 ZdZddd�Zedd� �Zedd� �Zed d
� �Zed d � �Z dd
d�Z
ddd�Z dd� Z d dd�Z
d dd�ZdS ) �_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)�self�contextr
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 �callback�ssldata�appdatar
r
r � do_handshakej s
z_SSLPipe.do_handshakec C sB | j tkrtd��| j tkr$td��t| _ || _| �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 tkr�xn| j�
| j �}|� |� |s�P q�W nJ| j t
kr�| j�� d | _t| _ | jr�| �� n| j tkr�|� | j�
� � W nz tjtjfk
�rn } zRt|dd �}|tjtjtjfk�rR| j tk�rP| j �rP| � |� � |tjk| _W d d }~X Y nX | jj�r�|� | j�
� � ||fS )NF�errno)r r r r �writer& r r+ r! r �read�max_size�appendr, Zunwrapr r �SSLError�CertificateError�getattr�SSL_ERROR_WANT_READ�SSL_ERROR_WANT_WRITE�SSL_ERROR_SYSCALLr �pending)r �datar$ r* r) �chunk�exc� 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|�}x�d| _y(|t|�k rp|| j�||d � �7 }W nh tjk
r� } zHt |dd �}|j
dkr�tj }|_ |tj tj
tjfkr�� |tj k| _W d d }~X Y nX | jjr�|�| j�� � |t|�k�s| 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"dd�Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zd#dd�Zdd� 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 �name�defaultr
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 �warnings�warn�ResourceWarningrX )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 |tttf�s$tdt|�j� ���|s,d S | j�|� d S )Nz+data: expecting a bytes-like instance, got ) �
isinstance�bytes� bytearrayr@ � TypeError�typerD 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+dd�Zdd� Zd,dd �Zd
d � Zd d
� Zdd� Zdd� Z dd� Z
dd� Z d-dd�Z 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 � _sslcontext�dict�_extra� collections�deque�_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}|r�y&| jr�t�| j |� n | j �
|� W q� tk
r� } z| �|d� d S d }~X Y q�X 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 �� rt�d| � | �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$t�d| � | j �� | _nd | _d| _| j�d� | 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 }t�d| |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 @�@)�peercert�cipher� 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 y�x�tt| j��D ]�}| jd \}}|rT| j�||�\}}n*|rl| j�| j�}d}n| j�| j �}d}x|D ]}| j �
|� q�W |t|�k r�||f| jd< | j j r�| j � � P | jd= | j
t|�8 _
q*W W nD tk
�r* } z$| j�r| �|� 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_backlog�Fatal error on transportc C sV t |t�r(| 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 �<module> s(
yx
........