o
    iY2                     @  s  U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	m
Z
 d dlmZ ddlmZmZ ddlmZ zddlmZ W n eyR   ddlmZ Y nw g d	ZG d
d dejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%G dd dejZ&e&j'de&j(de&j)de&j*de&j+de&j,de&j-de&j.de&j/de&j0de&j1de&j2de&j3de&j4de&j5diZ6de7d< e&j'e&j(e&j)e&j*e&j-e&j.e&j/e&j0e&j1e&j2e&j3e&j4hZ8e&j'e&j(e&j+hZ9ej:G dd  d Z;ej:G d!d" d"Z<dd#l=m>Z> dS )$    )annotationsN)	GeneratorSequence)Callable   )PayloadTooBigProtocolError)	BytesLike)
apply_mask)OpcodeOP_CONTOP_TEXT	OP_BINARYOP_CLOSEOP_PINGOP_PONGDATA_OPCODESCTRL_OPCODES	CloseCodeFrameClosec                   @  s$   e Zd ZdZd\ZZZd\ZZZ	dS )r   z#Opcode values for WebSocket frames.)r   r      )   	   
   N)
__name__
__module____qualname____doc__CONTTEXTBINARYCLOSEPINGPONG r%   r%   d/home/www/bk.finsightngr.online/finsigh_v2/venv200/lib/python3.10/site-packages/websockets/frames.pyr   &   s    
r   c                   @  sL   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdS )r   z-Close code values for WebSocket close frames.i  i  i  i  i  i  i  i  i  i  i  i  i  i  i  N)r   r   r   r   NORMAL_CLOSURE
GOING_AWAYPROTOCOL_ERRORUNSUPPORTED_DATANO_STATUS_RCVDABNORMAL_CLOSUREINVALID_DATAPOLICY_VIOLATIONMESSAGE_TOO_BIGMANDATORY_EXTENSIONINTERNAL_ERRORSERVICE_RESTARTTRY_AGAIN_LATERBAD_GATEWAYTLS_HANDSHAKEr%   r%   r%   r&   r   8   s"    r   OKz
going awayzprotocol errorzunsupported datazno status received [internal]zabnormal closure [internal]zinvalid frame payload datazpolicy violationzmessage too bigzmandatory extensionzinternal errorzservice restartztry again laterzbad gatewayz TLS handshake failure [internal]zdict[int, str]CLOSE_CODE_EXPLANATIONSc                   @  s   e Zd ZU dZded< ded< dZded< d	Zded
< d	Zded< d	Zded< e	e
jddZd&ddZedddd'ddZddd(d!d"Zd)d$d%ZdS )*r   aN  
    WebSocket frame.

    Attributes:
        opcode: Opcode.
        data: Payload data.
        fin: FIN bit.
        rsv1: RSV1 bit.
        rsv2: RSV2 bit.
        rsv3: RSV3 bit.

    Only these fields are needed. The MASK bit, payload length and masking-key
    are handled on the fly when parsing and serializing frames.

    r   opcoder	   dataTboolfinFrsv1rsv2rsv3ZWEBSOCKETS_MAX_LOG_SIZEZ75returnstrc              
   C  s  d}t | j dt | jdkrdnd }| jrdnd}| jtu r+tt| j }n| jtu rc| j}t || j	d krX| j	d d d }d
|dd	|  d
|| d g}d
dd |D }n`| jtu rqtt| j}nR| jrztt| j }d}W q ttfy   | j}t || j	d kr| j	d d d }d
|dd	|  d
|| d g}d
dd |D }d}Y qw d}t || j	kr| j	d d }|dd	|  d || d  }d
td|||g}| jj d| d| dS )zE
        Return a human-readable representation of a frame.

        Nz byter    s	continued       r   s      c                 s      | ]}|d V  qdS Z02xNr%   .0byter%   r%   r&   	<genexpr>       z Frame.__str__.<locals>.<genexpr>textc                 s  rG   rH   r%   rI   r%   r%   r&   rL      rM   binaryz''z...z, z [])lenr9   r;   r8   r   reprbytesdecoder   MAX_LOG_SIZEjoinr   r@   r   parseUnicodeDecodeErrorAttributeErrorfiltername)selfZcodinglengthZ	non_finalr9   rO   Zcutmetadatar%   r%   r&   __str__   s>   &

&
&"zFrame.__str__N)max_size
extensions
read_exact9Callable[[int], Generator[None, None, bytes | bytearray]]maskr`   
int | Nonera   %Sequence[extensions.Extension] | NoneGenerator[None, None, Frame]c             
   c  s   |dE dH }t d|\}}|d@ rdnd}|d@ rdnd}	|d@ r&dnd}
