관리-도구
편집 파일: sslproto.cpython-39.opt-2.pyc
a R�fCk � @ 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 �<module> s( y�~