U
'7`=x
@ sl d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z m
Z
m Z m Z m
Z
mZ d dlmZmZmZ d dlmZ ddlmZ ddlmZ dd lmZmZ dd
l m Z m Z m Z m Z m!Z!m"Z" dd l#m$Z$m%Z% dd l&m'Z' dd
l(m)Z)m*Z* ddl+m,Z, zd dl-Z-dZ.W n e/k
r& dZ.Y nX dZ0e1ej2Z3e4dZ5e4dZ6e4dZ7e8dddddddd d d d g
Z9e8d dd!d"dddd d d g Z:G d#d$ d$eZ;G d%d& d&eZ<G d'd( d(Z=G d)d* d*e j>Z?G d+d, d,e?Z@G d-d. d.e?ZAG d/d0 d0ZBG d1d2 d2ZCe@ZDeAZEe9ZFe:ZGz2esNdd3lHm@Z@mAZAm9Z9m:Z: e@ZIeAZJe9ZKe:ZLW n e/k
rf Y nX dS )4 N)IntEnum)AnyListOptionalTupleTypeUnion) CIMultiDictCIMultiDictProxyistr)URL )hdrs) BaseProtocol)
NO_EXTENSIONSBaseTimerContext)
BadStatusLineContentEncodingErrorContentLengthError
InvalidHeader LineTooLongTransferEncodingError) HttpVersion
HttpVersion10)internal_logger)
EMPTY_PAYLOAD StreamReader)
RawHeadersTF)
HeadersParser
HttpParserHttpRequestParserHttpResponseParserRawRequestMessageRawResponseMessagez [!#$%&'*+\-.^_`|~0-9A-Za-z]+zHTTP/(\d+).(\d+)s' [\x00-\x1F\x7F()<>@,;:\[\]={} \t\\\\\"]r" methodpathversionheaders raw_headersZ should_close compressionupgradechunkedZurlr# codereasonc @ s e Zd ZdZdZdZdZdS )
ParseStater r
N)__name__
__module__ __qualname__
PARSE_NONE PARSE_LENGTH
PARSE_CHUNKEDPARSE_UNTIL_EOF r8 r8 F/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/http_parser.pyr. [ s r. c @ s e Zd ZdZdZdZdZdZdS )
ChunkStater r
r/ r0 N)r1 r2 r3 PARSE_CHUNKED_SIZEPARSE_CHUNKED_CHUNKPARSE_CHUNKED_CHUNK_EOFPARSE_MAYBE_TRAILERSPARSE_TRAILERSr8 r8 r8 r9 r: c s
r: c @ s> e Zd Zd eeeddddZee edef dd d
Z dS ) r N)
max_line_size max_headersmax_field_sizereturnc C s || _ || _|| _d S N)rC rD rE )selfrC rD rE r8 r8 r9 __init__l s zHeadersParser.__init__CIMultiDictProxy[str]linesrF c C s t }g }d}|d }t|}|rz|dd\}}W n tk
rV t|d Y nX |d}| }t|r|t|t|| j krt
d | ddt
| j t
t|t|} |d7 }|| }|o|d dk}
|
rv|g} |
rj| t|7 } | | j kr(t
d | ddt
| j t
| | | |d7 }||k r`|| }|rh|d dk}
qd
}qjqd
| }n0| | j krt
d | ddt
| j t
| | }| d d } | d d }
|| |
|||f q t|t|fS )
Nr
:s zrequest header name {}utf8xmlcharrefreplacer ) zrequest header field {} utf-8surrogateescape)r lensplit
ValueErrorr striplstripHDRREsearchrE r formatdecodestrappendjoinaddr
tuple)rH rL r' r( Z lines_idxlineZ
line_countZbnameZbvalueZ
header_lengthZ continuationZ
bvalue_lstnamevaluer8 r8 r9
parse_headersv sz
zHeadersParser.parse_headers)rA rB rA )
r1 r2 r3 intrI r bytesr r rf r8 r8 r8 r9 r k s
r c @ s e Zd Zdee eej eeeeee ee ee e
ee e e
e
e
dddd Z
ejee ed
d d Zed
ddZddejejejfeeeee eeee e
ef dddZee ede ee
ee e
e
f d
ddZ e
ddddZ dS ) r N rA rB FT)protocollooplimitrC rD rE timerr, r$ readallpayload_exceptionresponse_with_bodyread_until_eofauto_decompressrF c C s || _ || _|| _|| _|| _|| _|| _| | _|
| _| | _ | | _
|
| _ g | _ d| _
d| _d | _d | _|| _|| _t|||| _d S )NrR F)rj rk rC rD rE rm r, r$ rn ro rp rq _lines_tail _upgradedZ_payload_payload_parser_auto_decompress_limitr _headers_parser)rH rj rk rl rC rD rE rm r, r$ rn ro rp rq rr r8 r8 r9 rI s( zHttpParser.__init__rK c C s d S rG r8 )rH rL r8 r8 r9
parse_message s zHttpParser.parse_messagerF c C sz | j d k r | j d | _ nZ| jr0| j| j | jrv| jd dkrP| jd z| | jW S tk
rt Y d S X d S )Nz
rR )rv feed_eofrt rs r_ rz ExceptionrH r8 r8 r9 r} s
zHttpParser.feed_eof
rR )dataSEPEMPTYCONTENT_LENGTH METH_CONNECTSEC_WEBSOCKET_KEY1rF c C s g }| j r | j | d }| _ t|}d} | j}
| |k r| jd kr| js||| } | | krp| jsp| d } q.| | kr| j|| | | d } | jd |krz| | j} W 5 | j X | j
|}
|
d k rz t |
}
W n t
k
r t|Y nX |
dk rt||| j
kr,t|| j| _t| d| j}| jd k sRt|
d k rf|
dksv| jr| jst| j| j|
| jd}t||
| j|| j| j| j| j | j d }|j sv|| _n||krt| j| j|
| jd}d| _t|| j| jd| j d | _nrt| d
d d krr|
d krr| j rrt| j| j|
| jd}t||
| j|| j| jd| j | j d }|j sv|| _nt }|| |f n|| d | _ |}qq.| jd kr| jr| jrtqq.|r| |k r| jrt| jd k stz | j!|| d \}}W n` t"k
rl } z@| j#d k rF| jj$%| #t&| n| jj$%| d}d}W 5 d }~X Y nX |rd} t|}d | _q.q.qq.|r| |k r|| d }n|}|| j|fS )
NrR r r/ r| r$ )rm rk rl )lengthr+ r$ r) r, rn rp rr T)r$ r) rn rr r, d )'rt rU rk rv ru findrs r_ clearrz r' getrg rW r r* getattrr$ rj AssertionErrorr+ r rm rx HttpPayloadParserr) r, rn rp rw donerq r feed_data
BaseExceptionro payload
set_exceptionr^ )rH r r r r r r ZmessagesZdata_lenZ start_posrk posmsgr r$ r Zpayload_parsereofexcr8 r8 r9 r s
zHttpParser.feed_datarJ c C s | j |\}}d}d}d}d}|tj}|r`| } | dkrFd}n| dkrTd}n | dkr`d}|tj}
|
r|
}
|
dkr|
}|tj} | rd| krd}||||||fS ) 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.
NFcloseTz
keep-aliver* )gzipdeflatebrr+ )ry rf r r Z
CONNECTIONlowerZCONTENT_ENCODINGZTRANSFER_ENCODING) rH rL r' r( Z
close_connencodingr* r+ ZconnvencZter8 r8 r9 rf s.
zHttpParser.parse_headers)valrF c C s
|| _ dS )zYSet connection upgraded (to websocket) mode.
:param bool val: new state.
N)ru )rH r r8 r8 r9 set_upgraded s zHttpParser.set_upgraded)NNri rA rB rA NNNFNTFT) r1 r2 r3 r r asyncioZAbstractEventLooprg r r^ boolr r rI abcabstractmethodr rh r rz r} r r r r r r r r rf r r8 r8 r8 r9 r sn
' . )r c @ s$ e Zd ZdZee edddZdS )r zRead request status line. Exception .http_exceptions.BadStatusLine
could be raised in case of any errors in status line.
Returns RawRequestMessage.
rK c C s\ |d dd}z|d d\}}}W n tk
rD t|d Y nX t|| jkrntdt| jtt||d\}}}|d\}} }
t
|st|zB| dr|d d d
d \} } t t
| t
| }
nt|W n tk
r t|Y nX | |\}}}}}}|d kr4|
tkr0d }nd
}t|||
||||||tj||
|d d
S )Nr rS rT r/ Status line is too long#?zHTTP/ .r
TF)r% Z query_stringZfragmentZencoded)r] rV rW r rU rC r r^ partitionMETHREmatch
startswithr rg r~ rf r r" r Zbuild)rH rL rc r$ r% r& Z path_partZ_hash_separatorZ url_fragmentZ_question_mark_separatorZqs_partZn1Zn2 version_or' r( r r) r* r+ r8 r8 r9 rz sf
z HttpRequestParser.parse_messageNr1 r2 r3 __doc__r rh r rz r8 r8 r8 r9 r s r c @ s$ e Zd ZdZee edddZdS )r! zRead response status line and headers.
BadStatusLine could be raised in case of any errors in status line.
Returns RawResponseMessagerK c
C sT |d dd}z|d d\}}W n tk
rB t|d Y nX z|d d\}}W n tk
rp d}Y nX t|| jkrtdt| jtt|t |}|d krt|t
t | dt | d}z t |}W n tk
r t|d Y nX |dkrt||
|\} }
} } }
}| d kr8|tk} t||| | |
| | |
| S ) Nr rS rT r
r r/ i )r] rV rW r rU rC r r^ VERSREr r rg grouprf r r# rX )rH rL rc r& Zstatusr- r r Zstatus_ir' r( r r) r* r+ r8 r8 r9 rz 3 s\
z HttpResponseParser.parse_messageNr r8 r8 r8 r9 r! - s r! c @ sj e Zd Zdeee eee ee ee eeedd
ddZdddd Z de
e
e
e ee
f d d
dZ dS )r NFT)
r r r+ r) r, r$ rn rp rr rF c
C s d| _ tj| _tj| _d| _d| _| | _ d| _
|rH|rH| j rHt ||}
n|}
|shtj| _|
d| _
n||rvtj
| _nn|d k rtj| _|| _ | j dkr|
d| _
n>|r|dkrtj| _n(|dkrtd tj| _|
d| _
|
| _d S )Nr rR FT )ZPUTZPOSTz6Content-Length or Transfer-Encoding header is required)_lengthr. r4 _typer: r< _chunk _chunk_size _chunk_tailrw r
DeflateBufferr} r6 r5 r7 r Zwarningr ) rH r r r+ r) r, r$ rn rp rr Z real_payloadr8 r8 r9 rI n sF
zHttpPayloadParser.__init__r{ c C sF | j tjkr| j n*| j tjkr.tdn| j tjkrBtdd S )Nz2Not enough data for satisfy content length header.z3Not enough data for satisfy transfer length header.) r r. r7 r r} r5 r r6 r r r8 r8 r9 r} s zHttpPayloadParser.feed_eofr ;)chunkr CHUNK_EXTrF c C s: | j tjkr| j}t|}||krT|| | _| j|| | jdkr| j dS n6d| _| j|d | | | j d||d fS n| j tjkr| j r| j | }d| _ |r6| j
t j kr|
|}|dkr|
|d|}|dkr|d | }n |d | }ztt|d} W n@ tk
r\ t|d | dd}
| j|
|
d Y nX ||d d }| dkrt j| _
nt j| _
| | _| j n
|| _ d S | j
t jkr"| j}t|}||kr|| | _| j|| d S d| _| j|d | | ||d }t j| _
| j | j
t jkrb|d d |krX|dd }t j | _
n
|| _ d S | j
t jkr|d d } | |kr| j d|dd fS | sd S | |d d
kr| | _ d S t j| _
| j
t jkr|
|}|dkr||d d }t j| _
q|| _ d S qn | j tjkr6| j|t| d S ) Nr )TrR TrR asciirT r/ )FrR r
) r r. r5 r rU r r r} r6 r r r: r< r rg rh rW r r] r r? r= r begin_http_chunk_receivingr> end_http_chunk_receivingr@ r7 ) rH r r r ZrequiredZ chunk_lenr iZsize_bsizer headr8 r8 r9 r s
zHttpPayloadParser.feed_data)NFNNNFTT)r r )
r1 r2 r3 r r rg r r^ rI r} rh r r r8 r8 r8 r9 r m s: 8
r c @ sr e Zd ZdZeee ddddZeddddZ e
e dd d
d Z dd d
dZ
dd ddZdd ddZdS )r zDDeflateStream decompress stream and feed data into specified stream.N)outr rF c C sb || _ d| _|| _d| _|dkr8ts,tdt | _n&|dkrJdt j
nt j
}t j |d| _d S )Nr Fr zGCan not decode content-encoding: brotli (br). Please install `brotlipy`r r Zwbits) r r r _started_decoding
HAS_BROTLIr brotliZ Decompressor decompressorzlib MAX_WBITS
decompressobj)rH r r Z zlib_moder8 r8 r9 rI / s zDeflateBuffer.__init__)r rF c C s | j | d S rG )r r )rH r r8 r8 r9 r @ s zDeflateBuffer.set_exception)r r rF c C s |sd S | j |7 _ | jsH| jdkrH|d d@ dkrHtjtj d| _z| j|}W n" tk
rz t d| j Y nX d| _|r| j
|t | d S )Nr r r z#Can not decode content-encoding: %sT)
r r r r r r r
decompressr~ r r r rU )rH r r r8 r8 r9 r C s&
zDeflateBuffer.feed_datar{ c C sR | j }|s| jdkrD| j|t| | jdkrD| j jsDtd| j d S )Nr r )
r flushr r r rU r r r r} )rH r r8 r8 r9 r} a s
zDeflateBuffer.feed_eofc C s | j d S rG )r r r r8 r8 r9 r k s z(DeflateBuffer.begin_http_chunk_receivingc C s | j d S rG )r r r r8 r8 r9 r n s z&DeflateBuffer.end_http_chunk_receiving)r1 r2 r3 r r r r^ rI r r rh rg r r} r r r8 r8 r8 r9 r , s
r )r r! r" r# )Mr r collectionsrestringr enumr typingr r r r r r Z multidictr r
r Zyarlr r r Z
base_protocolr Zhelpersr r Zhttp_exceptionsr r r r r r Z http_writerr r logr Zstreamsr r Ztypedefsr r r ImportError__all__setZ printableZASCIISETcompiler r rZ
namedtupler" r# r. r: r ABCr r r! r r ZHttpRequestParserPyZHttpResponseParserPyZRawRequestMessagePyZRawResponseMessagePyZ _http_parserZHttpRequestParserCZHttpResponseParserCZRawRequestMessageCZRawResponseMessageCr8 r8 r8 r9 <module> s
^ K@ @F