|d	@ r.dnd}zt|d
@ }W n tyI } ztd|d}~ww |d@ rPdnd|krXtd|d@ }|dkro|dE dH }t d|\}n|dkr|dE dH }t d|\}|dur||krt|||r|dE dH }||E dH }|rt||}| ||||	|
|}|du rg }t|D ]	}|j||d}q|	  |S )a  
        Parse a WebSocket frame.

        This is a generator-based coroutine.

        Args:
            read_exact: Generator-based coroutine that reads the requested
                bytes or raises an exception if there isn't enough data.
            mask: Whether the frame should be masked i.e. whether the read
                happens on the server side.
            max_size: Maximum payload size in bytes.
            extensions: List of extensions, applied in reverse order.

        Raises:
            EOFError: If the connection is closed without a full WebSocket frame.
            PayloadTooBig: If the frame's payload size exceeds ``max_size``.
            ProtocolError: If the frame contains incorrect values.

        r   N!BB   TF@             zinvalid opcodezincorrect masking   ~   !Hr   z!Q   )r`   )
structunpackr   
ValueErrorr   r   r
   reversedrT   check)clsrb   rd   r`   ra   r9   head1head2r;   r<   r=   r>   r8   excr]   
mask_bytesframe	extensionr%   r%   r&   rW      sH   


zFrame.parsera   rS   c          
      C  s&  |    |du r
g }|D ]}|| } qt }| jrdnd| jr#dndB | jr*dndB | jr1dndB | jB }|r;dnd}t	| j
}|dk rS|td|||B  n|d	k re|td
||dB | n|td||dB | |r~td}|| |rt| j
|}	n| j
}	||	 | S )aH  
        Serialize a WebSocket frame.

        Args:
            mask: Whether the frame should be masked i.e. whether the write
                happens on the client side.
            extensions: List of extensions, applied in order.

        Raises:
            ProtocolError: If the frame contains incorrect values.

        Nri   r   rj   rk   rl   ro   rh   i   z!BBHz!BBQrn   rq   )rv   encodeioBytesIOr;   r<   r=   r>   r8   rQ   r9   writerr   packsecretsZtoken_bytesr
   getvalue)
r\   rd   ra   r}   outputrx   ry   r]   r{   r9   r%   r%   r&   	serialize  s>   



zFrame.serializeNonec                 C  sP   | j s	| js	| jrtd| jtv r$t| jdkrtd| js&tddS dS )z
        Check that reserved bits and opcode have acceptable values.

        Raises:
            ProtocolError: If a reserved bit or the opcode is invalid.

        zreserved bits must be 0}   zcontrol frame too longzfragmented control frameN)	r<   r=   r>   r   r8   r   rQ   r9   r;   r\   r%   r%   r&   rv   N  s   
zFrame.checkr?   r@   )
rb   rc   rd   r:   r`   re   ra   rf   r?   rg   )rd   r:   ra   rf   r?   rS   r?   r   )r   r   r   r   __annotations__r;   r<   r=   r>   intosenvirongetrU   r_   classmethodrW   r   rv   r%   r%   r%   r&   r   z   s"   
 
2M<r   c                   @  sN   e Zd ZU dZded< ded< dddZedddZdddZdddZ	dS )r   z
    Code and reason for WebSocket close frames.

    Attributes:
        code: Close code.
        reason: Close reason.

    zCloseCode | intcoder@   reasonr?   c                 C  sz   d| j   krdk rn nd}nd| j   krdk r n nd}nt| j d}| j  d| d}| jr;| d	| j }|S )
zU
        Return a human-readable representation of a close code and reason.

          i  Z
registered  zprivate useunknownz ()rF   )r   r7   r   r   )r\   Zexplanationresultr%   r%   r&   r_   n  s   zClose.__str__r9   r	   c                 C  sz   t |tr	tdt|dkr-td|dd \}|dd  }| ||}|  |S t|dkr9| tj	dS t
d)z
        Parse the payload of a close frame.

        Args:
            data: Payload of the close frame.

        Raises:
            ProtocolError: If data is ill-formed.
            UnicodeDecodeError: If the reason isn't valid UTF-8.

        z.only compressed outgoing frames use memoryviewr   rp   Nr   rA   zclose frame too short)
isinstance
memoryviewAssertionErrorrQ   rr   rs   rT   rv   r   r+   r   )rw   r9   r   r   closer%   r%   r&   rW     s   

zClose.parserS   c                 C  s    |    td| j| j  S )z:
        Serialize the payload of a close frame.

        rp   )rv   rr   r   r   r   r   r   r%   r%   r&   r     s   zClose.serializer   c                 C  s8   | j tv sd| j   krdk std tddS dS )z
        Check that the close code has a valid value for a close frame.

        Raises:
            ProtocolError: If the close code is invalid.

        r   r   zinvalid status codeN)r   EXTERNAL_CLOSE_CODESr   r   r%   r%   r&   rv     s
   zClose.checkNr   )r9   r	   r?   r   )r?   rS   r   )
r   r   r   r   r   r_   r   rW   r   rv   r%   r%   r%   r&   r   `  s   
 	

r   r~   )?
__future__r   dataclassesenumr   r   r   rr   collections.abcr   r   typingr   
exceptionsr   r   r	   Zspeedupsr
   ImportErrorutils__all__IntEnumr   r   r   r    r   r!   r   r"   r   r#   r   r$   r   r   r   r   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r7   r   r   ZOK_CLOSE_CODES	dataclassr   r   rA   ra   r%   r%   r%   r&   <module>   s    

 fN