a
    њhg                     @  s<  d dl mZ d dlZd dlZd dl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 ddlmZmZmZmZmZmZmZmZmZmZ dd	lmZmZ dd
l m!Z! ddlm"Z"m#Z#m$Z$ g dZ%eeeef Z&G dd dej'Z(e(j)Z)e(j*Z*G dd dej'Z+e+j,Z,e+j-Z-e+j.Z.e+j/Z/dZ0G dd dZ1dS )    )annotationsN)	Generator)Union   )ConnectionClosedConnectionClosedErrorConnectionClosedOKInvalidStatePayloadTooBigProtocolError)	Extension)
OK_CLOSE_CODES	OP_BINARYOP_CLOSEOP_CONTOP_PINGOP_PONGOP_TEXTClose	CloseCodeFrame)RequestResponse)StreamReader)
LoggerLikeOriginSubprotocol)ProtocolSideStateSEND_EOFc                   @  s   e Zd ZdZed\ZZdS )r   z6A WebSocket connection is either a server or a client.   N)__name__
__module____qualname____doc__rangeSERVERCLIENT r)   r)   N/wd/webapps/venvs/v2025_4um/lib/python3.9/site-packages/websockets/protocol.pyr   /   s   r   c                   @  s    e Zd ZdZed\ZZZZdS )r   z6A WebSocket connection is in one of these four states.   N)	r"   r#   r$   r%   r&   
CONNECTINGOPENCLOSINGCLOSEDr)   r)   r)   r*   r   9   s   r       c                   @  s  e Zd ZdZedddddddd	d
ddZeddddZejdd	dddZeddddZ	eddddZ
eddddZdd	dddZd	dddZdd d	d!d"d#ZdJdd d	d!d%d&ZdKdd d	d!d'd(ZdLdd*d	d+d,d-Zdd	dd.d/Zdd	dd0d1ZdMd2d*d	d+d3d4Zd5dd6d7Zd8dd9d:Zd dd;d<Zd=dd>d?Zd=dd@dAZdBd	dCdDdEZdBd	dCdFdGZd	ddHdIZdS )Nr   a6  
    Sans-I/O implementation of a WebSocket connection.

    Args:
        side: :attr:`~Side.CLIENT` or :attr:`~Side.SERVER`.
        state: Initial state of the WebSocket connection.
        max_size: Maximum size of incoming messages in bytes;
            :obj:`None` disables the limit.
        logger: Logger for this connection; depending on ``side``,
            defaults to ``logging.getLogger("websockets.client")``
            or ``logging.getLogger("websockets.server")``;
            see the :doc:`logging guide <../../topics/logging>` for details.

    i   N)statemax_sizeloggerr   r   z
int | NonezLoggerLike | NoneNone)sider1   r2   r3   returnc                C  s   t  | _|d u r(td|j  }|| _|tj	| _
|| _|| _|| _d | _d| _d | _g | _d | _d | _d | _d | _d | _d| _t | _g | _g | _|  | _t| j d | _d S )Nzwebsockets.F) uuiduuid4idlogging	getLoggernamelowerr3   isEnabledForDEBUGdebugr5   r1   r2   cur_sizeexpect_continuation_frameorigin
extensionsZsubprotocol
close_rcvd
close_sentclose_rcvd_then_sentZhandshake_exceof_sentr   readereventswritesparseparsernext
parser_exc)selfr5   r1   r2   r3   r)   r)   r*   __init__Y   s0    	
	

zProtocol.__init__)r6   c                 C  s   | j S )a  
        State of the WebSocket connection.

        Defined in 4.1_, 4.2_, 7.1.3_, and 7.1.4_ of :rfc:`6455`.

        .. _4.1: https://datatracker.ietf.org/doc/html/rfc6455#section-4.1
        .. _4.2: https://datatracker.ietf.org/doc/html/rfc6455#section-4.2
        .. _7.1.3: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.3
        .. _7.1.4: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.4

        )_staterP   r)   r)   r*   r1      s    zProtocol.state)r1   r6   c                 C  s    | j r| j d|j || _d S )Nz= connection is %s)r@   r3   r<   rR   )rP   r1   r)   r)   r*   r1      s    c                 C  s*   | j turdS | jdu rtjS | jjS dS )z
        WebSocket close code received from the remote endpoint.

        Defined in 7.1.5_ of :rfc:`6455`.

        .. _7.1.5: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.5

        :obj:`None` if the connection isn't closed yet.

        N)r1   r/   rE   r   ABNORMAL_CLOSUREcoderS   r)   r)   r*   
