jj>ddlmZddlZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZddlmZmZmZddlmZmZmZd d lmZmZmZgd Zd Zd eddDZGddeZd,dZd-d Z Gd!d"eZ!d.d&Z"Gd'd(eZ#d/d+Z$dS)0) annotationsN)Sequence)AnyLiteral)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError)CONT CTRL_OPCODESFrame) BytesLike ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatesc,g|]}t|S)str).0bitss XC:\Projects\Muse2API\.venv\Lib\site-packages\websockets/extensions/permessage_deflate.py r#s>>>3t99>>>cReZdZdZedZ dddZddZddddZddZ dS)rz) Per-Message Deflate extension. permessage-deflateNremote_no_context_takeoverboollocal_no_context_takeoverremote_max_window_bitsintlocal_max_window_bitscompress_settingsdict[Any, Any] | NonereturnNonecl|i}|dvsJ|dvsJd|cxkrdksnJd|cxkrdksnJd|vsJ||_||_||_||_||_|js t j|j |_|js#t jd d|j i|j|_ d|_ dS) z? Configure the Per-Message Deflate extension. N)FTr% wbitsr6Fr) r)r+r,r.r/zlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr)r+r,r.r/s r"__init__zPerMessageDeflate.__init__)s.  $ " )]::::(M9999*0000b000000)////R///////////*D')B&&<#%:"!2. R-T5P4PQQQDL- +11(DL!&r$rc Hd|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.r>s r"__repr__zPerMessageDeflate.__repr__Vs_ C*.*I C C)-)G C C'+&A C C&*%?  C C C r$)max_sizeframerrD int | Nonecv|jtvr|S|jtur.|js|S|jrt d|jrd|_n>|js|S|jsd|_|jr tj |j |_ |jr5t|j dkrt|j tz}n|j }|dn|} |j ||}|j jrt%d||jrHt|j dkr0|j tdrt%d|n'#tj$r}t d |d}~wwxYw|jr |jr|` t)|j||jd|j|jS) z, Decode an incoming frame. z"RSV1 bit set in continuation frameFTr7iNrrzdecompression failed)opcoderrr=rsv1r finr)r8r9r,r:lendatabytes_EMPTY_UNCOMPRESSED_BLOCK decompressunconsumed_tailr errorrrsv2rsv3)r>rErDrL max_lengthexcs r"decodezPerMessageDeflate.decode_s << ' 'L <4  (  z J#$HIIIy .(-%:  9 -(,%. V#19T8TUUU 9 UZ4//$$'@@DD:D"*QQ  A<**4<rErLs r"encodezPerMessageDeflate.encodes= << ' 'L rdrerfrgr/s r"r?z'ClientPerMessageDeflateFactory.__init__Ks'.!7M2S2S2S2SQS2S2S2S2SNOO O " *%--*0000b0000NOO O  (W8I-I-I5  +E'*D'&<#&<#!2r$rncNt|j|j|j|jS)z, Build request parameters. )rmrdrerfrgrBs r"get_request_paramsz1ClientPerMessageDeflateFactory.get_request_paramsks. !  +  +  '  '    r$rlaccepted_extensionsSequence[Extension]rc0tfd|Drtdjt|d\}}}}jr|stdjr|sd}jn+|td|jkrtd j|td n.jdurn$|j}n|jkrtd |d krtd t|||pd|pdj S)zV Process response parameters. Return an extension instance. c38K|]}|jjkVdSr^rcr otherr>s r" zIClientPerMessageDeflateFactory.process_response_params..,HH5uzTY&HHHHHHr$zreceived duplicate Froz#expected server_no_context_takeoverTNzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr%z-unsupported client_max_window_bits = 8, use 9r4) anyr rcrtrdrerfrgrr/r>rlr|rdrerfrgs` r"process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramsws HHHH4GHHH H H F"#D#D#DEE E % 8 8 8  & & " "  * N- N&'LMMM  * 2- 2-1*  & . &-&'HIII'$*EEE&'KLLL  & .%1&'JKKK2 (D 0 0 &-)-)D&&'$*EEE&'KLLL !Q & &"#RSS S & & " (b " (b  "    r$)FFNTN) rdr*rer*rfrFrgrhr/rvr1r2)r1rn)rlrnr|r}r1r) r_r`rarbrrcr?r{rrr$r"rr/s2 =- . .D,1+0-1=A37 33333@     j j j j j j r$r extensions'Sequence[ClientExtensionFactory] | None Sequence[ClientExtensionFactory]c|g}td|Ds#t|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. Nc3@K|]}|jtjkVdSr^)rcr)r extension_factorys r"rz3enable_client_permessage_deflate..sB  "@"EEr$memLevel)r/)rlistrrs r"rrsw !+   *%% *#-q/   )  r$cFeZdZdZedZ dddZddZdS)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(FNrdr*rerfrFrgr/rvrequire_client_max_window_bitsr1r2c*|d|cxkrdksntd|d|cxkrdksntd|d|vrtd||rtd ||_||_||_||_||_||_dS) rxNr5r4z/server_max_window_bits must be between 9 and 15r%z/client_max_window_bits must be between 8 and 15r6zLcompress_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)ryrdrerfrgr/r)r>rdrerfrgr/rs r"r?z'ServerPerMessageDeflateFactory.__init__s'.!7M2S2S2S2SQS2S2S2S2SNOO O&.!7M2S2S2S2SQS2S2S2S2SNOO O  (W8I-I-I5  " ).L )>  +E'*D'&<#&<#!2.L+++r$rlrnr|r}2tuple[list[ExtensionParameter], PerMessageDeflate]c"tfd|Drtdjt|d\}}}}jr|sd}jr|sd}jn|j}n|jkrj}|dkrtdj |durj}n7|jrtdn|durj}nj|krj}t||||t|||pd |pd j fS) zi Process request parameters. Return response params and an extension instance. c38K|]}|jjkVdSr^rrs r"rzHServerPerMessageDeflateFactory.process_request_params..Jrr$zskipped duplicate TrNr%z-unsupported server_max_window_bits = 8, use 9zrequired client_max_window_bitsr4) rr rcrtrdrerfrgrrmrr/rs` r"process_request_paramsz5ServerPerMessageDeflateFactory.process_request_params?s HHHH4GHHH H H E"#C #C#CDD D $ 7 7 7  & & " ""  * 2- 2-1*  * 2- 2-1*  & . &-)-)D&&'$*EEE)-)D& !Q & &"#RSS S  & .%--)-)D&&-6N*+LMMMN'4//)-)D&&,/EEE)-)D& **&&    **&,"&,"&     r$)FFNNNF)rdr*rer*rfrFrgrFr/rvrr*r1r2)rlrnr|r}r1r)r_r`rarbrrcr?rrr$r"rrs}: =- . .D,1+0-1-137/4!M!M!M!M!MFr r r r r r r$r'Sequence[ServerExtensionFactory] | None Sequence[ServerExtensionFactory]c|g}td|Ds%t|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. Nc3@K|]}|jtjkVdSr^)rcr)r ext_factorys r"rz3enable_server_permessage_deflate..sB  :??r$ rr)rfrgr/)rrrrs r"rrs} %    *%% *')')#-q/   )   r$) rdr*rer*rfrFrgrhr1ri)rlrnror*r1rp)rrr1r)rrr1r)% __future__rr8collections.abcrtypingrr exceptionsrr r r r r framesrrrrrrbaserrr__all__rNrangerrrrmrtrrrrrr$r"rs#"""""" $$$$$$/.........AAAAAAAAAAKKKKKKKKKK   0>>q">>>v v v v v v v v r0::::zr r r r r %;r r r j0u u u u u %;u u u pr$