a њh>@sddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZmZmZmZmZgdZe d Zd d d d ddd dddZd d ddddZedZd d d dddZedZd d d ddddZedZedZd d d ddd d!Z ed"Z!ed#Z"d d d$d%d&Z#d'd d d d(d)d*d+Z$d d d d,dd-d.Z%d d/d0d1d2Z&ed3Z'd d d d4dd5d6Z(d d7d0d8d9Z)d d d d:dd;d<Z*d d d d=dd>d?Z+d d@d0dAdBZ,e,Z-dCdDd dEdFdGZ.dHd dIdJdKZ/e/Z0d d d dLddMdNZ1d dOd0dPdQZ2e2Z3dRd dSdTdUZ4e4Z5dRdVdSdWdXZ6d d dYdZd[Z7ed\Z8d d d ddd]d^Z9d d d dVdd_d`Z:d dad0dbdcZ;d d d dddedfZ[-!#$%&\'*+.^_`|~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|ddS)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]cCst|||\}}t||}d}t||dkrt||d}t||dkrz|}t|||\}}t|durt|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)r0r1r8namerBZ 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|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)rMrYrQr.r.r/rksrr zSequence[ExtensionParameter])rUrXr#cCs dtt|gdd|DS)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/ sz(build_extension_item..)joinrr)rUrXr.r.r/build_extension_items  r]zSequence[ExtensionHeader]) extensionsr#cCsddd|DS)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/rs 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_items rbzlist[Subprotocol]cCstt|ddS)z Parse a ``Sec-WebSocket-Protocol`` header. Return a list of WebSocket subprotocols. Raises: InvalidHeaderFormat: On invalid inputs. rzSec-WebSocket-Protocol)rMrbrQr.r.r/rs 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/rsrNonecCsJt|tstdt|tr$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/rs   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)rgcharsetr.r.r/rs 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_token68s 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]cCst|dd\}}|dkr,tdd|t||dkrHtdd|||d7}t||d\}}t||dzt|  }Wn t j ytddd Yn0z| d d\}}Wntytdd 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 Authorizationbasiczunsupported scheme:  zexpected space after schemerz#expected base64-encoded credentialsNr(z&expected username:password credentials)r<lowerr r3rrjrkbase64 b64decodeencodedecodebinasciiErrorsplitr+)r0schemer1basic_credentials user_passusernamepasswordr.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)recollections.abcrtypingrrr exceptionsrr r r r r rr__all__rrr3compiler4r7r9r<r>r?rArDrFrGrMrOrrRrSrrVrYrZparse_extension_listr]rZbuild_extension_listrbrZparse_subprotocol_listrZbuild_subprotocol_listrrrirjrkrrr.r.r.r/sh   "      B$     0