a یxd`@sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z mZddlmZmZddlmZmZmZgd Zd Zd d ed dDZGdddeZddddddddZddddddZGdddeZdd d!d"d#ZGd$d%d%eZd&d'd!d(d)ZdS)*) annotationsN)AnyDictListOptionalSequenceTupleUnion) exceptionsframes) ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescCsg|] }t|qS)str).0bitsrr`/var/www/html/Ranjet/env/lib/python3.9/site-packages/websockets/extensions/permessage_deflate.py rc@sheZdZdZedZddddddddd d Zd d d dZddddddddZdddddZ dS)rz) Per-Message Deflate extension. permessage-deflateNboolintzOptional[Dict[Any, Any]]None)remote_no_context_takeoverlocal_no_context_takeoverremote_max_window_bitslocal_max_window_bitscompress_settingsreturncCs|dur i}|dvsJ|dvs$Jd|kr8dks>nJd|krRdksXnJd|vsdJ||_||_||_||_||_|jstj|j d|_|jstjfd|j i|j|_ d|_ dS)z? Configure the Per-Message Deflate extension. N)FTrwbitsr,F) r%r&r'r(r)zlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr%r&r'r(r)rrr__init__!s*    zPerMessageDeflate.__init__rr*c 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(r4rrr__repr__JszPerMessageDeflate.__repr__)max_sizez frames.Frame Optional[int])framer:r*c Cs|jtjvr|S|jtjur4|js&|S|jrpd|_n<|js>|Stj|dd}|jsXd|_|j rpt j |j d|_ |j}|jr|t7}|durdn|}z|j ||}Wn2t jy}ztd|WYd}~n d}~00|j jrtd|d |jr|j r|` tj||d S) z, Decode an incoming frame. Frsv1Tr-Nrzdecompression failedzover size limit (? > z bytes)data)opcoder CTRL_OPCODESOP_CONTr3finr> dataclassesreplacer%r.r/r'r0r@_EMPTY_UNCOMPRESSED_BLOCK decompresserrorr ProtocolErrorunconsumed_tailZ PayloadTooBig)r4r<r:r@ max_lengthexcrrrdecodeSs6  "zPerMessageDeflate.decode)r<r*cCs|jtjvr|S|jtjurLtj|dd}|jrLtjfd|j i|j |_ |j |j |j tj}|jr|tr|dd}|jr|jr|` tj||dS)z, Encode an outgoing frame. Tr=r,Nr?)rAr rBrCrErFr&r.r1r(r)r2compressr@flush Z_SYNC_FLUSHrDendswithrG)r4r<r@rrrencodes     zPerMessageDeflate.encode)N) __name__ __module__ __qualname____doc__r namer5r9rNrTrrrrrs) :rr"r;Optional[Union[int, bool]]List[ExtensionParameter])server_no_context_takeoverclient_no_context_takeoverserver_max_window_bitsclient_max_window_bitsr*cCsdg}|r|d|r |d|r6|dt|f|durJ|dn|r`|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    rbSequence[ExtensionParameter]zTrzIClientPerMessageDeflateFactory.process_response_params..zreceived duplicate Frdz#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 NegotiationErrorrYrgr\r]r^r_rr)r4rarpr\r]r^r_rr8rprocess_response_paramsIsH            z6ClientPerMessageDeflateFactory.process_response_params)FFNTN) rUrVrWrXr rYr5rmrzrrrrrs  rz*Optional[Sequence[ClientExtensionFactory]]z Sequence[ClientExtensionFactory]) extensionsr*cCs:|dur g}tdd|Ds6t|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. Ncss|]}|jtjkVqdSrq)rYr)rZextension_factoryrrrrusz3enable_client_permessage_deflate..memLevel)r))rwlistrr{rrrrs rc @sFeZdZdZedZddddddddd d d Zd d ddddZdS)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://www.rfc-editor.org/rfc/rfc7692.html#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;rhr$)r\r]r^r_r)require_client_max_window_bitsr*cCs|dus&d|krdks&ntd|dusLd|krBdksLntd|durdd|vrdtd|durx|rxtd||_||_||_||_||_||_dS) riNrr+rjrkr,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)rlr\r]r^r_r)r)r4r\r]r^r_r)rrrrr5s$  z'ServerPerMessageDeflateFactory.__init__rcrnz2Tuple[List[ExtensionParameter], PerMessageDeflate]rocstfdd|Dr(tdjt|dd\}}}}jrJ|sJd}jrX|sXd}jdurdn |durtj}n|jkrj}jdur|dur؈j}n:|durj rtdn |durȈj}nj|kr؈j}t ||||t |||pd|pdj fS) zi Process request parameters. Return response params and an extension instance. c3s|]}|jjkVqdSrqrrrsr8rrrurzHServerPerMessageDeflateFactory.process_request_params..zskipped duplicate TrvNzrequired client_max_window_bitsr+) rwr rxrYrgr\r]r^r_rrbrr)ryrr8rprocess_request_params sX        z5ServerPerMessageDeflateFactory.process_request_params)FFNNNF)rUrVrWrXr rYr5rrrrrrs#rz*Optional[Sequence[ServerExtensionFactory]]z Sequence[ServerExtensionFactory]cCs>|dur g}tdd|Ds:t|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. Ncss|]}|jtjkVqdSrq)rYr)rZ ext_factoryrrrrusz3enable_server_permessage_deflate.. r|r})r^r_r))rwr~rrrrrr}s r) __future__rrEr.typingrrrrrrr r r r rbaserrr__all__rGrangererrbrgrrrrrrrrs& $=13