a ñ»jø@ã@s¾ddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZmZmZmZmZgd¢Ze d ƒZd d œd d ddd dœdd„Zd d ddœdd„Ze d¡Zd d d dœdd„Ze d¡Zd d d ddœdd„Ze d¡Ze d¡Zd d d ddœd d!„Z e d"¡Z!e d#¡Z"d d d$œd%d&„Z#d'd d d d(d)œd*d+„Z$d d d d,dœd-d.„Z%d d/d0œd1d2„Z&e d3¡Z'd d d d4dœd5d6„Z(d d7d0œd8d9„Z)d d d d:dœd;d<„Z*d d d d=dœd>d?„Z+d d@d0œdAdB„Z,e,Z-dCdDd dEœdFdG„Z.dHd dIœdJdK„Z/e/Z0d d d dLdœdMdN„Z1d dOd0œdPdQ„Z2e2Z3dRd dSœdTdU„Z4e4Z5dRdVdSœdWdX„Z6d d dYœdZd[„Z7e d\¡Z8d d d ddœd]d^„Z9d d d dVdœd_d`„Z:d dad0œdbdc„Z;d d d ddœdedf„Z[-!#$%&\'*+.^_`|~0-9a-zA-Z]+(?:/[-!#$%&\'*+.^_`|~0-9a-zA-Z]+)?ztuple[UpgradeProtocol, int]cCs8t ||¡}|dur"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_rer5rrrr:r6r;r.r.r/Úparse_upgrade_protocol s rSzlist[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)rMrSrQr.r.r/rs rztuple[ExtensionParameter, int]cCs t|||ƒ\}}t||ƒ}d}t||ƒdkr”t||dƒ}t||ƒdkrz|}t|||ƒ\}}t |¡durŠ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ú=rrCzinvalid quoted header content)r<r7r3rAr9rEr)r0r1r8ÚnamerBZ pos_beforer.r.r/Úparse_extension_item_param.s  ÿ rVztuple[ExtensionHeader, int]cCsht|||ƒ\}}t||ƒ}g}t||ƒdkrVt||dƒ}t|||ƒ\}}| |¡qtt|ƒ|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<r7r3rVrJrr )r0r1r8rUÚ parametersZ parameterr.r.r/Úparse_extension_itemRs  rYzlist[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)rMrYrQr.r.r/rksrr zSequence[ExtensionParameter])rUrXr#cCs d tt|ƒgdd„|Dƒ¡S)zc Build an extension definition. This is the reverse of :func:`parse_extension_item`. z; cSs*g|]"\}}|dur|n |›d|›‘qS)NrTr.)Ú.0rUrBr.r.r/Ú “sÿz(build_extension_item..)Újoinrr)rUrXr.r.r/Úbuild_extension_itemˆs  ýÿÿr]zSequence[ExtensionHeader])Ú extensionsr#cCsd dd„|Dƒ¡S)zl Build a ``Sec-WebSocket-Extensions`` header. This is the reverse of :func:`parse_extension`. ú, css|]\}}t||ƒVqdS)N)r])rZrUrXr.r.r/Ú ¢sz"build_extension..©r\)r^r.r.r/r›s ÿrztuple[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<rrrNr.r.r/Úparse_subprotocol_itemªs rbzlist[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)rMrbrQr.r.r/rºs rzSequence[Subprotocol])Ú subprotocolsr#cCs d |¡S)zl Build a ``Sec-WebSocket-Protocol`` header. This is the reverse of :func:`parse_subprotocol`. r_ra)rcr.r.r/rÊsrÚNonecCsJt|tƒstdƒ‚t|tƒr$tdƒ‚|D]}t |¡s(td|›ƒ‚q(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Ú TypeErrorrr9rEr+)rcZ subprotocolr.r.r/r×s   r)Ú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)rgÚcharsetr.r.r/rås rz[A-Za-z0-9-._~+/]+=*cCs2t ||¡}|dur"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_rer5rr:r6r;r.r.r/Ú parse_token68ös rjcCs|t|ƒkrt|d||ƒ‚dS)z8 Check that parsing reached the end of header. z trailing dataN)r2r)r0r1r8r.r.r/Ú parse_ends rkztuple[str, str]cCsàt|ddƒ\}}| ¡dkr,tdd|›ƒ‚t||ƒdkrHtdd||ƒ‚|d7}t||dƒ\}}t||dƒzt |  ¡¡  ¡}Wn t j y¢tddƒd ‚Yn0z|  d d¡\}}WntyÖtdd ƒd ‚Yn0||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Ú AuthorizationÚbasiczunsupported scheme: ú zexpected space after schemerz#expected base64-encoded credentialsNr(z&expected username:password credentials)r<Úlowerr r3rrjrkÚbase64Ú b64decodeÚencodeÚdecodeÚbinasciiÚErrorÚsplitr+)r0Úschemer1Úbasic_credentialsÚ user_passÚusernameÚpasswordr.r.r/rs@ þÿ þý þýr)rzr{r#cCs4d|vs J‚|›d|›}t | ¡¡ ¡}d|S)z€ Build an ``Authorization`` header for HTTP Basic Auth. This is the reverse of :func:`parse_authorization_basic`. r(zBasic )rpÚ b64encoderrrs)rzr{ryrxr.r.r/r?s r)=Ú __future__rrprtr)ÚreÚcollections.abcrÚtypingrrrÚ exceptionsrr r r r r rrÚ__all__rrr3Úcompiler4r7r9r<r>r?rArDrFrGrMrOrrRrSrrVrYrZparse_extension_listr]rZbuild_extension_listrbrZparse_subprotocol_listrZbuild_subprotocol_listrrrirjrkrrr.r.r.r/Úsh   û"  ÿ    Bÿ$     0