jgrddlmZddlZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZddlmZmZmZddlmZmZmZd d lmZmZmZgd Zd Zed dDcgc] }e| c}ZGddeZ ddZ ddZ!GddeZ" ddZ#GddeZ$ ddZ%ycc}w)) annotationsN)Sequence)AnyLiteral)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError)CONT CTRL_OPCODESFrame) BytesLike ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescjeZdZdZedZ d d dZd dZdd d dZd dZ y)rz) Per-Message Deflate extension. permessage-deflateNc|i}|dvsJ|dvsJd|cxkrdksJJd|cxkrdksJJd|vsJ||_||_||_||_||_|js&t j |j |_|js1t jd d|j i|j|_ d|_ y) z? Configure the Per-Message Deflate extension. N)FTr wbitsr$F) remote_no_context_takeoverlocal_no_context_takeoverremote_max_window_bitslocal_max_window_bitscompress_settingszlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr'r(r)r*r+s u/home/oem/CyberBrain26-unified/dev-env/.venv/lib/python3.12/site-packages/websockets/extensions/permessage_deflate.py__init__zPerMessageDeflate.__init__)s  $ " )]:::(M999*0b00000)/R/////////*D')B&&<#%:"!2..--T5P5P4PQDL--++111((DL!&c pd|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*r2s r3__repr__zPerMessageDeflate.__repr__VsV**.*I*I)JK))-)G)G(HI&&*&A&A%BC%%)%?%?$@  C r5)max_sizec|jtvr|S|jtur9|js|S|jr t d|j r[d|_nS|js|S|j sd|_|jr&tj|j |_ |j r5t|jdkrt|jtz}n |j}|dn|} |jj!||}|jj"r t%d||j rDt|jdk\r,|jj!tdr t%d||j r|jr|` t)|j||j d|j*|j,S#tj&$r}t d |d}~wwxYw) z, Decode an incoming frame. z"RSV1 bit set in continuation frameFTr%iNrrzdecompression failed)opcoderrr1rsv1r finr'r,r-r)r.lendatabytes_EMPTY_UNCOMPRESSED_BLOCK decompressunconsumed_tailr errorrrsv2rsv3)r2framer:r@ max_lengthexcs r3decodezPerMessageDeflate.decode_s <<< 'L <<4 (( zz#$HIIyy(-%:: 99(,%..#119T9T8TU 99UZZ4/$'@@D::D"*Q  A<<**4rBr? memoryviewrrFrG)r2rHr@s r3encodezPerMessageDeflate.encodes% <<< 'L <tfd|Drtdjt|d\}}}}jr |s tdj r|sd}j n'| td|j kDr tdj |Ctd jdurn)| j}n|jkDr td |d k(r td t|||xsd |xsd jS)zV Process response parameters. Return an extension instance. c3PK|]}|jjk(ywrSr^.0otherr2s r3 zIClientPerMessageDeflateFactory.process_response_params..H5uzzTYY&H#&zreceived duplicate Friz#expected server_no_context_takeoverTzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsrz-unsupported client_max_window_bits = 8, use 9r") anyr r^rkr`rarbrcrr+r2reaccepted_extensionsr`rarbrcs` r3process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramswsT H4GH H"%8 #DE E % 8  & & " "  * *-&'LMM  * *--1*  & & . &-&'HII'$*E*EE&'KLL  & & .%1&'JKK  ( (D 0 &-)-)D)D&'$*E*EE&'KLL !Q &"#RS S & & " (b " (b  " "   r5)FFNTN) r`rTrarTrbrYrcint | Literal[True] | Noner+dict[str, Any] | NonerVrW)rVSequence[ExtensionParameter])rerr~Sequence[Extension]rVr) rZr[r\r]rr^r4rqrr&r5r3rr/s2 - .D,1+0-1=A37 3$(3%)3!+ 3 !; 3 1 3 3@  j ,j 1j   j r5rch|g}td|Dst|tddigz}|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. c3VK|]!}|jtjk(#ywrS)r^r)rvextension_factorys r3rxz3enable_client_permessage_deflate..s*  "@"E"EE')memLevel)r+)r|listr extensionss r3rrsQ !+ *% *#-q/ )  r5cdeZdZdZedZ d ddZ ddZy) 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 9 and 15. 8 isn't supported due to a `limitation of zlib`_. 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``. .. _limitation of zlib: https://github.com/madler/zlib/issues/171 r Nc(|"d|cxkrdkstdtd|"d|cxkrdkstdtd|d|vr td| |r td ||_||_||_||_||_||_y) rnNr#r"z/server_max_window_bits must be between 9 and 15rz/client_max_window_bits must be between 8 and 15r$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)ror`rarbrcr+require_client_max_window_bits)r2r`rarbrcr+rs r3r4z'ServerPerMessageDeflateFactory.__init__s'.!7M2SQS2SNO O3TNO O&.!7M2SQS2SNO O3TNO O  (W8I-I5  " ).L>  +E'*D'&<#&<#!2.L+r5cptfd|Drtdjt|d\}}}}jr|sd}j r|sd}j n*| j }n|j kDr j }|dk(r tdj|durRj}nE|jr7td|dur j}nj|kr j}t||||t|||xsd|xsdjfS) zi Process request parameters. Return response params and an extension instance. c3PK|]}|jjk(ywrSrtrus r3rxzHServerPerMessageDeflateFactory.process_request_params..Jryrzzskipped duplicate Tr{rz-unsupported server_max_window_bits = 8, use 9zrequired client_max_window_bitsr") r|r r^rkr`rarbrcrrfrr+r}s` r3process_request_paramsz5ServerPerMessageDeflateFactory.process_request_params?sz H4GH H"%7 {#CD D $ 7  & & " ""  * *--1*  * *--1*  & & . &-)-)D)D&'$*E*EE)-)D)D& !Q &"#RS S  & & .%-)-)D)D&&-66*+LMM'4/)-)D)D&,,/EE)-)D)D& **&&   **&,"&,"&&    r5)FFNNNF)r`rTrarTrbrYrcrYr+rrrTrVrW)rerr~rrVz2tuple[list[ExtensionParameter], PerMessageDeflate])rZr[r\r]rr^r4rr&r5r3rrs: - .D,1+0-1-137/4!M$(!M%)!M!+ !M !+ !M 1 !M)-!M !MFr ,r 1r  < r r5rcl|g}td|Dst|tddddigz}|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. c3VK|]!}|jtjk(#ywrS)r^r)rv ext_factorys r3rxz3enable_server_permessage_deflate..s*  :???r rr)rbrcr+)r|rrrs r3rrsW % *% *')')#-q/ )   r5) r`rTrarTrbrYrcrrVzlist[ExtensionParameter])rerrirTrVz9tuple[bool, bool, int | None, int | Literal[True] | None])rz'Sequence[ClientExtensionFactory] | NonerVz Sequence[ClientExtensionFactory])rz'Sequence[ServerExtensionFactory] | NonerVz Sequence[ServerExtensionFactory])& __future__rr,collections.abcrtypingrr exceptionsrr r r r r framesrrrrrrbaserrr__all__rBrangerXrhrrfrkrrrr)bitss0r3rs " $/.AAKK 016q">3t9>v v r $ $'7   0: (:8<:>:zr %;r j7%0u %;u p7%m?sB4