o _c1@sddlmZddlZddlZddlZddlmZmZmZddl m Z m Z dZ dZ dZdd dZedZedZejGdddZejGdddZdddZdddZdS) ) annotationsN)Callable GeneratorOptional)datastructures exceptionsi ivaluebytesreturnstrcCs |jddS)zG Decode a bytestring for interpolating into an error message. backslashreplace)errors)decode)r rrc@sheZdZUdZded<ded<ded<dZd ed <dZd ed <edddZe dddZ dddZ dS)Responsez WebSocket handshake response. Attributes: status_code: Response code. reason_phrase: Response reason. headers: Response headers. body: Response body, if any. int status_coder reason_phraserrNzOptional[bytes]bodyrrr cCr)NzLResponse.exception is deprecated; use ClientConnection.handshake_exc insteadrrrrrr!r"zResponse.exceptionr#r$ read_exact read_to_eofGenerator[None, None, Response]c csz t|EdH}Wnty}ztd|d}~wwz |dd\}}}Wnty8tdt|dw|dkrFtdt|zt|} Wnty]tdt|dwd | krhd ksrntd t|t|std t||} t |EdH} d | vrt dd | krdksn| dks| dkrd} nHz| d} Wn t yd}Ynwt| }|durz |t EdH} Wn$t ytdt dw|t krtd|d||EdH} || | | | S)a Parse a WebSocket handshake response. This is a generator-based coroutine. The reason phrase and headers are expected to contain only ASCII characters. Other characters are represented with surrogate escapes. Args: read_line: generator-based coroutine that reads a LF-terminated line or raises an exception if there isn't enough data. read_exact: generator-based coroutine that reads the requested bytes or raises an exception if there isn't enough data. read_to_eof: generator-based coroutine that reads until the end of the stream. Raises: EOFError: if the connection is closed without a full HTTP response. SecurityError: if the response exceeds a security limit. LookupError: if the response isn't well formatted. ValueError: if the response isn't well formatted. Nz0connection closed while reading HTTP status liner&r'zinvalid HTTP status line: r(r)zinvalid HTTP status code: dizunsupported HTTP status code: zinvalid HTTP reason phrase: r,r-i0r.zbody too large: over z byteszbody too large: )r/r0r1r2rrH _value_re fullmatchrr3r4KeyErrorMAX_BODY RuntimeErrorr SecurityError)r5r#rLrMZ status_liner6r9Zraw_status_codeZ raw_reasonrIreasonrrKZraw_content_lengthcontent_lengthrrrr:sj       &     zResponse.parser cCs@d|jd|jd}||j7}|jdur||j7}|S)z< Serialize a WebSocket handshake response. z HTTP/1.1  z N)rIrJr;rr<rK)r responserrrr<s   zResponse.serializer>)r#r$rLr$rMr$r rNr?) r@rArBrCrDrKrrEr!rFr:r<rrrrrGs      ^rGr#r$-Generator[None, None, datastructures.Headers]c cst}ttdD]p}z t|EdH}Wnty'}ztd|d}~ww|dkr/|Sz |dd\}}WntyJtdt|dwt |sYtdt|| d}t |sltd t|| d }| d d }|||<q td ) a Parse HTTP headers. Non-ASCII characters are represented with surrogate escapes. Args: read_line: generator-based coroutine that reads a LF-terminated line or raises an exception if there isn't enough data. Raises: EOFError: if the connection is closed without complete headers. SecurityError: if the request exceeds a security limit. ValueError: if the request isn't well formatted. rNz,connection closed while reading HTTP headers:zinvalid HTTP header line: zinvalid HTTP header name: s zinvalid HTTP header value: r*r+ztoo many HTTP headers)rZHeadersrange MAX_HEADERSr/r0r1r2r _token_rerSstriprRrrrW) r#r_liner6raw_nameZ raw_valuenamer rrrr3!s4         r3Generator[None, None, bytes]ccsLz |tEdH}Wn tytdw|ds td|ddS)ao Parse a single line. CRLF is stripped from the return value. Args: read_line: generator-based coroutine that reads a LF-terminated line or raises an exception if there isn't enough data. Raises: EOFError: if the connection is closed without a CRLF. SecurityError: if the response exceeds a security limit. Nz line too longs zline without CRLF)MAX_LINErVrrWendswithr0)r#rdrrrr/Ts    r/)r r r r )r#r$r r\)r#r$r rg) __future__r dataclassesrertypingrrrrrr`rirUrcompilerarR dataclassrrGr3r/rrrrs&   c 3