close_code   s
    

zProtocol.close_codez
str | Nonec                 C  s(   | j turdS | jdu rdS | jjS dS )a  
        WebSocket close reason  received from the remote endpoint.

        Defined in 7.1.6_ of :rfc:`6455`.

        .. _7.1.6: https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.6

        :obj:`None` if the connection isn't closed yet.

        N )r1   r/   rE   reasonrS   r)   r)   r*   close_reason   s
    

zProtocol.close_reasonr   c                 C  sf   | j tu sJ d| jdurD| jdurD| jjtv rD| jjtv rDt}nt}|| j| j| j}| j	|_
|S )a  
        Exception to raise when trying to interact with a closed connection.

        Don't raise this exception while the connection :attr:`state`
        is :attr:`~websockets.protocol.State.CLOSING`; wait until
        it's :attr:`~websockets.protocol.State.CLOSED`.

        Indeed, the exception includes the close code and reason, which are
        known only once the connection is closed.

        Raises:
            AssertionError: If the connection isn't closed yet.

        zconnection isn't closed yetN)r1   r/   rE   rF   rU   r   r   r   rG   rO   	__cause__)rP   exc_typeexcr)   r)   r*   	close_exc   s$    

zProtocol.close_excbytes)datar6   c                 C  s   | j | t| j dS )aH  
        Receive data from the network.

        After calling this method:

        - You must call :meth:`data_to_send` and send this data to the network.
        - You should call :meth:`events_received` and process resulting events.

        Raises:
            EOFError: If :meth:`receive_eof` was called earlier.

        N)rI   Z	feed_datarN   rM   rP   r_   r)   r)   r*   receive_data   s    zProtocol.receive_datac                 C  s$   | j jrdS | j   t| j dS )a  
        Receive the end of the data stream from the network.

        After calling this method:

        - You must call :meth:`data_to_send` and send this data to the network;
          it will return ``[b""]``, signaling the end of the stream, or ``[]``.
        - You aren't expected to call :meth:`events_received`; it won't return
          any new events.

        :meth:`receive_eof` is idempotent.

        N)rI   eofZfeed_eofrN   rM   rS   r)   r)   r*   receive_eof  s    
zProtocol.receive_eofbool)r_   finr6   c                 C  sL   | j std| jtur.td| jj  | | _ | t	t
|| dS )a  
        Send a `Continuation frame`_.

        .. _Continuation frame:
            https://datatracker.ietf.org/doc/html/rfc6455#section-5.6

        Parameters:
            data: payload containing the same kind of data
                as the initial frame.
            fin: FIN bit; set it to :obj:`True` if this is the last frame
                of a fragmented message and to :obj:`False` otherwise.

        Raises:
            ProtocolError: If a fragmented message isn't in progress.

        unexpected continuation frameconnection is N)rB   r   rR   r-   r	   r1   r<   r=   
send_framer   r   rP   r_   re   r)   r)   r*   send_continuation#  s    
zProtocol.send_continuationTc                 C  sL   | j rtd| jtur.td| jj  | | _ | t	t
|| dS )a  
        Send a `Text frame`_.

        .. _Text frame:
            https://datatracker.ietf.org/doc/html/rfc6455#section-5.6

        Parameters:
            data: payload containing text encoded with UTF-8.
            fin: FIN bit; set it to :obj:`False` if this is the first frame of
                a fragmented message.

        Raises:
            ProtocolError: If a fragmented message is in progress.

        expected a continuation framerg   N)rB   r   rR   r-   r	   r1   r<   r=   rh   r   r   ri   r)   r)   r*   	send_text;  s    
