U uc<@sdZddlmZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZdd lmZmZmZmZmZd d ZGd d d eZGdddeZGdddZddZddZedZddZdS)zwHTTP Request Parser This server uses asyncore to accept connections and do initial processing but threads to do work. )BytesION)parse)unquote_to_bytes)OverflowableBuffer)ChunkedReceiverFixedStreamReceiver)HEADER_FIELD_RE ONLY_DIGIT_RE) BadRequestRequestEntityTooLargeRequestHeaderFieldsTooLargeServerNotImplementedfind_double_newlinecCst|dS)Nlatin-1)rdecode)Z bytestringr3/tmp/pip-unpacked-wheel-yalm6hg1/waitress/parser.pyunquote_bytes_to_wsgi$src@s eZdZdS) ParsingErrorN__name__ __module__ __qualname__rrrrr(src@s eZdZdS)TransferEncodingNotImplementedNrrrrrr,src@sleZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dZdZdZddZd d Zd d Zd dZddZdS)HTTPRequestParserzA structure that collects the HTTP request. Once the stream is completed, the instance is passed to a server task constructor. FrN1.0cCsi|_||_dS)z/ adj is an Adjustments object. N)headersadj)selfrrrr__init__HszHTTPRequestParser.__init__c Cs4|jr dSt|}|j}|dkr|jj}|j|}t|}d}|dkr`||_|t||}n|j|7_|}|j|kr|dt d||_ d|_|S|dkr|d|}| }|sd|_ d|_nz||Wnrt k r} zt| jd|_ d|_W5d} ~ XYntk rN} zt| jd|_ d|_W5d} ~ XYnHX|jdkrbd|_|jdkr|jj} |j| krtd| |_ d|_d|_|S||_|S||}|j|7_|jj} |j| krtd| |_ d|_n@|j r|j |_ d|_n(|jr,d|_|jr,t||jd<|SdS)z Receives the HTTP stream for one request. Returns the number of bytes consumed. Sets the completed flag once both the header and the body have been received. rNsGET / HTTP/1.0 zexceeds max_header of %sTzexceeds max_body of %sCONTENT_LENGTH) completedlenbody_rcvrZmax_request_header_size header_plusrheader_bytes_received parse_headerr errorlstripemptyrr argsrr content_lengthZmax_request_body_sizer headers_finishedreceivedbody_bytes_receivedchunkedstr__len__r) rdataZdatalenbrZ max_headersindexconsumedr%eZmax_bodyrrrr.Qs|           zHTTPRequestParser.receivedc Cs|d}|dkr4|d|}||dd}ntdd|ksLd|krTtd||_t|}|j}|D]}t|}|std |d d \}} d |krql| d } | dd  d} z|| d|  d7<Wqlt k r| d|| <YqlXqlt|\} } } | d|_| d} | d} | |_| |_t| \|_|_|_|_|_|jj|_|dd}| dkr|dkrd|_| dkrF|dd}dd|dD}|D]}|dkrtdq|r |ddkr d|_ t!|jj"}t#||_$n|rtd|dd}|d k|_%|d!krFd|_|j s|d"d#}t&|'dsttd$t(|}||_)|dkrt!|jj"}t*|||_$dS)%zp Parses the header_plus block of text (the headers plus the first line of the request).  rNzHTTP message header invalid  z#Bare CR or LF found in HTTP messagezInvalid headernamevalue_s -rs, Z CONNECTIONrz keep-aliveTz1.1ZTRANSFER_ENCODINGcSsg|]}|r|dqS)z )striplower).0encodingrrr sz2HTTPRequestParser.parse_header..,>r0z-Transfer-Encoding requested is not supported.r0ZEXPECTz 100-continuecloser!0zContent-Length is invalid)+findrstripr first_lineget_header_linesrrmatchgrouprBupperreplacerKeyErrorcrack_first_line request_uricommandversion split_uri proxy_schemeZ proxy_netlocpathqueryfragmentrZ url_schemegetrCconnection_closepopsplitrr0rZinbuf_overflowrr$expect_continuer encodeintr,r)rr%r6rMheaderlinesrlinekeyr>key1rVurirW connectionte encodingsrEbufexpectZclrrrr's                  zHTTPRequestParser.parse_headercCs |j}|dk r|StSdSN)r$getfilerrr$rrrget_body_streamKsz!HTTPRequestParser.get_body_streamcCs|j}|dk r|dSro)r$ZgetbufrIrqrrrrISszHTTPRequestParser.close)rrr__doc__r"r*rar-r%r0r,r&r/r$rWr(r^r r.r'rrrIrrrrr0s( wrcCsd}}}}}|dddkrZ|}d|kr@|dd\}}d|kr|dd\}}n6zt|\}}}}}Wntk rtdYnX|d|dt||d|dfS) Nrr:s//#?zBad URIr)r`rurlsplit UnicodeErrorrrr)rischemenetlocrZr[r\rrrrXZs" rXcCsg}|d}|D]h}|sqd|ks,d|kr>tdt|d|drp|s^tdt|d|d|7<q||q|S) zL Splits the header into lines, putting multi-line headers together. r9r;r<z'Bare CR or LF found in header line "%s"r)  zMalformed header line "%s"rH)r`rr1 startswithappend)rdrrerfrrrrN~s    rNsL([^ ]+) ((?:[^ :?#]+://[^ ?#/]*(?:[0-9]{1,5})?)?[^ ]+)(( HTTP/([0-9.]+))$|$)cCst|}|dk rx|t|krx|dr8|d}nd}|d}||krdtdt|d|d}|||fSdSdS) NrruzMalformed HTTP method "%s"rr:)rrr) first_line_rerOendr#rPrQrr1)rfmrWmethodrirrrrTs      rT) rsiorreurllibr urllib.parserZwaitress.buffersrZwaitress.receiverrrZwaitress.rfc7230rr Zwaitress.utilitiesr r r r rr ExceptionrrrrXrNcompilerrTrrrrs(     ,$