tnid6ddlmZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZddlmZmZmZd d lmZmZmZgd Zd Zd eddDZGddeZd,dZd-d ZGd!d"eZd.d&Z Gd'd(eZ!d/d+Z"dS)0) annotationsN)Sequence)AnyLiteral)frames)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError) BytesLike ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatesc,g|]}t|S)str).0bitss e/var/www/html/gpu-tools/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#wbitsr3Fr) 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_sizeframe frames.FramerA int | NonecN|jtjvr|S|jtjur|js|S|jrd|_n>|js|S|jsd|_|jr tj |j |_ |jr5t|j dkrt|j tz}n|j }|dn|} |j ||}|j jr|Jt%d||jr7t|j dkr|j tn'#tj$r}t)d|d}~wwxYw|jr |jr|` tj|j||jd|j|jS)z, Decode an incoming frame. FTr4iNrzdecompression failed)opcoder CTRL_OPCODESOP_CONTr:finrsv1r'r5r6r*r7lendatabytes_EMPTY_UNCOMPRESSED_BLOCK decompressunconsumed_tailr errorrFramersv2rsv3)r;rBrArL max_lengthexcs r decodezPerMessageDeflate.decode\s <6. . .L <6> ) )(  y .(-%:  9 -(,%. V#19T8TUUU 9 UZ4//$$'@@DD:D"*QQ  A<**4<zIClientPerMessageDeflateFactory.process_response_params..{,HH5uzTY&HHHHHHr"zreceived duplicate Frpz#expected server_no_context_takeoverTNzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr2) anyr rdrurerfrgrhrr-r;rmrrerfrgrhs` r process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramsps 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) rer(rfr(rgrDrhrir-rwr/r0)r/ro)rmrorrr/r) r`rarbrcrrdr<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_)rdr)rextension_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&FNrer(rfrgrDrhr-rwrequire_client_max_window_bitsr/r0c*|d|cxkrdksntd|d|cxkrdksntd|d|vrtd||rtd||_||_||_||_||_||_dS) ryNr#r2rzr{r3zLcompress_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)r|rerfrgrhr-r)r;rerfrgrhr-rs r r<z'ServerPerMessageDeflateFactory.__init__s'.!7M2S2S2S2SQS2S2S2S2SNOO O&.!7M2S2S2S2SQS2S2S2S2SNOO O  (W8I-I-I5  " ).L )>  +E'*D'&<#&<#!2.L+++r"rmrorr2tuple[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_bitsr2) rr rdrurerfrgrhrrnrr-rs` r process_request_paramsz5ServerPerMessageDeflateFactory.process_request_params2s 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)rer(rfr(rgrDrhrDr-rwrr(r/r0)rmrorrr/r)r`rarbrcrrdr<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_)rdr)r ext_factorys r rz3enable_server_permessage_deflate..sB  :??r" rr)rgrhr-)rrrrs r rrs} %    *%% *')')#-q/   )   r") rer(rfr(rgrDrhrir/rj)rmrorpr(r/rq)rrr/r)rrr/r)# __future__rr5collections.abcrtypingrrr exceptionsr r r r r rrrrbaserrr__all__rNrangersrrnrurrrrrr"r rs"""""" $$$$$$BAAAAAAAAAKKKKKKKKKK   0>>q">>>q q q q q q q q h0::::zm m m m m %;m m m `0o o o o o %;o o o dr"