zProtocol.send_textc                 C  sL   | j rtd| jtur.td| jj  | | _ | t	t
|| dS )a  
        Send a `Binary frame`_.

        .. _Binary frame:
            https://datatracker.ietf.org/doc/html/rfc6455#section-5.6

        Parameters:
            data: payload containing arbitrary binary data.
            fin: FIN bit; set it to :obj:`False` if this is the first frame of
                a fragmented message.

        Raises:
            ProtocolError: If a fragmented message is in progress.

        rk   rg   N)rB   r   rR   r-   r	   r1   r<   r=   rh   r   r   ri   r)   r)   r*   send_binaryR  s    
zProtocol.send_binaryrW   str)rU   rX   r6   c                 C  s   | j tur td| jj  |du rJ|dkr8tdttj	d}d}nt||}|
 }| tt| | jdu szJ || _t| _dS )a_  
        Send a `Close frame`_.

        .. _Close frame:
            https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.1

        Parameters:
            code: close code.
            reason: close reason.

        Raises:
            ProtocolError: If the code isn't valid or if a reason is provided
                without a code.

        rg   NrW   z#cannot send a reason without a coder0   )rR   r-   r	   r1   r<   r=   r   r   r   ZNO_STATUS_RCVD	serializerh   r   r   rE   rF   r.   rP   rU   rX   closer_   r)   r)   r*   
send_closei  s    

zProtocol.send_closec                 C  s>   | j tur*| j tur*td| jj  | tt	| dS )z
        Send a `Ping frame`_.

        .. _Ping frame:
            https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.2

        Parameters:
            data: payload containing arbitrary binary data.

        rg   N)
rR   r-   r.   r	   r1   r<   r=   rh   r   r   r`   r)   r)   r*   	send_ping  s    zProtocol.send_pingc                 C  s>   | j tur*| j tur*td| jj  | tt	| dS )z
        Send a `Pong frame`_.

        .. _Pong frame:
            https://datatracker.ietf.org/doc/html/rfc6455#section-5.5.3

        Parameters:
            data: payload containing arbitrary binary data.

        rg   N)
rR   r-   r.   r	   r1   r<   r=   rh   r   r   r`   r)   r)   r*   	send_pong  s    zProtocol.send_pongintc                 C  s   | j tu rR|tjkrRt||}| }| tt| || _	| j
durLd| _t| _ | jtu rj| jsj|   |  | _t| j dS )a?  
        `Fail the WebSocket connection`_.

        .. _Fail the WebSocket connection:
            https://datatracker.ietf.org/doc/html/rfc6455#section-7.1.7

        Parameters:
            code: close code
            reason: close reason

        Raises:
            ProtocolError: If the code isn't valid.
        NT)r1   r-   r   rT   r   ro   rh   r   r   rF   rE   rG   r.   r5   r'   rH   send_eofdiscardrM   rN   rp   r)   r)   r*   fail  s    




zProtocol.failzlist[Event]c                 C  s   | j g  }| _ |S )a*  
        Fetch events generated from data received from the network.

        Call this method immediately after any of the ``receive_*()`` methods.

        Process resulting events, likely by passing them to the application.

        Returns:
            Events read from the connection.
        )rJ   )rP   rJ   r)   r)   r*   events_received  s    zProtocol.events_receivedzlist[bytes]c                 C  s   | j g  }| _ |S )a  
        Obtain data to send to the network.

        Call this method immediately after any of the ``receive_*()``,
        ``send_*()``, or :meth:`fail` methods.

        Write resulting data to the connection.

        The empty bytestring :data:`~websockets.protocol.SEND_EOF` signals
        the end of the data stream. When you receive it, half-close the TCP
        connection.

        Returns:
            Data to write to the connection.

        )rK   )rP   rK   r)   r)   r*   data_to_send  s    zProtocol.data_to_sendc                 C  s>   | j tu rdS | j tu rdS | j tu r*dS | j tu s8J | jS )a  
        Tell if the TCP connection is expected to close soon.

        Call this method immediately after any of the ``receive_*()``,
        ``send_close()``, or :meth:`fail` methods.

        If it returns :obj:`True`, schedule closing the TCP connection after a
        short timeout if the other side hasn't already closed it.

        Returns:
            Whether the TCP connection is expected to close soon.

        FT)r1   r-   r.   r/   r,   rH   rS   r)   r)   r*   close_expected  s    "


