o _id@sddlmZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZddlmZmZmZd d lmZmZmZgd Zd Zd deddDZGdddeZd1ddZd2d"d#ZGd$d%d%eZd3d)d*Z Gd+d,d,eZ!d4d/d0Z"dS)5) annotationsN)Sequence)AnyLiteral)frames)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError) BytesLike ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescCsg|]}t|qS)str).0bitsrrj/sda-disk/www/egybert/egybert_env/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)FTr wbitsr.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(r6rrr__repr__SszPerMessageDeflate.__repr__)max_sizeframe frames.Framer; int | Nonec Cs>|jtjvr|S|jtjur|js|S|jrd|_n|js|S|js%d|_|jr1tj |j d|_ |jrCt |j dkrCt|j t}n|j }|durLdn|}z(|j ||}|j jre|dus`Jtd||jrut |j dkru|j tWntjy}ztd|d}~ww|jr|jr|` t|j||jd|j|jS)z, Decode an incoming frame. FTr/iNrzdecompression failed)opcoder CTRL_OPCODESOP_CONTr5finrsv1r#r0r1r&r2lendatabytes_EMPTY_UNCOMPRESSED_BLOCK decompressunconsumed_tailr errorr Framersv2rsv3)r6r<r;rE max_lengthexcrrrdecode\sP       zPerMessageDeflate.decodecCs|jtjvr|S|jtjur|jrtjdd|j i|j|_ |j |j |j tj }|jrO|ddtks:Jt|dkrG|dd}nt|dd}|jrW|jrW|` t|j||j|jtju|j|jS)z, Encode an outgoing frame. r.Nir)r?rr@rAr%r0r3r(r)r4compressrEflush Z_SYNC_FLUSHrBrGrD memoryviewrKrLrM)r6r<rErrrencodes2     zPerMessageDeflate.encodeN) r#r$r%r$r&r'r(r'r)r*r+r,)r+r)r<r=r;r>r+r=)r<r=r+r=) __name__ __module__ __qualname____doc__rnamer7r:rPrVrrrrr!s * Irserver_no_context_takeoverr$client_no_context_takeoverserver_max_window_bitsr>client_max_window_bitsint | Literal[True] | 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. )r]N)r^Nr_T)r`Nr`)appendr)r]r^r_r`paramsrrr_build_parameterss    rerdSequence[ExtensionParameter] is_server9tuple[bool, bool, int | None, int | Literal[True] | 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. FNr]Tr^r_r`)rr _MAX_WINDOW_BITS_VALUESr'r )rdrgr]r^r_r`r\valuerrr_extract_parameterssJ         rkc@sBeZdZdZedZ     ddddZdddZdddZdS) 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$r^r_r>r`rar)dict[str, Any] | Noner+r,cCs|dusd|krdkstdtd|dus0|dus0d|kr+dks0tdtd|dur{zIClientPerMessageDeflateFactory.process_response_params..zreceived duplicate Frgz#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 r\rkr]r^r_r`rr)r6rdrrr]r^r_r`rr9rprocess_response_paramspsJ        z6ClientPerMessageDeflateFactory.process_response_params)FFNTN) r]r$r^r$r_r>r`rar)rlr+r,)r+rf)rdrfrrrsr+r) rXrYrZr[rr\r7rqr}rrrrr*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|] }|jtjkVqdSrW)r\r)rextension_factoryrrrrx  z3enable_client_permessage_deflate..memLevel)r))r{listrr~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"FNr]r$r^r_r>r`r)rlrequire_client_max_window_bitsr+r,cCs|dusd|krdkstdtd|dus,d|kr'dks,tdtd|dur8d|vr8td|durB|rBtd||_||_||_||_||_||_dS) rmNr r-rnror.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)rpr]r^r_r`r)r)r6r]r^r_r`r)rrrrr7s,   z'ServerPerMessageDeflateFactory.__init__rdrfrrrs2tuple[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. c3rtrWrurvr9rrrx=ryzHServerPerMessageDeflateFactory.process_request_params..zskipped duplicate TrzNzrequired client_max_window_bitsr-) r{r r\rkr]r^r_r`rrerr)r|rr9rprocess_request_params2s\       z5ServerPerMessageDeflateFactory.process_request_params)FFNNNF)r]r$r^r$r_r>r`r>r)rlrr$r+r,)rdrfrrrsr+r)rXrYrZr[rr\r7rrrrrrs #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. NcsrrW)r\r)r ext_factoryrrrrxrz3enable_server_permessage_deflate.. rr)r_r`r))r{rrrrrrrs r) r]r$r^r$r_r>r`rar+rb)rdrfrgr$r+rh)r~rr+r)r~rr+r)# __future__rr0collections.abcrtypingrrr exceptionsrr r r r r rrrbaserrr__all__rGrangerirrerkrrrrrrrrs*     5 = 13