U g;> @sddlmZddlZddlZddlZddlZddlmZmZm Z m Z ddl m Z m Z ddlmZmZmZmZmZmZddd d d d d ddddg Ze dZdddddddZddddddZedZddddddZedZdddd d!d"d#Zed$Zed%Zdddd d!d&d'Zed(Z ed)Z!ddd*d+d,Z"d-dddd.d/d0d1Z#dddd2d!d3d4Z$dd5d6d7dZ%ed8Z&dddd9d!d:d;Z'ddd!d?d@Z)ddddAd!dBdCZ*ddDd6dEd Z+e+Z,dFdGddHdIdJZ-dKddLdMd Z.e.Z/ddddNd!dOdPZ0ddQd6dRd Z1e1Z2dSddTdUd Z3e3Z4dSdVdTdWdZ5dddXdYdZ6edZZ7dddd d!d[d\Z8ddddVd!d]d^Z9dd_d6d`dZ:ddddadbdZ;dS)c) annotationsN)CallableSequenceTypeVarcast)InvalidHeaderFormatInvalidHeaderValue)ConnectionOptionExtensionHeader ExtensionNameExtensionParameter SubprotocolUpgradeProtocol build_hostparse_connection parse_upgradeparse_extensionbuild_extensionparse_subprotocolbuild_subprotocolvalidate_subprotocolsbuild_www_authenticate_basicparse_authorization_basicbuild_authorization_basicTstrintbool)hostportsecurereturncCs\zt|}Wntk r"YnX|jdkr:d|d}||rDdndkrX|d|}|S)z# Build a ``Host`` header. []iP:) ipaddress ip_address ValueErrorversion)rr r!addressr-6/tmp/pip-unpacked-wheel-dx_q7dq3/websockets/headers.pyr&s  z str | None)headerposr"cCs|t|krdS||S)z Return the next character from ``header`` at the given position. Return :obj:`None` at the end of ``header``. We never need to peek more than one character ahead. N)len)r/r0r-r-r. peek_aheadBs r2z[\t ]*cCs t||}|dk st|S)z Parse optional whitespace from ``header`` at the given position. Return the new position. The whitespace itself isn't returned because it isn't significant. N)_OWS_rematchAssertionErrorend)r/r0r4r-r-r. parse_OWSQs  r7z[-!#$%&\'*+.^_`|~0-9a-zA-Z]+ztuple[str, int])r/r0 header_namer"cCs2t||}|dkr"t|d||||fS)z Parse a token from ``header`` at the given position. Return the token value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. Nzexpected token) _token_rer4rgroupr6r/r0r8r4r-r-r. parse_tokencs r<zC"(?:[\x09\x20-\x21\x23-\x5b\x5d-\x7e]|\\[\x09\x20-\x7e\x80-\xff])*"z\\([\x09\x20-\x7e\x80-\xff])cCsBt||}|dkr"t|d||td|dd|fS)z Parse a quoted string from ``header`` at the given position. Return the unquoted value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. Nzexpected quoted stringz\1r)_quoted_string_rer4r _unquote_resubr:r6r;r-r-r.parse_quoted_string{s rAz[\x09\x20-\x7e\x80-\xff]*z ([\x22\x5c]))valuer"cCs.t|}|dkrtddtd|dS)zh Format ``value`` as a quoted string. This is the reverse of :func:`parse_quoted_string`. Nz-invalid characters for quoted-string encoding"z\\\1) _quotable_re fullmatchr* _quote_rer@)rBr4r-r-r.build_quoted_strings rGz(Callable[[str, int, str], tuple[T, int]]zlist[T]) parse_itemr/r0r8r"cCst||dkrt||d}qg}||||\}}||t||}|t|krTqt||dkrrt||d}nt|d||t||dkrt||d}q|t|kr"qq"|t|kst|S)a Parse a comma-separated list from ``header`` at the given position. This is appropriate for parsing values with the following grammar: 1#item ``parse_item`` parses one item. ``header`` is assumed not to start or end with whitespace. (This function is designed for parsing an entire header value and :func:`~websockets.http.read_headers` strips whitespace from values.) Return a list of items. Raises: InvalidHeaderFormat: On invalid inputs. ,rzexpected comma)r2r7appendr1rr5)rHr/r0r8itemsitemr-r-r. parse_lists"    rMztuple[ConnectionOption, int]cCst|||\}}tt||fS)z Parse a Connection option from ``header`` at the given position. Return the protocol value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. )r<rr r/r0r8rLr-r-r.parse_connection_options rOzlist[ConnectionOption])r/r"cCstt|ddS)z Parse a ``Connection`` header. Return a list of HTTP connection options. Args header: value of the ``Connection`` header. Raises: InvalidHeaderFormat: On invalid inputs. r Connection)rMrOr/r-r-r.rs z>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?ztuple[UpgradeProtocol, int]cCs8t||}|dkr"t|d||tt||fS)z Parse an Upgrade protocol from ``header`` at the given position. Return the protocol value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. Nzexpected protocol) _protocol_rer4rrrr:r6r;r-r-r.parse_upgrade_protocols rSzlist[UpgradeProtocol]cCstt|ddS)z Parse an ``Upgrade`` header. Return a list of HTTP protocols. Args: header: Value of the ``Upgrade`` header. Raises: InvalidHeaderFormat: On invalid inputs. rZUpgrade)rMrSrQr-r-r.rs ztuple[ExtensionParameter, int]cCst|||\}}t||}d}t||dkrt||d}t||dkrz|}t|||\}}t|dkrt|d||nt|||\}}t||}||f|fS)z Parse a single extension parameter from ``header`` at the given position. Return a ``(name, value)`` pair and the new position. Raises: InvalidHeaderFormat: On invalid inputs. N=rrCzinvalid quoted header content)r<r7r2rAr9rEr)r/r0r8namerBZ pos_beforer-r-r.parse_extension_item_param's$   rVztuple[ExtensionHeader, int]cCsht|||\}}t||}g}t||dkrVt||d}t|||\}}||qtt||f|fS)a Parse an extension definition from ``header`` at the given position. Return an ``(extension name, parameters)`` pair, where ``parameters`` is a list of ``(name, value)`` pairs, and the new position. Raises: InvalidHeaderFormat: On invalid inputs. ;r)r<r7r2rVrJrr )r/r0r8rU parametersZ parameterr-r-r.parse_extension_itemKs  rYzlist[ExtensionHeader]cCstt|ddS)a Parse a ``Sec-WebSocket-Extensions`` header. Return a list of WebSocket extensions and their parameters in this format:: [ ( 'extension name', [ ('parameter name', 'parameter value'), .... ] ), ... ] Parameter values are :obj:`None` when no value is provided. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Extensions)rMrYrQr-r-r.rdsr zlist[ExtensionParameter])rUrXr"cCs dtt|gdd|DS)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; cSs*g|]"\}}|dkr|n |d|qS)NrTr-).0rUrBr-r-r. sz(build_extension_item..)joinrr)rUrXr-r-r.build_extension_items  r]zSequence[ExtensionHeader]) extensionsr"cCsddd|DS)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. , css|]\}}t||VqdS)N)r])rZrUrXr-r-r. sz"build_extension..r\)r^r-r-r.rs ztuple[Subprotocol, int]cCst|||\}}tt||fS)z Parse a subprotocol from ``header`` at the given position. Return the subprotocol value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. )r<rrrNr-r-r.parse_subprotocol_items rbzlist[Subprotocol]cCstt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Protocol)rMrbrQr-r-r.rs zSequence[Subprotocol]) subprotocolsr"cCs d|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. r_ra)rcr-r-r.rsNonecCsJt|tstdt|tr$td|D]}t|s(td|q(dS)zT Validate that ``subprotocols`` is suitable for :func:`build_subprotocol`. zsubprotocols must be a listz&subprotocols must be a list, not a strzinvalid subprotocol: N) isinstancer TypeErrorrr9rEr*)rcZ subprotocolr-r-r.rs   )realmr"cCs t|}td}d|d|S)z Build a ``WWW-Authenticate`` header for HTTP Basic Auth. Args: realm: Identifier of the protection space. zUTF-8z Basic realm=z , charset=)rG)rgcharsetr-r-r.rs z[A-Za-z0-9-._~+/]+=*cCs2t||}|dkr"t|d||||fS)z Parse a token68 from ``header`` at the given position. Return the token value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. Nzexpected token68) _token68_rer4rr:r6r;r-r-r. parse_token68s rjcCs|t|krt|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r1r)r/r0r8r-r-r. parse_ends rkztuple[str, str]cCst|dd\}}|dkr,tdd|t||dkrHtdd|||d7}t||d\}}t||dzt|  }Wn"t j k rtddd YnXz| d d\}}Wn tk rtdd d YnX||fS) a! Parse an ``Authorization`` header for HTTP Basic Auth. Return a ``(username, password)`` tuple. Args: header: Value of the ``Authorization`` header. Raises: InvalidHeaderFormat: On invalid inputs. InvalidHeaderValue: On unsupported inputs. r Authorizationbasiczunsupported scheme:  zexpected space after schemerz#expected base64-encoded credentialsNr'z&expected username:password credentials)r<lowerr r2rrjrkbase64 b64decodeencodedecodebinasciiErrorsplitr*)r/schemer0basic_credentials user_passusernamepasswordr-r-r.rsF  )rzr{r"cCs4d|ks t|d|}t|}d|S)z Build an ``Authorization`` header for HTTP Basic Auth. This is the reverse of :func:`parse_authorization_basic`. r'zBasic )r5rp b64encoderrrs)rzr{ryrxr-r-r.r8s )< __future__rrprtr(retypingrrrr exceptionsrr r r r r rr__all__rrr2compiler3r7r9r<r>r?rArDrFrGrMrOrrRrSrrVrYrZparse_extension_listr]rZbuild_extension_listrbrZparse_subprotocol_listrZbuild_subprotocol_listrrrirjrkrrr-r-r-r.sz        B$     0