a
XC?hCk � @ s� d dl Z d dlZz d dlZW n ey2 dZY n0 ddlmZ ddlmZ ddlmZ ddlm Z dd� Z
d Z d
Z d Z
d ZG d
d� de�ZG dd� dejej�ZG dd� dej�ZdS )� N� )� 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/python39/lib64/python3.9/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__7 s
z_SSLPipe.__init__c C s | j S �N)r �r r r r
r M s z_SSLPipe.contextc C s | j S r )r r r r r
�
ssl_objectR s z_SSLPipe.ssl_objectc C s | j S 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 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_handshakei 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
�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 �}|� |� |s�q�q�nJ| j t
kr�| j�� d | _t| _ | jr�| �� n| j tkr�|� | j�
� � W n| tjtjf�yn } zZt|dd �}|tjtjtjfv�rN| j tk�rL| j �rL| � |� � |tjk| _W Y d }~n
d }~0 0 | 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|�}d| _z(|t|�k rn|| j�||d � �7 }W nj tjy� } zPt |dd �}|j
dkr�tj }|_ |tj tj
tjfvr�� |tj k| _W Y d }~n
d }~0 0 | jjr�|�| j�� � |t|�k�s| jrB�qqB||fS )NFr2 ZPROTOCOL_IS_SHUTDOWN)r r �len�
memoryviewr r r3 r r7 r9 �reasonr: r2 r; r< r r= r6 r4 )r r> �offsetr, Zviewr@ rA r r r
� feed_appdata� s4
�"z_SSLPipe.feed_appdata)N)N)N)F)r )�__name__�
__module__� __qualname__r5 r �propertyr r! r" r$ r. r0 r1 r) rF r r r r
r # s
Kr c @ s� e Zd ZejjZdd� Zd$dd�Zdd� Z dd � Z
d
d � Z d d
� Z e
jfdd�Zdd� Zdd� Zdd� Zd%dd�Zdd� 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 )rM �_get_extra_info�r �name�defaultr r r
�get_extra_info& s z$_SSLProtocolTransport.get_extra_infoc C s | j �|� d S r )rM �_set_app_protocol)r �protocolr r r
� set_protocol* s z"_SSLProtocolTransport.set_protocolc C s | j jS r )rM �
_app_protocolr r r r
� get_protocol- s z"_SSLProtocolTransport.get_protocolc C s | j S r )rN r r r r
�
is_closing0 s z _SSLProtocolTransport.is_closingc C s d| _ | j�� d S �NT)rN rM �_start_shutdownr r r r
�close3 s z_SSLProtocolTransport.closec C s& | j s"|d| ��t| d� | �� d S )Nzunclosed transport )�source)rN �ResourceWarningr] )r Z_warnr r r
�__del__> s z _SSLProtocolTransport.__del__c C s | j j}|d u rtd��|�� S )Nz*SSL transport has not been initialized yet)rM �
_transportr' �
is_reading)r Ztrr r r
rb C s z _SSLProtocolTransport.is_readingc C s | j j�� d S r )rM ra �
pause_readingr r r r
rc I s z#_SSLProtocolTransport.pause_readingc C s | j j�� d S r )rM ra �resume_readingr r r r
rd Q s z$_SSLProtocolTransport.resume_readingc C s | j j�||� d S r )rM ra �set_write_buffer_limits)r ZhighZlowr r r
re Y s z-_SSLProtocolTransport.set_write_buffer_limitsc C s | j j�� S r )rM ra �get_write_buffer_sizer r r r
rf n s z+_SSLProtocolTransport.get_write_buffer_sizec C s | j j�� S r )rM ra �get_write_buffer_limitsr r r r
rg r s z-_SSLProtocolTransport.get_write_buffer_limitsc C s
| j jjS r )rM ra �_protocol_pausedr r r r
rh x 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� bytearrayrC � TypeError�typerG rM �_write_appdata�r r> r r r
r3 } 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[ )rM �_abortrN r r r r
�abort� s
z_SSLProtocolTransport.abort)N)NN)rG rH rI r Z
_SendfileModeZFALLBACKZ_sendfile_compatibler rT rW rY rZ r] �warnings�warnr` rb rc rd re rf rg rJ rh r3 rp rr r r r r
rK s&
rK 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 u rtd��|d u r 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�_waiterrL rU rK �_app_transport�_sslpipe�_session_established�
_in_handshake� _in_shutdownra �_call_connection_made�_ssl_handshake_timeout) r rO � app_protocolr Zwaiterr r
Zcall_connection_madeZssl_handshake_timeoutr r r
r � sB ���
zSSLProtocol.__init__c C s || _ t|tj�| _d S r )rX ri r ZBufferedProtocol�_app_protocol_is_buffer)r r� r r r
rU � 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_result�r r@ r r r
�_wakeup_waiter� s
zSSLProtocol._wakeup_waiterc C s& || _ t| j| j| j�| _| �� d S r )ra 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 ur2d| j_d | _d | _t| dd �rT| j �
� | � |� d | _d | _ d S )NFT�_handshake_timeout_handle)
r� rL � call_soonrX �connection_lostr~ rN ra r9 r� �cancelr� r r� r r r
r� � s
zSSLProtocol.connection_lostc C s | j �� d S r )rX �
pause_writingr r r r
r� s zSSLProtocol.pause_writingc C s | j �� d S r )rX �resume_writingr r r r
r� s zSSLProtocol.resume_writingc C s | j d u rd S z| j �|�\}}W nH ttfy: � Y n2 tyj } z| �|d� W Y d }~d S d }~0 0 |D ]}| j�|� qp|D ]�}|�rz&| jr�t �
| j |� n | j � |� W nL ttfy� � Y n6 t�y } z | �|d� W Y d }~ d S d }~0 0 q�| �
� �qq�d S )NzSSL error in data receivedz/application protocol failed to receive SSL data)r r) �
SystemExit�KeyboardInterrupt�
BaseException� _fatal_errorra r3 r� r Z _feed_data_to_buffered_protorX �
data_receivedr\ )r r> r, r- �er? Zexr r r
r� s8
�� zSSLProtocol.data_receivedc C s^ zL| j �� rt�d| � | �t� | js@| j�� }|r@t� d� W | j
� � n | j
� � 0 d S )Nz%r received EOFz?returning true from eof_received() has no effect when using ssl) rL � get_debugr �debugr� �ConnectionResetErrorr� rX � eof_receivedZwarningra r] )r Z keep_openr r r
r� 2 s
zSSLProtocol.eof_receivedc C s4 || j v r| j | S | jd ur,| j�||�S |S d S r )rx ra rT rQ r r r
rP H 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\ P s
zSSLProtocol._start_shutdownc C s. | j �|df� | jt|�7 _| �� d S )Nr )r{ r6 r| rB �_process_write_backlogro r r r
rn Y 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 )rL r� r r� �time�_handshake_start_timer� r{ r6 Z
call_laterr� �_check_handshake_timeoutr� r� r r r r
r� ^ s
��z SSLProtocol._start_handshakec C s* | j du r&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� m s
��z$SSLProtocol._check_handshake_timeoutc
C s d| _ | j�� | jj}z|d ur&|�|�� }W n^ ttfyH � Y nH ty� } z0t |t
j �rhd}nd}| � ||� W Y d }~d S d }~0 0 | 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� r r! Z getpeercertr� r� r� ri r r8 r� rL r� r� r� r r� rx �updater� r� r� rX r� r~ r� r� r� r� )r Z
handshake_excZsslobjr� r@ r� Zdtr r r
�_on_handshake_completev s8
�z"SSLProtocol._on_handshake_completec
C sB | j d u s| jd u rd S z�tt| j��D ]�}| 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\ ttfy� � Y nF t�y< } z,| j�r | �|� n | �|d� W Y d }~n
d }~0 0 d S )Nr r z Fatal error on SSL transport)ra r �rangerB r{ rF r. r� r0 � _finalizer3 Z_pausedrd r| r� r� r� r� r� )r �ir> rE r, r? r@ r r r
r� � s: �
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� rV ) ri �OSErrorrL r� r r� Zcall_exception_handlerra Z _force_close)r r@ r� r r r
r� � s
�zSSLProtocol._fatal_errorc C s d | _ | jd ur| j�� d S r )r ra r] r r r r
r� � s
zSSLProtocol._finalizec C s0 z | j d ur| j �� W | �� n
| �� 0 d S r )ra rr r� r r r r
rq � s
zSSLProtocol._abort)FNTN)N)N)r� )rG rH rI r rU r� r� r� r� r� r� r� rP r\ rn r� r� r� r� r� r� rq r r r r
ru � s, �
.
&
)+
ru )ry rs r � ImportError� r r r �logr r r r( r# r/ �objectr Z_FlowControlMixinZ TransportrK �Protocolru r r r r
� s(
y�~