a یxd>@sddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZgdZe dZd d d d d d dZd d ddddZedZd d d dddZedZd d d ddddZedZedZ d d d ddddZ!edZ"ed Z#d d d!d"d#Z$d$d d d d%d&d'd(Z%d d d d)dd*d+Z&d d,d-d.d/Z'ed0Z(d d d d1dd2d3Z)d d4d-d5d6Z*d d d d7dd8d9Z+d d d d:dd;d<Z,d d=d-d>d?Z-e-Z.d@dAd dBdCdDZ/dEd dFdGdHZ0e0Z1d d d dIddJdKZ2d dLd-dMdNZ3e3Z4dOd dPdQdRZ5e5Z6dOdSdPdTdUZ7d d dVdWdXZ8edYZ9d d d dddZd[Z:d d d dSdd\d]Z;d d^d-d_d`ZzC"(?:[\x09\x20-\x21\x23-\x5b\x5d-\x7e]|\\[\x09\x20-\x7e\x80-\xff])*"z\\([\x09\x20-\x7e\x80-\xff])cCsDt||}|dur$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_rer6r r; _unquote_resubr<r7r=r/r/r0parse_quoted_string{s rCz[\x09\x20-\x7e\x80-\xff]*z ([\x22\x5c]))valuer$cCs.t|}|durtddtd|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_rerB)rDr6r/r/r0build_quoted_strings rIz(Callable[[str, int, str], Tuple[T, int]]zList[T]) parse_itemr1r2r9r$cCst||dkrt||d}qg}||||\}}||t||}|t|krTqt||dkrrt||d}nt|d||t||dkrt||d}q|t|kr"qq"|t|ksJ|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. ,r zexpected comma)r4r8appendr3r r;)rJr1r2r9itemsitemr/r/r0 parse_lists&    rOzTuple[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>r r r1r2r9rNr/r/r0parse_connection_options rQzList[ConnectionOption])r1r$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)rOrQr1r/r/r0rs rz>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?zTuple[UpgradeProtocol, int]cCs:t||}|dur$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_rer6r r;r rr<r7r=r/r/r0parse_upgrade_protocol s rUzList[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)rOrUrSr/r/r0rs rzTuple[ExtensionParameter, int]cCst|||\}}t||}d}t||dkrt||d}t||dkr||}t|||\}}t|durt|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=r rEzinvalid quoted header content)r>r8r4rCr:rGr r;)r1r2r9namerDZ pos_beforer/r/r0parse_extension_item_param-s   rXzTuple[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>r8r4rXrLr r)r1r2r9rW parametersZ parameterr/r/r0parse_extension_itemQs  r[zList[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)rOr[rSr/r/r0rjsrrzList[ExtensionParameter])rWrZr$cCs dtt|gdd|DS)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; cSs*g|]"\}}|dur|n |d|qS)NrVr/).0rWrDr/r/r0 sz(build_extension_item..)joinr r)rWrZr/r/r0build_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_)r\rWrZr/r/r0 sz"build_extension..r^)r`r/r/r0rs rzTuple[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>r rrPr/r/r0parse_subprotocol_items rdzList[Subprotocol]cCstt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: on invalid inputs. rzSec-WebSocket-Protocol)rOrdrSr/r/r0rs rzSequence[Subprotocol]) subprotocolsr$cCs d|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rarc)rer/r/r0rsrNonecCsJt|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 TypeErrorrr:rGr,)reZ subprotocolr/r/r0rs   r)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=)rI)richarsetr/r/r0rs rz[A-Za-z0-9-._~+/]+=*cCs4t||}|dur$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_rer6r r;r<r7r=r/r/r0 parse_token68s rlcCs |t|krt|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r3r r;)r1r2r9r/r/r0 parse_ends rmzTuple[str, str]cCst|dd\}}|dkr.tdd|t||dkrLtdd|||d7}t||d\}}t||dzt |  }Wn"t j ytddd Yn0z|d d\}}Wn tytdd d Yn0||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 schemer z#expected base64-encoded credentialsNr)z&expected username:password credentials)r>lowerr ZInvalidHeaderValuer4r;rlrmbase64 b64decodeencodedecodebinasciiErrorsplitr,)r1schemer2basic_credentials user_passusernamepasswordr/r/r0rs@   r)r|r}r$cCs4d|vs J|d|}t|}d|S)z Build an ``Authorization`` header for HTTP Basic Auth. This is the reverse of :func:`parse_authorization_basic`. r)zBasic )rr b64encodertru)r|r}r{rzr/r/r0r@s r)> __future__rrrrvr*retypingrrrrrrr r r r rrrr__all__rrr4compiler5r8r:r>r@rArCrFrHrIrOrQrrTrUrrXr[rZparse_extension_listr_rZbuild_extension_listrdrZparse_subprotocol_listrZbuild_subprotocol_listrrrkrlrmrrr/r/r/r0sd $        D$     0