jd>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 r/workspace/show-city-briefing-agent/.venv/lib/python3.11/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 jdd|j i|j|_ d|_ dS) z? Configure the Per-Message Deflate extension. N)FTr%wbitsr5Fr) 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__Ss_ C*.*I C C)-)G C C'+&A C C&*%?  C C C r$)max_sizeframerrC 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 frameFTr6iNrrzdecompression failed)opcoderrr<rsv1r finr)r7r8r,r9lendatabytes_EMPTY_UNCOMPRESSED_BLOCK decompressunconsumed_tailr errorrrsv2rsv3)r=rDrCrK max_lengthexcs r"decodezPerMessageDeflate.decode\s << ' 'L <4  (  z J#$HIIIy .(-%:  9 -(,%. V#19T8TUUU 9 UZ4//$$'@@DD:D"*QQ  A<**4<rBrUr\rr$r"rr!s =- . .D48 &&&&&&&&&&T     $ I I I I I I V- - - - - - r$rserver_no_context_takeoverr*client_no_context_takeoverserver_max_window_bitsrEclient_max_window_bitsint | Literal[True] | Noner1list[ExtensionParameter]c2g}|r|d|r|d|r$|dt|f|dur|dn&|r$|dt|f|S)zS Build a list of ``(name, value)`` pairs for some compression parameters. )rcN)rdNreT)rfNrf)appendr)rcrdrerfparamss r"_build_parametersrls(*F!< :;;;!< :;;;O /5K1L1LMNNN%% 67777 O /5K1L1LMNNN Mr$rkSequence[ExtensionParameter] is_server9tuple[bool, bool, int | None, int | Literal[True] | None]cd}d}d}d}|D]\}}|dkr&|rt||d}!t|||dkr&|rt||d}Mt|||dkr:|t||tvrt|}t|||dkrA|t||r|d}|tvrt|}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. FNrcTrdrerf)rr _MAX_WINDOW_BITS_VALUESr-r )rkrnrcrdrerfrbvalues r"_extract_parametersrss(-',)-9=$-$- e / / /) /(...}-1**+D%888 1 1 1) /(...}-1**+D%888 - - -%1(...///),U&&+D%888 - - -%1(... 9U])-&&111),U&&+D%888't,, , #"  r$cLeZdZdZedZ dddZddZddZdS)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(FNTrcr*rdrerErfrgr/dict[str, Any] | Noner1r2c|d|cxkrdksntd|"|dusd|cxkrdksntd|d|vrtd||_||_||_||_||_dS) G Configure the Per-Message Deflate extension factory. Nr%r4/server_max_window_bits must be between 8 and 15T/client_max_window_bits must be between 8 and 15r5zLcompress_settings must not include wbits, set client_max_window_bits instead) ValueErrorrcrdrerfr/)r=rcrdrerfr/s r"r>z'ClientPerMessageDeflateFactory.__init__Fs'.!7M2S2S2S2SQS2S2S2S2SNOO O " *%--*0000b0000NOO O  (W8I-I-I5  +E'*D'&<#&<#!2r$rmcNt|j|j|j|jS)z, Build request parameters. )rlrcrdrerfrAs r"get_request_paramsz1ClientPerMessageDeflateFactory.get_request_paramsfs. !  +  +  '  '    r$rkaccepted_extensionsSequence[Extension]rctfd|Drtdjt|d\}}}}jr|stdjr|sd}jn+|td|jkrtd j|td n.jdurn$|j}n|jkrtd t|||pd |pd j S) zV Process response parameters. Return an extension instance. c38K|]}|jjkVdSr]rbr otherr=s r" zIClientPerMessageDeflateFactory.process_response_params..},HH5uzTY&HHHHHHr$zreceived duplicate Frnz#expected server_no_context_takeoverTNzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr4) anyr rbrsrcrdrerfrr/r=rkr}rcrdrerfs` r"process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramsrs 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 & & " (b " (b  "    r$)FFNTN) rcr*rdr*rerErfrgr/rur1r2)r1rm)rkrmr}r~r1r) r^r_r`rarrbr>r|rrr$r"rr,s. =- . .D,1+0-1=A37 33333@     g g g g g g 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])rbr)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 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(FNrcr*rdrerErfr/rurequire_client_max_window_bitsr1r2c*|d|cxkrdksntd|d|cxkrdksntd|d|vrtd||rtd||_||_||_||_||_||_dS) rwNr%r4rxryr5zLcompress_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)rzrcrdrerfr/r)r=rcrdrerfr/rs r"r>z'ServerPerMessageDeflateFactory.__init__s'.!7M2S2S2S2SQS2S2S2S2SNOO O&.!7M2S2S2S2SQS2S2S2S2SNOO O  (W8I-I-I5  " ).L )>  +E'*D'&<#&<#!2.L+++r$rkrmr}r~2tuple[list[ExtensionParameter], PerMessageDeflate]ctfd|Drtdjt|d\}}}}jr|sd}jr|sd}jn|j}n|jkrj}j |durj}n7|jrtdn|durj}nj|krj}t||||t|||pd|pdj fS)zi Process request parameters. Return response params and an extension instance. c38K|]}|jjkVdSr]rrs r"rzHServerPerMessageDeflateFactory.process_request_params..?rr$zskipped duplicate TrNzrequired client_max_window_bitsr4) rr rbrsrcrdrerfrrlrr/rs` r"process_request_paramsz5ServerPerMessageDeflateFactory.process_request_params4s HHHH4GHHH H H E"#C #C#CDD D $ 7 7 7  & & " ""  * 2- 2-1*  * 2- 2-1*  & . &-)-)D&&'$*EEE)-)D&  & .%--)-)D&&-6N*+LMMMN'4//)-)D&&,/EEE)-)D& **&&    **&,"&,"&     r$)FFNNNF)rcr*rdr*rerErfrEr/rurr*r1r2)rkrmr}r~r1r)r^r_r`rarrbr>rrr$r"rrs}4 =- . .D,1+0-1-137/4!M!M!M!M!MFo o o o o o 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])rbr)r ext_factorys r"rz3enable_server_permessage_deflate..sB  :??r$ rr)rerfr/)rrrrs r"rrs} %    *%% *')')#-q/   )   r$) rcr*rdr*rerErfrgr1rh)rkrmrnr*r1ro)rrr1r)rrr1r)% __future__rr7collections.abcrtypingrr exceptionsrr r r r r framesrrrrrrbaserrr__all__rMrangerqrrlrsrrrrrr$r"rs#"""""" $$$$$$/.........AAAAAAAAAAKKKKKKKKKK   0>>q">>>s s s s s s s s l0::::zm m m m m %;m m m `0o o o o o %;o o o dr$