Current File : //opt/alt/python36/lib64/python3.6/asyncio/__pycache__/sslproto.cpython-36.opt-2.pyc
3
��2a�e �
@ s� d dl Z d dlZy d dlZW n ek
r4 dZY nX ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
dd � Z d
d � Z d Z
d
ZdZdZG dd� de�ZG dd� dejej�ZG dd� dej�ZdS )� N� )� base_events)�compat)� protocols)�
transports)�loggerc C sj | r t d��ttd�r*tj� }|sfd|_n<tjtj�}| jtjO _| jtj O _|j
� tj |_ |S )Nz(Server side SSL needs a valid SSLContext�create_default_contextF)
�
ValueError�hasattr�sslr �check_hostnameZ
SSLContextZPROTOCOL_SSLv23�optionsZ OP_NO_SSLv2Z OP_NO_SSLv3Zset_default_verify_pathsZ
CERT_REQUIRED� verify_mode)� server_side�server_hostname�
sslcontext� r �-/opt/alt/python36/lib64/python3.6/sslproto.py�_create_transport_context s
r c C s
t td�S )N� MemoryBIO)r
r r r r r �_is_sslproto_available% s r Z UNWRAPPEDZ DO_HANDSHAKEZWRAPPEDZSHUTDOWNc @ sz e Zd ZdZddd�Zedd� �Zedd � �Zed
d � �Zed d
� �Z d dd�Z
d dd�Z dd� Z d dd�Z
d dd�ZdS ) �_SSLPipe� i Nc C sH || _ || _|| _t| _tj� | _tj� | _d | _ d| _
d | _ d | _ d S )NF)
�_context� _server_side�_server_hostname�
_UNWRAPPED�_stater r � _incoming� _outgoing�_sslobj�
_need_ssldata�
_handshake_cb� _shutdown_cb)�self�contextr r r r r �__init__D s
z_SSLPipe.__init__c C s | j S )N)r )r$ r r r r% Z s z_SSLPipe.contextc C s | j S )N)r )r$ r r r �
ssl_object_ s z_SSLPipe.ssl_objectc C s | j S )N)r! )r$ r r r � need_ssldatag s z_SSLPipe.need_ssldatac C s
| j tkS )N)r �_WRAPPED)r$ r r r �wrappedm 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_handshakev s
z_SSLPipe.do_handshakec C sB | j tkrtd��| j tkr$td��t| _ || _| jd�\}}|S )Nzno security layer presentzshutdown in progressr+ )r r r- � _SHUTDOWNr# r/ )r$ r0 r1 r2 r r r �shutdown� s
z_SSLPipe.shutdownc C s | j j� | jd�\}}d S )Nr+ )r Z write_eofr/ )r$ r1 r2 r r r �feed_eof� s
z_SSLPipe.feed_eofFc C s� | j tkr"|r|g}ng }g |fS d| _|r8| jj|� g }g }y�| j tkrx| jj� t| _ | j rl| j d � |rx||fS | j tkr�xn| jj
| j �}|j |� |s�P q�W nJ| j t
kr�| jj� d | _t| _ | jr�| j� n| j tkr�|j | jj
� � W nx tjtjfk
�rl } zRt|dd �tjtjtjfk�rN| j tk�rL| j �rL| j |� � |jtjk| _W Y d d }~X nX | jj�r�|j | jj
� � ||fS )NF�errno)r r r! r �writer. r r3 r) r" �read�max_size�appendr4 Zunwrapr# r �SSLError�CertificateError�getattr�SSL_ERROR_WANT_READ�SSL_ERROR_WANT_WRITE�SSL_ERROR_SYSCALLr7 r �pending)r$ �datar, r2 r1 �chunk�excr r r r/ � sV
z_SSLPipe.feed_ssldatar c C s | j tkr6|t|�k r&||d � g}ng }|t|�fS g }t|�}x�d| _y(|t|�k rp|| jj||d � �7 }W n\ tjk
r� } z>|j dkr�tj
|_ |j tj
tj tj
fkr�� |j tj
k| _W Y d d }~X nX | jjr�|j| jj� � |t|�ks�| jrDP qDW ||fS )NFZPROTOCOL_IS_SHUTDOWN)r r �len�
memoryviewr! r r8 r r< �reasonr? r7 r@ rA r rB r; r9 )r$ rC �offsetr1 ZviewrE r r r � feed_appdata� s2
z_SSLPipe.feed_appdatai )N)N)N)F)r )�__name__�
__module__� __qualname__r: r& �propertyr% r' r( r* r3 r5 r6 r/ rJ r r r r r 0 s
Jr c @ s� e Zd Zdd� Zd dd�Zdd� Zdd � Zd
d � Zd d
� Ze j
rHdd� Z dd� Z dd� Z
d dd�Zdd� Zdd� Zdd� Zd d � ZdS ) �_SSLProtocolTransportc C s || _ || _d| _d S )NF)�_loop�
_ssl_protocol�_closed)r$ �loopZ ssl_protocolr r r r& ) s z _SSLProtocolTransport.__init__Nc C s | j j||�S )N)rQ �_get_extra_info)r$ �name�defaultr r r �get_extra_info/ s z$_SSLProtocolTransport.get_extra_infoc C s || j _d S )N)rQ �
_app_protocol)r$ �protocolr r r � set_protocol3 s z"_SSLProtocolTransport.set_protocolc C s | j jS )N)rQ rX )r$ r r r � get_protocol6 s z"_SSLProtocolTransport.get_protocolc C s | j S )N)rR )r$ r r r �
is_closing9 s z _SSLProtocolTransport.is_closingc C s d| _ | jj� d S )NT)rR rQ �_start_shutdown)r$ r r r �close< s z_SSLProtocolTransport.closec C s& | j s"tjd| t| d� | j� d S )Nzunclosed transport %r)�source)rR �warnings�warn�ResourceWarningr^ )r$ r r r �__del__K s z _SSLProtocolTransport.__del__c C s | j jj� d S )N)rQ �
_transport�
pause_reading)r$ r r r re Q s z#_SSLProtocolTransport.pause_readingc C s | j jj� d S )N)rQ rd �resume_reading)r$ r r r rf Y s z$_SSLProtocolTransport.resume_readingc C s | j jj||� d S )N)rQ rd �set_write_buffer_limits)r$ ZhighZlowr r r rg a s z-_SSLProtocolTransport.set_write_buffer_limitsc C s | j jj� S )N)rQ rd �get_write_buffer_size)r$ r r r rh v s z+_SSLProtocolTransport.get_write_buffer_sizec C s< t |tttf�s$tdjt|�j���|s,d S | jj |� d S )Nz/data: expecting a bytes-like instance, got {!r})
�
isinstance�bytes� bytearrayrG � TypeError�format�typerK rQ �_write_appdata)r$ rC r r r r8 z s z_SSLProtocolTransport.writec C s dS )NFr )r$ r r r �
can_write_eof� s z#_SSLProtocolTransport.can_write_eofc C s | j j� d S )N)rQ �_abort)r$ r r r �abort� s z_SSLProtocolTransport.abort)N)NN)rK rL rM r& rW rZ r[ r\ r^ r ZPY34rc re rf rg rh r8 rp rr r r r r rO & s
rO c @ s� e Zd Zd'dd�Zd(dd�Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dd� Z
d)dd�Z dd� Z dd� Z
dd� Zd d � Zd d � Zd*d!d"�Zd#d$� Zd%d&� ZdS )+� SSLProtocolFNTc C s� t d krtd��|s t||�}|| _|r6| r6|| _nd | _|| _t|d�| _tj � | _
d| _ || _ || _
|| _t| j
| �| _d | _d| _d| _d| _d | _|| _d S )Nz stdlib ssl module not available)r r F)r r- r r r � _sslcontext�dict�_extra� collections�deque�_write_backlog�_write_buffer_size�_waiterrP rX rO �_app_transport�_sslpipe�_session_established�
_in_handshake� _in_shutdownrd �_call_connection_made)r$ rS Z app_protocolr Zwaiterr r Zcall_connection_mader r r r& � s,
zSSLProtocol.__init__c C sD | j d krd S | j j� s:|d k r.| j j|� n | j jd � d | _ d S )N)r{ Z cancelledZ
set_exceptionZ
set_result)r$ rE r r r �_wakeup_waiter� s
zSSLProtocol._wakeup_waiterc C s& || _ t| j| j| j�| _| j� d S )N)rd r rt r r r} �_start_handshake)r$ � transportr r r �connection_made� s
zSSLProtocol.connection_madec C s8 | j r d| _ | jj| jj|� d | _d | _| j|� d S )NF)r~ rP � call_soonrX �connection_lostrd r| r� )r$ rE r r r r� � s zSSLProtocol.connection_lostc C s | j j� d S )N)rX �
pause_writing)r$ r r r r� � s zSSLProtocol.pause_writingc C s | j j� d S )N)rX �resume_writing)r$ r r r r� � s zSSLProtocol.resume_writingc C s� | j d krd S y| j j|�\}}W nH tjk
rj } z*| jj� rTtjd| |j|j � | j
� d S d }~X nX x|D ]}| j j |� qrW x(|D ] }|r�| j
j|� q�| j� P q�W d S )Nz %r: SSL error %s (reason %s))r} r/ r r< rP � get_debugr �warningr7 rH rq rd r8 rX �
data_receivedr] )r$ rC r1 r2 �erD r r r r� � s"
zSSLProtocol.data_receivedc
C sT zB| j j� rtjd| � | jt� | js@| jj� }|r@tj d� W d | j
j � X d S )Nz%r received EOFz?returning true from eof_received() has no effect when using ssl) rP r� r �debugr� �ConnectionResetErrorr rX � eof_receivedr� rd r^ )r$ Z keep_openr r r r� s
zSSLProtocol.eof_receivedc C s4 || j kr| j | S | jd k r,| jj||�S |S d S )N)rv rd rW )r$ rU rV r r r rT ! s
zSSLProtocol._get_extra_infoc C s. | j r
d S | jr| j� nd| _ | jd� d S )NTr+ )r� r rq ro )r$ r r r r] ) s
zSSLProtocol._start_shutdownc C s. | j j|df� | jt|�7 _| j� d S )Nr )ry r; rz rF �_process_write_backlog)r$ rC r r r ro 2 s zSSLProtocol._write_appdatac C sH | j j� r$tjd| � | j j� | _nd | _d| _| jjd� | j � d S )Nz%r starts SSL handshakeTr+ r )r+ r )
rP r� r r� �time�_handshake_start_timer ry r; r� )r$ r r r r� 7 s
z SSLProtocol._start_handshakec C sT d| _ | jj}yF|d k r |�|j� }t| jd�sR| jrR| jjtj krRtj
|| j� W n~ t k
r� } zb| j j
� r�t|tj�r�tjd| dd� ntjd| dd� | jj� t|t�r�| j|� d S � W Y d d }~X nX | j j
� �r| j j� | j }tjd| |d � | jj||j� |j � |d � | j �r4| j j | j � | j� d| _!| j j"| j#� d S )
NFr z5%r: SSL handshake failed on verifying the certificateT)�exc_infoz%r: SSL handshake failedz %r: SSL handshake took %.1f msg @�@)�peercert�cipher� compressionr' )$r r} r' Z getpeercertr
rt r r r Z CERT_NONEZmatch_hostname�
BaseExceptionrP r� ri r= r r� rd r^ � Exceptionr� r� r� r� rv �updater� r� r� rX r� r| r~ r� r� )r$ Z
handshake_excZsslobjr� rE Zdtr r r �_on_handshake_completeC sD
z"SSLProtocol._on_handshake_completec C s> | j d ks| jd krd S y�x�tt| j��D ]�}| jd \}}|rT| jj||�\}}n*|rl| jj| j�}d}n| jj| j �}d}x|D ]}| j j
|� q�W |t|�k r�||f| jd< | j j r�| j j � P | jd= | j
t|�8 _
q*W W nR tk
�r8 } z4| j�r| j|� n | j|d� t|t��s(� W Y d d }~X nX d S )Nr r z Fatal error on SSL transport)rd r} �rangerF ry rJ r3 r� r5 � _finalizer8 Z_pausedrf rz r� r � _fatal_errorri r� )r$ �irC rI r1 rD rE r r r r� w s8
z"SSLProtocol._process_write_backlog�Fatal error on transportc C sX t |tj�r*| jj� rBtjd| |dd� n| jj||| j| d�� | jrT| jj |� d S )Nz%r: %sT)r� )�messageZ exceptionr� rY )
ri r Z_FATAL_ERROR_IGNORErP r� r r� Zcall_exception_handlerrd Z _force_close)r$ rE r� r r r r� � s
zSSLProtocol._fatal_errorc C s d | _ | jd k r| jj� d S )N)r} rd r^ )r$ r r r r� � s
zSSLProtocol._finalizec
C s( z| j d k r| j j� W d | j� X d S )N)rd rr r� )r$ r r r rq � s
zSSLProtocol._abort)FNT)N)N)r� )rK rL rM r& r� r� r� r� r� r� r� rT r] ro r� r� r� r� r� rq r r r r rs � s$
"
4,
rs )rw r` r � ImportError� r r r r �logr r r r r. r) r4 �objectr Z_FlowControlMixinZ TransportrO ZProtocolrs r r r r �<module> s*
wn
Mini Shell