o `^h;>@sddlmZddlZddlZddlZddlZddlmZmZm Z m Z ddl m Z m Z ddlmZmZmZmZmZmZgdZe dZdiddZdjddZedZdkddZedZdlddZedZed Zdld!d"Zed#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-dwdKdLZ.e.Z/dxdNdOZ0dydQdRZ1e1Z2dzdUdVZ3e3Z4d{dXdYZ5d|d[d\Z6ed]Z7dld^d_Z8d}d`daZ9d~dcddZ:ddgdhZ;dS)) annotationsN)CallableSequenceTypeVarcast)InvalidHeaderFormatInvalidHeaderValue)ConnectionOptionExtensionHeader ExtensionNameExtensionParameter SubprotocolUpgradeProtocol) build_hostparse_connection parse_upgradeparse_extensionbuild_extensionparse_subprotocolbuild_subprotocolvalidate_subprotocolsbuild_www_authenticate_basicparse_authorization_basicbuild_authorization_basicThoststrportintsecureboolreturncCsZzt|}Wn tyYn w|jdkrd|d}||r!dndkr+|d|}|S)z# Build a ``Host`` header. []iP:) ipaddress ip_address ValueErrorversion)rrr addressr-P/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/websockets/headers.pyr&s   rheaderpos str | NonecCs|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 r3z[\t ]*cCs t||}|dus J|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_rematchend)r/r0r5r-r-r. parse_OWSQs r7z[-!#$%&\'*+.^_`|~0-9a-zA-Z]+ header_nametuple[str, int]cC2t||}|durt|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_rer5rgroupr6r/r0r8r5r-r-r. parse_tokenc r>zC"(?:[\x09\x20-\x21\x23-\x5b\x5d-\x7e]|\\[\x09\x20-\x7e\x80-\xff])*"z\\([\x09\x20-\x7e\x80-\xff])cCsBt||}|durt|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_rer5r _unquote_resubr<r6r=r-r-r.parse_quoted_string{s  rDz[\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_rerC)rEr5r-r-r.build_quoted_strings rJ 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)r3r7appendr2r)rKr/r0r8itemsitemr-r-r. parse_lists*    rRtuple[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 r/r0r8rQr-r-r.parse_connection_option rVlist[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)rRrVr/r-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_rer5rrrr<r6r=r-r-r.parse_upgrade_protocols r_list[UpgradeProtocol]cCrY)z Parse an ``Upgrade`` header. Return a list of HTTP protocols. Args: header: Value of the ``Upgrade`` header. Raises: InvalidHeaderFormat: On invalid inputs. rUpgrade)rRr_r[r-r-r.rr\rtuple[ExtensionParameter, int]cCst|||\}}t||}d}t||dkrJt||d}t||dkr=|}t|||\}}t|durr7r3rDr;rHr)r/r0r8namerE pos_beforer-r-r.parse_extension_item_param's    rftuple[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)r>r7r3rfrOrr )r/r0r8rd parameters parameterr-r-r.parse_extension_itemKs  rklist[ExtensionHeader]cCrY)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)rRrkr[r-r-r.rdsrrdr rilist[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)Nrcr-).0rdrEr-r-r. sz(build_extension_item..)joinrr)rdrir-r-r.build_extension_items rq extensionsSequence[ExtensionHeader]cCsddd|DS)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. , css|] \}}t||VqdS)N)rq)rnrdrir-r-r. s z"build_extension..rp)rrr-r-r.rs rtuple[Subprotocol, int]cCrT)z Parse a subprotocol from ``header`` at the given position. Return the subprotocol value and the new position. Raises: InvalidHeaderFormat: On invalid inputs. )r>rrrUr-r-r.parse_subprotocol_itemrWrxlist[Subprotocol]cCrY)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Protocol)rRrxr[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`. rtrv)rzr-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;rHr*)rz 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=)rJ)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_rer5rr<r6r=r-r-r. parse_token68r?rcCs|t|kr t|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r2r)r/r0r8r-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 r3rrrbase64 b64decodeencodedecodebinasciiErrorsplitr*)r/schemer0basic_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.r8s r)rrrrr r!r"r)r/rr0rr"r1)r/rr0rr"r)r/rr0rr8rr"r9)rErr"r) rKrLr/rr0rr8rr"rM)r/rr0rr8rr"rS)r/rr"rX)r/rr0rr8rr"r])r/rr"r`)r/rr0rr8rr"rb)r/rr0rr8rr"rg)r/rr"rl)rdr rirmr"r)rrrsr"r)r/rr0rr8rr"rw)r/rr"ry)rzr{r"r)rzr{r"r|)rrr"r)r/rr0rr8rr"r|)r/rr"r)rrrrr"r)< __future__rrrr(retypingrrrr exceptionsrr r r r r rr__all__rrr3compiler4r7r;r>rArBrDrGrIrJrRrVrr^r_rrfrkrparse_extension_listrqrbuild_extension_listrxrparse_subprotocol_listrbuild_subprotocol_listrrrrrrrr-r-r-r.sf              B     $            0