U
    g                      @   sR  U d dl Z d dlmZmZmZmZmZmZmZm	Z	m
Z
 ddlmZmZmZmZ ddlmZmZmZmZmZ ddlmZ ddlmZmZmZmZmZmZmZm Z  ddl!m"Z"m#Z#m$Z$m%Z% d	gZ&e 'e(d
Z)e 'dZ*ee+ ee+ dddZ,ee+ eee+e+f  dddZ-e 'e(d
Z.eee dddZ/e 'e(d
Z0ee
eedf dddZ1G dd dZ2e 'e(d
Z3G dd dZ4G dd dZ5eddddZ6ee
e	e$ ee	e$ e	e$ f f e
edef ee7edef f f f Z8eefe/e efe1e efe1eefe6eefe6eefe6e efe6e efe6e efe6ee4e2e5di
Z9e8e:d	< dS )     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 ]d}t|}|rb|d kr2tdt|tsDt|}|d7 }||| d  7 }q|d k	rp|V  |}q|d k	r|V  d S )Nz%continuation line at start of headers    )iterobs_fold_rematchr   
isinstance	bytearrayend)r$   itlastliner)    r0   0/tmp/pip-unpacked-wheel-ib4s2i_e/h11/_readers.py_obsolete_line_fold+   s     

r2   c                 c   s2   t | D ]$}tt|d|}|d |d fV  qd S )Nzillegal header line: {!r}
field_nameZfield_value)r2   r!   header_field_re)r$   r/   matchesr0   r0   r1   _decode_header_lines@   s    r6   bufr%   c                 C   sj   |   }|d kr$|  r tdd S |s0tdtt|d d|d }tf tt|dd  dd|S )Nillegal request linezno request line receivedr   zillegal request line: {!r}r   T)headers_parsed)maybe_extract_lines+is_next_line_obviously_invalid_request_liner   r!   request_line_rer   listr6   )r8   r$   r5   r0   r0   r1   maybe_read_from_IDLE_clientK   s&        r@   c                 C   s   |   }|d kr$|  r tdd S |s0tdtt|d d|d }|d d krVdn|d }|d d krndn|d }t|d	 }|d
k rtnt}|tt	|dd  d|||dS )Nr9   zno response line receivedr   zillegal 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?   r6   )r8   r$   r5   rA   rB   rD   class_r0   r0   r1   $maybe_read_from_SEND_RESPONSE_server^   s*    rI   c                   @   sD   e Zd ZeddddZeeeedf dddZ	e
dd	d
ZdS )ContentLengthReaderN)lengthr%   c                 C   s   || _ || _d S N)_length
_remaining)selfrK   r0   r0   r1   __init__{   s    zContentLengthReader.__init__r7   c                 C   sD   | j dkrt S || j }|d kr(d 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__r%   c                 C   s   t d| j| j | jd S )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   rZ   r0   r0   r0   r1   rJ   z   s   	rJ   c                   @   sB   e Zd ZddddZeeeedf dddZe	ddd	Z
dS )
ChunkedReaderNrW   c                 C   s   d| _ d| _d| _d S )Nr   F)_bytes_in_chunk_bytes_to_discard_reading_trailerrY   r0   r0   r1   rP      s    zChunkedReader.__init__r7   c                 C   s@  | j r,| }|d krd S ttt|dS | jdkrn|| j}|d krNd S |  jt|8  _| jdkrnd S | jdks|t| j	dkr|
 }|d krd S tt|d|}t|d dd| _	| j	dkrd| _ | |S d}nd}| j	dkst|| j	}|d krd S |  j	t|8  _	| j	dkr.d	| _d}nd}t|||d
S )N)r:   r   zillegal chunk header: {!r}
chunk_size   )baseTF   )rR   chunk_start	chunk_end)ra   r<   r   r?   r6   r`   rS   rT   AssertionErrorr_   Zmaybe_extract_next_liner!   chunk_header_rerG   r   )rO   r8   r$   rR   r   r5   rf   rg   r0   r0   r1   rV      sN    




zChunkedReader.__call__c                 C   s   t dd S )NzVpeer closed connection without sending complete message body (incomplete chunked read))r   rY   r0   r0   r1   rZ      s    zChunkedReader.read_eof)r[   r\   r]   rP   r   r
   r   r   rV   r   rZ   r0   r0   r0   r1   r^      s   .r^   c                   @   s.   e Zd Zeee dddZedddZdS )Http10Readerr7   c                 C   s    | d}|d krd S t|dS )Niɚ;rQ   )rS   r   rU   r0   r0   r1   rV      s    
zHttp10Reader.__call__rW   c                 C   s   t  S rL   )r   rY   r0   r0   r1   rZ      s    zHttp10Reader.read_eofN)	r[   r\   r]   r   r   r   rV   r   rZ   r0   r0   r0   r1   rj      s   rj   c                 C   s   | rt dd S )NzGot data when expecting EOF)r   )r8   r0   r0   r1   expect_nothing   s    rk   .)chunkedzcontent-lengthzhttp/1.0);retypingr   r   r   r   r   r   r   r	   r
   Z_abnfr   r   r   r   Z_eventsr   r   r   r   r   Z_receivebufferr   _stater   r   r   r   r   r   r   r   Z_utilr   r   r    r!   __all__compileencoder4   r(   bytesr2   r6   r>   r@   rF   rI   rJ   ri   r^   rj   rk   strZReadersTyper"   __annotations__r0   r0   r0   r1   <module>   sl    ,(

>"         