o
    Dh`8                  
   @   s  d dl Z ddlT ddlmZ ddlmZmZmZmZm	Z	 ddl
mZ ddlT ddlmZ ddlmZmZmZmZmZmZmZ d	d
lmZ eefeddddgddfeefeddgdddfeefedg dddfeefeddgdddfeefedg dddfgZ dd Z!dd Z"dd  Z#d!d" Z$d#d$ Z%d%d& Z&d'd( Z'd)d* Z(d+d, Z)d-d. Z*d/d0 Z+dLd2d3Z,d4d5 Z-d6d7 Z.d8d9 Z/d:d; Z0d<d= Z1d>d? Z2d@dA Z3dBdC Z4dDdE Z5dFdG Z6dHdI Z7dJdK Z8dS )M    N   )*)normalize_and_validate)_obsolete_line_foldChunkedReaderContentLengthReaderHttp10ReaderREADERSReceiveBuffer)LocalProtocolError)ChunkedWriterContentLengthWriterHttp10Writerwrite_any_responsewrite_headerswrite_requestWRITERS   )normalize_data_eventsGETz/aHostfoo
Connectionclosemethodtargetheaderss1   GET /a HTTP/1.1
host: foo
connection: close

      OK)status_coder    reasons&   HTTP/1.1 200 OK
connection: close

s   HTTP/1.1 200 OK

e   )Upgrade	websockets   Upgrades,   HTTP/1.1 101 Upgrade
upgrade: websocket

s   HTTP/1.1 101 Upgrade

c                 C   s   g }| ||j  d|S )N    )appendjoin)writerobjgot_list r.   L/var/www/html/env_mimamsha/lib/python3.10/site-packages/h11/tests/test_io.pydowrite=   s   
r0   c                 C   s   t | |}||ksJ d S N)r0   )r+   r,   expectedgotr.   r.   r/   twC   s   
r4   c                 C   s   t  }|| 7 }|S r1   r
   )databufr.   r.   r/   makebufH   s   r7   c                    s    fdd}t |}|| | |rJ t }tt|D ]}| |d u s'J ||||d  7 }q|| | t |}|d7 }|| | t|dksNJ d S )Nc                    sP   |  ksJ t | dg D ]\}}t|| t|tu sJ t|tu s%J qd S )Nr    )getattrprinttypebytes)r3   namevaluer2   r.   r/   checkO   s   
ztr.<locals>.checkr      trailing)r7   r   rangelenr;   )readerr5   r2   r?   r6   ir.   r>   r/   trN   s   
rE   c                  C   s,   t D ]\\} }}}tt| |f || qd S r1   )SIMPLE_CASESr4   r   rolestateeventbinaryr.   r.   r/   test_writers_simplek      rL   c                  C   s,   t D ]\\} }}}tt| |f || qd S r1   )rF   rE   r	   rG   r.   r.   r/   test_readers_simplep   rM   rN   c                	   C   s   t ttddgd t tg d tt t ttddddgd	d
d  W d    n1 s.w   Y  tt t tt	ddgd	dd  W d    d S 1 sPw   Y  d S )Nr   bar)bazquuxs   foo: bar
baz: quux

s   
r   /r   r   1.0r   r   r    http_versionr!   )r#   r    rV   )
r4   r   r   pytestraisesr   r   Requestr   Responser.   r.   r.   r/   test_writers_unusualu   s6   
"r[   c                	   C   sL  t tttf dtdddgdd t tttf dtddg dd t tttf dtd	dgdd
d t tttf dtd	dgdd
d t tttf dtd	dgdd
d t tttf dtd	dgdd
d t tttf dtd	dgddd t tttf dtddg dd t	t
 t tttf dd  W d    n1 sw   Y  t	t
 t tttf dd  W d    n1 sw   Y  t	t
 t tttf dd  W d    n1 sw   Y  t	t
 t tttf dd  W d    n1 sw   Y  t	t
 t tttf dd  W d    d S 1 sw   Y  d S )Ns$   HEAD /foo HTTP/1.0
Some: header

HEAD/fooSomeheaderrT   rU   s   HEAD /foo HTTP/1.0

s!   HTTP/1.0 200 OK
Some: header

r!   r"   )r#   r    rV   r$   s$   HTTP/1.0 200 OK
Foo: a a a a a 

)Fooz	a a a a as   HTTP/1.0 200 OK
Foo:

)ra    s   HTTP/1.0 200 OK
Foo: 	 	 

s   HTTP/1.0 200
Foo: bar

)ra   rP   r(   s   HEAD /foo HTTP/1.1
Host: example.com
Some: multi-line
 header
	nonsense
    	   		I guess
Connection: close
More-nonsense: in the
    last header  

)r   zexample.com)r_   z"multi-line header nonsense I guessr   )zMore-nonsensezin the last headerr   s&   HEAD /foo HTTP/1.1
  folded: line

s#   HEAD /foo HTTP/1.1
foo  : line

s"   HEAD /foo HTTP/1.1
foo	: line

s   HEAD /foo HTTP/1.1
: line

)rE   r	   CLIENTIDLErY   SERVERSEND_RESPONSErZ   rW   rX   r   r.   r.   r.   r/   test_readers_unusual   s   







	

	
	



$rh   c                   C   s&   t tg ddtddgksJ d S )N)   aaas   bbbs     ccc   dddri   s   bbb cccrj   )listr   	bytearrayr.   r.   r.   r/   test__obsolete_line_fold_bytes  s
   rm   c                 c   sJ    	 | |}|d u rn|V  t |tu rnq|r#|rJ |  V  d S d S r1   )r:   EndOfMessageread_eof)rC   r6   do_eofrJ   r.   r.   r/   _run_reader_iter  s   	rq   c                  G   s   t t|  }t|S r1   )rk   rq   r   )argseventsr.   r.   r/   _run_reader+  s   rt   Fc           	      C   s   t d t|}t|  |||ksJ t d |  }t }g }tt|D ]}|t||d7 }||||d  7 }q%|t|||7 }t||ksJJ tdd |D }|rh|sjt|d }t|  |d|kslJ d S d S d S )NzTest 1zTest 2Fr   c                 s   s    | ]	}t |tu V  qd S r1   )r:   rn   ).0rJ   r.   r.   r/   	<genexpr>A  s    z t_body_reader.<locals>.<genexpr>r@   )r9   r7   rt   r   rA   rB   r   any)	thunkr5   r2   rp   r6   rC   rs   rD   is_completer.   r.   r/   t_body_reader0  s$   rz   c                   C   s4   t dd dt g t dd dtddt g d S )Nc                   S      t dS )Nr   r   r.   r.   r.   r/   <lambda>H      z*test_ContentLengthReader.<locals>.<lambda>r(   c                   S   r{   )N
   r|   r.   r.   r.   r/   r}   K  r~   s
   0123456789r5   )rz   rn   Datar.   r.   r.   r/   test_ContentLengthReaderG  s   r   c                   C   sL   t tdt gdd t tdtddgdd t tdtddt gdd d S )Nr(   T)rp   s   asdfr   F)rz   r   rn   r   r.   r.   r.   r/   test_Http10ReaderQ  s
   
r   c                   C   s  t tdt g t tdtdgdg t tdtddtdgdg t tdtddt g t td	td
dt g tt t tdtddg W d    n1 sTw   Y  tt t tdd  W d    n1 sow   Y  t tdtddt g d S )N   0

s   0
Some: header

r^   r    s3   5
01234
10
0123456789abcdef
0
Some: header

s   012340123456789abcdefr   s%   5
01234
10
0123456789abcdef
0

s   aA
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
0

s   xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxsi   9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999
xxxs   xxxs   10 
xxxsG   5; hello=there
xxxxx
0; random="junk"; some=more; canbe=lonnnnngg

s   xxxxx)rz   r   rn   r   rW   rX   r   r.   r.   r.   r/   test_ChunkedReaderY  sD   
r   c                  C   s  t d} t| tdddksJ t| tdddksJ t| t dks&J t d} tt t| tdd W d    n1 sBw   Y  t d} t| tdd tt t| tdd W d    n1 skw   Y  t d} t| tdd tt t| t  W d    n1 sw   Y  t d} t| tdddk t| tdddk tt t| tdgd	 W d    d S 1 sw   Y  d S )
N   s   123r   s   45r(   s   123456s   456Etagasdfr   )r   r0   r   rn   rW   rX   r   wr.   r.   r/   test_ContentLengthWriter  s0   "r   c                  C   s   t  } t| tdddksJ t| tdddksJ t| tdddks'J t| t dks1J t| tdd	gd
dks?J d S )Nri   r   s   3
aaa
s   aaaaaaaaaaaaaaaaaaaas   14
aaaaaaaaaaaaaaaaaaaa
r(   r   r   )abr   s   0
etag: asdf
a: b

)r   r0   r   rn   r   r.   r.   r/   test_ChunkedWriter  s   r   c                  C   st   t  } t| tdddksJ t| t dksJ tt t| tdgd W d    d S 1 s3w   Y  d S )Ns   1234r   r(   r   r   )r   r0   r   rn   rW   rX   r   r   r.   r.   r/   test_Http10Writer  s   "r   c                   C   D   t t ttttf dd  W d    d S 1 sw   Y  d S )Ns   HTTP/1.0 200 OK xxxx

)rW   rX   r   rE   r	   rf   rg   r.   r.   r.   r/   &test_reject_garbage_after_request_line  s   "r   c                   C   r   )Ns&   HEAD /foo HTTP/1.1 xxxxxx
Host: a

rW   rX   r   rE   r	   rd   re   r.   r.   r.   r/   'test_reject_garbage_after_response_line     
"r   c                   C   r   )Ns%   HEAD /foo HTTP/1.1
Host: foo bar

r   r.   r.   r.   r/   "test_reject_garbage_in_header_line  r   r   c               	   C   sh   dD ]/} t d}||  |d tt tttt	f |d  W d    n1 s,w   Y  qd S )Ns     s   HEAD /s    HTTP/1.1
Host: foobar

)
rl   r)   extendrW   rX   r   rE   r	   rd   re   )bad_charmessager.   r.   r/   test_reject_non_vchar_in_path  s   

r   c                   C   s&   t tttf dtddddgd d S )Ns\   HEAD /foo HTTP/1.1
Host: foo
Set-Cookie: ___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900

r\   r]   r   )z
Set-Cookiez-___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900r   )rE   r	   rd   re   rY   r.   r.   r.   r/   "test_allow_some_garbage_in_cookies  s   
r   c                   C   s   t ttddgd d S )NrO   rc   s   host: example.com
foo: bar

)r4   r   r   r.   r.   r.   r/   test_host_comes_first  s
   
r   )F)9rW   _events_headersr   _readersr   r   r   r   r	   _receivebufferr   _state_utilr   _writersr   r   r   r   r   r   r   helpersr   rd   re   rY   rf   rg   rZ   InformationalResponserF   r0   r4   r7   rE   rL   rN   r[   rh   rm   rq   rt   rz   r   r   r   r   r   r   r   r   r   r   r   r   r.   r.   r.   r/   <module>   sx    $	
#{

7			
