관리-도구
편집 파일: sslproto.cpython-38.opt-2.pyc
U e5dJj � @ s� d dl Z d dlZz 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 �(/usr/lib64/python3.8/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 r )r r! r r r � ssl_objectS s z_SSLPipe.ssl_objectc C s | j S r )r r! r r r � need_ssldata[ s z_SSLPipe.need_ssldatac C s | j tkS r )r �_WRAPPEDr! 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 �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 nz tjtjfk �rl } zRt|dd �}|tjtjtjfk�rP| j tk�rN| j �rN| � |� � |tjk| _W 5 d }~X Y nX | jj�r�|� | j� � � ||fS )NF�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* � sZ � 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 nh tjk r� } zHt |dd �}|j dkr�tj }|_ |tj tj tjfkr�� |tj k| _W 5 d }~X Y nX | jjr�|�| j�� � |t|�k�s | jrB�q qB||fS )NFr3 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� s4 � z_SSLPipe.feed_appdata)N)N)N)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� Zedd� �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_closing1 s z _SSLProtocolTransport.is_closingc C s d| _ | j�� d S �NT)rO rN �_start_shutdownr! r r r �close4 s z_SSLProtocolTransport.closec C s& | j s"|d| ��t| d� | �� d S )Nzunclosed transport )�source)rO �ResourceWarningr^ )r Z_warnr r r �__del__? s z _SSLProtocolTransport.__del__c C s | j j}|d krtd��|�� S )Nz*SSL transport has not been initialized yet)rN � _transportr( � is_reading)r Ztrr r r rc D s z _SSLProtocolTransport.is_readingc C s | j j�� d S r )rN rb � pause_readingr! r r r rd J s z#_SSLProtocolTransport.pause_readingc C s | j j�� d S r )rN rb �resume_readingr! r r r re R s z$_SSLProtocolTransport.resume_readingc C s | j j�||� d S r )rN rb �set_write_buffer_limits)r ZhighZlowr r r rf Z s z-_SSLProtocolTransport.set_write_buffer_limitsc C s | j j�� S r )rN rb �get_write_buffer_sizer! r r r rg o s z+_SSLProtocolTransport.get_write_buffer_sizec C s | j jjS r )rN rb �_protocol_pausedr! r r r rh s 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� bytearrayrD � TypeError�typerH rN �_write_appdata�r r? r r r r4 x 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.abort)N)NN)rH rI rJ r Z _SendfileModeZFALLBACKZ_sendfile_compatibler rU rX rZ r[ r^ �warnings�warnra rc rd re rf rg rK rh r4 rp rr r r r r rL s$ rL 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�_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 � s@ � � zSSLProtocol.__init__c C s || _ t|tj�| _d S r )rY ri r ZBufferedProtocol�_app_protocol_is_buffer)r r� r r r rV � s �z SSLProtocol._set_app_protocolc C sD | j d krd S | j �� s:|d k r.| 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 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 k r2d| j_d | _d | _t| dd �rT| 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 s | j �� d S r )rY � pause_writingr! r r r r� � s zSSLProtocol.pause_writingc C s | j �� d S r )rY �resume_writingr! r r r r� s zSSLProtocol.resume_writingc C s" | j d krd S z| j �|�\}}W nL ttfk r< � Y n4 tk rn } z| �|d� W Y �d S d }~X Y nX |D ]}| j�|� qt|D ]�}|�rz&| jr�t � | j |� n | j � |� W nP ttfk r� � Y n8 tk �r } z| �|d� W Y � d S d }~X Y nX q�| � � �q q�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� s< � � zSSLProtocol.data_receivedc C sT zB| j�� rt�d| � | �t� | js@| j � � }|r@t� d� W 5 | j �� X d S )Nz%r received EOFz?returning true from eof_received() has no effect when using ssl) rb r^ rM � get_debugr �debugr� �ConnectionResetErrorr� rY � eof_receivedZwarning)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 r )rx rb rU rR r r r rQ C 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] K s zSSLProtocol._start_shutdownc C s. | j �|df� | jt|�7 _| �� d S )Nr )r{ r7 r| rC �_process_write_backlogro r r r rn T 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 )rM r� r r� �time�_handshake_start_timer� r{ r7 Z call_laterr� �_check_handshake_timeoutr� r� r! r r r r� Y 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� h s �z$SSLProtocol._check_handshake_timeoutc C s d| _ | j�� | jj}z|d k r&|�|�� }W nb ttfk rJ � Y nJ tk r� } z,t |t j �rld}nd}| � ||� W Y �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� r r" Z getpeercertr� r� r� ri r r9 r� rM r� r� r� r r� rx �updater� r� r� rY r� r~ r� r� r� r� )r Z handshake_excZsslobjr� rA r� Zdtr r r �_on_handshake_completeq s8 �z"SSLProtocol._on_handshake_completec C sB | j d ks| jd krd 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\ ttfk r� � Y nD tk �r< } z$| j�r | �|� n | �|d� W 5 d }~X Y nX d S )Nr r z Fatal error on SSL transport)rb r �rangerC r{ rG r/ r� r1 � _finalizer4 Z_pausedre r| r� r� r� r� r� )r �ir? rF r- r@ rA 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� rW ) ri �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 k r| j�� d S r )r rb r^ r! r r r r� � s zSSLProtocol._finalizec C s( z| jd k r| j�� W 5 | � � X d S r )r� rb rr r! r r r rq � s zSSLProtocol._abort)FNTN)N)N)r� )rH rI rJ r rV r� r� r� r� r� r� r� rQ r] rn r� r� r� r� r� r� rq r r r r ru � s. � . & )+ ru )ry rs r � ImportError� r r r r �logr r r r) r$ r0 �objectr Z_FlowControlMixinZ TransportrL ZProtocolru r r r r �<module> s* y�x