
    vg>d                    .   d dl mZ d dlZd dlmZ d dlmZ ddlmZ ddl	m
Z
mZmZmZmZmZ ddlmZmZ d	d
lmZmZmZ g dZdZd  edd          D             Z G d de          Zd,dZd-d Z G d! d"e          Zd.d&Z G d' d(e          Zd/d+Z dS )0    )annotationsN)Sequence)Any   )frames)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationErrorPayloadTooBigProtocolError)ExtensionNameExtensionParameter   )ClientExtensionFactory	ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflates     c                ,    g | ]}t          |          S  )str).0bitss     e/var/www/pixelcanvas.ch/venv/lib/python3.11/site-packages/websockets/extensions/permessage_deflate.py
<listcomp>r      s    >>>3t99>>>          c                  R    e Zd ZdZ ed          Z	 dddZddZddddZddZ	dS )r   z)
    Per-Message Deflate extension.

    permessage-deflateNremote_no_context_takeoverboollocal_no_context_takeoverremote_max_window_bitsintlocal_max_window_bitscompress_settingsdict[Any, Any] | NonereturnNonec                l   |i }|dv sJ |dv sJ d|cxk    rdk    sn J d|cxk    rdk    sn J d|vsJ || _         || _        || _        || _        || _        | j         s t          j        | j                   | _        | j        s#t          j        dd| j         i| j        | _	        d| _
        dS )	z?
        Configure the Per-Message Deflate extension.

        N)FTr!      wbitsr1   Fr   )r%   r'   r(   r*   r+   zlibdecompressobjdecodercompressobjencoderdecode_cont_data)selfr%   r'   r(   r*   r+   s         r   __init__zPerMessageDeflate.__init__)   s,    $ ")]::::(M9999*0000b000000)////R///////////*D')B&&<#%:"!2. 	R-T5P4PQQQDL- 	+  11( DL !&r    r   c           	     H    d| j          d| j         d| j         d| j         d	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*   r9   s    r   __repr__zPerMessageDeflate.__repr__S   s_    C*.*IC C)-)GC C '+&AC C &*%?	C C C	
r    )max_sizeframeframes.Framer?   
int | Nonec               N   |j         t          j        v r|S |j         t          j        u r| j        s|S |j        rd| _        n>|j        s|S |j        sd| _        | j        r t          j	        | j
                   | _        |j        r5t          |j                  dk     rt          |j                  t          z   }n|j        }|dn|}	 | j                            ||          }| j        j        r|J t%          d|          |j        r7t          |j                  dk    r| j                            t                     n'# t          j        $ r}t)          d          |d}~ww xY w|j        r	| j        r| `t          j        |j         ||j        d|j        |j                  S )z,
        Decode an incoming frame.

        FTr2   i  Nr   zdecompression failed)opcoder   CTRL_OPCODESOP_CONTr8   finrsv1r%   r3   r4   r(   r5   lendatabytes_EMPTY_UNCOMPRESSED_BLOCK
decompressunconsumed_tailr   errorr   Framersv2rsv3)r9   r@   r?   rJ   
max_lengthexcs         r   decodezPerMessageDeflate.decode\   s    <6...L
 <6>))( y .(-% : 9 -(,% . V#19T8TUUU
 9 	UZ4// $$'@@DD:D"*QQ
		A<**4<<D|+ 4+++#D(333y CS__44''(ABBBz 	A 	A 	A 677S@	A 9 	8 	|LIJJ
 
 	
s   	A9E E'E""E'c                l   |j         t          j        v r|S |j         t          j        ur*| j        r#t          j        dd| j         i| j        | _	        | j	        
                    |j                  | j	                            t
          j                  z   }|j        rJ|dd         t          k    sJ t!          |          dk     r|dd         }nt#          |          dd         }|j        r	| j        r| `	t          j        |j         ||j        |j         t          j        u|j        |j                  S )z,
        Encode an outgoing frame.

        r1   Ni   r   )rD   r   rE   rF   r'   r3   r6   r*   r+   r7   compressrJ   flushZ_SYNC_FLUSHrG   rL   rI   
memoryviewrP   rQ   rR   )r9   r@   rJ   s      r   encodezPerMessageDeflate.encode   sF    <6...L
 <v~--- #/    55 ,    |$$UZ004<3E3EdFW3X3XX9 	- 9 999994yy4CRCy!$'', 9 	7 	|LIL.JJ
 
 	
