3 ��2a�e� @s�ddlZddlZyddlZWnek r4dZYnXddlmZddlmZddlmZddlmZddl m Z dd �Zd d�ZdZ d ZdZdZGdd�de�ZGdd�dejej�ZGdd�dej�ZdS)�N�)�base_events)�compat)� protocols)� transports)�loggercCsj|rtd��ttd�r*tj�}|sfd|_n|j dkr�tj |_|jtj tjtj fkr��|jtj k|_WYdd}~XnX|jjr�|j|jj��|t|�ks�|jrDPqDW||fS)NFZPROTOCOL_IS_SHUTDOWN)rr�len� memoryviewr!r r8rr<�reasonr?r7r@rArrBr;r9)r$rC�offsetr1ZviewrErrr�feed_appdata�s2 z_SSLPipe.feed_appdatai)N)N)N)F)r)�__name__� __module__�__qualname__r:r&�propertyr%r'r(r*r3r5r6r/rJrrrrr0s Jrc@s�eZdZdd�Zddd�Zdd�Zdd �Zd d�Zdd �Ze j rHdd�Zdd�Zdd�Z ddd�Zdd�Zdd�Zdd�Zdd�ZdS) �_SSLProtocolTransportcCs||_||_d|_dS)NF)�_loop� _ssl_protocol�_closed)r$�loopZssl_protocolrrrr&)sz_SSLProtocolTransport.__init__NcCs|jj||�S)N)rQ�_get_extra_info)r$�name�defaultrrr�get_extra_info/sz$_SSLProtocolTransport.get_extra_infocCs||j_dS)N)rQ� _app_protocol)r$�protocolrrr�set_protocol3sz"_SSLProtocolTransport.set_protocolcCs|jjS)N)rQrX)r$rrr�get_protocol6sz"_SSLProtocolTransport.get_protocolcCs|jS)N)rR)r$rrr� is_closing9sz _SSLProtocolTransport.is_closingcCsd|_|jj�dS)NT)rRrQ�_start_shutdown)r$rrr�close<sz_SSLProtocolTransport.closecCs&|js"tjd|t|d�|j�dS)Nzunclosed transport %r)�source)rR�warnings�warn�ResourceWarningr^)r$rrr�__del__Ksz_SSLProtocolTransport.__del__cCs|jjj�dS)N)rQ� _transport� pause_reading)r$rrrreQsz#_SSLProtocolTransport.pause_readingcCs|jjj�dS)N)rQrd�resume_reading)r$rrrrfYsz$_SSLProtocolTransport.resume_readingcCs|jjj||�dS)N)rQrd�set_write_buffer_limits)r$ZhighZlowrrrrgasz-_SSLProtocolTransport.set_write_buffer_limitscCs|jjj�S)N)rQrd�get_write_buffer_size)r$rrrrhvsz+_SSLProtocolTransport.get_write_buffer_sizecCs<t|tttf�s$tdjt|�j��|s,dS|jj |�dS)Nz/data: expecting a bytes-like instance, got {!r}) � isinstance�bytes� bytearrayrG� TypeError�format�typerKrQ�_write_appdata)r$rCrrrr8zsz_SSLProtocolTransport.writecCsdS)NFr)r$rrr� can_write_eof�sz#_SSLProtocolTransport.can_write_eofcCs|jj�dS)N)rQ�_abort)r$rrr�abort�sz_SSLProtocolTransport.abort)N)NN)rKrLrMr&rWrZr[r\r^rZPY34rcrerfrgrhr8rprrrrrrrO&s rOc@s�eZdZd'dd�Zd(dd�Zdd �Zd d�Zdd �Zdd�Zdd�Z dd�Z d)dd�Zdd�Zdd�Z dd�Zdd�Zdd�Zd*d!d"�Zd#d$�Zd%d&�ZdS)+�SSLProtocolFNTcCs�tdkrtd��|st||�}||_|r6|r6||_nd|_||_t|d�|_tj �|_ d|_||_||_ ||_t|j |�|_d|_d|_d|_d|_d|_||_dS)Nzstdlib ssl module not available)rrF)rr-rrr�_sslcontext�dict�_extra�collections�deque�_write_backlog�_write_buffer_size�_waiterrPrXrO�_app_transport�_sslpipe�_session_established� _in_handshake�_in_shutdownrd�_call_connection_made)r$rSZapp_protocolrZwaiterrrZcall_connection_maderrrr&�s, zSSLProtocol.__init__cCsD|jdkrdS|jj�s:|dk r.|jj|�n|jjd�d|_dS)N)r{Z cancelledZ set_exceptionZ set_result)r$rErrr�_wakeup_waiter�s zSSLProtocol._wakeup_waitercCs&||_t|j|j|j�|_|j�dS)N)rdrrtrrr}�_start_handshake)r$� transportrrr�connection_made�s zSSLProtocol.connection_madecCs8|jrd|_|jj|jj|�d|_d|_|j|�dS)NF)r~rP� call_soonrX�connection_lostrdr|r�)r$rErrrr��szSSLProtocol.connection_lostcCs|jj�dS)N)rX� pause_writing)r$rrrr��szSSLProtocol.pause_writingcCs|jj�dS)N)rX�resume_writing)r$rrrr��szSSLProtocol.resume_writingcCs�|jdkrdSy|jj|�\}}WnHtjk rj}z*|jj�rTtjd||j|j �|j �dSd}~XnXx|D]}|jj|�qrWx(|D] }|r�|j j|�q�|j�Pq�WdS)Nz%r: SSL error %s (reason %s))r}r/rr<rP� get_debugr�warningr7rHrqrdr8rX� data_receivedr])r$rCr1r2�erDrrrr��s" zSSLProtocol.data_receivedc CsTzB|jj�rtjd|�|jt�|js@|jj�}|r@tj d�Wd|j j�XdS)Nz%r received EOFz?returning true from eof_received() has no effect when using ssl)rPr�r�debugr��ConnectionResetErrorrrX�eof_receivedr�rdr^)r$Z keep_openrrrr�s zSSLProtocol.eof_receivedcCs4||jkr|j|S|jdk r,|jj||�S|SdS)N)rvrdrW)r$rUrVrrrrT!s zSSLProtocol._get_extra_infocCs.|jr dS|jr|j�nd|_|jd�dS)NTr+)r�rrqro)r$rrrr])s zSSLProtocol._start_shutdowncCs.|jj|df�|jt|�7_|j�dS)Nr)ryr;rzrF�_process_write_backlog)r$rCrrrro2szSSLProtocol._write_appdatacCsH|jj�r$tjd|�|jj�|_nd|_d|_|jjd�|j �dS)Nz%r starts SSL handshakeTr+r)r+r) rPr�rr��time�_handshake_start_timerryr;r�)r$rrrr�7s zSSLProtocol._start_handshakecCsTd|_|jj}yF|dk r|�|j�}t|jd�sR|jrR|jjtj krRtj ||j�Wn~tk r�}zb|jj �r�t|tj�r�tjd|dd�ntjd|dd�|jj�t|t�r�|j|�dS�WYdd}~XnX|jj ��r|jj�|j}tjd||d�|jj||j�|j�|d �|j�r4|jj|j �|j�d|_!|jj"|j#�dS) NFrz5%r: SSL handshake failed on verifying the certificateT)�exc_infoz%r: SSL handshake failedz%r: SSL handshake took %.1f msg@�@)�peercert�cipher�compressionr')$rr}r'Zgetpeercertr rtrrrZ CERT_NONEZmatch_hostname� BaseExceptionrPr�rir=rr�rdr^� Exceptionr�r�r�r�rv�updater�r�r�rXr�r|r~r�r�)r$Z handshake_excZsslobjr�rEZdtrrr�_on_handshake_completeCsD z"SSLProtocol._on_handshake_completecCs>|jdks|jdkrdSy�x�tt|j��D]�}|jd\}}|rT|jj||�\}}n*|rl|jj|j�}d}n|jj|j �}d}x|D]}|jj |�q�W|t|�kr�||f|jd<|jjr�|jj�P|jd=|j t|�8_ q*WWnRtk �r8}z4|j�r|j|�n|j|d�t|t��s(�WYdd}~XnXdS)NrrzFatal error on SSL transport)rdr}�rangerFryrJr3r�r5� _finalizer8Z_pausedrfrzr�r�_fatal_errorrir�)r$�irCrIr1rDrErrrr�ws8 z"SSLProtocol._process_write_backlog�Fatal error on transportcCsXt|tj�r*|jj�rBtjd||dd�n|jj|||j|d��|jrT|jj |�dS)Nz%r: %sT)r�)�messageZ exceptionr�rY) rirZ_FATAL_ERROR_IGNORErPr�rr�Zcall_exception_handlerrdZ_force_close)r$rEr�rrrr��s zSSLProtocol._fatal_errorcCsd|_|jdk r|jj�dS)N)r}rdr^)r$rrrr��s zSSLProtocol._finalizec Cs(z|jdk r|jj�Wd|j�XdS)N)rdrrr�)r$rrrrq�s zSSLProtocol._abort)FNT)N)N)r�)rKrLrMr&r�r�r�r�r�r�r�rTr]ror�r�r�r�r�rqrrrrrs�s$ " 4, rs)rwr`r�ImportError�rrrr�logrrrrr.r)r4�objectrZ_FlowControlMixinZ TransportrOZProtocolrsrrrr�s* wn