vgV>xddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZmZmZmZmZgdZe d ZdQdZdRdZejdZdSdZejdZdTdZejdZejdZdTdZ ejdZ!ejd Z"dUd"Z#dVd&Z$dWd(Z%dXd*Z&ejd+Z'dYd-Z(dZd/Z)d[d1Z*d\d3Z+d]d5Z,e,Z-d^d:Z.d_d=Z/e/Z0d`d?Z1dadAZ2e2Z3dbdDZ4e4Z5dcdFZ6dddHZ7ejdIZ8dTdJZ9dedKZ:dfdMZ;dgdPZ VS ! !S ( (a(( VS ! !S ( ( EJvsK88 c T$$ #f++    fc " "c ) )FC!G,,CC%k3CVSQQ Q%%,,FC!G,,C%%,, #f++   -4 #f++     Lr6tuple[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. )rBrr r0r1r<rWs r/parse_connection_optionr\s/FC55ID#  $ ' ' ,,r6list[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)rXr\r0s r/rrs -vq, G GGr6z>[-!#$%&\'*+.^_`|~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_rer9r rrr@r:rAs r/parse_upgrade_protocolrdsX   vs + +E }!+/BFCPPP  / / <>JE3$$ %=# r6tuple[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)rBr;r5rmrUrr )r0r1r<rk parameters parameters r/parse_extension_itemrsLsFC55ID# FC CJ VS ! !S ( (a((3FCMM 3)$$$ VS ! !S ( (  % %z 2C 77r6list[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)rXrsr`s r/rres. *FA7Q R RRr6rkr rqlist[ExtensionParameter]cndtt|gd|DzS)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; c*g|]\}}||n|d|S)Nrj).0rkrIs r/ z(build_extension_item..sD   eMDD$'8'8'8'8   r6)joinrr)rkrqs r/build_extension_itemr}sK 99 c4   *      r6 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})rzrkrqs r/ z"build_extension..sB3C4T:..r6r|)r~s r/rrs6 99GQ  r6tuple[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. )rBrrr[s r/parse_subprotocol_itemrs.FC55ID#  T " "C ''r6list[Subprotocol]c0tt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Protocol)rXrr`s r/rrs ,fa9Q R RRr6 subprotocolsSequence[Subprotocol]c,d|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rr)rs r/rrs 99\ " ""r6Nonect|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?rMr-)r subprotocols r/rrs lH - -75666,$$B@AAA#DD "";// DB[BBCC C DDDr6realmcNt|}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=)rO)rcharsets r/rrs4  & &E!'**G 4% 4 47 4 44r6z[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_rer9r r@r:rAs r/ parse_token68rsN   fc * *E }!+/A63OOO ;;==%))++ %%r6cP|t|krt|d||dS)z8 Check that parsing reached the end of header. z trailing dataN)r4r )r0r1r<s r/ parse_endrs1  S[[!+LLLr6tuple[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)rBlowerr r5r rrbase64 b64decodeencodedecodebinasciiErrorsplitr-)r0schemer1basic_credentials user_passusernamepasswords r/rr s 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 r/rr9s] h    ((h((I()9)9););<<CCEE ' ''r6)rrrr r!r"r#r)r0rr1r r#r2)r0rr1r r#r )r0rr1r r<rr#r=)rIrr#r) rPrQr0rr1r r<rr#rR)r0rr1r r<rr#rY)r0rr#r])r0rr1r r<rr#ra)r0rr#re)r0rr1r r<rr#rh)r0rr1r r<rr#rn)r0rr#rt)rkr rqrvr#r)r~rr#r)r0rr1r r<rr#r)r0rr#r)rrr#r)rrr#r)rrr#r)r0rr1r r<rr#r)r0rr#r)rrrrr#r)= __future__rrrr*recollections.abcrtypingrrr exceptionsr r r r r rrr__all__rrr5compiler8r;r?rBrErFrHrLrNrOrXr\rrcrdrrmrsrparse_extension_listr}rbuild_extension_listrrparse_subprotocol_listrbuild_subprotocol_listrrrrrrrryr6r/rs""""""  $$$$$$**********????????    GCLL8 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----` ( ( ( ( ( (r6