zProtocol.close_expectedzGenerator[None]c              
   c  s  z| j  E dH r,| jr$| jd td| jdu r<d}n| jdu rN| j}n| j| j }tj| j j	| j
tu || jdE dH }| jr| jd| | | qW nN ty } z&| tjt| || _W Y d}~nd}~0  ty } z$| tjt| || _W Y d}~nd}~0  tyb } z.| tj|j d|j  || _W Y d}~nd}~0  ty } z0|| j | tjt| || _W Y d}~nPd}~0  ty } z.| jjddd	 | tj || _W Y d}~n
d}~0 0 dV  td
dS )a-  
        Parse incoming data into frames.

        :meth:`receive_data` and :meth:`receive_eof` run this generator
        coroutine until it needs more data or reaches EOF.

        :meth:`parse` never raises an exception. Instead, it sets the
        :attr:`parser_exc` and yields control.

        N< EOFzunexpected end of stream)maskr2   rD   z< %sz at position zparser failedT)exc_infoz"parse() shouldn't step after error) rI   at_eofr@   r3   EOFErrorr2   rA   r   rL   Z
read_exactr5   r'   rD   
recv_framer   rx   r   ZPROTOCOL_ERRORrn   rO   rT   UnicodeDecodeErrorZINVALID_DATArX   startr
   Zset_current_sizeZMESSAGE_TOO_BIG	ExceptionerrorINTERNAL_ERRORAssertionError)rP   r2   framer\   r)   r)   r*   rL   /  sL    

zProtocol.parsec                 c  s   | j tu p| jtu | jksJ | j E dH s:| j  q| jrL| j	d | j t
u rh| jturh|   t| _dV  tddS )z
        Discard incoming data.

        This coroutine replaces :meth:`parse`:

        - after receiving a close frame, during a normal closure (1.4);
        - after sending a close frame, during an abnormal closure (7.1.7).

        Nr|   z"discard() shouldn't step after EOF)r5   r'   r1   r,   rH   rI   r   rw   r@   r3   r(   rv   r/   r   rS   r)   r)   r*   rw   w  s    zProtocol.discardr   )r   r6   c                 C  sr  |j tu s|j tu r<| jdur&td|js8t|j| _n&|j tu r|| jdu rXtd|jrfd| _n|  jt|j7  _n|j t	u rt
t|j}| | n|j tu rn|j tu rPt|j| _| jtu r| jdusJ d| _| jdurtd| jtu r&| t
t|j | j| _d| _t| _| jtu r:|   |  | _t| j ntd|j d| j| dS )	z-
        Process an incoming frame.

        Nrk   rf   Fzincomplete fragmented messageTzunexpected opcode: 02x)opcoder   r   rA   r   re   lenr_   r   r   r   r   rh   r   r   rL   rE   r1   r.   rF   rG   r-   r5   r'   rv   rw   rM   rN   r   rJ   append)rP   r   Z
pong_framer)   r)   r*   r     sD    







zProtocol.recv_framec                 C  s6   | j r| j d| | j|j| jtu | jd d S )Nz> %s)r}   rD   )r@   r3   rK   r   ro   r5   r(   rD   )rP   r   r)   r)   r*   rh     s    zProtocol.send_framec                 C  s2   | j r
J d| _ | jr"| jd | jt d S )NTz> EOF)rH   r@   r3   rK   r   r    rS   r)   r)   r*   rv     s
    
zProtocol.send_eof)T)T)NrW   )rW   )r"   r#   r$   r%   r-   rQ   propertyr1   setterrV   rY   r]   ra   rc   rj   rl   rm   rr   rs   rt   rx   ry   rz   r{   rL   rw   r   rh   rv   r)   r)   r)   r*   r   I   s@   D%$1-HR
r   )2
__future__r   enumr:   r7   collections.abcr   typingr   
exceptionsr   r   r   r	   r
   r   rD   r   framesr   r   r   r   r   r   r   r   r   r   Zhttp11r   r   streamsr   r   r   r   __all__EventIntEnumr   r'   r(   r   r,   r-   r.   r/   r    r   r)   r)   r)   r*   <module>   s.    0