File: //usr/local/lib/python3.10/dist-packages/h11/__pycache__/_readers.cpython-310.pyc
o
���g� � @ s\ U d dl Z d dlmZmZmZmZmZmZmZm Z m
Z
ddl m Z m
Z
mZmZ ddlmZmZmZmZmZ ddlmZ ddlmZmZmZm Z m Z m Z m Z m Z ddl!m"Z"m#Z#m$Z$m%Z% d gZ&e �'e
�(d
��Z)e �'d �Z*d ee+ d
ee+ fdd�Z,d ee+ d
eee+e+f fdd�Z-e �'e�(d
��Z.ded
ee fdd�Z/e �'e�(d
��Z0ded
e
eedf fdd�Z1G dd� d�Z2e �'e �(d
��Z3G dd� d�Z4G dd � d �Z5ded
dfd d �Z6ee
e e$ ee e$ e e$ f f e
ed ef ee7ed ef f f f Z8ee fe/e e fe1e e fe1eefe6ee fe6eefe6e efe6e e fe6e efe6e e4e2e5d �i
Z9e8e:d < dS )!� N) �Any�Callable�Dict�Iterable�NoReturn�Optional�Tuple�Type�Union� )� chunk_header� header_field� request_line� status_line)�Data� EndOfMessage�InformationalResponse�Request�Response)�
ReceiveBuffer)�CLIENT�CLOSED�DONE�IDLE�
MUST_CLOSE� SEND_BODY�
SEND_RESPONSE�SERVER)�LocalProtocolError�RemoteProtocolError�Sentinel�validate�READERS�asciis [ \t]+�lines�returnc c s� � t | �}d }|D ]2}t�|�}|r2|d u rtd��t|t�s#t|�}|d7 }|||�� d � 7 }q |d ur9|V |}q |d urE|V d S d S )Nz%continuation line at start of headers� )�iter� obs_fold_re�matchr �
isinstance� bytearray�end)r$ �it�last�liner) � r0 �7/usr/local/lib/python3.10/dist-packages/h11/_readers.py�_obsolete_line_fold+ s$ �
�r2 c c s4 � t | �D ]}tt|d|�}|d |d fV qd S )Nzillegal header line: {!r}�
field_name� field_value)r2 r! �header_field_re)r$ r/ �matchesr0 r0 r1 �_decode_header_lines@ s
� �r7 �bufc C sj | � � }|d u r| �� rtd��d S |std��tt|d d|d �}tdtt|dd � ��dd�|��S ) N�illegal request linezno request line receivedr zillegal request line: {!r}r T)�headers�_parsedr0 )�maybe_extract_lines�+is_next_line_obviously_invalid_request_liner r! �request_line_rer �listr7 )r8 r$ r6 r0 r0 r1 �maybe_read_from_IDLE_clientK s ���r@ c C s� | � � }|d u r| �� rtd��d S |std��tt|d d|d �}|d d u r+dn|d }|d d u r7dn|d }t|d �}|d
k rGtnt}|tt |d d � ��d |||d
�S )Nr9 zno response line receivedr zillegal status line: {!r}� http_versions 1.1�reason� � status_code�� r T)r: r; rD rB rA )
r< r= r r! �status_line_re�intr r r? r7 )r8 r$ r6 rA rB rD �class_r0 r0 r1 �$maybe_read_from_SEND_RESPONSE_server^ s* � ��rI c @ sH e Zd Zdeddfdd�Zdedeeedf fdd�Z de
fd d
�Z dS ) �ContentLengthReader�lengthr% Nc C s || _ || _d S �N)�_length�
_remaining)�selfrK r0 r0 r1 �__init__{ s
z ContentLengthReader.__init__r8 c C sD | j dkrt� S |�| j �}|d u rd S | j t|�8 _ t|d�S )Nr ��data)rN r �maybe_extract_at_most�lenr �rO r8 rR r0 r0 r1 �__call__ s
z ContentLengthReader.__call__c C s t d�| j| j | j���)Nz]peer closed connection without sending complete message body (received {} bytes, expected {}))r �formatrM rN �rO r0 r0 r1 �read_eof� s ��z ContentLengthReader.read_eof) �__name__�
__module__� __qualname__rG rP r r
r r rV r rY r0 r0 r0 r1 rJ z s rJ c @ s@ e Zd Zd
dd�Zdedeeedf fdd�Zde fdd �Z
dS ) �
ChunkedReaderr% Nc C s d| _ d| _d| _d S )Nr F)�_bytes_in_chunk�_bytes_to_discard�_reading_trailerrX r0 r0 r1 rP � s
zChunkedReader.__init__r8 c C s< | j r|�� }|d u r
d S ttt|��d�S | jdkr7|�| j�}|d u r'd S | jt|�8 _| jdkr7d S | jdks>J �| jdkrl|� � }|d u rMd S t
t |d|�}t |d dd�| _| jdkrid| _ | |�S d}nd}| jdksuJ �|�| j�}|d u r�d S | jt|�8 _| jdkr�d | _d}nd}t
|||d
�S ) N)r: r zillegal chunk header: {!r}�
chunk_size� )�baseTF� )rR � chunk_start� chunk_end)r` r< r r? r7 r_ rS rT r^ �maybe_extract_next_liner! �chunk_header_rerG r )rO r8 r$ rR r r6 re rf r0 r0 r1 rV � sN
�
zChunkedReader.__call__c C s t d��)NzVpeer closed connection without sending complete message body (incomplete chunked read))r rX r0 r0 r1 rY � s �zChunkedReader.read_eof)r% N) rZ r[ r\ rP r r
r r rV r rY r0 r0 r0 r1 r] � s
.r] c @ s0 e Zd Zdedee fdd�Zdefdd�ZdS )� Http10Readerr8 r% c C s |� d�}|d u r d S t|d�S )Ni�ɚ;rQ )rS r rU r0 r0 r1 rV � s
zHttp10Reader.__call__c C s t � S rL )r rX r0 r0 r1 rY � s zHttp10Reader.read_eofN) rZ r[ r\ r r r rV r rY r0 r0 r0 r1 ri � s ri c C s | rt d��d S )NzGot data when expecting EOF)r )r8 r0 r0 r1 �expect_nothing� s rj .)�chunkedzcontent-lengthzhttp/1.0);�re�typingr r r r r r r r r
�_abnfr r
r r �_eventsr r r r r �_receivebufferr �_stater r r r r r r r �_utilr r r r! �__all__�compile�encoder5 r( �bytesr2 r7 r> r@ rF rI rJ rh r] ri rj �str� ReadersTyper" �__annotations__r0 r0 r0 r1 �<module> sb
, (
�
��
� > "����