o :h>@sddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZgdZe dZdiddZdjddZedZdkddZedZdlddZedZed Z dld!d"Z!ed#Z"ed$Z#dmd&d'Z$dnd+d,Z%dod.d/Z&dpd1d2Z'ed3Z(dqd5d6Z)drd8d9Z*dsd;d<Z+dtd>d?Z,dudAdBZ-e-Z.dvdGdHZ/dwdKdLZ0e0Z1dxdNdOZ2dydQdRZ3e3Z4dzdUdVZ5e5Z6d{dXdYZ7d|d[d\Z8ed]Z9dld^d_Z:d}d`daZ;d~dcddZ _unquote_resubr?r8r@r/r/r0parse_quoted_string{s  rFz[\x09\x20-\x7e\x80-\xff]*z ([\x22\x5c])valuecCs.t|}|dur tddtd|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_rerE)rGr7r/r/r0build_quoted_strings rL parse_item(Callable[[str, int, str], Tuple[T, int]]List[T]cCst||dkrt||d}t||dksg} ||||\}}||t||}|t|kr1n4t||dkr@t||d}nt|d||t||dkr]t||d}t||dksO|t|krdnq|t|ksmJ|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 Tzexpected comma)r5r9appendr4r r>)rMr1r2r:itemsitemr/r/r0 parse_lists.    rTTuple[ConnectionOption, int]cCt|||\}}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. )rAr r r1r2r:rSr/r/r0parse_connection_option rXList[ConnectionOption]cCtt|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)rTrXr1r/r/r0r rz>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?Tuple[UpgradeProtocol, int]cCs:t||}|durt|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_rer7r r>r rr?r8r@r/r/r0parse_upgrade_protocol s raList[UpgradeProtocol]cCr[)z Parse an ``Upgrade`` header. Return a list of HTTP protocols. Args: header: value of the ``Upgrade`` header. Raises: InvalidHeaderFormat: on invalid inputs. rUpgrade)rTrar]r/r/r0rr^rTuple[ExtensionParameter, int]cCst|||\}}t||}d}t||dkrKt||d}t||dkr>|}t|||\}}t|dur=t|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 rHzinvalid quoted header content)rAr9r5rFr=rJr r>)r1r2r:namerG pos_beforer/r/r0parse_extension_item_param-s    rhTuple[ExtensionHeader, int]cCstt|||\}}t||}g}t||dkr1t||d}t|||\}}||t||dkstt||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 )rAr9r5rhrQr r)r1r2r:rf parameters parameterr/r/r0parse_extension_itemQs  rmList[ExtensionHeader]cCr[)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)rTrmr]r/r/r0rjsrrfrrkList[ExtensionParameter]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)Nrer/).0rfrGr/r/r0 sz(build_extension_item..)joinr r)rfrkr/r/r0build_extension_items rs extensionsSequence[ExtensionHeader]cCsddd|DS)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. , css|] \}}t||VqdS)N)rs)rprfrkr/r/r0 s z"build_extension..rr)rtr/r/r0rs rTuple[Subprotocol, int]cCrV)z Parse a subprotocol from ``header`` at the given position. Return the subprotocol value and the new position. Raises: InvalidHeaderFormat: on invalid inputs. )rAr rrWr/r/r0parse_subprotocol_itemrYrzList[Subprotocol]cCr[)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: on invalid inputs. rzSec-WebSocket-Protocol)rTrzr]r/r/r0rs r subprotocolsSequence[Subprotocol]cCs d|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rvrx)r|r/r/r0rs rNonecCsJt|ts tdt|trtd|D]}t|s"td|qdS)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=rJr,)r| subprotocolr/r/r0rs   rrealmcCs 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=)rL)rcharsetr/r/r0rs rz[A-Za-z0-9-._~+/]+=*cCr<)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_rer7r r>r?r8r@r/r/r0 parse_token68s rcCs |t|krt|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r4r r>)r1r2r:r/r/r0 parse_ends rTuple[str, str]cCst|dd\}}|dkrtdd|t||dkr&tdd|||d7}t||d\}}t||dz t |  }Wnt j yRtddd wz |d d\}}W||fStyntdd d w) 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)rAlowerr InvalidHeaderValuer5r>rrbase64 b64decodeencodedecodebinasciiErrorsplitr,)r1schemer2basic_credentials user_passusernamepasswordr/r/r0rsD   rrrcCs4d|vsJ|d|}t|}d|S)z Build an ``Authorization`` header for HTTP Basic Auth. This is the reverse of :func:`parse_authorization_basic`. r)zBasic )r b64encoderr)rrrrr/r/r0r@s r)rrr r!r"r#r$r)r1rr2r!r$r3)r1rr2r!r$r!)r1rr2r!r:rr$r;)rGrr$r) rMrNr1rr2r!r:rr$rO)r1rr2r!r:rr$rU)r1rr$rZ)r1rr2r!r:rr$r_)r1rr$rb)r1rr2r!r:rr$rd)r1rr2r!r:rr$ri)r1rr$rn)rfrrkror$r)rtrur$r)r1rr2r!r:rr$ry)r1rr$r{)r|r}r$r)r|r}r$r~)rrr$r)r1rr2r!r:rr$r~)r1rr$r)rrrrr$r)> __future__rrrr*retypingrrrrrrr r r r rrrr__all__rrr5compiler6r9r=rArCrDrFrIrKrLrTrXrr`rarrhrmrparse_extension_listrsrbuild_extension_listrzrparse_subprotocol_listrbuild_subprotocol_listrrrrrrrr/r/r/r0sf $              D     $            0