RPf>ddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZgdZe dZd*d Zd+d Zej4d Zd,d Zej4d Zd-dZej4dZej4dZ d-dZ!ej4dZ"ej4dZ#d.dZ$ d/dZ% d0dZ&d1dZ'ej4dZ( d2dZ)d3dZ* d4dZ+ d5dZ,d6dZ-e-Z. d7dZ/d8dZ0e0Z1 d9d Z2d:d!Z3e3Z4d;d"Z5e5Z6dd'Z;d?d(Zitemsitems r- parse_listrSs> VS !S (a( VS !S ( E vsK8 c T$ #f+   fc "c )FC!G,C00-vs  %,FC!G,C%, #f+  1 8 #f+   Lr3cFt|||\}}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?r r r0r1r>rRs r-parse_connection_optionrVs)FC5ID#  $ ' ,,r3c&tt|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)rSrVr0s r-rrs -vq, GGr3z>[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?ctj||}|tj|d||t t |j |jfS)z Parse an Upgrade protocol from ``header`` at the given position. Return the protocol value and the new position. Raises: InvalidHeaderFormat: on invalid inputs. zexpected protocol) _protocol_rer6r r;r rr<r7r=s r-parse_upgrade_protocolr\ sX   vs +E },, ,fc    / <namerK pos_befores r-parse_extension_item_paramrc-sFC5ID# FC CE&##%a( fc "c )J,VS+FJE3""5)1 44!@&*%VS+>JE3$ %=# r3ct|||\}}t||}g}t||dk(r@t||dz}t|||\}}|j |t||dk(r@t 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 )r?r8r2rcrOr r)r0r1r>ra parameters parameters r-parse_extension_itemrhQsFC5ID# FC CJ VS !S (a(3FCM 3)$ VS !S (  %z 2C 77r3c&tt|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)rSrhrYs r-rrjs. *FA7Q RRr3c djtt|g|Dcgc]\}}||n|d|c}}zScc}}w)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; r`)joinr str)rarfrKs r-build_extension_itemrms] 99 c4 * eMD$q'8 8    sA c2djd|DS)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. , c3:K|]\}}t||yw)N)rm).0rarfs r- z"build_extension..s!3C4T:.srk) extensionss r-rrs" 99GQ r3cFt|||\}}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. )r?r rrUs r-parse_subprotocol_itemrvs(FC5ID#  T "C ''r3c&tt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: on invalid inputs. rzSec-WebSocket-Protocol)rSrvrYs r-rrs ,fa9Q RRr3c$dj|S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. rors) subprotocolss r-rrs 99\ ""r3ct|ts tdt|tr td|D]%}tj |rt d|y)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 TypeErrorrlr:rIr()ry subprotocols r-rrsb lH -566,$@AA#D "";/4[MBC CDr3c>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=)rL)realmcharsets r-rrs+  &E!'*G % 7) 44r3z[A-Za-z0-9-._~+/]+=*ctj||}|tj|d|||j |j fS)z Parse a token68 from ``header`` at the given position. Return the token value and the new position. Raises: InvalidHeaderFormat: on invalid inputs. zexpected token68) _token68_rer6r r;r<r7r=s r- parse_token68rsQ   fc *E },, +VS   ;;=%))+ %%r3cP|t|krtj|d||y)z8 Check that parsing reached the end of header. z trailing dataN)r/r r;)r0r1r>s r- parse_endrs-  S[,,[/6SVWWr3cHt|dd\}}|jdk7rtjdd|t ||dk7rtj dd|||dz }t ||d\}}t||d tj|jj} |jd d\}}||fS#tj$rtjddd wxYw#t$rtjdd d wxYw) 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)r?lowerr InvalidHeaderValuer2r;rrbase64 b64decodeencodedecodebinasciiErrorsplitr()r0schemer1basic_credentials user_passusernamepasswords r-rrsM fa9KFC ||~ ++ "6( +  &##%,, :FC  1HC*63Hs fc?+$$%6%=%=%?@GGI &__S!4( X  >>++  1   ++  4  s1C9D+C=!D!cd|vsJ|d|}tj|jj}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-rr@sQ h  *AhZ(I(()9)9);<CCE ' ''r3)r)rlr*intr+boolreturnrl)r0rlr1rrz Optional[str])r0rlr1rrr)r0rlr1rr>rlrzTuple[str, int])rKrlrrl) rPz(Callable[[str, int, str], Tuple[T, int]]r0rlr1rr>rlrzList[T])r0rlr1rr>rlrzTuple[ConnectionOption, int])r0rlrzList[ConnectionOption])r0rlr1rr>rlrzTuple[UpgradeProtocol, int])r0rlrzList[UpgradeProtocol])r0rlr1rr>rlrzTuple[ExtensionParameter, int])r0rlr1rr>rlrzTuple[ExtensionHeader, int])r0rlrzList[ExtensionHeader])rarrfzList[ExtensionParameter]rrl)rtzSequence[ExtensionHeader]rrl)r0rlr1rr>rlrzTuple[Subprotocol, int])r0rlrzList[Subprotocol])rySequence[Subprotocol]rrl)ryrrNone)rrlrrl)r0rlr1rr>rlrr)r0rlrzTuple[str, str])rrlrrlrrl)> __future__rrrr%retypingrrrrrr r r r rrrrr__all__rrr2compiler5r8r:r?rBrCrErHrJrLrSrVrr[r\rrcrhrparse_extension_listrmrbuild_extension_listrvrparse_subprotocol_listrbuild_subprotocol_listrrrrrrrr3r-rsj"  KKK  CL8 7 "**Y   BJJ6 7  & BJJJ bjj89 D$rzz67  BJJ '  5A8A A A A  AH -  - -(+ -! - H rzzE = ==(+= =( D ! !!(+!#!H8 88(+8 82S4' %=& ' (  ( ((+ ( ( S+#+ D 5bjj01 &$X-` (r3