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 deZe d"de dddZe d#de dddZeed re d$de ddd
Ze d%de ddd ZG d d d ejZG d d deejZG d d dZG d d d ZdS )& StreamReader StreamWriterStreamReaderProtocolopen_connection start_serverIncompleteReadErrorLimitOverrunError NZAF_UNIXopen_unix_connectionstart_unix_server )
coroutines)compat)events) protocols) coroutine)logger c s$ e Zd Z fddZdd Z ZS )r c s( t jdt||f || _|| _d S )Nz-%d bytes read on a total of %r expected bytes)super__init__lenpartialexpected)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 fddZdd 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 )looplimitc + 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 ) hostportr& r' kwdsreader 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 ddZdd 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 fddZdd 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| |jdd k | _| jd k r`t|| | j | j| _| j| j | j}tj |r`| jj
| d S )NZ
sslcontext) rG
set_transportget_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 dddZ 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 appendjoin)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 namedefaultr 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 ZedfddZdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
edd Zedd Zed'dd Zed)d d Zed d! Zejred"d# Zed$d% Zejrd&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 byteseofzl=%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_readingNotImplementedError)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||jrv| 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 sepseplenlineer r r readline s
zStreamReader.readlinerx c c s t |}|dkrtd| jd k r(| jd}xt | j}|| |kr|| jj||}|dkr\P |d | }|| jkr|td|| jrt| j}| jj t
|d | j dE d H q.W || jkrtd|| 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} offsetZbuflenZisepchunkr r r ry s:
zStreamReader.readuntilr c c s | j d k r| j |dkr dS |dk rZg }x&| j| jE d H }|sBP |j| q*W dj|S | j rz| j rz| jdE d H t| jd | }| jd |= | j |S )Nr read)
rl r rg rX rY ri rj rw r rr )r nZblocksblockrQ 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||| jdE d H q.W t| j|krt| 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__Zsockethasattrrt r r
r r r logr rn EOFErrorr Exceptionr r r r r
ZProtocolr5 r r r r r r r s@
" B3G