File: //lib64/python3.6/site-packages/tornado/__pycache__/http1connection.cpython-36.pyc
3
��fO� � @ s d Z ddlZddlZddlZddlZddlmZmZmZ ddl m
Z
m Z ddl m
Z
ddl mZ ddl mZ ddlmZmZ dd lmZ dd
lmZmZmZmZmZmZm Z G d d � d e �Z G d
d� de �Z G dd� de �Z!G dd� dej"�Z#G dd� dej$�Z%G dd� de �Z&dS )zFClient and server implementations of HTTP/1.x.
.. versionadded:: 4.0
� N)�Future�future_add_done_callback�"future_set_result_unless_cancelled)�
native_str�utf8)�gen)�httputil)�iostream)�gen_log�app_log)�GzipDecompressor)�cast�Optional�Type� Awaitable�Callable�Union�Tuplec @ s e Zd Zdd�dd�ZdS )�_QuietExceptionN)�returnc C s d S )N� )�selfr r �'/usr/lib64/python3.6/http1connection.py�__init__+ s z_QuietException.__init__)�__name__�
__module__� __qualname__r r r r r r * s r c @ sJ e Zd ZdZejdd�dd�Zdd�dd�Zd ee e
j dd
�d d �Z dS )
�_ExceptionLoggingContextz�Used with the ``with`` statement when calling delegate methods to
log any exceptions with the given logger. Any exceptions caught are
converted to _QuietException
N)�loggerr c C s
|| _ d S )N)r )r r r r r r 5 s z!_ExceptionLoggingContext.__init__)r c C s d S )Nr )r r r r � __enter__8 s z"_ExceptionLoggingContext.__enter__z Optional[Type[BaseException]])�typ�value�tbr c C s2 |d k r.|d k st �| jjd|||fd� t�d S )NzUncaught exception)�exc_info)�AssertionErrorr �errorr )r r r! r" r r r �__exit__; s z!_ExceptionLoggingContext.__exit__)
r r r �__doc__�loggingZLoggerr r r �
BaseException�types�
TracebackTyper& r r r r r / s r c
@ s. e Zd ZdZdeeeeeeedd�dd�ZdS )�HTTP1ConnectionParameterszDParameters for `.HTTP1Connection` and `.HTTP1ServerConnection`.
FN)�
no_keep_alive�
chunk_size�max_header_size�header_timeout�
max_body_size� body_timeout�
decompressr c C s6 || _ |p d| _|pd| _|| _|| _|| _|| _dS )a;
:arg bool no_keep_alive: If true, always close the connection after
one request.
:arg int chunk_size: how much data to read into memory at once
:arg int max_header_size: maximum amount of data for HTTP headers
:arg float header_timeout: how long to wait for all headers (seconds)
:arg int max_body_size: maximum amount of data for body
:arg float body_timeout: how long to wait while reading body (seconds)
:arg bool decompress: if true, decode incoming
``Content-Encoding: gzip``
i N)r- r. r/ r0 r1 r2 r3 )r r- r. r/ r0 r1 r2 r3 r r r r K s
z"HTTP1ConnectionParameters.__init__)FNNNNNF)r r r r' �bool�int�floatr r r r r r, G s r, c @ s� e Zd ZdZd>ejeeedd�dd�Z e
j e e d�dd�Z
e
j ed�d d
�Zdd �d d
�Zeeg df dd�dd�Zdd �dd�Zdd �dd�Zejd �dd�Zedd�dd�Zedd�dd �Zd?ee
je
j f e
j e d d �d d �Z e e d!�d"d#�Z e d d!�d$d%�Z!dd �d&d'�Z"d dd(�d)d*�Z#e
je
j ed+�d,d-�Z$d.dd(�d/d0�Z%e e&e'e
j f d1�d2d3�Z(ee
j e
j ee d d4�d5d6�Z)ee
j dd7�d8d9�Z*e
j dd�d:d;�Z+e
j dd�d<d=�Z,dS )@�HTTP1Connectionz�Implements the HTTP/1.x protocol.
This class can be on its own for clients, or via `HTTP1ServerConnection`
for servers.
N)�stream� is_client�params�contextr c C s� || _ || _|dkrt� }|| _|| _|j| _| jjp<| jj| _| jj | _
d| _ d| _ t
� | _d| _| j� d| _d| _d| _d| _d| _d| _dS )a
:arg stream: an `.IOStream`
:arg bool is_client: client or server
:arg params: a `.HTTP1ConnectionParameters` instance or ``None``
:arg context: an opaque application-defined object that can be accessed
as ``connection.context``.
NF)r9 r8 r, r: r; r- r1 Zmax_buffer_size�_max_body_sizer2 �
_body_timeout�_write_finished�_read_finishedr �_finish_future�_disconnect_on_finish�_clear_callbacks�_request_start_line�_response_start_line�_request_headers�_chunking_output�_expected_content_remaining�_pending_write)r r8 r9 r: r; r r r r p s(
zHTTP1Connection.__init__)�delegater c C s | j jrt|| j j�}| j|�S )az Read a single HTTP response.
Typical client-mode usage is to write a request using `write_headers`,
`write`, and `finish`, and then call ``read_response``.
:arg delegate: a `.HTTPMessageDelegate`
Returns a `.Future` that resolves to a bool after the full response has
been read. The result is true if the stream is still open.
)r: r3 �_GzipMessageDelegater. �
_read_message)r rI r r r �
read_response� s z HTTP1Connection.read_responsec 4 � s� d}�z��y"| j jd| jjd�}| jjd kr8|I d H }nLy,tj| j jj� | jj |t j
d�I d H }W n tj k
r� | j � dS X | j
|�\}}| jr�tj|�}|| _|}d| _n*tj|�} | | _|| _| }| j| |� | _d}tt��& |j||�}
|
d k �r |
I d H W d Q R X | j d k�r*d}dS d} | j�r�t|tj��sHt �| jd k �rf| jj dk�rfd} |j } | dk�rzd} | dk�r�| d k �r�d
|k�s�d |k�r�tj d | ��| j |�I d H n&|j!d
�dk�r�| j" �r�| j j#d� | �s�| j$| j�r |j nd||�}
|
d k �r�| j%d k�r,|
I d H n\y*tj| j jj� | j% |
t j
d�I d H W n0 tj k
�r� t&j'd| j(� | j j � dS X d| _)| j" �s�| j�r�d}tt�� |j*� W d Q R X | j+j,� �r| j d k �r| j j-� �r| j j.| j/� | j+I d H | j�r| j�r| j � | j d k�r*dS W nV tj k
�r� } z6t&j'd| j(|� | j�sl| j j#d�I d H | j � dS d }~X nX W d |�r�tt�� |j0� W d Q R X d }| j1� X dS )NFs
?
?
)� max_bytes)Zquiet_exceptionsT�HEADi0 �d �� zContent-LengthzTransfer-Encodingz!Response code %d cannot have bodyZExpectz 100-continues HTTP/1.1 100 (Continue)
r z Timeout reading body from %sz"Malformed HTTP message from %s: %ss HTTP/1.1 400 Bad Request
)2r8 Zread_until_regexr: r/ r0 r Z with_timeout�io_loopZtimer �StreamClosedError� TimeoutError�close�_parse_headersr9 r Zparse_response_start_linerD rA Zparse_request_start_linerC rE �_can_keep_aliver r �headers_received�
isinstance�ResponseStartLiner$ �method�code�HTTPInputErrorrK �getr> �write�
_read_bodyr= r
�infor; r? �finishr@ �done�closed�set_close_callback�_on_connection_close�on_connection_closerB )r rI Zneed_delegate_closeZ
header_futureZ header_dataZstart_line_str�headersZresp_start_line�
start_lineZreq_start_lineZheader_recv_futureZ skip_bodyr[ Z body_future�er r r rK � s�
z HTTP1Connection._read_message)r c C s, d| _ d| _d| _| jdk r(| jjd� dS )z�Clears the callback attributes.
This allows the request handler to be garbage collected more
quickly in CPython by breaking up reference cycles.
N)�_write_callback�
_write_future�_close_callbackr8 rd )r r r r rB . s
z HTTP1Connection._clear_callbacks)�callbackr c C s
|| _ dS )a$ Sets a callback that will be run when the connection is closed.
Note that this callback is slightly different from
`.HTTPMessageDelegate.on_connection_close`: The
`.HTTPMessageDelegate` method is called when the connection is
closed while recieving a message. This callback is used when
there is not an active delegate (for example, on the server
side this callback is used if the client closes the connection
after sending its request but before receiving all the
response.
N)rl )r rm r r r rd : s z"HTTP1Connection.set_close_callbackc C s> | j d k r | j }d | _ |� | jj� s2t| jd � | j� d S )N)rl r@ rb r rB )r rm r r r re H s
z$HTTP1Connection._on_connection_closec C s6 | j d k r| j j� | j� | jj� s2t| jd � d S )N)r8 rT rB r@ rb r )r r r r rT T s
zHTTP1Connection.closec C s. | j � | j}d| _| jj� s*t| jd� |S )a> Take control of the underlying stream.
Returns the underlying `.IOStream` object and stops all further
HTTP processing. May only be called during
`.HTTPMessageDelegate.headers_received`. Intended for implementing
protocols like websockets that tunnel over an HTTP handshake.
N)rB r8 r@ rb r )r r8 r r r �detach[ s
zHTTP1Connection.detach)�timeoutr c C s
|| _ dS )ztSets the body timeout for a single request.
Overrides the value from `.HTTP1ConnectionParameters`.
N)r= )r ro r r r �set_body_timeoutj s z HTTP1Connection.set_body_timeout)r1 r c C s
|| _ dS )zwSets the body size limit for a single request.
Overrides the value from `.HTTP1ConnectionParameters`.
N)r< )r r1 r r r �set_max_body_sizeq s z!HTTP1Connection.set_max_body_sizez Future[None])rh rg �chunkr c C sv g }| j rht|tj�st�|| _|jtd|d |d f �� |jd kobd|kobd|kpb|d d k| _ n�t|tj
�sxt�| jd
k s�t�| j d
k s�t�|| _ |jtd |d |d f �� | jj
d
ko�|jd!ko�|jdk s�|jdko�d|ko�d|k| _ | jj
d
k�r| j�rd|d< | jj
dk�r@| j jdd�j� dk�r@d|d< | j �rPd |d< | j �r�| jjdk�s|ttj
|�jdk�r�d| _n d|k�r�t|d �| _nd
| _dd� |j� D �}|jdd� |D �� x(|D ] }d |k�r�td t|� ���q�W d
}| jj� �r(t� }| _ |j t j � � |j � nJt� }| _ d j!|�d }|�rV|| j"|�7 }| jj#|�| _$t%| j$| j&� |S )"z+Implements `.HTTPConnection.write_headers`.z%s %s HTTP/1.1r � �POST�PUT�PATCHzContent-LengthzTransfer-Encoding�chunkedNzHTTP/1.1 %d %s� zHTTP/1.1�� �0 rO rP rT �
ConnectionzHTTP/1.0� z
keep-alivez
Keep-AliverN c s s& | ] \}}t |�d t |� V qdS )z: N)r )�.0�n�vr r r � <genexpr>� s z0HTTP1Connection.write_headers.<locals>.<genexpr>c s s | ]}|j d �V qdS )�latin1N)�encode)r} �lr r r r� � s �
zNewline in header: s
s
)rt ru rv )ry rz )'r9 rX r �RequestStartLiner$ rC �appendr rZ rF rY rE rD �versionr[ rA r] �lowerr
rG r5 Zget_all�extend�
ValueError�reprr8 rc r rk �
set_exceptionr rR � exception�join�
_format_chunkr^ rH r �_on_write_complete) r rh rg rr �linesZ header_lines�line�future�datar r r �
write_headersx sf
z HTTP1Connection.write_headers)rr r c C sh | j d k r:| j t|�8 _ | j dk r:| jj� tjd��| jr`|r`tdt|� �d | d S |S d S )Nr z,Tried to write more data than Content-Lengthz%xs
)rG �lenr8 rT r �HTTPOutputErrorrF r )r rr r r r r� � s
z HTTP1Connection._format_chunkc C sh d}| j j� r6t� }| _| jjtj� � | jj� n.t� }| _| j j| j |��| _
t | j
| j � |S )z�Implements `.HTTPConnection.write`.
For backwards compatibility it is allowed but deprecated to
skip `write_headers` and instead call `write()` with a
pre-encoded header block.
N)
r8 rc r rk r� r rR r� r^ r� rH r r� )r rr r� r r r r^ � s
zHTTP1Connection.writec C s� | j dk r:| j dkr:| jj� r:| jj� tjd| j ��| jrf| jj� sf| jjd�| _| jj | j
� d| _ | j sxd| _
| jjd� | jdkr�| jd� nt| j| j� dS )z$Implements `.HTTPConnection.finish`.Nr z0Tried to write %d bytes less than Content-Lengths 0
T)rG r8 rc rT r r� rF r^ rH Zadd_done_callbackr� r> r? rA � set_nodelay�_finish_requestr )r r r r ra � s$
zHTTP1Connection.finish)r� r c C sn |j � }|d k r&t|tj� r&|j� | jd k rJ| j}d | _| jjj|� | j d k rj| j }d | _ t
|d � d S )N) r� rX r rR �resultrj r8 rQ Z add_callbackrk r )r r� �excrm r r r r� s
z"HTTP1Connection._on_write_complete)rh rg r c C sp | j jr dS |jd�}|d k r&|j� }|jdkr8|dkS d|ksd|jdd�j� dksdt|d d �d
krl|d kS dS )NFr{ zHTTP/1.1rT zContent-LengthzTransfer-Encodingr| rw rZ rN �GETz
keep-alive)rN r� )r: r- r] r� r� �getattr)r rh rg Zconnection_headerr r r rV " s
z HTTP1Connection._can_keep_alivezOptional[Future[None]]c C sH | j � | j r"| jr"| j� d S | jjd� | jj� sDt| jd � d S )NF) rB r9 rA rT r8 r� r@ rb r )r r� r r r r� 6 s
z HTTP1Connection._finish_request)r� r c C sL t |jd��jd�}|jd�}|d |� jd�}tjj||d � �}||fS )Nr� z
�
�
)r �decode�lstrip�find�rstripr � HTTPHeaders�parse)r r� Zdata_strZeolrh rg r r r rU A s
z HTTP1Connection._parse_headers)r[ rg rI r c s8 d|kr�d|krt jd��d|d krjtjd|d �� t� fdd�� D ��r^t jd|d ��� d |d< yt|d �}W n& tk
r� t jd
|d ��Y nX tt|�| jkr�t jd ��nd }|d kr�d|ks�|dkr�t jd
| ��d }|d k �r| j ||�S |j
dd�j � dk�r"| j |�S | j
�r4| j|�S d S )NzContent-LengthzTransfer-Encodingz7Response with both Transfer-Encoding and Content-Length�,z,\s*c 3 s | ]}|� d kV qdS )r Nr )r} �i)�piecesr r r� ` s z-HTTP1Connection._read_body.<locals>.<genexpr>z$Multiple unequal Content-Lengths: %rr z*Only integer Content-Length is allowed: %szContent-Length too longry z*Response with code %d should not have bodyr| rw )Nr )r r\ �re�split�anyr5 r� r
r< �_read_fixed_bodyr] r� �_read_chunked_bodyr9 �_read_body_until_close)r r[ rg rI �content_lengthr )r� r r_ M s@
zHTTP1Connection._read_body)r� rI r c � s| xv|dkrv| j jt| jj|�dd�I d H }|t|�8 }| j sD| jrtt ��" |j
|�}|d k rj|I d H W d Q R X qW d S )Nr T)�partial) r8 �
read_bytes�minr: r. r� r> r9 r r �
data_received)r r� rI �body�retr r r r� � s
z HTTP1Connection._read_fixed_bodyc � s d}�x
| j jddd�I d H }t|j� d�}|dkr\| j jd�I d H }|dkrXtjd��d S ||7 }|| jkrxtjd��|}xr|r�| j jt|| j j
�d d
�I d H }|t |�8 }| j s�| j
r~tt��" |j|�}|d k r�|I d H W d Q R X q~W | j jd�I d H }|dkst�qW d S ) Nr s
�@ )rM � rx z%improperly terminated chunked requestzchunked body too largeT)r� )r8 Z
read_untilr5 �stripr� r r\ r< r� r: r. r� r> r9 r r r� r$ ) r rI Z
total_sizeZ
chunk_len_strZ chunk_lenZcrlfZ
bytes_to_readrr r� r r r r� � s2
z"HTTP1Connection._read_chunked_bodyc � sR | j j� I d H }| j s | jrNtt��" |j|�}|d k rD|I d H W d Q R X d S )N)r8 Zread_until_closer> r9 r r r� )r rI r� r� r r r r� � s
z&HTTP1Connection._read_body_until_close)NN)N)-r r r r' r �IOStreamr4 r, �objectr r �HTTPMessageDelegater rL rK rB r r rd re rT rn r6 rp r5 rq r r� rY r� �bytesr� r� r^ ra r� rV r� r �strrU r_ r� r� r� r r r r r7 i s@ /z V 6 r7 c @ sx e Zd ZdZejedd�dd�Zeej ej
f ej e e
d d�dd�Zedd �d
d �Zdd �d
d�Zdd �dd�ZdS )rJ zIWraps an `HTTPMessageDelegate` to decode ``Content-Encoding: gzip``.
N)rI r. r c C s || _ || _d | _d S )N)� _delegate� _chunk_size�
_decompressor)r rI r. r r r r � s z _GzipMessageDelegate.__init__)rh rg r c C s: |j d�dkr,t� | _|jd|d � |d= | jj||�S )NzContent-EncodingZgzipzX-Consumed-Content-Encoding)r] r r� �addr� rW )r rh rg r r r rW � s
z%_GzipMessageDelegate.headers_received)rr r c � sr | j rP|}xb|rL| j j|| j�}|rB| jj|�}|d k rB|I d H | j j}q W n | jj|�}|d k rn|I d H d S )N)r� r3 r� r� r� Zunconsumed_tail)r rr Zcompressed_dataZ decompressedr� r r r r� � s
z"_GzipMessageDelegate.data_received)r c C s* | j d k r | j j� }|r td��| jj� S )Nz9decompressor.flush returned data; possile truncated input)r� �flushr� r� ra )r �tailr r r ra � s
z_GzipMessageDelegate.finishc C s
| j j� S )N)r� rf )r r r r rf � s z(_GzipMessageDelegate.on_connection_close)r r r r' r r� r5 r r r� rY r� r r rW r� r� ra rf r r r r rJ � s
rJ c @ sZ e Zd ZdZdejeedd�dd�Zdd�dd�Z e
j dd �d
d �Z e
j dd �d d
�Z
dS )�HTTP1ServerConnectionzAn HTTP/1.x server.N)r8 r: r; r c C s* || _ |dkrt� }|| _|| _d| _dS )z�
:arg stream: an `.IOStream`
:arg params: a `.HTTP1ConnectionParameters` or None
:arg context: an opaque application-defined object that is accessible
as ``connection.context``
N)r8 r, r: r; �_serving_future)r r8 r: r; r r r r � s z HTTP1ServerConnection.__init__)r c � sB | j j� | jdk st�y| jI dH W n tk
r< Y nX dS )zmCloses the connection.
Returns a `.Future` that resolves after the serving loop has exited.
N)r8 rT r� r$ � Exception)r r r r rT s
zHTTP1ServerConnection.close)rI r c C s> t |tj�st�tj| j|��}|| _| jj j
|dd� � dS )znStarts serving requests on this connection.
:arg delegate: a `.HTTPServerConnectionDelegate`
c S s | j � S )N)r� )�fr r r �<lambda>' s z5HTTP1ServerConnection.start_serving.<locals>.<lambda>N) rX r � HTTPServerConnectionDelegater$ r Zconvert_yielded�_server_request_loopr� r8 rQ Z
add_future)r rI Zfutr r r �
start_serving s z#HTTP1ServerConnection.start_servingc � s� z�x�t | jd| j| j�}|j| |�}y|j|�I d H }W nb tjtjt j
fk
rX d S t k
rr |j � d S t
k
r� tjddd� |j � d S X |s�d S t jd�I d H qW W d |j| � X d S )NFzUncaught exceptionT)r# r )r7 r8 r: r; Z
start_requestrL r rR ZUnsatisfiableReadError�asyncioZCancelledErrorr rT r� r
r% ZsleepZon_close)r rI ZconnZrequest_delegater� r r r r� ) s, z*HTTP1ServerConnection._server_request_loop)NN)r r r r' r r� r, r� r rT r r� r� r� r r r r r� � s
r� )'r' r� r( r� r* Ztornado.concurrentr r r Ztornado.escaper r Ztornador r r Z tornado.logr
r Z tornado.utilr Ztypingr
r r r r r r r� r r� r r, ZHTTPConnectionr7 r� rJ r� r r r r �<module> s, $" Z;