3
\�_ � @ sH d ddddddgZ ddlZeed �r2e jd
d g� d d
lmZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d!Z G dd� de
�ZG dd� de�Ze d"de d�dd��Ze d#de d�dd��Zeed ��re d$de d�dd
��Ze d%de d�dd ��ZG d d � d ej�ZG d d� deej�ZG d d� d�ZG d d � d �ZdS )&� StreamReader� StreamWriter�StreamReaderProtocol�open_connection� start_server�IncompleteReadError�LimitOverrunError� NZAF_UNIX�open_unix_connection�start_unix_server� )�
coroutines)�compat)�events)� protocols)� coroutine)�logger� � c s$ e Zd Z� fdd�Zdd� Z� ZS )r c s( t � jdt|�|f � || _|| _d S )Nz-%d bytes read on a total of %r expected bytes)�super�__init__�len�partial�expected)�selfr r )� __class__� �'/usr/lib64/python3.6/asyncio/streams.pyr s z IncompleteReadError.__init__c C s t | �| j| jffS )N)�typer r )r r r r �
__reduce__& s z IncompleteReadError.__reduce__)�__name__�
__module__� __qualname__r r �
__classcell__r r )r r r s c s$ e Zd Z� fdd�Zdd� Z� ZS )r c s t � j|� || _d S )N)r r �consumed)r �messager# )r r r r 0 s zLimitOverrunError.__init__c C s t | �| jd | jffS )Nr )r �argsr# )r r r r r 4 s z LimitOverrunError.__reduce__)r r r! r r r" r r )r r r * s )�loop�limitc + sb |d krt j� }t||d�}t||d�� |j� fdd�| |f|�E d H \}}t|� ||�}||fS )N)r' r& )r& c s � S )Nr r )�protocolr r �Q s z!open_connection..)r �get_event_loopr r Zcreate_connectionr ) �host�portr& r' �kwds�reader� transport�_�writerr )r( r r 8 s c + s8 �d krt j� �� ��fdd�}�j|||f|�E d H S )Nc s t ��d�} t| � �d�}|S )N)r' r& )r& )r r )r. r( )�client_connected_cbr' r& r r �factoryq s z start_server..factory)r r* Z
create_server)r2 r+ r, r& r' r- r3 r )r2 r' r& r r V s c + s` |d krt j� }t||d�}t||d�� |j� fdd�| f|�E d H \}}t|� ||�}||fS )N)r' r& )r& c s � S )Nr r )r( r r r) � s z&open_unix_connection..)r r* r r Zcreate_unix_connectionr )�pathr& r' r- r. r/ r0 r1 r )r( r r } s c + s6 �d krt j� �� ��fdd�}�j||f|�E d H S )Nc s t ��d�} t| � �d�}|S )N)r' r& )r& )r r )r. r( )r2 r' r& r r r3 � s z"start_unix_server..factory)r r* Zcreate_unix_server)r2 r4 r& r' r- r3 r )r2 r' r& r r
� s c @ s: e Zd Zd dd�Zdd� Zdd� Zdd � Zed
d � �ZdS )
�FlowControlMixinNc C s0 |d krt j� | _n|| _d| _d | _d| _d S )NF)r r* �_loop�_paused�
_drain_waiter�_connection_lost)r r& r r r r � s zFlowControlMixin.__init__c C s d| _ | jj� r tjd| � d S )NTz%r pauses writing)r7 r6 � get_debugr �debug)r r r r �
pause_writing� s
z FlowControlMixin.pause_writingc C sF d| _ | jj� r tjd| � | j}|d k rBd | _|j� sB|jd � d S )NFz%r resumes writing)r7 r6 r: r r; r8 �done�
set_result)r �waiterr r r �resume_writing� s
z FlowControlMixin.resume_writingc C sV d| _ | jsd S | j}|d kr"d S d | _|j� r4d S |d krH|jd � n
|j|� d S )NT)r9 r7 r8 r= r> �
set_exception)r �excr? r r r �connection_lost� s z FlowControlMixin.connection_lostc c s< | j rtd��| jsd S | j}| jj� }|| _|E d H d S )NzConnection lost)r9 �ConnectionResetErrorr7 r8 r6 �
create_future)r r? r r r �
_drain_helper� s
z FlowControlMixin._drain_helper)N) r r r! r r< r@ rC r rF r r r r r5 � s
r5 c sB e Zd Zd � fdd� Zdd� Z� fdd�Zdd � Zd
d � Z� ZS )
r Nc s* t � j|d� || _d | _|| _d| _d S )N)r& F)r r �_stream_reader�_stream_writer�_client_connected_cb� _over_ssl)r Z
stream_readerr2 r& )r r r r � s
z StreamReaderProtocol.__init__c C sd | j j|� |jd�d k | _| jd k r`t|| | j | j�| _| j| j | j�}tj |�r`| jj
|� d S )NZ
sslcontext) rG �
set_transport�get_extra_inforJ rI r r6 rH r Z iscoroutineZ create_task)r r/ �resr r r �connection_made� s
z$StreamReaderProtocol.connection_madec sF | j d k r*|d kr | j j� n | j j|� t� j|� d | _ d | _d S )N)rG �feed_eofrA r rC rH )r rB )r r r rC � s
z$StreamReaderProtocol.connection_lostc C s | j j|� d S )N)rG � feed_data)r �datar r r �
data_received s z"StreamReaderProtocol.data_receivedc C s | j j� | jrdS dS )NFT)rG rO rJ )r r r r � eof_received s
z!StreamReaderProtocol.eof_received)NN) r r r! r rN rC rR rS r" r r )r r r � s
c @ sf e Zd Zdd� Zdd� Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z ddd�Z e
dd� �ZdS )r c C s || _ || _|| _|| _d S )N)�
_transport� _protocol�_readerr6 )r r/ r( r. r& r r r r s zStreamWriter.__init__c C s: | j jd| j g}| jd k r,|jd| j � ddj|� S )Nz transport=%rz reader=%rz<%s>� )r r rT rV �append�join)r �infor r r �__repr__! s
zStreamWriter.__repr__c C s | j S )N)rT )r r r r r/ ' s zStreamWriter.transportc C s | j j|� d S )N)rT �write)r rQ r r r r\ + s zStreamWriter.writec C s | j j|� d S )N)rT �
writelines)r rQ r r r r] . s zStreamWriter.writelinesc C s
| j j� S )N)rT � write_eof)r r r r r^ 1 s zStreamWriter.write_eofc C s
| j j� S )N)rT �
can_write_eof)r r r r r_ 4 s zStreamWriter.can_write_eofc C s
| j j� S )N)rT �close)r r r r r` 7 s zStreamWriter.closeNc C s | j j||�S )N)rT rL )r �name�defaultr r r rL : s zStreamWriter.get_extra_infoc c sN | j d k r | j j� }|d k r |�| jd k r:| jj� r:d V | jj� E d H d S )N)rV � exceptionrT Z
is_closingrU rF )r rB r r r �drain= s
zStreamWriter.drain)N)r r r! r r[ �propertyr/ r\ r] r^ r_ r` rL r rd r r r r r s
c @ s� e Zd Zedfdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
edd� �Zedd� �Zed'dd ��Zed)d d ��Zed d!� �Zejr�ed"d#� �Zed$d%� �Zejr�d&d#� ZdS )*r Nc C sZ |dkrt d��|| _|d kr*tj� | _n|| _t� | _d| _d | _d | _ d | _
d| _ d S )Nr zLimit cannot be <= 0F) �
ValueError�_limitr r* r6 � bytearray�_buffer�_eof�_waiter�
_exceptionrT r7 )r r' r& r r r r X s zStreamReader.__init__c C s� dg}| j r |jdt| j � � | jr0|jd� | jtkrJ|jd| j � | jr`|jd| j � | jrv|jd| j � | jr�|jd| j � | j r�|jd� d d
j
|� S ) Nr z%d bytes�eofzl=%dzw=%rze=%rzt=%rZpausedz<%s>rW ) ri rX r rj rg �_DEFAULT_LIMITrk rl rT r7 rY )r rZ r r r r[ k s
zStreamReader.__repr__c C s | j S )N)rl )r r r r rc } s zStreamReader.exceptionc C s0 || _ | j}|d k r,d | _|j� s,|j|� d S )N)rl rk � cancelledrA )r rB r? r r r rA � s zStreamReader.set_exceptionc C s* | j }|d k r&d | _ |j� s&|jd � d S )N)rk ro r> )r r? r r r �_wakeup_waiter� s
zStreamReader._wakeup_waiterc C s
|| _ d S )N)rT )r r/ r r r rK � s zStreamReader.set_transportc C s* | j r&t| j�| jkr&d| _ | jj� d S )NF)r7 r ri rg rT �resume_reading)r r r r �_maybe_resume_transport� s z$StreamReader._maybe_resume_transportc C s d| _ | j� d S )NT)rj rp )r r r r rO � s zStreamReader.feed_eofc C s | j o | j S )N)rj ri )r r r r �at_eof� s zStreamReader.at_eofc C sv |sd S | j j|� | j� | jd k rr| j rrt| j �d| j krry| jj� W n tk
rj d | _Y nX d| _d S )Nr T) ri �extendrp rT r7 r rg Z
pause_reading�NotImplementedError)r rQ r r r rP � s
zStreamReader.feed_datac c sV | j d k rtd| ��| jr,d| _| jj� | jj� | _ z| j E d H W d d | _ X d S )NzH%s() called while another coroutine is already waiting for incoming dataF)rk � RuntimeErrorr7 rT rq r6 rE )r Z func_namer r r �_wait_for_data� s
zStreamReader._wait_for_datac c s� d}t |�}y| j|�E d H }W n� tk
rB } z|jS d }~X nf tk
r� } zJ| jj||j�rv| jd |j| �= n
| jj� | j � t
|j d ��W Y d d }~X nX |S )N�
r ) r � readuntilr r r ri �
startswithr# �clearrr rf r% )r �sep�seplen�line�er r r �readline� s
zStreamReader.readlinerx c c s� t |�}|dkrtd��| jd k r(| j�d}x�t | j�}|| |kr|| jj||�}|dkr\P |d | }|| jkr|td|��| jr�t| j�}| jj � t
|d ��| j d�E d H q.W || jkr�td|��| jd || � }| jd || �= | j � t|�S )Nr z,Separator should be at least one-byte stringr z2Separator is not found, and chunk exceed the limitry z2Separator is found, but chunk is longer than limit�����)
r rf rl ri �findrg r rj �bytesr{ r rw rr )r Z separatorr} �offsetZbuflenZisep�chunkr r r ry � s:
zStreamReader.readuntilr c c s� | j d k r| j �|dkr dS |dk rZg }x&| j| j�E d H }|sBP |j|� q*W dj|�S | j rz| j rz| jd�E d H t| jd |� �}| jd |�= | j � |S )Nr � �read)
rl r� rg rX rY ri rj rw r� rr )r �nZblocks�blockrQ r r r r� P s$
zStreamReader.readc c s� |dk rt d��| jd k r | j�|dkr,dS xFt| j�|k rr| jr`t| j�}| jj� t||��| jd�E d H q.W t| j�|kr�t| j�}| jj� n t| jd |� �}| jd |�= | j � |S )Nr z*readexactly size can not be less than zeror� � readexactly)
rf rl r ri rj r� r{ r rw rr )r r� Z
incompleterQ r r r r� � s&
zStreamReader.readexactlyc C s | S )Nr )r r r r � __aiter__� s zStreamReader.__aiter__c c s | j � E d H }|dkrt�|S )Nr� )r� �StopAsyncIteration)r �valr r r � __anext__� s zStreamReader.__anext__c C s | S )Nr )r r r r r� � s )rx r� )r� )r r r! rn r r[ rc rA rp rK rr rO rs rP r rw r� ry r� r� r
ZPY35r� r� ZPY352r r r r r V s, [ 2 * i )NN)NN)N)N)�__all__Zsocket�hasattrrt � r r
r r r �logr rn �EOFErrorr � Exceptionr r r r r
ZProtocolr5 r r r r r r r � s@
" B3G