U g@sddlmZddlZddlZddlZddlmZddlmZddgZ e ej dd Z e ej d d Zd d dddZedZedZdddddZdddddZdddddZdd dddZdS)) annotationsN)Headers) SecurityError read_request read_responseZWEBSOCKETS_MAX_NUM_HEADERSZ128ZWEBSOCKETS_MAX_LINE_LENGTHZ8192bytesstr)valuereturncCs |jddS)zG Decode a bytestring for interpolating into an error message. backslashreplace)errors)decode)r r:/tmp/pip-unpacked-wheel-dx_q7dq3/websockets/legacy/http.pydsrs[-!#$%&\'*+.^_`|~0-9a-zA-Z]+s[\x09\x20-\x7e\x80-\xff]*zasyncio.StreamReaderztuple[str, Headers])streamr c szt|IdH}Wn,tk r>}ztd|W5d}~XYnXz|dd\}}}Wn(tk r~tdt|dYnX|dkrtdt||dkrtd t||d d }t|IdH}||fS) a Read an HTTP/1.1 GET request and return ``(path, headers)``. ``path`` isn't URL-decoded or validated in any way. ``path`` and ``headers`` are expected to contain only ASCII characters. Other characters are represented with surrogate escapes. :func:`read_request` doesn't attempt to read the request body because WebSocket handshake requests don't have one. If the request contains a body, it may be read from ``stream`` after this coroutine returns. Args: stream: Input to read the request from. Raises: EOFError: If the connection is closed without a full HTTP request. SecurityError: If the request exceeds a security limit. ValueError: If the request isn't well formatted. Nz1connection closed while reading HTTP request line rzinvalid HTTP request line: sGETzunsupported HTTP method: HTTP/1.1unsupported HTTP version: asciisurrogateescape) read_lineEOFErrorsplit ValueErrorrr read_headers)rZ request_lineexcmethodraw_pathversionpathheadersrrrr-s ztuple[int, str, Headers]c s6zt|IdH}Wn,tk r>}ztd|W5d}~XYnXz|dd\}}}Wn(tk r~tdt|dYnX|dkrtdt|z t|}Wn(tk rtdt|dYnXd |krd ksntd t|t|std t||}t |IdH}|||fS) a Read an HTTP/1.1 response and return ``(status_code, reason, headers)``. ``reason`` and ``headers`` are expected to contain only ASCII characters. Other characters are represented with surrogate escapes. :func:`read_request` doesn't attempt to read the response body because WebSocket handshake responses don't have one. If the response contains a body, it may be read from ``stream`` after this coroutine returns. Args: stream: Input to read the response from. Raises: EOFError: If the connection is closed without a full HTTP response. SecurityError: If the response exceeds a security limit. ValueError: If the response isn't well formatted. Nz0connection closed while reading HTTP status linerrzinvalid HTTP status line: rrzinvalid HTTP status code: dizunsupported HTTP status code: zinvalid HTTP reason phrase: ) rrrrrint _value_re fullmatchrr) rZ status_linerr Zraw_status_codeZ raw_reason status_codereasonr"rrrr^s*  rc st}ttdD]}zt|IdH}Wn,tk rT}ztd|W5d}~XYnX|dkrdq z|dd\}}Wn(tk rtdt|dYnXt |stdt|| d}t |std t|| d }| d d }|||<qt d |S) zo Read HTTP headers from ``stream``. Non-ASCII characters are represented with surrogate escapes. Nz,connection closed while reading HTTP headers:zinvalid HTTP header line: zinvalid HTTP header name: s zinvalid HTTP header value: rrztoo many HTTP headers)rrangeMAX_NUM_HEADERSrrrrr _token_rer&stripr%rr) rr"_linerZraw_name raw_valuenamer rrrrs,       rcs@|IdH}t|tkr"td|ds4td|ddS)z[ Read a single line from ``stream``. CRLF is stripped from the return value. Nz line too longs zline without CRLF)readlinelenMAX_LINE_LENGTHrendswithr)rr1rrrrs   r) __future__rZasyncioosreZdatastructuresr exceptionsr__all__r$environgetr-r7rcompiler.r%rrrrrrrrs    14(