B â•]bø>ã @sÈddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZddd d d d d ddddg Ze dƒZdddddœdd„Zddddœdd„Ze d¡Zddddœdd„Ze d¡Zdddd d!œd"d#„Ze d$¡Ze d%¡Z dddd d!œd&d'„Z!e d(¡Z"e d)¡Z#ddd*œd+d,„Z$d-dddd.d/œd0d1„Z%dddd2d!œd3d4„Z&dd5d6œd7d„Z'e d8¡Z(dddd9d!œd:d;„Z)ddd!œd?d@„Z+ddddAd!œdBdC„Z,ddDd6œdEd „Z-e-Z.dFdGddHœdIdJ„Z/dKddLœdMd „Z0e0Z1ddddNd!œdOdP„Z2ddQd6œdRd „Z3e3Z4dSddTœdUd „Z5e5Z6dSdVdTœdWd„Z7dddXœdYd„Z8e dZ¡Z9dddd d!œd[d\„Z:ddddVd!œd]d^„Z;dd_d6œd`d„Z[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?zTuple[UpgradeProtocol, int]cCs:t ||¡}|dkr$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_rer4r r:r rr;r6)r/r0r8r4r-r-r.Úparse_upgrade_protocol s  rPzList[UpgradeProtocol]cCstt|ddƒS)zÆ Parse an ``Upgrade`` header. Return a list of HTTP protocols. Args: header: value of the ``Upgrade`` header. Raises: InvalidHeaderFormat: on invalid inputs. rZUpgrade)rMrP)r/r-r-r.rs zTuple[ExtensionParameter, int]cCs¢t|||ƒ\}}t||ƒ}d}t||ƒdkr–t||dƒ}t||ƒdkr||}t|||ƒ\}}t |¡dkrŒt |d||¡‚nt|||ƒ\}}t||ƒ}||f|fS)zÌ Parse a single extension parameter from ``header`` at the given position. Return a ``(name, value)`` pair and the new position. Raises: InvalidHeaderFormat: on invalid inputs. Nú=r rCzinvalid quoted header content)r<r7r2rAr9rEr r:)r/r0r8ÚnamerBZ pos_beforer-r-r.Úparse_extension_item_param-s   rSzTuple[ExtensionHeader, int]cCslt|||ƒ\}}t||ƒ}g}x:t||ƒdkrXt||dƒ}t|||ƒ\}}| |¡q Wtt|ƒ|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<r7r2rSrJr r)r/r0r8rRÚ parametersZ parameterr-r-r.Úparse_extension_itemQs rVzList[ExtensionHeader]cCstt|ddƒS)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)rMrV)r/r-r-r.rjsrzList[ExtensionParameter])rRrUr$cCs d tt|ƒgdd„|Dƒ¡S)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; cSs*g|]"\}}|dkr|n |›d|›‘qS)NrQr-)Ú.0rRrBr-r-r.ú ”sz(build_extension_item..)Újoinr r)rRrUr-r-r.Úbuild_extension_item‡s  rZzSequence[ExtensionHeader])Ú extensionsr$cCsd dd„|Dƒ¡S)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. z, css|]\}}t||ƒVqdS)N)rZ)rWrRrUr-r-r.ú ¢sz"build_extension..)rY)r[r-r-r.ršszTuple[Subprotocol, int]cCst|||ƒ\}}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 r)r/r0r8rLr-r-r.Úparse_subprotocol_item©s r]zList[Subprotocol]cCstt|ddƒS)z  Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: on invalid inputs. rzSec-WebSocket-Protocol)rMr])r/r-r-r.r¹s zSequence[Subprotocol])Ú subprotocolsr$cCs d |¡S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. z, )rY)r^r-r-r.rÉsÚNonecCsNt|tƒstdƒ‚t|tƒr$tdƒ‚x$|D]}t |¡s*td|›ƒ‚q*WdS)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Ú TypeErrorrr9rEr+)r^Z subprotocolr-r-r.rÖs    )Úrealmr$cCs 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=)rG)rbÚcharsetr-r-r.räs z[A-Za-z0-9-._~+/]+=*cCs4t ||¡}|dkr$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_rer4r r:r;r6)r/r0r8r4r-r-r.Ú parse_token68õs  recCs |t|ƒkrt |d||¡‚dS)z8 Check that parsing reached the end of header. z trailing dataN)r1r r:)r/r0r8r-r-r.Ú parse_ends rfzTuple[str, str]cCsìt|ddƒ\}}| ¡dkr.t dd|›¡‚t||ƒdkrLt dd||¡‚|d7}t||dƒ\}}t||dƒyt  |  ¡¡  ¡}Wn$t j k rªt dd¡d ‚YnXy| d d¡\}}Wn"tk rât dd ¡d ‚YnX||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. rZ AuthorizationZbasiczunsupported scheme: ú zexpected space after schemer z#expected base64-encoded credentialsNr)z&expected username:password credentials)r<Úlowerr ZInvalidHeaderValuer2r:rerfÚbase64Z b64decodeÚencodeÚdecodeÚbinasciiÚErrorÚsplitr+)r/Zschemer0Úbasic_credentialsÚ user_passÚusernameÚpasswordr-r-r.rs4      )rqrrr$cCs4d|ks t‚|›d|›}t | ¡¡ ¡}d|S)z€ Build an ``Authorization`` header for HTTP Basic Auth. This is the reverse of :func:`parse_authorization_basic`. r)zBasic )r5riZ b64encoderjrk)rqrrrpror-r-r.r@s )>Z __future__rrirlr*ÚreÚtypingrrrrrrr Úr r r rrrrÚ__all__rrr2Úcompiler3r7r9r<r>r?rArDrFrGrMrNrrOrPrrSrVrZparse_extension_listrZrZbuild_extension_listr]rZparse_subprotocol_listrZbuild_subprotocol_listrrrdrerfrrr-r-r-r.Úst $        D$     0