&th>xddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZgdZe dZdPdZdQdZejdZdRdZejdZdSdZejdZejdZ dSdZ!ejdZ"ejdZ#dTd!Z$dUd%Z%dVd'Z&dWd)Z'ejd*Z(dXd,Z)dYd.Z*dZd0Z+d[d2Z,d\d4Z-e-Z.d]d9Z/d^d<Z0e0Z1d_d>Z2d`d@Z3e3Z4dadCZ5e5Z6dbdEZ7dcdGZ8ejdHZ9dSdIZ:dddJZ;dedLZr;s r1 parse_tokenrEcsP OOFC ( (E },[:JFTWXXX ;;==%))++ %%r8zC"(?:[\x09\x20-\x21\x23-\x5b\x5d-\x7e]|\\[\x09\x20-\x7e\x80-\xff])*"z\\([\x09\x20-\x7e\x80-\xff])ct||}|tj|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_rer;r rB _unquote_resubrCr<rDs r1parse_quoted_stringrK{sq  # #FC 0 0E }, 163    ??5%++--""5 6 6 CCr8z[\x09\x20-\x7e\x80-\xff]*z ([\x22\x5c])valuect|}|tddtd|zdzS)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_rerJ)rLr;s r1build_quoted_stringrRsI  " "5 ) )E }HIII w.. . 44r8 parse_item(Callable[[str, int, str], Tuple[T, int]]List[T]c\t||dkr't||dz}t||dk'g} ||||\}}||t||}|t|krnt||dkrt||dz}nt j|d||t||dkr't||dz}t||dk'|t|krn|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 Tzexpected comma)r7r=appendr6r rB)rSr2r3r>itemsitems r1 parse_listr[sb> VS ! !S ( (a(( VS ! !S ( ( EJvsK88 c T$$ #f++    fc " "c ) )FC!G,,CC0-vs  %%,,FC!G,,C%%,, #f++   18 #f++     Lr8Tuple[ConnectionOption, int]cXt|||\}}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. )rEr r r2r3r>rZs r1parse_connection_optionr_s/FC55ID#  $ ' ' ,,r8List[ConnectionOption]c0tt|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)r[r_r2s r1rrs -vq, G GGr8z>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?Tuple[UpgradeProtocol, int]ct||}|tj|d||t t ||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_rer;r rBr rrCr<rDs r1parse_upgrade_protocolrg sd   vs + +E }, ,fc     / / <namerL pos_befores r1parse_extension_item_paramrp-sFC55ID# FC CE&##%%a(( fc " "c ) )J,VS+FFJE3""5))1 4!@&*2 %VS+>>JE3$$ %=# r8Tuple[ExtensionHeader, int]cHt|||\}}t||}g}t||dkrPt||dz}t|||\}}||t||dkPt t ||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 )rEr=r7rprXr r)r2r3r>rn parameters parameters r1parse_extension_itemrvQsFC55ID# FC CJ VS ! !S ( (a((3FCMM 3)$$$ VS ! !S ( (  % %z 2C 77r8List[ExtensionHeader]c0tt|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)r[rvrcs r1rrjs. *FA7Q R RRr8rnrrtList[ExtensionParameter]cndtt|gd|DzS)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; c*g|]\}}||n|d|S)Nrm).0rnrLs r1 z(build_extension_item..sD   eMDD$'8'8'8'8   r8)joinr r )rnrts r1build_extension_itemrsK 99 c4   *      r8 extensionsSequence[ExtensionHeader]c@dd|DS)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. , c3<K|]\}}t||VdS)N)r)r}rnrts r1 z"build_extension..sB3C4T:..r8r)rs r1rrs6 99GQ  r8Tuple[Subprotocol, int]cXt|||\}}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. )rEr rr^s r1parse_subprotocol_itemrs.FC55ID#  T " "C ''r8List[Subprotocol]c0tt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: on invalid inputs. rzSec-WebSocket-Protocol)r[rrcs r1rrs ,fa9Q R RRr8 subprotocolsSequence[Subprotocol]c,d|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rr)rs r1rrs 99\ " ""r8Nonect|tstdt|trtd|D].}t|st d|/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 TypeErrorr rArPr/)r subprotocols r1rrs lH - -75666,$$B@AAA#DD "";// DB[BBCC C DDDr8realmcNt|}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=)rR)rcharsets r1rrs4  & &E!'**G 4% 4 47 4 44r8z[A-Za-z0-9-._~+/]+=*ct||}|tj|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_rer;r rBrCr<rDs r1 parse_token68rsZ   fc * *E }, +VS    ;;==%))++ %%r8cZ|t|krtj|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r6r rB)r2r3r>s r1 parse_endrs5  S[[,[/6SVWWWr8Tuple[str, str]ct|dd\}}|dkrtjdd|t ||dkrtjdd|||dz }t ||d\}}t||d tj |  }n)#tj $rtjddd wxYw |d d\}}n$#t$rtjdd d wxYw||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)rElowerr InvalidHeaderValuer7rBrrbase64 b64decodeencodedecodebinasciiErrorsplitr/)r2schemer3basic_credentials user_passusernamepasswords r1rrs fa99KFC ||~~  +  +6 + +   &##%%, :FC   1HC*63HHs fc?+++$%6%=%=%?%?@@GGII >+  1    &__S!44(( +  4    X s8C&C:>D!D9rrcd|vsJ|d|}tj|}d|zS)z Build an ``Authorization`` header for HTTP Basic Auth. This is the reverse of :func:`parse_authorization_basic`. r+zBasic )r b64encoderr)rrrrs r1rr@s] h    ((h((I()9)9););<<CCEE ' ''r8)rr r!r"r#r$r%r )r2r r3r"r%r4)r2r r3r"r%r")r2r r3r"r>r r%r?)rLr r%r ) rSrTr2r r3r"r>r r%rU)r2r r3r"r>r r%r\)r2r r%r`)r2r r3r"r>r r%rd)r2r r%rh)r2r r3r"r>r r%rk)r2r r3r"r>r r%rq)r2r r%rw)rnrrtryr%r )rrr%r )r2r r3r"r>r r%r)r2r r%r)rrr%r )rrr%r)rr r%r )r2r r3r"r>r r%r)r2r r%r)rr rr r%r )> __future__rrrr,retypingrrrrrr r r r rrrrr__all__rrr7compiler:r=rArErHrIrKrOrQrRr[r_rrfrgrrprvrparse_extension_listrrbuild_extension_listrrparse_subprotocol_listrbuild_subprotocol_listrrrrrrrr|r8r1rs""""""  KKKKKKKKKKKKKKKKKK    GCLL8 7 7 7 7 "*Y       BJ6 7 7  & & & & BJJ bj899 DDDD$rz677  BJ ' '  5 5 5 5AAAAH - - - - H H H H rzE ====( D D D D !!!!H88882SSSS4'&    ' ( ( ( ( S S S S+####+ D D D D 5 5 5 5bj011 &&&&$XXXX----` ( ( ( ( ( (r8