U
'�7`=x �
@ sl d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z m
Z
m Z m Z m
Z
mZ d dlmZmZmZ d dlmZ ddlmZ ddlmZ dd lmZmZ dd
l m Z m Z m Z m Z m!Z!m"Z" dd l#m$Z$m%Z% dd l&m'Z' dd
l(m)Z)m*Z* ddl+m,Z, zd dl-Z-dZ.W n e/k
�r& dZ.Y nX dZ0e1ej2�Z3e�4d�Z5e�4d�Z6e�4d�Z7e�8dddddddd d d d g
�Z9e�8d dd!d"dddd d d g �Z:G d#d$� d$e�Z;G d%d&� d&e�Z<G d'd(� d(�Z=G d)d*� d*e j>�Z?G d+d,� d,e?�Z@G d-d.� d.e?�ZAG d/d0� d0�ZBG d1d2� d2�ZCe@ZDeAZEe9ZFe:ZGz2e�sNdd3lHm@Z@mAZAm9Z9m:Z: e@ZIeAZJe9ZKe:ZLW n e/k
�rf Y nX dS )4� N)�IntEnum)�Any�List�Optional�Tuple�Type�Union)� CIMultiDict�CIMultiDictProxy�istr)�URL� )�hdrs)� BaseProtocol)�
NO_EXTENSIONS�BaseTimerContext)�
BadStatusLine�ContentEncodingError�ContentLengthError�
InvalidHeader� LineTooLong�TransferEncodingError)� HttpVersion�
HttpVersion10)�internal_logger)�
EMPTY_PAYLOAD� StreamReader)�
RawHeadersTF)�
HeadersParser�
HttpParser�HttpRequestParser�HttpResponseParser�RawRequestMessage�RawResponseMessagez [!#$%&'*+\-.^_`|~0-9A-Za-z]+zHTTP/(\d+).(\d+)s' [\x00-\x1F\x7F()<>@,;:\[\]={} \t\\\\\"]r" �method�path�version�headers� raw_headersZ should_close� compression�upgrade�chunkedZurlr# �code�reasonc @ s e Zd ZdZdZdZdZdS )�
ParseStater r
� � N)�__name__�
__module__� __qualname__�
PARSE_NONE� PARSE_LENGTH�
PARSE_CHUNKED�PARSE_UNTIL_EOF� r8 r8 �F/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/http_parser.pyr. [ s r. c @ s e Zd ZdZdZdZdZdZdS )�
ChunkStater r
r/ r0 � N)r1 r2 r3 �PARSE_CHUNKED_SIZE�PARSE_CHUNKED_CHUNK�PARSE_CHUNKED_CHUNK_EOF�PARSE_MAYBE_TRAILERS�PARSE_TRAILERSr8 r8 r8 r9 r: c s
r: c @ s> e Zd Zd eeedd�dd�Zee edef d�d d
�Z dS ) r � � � N)�
max_line_size� max_headers�max_field_size�returnc C s || _ || _|| _d S �N)rC rD rE )�selfrC rD rE r8 r8 r9 �__init__l s zHeadersParser.__init__�CIMultiDictProxy[str]��linesrF c C s� t � }g }d}|d }t|�}|�r�z|�dd�\}}W n tk
rV t|�d �Y nX |�d�}|�� }t�|�r|t|��t|�| j kr�t
d� |� dd��t
| j �t
t|����t|�} |d7 }|| }|o�|d dk}
|
�rv|g} |
�rj| t|�7 } | | j k�r(t
d � |� dd��t
| j �t
| ���| �|� |d7 }||k �r`|| }|�rh|d dk}
q�d
}�qjq�d
�| �}n0| | j k�r�t
d � |� dd��t
| j �t
| ���|�� }|� d d �} |� d d �}
|�| |
� |�||f� q t|�t|�fS )
Nr
� :s zrequest header name {}�utf8�xmlcharrefreplacer )� � zrequest header field {}� �utf-8�surrogateescape)r �len�split�
ValueErrorr �strip�lstrip�HDRRE�searchrE r �format�decode�str�append�join�addr
�tuple)rH rL r' r( Z lines_idx�lineZ
line_countZbnameZbvalueZ
header_lengthZ continuationZ
bvalue_lst�name�valuer8 r8 r9 �
parse_headersv sz
�
�
��
�� zHeadersParser.parse_headers)rA rB rA )
r1 r2 r3 �intrI r �bytesr r rf r8 r8 r8 r9 r k s ��
�r c @ s� e Zd Zdee eej eeeeee ee ee e
ee e e
e
e
dd�dd �Z
ejee ed
�d d ��Zed
�dd�Zddejejejfeeeee eeee e
ef d�dd�Zee ede ee
ee e
e
f d
�dd�Z e
dd�dd�Z dS ) r N� rA rB FT)�protocol�loop�limitrC rD rE �timerr, r$ �readall�payload_exception�response_with_body�read_until_eof�auto_decompressrF c C s� || _ || _|| _|| _|| _|| _|| _| | _|
| _| | _ | | _
|
| _ g | _ d| _
d| _d | _d | _|| _|| _t|||�| _d S )NrR F)rj rk rC rD rE rm r, r$ rn ro rp rq �_lines�_tail� _upgradedZ_payload�_payload_parser�_auto_decompress�_limitr �_headers_parser)rH rj rk rl rC rD rE rm r, r$ rn ro rp rq rr r8 r8 r9 rI � s( zHttpParser.__init__rK c C s d S rG r8 )rH rL r8 r8 r9 �
parse_message� s zHttpParser.parse_message�rF c C sz | j d k r | j �� d | _ nZ| jr0| j�| j� | jrv| jd dkrP| j�d� z| �| j�W S tk
rt Y d S X d S )N���z
rR )rv �feed_eofrt rs r_ rz � Exception�rH r8 r8 r9 r} � s
zHttpParser.feed_eof�
rR )�data�SEP�EMPTY�CONTENT_LENGTH� METH_CONNECT�SEC_WEBSOCKET_KEY1rF c C s� g }| j r | j | d }| _ t|�}d} | j}
| |k �r�| jd k�r�| j�s�|�|| �} | | krp| jsp| d } q.| | k�r�| j�|| | � � | d } | jd |k�r�z| � | j�} W 5 | j�� X | j
� |�}
|
d k �rz t |
�}
W n t
k
�r t|��Y nX |
dk �rt|��|| j
k�r,t|��| j| _t| d| j�}| jd k �sRt�|
d k �rf|
dk�sv| j�r�| j�s�t| j| j|
| jd�}t||
| j|| j| j| j| j | j d� }|j �sv|| _n�||k�rt| j| j|
| jd�}d| _t|| j| jd| j d �| _nrt| d
d �d k�rr|
d k�rr| j �rrt| j| j|
| jd�}t||
| j|| j| jd| j | j d� }|j �sv|| _nt }|�| |f� n|| d � | _ |}�q�q.| jd k�r�| j�r�| j�r�t��q�q.|�r�| |k �r�| j�r�t�| jd k �s�t�z | j�!|| d � �\}}W n` t"k
�rl } z@| j#d k �rF| jj$�%| �#t&|��� n| jj$�%|� d}d}W 5 d }~X Y nX |�r�d} t|�}d | _q.q.�q�q.|�r�| |k �r�|| d � }n|}|| j|fS )
NrR r r/ r| r$ )rm rk rl )�lengthr+ r$ r) r, rn rp rr T)r$ r) rn rr r, �d �� )'rt rU rk rv ru �findrs r_ �clearrz r' �getrg rW r r* �getattrr$ rj �AssertionErrorr+ r rm rx �HttpPayloadParserr) r, rn rp rw �donerq r � feed_data�
BaseExceptionro �payload�
set_exceptionr^ )rH r� r� r� r� r� r� ZmessagesZdata_lenZ start_posrk �pos�msgr� r$ r� Zpayload_parser�eof�excr8 r8 r9 r� s
������
��
����� �zHttpParser.feed_datarJ c C s� | j �|�\}}d}d}d}d}|�tj�}|r`|�� } | dkrFd}n| dkrTd}n | dkr`d}|�tj�}
|
r�|
�� }
|
dkr�|
}|�tj�} | r�d| �� kr�d}||||||fS ) z�Parses RFC 5322 headers from a stream.
Line continuations are supported. Returns list of header name
and value pairs. Header name is in upper case.
NF�closeTz
keep-aliver* )�gzip�deflate�brr+ )ry rf r� r Z
CONNECTION�lowerZCONTENT_ENCODINGZTRANSFER_ENCODING) rH rL r' r( Z
close_conn�encodingr* r+ Zconn�v�encZter8 r8 r9 rf � s.
zHttpParser.parse_headers)�valrF c C s
|| _ dS )zYSet connection upgraded (to websocket) mode.
:param bool val: new state.
N)ru )rH r� r8 r8 r9 � set_upgraded� s zHttpParser.set_upgraded)NNri rA rB rA NNNFNTFT) r1 r2 r3 r r �asyncioZAbstractEventLooprg r r^ �boolr r� rI �abc�abstractmethodr rh r rz r} r r� r� r� r r r� r rf r� r8 r8 r8 r9 r � sn �
� '�� .�� )r c @ s$ e Zd ZdZee ed�dd�ZdS )r z�Read request status line. Exception .http_exceptions.BadStatusLine
could be raised in case of any errors in status line.
Returns RawRequestMessage.
rK c C s\ |d � dd�}z|�d d�\}}}W n tk
rD t|�d �Y nX t|�| jkrntdt| j�tt|����|�d�\}}}|�d�\}} }
t �
|�s�t|��zB|� d�r�|d d � �d
d �\} } t t
| �t
| ��}
nt|��W n tk
r� t|��Y nX | �|�\}}}}}}|d k�r4|
tk�r0d }nd
}t|||
||||||tj||
|d d��
S )Nr rS rT r/ �Status line is too long�#�?zHTTP/� �.r
TF)r% Z query_stringZfragmentZencoded)r] rV rW r rU rC r r^ � partition�METHRE�match�
startswithr rg r~ rf r r" r Zbuild)rH rL rc r$ r% r& Z path_partZ_hash_separatorZ url_fragmentZ_question_mark_separatorZqs_partZn1Zn2� version_or' r( r� r) r* r+ r8 r8 r9 rz � sf
�
�
��z HttpRequestParser.parse_messageN�r1 r2 r3 �__doc__r rh r rz r8 r8 r8 r9 r � s r c @ s$ e Zd ZdZee ed�dd�ZdS )r! z�Read response status line and headers.
BadStatusLine could be raised in case of any errors in status line.
Returns RawResponseMessagerK c
C sT |d � dd�}z|�d d�\}}W n tk
rB t|�d �Y nX z|�d d�\}}W n tk
rp d}Y nX t|�| jkr�tdt| j�tt|����t� |�}|d kr�t|��t
t |� d��t |� d���}z t |�}W n tk
r� t|�d �Y nX |dk�rt|��| �
|�\} }
} } }
}| d k�r8|tk} t|||�� | |
| | |
|� S ) Nr rS rT r
� r� r/ i� )r] rV rW r rU rC r r^ �VERSREr� r rg �grouprf r r# rX )rH rL rc r& Zstatusr- r� r� Zstatus_ir' r( r� r) r* r+ r8 r8 r9 rz 3 s\
�
�
�z HttpResponseParser.parse_messageNr� r8 r8 r8 r9 r! - s r! c @ sj e Zd Zdeee eee ee ee eeedd�
dd�Zdd�dd �Z de
e
e
e ee
f d �d
d�Z dS )r� NFT)
r� r� r+ r) r, r$ rn rp rr rF c
C s� d| _ tj| _tj| _d| _d| _| | _ d| _
|rH|rH| j rHt ||�}
n|}
|shtj| _|
� � d| _
n||rvtj
| _nn|d k r�tj| _|| _ | j dkr�|
� � d| _
n>|r�|dkr�tj| _n(|dkr�t�d� tj| _|
� � d| _
|
| _d S )Nr rR FT�� )ZPUTZPOSTz6Content-Length or Transfer-Encoding header is required)�_lengthr. r4 �_typer: r< �_chunk� _chunk_size� _chunk_tailrw r� �
DeflateBufferr} r6 r5 r7 r Zwarningr� ) rH r� r� r+ r) r, r$ rn rp rr Z real_payloadr8 r8 r9 rI n sF �
�zHttpPayloadParser.__init__r{ c C sF | j tjkr| j�� n*| j tjkr.td��n| j tjkrBtd��d S )Nz2Not enough data for satisfy content length header.z3Not enough data for satisfy transfer length header.) r� r. r7 r� r} r5 r r6 r r r8 r8 r9 r} � s � �zHttpPayloadParser.feed_eofr� � ;)�chunkr� � CHUNK_EXTrF c C s: | j tjkr�| j}t|�}||krT|| | _| j�||� | jdkr�| j�� dS n6d| _| j�|d |� |� | j�� d||d � fS �n�| j tjk�r| j r�| j | }d| _ |�r6| j
t j k�r�|�
|�}|dk�r�|�
|d|�}|dkr�|d |� }n |d |� }ztt|�d�} W n@ tk
�r\ t|d |� �dd��}
| j�|
� |
d �Y nX ||d d � }| dk�r�t j| _
nt j| _
| | _| j�� n
|| _ d S | j
t jk�r"| j}t|�}||k�r�|| | _| j�||� d S d| _| j�|d |� |� ||d � }t j| _
| j�� | j
t jk�rb|d d� |k�rX|dd � }t j | _
n
|| _ d S | j
t jk�r�|d d� } | |k�r�| j�� d|dd � fS | �s�d S | |d d
� k�r�| | _ d S t j| _
| j
t jkr�|�
|�}|dk�r||d d � }t j| _
q�|| _ d S q�n | j tjk�r6| j�|t|�� d S ) Nr )TrR TrR � �asciirT r/ )FrR r
) r� r. r5 r� rU r� r� r} r6 r� r� r: r< r� rg rh rW r r] r� r? r= r� �begin_http_chunk_receivingr> �end_http_chunk_receivingr@ r7 ) rH r� r� r� ZrequiredZ chunk_lenr� �iZsize_b�sizer� �headr8 r8 r9 r� � s�
�
zHttpPayloadParser.feed_data)NFNNNFTT)r� r� )
r1 r2 r3 r r rg r� r^ rI r} rh r r� r8 r8 r8 r9 r� m s: �� 8
�
�r� c @ sr e Zd ZdZeee dd�dd�Zedd�dd�Z e
e dd �d
d �Z dd �d
d�Z
dd �dd�Zdd �dd�ZdS )r� zDDeflateStream decompress stream and feed data into specified stream.N)�outr� rF c C sb || _ d| _|| _d| _|dkr8ts,td��t�� | _n&|dkrJdt j
nt j
}t j |d�| _d S )Nr Fr� zGCan not decode content-encoding: brotli (br). Please install `brotlipy`r� r� �Zwbits) r� r� r� �_started_decoding�
HAS_BROTLIr �brotliZ Decompressor� decompressor�zlib� MAX_WBITS�
decompressobj)rH r� r� Z zlib_moder8 r8 r9 rI / s � zDeflateBuffer.__init__)r� rF c C s | j �|� d S rG )r� r� )rH r� r8 r8 r9 r� @ s zDeflateBuffer.set_exception)r� r� rF c C s� |sd S | j |7 _ | jsH| jdkrH|d d@ dkrHtjtj d�| _z| j�|�}W n" tk
rz t d| j ��Y nX d| _|r�| j
� |t |�� d S )Nr� r � � r� z#Can not decode content-encoding: %sT)
r� r� r� r� r� r� r� �
decompressr~ r r� r� rU )rH r� r� r8 r8 r9 r� C s& ����
zDeflateBuffer.feed_datar{ c C sR | j �� }|s| jdkrD| j�|t|�� | jdkrD| j jsDtd��| j� � d S )Nr r� )
r� �flushr� r� r� rU r� r� r r} )rH r� r8 r8 r9 r} a s
zDeflateBuffer.feed_eofc C s | j �� d S rG )r� r� r r8 r8 r9 r� k s z(DeflateBuffer.begin_http_chunk_receivingc C s | j �� d S rG )r� r� r r8 r8 r9 r� n s z&DeflateBuffer.end_http_chunk_receiving)r1 r2 r3 r� r r r^ rI r� r� rh rg r� r} r� r� r8 r8 r8 r9 r� , s
r� )r r! r" r# )Mr� r� � collections�re�stringr� �enumr �typingr r r r r r Z multidictr r
r Zyarlr r� r Z
base_protocolr Zhelpersr r Zhttp_exceptionsr r r r r r Z http_writerr r �logr Zstreamsr r Ztypedefsr r� r� � ImportError�__all__�setZ printableZASCIISET�compiler� r� rZ �
namedtupler" r# r. r: r �ABCr r r! r� r� ZHttpRequestParserPyZHttpResponseParserPyZRawRequestMessagePyZRawResponseMessagePyZ _http_parserZHttpRequestParserCZHttpResponseParserCZRawRequestMessageCZRawResponseMessageCr8 r8 r8 r9 �<module> s�
����^ K@ @F