B ]b1@sddlmZddlZddlZddlZddlmZmZmZddl m Z m Z dZ dZ dZd d d d d ZedZedZejGdddZejGdddZdddddZdddddZdS)) annotationsN)Callable GeneratorOptional)datastructures exceptionsiibytesstr)valuereturncCs |jddS)zG Decode a bytestring for interpolating into an error message. backslashreplace)errors)decode)r r0lib/python3.7/site-packages/websockets/http11.pydsrs[-!#$%&\'*+.^_`|~0-9a-zA-Z]+s[\x09\x20-\x7e\x80-\xff]*c@sbeZdZUdZded<ded<dZded<edd d d Zed d dddZ dd ddZ dS)Requestz WebSocket handshake request. Attributes: path: Request path, including optional query. headers: Request headers. r pathzdatastructures.HeadersheadersNzOptional[Exception] _exception)r cCstdt|jS)NzKRequest.exception is deprecated; use ServerConnection.handshake_exc instead)warningswarnDeprecationWarningr)selfrrr exceptionEszRequest.exceptionz-Callable[[int], Generator[None, None, bytes]]zGenerator[None, None, Request]) read_liner c csyt|EdH}Wn,tk r>}ztd|Wdd}~XYnXy|dd\}}}Wn(tk r~tdt|dYnX|dkrtdt||dkrtd t||d d }t|EdH}d |krtd d|krtd|||S)a Parse a WebSocket handshake request. This is a generator-based coroutine. The request path isn't URL-decoded or validated in any way. The request path and headers are expected to contain only ASCII characters. Other characters are represented with surrogate escapes. :meth:`parse` 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 the data stream after :meth:`parse` returns. 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 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 zinvalid HTTP request line: sGETzunsupported HTTP method: sHTTP/1.1zunsupported HTTP version: asciisurrogateescapezTransfer-Encodingz!transfer codings aren't supportedzContent-Lengthzunsupported request body) parse_lineEOFErrorsplit ValueErrorrr parse_headersNotImplementedError) clsrZ request_lineexcmethodZraw_pathversionrrrrrparseNs&# z Request.parser cCs$d|jd}||j7}|S)z; Serialize a WebSocket handshake request. zGET z HTTP/1.1 )rencoder serialize)rZrequestrrrr.szRequest.serialize) __name__ __module__ __qualname____doc____annotations__rpropertyr classmethodr,r.rrrrr5s   ?rc@szeZdZUdZded<ded<ded<dZd ed <dZd ed <ed d ddZe dddddddZ dd ddZ 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_phrasezdatastructures.HeadersrNzOptional[bytes]bodyzOptional[Exception]r)r cCstdt|jS)NzLResponse.exception is deprecated; use ClientConnection.handshake_exc instead)rrrr)rrrrrszResponse.exceptionz-Callable[[int], Generator[None, None, bytes]]zGenerator[None, None, Response])r read_exact read_to_eofr c cs$yt|EdH}Wn,tk r>}ztd|Wdd}~XYnXy|dd\}}}Wn(tk r~tdt|dYnX|dkrtdt|y t|} Wn(tk rtdt|dYnXd | krd ksntd t|t|std t||} t |EdH} d | kr>t dd | krVdksln| dksl| dkrrd} ny | d} Wnt k rd}Yn Xt| }|dkry|t EdH} Wn(t k rtdt dYnXn,|t krtd|dn||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 linerrzinvalid HTTP status line: sHTTP/1.1zunsupported HTTP version: zinvalid HTTP status code: dizunsupported HTTP status code: zinvalid HTTP reason phrase: zTransfer-Encodingz!transfer codings aren't supportedi0zContent-Lengthzbody too large: over z byteszbody too large: )r"r#r$r%rr7 _value_re fullmatchrr&r'KeyErrorMAX_BODY RuntimeErrorr SecurityError)r(rr;r<Z status_liner)r+Zraw_status_codeZ raw_reasonr8reasonrr:Zraw_content_lengthZcontent_lengthrrrr,sT     .    zResponse.parser cCs@d|jd|jd}||j7}|jdk r<||j7}|S)z< Serialize a WebSocket handshake response. z HTTP/1.1  z N)r8r9r-rr.r:)rZresponserrrr.s   zResponse.serialize) r/r0r1r2r3r:rr4rr5r,r.rrrrr6s    _r6z-Callable[[int], Generator[None, None, bytes]]z-Generator[None, None, datastructures.Headers])rr c cst}xttdD]}yt|EdH}Wn,tk rZ}ztd|Wdd}~XYnX|dkrfPy|dd\}}Wn(tk rtdt|dYnXt |stdt|| d}t |std t|| d }| d d }|||<qWtd |S) 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"r#r$r%r _token_rerAstripr@rrrE) rr_liner)Zraw_nameZ raw_valuenamer rrrr&#s,       r&zGenerator[None, None, bytes]ccsPy|tEdH}Wntk r0tdYnX|dsDtd|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_LINErDrrEendswithr#)rrOrrrr"Vs r")Z __future__rZ dataclassesrertypingrrrrrrKrRrCrcompilerLr@Z dataclassrr6r&r"rrrrs  d 3