o DRhU`@sddlmZddlZddlZddlmZmZddlmZddl m Z m Z m Z m Z mZmZddlmZmZdd lmZmZmZgd Zd Zd d eddDZGdddeZd0ddZd1d!d"ZGd#d$d$eZd2d(d)ZGd*d+d+eZd3d.d/Z dS)4) annotationsN)AnySequence)frames)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError) ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescCsg|]}t|qS)str).0bitsrrj/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/websockets/extensions/permessage_deflate.py src@sJeZdZdZedZ ddddZdddZdddddZd ddZ dS)!rz) Per-Message Deflate extension. permessage-deflateNremote_no_context_takeoverboollocal_no_context_takeoverremote_max_window_bitsintlocal_max_window_bitscompress_settingsdict[Any, Any] | NonereturnNonecCs|duri}|dvs J|dvsJd|krdks JJd|kr+dks.JJd|vs4J||_||_||_||_||_|jsOtj|j d|_|js`tjdd|j i|j|_ d|_ dS) z? Configure the Per-Message Deflate extension. N)FTrwbitsr,Fr) r!r#r$r&r'zlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr!r#r$r&r'rrr__init__)s*     zPerMessageDeflate.__init__rc 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__SszPerMessageDeflate.__repr__)max_sizeframe frames.Framer9 int | Nonec Cs|jtjvr|S|jtjur|js|S|jrd|_n|js|Stj|dd}|js,d|_|j r8t j |j d|_ |j}|jrB|t7}|durHdn|}z |j ||}Wnt jye}ztd|d}~ww|j jrrtd|d |jrz|j rz|` 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:r9r@ max_lengthexcrrrdecode\s<    zPerMessageDeflate.decodecCs|jtjvr|S|jtjur&tj|dd}|jr&tjdd|j i|j |_ |j |j |j tj}|jrB|trB|dd}|jrJ|jrJ|` tj||dS)z, Encode an outgoing frame. Tr=r,Nr?r)rArrBrCrErFr#r.r1r&r'r2compressr@flush Z_SYNC_FLUSHrDendswithrG)r4r:r@rrrencodes    zPerMessageDeflate.encodeN) r!r"r#r"r$r%r&r%r'r(r)r*)r)r)r:r;r9r<r)r;)r:r;r)r;) __name__ __module__ __qualname____doc__r namer5r8rMrSrrrrr!s * :rserver_no_context_takeoverr"client_no_context_takeoverserver_max_window_bitsr<client_max_window_bitsint | bool | Noner)list[ExtensionParameter]cCsfg}|r |d|r|d|r|dt|f|dur&|d|S|r1|dt|f|S)zS Build a list of ``(name, value)`` pairs for some compression parameters. )rZN)r[Nr\T)r]Nr])appendr)rZr[r\r]paramsrrr_build_parameterss    rbraSequence[ExtensionParameter] is_server0tuple[bool, bool, int | None, int | bool | None]cCsd}d}d}d}|D]p\}}|dkr$|rt||durd}q t|||dkr:|r.t||dur5d}q t|||dkrT|durFt||tvrOt|}q t|||dkrw|dur`t||ri|durid}q |tvrrt|}q t||t|||||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. FNrZTr[r\r])rr _MAX_WINDOW_BITS_VALUESr%r)rardrZr[r\r]rYvaluerrr_extract_parameterssJ         rhc@sBeZdZdZedZ     ddddZdddZd ddZdS)!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 FNTrZr"r[r\r<r]r^r'dict[str, Any] | Noner)r*cCs|dusd|krdkstdtd|dus0|dus0d|kr+dks0tdtd|dur^zIClientPerMessageDeflateFactory.process_response_params..zreceived duplicate Frdz#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 rYrhrZr[r\r]rr'r4rarorZr[r\r]rr7rprocess_response_paramsSsJ        z6ClientPerMessageDeflateFactory.process_response_params)FFNTN) rZr"r[r"r\r<r]r^r'rir)r*)r)r_)rarcrorpr)r) rUrVrWrXr rYr5rnrzrrrrr s  r extensions'Sequence[ClientExtensionFactory] | None Sequence[ClientExtensionFactory]cCs:|durg}tdd|Dst|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. Ncs|] }|jtjkVqdSrT)rYr)rextension_factoryrrrru  z3enable_client_permessage_deflate..memLevel)r')rxlistrr{rrrrs rc@s:eZdZdZedZ      ddddZdddZdS)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 FNrZr"r[r\r<r]r'rirequire_client_max_window_bitsr)r*cCs|dusd|krdkstdtd|dus,d|kr'dks,tdtd|dur8d|vr8td|durB|rBtd||_||_||_||_||_||_dS) rjNrr+rkrlr,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)rmrZr[r\r]r'r)r4rZr[r\r]r'rrrrr5s,   z'ServerPerMessageDeflateFactory.__init__rarcrorp2tuple[list[ExtensionParameter], PerMessageDeflate]cstfdd|Drtdjt|dd\}}}}jr$|s$d}jr+|s+d}jdur1n|dur9j}n|jkrAj}jdurN|durMj}n|durZjrYtdn|durbj}nj|krjj}t ||||t |||pvd|pydj fS) zi Process request parameters. Return response params and an extension instance. c3rqrTrrrsr7rrru rvzHServerPerMessageDeflateFactory.process_request_params..zskipped duplicate TrwNzrequired client_max_window_bitsr+) rxr rYrhrZr[r\r]rrbrr'ryrr7rprocess_request_paramss\       z5ServerPerMessageDeflateFactory.process_request_params)FFNNNF)rZr"r[r"r\r<r]r<r'rirr"r)r*)rarcrorpr)r)rUrVrWrXr rYr5rrrrrrs #r'Sequence[ServerExtensionFactory] | None Sequence[ServerExtensionFactory]cCs>|durg}tdd|Dst|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. Ncsr~rT)rYr)r ext_factoryrrrrurz3enable_server_permessage_deflate.. rr)r\r]r')rxrrrrrrrs r) rZr"r[r"r\r<r]r^r)r_)rarcrdr"r)re)r{r|r)r})r{rr)r)! __future__rrEr.typingrrr exceptionsrrr r r r r rbaserrr__all__rGrangerfrrbrhrrrrrrrrs*     = 13