| Current File : //opt/alt/python310/lib64/python3.10/asyncio/__pycache__/sslproto.cpython-310.opt-2.pyc |
o
�=?hCk � @ 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� de�ZG dd� dejej�ZG dd� dej�ZdS )� N� )� constants)� protocols)�
transports)�loggerc C s" | rt d��t�� }|sd|_|S )Nz(Server side SSL needs a valid SSLContextF)�
ValueError�sslZcreate_default_contextZcheck_hostname)� server_side�server_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 dZddd�Zedd� �Zedd� �Zed d
� �Zed d � �Z dd
d�Z
ddd�Z dd� Z ddd�Z
ddd�ZdS ) �_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)�self�contextr 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_ssldata�r �callback�ssldata�appdatar r r
� do_handshakei s
�z_SSLPipe.do_handshakec C sD | j tkr
td��| j tkrtd��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}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 )NFT�errno)r r r r �writer) r r/ r$ r �read�max_size�appendr0 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* � 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$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 )rN �_get_extra_info�r �name�defaultr 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 |tttf�stdt|�j� ���|sd S | j�|� d S )Nz+data: expecting a bytes-like instance, got ) �
isinstance�bytes� bytearrayrD � TypeError�typerH rN �_write_appdata�r 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^ �warnings�warnra rc re rg rh rj rk rK rl r4 rt rv r r r r
rL s(
rL c @ s� e Zd Z d+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 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 � _sslcontext�dict�_extra� collections�deque�_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_result�r 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* �
SystemExit�KeyboardInterrupt�
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
t�d| � | �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 �� rt�d| � | j �� | _nd | _d| _| j�d� | 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 }t�d| |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 @�@)�peercert�cipher� 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| j��D ]\}| 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_backlog�Fatal error on transportc C sZ t |t�r| 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
�<module> s, � y�~