U Mf> @sddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZddd d d d d ddddg Ze dZdddddddZddddddZedZddddddZedZdddd d!d"d#Zed$Zed%Z dddd d!d&d'Z!ed(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 Z0e0Z1ddddNd!dOdPZ2ddQd6dRd Z3e3Z4dSddTdUd Z5e5Z6dSdVdTdWdZ7dddXdYdZ8edZZ9dddd d!d[d\Z:ddddVd!d]d^Z;dd_d6d`dZr/r/r0parse_quoted_string{s rDz[\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_rerC)rEr6r/r/r0build_quoted_strings rJz(Callable[[str, int, str], Tuple[T, int]]zList[T]) parse_itemr1r2r:r$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. ,r zexpected comma)r4r9appendr3r r<r7)rKr1r2r:itemsitemr/r/r0 parse_lists,    rPzTuple[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 r1r2r:rOr/r/r0parse_connection_options rRzList[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)rPrRr1r/r/r0rs z>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?zTuple[UpgradeProtocol, int]cCs:t||}|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_rer6r r<r rr=r8r>r/r/r0parse_upgrade_protocol s rVzList[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)rPrVrTr/r/r0rs zTuple[ExtensionParameter, int]cCst|||\}}t||}d}t||dkrt||d}t||dkr||}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=r rFzinvalid quoted header content)r?r9r4rDr;rHr r<)r1r2r:namerEZ pos_beforer/r/r0parse_extension_item_param-s$   rYzTuple[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?r9r4rYrMr r)r1r2r:rX 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)rPr\rTr/r/r0rjsrzList[ExtensionParameter])rXr[r$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)NrWr/).0rXrEr/r/r0 sz(build_extension_item..)joinr r)rXr[r/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]rXr[r/r/r0 sz"build_extension..r_)rar/r/r0rs 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?r rrQr/r/r0parse_subprotocol_items rezList[Subprotocol]cCstt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: on invalid inputs. rzSec-WebSocket-Protocol)rPrerTr/r/r0rs zSequence[Subprotocol]) subprotocolsr$cCs d|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rbrd)rfr/r/r0rsNonecCsJt|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;rHr,)rfZ subprotocolr/r/r0rs   )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=)rJ)rjcharsetr/r/r0rs z[A-Za-z0-9-._~+/]+=*cCs4t||}|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_rer6r r<r=r8r>r/r/r0 parse_token68s rmcCs |t|krt|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r3r r<)r1r2r:r/r/r0 parse_ends rnzTuple[str, str]cCst|dd\}}|dkr.tdd|t||dkrLtdd|||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 schemer z#expected base64-encoded credentialsNr)z&expected username:password credentials)r?lowerr ZInvalidHeaderValuer4r<rmrnbase64 b64decodeencodedecodebinasciiErrorsplitr,)r1schemer2basic_credentials user_passusernamepasswordr/r/r0rsF  )r}r~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 )r7rs b64encoderurv)r}r~r|r{r/r/r0r@s )> __future__rrsrwr*retypingrrrrrrr r r r rrrr__all__rrr4compiler5r9r;r?rArBrDrGrIrJrPrRrrUrVrrYr\rZparse_extension_listr`rZbuild_extension_listrerZparse_subprotocol_listrZbuild_subprotocol_listrrrlrmrnrrr/r/r/r0sz $        D$     0