a
    یxd0                     @  sx  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mZm	Z	m
Z
mZmZ ddlmZmZ ddlmZ zddlmZ W n ey   ddlmZ Y n0 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%ddddddddddddddddZ&h dZ'h dZ(e)e*e+fZ,ej-G dd dZ.dd d!d"d#Z/dd$d!d%d&Z0ej-G d'd( d(Z1dS ))    )annotationsN)Callable	GeneratorOptionalSequenceTuple   )
exceptions
extensions)Data)
apply_mask)OpcodeOP_CONTOP_TEXT	OP_BINARYOP_CLOSEOP_PINGOP_PONGDATA_OPCODESCTRL_OPCODESFrameprepare_dataprepare_ctrl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(   I/var/www/html/Ranjet/env/lib/python3.9/site-packages/websockets/frames.pyr   %   s   
r   OKz
going awayzprotocol errorzunsupported typezno status code [internal]z'connection closed abnormally [internal]zinvalid datazpolicy violationzmessage too bigzextension requiredzunexpected errorzservice restartztry again laterzbad gatewayzTLS failure [internal])          i                  i  >   r+   r,   r-   r.   r0   r1   r2   r3   r4   r5   r6   r7   >   r+   r,   r/   c                   @  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< ddddZ	e
dddddddddddZddddddddZd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   opcodebytesdataTboolfinFrsv1rsv2rsv3strreturnc              
   C  s  d}t | j dt | jdkr"dnd }| jr4dnd}| jtu rRt| j }n| jtu r| j}t |dkrd|dd	 d
|dd g}ddd |D }n| jt	u rt
t| j}n| jrFzt| j }d}W nd ttfyB   | j}t |dkr&d|dd	 d
|dd g}ddd |D }d}Y n0 nd}t |dkrt|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          s     i c                 s  s   | ]}|d V  qdS 02xNr(   .0byter(   r(   r)   	<genexpr>   rG   z Frame.__str__.<locals>.<genexpr>textc                 s  s   | ]}|d V  qdS rJ   r(   rL   r(   r(   r)   rO      rG   binaryz''K   0   z...iz, z [])lenr:   r<   r8   r   reprdecoder   joinr   r@   r   parseUnicodeDecodeErrorAttributeErrorfiltername)selfZcodinglengthZ	non_finalr:   rQ   metadatar(   r(   r)   __str__   s6    &

 
 zFrame.__str__N)max_sizer
   z-Callable[[int], Generator[None, None, bytes]]zOptional[int]z(Optional[Sequence[extensions.Extension]]zGenerator[None, None, Frame])
read_exactmaskrb   r
   rB   c             
   c  s  |dE dH }t d|\}}|d@ r*dnd}|d@ r:dnd}	|d@ rJdnd}
|d	@ rZdnd}zt|d
@ }W n0 ty } ztd|W Y d}~n
d}~0 0 |d@ rdnd|krtd|d@ }|dkr|dE dH }t d|\}n&|dkr|dE dH }t d|\}|dur>||kr>td| d| d|rR|dE dH }||E dH }|rpt||}| ||||	|
|}|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.
            UnicodeDecodeError: if the frame contains invalid UTF-8.
            PayloadTooBig: if the frame's payload size exceeds ``max_size``.
            ProtocolError: if the frame contains incorrect values.

        r   N!BB   TF@       rH      zinvalid opcodezincorrect masking   ~   !Hr   z!Qzover size limit (z > z bytes)   )rb   )structunpackr   
ValueErrorr	   ProtocolErrorZPayloadTooBigr   reversedrW   check)clsrc   rd   rb   r
   r:   head1head2r<   r=   r>   r?   r8   excr_   