r    N)r%   r&   r'   r&   r(   r)   r*   r)   r+   r,   r-   r.   )r-   r   )r@   rA   r?   rB   r-   rA   )r@   rA   r-   rA   )
__name__
__module____qualname____doc__r   namer:   r>   rU   r\   r   r    r   r   r   !   s         
 =-..D 48&& && && && &&T
 
 
 
  $	F
 F
 F
 F
 F
 F
P,
 ,
 ,
 ,
 ,
 ,
r    r   server_no_context_takeoverr&   client_no_context_takeoverserver_max_window_bitsrB   client_max_window_bitsint | bool | Noner-   list[ExtensionParameter]c                2   g }| r|                     d           |r|                     d           |r$|                     dt          |          f           |du r|                     d           n&|r$|                     dt          |          f           |S )zS
    Build a list of ``(name, value)`` pairs for some compression parameters.

    )rc   N)rd   Nre   T)rf   Nrf   )appendr   )rc   rd   re   rf   paramss        r   _build_parametersrl      s     (*F! <:;;;! <:;;; O/5K1L1LMNNN%%67777	 O/5K1L1LMNNNMr    rk   Sequence[ExtensionParameter]	is_server0tuple[bool, bool, int | None, int | bool | None]c                  d}d}d}d}| D ]\  }}|dk    r&|rt          |          |d}!t          ||          |dk    r&|rt          |          |d}Mt          ||          |dk    r:|t          |          |t          v rt          |          }t          ||          |dk    rA|t          |          |r|d}|t          v rt          |          }t          ||          t	          |          ||||fS )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.

    FNrc   Trd   re   rf   )r   r
   _MAX_WINDOW_BITS_VALUESr)   r	   )rk   rn   rc   rd   re   rf   rb   values           r   _extract_parametersrs      s    (-',)-04 $- $-e///) /(...}-1**+D%888111) /(...}-1**+D%888---%1(...///),U&&+D%888---%1(... 9U])-&&111),U&&+D%888 't,,, 	#"	 r    c                  L    e Zd ZdZ ed          Z	 	 	 	 	 dddZddZddZdS )r   a  
    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$   FNTrc   r&   rd   re   rB   rf   rg   r+   dict[str, Any] | Noner-   r.   c                    |d|cxk    rdk    sn t          d          |"|du sd|cxk    rdk    sn t          d          |d|v rt          d          || _        || _        || _        || _        || _        dS )	G
        Configure the Per-Message Deflate extension factory.

        Nr!   r0   /server_max_window_bits must be between 8 and 15T/client_max_window_bits must be between 8 and 15r1   zLcompress_settings must not include wbits, set client_max_window_bits instead)
ValueErrorrc   rd   re   rf   r+   )r9   rc   rd   re   rf   r+   s         r   r:   z'ClientPerMessageDeflateFactory.__init__B  s     '.!7M2S2S2S2SQS2S2S2S2SNOOO"*%--*0000b0000NOOO(W8I-I-I5  
 +E'*D'&<#&<#!2r    rh   c                N    t          | j        | j        | j        | j                  S )z,
        Build request parameters.

        )rl   rc   rd   re   rf   r=   s    r   get_request_paramsz1ClientPerMessageDeflateFactory.get_request_paramsb  s.    
 !++''	
 
 	
r    rk   rm   accepted_extensionsSequence[Extension]r   c                    t           fd|D                       rt          d j                   t          |d          \  }}}} j        r|st          d           j        r|sd} j        n+|t          d          | j        k    rt          d	           j        |t          d
          n. j        du rn$| j        }n| j        k    rt          d          t          |||pd|pd j	                  S )zV
        Process response parameters.

        Return an extension instance.

        c              3  8   K   | ]}|j         j         k    V  d S r]   rb   r   otherr9   s     r   	<genexpr>zIClientPerMessageDeflateFactory.process_response_params.<locals>.<genexpr>y  ,      HH5uzTY&HHHHHHr    zreceived duplicate Frn   z#expected server_no_context_takeoverTNzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr0   )
