U Mf`@sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z mZddlmZmZddlmZmZmZd d d d d gZdZddeddDZGdd d eZddddddddZddddddZGd d d eZd!d"d#d$d ZGd%d d eZd&d'd#d(d ZdS))) annotationsN)AnyDictListOptionalSequenceTupleUnion) exceptionsframes) ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactoryPerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescCsg|] }t|qS)str).0bitsrrL/tmp/pip-unpacked-wheel-yzabpfcc/websockets/extensions/permessage_deflate.py src@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|dkr i}|dkst|dks$td|kr8dks>ntd|krRdksXntd|ksdt||_||_||_||_||_|jstj|j d|_|jstj fd|j i|j|_ d|_ dS)z? Configure the Per-Message Deflate extension. N)FTrwbitsr+F) AssertionErrorr$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__Js$zPerMessageDeflate.__repr__)max_sizez frames.Frame Optional[int])framer:r)c Cs|jtjkr|S|jtjkr4|js&|S|jrpd|_n<|js>|Stj|dd}|jsXd|_|j rpt j |j d|_ |j}|jr|t7}|dkrdn|}z|j ||}Wn0t jk r}ztd|W5d}~XYnX|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|jtjkr|S|jtjk rLtj|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) :r!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|dkrJ|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]zTszIClientPerMessageDeflateFactory.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  z*Optional[Sequence[ClientExtensionFactory]]z Sequence[ClientExtensionFactory]) extensionsr)cCs:|dkr 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 c @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|dks&d|krdks&ntd|dksLd|krBdksLntd|dk rdd|krdtd|dkrx|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}jdkrdn |dkrtj}n|jkrj}jdkr|dkr؈j}n:|dkrj rtdn |dkrȈj}nj|kr؈j}t ||||t |||pd|pdj fS) zi Process request parameters. Return response params and an extension instance. c3s|]}|jjkVqdSrqrrrsr8rrruszHServerPerMessageDeflateFactory.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#z*Optional[Sequence[ServerExtensionFactory]]z Sequence[ServerExtensionFactory]cCs>|dkr 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 ) __future__rrEr.typingrrrrrrr r r r rbaserrr__all__rGrangererrbrgrrrrrrrrs0 $=13