o KK&h>@sddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZmZmZmZmZgdZe d Zd d dmddZdnddZedZdoddZedZdpd!d"Zed#Zed$Zdpd%d&Z ed'Z!ed(Z"dqd*d+Z#drd/d0Z$dsd2d3Z%dtd5d6Z&ed7Z'dud9d:Z(dvdr.r.r/parse_quoted_strings  rEz[\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_rerD)rFr6r.r.r/build_quoted_strings rK parse_item(Callable[[str, int, str], tuple[T, int]]list[T]cCst||dkrt||d}t||dksg} ||||\}}||t||}|t|kr1n3t||dkr@t||d}nt|d||t||dkr\t||d}t||dksN|t|krcnq|t|kslJ|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. ,rTzexpected comma)r4r8appendr3r)rLr0r1r9itemsitemr.r.r/ parse_lists*    rStuple[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. )r?rr r0r1r9rRr.r.r/parse_connection_option rWlist[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)rSrWr0r.r.r/r rz>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?tuple[UpgradeProtocol, int]cCs8t||}|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_rer6rrrr=r7r>r.r.r/parse_upgrade_protocol s r`list[UpgradeProtocol]cCrZ)z Parse an ``Upgrade`` header. Return a list of HTTP protocols. Args: header: Value of the ``Upgrade`` header. Raises: InvalidHeaderFormat: On invalid inputs. rUpgrade)rSr`r\r.r.r/rr]rtuple[ExtensionParameter, int]cCst|||\}}t||}d}t||dkrJt||d}t||dkr=|}t|||\}}t|dursz(build_extension_item..)joinrr)rerjr.r.r/build_extension_items rr extensionsSequence[ExtensionHeader]cCsddd|DS)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. , css|] \}}t||VqdS)N)rr)rorerjr.r.r/ s z"build_extension..rq)rsr.r.r/rs rtuple[Subprotocol, int]cCrU)z Parse a subprotocol from ``header`` at the given position. Return the subprotocol value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. )r?rrrVr.r.r/parse_subprotocol_itemrXrylist[Subprotocol]cCrZ)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Protocol)rSryr\r.r.r/rs r subprotocolsSequence[Subprotocol]cCs d|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rurw)r{r.r.r/rs 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<rIr+)r{ subprotocolr.r.r/rs   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=)rK)rcharsetr.r.r/rs 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_rer6rr=r7r>r.r.r/ parse_token68r@rcCs|t|kr t|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r3r)r0r1r9r.r.r/ parse_ends rtuple[str, str]cCst|dd\}}|dkrtdd|t||dkr$tdd|||d7}t||d\}}t||dz t|  }Wnt j yOtddd wz | d d\}}W||fStyjtdd 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 schemerz#expected base64-encoded credentialsNr(z&expected username:password credentials)r?lowerr r4rrrbase64 b64decodeencodedecodebinasciiErrorsplitr+)r0schemer1basic_credentials user_passusernamepasswordr.r.r/rsD   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.r/r?s r) rrrr r!r"rr"r#r)r0rr1r r#r2)r0rr1r r#r )r0rr1r r9rr#r:)rFrr#r) rLrMr0rr1r r9rr#rN)r0rr1r r9rr#rT)r0rr#rY)r0rr1r r9rr#r^)r0rr#ra)r0rr1r r9rr#rc)r0rr1r r9rr#rh)r0rr#rm)rer rjrnr#r)rsrtr#r)r0rr1r r9rr#rx)r0rr#rz)r{r|r#r)r{r|r#r})rrr#r)r0rr1r r9rr#r})r0rr#r)rrrrr#r)= __future__rrrr)recollections.abcrtypingrrr exceptionsrr r r r r rr__all__rrr4compiler5r8r<r?rBrCrErHrJrKrSrWrr_r`rrgrlrparse_extension_listrrrbuild_extension_listryrparse_subprotocol_listrbuild_subprotocol_listrrrrrrrr.r.r.r/sj     "         B     $            0