Z_ @ s d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l
m Z d d l m
Z
d d l m Z m Z m Z m Z m Z m Z d d l m Z m Z d d
l m Z d d l m Z m Z y d d l Z d Z Wn e k
rd
Z Yn Xd d d d d f Z e! e j" Z# e j$ d Z% e j$ d Z& e j$ d Z' e j( d d d d d d d d d d d g
Z) e j( d d d d! d d d d d d g Z* Gd" d# d# e Z+ Gd$ d% d% e Z, Gd& d d Z- Gd' d( d( e- Z. Gd) d* d* e- Z/ Gd+ d, d, Z0 Gd- d. d. Z1 e. Z2 e/ Z3 y- d d/ l4 m5 Z5 m6 Z6 e
re5 Z2 e6 Z3 Wn e k
rYn Xd S)0 N)IntEnum) CIMultiDict)URL )hdrs)
NO_EXTENSIONS)
BadStatusLineContentEncodingErrorContentLengthError
InvalidHeader LineTooLongTransferEncodingError) HttpVersion
HttpVersion10)internal_logger)
EMPTY_PAYLOAD StreamReaderTF
HttpParserHttpRequestParserHttpResponseParserRawRequestMessageRawResponseMessagez
[A-Z0-9$-_.]+zHTTP/(\d+).(\d+)s' [\x00-\x1F\x7F()<>@,;:\[\]={} \t\\\\\"]methodpathversionheaders raw_headersZ should_close compressionupgradechunkedZurlcodereasonc @ s( e Z d Z d Z d Z d Z d Z d S)
ParseStater r N)__name__
__module__ __qualname__
PARSE_NONE PARSE_LENGTH
PARSE_CHUNKEDPARSE_UNTIL_EOF r, r, /http_parser.pyr" / s r" c @ s. e Z d Z d Z d Z d Z d Z d Z d S)
ChunkStater r r# r$ N)r% r& r' PARSE_CHUNKED_SIZEPARSE_CHUNKED_CHUNKPARSE_CHUNKED_CHUNK_EOFPARSE_MAYBE_TRAILERSPARSE_TRAILERSr, r, r, r- r. 7 s
r. c @ s e Z d Z d d d d d d d d d d d d d d d
Z d d Z d
d e j e j e j d d
Z d d Z
d S)r Ni i FTc C s | | _ | | _ | | _ | | _ | | _ | | _ | | _ | | _ | | _ |
| _ | | _
| | _ g | _ d | _
d | _ d | _ d | _ |
| _ d S)N F)protocolloop
max_line_size max_headersmax_field_sizetimerr r readallpayload_exceptionresponse_with_bodyread_until_eof_lines_tail _upgradedZ_payload_payload_parser_auto_decompress)selfr6 r7 r8 r9 r: r; r r r< r= r> r? auto_decompressr, r, r- __init__A s$ zHttpParser.__init__c C s | j d k r( | j j d | _ nu | j rD | j j | j | j r | j d d k rp | j j d y | j | j SWn t k
r d SYn Xd S)Nr z
)rC feed_eofrA r@ append
parse_message Exception)rE r, r, r- rJ [ s
zHttpParser.feed_eofs
r5 c " C s g } | j r& | j | d } | _ t | } d } | j }
xx| | k r| j d k r| j r| j | | } | | k r | j r | d } qD | | k r| j j | | | | d } | j d | k rz | j | j } Wd | j j X| j
j | }
|
d k roy t |
}
Wn t
k
rVt | Yn X|
d k rot | | | j
k rt | | j | _ t | d | j } |
d k r|
d k s| j rW| j rWt | j d | j d |
} t | d |
d | j d | d
| j d | j d | j d
| j d | j } | j sn| | _ n| | k rt | j d | j d |
} d | _ t | d | j d
| j d d d | j | _ n t | d d d k rh|
d k rh| j rht | j d | j d |
} t | d |
d | j d | d
| j d | j d d d
| j d | j } | j sn| | _ n t } | j | | f q| | d | _ | } PqD | j d k r| j r| j st PqD | r| | k r| j st y& | j j | | d \ } } Wnx t! k
r} zX | j" d k r^| j j# j$ | j" t% | n | j j# j$ | d } d } WYd d } ~ Xn X| rd } t | } d | _ qD qD PqD W| r| | k r| | d } n | } | | j | f S)Nr5 r r# r r r; r7 lengthr r r r< r> rF Td rI )&rA lenr7 rC rB findr@ rK rL clearr getint
ValueErrorr r getattrr r r r6 r; HttpPayloadParserr r r< r> rD doner? r AssertionError feed_data
BaseExceptionr= payload
set_exceptionstr)rE dataSEPZEMPTYCONTENT_LENGTH METH_CONNECTSEC_WEBSOCKET_KEY1ZmessagesZdata_lenZ start_posr7 posmsgrN r r] Zpayload_parsereofexcr, r, r- r[ l s
zHttpParser.feed_datac
C s" t } g } d } | d } t | } x| r!t | } y | j d d \ } } Wn! t k
r t | d Yn X| j d } t j | r t | | d 7} | | } | o | d d k }
|
r| g } x |
r| t | 7} | | j k r2t d j
| j d d
| j | j | | d 7} | | k rx| | } | r| d d k }
q d } Pq Wd j
| } n6 | | j k rt d j
| j d d
| j | j } | j d d
} | j d d
} | j | | | j | | f q. Wd }
d } d } d } t | } | j t j } | r| j } | d k rd }
n' | d k rd }
n | d k rd } | j t j } | r| j } | d k r| } | j t j } | r d | j k r d } | | |
| | | f S)zParses RFC 5322 headers from a stream.
Line continuations are supported. Returns list of header name
and value pairs. Header name is in upper case.
r :Ns r zrequest header field {}utf8xmlcharrefreplacer5 zutf-8surrogateescapeFcloseTz
keep-aliver gzipdeflatebrr )rj rk )rj rk )rp rq rr )r rQ splitrV r stripHDRREsearchr: r formatdecoderK joinaddtuplerT r Z
CONNECTIONlowerZCONTENT_ENCODINGZTRANSFER_ENCODING)rE linesr r Z lines_idxlineZ
line_countZ
header_lengthZbnameZbvalueZ continuationnamevalueZ
close_connencodingr r connvencZter, r, r-
parse_headers s
zHttpParser.parse_headers) r% r& r' rG rJ r rb rc rd r[ r r, r, r, r- r ? s c @ s" e Z d Z d Z d d Z d S)HttpRequestParserPyzRead request status line. Exception .http_exceptions.BadStatusLine
could be raised in case of any errors in status line.
Returns RawRequestMessage.
c C s t | d | j k r+ t d | j | d j d d } y | j d d \ } } } Wn! t k
r t | d Yn X| j } t j | s t | y_ | j
d r | d d j d d \ } } t t | t | } n t | Wn t
k
r*t | Yn X| j | \ } } }
} } }
|
d k rs| t k rmd
}
n d }
t | | | | | |
| | |
t |
S) Nr zStatus line is too longzutf-8rn r# zHTTP/ .r TF)rQ r8 r rx rs rV r upperMETHREmatch
startswithr rU rM r r r r )rE r} r~ r r r Zn1Zn2r r ro r r r r, r, r- rL _ s4
"
! z!HttpRequestParserPy.parse_messageN)r% r& r' __doc__rL r, r, r, r- r Y s r c @ s" e Z d Z d Z d d Z d S)HttpResponseParserPyzRead response status line and headers.
BadStatusLine could be raised in case of any errors in status line.
Returns RawResponseMessagec
" C s t | d | j k r+ t d | j | d j d d } y | j d d \ } } Wn! t k
r t | d Yn8 Xy | j d d \ } } Wn t k
r d } Yn Xt j | } | d k r t | t t
| j d t
| j d } y t
| } Wn! t k
r?t | d Yn X| d k rXt | | j | \ } } } }
} } | d k r| t
k } t | | | j | | | |
| | S) Nr zStatus line is too longzutf-8rn r rH r# i )rQ r8 r rx rs rV r VERSREr r rU groupr r r rt )
rE r} r~ r statusr! r r r ro r r r r, r, r- rL s8
-
! z"HttpResponseParserPy.parse_messageN)r% r& r' r rL r, r, r, r- r s r c
@ sR e Z d Z d d d d d d d d d d Z d d Z d d d
d Z d S) rX NFTc
C s[ | | _ d | _ t j | _ t j | _ d | _ d | _ | | _
d | _ | rr | rr | j
rr t | | } | s t j | _ | j
d | _ n | r t j | _ n | d k r t j | _ | | _ | j d k rN| j
d | _ nY | r| d k rt j | _ n8 | d k rNt j d t j | _ | j
d | _ | | _ d S)
Nr r5 FT PUTPOSTz6Content-Length or Transfer-Encoding header is required)r r )r] _lengthr" r( _typer. r0 _chunk _chunk_size _chunk_tailrD rY
DeflateBufferrJ r* r) r+ r warning)
rE r] rN r r r r r< r> rF r, r, r- rG s>
zHttpPayloadParser.__init__c C se | j t j k r" | j j n? | j t j k rC t d n | j t j k ra t d d S)Nz2Not enough data for satisfy content length header.z3Not enough data for satisfy transfer length header.) r r" r+ r] rJ r) r
r* r
)rE r, r, r- rJ s zHttpPayloadParser.feed_eofs
;c
C s | j t j k r | j } t | } | | k rv | | | _ | j j | | | j d k r | j j d Sqd | _ | j j | d | | | j j d | | d f SnG| j t j k r| j r | j | } d | _ x| r| j
t j k r)| j
| } | d k r | j
| d | } | d k r[| d | } n | d | } y t t | d } WnA t k
rt | d | } | j j | | d Yn X| | d d } | d k rt j | _
q)t j | _
| | _ | j j n
| | _ d S| j
t j k r| j } t | } | | k r| | | _ | j j | | d Sd | _ | j j | d | | | | d } t j | _
| j j | j
t j k r#| d d | k r| d d } t j | _
n
| | _ d
S| j
t j k rx| d d | k rl| j j d | d d f St j | _
| j
t j k r | j
| } | d k r| | d d } t j | _
q | | _ d Sq Wn+ | j t j k r| j j | t | d S)
Nr Tr5 r# F)Tr5 )FN)FN)FN)FN)FN)r r" r) r rQ r] r[ rJ r* r r r. r0 rR rU bytesrV r
r^ r3 r1 r begin_http_chunk_receivingr2 end_http_chunk_receivingr4 r+ )
rE chunkra Z CHUNK_EXTZrequiredZ chunk_lenre isizerh r, r, r- r[ s
zHttpPayloadParser.feed_data)r% r& r' rG rJ r[ r, r, r, r- rX s
*
rX c @ s^ e Z d Z d Z d d Z d d Z d d Z d d Z d
d Z d d
Z d S)r zDDeflateStream decompress stream and feed data into specified stream.c C s | | _ d | _ | | _ d | _ | d k rT t sB t d t j | _ n8 | d k rm d t j
n t j
} t j d | | _ d S)Nr Frr zGCan not decode content-encoding: brotli (br). Please install `brotlipy`rp r Zwbits) outr r _started_decoding
HAS_BROTLIr brotliZ Decompressor decompressorzlib MAX_WBITS
decompressobj)rE r r Z zlib_moder, r, r- rG a s #zDeflateBuffer.__init__c C s | j j | d S)N)r r^ )rE rh r, r, r- r^ r s zDeflateBuffer.set_exceptionc C s | j | 7_ y | j j | } Wn t k
r | j r | j d k r t j | _ y | j j | } Wq t k
r t d | j Yq Xn t d | j Yn X| r d | _ | j j
| t | d S)Nrq z#Can not decode content-encoding: %sT) r r
decompressrM r r r r r r r[ rQ )rE r r r, r, r- r[ u s
zDeflateBuffer.feed_datac C sv | j j } | s$ | j d k re | j j | t | | j d k re | j j re t d | j j d S)Nr rr rq )
r flushr r r[ rQ r rg r rJ )rE r r, r, r- rJ s zDeflateBuffer.feed_eofc C s | j j d S)N)r r )rE r, r, r- r s z(DeflateBuffer.begin_http_chunk_receivingc C s | j j d S)N)r r )rE r, r, r- r s z&DeflateBuffer.end_http_chunk_receivingN)
r% r& r' r rG r^ r[ rJ r r r, r, r, r- r ^ s
r )HttpRequestParserCHttpResponseParserC)7 collectionsrestringr enumr Z multidictr Zyarlr rH r Zhelpersr Zhttp_exceptionsr r r
r r r
Z http_writerr r logr Zstreamsr r r r ImportError__all__set printableZASCIISETcompiler r ru
namedtupler r r" r. r r r rX r r r Z _http_parserr r r, r, r, r- <module> s` .
01<