a j(g@sddlmZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZddlmZmZd d lmZmZmZgd Zd Zd deddDZGdddeZddddddddZddddddZGdd d eZd!d"d#d$d%ZGd&d'd'eZ d(d)d#d*d+Z!dS),) annotationsN)Sequence)AnyLiteral)frames)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError) ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescCsg|] }t|qS)str).0bitsrrNE:\soccer\ml_env\Lib\site-packages\websockets/extensions/permessage_deflate.py rc@sheZdZdZedZddddddddd d Zd d d dZddddddddZdddddZ dS)rz) Per-Message Deflate extension. permessage-deflateNboolintzdict[Any, Any] | NoneNone)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)FTr wbitsr-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)r5rrr__repr__SszPerMessageDeflate.__repr__)max_sizez frames.Frame int | None)framer;r+c CsP|jtjvr|S|jtjur4|js&|S|jrbd|_n.|js>|S|jsJd|_|jrbtj |j d|_ |jrt |j dkrt|j t}n|j }|durdn|}zP|j ||}|j jr|dusJtd||jrt |j dkr|j tWn2tjy}ztd|WYd}~n d}~00|jr4|jr4|` t|j||jd|j|jS)z, Decode an incoming frame. FTr.iNrzdecompression failed)opcoder CTRL_OPCODESOP_CONTr4finZrsv1r&r/r0r(r1lendatabytes_EMPTY_UNCOMPRESSED_BLOCK decompressunconsumed_tailr errorr Framersv2rsv3)r5r=r;rC max_lengthexcrrrdecode\sH     zPerMessageDeflate.decode)r=r+cCs|jtjvr|S|jtjur>|jr>tjfd|j i|j|_ |j |j |j tj }|jr|ddtkstJt|dkr|dd}nt|dd}|jr|jr|` t|j||j|jtju|j|jS)z, Encode an outgoing frame. r-Ni)r>rr?r@r'r/r2r)r*r3compressrCflush Z_SYNC_FLUSHrArErB memoryviewrIrJrK)r5r=rCrrrencodes2     zPerMessageDeflate.encode)N) __name__ __module__ __qualname____doc__rnamer6r:rNrTrrrrr!s* Hrr#r<int | Literal[True] | Nonezlist[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    raSequence[ExtensionParameter]z9tuple[bool, bool, int | None, int | Literal[True] | None])r` is_serverr+cCs d}d}d}d}|D]\}}|dkrJ|r0t||dur>d}qt||q|dkrx|r^t||durld}qt||q|dkr|durt||tvrt|}qt||q|dkr|durt||r|durd}q|tvrt|}qt||qt|q||||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. FNr[Tr\r]r^)rr _MAX_WINDOW_BITS_VALUESr$r )r`rcr[r\r]r^rYvaluerrr_extract_parameterssJ          rfc@sReZdZdZedZdddddd d d d d ZddddZddddddZdS)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"FNTr#r<rZdict[str, Any] | Noner%)r[r\r]r^r*r+cCs|dus&d|krdks&ntd|dusT|dusTd|krJdksTntd|durld|vrltd||_||_||_||_||_dS) G Configure the Per-Message Deflate extension factory. Nr r,/server_max_window_bits must be between 8 and 15T/client_max_window_bits must be between 8 and 15r-zLcompress_settings must not include wbits, set client_max_window_bits instead) ValueErrorr[r\r]r^r*)r5r[r\r]r^r*rrrr6Bs( z'ClientPerMessageDeflateFactory.__init__rbr7cCst|j|j|j|jS)z, Build request parameters. )rar[r\r]r^r9rrrget_request_paramsbs z1ClientPerMessageDeflateFactory.get_request_paramsSequence[Extension]rr`accepted_extensionsr+cstfdd|Dr&tdjt|dd\}}}}jrL|sLtdjrZ|sZd}jdurfn$|durxtd n|jkrtd jdur|durtd n.jdurn"|durˆj}n|jkrtd t|||pd |pd j S)zV Process response parameters. Return an extension instance. c3s|]}|jjkVqdSNrYrotherr9rr yrzIClientPerMessageDeflateFactory.process_response_params..zreceived duplicate Frcz#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 rYrfr[r\r]r^rr*r5r`ror[r\r]r^rr9rprocess_response_paramsnsH          z6ClientPerMessageDeflateFactory.process_response_params)FFNTN) rUrVrWrXrrYr6rlrxrrrrr(s  rz'Sequence[ClientExtensionFactory] | Nonez 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|]}|jtjkVqdSrp)rYr)rZextension_factoryrrrrtsz3enable_client_permessage_deflate..memLevel)r*)rvlistrryrrrrs 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://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"FNr#r<rgr%)r[r\r]r^r*require_client_max_window_bitsr+cCs|dus&d|krdks&ntd|dusLd|krBdksLntd|durdd|vrdtd|durx|rxtd||_||_||_||_||_||_dS) rhNr r,rirjr-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)rkr[r\r]r^r*r~)r5r[r\r]r^r*r~rrrr6 s$  z'ServerPerMessageDeflateFactory.__init__rbrmz2tuple[list[ExtensionParameter], PerMessageDeflate]rncstfdd|Dr&tdjt|dd\}}}}jrH|sHd}jrV|sVd}jdurbn |durrj}n|jkrj}jdur|durԈj}n8|durjrtdn |durĈj}nj|krԈj}t ||||t |||pd|pdj fS) zi Process request parameters. Return response params and an extension instance. c3s|]}|jjkVqdSrprqrrr9rrrt;rzHServerPerMessageDeflateFactory.process_request_params..zskipped duplicate TruNzrequired client_max_window_bitsr,) rvr rYrfr[r\r]r^r~rarr*rwrr9rprocess_request_params0sX        z5ServerPerMessageDeflateFactory.process_request_params)FFNNNF)rUrVrWrXrrYr6rrrrrrs#rz'Sequence[ServerExtensionFactory] | Nonez 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|]}|jtjkVqdSrp)rYr)rZ ext_factoryrrrrtsz3enable_server_permessage_deflate.. rzr{)r]r^r*)rvr|rr}rrrrs r)" __future__rr/collections.abcrtypingrrr exceptionsrr r r r r rrbaserrr__all__rErangerdrrarfrrrrrrrrs(    3=13