U gU`@sddlmZddlZddlZddlmZmZddlmZddl m Z m Z m Z m Z mZmZddlmZmZdd lmZmZmZd d d d dgZdZddeddDZGdd d eZddddddddZdddddd ZGd!d d eZd"d#d$d%d ZGd&d d eZd'd(d$d)dZ dS)*) annotationsN)AnySequence)frames)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError) ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactoryPerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescCsg|] }t|qS)str).0bitsrrL/tmp/pip-unpacked-wheel-dx_q7dq3/websockets/extensions/permessage_deflate.py src@sheZdZdZedZddddddddd d Zd d d dZddddddddZdddddZ dS)rz) Per-Message Deflate extension. permessage-deflateNboolintzdict[Any, Any] | NoneNone)remote_no_context_takeoverlocal_no_context_takeoverremote_max_window_bitslocal_max_window_bitscompress_settingsreturncCs|dkr i}|dkst|dks$td|kr8dks>ntd|krRdksXntd|ksdt||_||_||_||_||_|jstj|j d|_|jstj fd|j i|j|_ d|_ dS)z? Configure the Per-Message Deflate extension. N)FTrwbitsr+F) AssertionErrorr$r%r&r'r(zlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr$r%r&r'r(rrr__init__)s*    zPerMessageDeflate.__init__rr)c Cs&d|jd|jd|jd|jd S)Nz-PerMessageDeflate(remote_no_context_takeover=z, local_no_context_takeover=z, remote_max_window_bits=z, local_max_window_bits=))r$r%r&r'r4rrr__repr__Ss$zPerMessageDeflate.__repr__)max_sizez frames.Frame int | None)framer:r)c Cs |jtjkr|S|jtjkr4|js&|S|jrpd|_n<|js>|Stj|dd}|jsXd|_|j rpt j |j d|_ |j}|jr|t7}|dkrdn|}z|j ||}Wn.t jk r}ztd|W5d}~XYnX|j jrtd|d |jr|j r|` tj||d S) z, Decode an incoming frame. Frsv1Tr,Nrzdecompression failedzover size limit (? > z bytes)data)opcoder CTRL_OPCODESOP_CONTr3finr> dataclassesreplacer$r.r/r&r0r@_EMPTY_UNCOMPRESSED_BLOCK decompresserrorr unconsumed_tailr )r4r<r:r@ max_lengthexcrrrdecode\s6   zPerMessageDeflate.decode)r<r)cCs|jtjkr|S|jtjk rLtj|dd}|jrLtjfd|j i|j |_ |j |j |j tj}|jr|tr|dd}|jr|jr|` tj||dS)z, Encode an outgoing frame. Tr=r+Nr?)rArrBrCrErFr%r.r1r'r(r2compressr@flush Z_SYNC_FLUSHrDendswithrG)r4r<r@rrrencodes     zPerMessageDeflate.encode)N) __name__ __module__ __qualname____doc__r namer5r9rMrSrrrrr!s* :r!r;int | bool | Nonelist[ExtensionParameter])server_no_context_takeoverclient_no_context_takeoverserver_max_window_bitsclient_max_window_bitsr)cCsdg}|r|d|r |d|r6|dt|f|dkrJ|dn|r`|dt|f|S)zS Build a list of ``(name, value)`` pairs for some compression parameters. )r[N)r\Nr]T)r^Nr^)appendr)r[r\r]r^paramsrrr_build_parameterss    raSequence[ExtensionParameter]z0tuple[bool, bool, int | None, int | bool | None])r` is_serverr)cCs d}d}d}d}|D]\}}|dkrJ|r0t||dkr>d}qt||q|dkrx|r^t||dkrld}qt||q|dkr|dk rt||tkrt|}qt||q|dkr|dk rt||r|dkrd}q|tkrt|}qt||qt|q||||fS)z Extract compression parameters from a list of ``(name, value)`` pairs. If ``is_server`` is :obj:`True`, ``client_max_window_bits`` may be provided without a value. This is only allowed in handshake requests. FNr[Tr\r]r^)rr _MAX_WINDOW_BITS_VALUESr"r)r`rcr[r\r]r^rXvaluerrr_extract_parameterssJ          rfc@sReZdZdZedZdddddd d d d d ZddddZddddddZdS)ra Client-side extension factory for the Per-Message Deflate extension. Parameters behave as described in `section 7.1 of RFC 7692`_. .. _section 7.1 of RFC 7692: https://datatracker.ietf.org/doc/html/rfc7692#section-7.1 Set them to :obj:`True` to include them in the negotiation offer without a value or to an integer value to include them with this value. Args: server_no_context_takeover: Prevent server from using context takeover. client_no_context_takeover: Prevent client from using context takeover. server_max_window_bits: Maximum size of the server's LZ77 sliding window in bits, between 8 and 15. client_max_window_bits: Maximum size of the client's LZ77 sliding window in bits, between 8 and 15, or :obj:`True` to indicate support without setting a limit. compress_settings: Additional keyword arguments for :func:`zlib.compressobj`, excluding ``wbits``. r FNTr!r;rYdict[str, Any] | Noner#)r[r\r]r^r(r)cCs|dks&d|krdks&ntd|dksT|dksTd|krJdksTntd|dk rld|krltd||_||_||_||_||_dS) G Configure the Per-Message Deflate extension factory. Nrr*/server_max_window_bits must be between 8 and 15T/client_max_window_bits must be between 8 and 15r+zLcompress_settings must not include wbits, set client_max_window_bits instead) ValueErrorr[r\r]r^r()r4r[r\r]r^r(rrrr5's* z'ClientPerMessageDeflateFactory.__init__rZr6cCst|j|j|j|jS)z, Build request parameters. )rar[r\r]r^r8rrrget_request_paramsGs z1ClientPerMessageDeflateFactory.get_request_paramsrbSequence[Extension]rr`accepted_extensionsr)cstfdd|Dr&tdjt|dd\}}}}jrL|sLtdjrZ|sZd}jdkrfn$|dkrxtd n|jkrtd jdkr|dk rtd n.jdkrn"|dkrˆj}n|jkrtd t|||pd |pd j S)zV Process response parameters. Return an extension instance. c3s|]}|jjkVqdSNrXrotherr8rr ^szIClientPerMessageDeflateFactory.process_response_params..zreceived duplicate Frcz#expected server_no_context_takeoverTNzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr*) anyr rXrfr[r\r]r^rr(r4r`ror[r\r]r^rr8rprocess_response_paramsSsH          z6ClientPerMessageDeflateFactory.process_response_params)FFNTN) rTrUrVrWr rXr5rlrxrrrrr s  z'Sequence[ClientExtensionFactory] | Nonez Sequence[ClientExtensionFactory]) extensionsr)cCs:|dkr g}tdd|Ds6t|tddidg}|S)z Enable Per-Message Deflate with default settings in client extensions. If the extension is already present, perhaps with non-default settings, the configuration isn't changed. Ncss|]}|jtjkVqdSrp)rXr)rZextension_factoryrrrrtsz3enable_client_permessage_deflate..memLevel)r()rvlistrryrrrrs c @sFeZdZdZedZddddddddd d d Zd d ddddZdS)ra Server-side extension factory for the Per-Message Deflate extension. Parameters behave as described in `section 7.1 of RFC 7692`_. .. _section 7.1 of RFC 7692: https://datatracker.ietf.org/doc/html/rfc7692#section-7.1 Set them to :obj:`True` to include them in the negotiation offer without a value or to an integer value to include them with this value. Args: server_no_context_takeover: Prevent server from using context takeover. client_no_context_takeover: Prevent client from using context takeover. server_max_window_bits: Maximum size of the server's LZ77 sliding window in bits, between 8 and 15. client_max_window_bits: Maximum size of the client's LZ77 sliding window in bits, between 8 and 15. compress_settings: Additional keyword arguments for :func:`zlib.compressobj`, excluding ``wbits``. require_client_max_window_bits: Do not enable compression at all if client doesn't advertise support for ``client_max_window_bits``; the default behavior is to enable compression without enforcing ``client_max_window_bits``. r FNr!r;rgr#)r[r\r]r^r(require_client_max_window_bitsr)cCs|dks&d|krdks&ntd|dksLd|krBdksLntd|dk rdd|krdtd|dkrx|rxtd||_||_||_||_||_||_dS) rhNrr*rirjr+zLcompress_settings must not include wbits, set server_max_window_bits insteadzVrequire_client_max_window_bits is enabled, but client_max_window_bits isn't configured)rkr[r\r]r^r(r~)r4r[r\r]r^r(r~rrrr5s$  z'ServerPerMessageDeflateFactory.__init__rbrmz2tuple[list[ExtensionParameter], PerMessageDeflate]rncstfdd|Dr&tdjt|dd\}}}}jrH|sHd}jrV|sVd}jdkrbn |dkrrj}n|jkrj}jdkr|dkrԈj}n8|dkrjrtdn |dkrĈj}nj|krԈj}t ||||t |||pd|pdj fS) zi Process request parameters. Return response params and an extension instance. c3s|]}|jjkVqdSrprqrrr8rrrt szHServerPerMessageDeflateFactory.process_request_params..zskipped duplicate TruNzrequired client_max_window_bitsr*) rvr rXrfr[r\r]r^r~rarr(rwrr8rprocess_request_paramssX        z5ServerPerMessageDeflateFactory.process_request_params)FFNNNF)rTrUrVrWr rXr5rrrrrrs#z'Sequence[ServerExtensionFactory] | Nonez Sequence[ServerExtensionFactory]cCs>|dkr g}tdd|Ds:t|tddddidg}|S)z Enable Per-Message Deflate with default settings in server extensions. If the extension is already present, perhaps with non-default settings, the configuration isn't changed. Ncss|]}|jtjkVqdSrp)rXr)rZ ext_factoryrrrrtsz3enable_server_permessage_deflate.. rzr{)r]r^r()rvr|rr}rrrrs )! __future__rrEr.typingrrr exceptionsrrr r r r r rbaserrr__all__rGrangerdrrarfrrrrrrrrs2   =13