Jtj>~ddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZmZmZmZmZgdZe d Zd d dTdZdUdZejdZdVdZejdZdWdZejdZejd ZdWd!Z ejd"Z!ejd#Z"dXd%Z#dYd)Z$dZd+Z%d[d-Z&ejd.Z'd\d0Z(d]d2Z)d^d4Z*d_d6Z+d`d8Z,e,Z-dad=Z.dbd@Z/e/Z0dcdBZ1dddDZ2e2Z3dedGZ4e4Z5dfdIZ6dgdKZ7ejdLZ8dWdMZ9dhdNZ:didPZ;djdSZ VS ! !S ( (a(( VS ! !S ( ( EJvsK88 c T$$ #f++    fc " "c ) )FC!G,,CC%k3CVSQQ Q%%,,FC!G,,C%%,, #f++   -4 #f++     Lr7tuple[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. )rCrr r1r2r=rXs r0parse_connection_optionr]s/FC55ID#  $ ' ' ,,r7list[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)rYr]r1s r0rrs -vq, G GGr7z>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?tuple[UpgradeProtocol, int]ct||}|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_rer:r rrrAr;rBs r0parse_upgrade_protocolre sX   vs + +E }!+/BFCPPP  / / <>JE3$$ %=# r7tuple[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)rCr<r6rnrVrr )r1r2r=rl parameters parameters r0parse_extension_itemrtRsFC55ID# FC CJ VS ! !S ( (a((3FCMM 3)$$$ VS ! !S ( (  % %z 2C 77r7list[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)rYrtras r0rrks. *FA7Q R RRr7rlr rrSequence[ExtensionParameter]cndtt|gd|DzS)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; c*g|]\}}||n|d|S)Nrk).0rlrJs r0 z(build_extension_item..sD   eMDD$'8'8'8'8   r7)joinrr)rlrrs r0build_extension_itemr~sK 99 c4   *      r7 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{rlrrs r0 z"build_extension..sB3C4T:..r7r})rs r0rrs6 99GQ  r7tuple[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. )rCrrr\s r0parse_subprotocol_itemrs.FC55ID#  T " "C ''r7list[Subprotocol]c0tt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Protocol)rYrras r0rrs ,fa9Q R RRr7 subprotocolsSequence[Subprotocol]c,d|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rr)rs r0rrs 99\ " ""r7Nonect|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 TypeErrorrr@rNr.)r subprotocols r0rrs lH - -75666,$$B@AAA#DD "";// DB[BBCC C DDDr7realmcNt|}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=)rP)rcharsets r0rrs4  & &E!'**G 4% 4 47 4 44r7z[A-Za-z0-9-._~+/]+=*ct||}|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_rer:r rAr;rBs r0 parse_token68rsN   fc * *E }!+/A63OOO ;;==%))++ %%r7cP|t|krt|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r5r )r1r2r=s r0 parse_endrs1  S[[!+LLLr7tuple[str, str]cXt|dd\}}|dkrtdd|t||dkrt dd|||dz }t ||d\}}t ||d tj|  }n$#tj $rtddd wxYw | d d\}}n#t$rtdd 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 schemerz#expected base64-encoded credentialsNr*z&expected username:password credentials)rClowerr r6r rrbase64 b64decodeencodedecodebinasciiErrorsplitr.)r1schemer2basic_credentials user_passusernamepasswords r0rrs fa99KFC ||~~    +6 + +   &##%%! :FC   1HC*63HHs fc?+++$%6%=%=%?%?@@GGII >  1    &__S!44((   4    X s8C !C+/D D%rrcd|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 r0rr?s] h    ((h((I()9)9););<<CCEE ' ''r7) rrr r!r"r#rr#r$r)r1rr2r!r$r3)r1rr2r!r$r!)r1rr2r!r=rr$r>)rJrr$r) rQrRr1rr2r!r=rr$rS)r1rr2r!r=rr$rZ)r1rr$r^)r1rr2r!r=rr$rb)r1rr$rf)r1rr2r!r=rr$ri)r1rr2r!r=rr$ro)r1rr$ru)rlr rrrwr$r)rrr$r)r1rr2r!r=rr$r)r1rr$r)rrr$r)rrr$r)rrr$r)r1rr2r!r=rr$r)r1rr$r)rrrrr$r)= __future__rrrr+recollections.abcrtypingrrr exceptionsr r r r r rrr__all__rrr6compiler9r<r@rCrFrGrIrMrOrPrYr]rrdrerrnrtrparse_extension_listr~rbuild_extension_listrrparse_subprotocol_listrbuild_subprotocol_listrrrrrrrrzr7r0rs""""""  $$$$$$**********????????    GCLL!& D 7 7 7 7 "*Y       BJ6 7 7  & & & & BJJ bj899  D D D D rz677  BJ ' '  5 5 5 5????D - - - - H H H H rzE ====$ D D D D !!!!H88882SSSS4'&    ' ( ( ( ( S S S S+####+ D D D D 5 5 5 5bj011  & & & & MMMM----` ( ( ( ( ( (r7