U g@4@sZddlmZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZdd d d gZd jejZejd dedeZejddedeZeejddZeejddZeejddZdddddZedZedZej Gdd d Z!ej Gdd d Z"d d!d"d#d$Z#d d%d"d&d'Z$dS)() annotationsN)Callable Generator)Headers) SecurityError)versionSERVER USER_AGENTRequestResponsez{}.{}ZWEBSOCKETS_USER_AGENTzPython/z websockets/ZWEBSOCKETS_SERVERZWEBSOCKETS_MAX_NUM_HEADERSZ128ZWEBSOCKETS_MAX_LINE_LENGTHZ8192ZWEBSOCKETS_MAX_BODY_SIZEZ 1_048_576bytesstr)valuereturncCs |jddS)zG Decode a bytestring for interpolating into an error message. backslashreplace)errors)decode)rr5/tmp/pip-unpacked-wheel-dx_q7dq3/websockets/http11.pyd,srs[-!#$%&\'*+.^_`|~0-9a-zA-Z]+s[\x09\x20-\x7e\x80-\xff]*c@sbeZdZUdZded<ded<dZded<edd d d Zed d dddZ dd ddZ dS)r z WebSocket handshake request. Attributes: path: Request path, including optional query. headers: Request headers. rpathrheadersNException | None _exceptionrcCstdt|jS)NzIRequest.exception is deprecated; use ServerProtocol.handshake_exc insteadwarningswarnDeprecationWarningrselfrrr exceptionXs zRequest.exception-Callable[[int], Generator[None, None, bytes]]zGenerator[None, None, Request] read_linerc cszt|EdH}Wn,tk r>}ztd|W5d}~XYnXz|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: HTTP/1.1)unsupported protocol; expected HTTP/1.1: sGETz+unsupported HTTP method; expected GET; got asciisurrogateescapeTransfer-Encoding!transfer codings aren't supportedContent-Lengthzunsupported request body) parse_lineEOFErrorsplit ValueErrorrr parse_headersNotImplementedError) clsr%Z request_lineexcmethodraw_pathprotocolrrrrrparseas*#  z Request.parser cCs$d|jd}||j7}|S)z; Serialize a WebSocket handshake request. zGET z HTTP/1.1 )rencoder serialize)r!requestrrrr<szRequest.serialize) __name__ __module__ __qualname____doc____annotations__rpropertyr" classmethodr:r<rrrrr Hs  ?c@szeZdZUdZded<ded<ded<dZd ed <dZd ed <ed d ddZe dddddddZ dd ddZ dS)r z WebSocket handshake response. Attributes: status_code: Response code. reason_phrase: Response reason. headers: Response headers. body: Response body, if any. int status_coder reason_phraserrNz bytes | NonebodyrrrcCstdt|jS)NzJResponse.exception is deprecated; use ClientProtocol.handshake_exc insteadrr rrrr"s zResponse.exceptionr#zGenerator[None, None, Response])r% read_exact read_to_eofrc cs$zt|EdH}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||d d} t |EdH} d| krBt dd | krZdkspn| dksp| dkrvd} nz | d} Wnt k rd}Yn Xt| }|dkrz|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 liner&r'zinvalid HTTP status line: r(r)z+invalid status code; expected integer; got diXu-invalid status code; expected 100–599; got zinvalid HTTP reason phrase: r*r+r,r-i0r.zbody too large: over z byteszbody too large: )r/r0r1r2rrE _value_re fullmatchrr3r4KeyError MAX_BODY_SIZE RuntimeErrorr)r5r%rIrJZ status_liner6r9Zraw_status_codeZ raw_reasonrFreasonrrHZraw_content_lengthcontent_lengthrrrr:s\        .    zResponse.parser cCs@d|jd|jd}||j7}|jdk r<||j7}|S)z< Serialize a WebSocket handshake response. z HTTP/1.1  z N)rFrGr;rr<rH)r!responserrrr<(s   zResponse.serialize) r>r?r@rArBrHrrCr"rDr:r<rrrrr s    ]r#zGenerator[None, None, Headers]r$c cst}ttdD]}zt|EdH}Wn,tk rT}ztd|W5d}~XYnX|dkrdq z|dd\}}Wn(tk rtdt|dYnXt |stdt|| d}t |std t|| d }| d d }|||<qt d |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)rrangeMAX_NUM_HEADERSr/r0r1r2r _token_rerOstriprNrr) r%r_liner6Zraw_name raw_valuenamerrrrr36s,      r3zGenerator[None, None, bytes]ccsNz|tEdH}Wntk r.tdYnX|dsBtd|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_LINE_LENGTHrRrendswithr0)r%r^rrrr/is r/)% __future__rZ dataclassesosresysrtypingrrZdatastructuresr exceptionsrrZwebsockets_version__all__format version_infoPYTHON_VERSIONenvirongetr r rErZrbrQrcompiler[rNZ dataclassr r r3r/rrrrs@       d 3