Edit File: http_parser.cpython-37.pyc
B �k�aN � @ s� U d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl m Z m Z m Z mZmZmZmZmZmZmZmZmZ d dlmZmZmZ d dlm Z ddl m Z dd l m Z dd l!m"Z"m#Z# dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ dd l,m-Z-m.Z. dd l/m0Z0 ddl1m2Z2m3Z3 ddl4m5Z5m6Z6 yd dl7Z7dZ8W n e9k �rT dZ8Y nX dZ:e;ej<�Z=e5ee> e?d< e�@d�ZAe5ee> e?d< e�@d�ZBe5ee> e?d< e�@d�ZCe5eeD e?d< G dd� de�ZEe�Fd d d d d d!d"d#d$d%g �ZGed&eEeG�ZHG d'd(� d(e �ZIG d)d*� d*e �ZJG d+d,� d,�ZKG d-d.� d.e jLe eH �ZMG d/d0� d0eMeE �ZNG d1d2� d2eMeG �ZOG d3d4� d4�ZPG d5d6� d6�ZQeNZReOZSeEZTeGZUy2e"�s�dd7lVmNZNmOZOmEZEmGZG eNZWeOZXeEZYeGZZW n e9k �r� Y nX dS )8� N)�suppress)�IntEnum) �Any�Generic�List� NamedTuple�Optional�Pattern�Set�Tuple�Type�TypeVar�Union�cast)� CIMultiDict�CIMultiDictProxy�istr)�URL� )�hdrs)� BaseProtocol)� NO_EXTENSIONS�BaseTimerContext)�BadHttpMessage� BadStatusLine�ContentEncodingError�ContentLengthError� InvalidHeader� LineTooLong�TransferEncodingError)� HttpVersion� HttpVersion10)�internal_logger)� EMPTY_PAYLOAD� StreamReader)�Final� RawHeadersTF)� HeadersParser� HttpParser�HttpRequestParser�HttpResponseParser�RawRequestMessage�RawResponseMessage�ASCIISETz [!#$%&'*+\-.^_`|~0-9A-Za-z]+�METHREzHTTP/(\d+).(\d+)�VERSREs' [\x00-\x1F\x7F()<>@,;:\[\]={} \t\\\\\"]�HDRREc @ sb e Zd ZU eed< eed< eed< ded< eed< eed< ee ed< eed < eed < e ed < d S ) r+ �method�path�versionzCIMultiDictProxy[str]�headers� raw_headers� should_close� compression�upgrade�chunked�urlN) �__name__� __module__� __qualname__�str�__annotations__r r& �boolr r � rA rA �F/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/http_parser.pyr+ K s r+ r, r3 �code�reasonr4 r5 r6 r7 r8 r9 �_MsgTc @ s e Zd ZdZdZdZdZdS )� ParseStater r � � N)r; r< r= � PARSE_NONE� PARSE_LENGTH� PARSE_CHUNKED�PARSE_UNTIL_EOFrA rA rA rB rF k s rF c @ s e Zd ZdZdZdZdZdZdS )� ChunkStater r rG rH � N)r; r< r= �PARSE_CHUNKED_SIZE�PARSE_CHUNKED_CHUNK�PARSE_CHUNKED_CHUNK_EOF�PARSE_MAYBE_TRAILERS�PARSE_TRAILERSrA rA rA rB rM s s rM 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)rV rW rX )�selfrV rW rX rA rA rB �__init__| s zHeadersParser.__init__zCIMultiDictProxy[str])�linesrY c C s� t � }g }d}|d }t|�}�x�|�r�y|�dd�\}}W n tk rZ t|�d �Y nX |�d�}|�� }t�|�r�t|��t|�| j kr�t d� |� dd��t | j �t t|����t|�} |d7 }|| }|o�|d dk} | �r||g} x�| �rn| t|�7 } | | j k�r.t d � |� dd��t | j �t | ���| �|� |d7 }||k �rf|| }|�rl|d dk} q�d }P q�W d �| �}n0| | j k�r�t d � |� dd��t | j �t | ���|�� }|� d d �} |� d d �} |�| | � |�||f� q"W t|�t|�fS ) Nr � :s zrequest header name {}�utf8�xmlcharrefreplacer )� � zrequest header field {}� zutf-8�surrogateescape)r �len�split� ValueErrorr �strip�lstripr0 �searchrX r �format�decoder> �append�join�addr �tuple)rZ r\ r4 r5 Z lines_idx�lineZ line_countZbnameZbvalue� header_lengthZ continuationZ bvalue_lst�name�valuerA rA rB � parse_headers� sn zHeadersParser.parse_headers)rT rU rT ) r; r< r= �intr[ r �bytesr r& rt rA rA rA rB r' { s r' c @ s e Zd Zdee eej eeeeee ee ee e ee e e e e dd�dd �Z ejee ed �d d ��Zee d �dd�Zddejejejfeeeee eeeeeef e ef d�dd�Z ee ede ee ee e e f d �dd�Z e dd�dd�Z dS ) r( N� � � � FT)�protocol�loop�limitrV rW rX �timerrC r1 �readall�payload_exception�response_with_body�read_until_eof�auto_decompressrY c C s� || _ || _|| _|| _|| _|| _|| _| | _| | _| | _ | | _ | | _ g | _ d| _ d| _d | _d | _|| _|| _t|||�| _d S )Nrb F)rz r{ rV rW rX r} rC r1 r~ r r� r� �_lines�_tail� _upgraded�_payload�_payload_parser�_auto_decompress�_limitr' �_headers_parser)rZ rz r{ r| rV rW rX r} rC r1 r~ r r� r� r� rA rA rB r[ � s( zHttpParser.__init__)r\ rY c C s d S )NrA )rZ r\ rA rA rB � parse_message s zHttpParser.parse_message)rY c C sp | j d k r | j �� d | _ nP| jr0| j�| j� | jrl| jd dkrP| j�d� tt�� | �| j�S Q R X d S )N���z rb )r� �feed_eofr� r� rl r � Exceptionr� )rZ rA rA rB r� s zHttpParser.feed_eofs rb )�data�SEP�EMPTY�CONTENT_LENGTH� METH_CONNECT�SEC_WEBSOCKET_KEY1rY c s� g }| j r | j | d }| _ t|�}d} | j} �x6| |k �rf| jd k�rx| j�sx|�|| �} | | krt| jst| d } q2| | k�rb| j�|| | � � | d } | jd |k�rvz| �| j��W d | j� � X t t d�� �fdd� } | � } |�j kr�t |���j| _t�d| j�}| jd k �s t�| d k �r2| dk�sB�j�r��j�s�t| j| j| | jd �}t|| �j|�j| j| j| j| j d � }|j �sR|| _n�||k�r�t �t ��s�t�t| j| j| | jd �}d | _t|�j�jd | j d �| _nrt�d d�dk�rN| d k�rN| j �rNt| j| j| | jd �}t|| �j|�j| jd | j| j d � }|j �sR|| _nt!}|��|f� n|| d � | _ |}P q2| jd k�r�| j�r�| j�r�t�P q2|�rb| |k �rb| j�r�t�| jd k �s�t�y | j�"|| d � �\}}W n` t#k �rD } z@| j$d k �r | jj%�&| �$t'|��� n| jj%�&|� d }d}W d d }~X Y nX |�rdd} t|�}d | _q2q2P q2W |�r�| |k �r�|| d � }n|}|| j|fS )Nrb r rG r� )rY c sV �j �� �} | d krd S y t| �}W n tk r@ t� ��Y nX |dk rRt� ��|S )Nr )r4 �getru rf r )Z length_hdr�length)r� �msgrA rB �get_content_length? s z0HttpParser.feed_data.<locals>.get_content_lengthr1 )r} r{ r| )r� r9 r1 r7 rC r~ r� r� T)r1 r7 r~ r� rC �d �� )(r� rd r{ r� r� �findr� rl r� �clearr ru r4 r r8 �getattrr1 rz �AssertionErrorr9 r$ r} r� �HttpPayloadParserr7 rC r~ r� r� �done� isinstancer+ r� r# � feed_data� BaseExceptionr �payload� set_exceptionr> )rZ r� r� r� r� r� r� �messagesZdata_lenZ start_posr{ �posr� r� r1 r� Zpayload_parser�eof�excrA )r� r� rB r� s� zHttpParser.feed_datazCIMultiDictProxy[str]c C s� | j �|�\}}d}d}d}d}|�tj�}|r`|�� } | dkrFd}n| dkrTd}n | dkr`d}|�tj�} | r�| �� } | dkr�| }|�tj�} | dk r�d| �� kr�d}ntd ��tj |kr�td ��||||||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-aliver8 )�gzip�deflate�brr9 z'Request has invalid `Transfer-Encoding`z6Content-Length can't be present with Transfer-Encoding) r� rt r� r Z CONNECTION�lowerZCONTENT_ENCODINGZTRANSFER_ENCODINGr r� ) rZ r\ r4 r5 Z close_conn�encodingr8 r9 Zconn�v�enc�terA rA rB rt � s8 zHttpParser.parse_headers)�valrY c C s || _ dS )zZSet connection upgraded (to websocket) mode. :param bool val: new state. N)r� )rZ r� rA rA rB � set_upgraded� s zHttpParser.set_upgraded)NNrw rx ry rx NNNFNTFT) r; r< r= r r �asyncioZAbstractEventLoopru r r> r@ r r� r[ �abc�abstractmethodr rv rE r� r� r r� r� r� r r r$ r� r& rt r� rA rA rA rB r( � s: <$ .$.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. )r\ rY c C s\ |d � dd�}y|�d d�\}}}W n tk rD t|�d �Y nX t|�| jkrntdt| j�tt|����|�d�\}}}|�d�\}} } t � |�s�t|��yB|� 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 zutf-8rc rG zStatus line is too long�#�?zHTTP/� �.r TF)r2 Z query_string�fragment�encoded)rk re rf r rd rV r r> � partitionr. �match� startswithr ru r� rt r! r+ r �build)rZ r\ rp r1 r2 r3 Z path_partZ_hash_separatorZ url_fragmentZ_question_mark_separatorZqs_partZn1Zn2� version_or4 r5 r� r7 r8 r9 rA rA rB r� sN z HttpRequestParser.parse_messageN)r; r< r= �__doc__r rv r+ r� rA rA rA rB 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 RawResponseMessage. )r\ rY c C sT |d � dd�}y|�d d�\}}W n tk rB t|�d �Y nX y|�d d�\}}W n tk rp d}Y nX t|�| jkr�tdt| j�tt|����t� |�}|d kr�t|��t t |� d��t |� d���}y t |�}W n tk r� t|�d �Y nX |dk�rt|��| � |�\} } } } } }| d k�r8|tk} t|||�� | | | | | |� S ) Nr zutf-8rc r � zStatus line is too longrG i� )rk re rf r rd rV r r> r/ r� r ru �grouprt r! r, rg )rZ r\ rp r3 �statusrD r� r� Zstatus_ir4 r5 r� r7 r8 r9 rA rA rB r� V sF z HttpResponseParser.parse_messageN)r; r< r= r� r rv r, r� rA rA rA rB r* O 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� r9 r7 rC r1 r~ r� r� rY 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 rb FT�� )ZPUTZPOSTz6Content-Length or Transfer-Encoding header is required)�_lengthrF rI �_typerM rO �_chunk� _chunk_size� _chunk_tailr� r� � DeflateBufferr� rK rJ rL r" �warningr� ) rZ r� r� r9 r7 rC r1 r~ r� r� Z real_payloadrA rA rB r[ � s@ zHttpPayloadParser.__init__)rY 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� rF rL r� r� rJ r rK r )rZ rA rA rB r� � s zHttpPayloadParser.feed_eof� � ;)�chunkr� � CHUNK_EXTrY c C sB | j tjkr�| j}t|�}||krT|| | _| j�||� | jdkr�| j�� dS n6d| _| j�|d |� |� | j�� d||d � fS �n�| j tjk�r | j r�| j | }d| _ �x�|�r| j t j k�r�|� |�}|dk�r�|� |d|�}|dk�r|d |� }n |d |� }ytt|�d�} W n@ tk �rb 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�rh|d d� |k�r^|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�W n | j tjk�r>| j�|t|�� d S ) Nr )Trb Trb � �asciirc rG )Frb r ) r� rF rJ r� rd r� r� r� rK r� r� rM rO r� ru rv rf r rk r� rR rP r� �begin_http_chunk_receivingrQ �end_http_chunk_receivingrS rL ) rZ r� r� r� �requiredZ chunk_lenr� �iZsize_b�sizer� �headrA rA rB r� � s� zHttpPayloadParser.feed_data)NFNNNFTT)r� r� ) r; r< r= r$ r ru r@ r> r[ r� rv r r� rA rA rA rB r� � s $- r� c @ s| e Zd ZU dZeed< eee dd�dd�Z e dd�dd �Z e e dd �d d �Zdd �dd�Zdd �dd�Zdd �dd�ZdS )r� zDDeflateStream decompress stream and feed data into specified stream.� decompressorN)�outr� rY c C sn || _ d| _|| _d| _|dkrDts,td��G dd� d�}|� | _n&|dkrVdtj ntj}tj |d �| _d S ) Nr Fr� zECan not decode content-encoding: brotli (br). Please install `Brotli`c @ s8 e Zd Zdd�dd�Zeed�dd�Zed�dd �ZdS ) z-DeflateBuffer.__init__.<locals>.BrotliDecoderN)rY c S s t �� | _d S )N)�brotliZ Decompressor�_obj)rZ rA rA rB r[ e s z6DeflateBuffer.__init__.<locals>.BrotliDecoder.__init__)r� rY c S s0 t | jd�r tt| j�|��S tt| j�|��S )N� decompress)�hasattrr� r rv r� �process)rZ r� rA rA rB r� h s z8DeflateBuffer.__init__.<locals>.BrotliDecoder.decompressc S s t | jd�r tt| j�� �S dS )N�flushrb )r� r� r rv r� )rZ rA rA rB r� m s z3DeflateBuffer.__init__.<locals>.BrotliDecoder.flush)r; r< r= r[ rv r� r� rA rA rA rB � BrotliDecodera s r� r� r� )�wbits) r� r� r� �_started_decoding� HAS_BROTLIr r� �zlib� MAX_WBITS� decompressobj)rZ r� r� r� Z zlib_moderA rA rB r[ T s zDeflateBuffer.__init__)r� rY c C s | j �|� d S )N)r� r� )rZ r� rA rA rB r� w s zDeflateBuffer.set_exception)r� r� rY c C s� |sd S | j |7 _ | jsH| jdkrH|d d@ dkrHtjtj d�| _y| 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� r� r� r r� r� rd )rZ r� r� rA rA rB r� z s zDeflateBuffer.feed_data)rY c C sR | j �� }|s| jdkrD| j�|t|�� | jdkrD| j jsDtd��| j� � d S )Nr r� ) r� r� r� r� r� rd r� r� r r� )rZ r� rA rA rB r� � s zDeflateBuffer.feed_eofc C s | j �� d S )N)r� r� )rZ rA rA rB r� � s z(DeflateBuffer.begin_http_chunk_receivingc C s | j �� d S )N)r� r� )rZ rA rA rB r� � s z&DeflateBuffer.end_http_chunk_receiving)r; r< r= r� r r? r$ r r> r[ r� r� rv ru r� r� r� r� rA rA rA rB r� O s # r� )r) r* r+ r, )[r� r� � collections�re�stringr� � contextlibr �enumr �typingr r r r r r r r r r r r Z multidictr r r Zyarlr r� r Z base_protocolr Zhelpersr r Zhttp_exceptionsr r r r r r r Z http_writerr r! �logr" Zstreamsr# r$ Ztypedefsr% r& r� r� � ImportError�__all__�set� printabler- r> r? �compiler. r/ r0 rv r+ � namedtupler, rE rF rM r' �ABCr( r) r* r� r� ZHttpRequestParserPyZHttpResponseParserPyZRawRequestMessagePyZRawResponseMessagePyZ _http_parserZHttpRequestParserCZHttpResponseParserCZRawRequestMessageCZRawResponseMessageCrA rA rA rB �<module> s� 8 $ ^ +MA @Z