o 4Zc< @sddlZddlTddlmZmZddlmZmZmZm Z m Z ddl m Z ddl TddlmZddlmZmZmZmZmZmZmZd d lmZeefed d d dgddfeefeddgdddfeefedgdddfeefe ddgdddfeefe dgdddfgZ!ddZ"ddZ#dd Z$d!d"Z%d#d$Z&d%d&Z'd'd(Z(d)d*Z)d+d,Z*d-d.Z+d/d0Z,dLd2d3Z-d4d5Z.d6d7Z/d8d9Z0d:d;Z1dd?Z3d@dAZ4dBdCZ5dDdEZ6dFdGZ7dHdIZ8dJdKZ9dS)MN)*)Headersnormalize_and_validate)_obsolete_line_fold ChunkedReaderContentLengthReader Http10ReaderREADERS ReceiveBuffer)LocalProtocolError) ChunkedWriterContentLengthWriter Http10Writerwrite_any_response write_headers write_requestWRITERS)normalize_data_eventsGETz/aHostfoo Connectionclosemethodtargetheaderss1GET /a HTTP/1.1 Host: foo Connection: close OK) status_coder!reasons&HTTP/1.1 200 OK Connection: close sHTTP/1.1 200 OK e)Upgrade websocketsUpgrades,HTTP/1.1 101 Upgrade Upgrade: websocket sHTTP/1.1 101 Upgrade cCsg}|||jd|S)N)appendjoin)writerobjgot_listr/G/var/www/chikooza/env/lib/python3.10/site-packages/h11/tests/test_io.pydowrite=s  r1cCst||}||ks JdSN)r1)r,r-expectedgotr/r/r0twCs r5cCst}||7}|Sr2r )databufr/r/r0makebufHsr8csfdd}t|}||||rJt}tt|D]}||dus'J||||d7}q|||t|}|d7}|||t|dksNJdS)NcsP|ksJt|dgD]\}}t||t|tusJt|tus%Jq dS)Nr!)getattrprinttypebytes)r4namevaluer3r/r0checkOs  ztr..checkrtrailing)r8r rangelenr<)readerr6r3r@r7ir/r?r0trNs    rFcCs,tD]\\}}}}tt||f||qdSr2) SIMPLE_CASESr5rrolestateeventbinaryr/r/r0test_writers_simplekrMcCs,tD]\\}}}}tt||f||qdSr2)rGrFr rHr/r/r0test_readers_simpleprNrOc Cstttddgdtttgdtttttddddgd d dWdn1s0wYtttt t d dgd d dWddS1sRwYdS) Nrbar)bazquuxsfoo: bar baz: quux s r/rr1.0rr r! http_versionr")r$r!rW) r5rrrpytestraisesr rRequestrResponser/r/r/r0test_writers_unusualus6   "r\c Csttttfdtdddgddttttfdtddgddttttfdtd dgdd d ttttfd td d gdd d ttttfdtd dgdd d ttttfdtd dgdd d ttttfdtd dgddd ttttfdtd dgddd ttttfdtd ddgddd ttttfdtd ddgddd ttttfdtddgddt t ttttfddWdn1swYt t ttttfd dWdn1swYt t ttttfd!dWdn 1swYt t ttttfd!dWdn 1s5wYt t ttttfd"dWddS1sVwYdS)#Ns$HEAD /foo HTTP/1.0 Some: header HEAD/fooSomeheaderrUrVsHEAD /foo HTTP/1.0 s!HTTP/1.0 200 OK Some: header r"r#)r$r!rWr%s$HTTP/1.0 200 OK Foo: a a a a a )Fooz a a a a asHTTP/1.0 200 OK Foo: )rbsHTTP/1.0 200 OK Foo: sHTTP/1.0 200 Foo: bar )rbrQr)s#HTTP/1.1 200 OK SomeHeader: val ) SomeHeadervalz1.1OKs5HTTP/1.1 200 OK SomeHeader1: val1 SomeHeader2: val2 ) SomeHeader1val1) SomeHeader2val2s7HTTP/1.1 200 OK SomeHeader1: val1 SomeHeader2: val2 sHEAD /foo HTTP/1.1 Host: example.com Some: multi-line header nonsense I guess Connection: close More-nonsense: in the last header )rz example.com)r`z"multi-line header nonsense I guessr)z More-nonsensezin the last headerrs&HEAD /foo HTTP/1.1 folded: line s#HEAD /foo HTTP/1.1 foo : line s"HEAD /foo HTTP/1.1 foo : line sHEAD /foo HTTP/1.1 : line ) rFr CLIENTIDLErZSERVER SEND_RESPONSEr[rXrYr r/r/r/r0test_readers_unusuals                        $rpcCs&ttgddtddgksJdS)N)aaasbbbs cccdddrqsbbb cccrr)listr bytearrayr/r/r/r0test__obsolete_line_fold_bytes4s ruccsJ ||}|dur n |Vt|turnq|r#|rJ|VdSdSr2)r; EndOfMessageread_eof)rDr7do_eofrKr/r/r0_run_reader_iterAs  rycGstt|}t|Sr2)rsryr)argseventsr/r/r0 _run_readerPs r|Fc Cstdt|}t||||ksJtd|}t}g}tt|D]}|t||d7}||||d7}q%|t|||7}t||ksJJtdd|D}|rh|sjt|d}t||d|kslJdSdSdS)NzTest 1zTest 2Frcss|] }t|tuVqdSr2)r;rv).0rKr/r/r0 fsz t_body_reader..rA)r:r8r|r rBrCrany) thunkr6r3rxr7rDr{rE is_completer/r/r0 t_body_readerUs$ rcCs4tdddtgtdddtddtgdS)NcStdS)Nrrr/r/r/r0mz*test_ContentLengthReader..r)cSr)N rr/r/r/r0rprs 0123456789r6)rrvDatar/r/r/r0test_ContentLengthReaderls rcCsLttdtgddttdtddgddttdtddtgdddS)Nr)T)rxsasdfrF)rr rvrr/r/r/r0test_Http10Readervs  rcCsttdtgttdtdgdgttdtddtdgdgttdtddtgttd td dtgttttd td dgWdn1sTwYttttd dWdn1sowYttdtddtgdS)N0 s0 Some: header r_r!s35 01234 10 0123456789abcdef 0 Some: header s012340123456789abcdefrs%5 01234 10 0123456789abcdef 0 saA xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx 0 sxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxsi9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999 xxxsxxxs10 xxxsG5; hello=there xxxxx 0; random="junk"; some=more; canbe=lonnnnngg sxxxxx)rrrvrrXrYr r/r/r/r0test_ChunkedReader~sD     rcCstd}t|tdddksJt|tdddksJt|tdks&Jtd}ttt|tddWdn1sBwYtd}t|tddttt|tddWdn1skwYtd}t|tddttt|tWdn1swYtd}t|tdddkt|tdddkttt|tdgd WddS1swYdS) Ns123rs45r)s123456s456Etagasdfr)rr1rrvrXrYr wr/r/r0test_ContentLengthWriters0    "rcCst}t|tdddksJt|tdddksJt|tdddks'Jt|tdks1Jt|tdd gd d ks?JdS) Nrqrs3 aaa saaaaaaaaaaaaaaaaaaaas14 aaaaaaaaaaaaaaaaaaaa r)rr)abrs0 Etag: asdf a: b )rr1rrvrr/r/r0test_ChunkedWriters rcCstt}t|tdddksJt|tdksJttt|tdgdWddS1s3wYdS)Ns1234rr)rr)rr1rrvrXrYr rr/r/r0test_Http10Writers  "rcCDttttttfddWddS1swYdS)NsHTTP/1.0 200 OKxxxx )rXrYr rFr rnror/r/r/r0&test_reject_garbage_after_request_lines "rcCr)Ns&HEAD /foo HTTP/1.1 xxxxxx Host: a rXrYr rFr rlrmr/r/r/r0'test_reject_garbage_after_response_line  "rcCr)Ns%HEAD /foo HTTP/1.1 Host: foobar rr/r/r/r0"test_reject_garbage_in_header_linerrc CshdD]/}td}|||dtttttt f|dWdn1s,wYqdS)Ns sHEAD /s HTTP/1.1 Host: foobar ) rtr*extendrXrYr rFr rlrm)bad_charmessager/r/r0test_reject_non_vchar_in_paths   rcCs&ttttfdtddddgddS)Ns\HEAD /foo HTTP/1.1 Host: foo Set-Cookie: ___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900 r]r^r)z Set-Cookiez-___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900r)rFr rlrmrZr/r/r/r0"test_allow_some_garbage_in_cookies s rcCstttddgddS)NrPrksHost: example.com foo: bar )r5rrr/r/r/r0test_host_comes_firsts  r)F):rX_events_headersrr_readersrrrr r _receivebufferr _state_utilr _writersrrrrrrrhelpersrrlrmrZrnror[InformationalResponserGr1r5r8rFrMrOr\rpruryr|rrrrrrrrrrrrrr/r/r/r0sz  $    #!   7