File: //usr/local/lib/python3.10/asyncio/__pycache__/sslproto.cpython-310.opt-2.pyc
o
�iCk � @ 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 �-/usr/local/lib/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 �unwrapr 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 r9 r; �reasonr< r3 r= r> r r? r7 r5 )r r@ �offsetr- �viewrB rC 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* rI 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 )rP �_get_extra_info�r �name�defaultr r r
�get_extra_info& s z$_SSLProtocolTransport.get_extra_infoc C s | j �|� d S r )rP �_set_app_protocol)r �protocolr r r
� set_protocol* s z"_SSLProtocolTransport.set_protocolc C s | j jS r )rP �
_app_protocolr! r r r
� get_protocol- s z"_SSLProtocolTransport.get_protocolc C s | j S r )rQ r! r r r
�
is_closing0 s z _SSLProtocolTransport.is_closingc C s d| _ | j�� d S �NT)rQ rP �_start_shutdownr! r r r
�close3 s z_SSLProtocolTransport.closec C s* | j s|d| ��t| d� | �� d S d S )Nzunclosed transport )�source)rQ �ResourceWarningr` )r �_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)rP �
_transportr( �
is_reading)r �trr r r
rf C s z _SSLProtocolTransport.is_readingc C � | j j�� d S r )rP re �
pause_readingr! r r r
ri I � z#_SSLProtocolTransport.pause_readingc C rh r )rP re �resume_readingr! r r r
rk Q rj z$_SSLProtocolTransport.resume_readingc C s | j j�||� d S r )rP re �set_write_buffer_limits)r �high�lowr r r
rl Y s z-_SSLProtocolTransport.set_write_buffer_limitsc C � | j j�� S r )rP re �get_write_buffer_sizer! r r r
rp n s z+_SSLProtocolTransport.get_write_buffer_sizec C ro r )rP re �get_write_buffer_limitsr! r r r
rq r s z-_SSLProtocolTransport.get_write_buffer_limitsc C s
| j jjS r )rP re �_protocol_pausedr! r r r
rr 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� bytearrayrE � TypeError�typerJ rP �_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^ )rP �_abortrQ r! r r r
�abort� s
z_SSLProtocolTransport.abortr )NN)rJ rK rL r Z
_SendfileModeZFALLBACKZ_sendfile_compatibler rW rZ r\ r] r` �warnings�warnrd rf ri rk rl rp rq rM rr r4 rz r| r r r r
rN s(
rN 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�_waiterrO rX rN �_app_transport�_sslpipe�_session_established�
_in_handshake� _in_shutdownre �_call_connection_made�_ssl_handshake_timeout) r rR � app_protocolr �waiterr r
Zcall_connection_madeZssl_handshake_timeoutr r r
r � sB ���
zSSLProtocol.__init__c C s || _ t|tj�| _d S r )r[ rs r ZBufferedProtocol�_app_protocol_is_buffer)r r� r r r
rX � 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� � cancelled�
set_exception�
set_result�r rB r r r
�_wakeup_waiter� s
zSSLProtocol._wakeup_waiterc C s( || _ t| j| j| j�| _| �� d S r )re r r� 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� rO � call_soonr[ �connection_lostr� rQ re r; r� �cancelr� r� r� r r r
r� � s
zSSLProtocol.connection_lostc C � | j �� d S r )r[ �
pause_writingr! r r r
r� � zSSLProtocol.pause_writingc C r� r )r[ �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_errorre r4 r� r Z _feed_data_to_buffered_protor[ �
data_receivedr_ )r r@ r- r. �erA �exr 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) rO � get_debugr �debugr� �ConnectionResetErrorr� r[ � eof_received�warningre 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� re rW rT r r r
rS H s
zSSLProtocol._get_extra_infoc C s0 | j rd S | jr| �� d S d| _ | �d� d S )NTr& )r� r� r{ rx r! r r r
r_ P s zSSLProtocol._start_shutdownc C s. | j �|df� | jt|�7 _| �� d S )Nr )r� r7 r� rD �_process_write_backlogry r r r
rx 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 )rO 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� rs r r: r� rO r� r� r� r r� r� �updater� r� r� r[ r� r� r� r� r� r� )r Z
handshake_excZsslobjr� rB 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)re r� �rangerD r� rI r/ r� r1 � _finalizer4 Z_pausedrk r� r� r� r� r� r� )r �ir@ rG r- rA rB 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)�message� exceptionr� rY ) rs �OSErrorrO r� r r� Zcall_exception_handlerre Z _force_close)r rB r� r r r
r� � s
���zSSLProtocol._fatal_errorc C s" d | _ | jd ur| j�� d S d S r )r� re r` r! r r r
r� � s
�zSSLProtocol._finalizec C s< z| j d ur| j �� W | �� d S W | �� d S | �� w r )re r| r� r! r r r
r{ � s
�zSSLProtocol._abort)FNTNr )r� )rJ rK rL r rX r� r� r� r� r� r� r� rS r_ rx r� r� r� r� r� r� r{ r r r r
r � s0
�.
&
)
+ r )r� r} r � ImportError� r r r �logr r r r) r$ r0 �objectr Z_FlowControlMixinZ TransportrN ZProtocolr r r r r
�<module> s, � y�~