anyr   rb   rs   rc   rd   re   rf   r   r+   r9   rk   r}   rc   rd   re   rf   s   `      r   process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramsn  s    HHHH4GHHHHH 	F"#D#D#DEEE  %888	
&&"" * 	N- N&'LMMM * 	2- 2-1* &. &-&'HIII'$*EEE&'KLLL &.%1&'JKKK 2 (D00 &-)-)D&&'$*EEE&'KLLL &&"(b"(b"
 
 	
r    )FFNTN)rc   r&   rd   r&   re   rB   rf   rg   r+   ru   r-   r.   )r-   rh   )rk   rm   r}   r~   r-   r   )	r^   r_   r`   ra   r   rb   r:   r|   r   r   r    r   r   r   (  s         . =-..D ,1+0-148373 3 3 3 3@

 

 

 

g
 g
 g
 g
 g
 g
r    r   
extensions'Sequence[ClientExtensionFactory] | None Sequence[ClientExtensionFactory]c                    | g } t          d | D                       s#t          |           t          ddi          gz   } | 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.

    Nc              3  @   K   | ]}|j         t          j         k    V  d S r]   )rb   r   )r   extension_factorys     r   r   z3enable_client_permessage_deflate.<locals>.<genexpr>  sB         	"@"EE     r    memLevel   )r+   )r   listr   r   s    r   r   r     sw     
  !+     
 *%%*#-q/  )
 


 r    c                  F    e Zd ZdZ ed          Z	 	 	 	 	 	 dddZddZdS )r   a  
    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$   FNrc   r&   rd   re   rB   rf   r+   ru   require_client_max_window_bitsr-   r.   c                *   |d|cxk    rdk    sn t          d          |d|cxk    rdk    sn t          d          |d|v rt          d          ||rt          d          || _        || _        || _        || _        || _        || _        dS )	rw   Nr!   r0   rx   ry   r1   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)rz   rc   rd   re   rf   r+   r   )r9   rc   rd   re   rf   r+   r   s          r   r:   z'ServerPerMessageDeflateFactory.__init__  s     '.!7M2S2S2S2SQS2S2S2S2SNOOO&.!7M2S2S2S2SQS2S2S2S2SNOOO(W8I-I-I5   ").L)>  
 +E'*D'&<#&<#!2.L+++r    rk   rm   r}   r~   2tuple[list[ExtensionParameter], PerMessageDeflate]c                    t           fd|D                       rt          d j                   t          |d          \  }}}} j        r|sd} j        r|sd} j        n| j        }n| j        k    r j        } j        |du r j        }n7| j        rt          d          n|du r j        }n j        |k     r j        }t          ||||          t          |||pd|pd j                  fS )zi
        Process request parameters.

        Return response params and an extension instance.

        c              3  8   K   | ]}|j         j         k    V  d S r]   r   r   s     r   r   zHServerPerMessageDeflateFactory.process_request_params.<locals>.<genexpr>;  r   r    zskipped duplicate Tr   Nzrequired client_max_window_bitsr0   )r   r   rb   rs   rc   rd   re   rf   r   rl   r   r+   r   s   `      r   process_request_paramsz5ServerPerMessageDeflateFactory.process_request_params0  s    HHHH4GHHHHH 	E"#C	#C#CDDD  $777	
&&""" * 	2- 2-1* * 	2- 2-1* &. &-)-)D&&'$*EEE)-)D& &.%--)-)D& &-6 N*+LMMMN'4//)-)D&&,/EEE)-)D& **&&	  **&,"&,"& 
 	
r    )FFNNNF)rc   r&   rd   r&   re   rB   rf   rB   r+   ru   r   r&   r-   r.   )rk   rm   r}   r~   r-   r   )r^   r_   r`   ra   r   rb   r:   r   r   r    r   r   r     s}         4 =-..D ,1+0-1-137/4!M !M !M !M !MFo
 o
 o
 o
 o
 o
r    r   'Sequence[ServerExtensionFactory] | None Sequence[ServerExtensionFactory]c                    | g } t          d | D                       s%t          |           t          ddddi          gz   } | 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.

    Nc              3  @   K   | ]}|j         t          j         k    V  d S r]   )rb   r   )r   ext_factorys     r   r   z3enable_server_permessage_deflate.<locals>.<genexpr>  sB         	:??     r       r   r   )re   rf   r+   )r   r   r   r   s    r   r   r     s}     
  %     

 *%%*')')#-q/  )
 

 r    )
rc   r&   rd   r&   re   rB   rf   rg   r-   rh   )rk   rm   rn   r&   r-   ro   )r   r   r-   r   )r   r   r-   r   )!
__future__r   r3   collections.abcr   typingr    r   
exceptionsr   r	   r
   r   r   r   r   r   baser   r   r   __all__rL   rangerq   r   rl   rs   r   r   r   r   r   r    r   <module>r      s   " " " " " "  $ $ $ $ $ $                            7 6 6 6 6 6 6 6 K K K K K K K K K K   0 >>q">>> o
 o
 o
 o
 o
	 o
 o
 o
d   0: : : :zm
 m
 m
 m
 m
%; m
 m
 m
`   0o
 o
 o
 o
 o
%; o
 o
 o
d     r    