mask_bytesframe	extensionr(   r(   r)   rY      sF    "



zFrame.parse)r
   )rd   r
   rB   c          
      C  s(  |    |du rg }|D ]}|| } qt }| jr:dnd| jrFdndB | jrTdndB | jrbdndB | jB }|rvdnd}t	| j
}|dk r|td|||B  n>|d	k r|td
||dB | n|td||dB | |rtd}|| |rt| 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.

        Nrf   r   rg   rh   rH   rk   re   i   z!BBHz!BBQrj   rm   )rs   encodeioBytesIOr<   r=   r>   r?   r8   rU   r:   writern   packsecretsZtoken_bytesr   getvalue)
r^   rd   r
   rz   outputru   rv   r_   rx   r:   r(   r(   r)   	serialize   s>    



zFrame.serializeNonec                 C  sR   | j s| js| jrtd| jtv rNt| jdkr>td| j	sNtd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	   rq   r8   r   rU   r:   r<   r^   r(   r(   r)   rs   7  s    


zFrame.check)r   r   r    r!   __annotations__r<   r=   r>   r?   ra   classmethodrY   r   rs   r(   r(   r(   r)   r   g   s   
/P;r   r   zTuple[int, bytes]r:   rB   c                 C  s6   t | trt| dfS t | tr*t| fS tddS )a  
    Convert a string or byte-like object to an opcode and a bytes-like object.

    This function is designed for data frames.

    If ``data`` is a :class:`str`, return ``OP_TEXT`` and a :class:`bytes`
    object encoding ``data`` in UTF-8.

    If ``data`` is a bytes-like object, return ``OP_BINARY`` and a bytes-like
    object.

    Raises:
        TypeError: if ``data`` doesn't have a supported type.

    utf-8data must be str or bytes-likeN)
isinstancer@   r   r{   	BytesLiker   	TypeErrorr:   r(   r(   r)   r   I  s
    

r   r9   c                 C  s2   t | tr| dS t | tr&t| S tddS )ai  
    Convert a string or byte-like object to bytes.

    This function is designed for ping and pong frames.

    If ``data`` is a :class:`str`, return a :class:`bytes` object encoding
    ``data`` in UTF-8.

    If ``data`` is a bytes-like object, return a :class:`bytes` object.

    Raises:
        TypeError: if ``data`` doesn't have a supported type.

    r   r   N)r   r@   r{   r   r9   r   r   r(   r(   r)   r   a  s
    


r   c                   @  s`   e Zd ZU dZded< ded< ddddZed	d d
ddZd	dddZddddZ	dS )r   z
    Code and reason for WebSocket close frames.

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

    intcoder@   reasonrA   c                 C  sz   d| j   krdk r n nd}n.d| j   kr6dk r@n nd}nt| j d}| j  d| d}| jrv| d	| j }|S )
zU
        Return a human-readable representation of a close code and reason.

          i  
registered  zprivate useunknownz ()rI   )r   CLOSE_CODESgetr   )r^   Zexplanationresultr(   r(   r)   ra     s    zClose.__str__r9   r   c                 C  sn   t |dkrJtd|dd \}|dd d}| ||}|  |S t |dkr`| ddS tddS )	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.

        r   rl   Nr   r   r/   rC   zclose frame too short)rU   rn   ro   rW   rs   r	   rq   )rt   r:   r   r   closer(   r(   r)   rY     s    

zClose.parsec                 C  s"   |    td| j| jd S )z:
        Serialize the payload of a close frame.

        rl   r   )rs   rn   r   r   r   r{   r   r(   r(   r)   r     s    zClose.serializer   c                 C  s0   | j tv s,d| j   kr dk s,n td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	   rq   r   r(   r(   r)   rs     s    "zClose.checkN)
r   r   r    r!   r   ra   r   rY   r   rs   r(   r(   r(   r)   r   x  s   
	r   )2
__future__r   dataclassesenumr|   r   rn   typingr   r   r   r   r   rC   r	   r
   r   Zspeedupsr   ImportErrorutils__all__IntEnumr   r"   r   r#   r   r$   r   r%   r   r&   r   r'   r   r   r   r   r   ZOK_CLOSE_CODESr9   	bytearray
memoryviewr   	dataclassr   r   r   r   r(   r(   r(   r)   <module>   s`   


 b