File: //usr/local/CyberCP/lib/python3.10/site-packages/websockets/__pycache__/protocol.cpython-310.pyc
o
�h�g � @ s@ d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ ddl m
Z
m Z m Z m
Z
mZmZ ddlmZ ddlmZmZmZmZmZmZmZmZmZm Z dd l m Z m Z dd
l m!Z! dd lm"Z"m#Z#m$Z$ g d �Z%ee e e f Z& G d
d� dej'�Z(e(j)Z)e(j*Z*G dd� dej'�Z+e+j,Z,e+j-Z-e+j.Z.e+j/Z/dZ0 G dd� d�Z1dS )� )� 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 Zd ZdZed�\ZZdS )r z6A WebSocket connection is either a server or a client.� N)�__name__�
__module__� __qualname__�__doc__�range�SERVER�CLIENT� r) r) �F/usr/local/CyberCP/lib/python3.10/site-packages/websockets/protocol.pyr / s r c @ s e Zd ZdZed�\ZZZZdS )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 Zd ZdZeddd�dMdd�ZedNdd��ZejdOdd��ZedPdd��Z edQdd��Z
edRdd��Z dSd d �Z dTd d!�Z
dUd$d%�ZdVdUd'd(�ZdVdUd)d*�ZdWdXd/d0�ZdSd1d2�ZdSd3d4�ZdYdZd6d7�Zd[d9d:�Zd\d<d=�Zd]d>d?�Zd^dAdB�Zd^dCdD�Zd_dGdH�Zd_dIdJ�ZdTdKdL�Z dS )`r 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�logger�sider r1 r r2 �
int | Noner3 �LoggerLike | None�return�Nonec C s� t �� | _ |d u rt�d|j�� � ��}|| _ |�tj �| _
|| _ || _ || _
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�debugr4 r1 r2 �cur_size�expect_continuation_frame�origin�
extensions� subprotocol�
close_rcvd�
close_sent�close_rcvd_then_sent�
handshake_exc�eof_sentr �reader�events�writes�parse�parser�next�
parser_exc)�selfr4 r1 r2 r3 r) r) r* �__init__Y s6
zProtocol.__init__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�rT r) r) r* r1 � s
zProtocol.statec C s | j r | j� d|j� || _d S )Nz= connection is %s)rB r3 r> rV )rT r1 r) r) r* r1 � s
c C s& | j turdS | jdu rtjS | jjS )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/ rH r �ABNORMAL_CLOSURE�coderW r) r) r* �
close_code� s
zProtocol.close_code�
str | Nonec C s$ | j turdS | jdu rdS | jjS )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/ rH �reasonrW r) r) r* � close_reason� s
zProtocol.close_reasonr c C sf | j tu s J d��| jdur"| jdur"| jjtv r"| jjtv r"t}nt}|| 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.
zconnection isn't closed yetN) r1 r/ rH rI rY r
r r rJ rS � __cause__)rT �exc_type�excr) r) r* � close_exc� s
�zProtocol.close_exc�data�bytesc C s | j �|� t| j� dS )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)rM � feed_datarR rQ �rT rc r) r) r* � receive_data� s
zProtocol.receive_datac C s$ | j jrdS | j �� t| j� dS )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)rM �eof�feed_eofrR rQ rW r) r) r* � receive_eof s
zProtocol.receive_eof�fin�boolc C sL | j std��| jturtd| jj�� � ���| | _ | �t t
||�� dS )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) rD r rV r- r r1 r> r? �
send_framer r �rT rc rk r) r) r* �send_continuation# s
zProtocol.send_continuationTc C �L | j rtd��| jturtd| jj�� � ���| | _ | �t t
||�� dS )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 framern N) rD r rV r- r r1 r> r? ro r r rp r) r) r* � send_text; �
zProtocol.send_textc C rr )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.
rs rn N) rD r rV r- r r1 r> r? ro r r rp r) r) r* � send_binaryR ru zProtocol.send_binaryr\ rY r] �strc C s� | j turtd| jj�� � ���|du r%|dkr td��ttj d�}d}n t||�}|�
� }| � t t
|�� | jdu s=J �|| _t| _dS )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.
rn Nr\ z#cannot send a reason without a coder0 )rV r- r r1 r> r? r r r �NO_STATUS_RCVD� serializero r r rH rI r. �rT rY r] �closerc r) r) r* �
send_closei s
zProtocol.send_closec C �> | j tur| j turtd| jj�� � ���| �tt |�� dS )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.
rn N)
rV r- r. r r1 r> r? ro r r rf r) r) r* � send_ping� � zProtocol.send_pingc C r} )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.
rn N)
rV r- r. r r1 r> r? ro r r rf r) r) r* � send_pong� r zProtocol.send_pong�intc C s� | j tu r)|tjkr)t||�}|�� }| �tt|�� || _ | j
dur&d| _ t | _ | j
tu r5| js5| �� | �� | _t| j� dS )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 rX r ry ro r r rI rH rJ r. r4 r' rL �send_eof�discardrQ rR rz r) r) r* �fail� s
z
Protocol.fail� list[Event]c C � | 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.
)rN )rT rN r) r) r* �events_received� s zProtocol.events_received� list[bytes]c C r� )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.
)rO )rT rO r) r) r* � data_to_send� s zProtocol.data_to_sendc C s> | j tu rdS | j tu rdS | j tu rdS | j tu s J �| jS )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, rL rW r) r) r* �close_expected s
"
zProtocol.close_expected�Generator[None]c
c s� � zO | j �� E dH r| jr| j�d� td��| jdu r d}n| jdu r)| j}n| j| j }tj| j j | j
t u || j d�E dH }| jrK| j�d|� | �
|� q tyn } z| �tjt|�� || _W Y d}~n�d}~w ty� } z| �tjt|�� || _W Y d}~nld}~w ty� } z| �tj|j� d|j� �� || _W Y d}~nJd}~w ty� } z|�| j� | �tjt|�� || _W Y d}~n'd}~w t y� } z| jj ddd � | �tj � || _W Y d}~nd}~ww dV t d
��) 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.
TN�< EOFzunexpected end of stream)�maskr2 rF z< %sz
at position z
parser failed)�exc_infoz"parse() shouldn't step after error) rM �at_eofrB r3 �EOFErrorr2 rC r rP �
read_exactr4 r' rF �
recv_framer r� r �PROTOCOL_ERRORrw rS rX �UnicodeDecodeError� INVALID_DATAr] �startr
�set_current_size�MESSAGE_TOO_BIG� Exception�error�INTERNAL_ERROR�AssertionError)rT r2 �framera r) r) r* rP / s^ �
�
� �� � � ��zProtocol.parsec c s� � | j tu p
| jtu | jksJ �| j�� E dH s%| j�� | j�� E dH r| jr.| j �d� | j t
u r<| jtur<| � � t | _dV t
d��)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)r4 r' r1 r, rL rM r� r� rB r3 r( r� r/ r� rW r) r) r* r� w s �
� zProtocol.discardr� r c C sj |j tu s
|j tu r | jdurtd��|js t|j�| _n�|j tu r=| jdu r+td��|jr2d| _n{| jt|j�7 _np|j t u rNt
t |j�}| � |� n_|j t u rTnY|j t
u r�t�|j�| _| jtu ro| jduslJ �d| _| jdurxtd��| jtu r�| � t
t
|j�� | j| _d| _t| _| jtu r�| �� | �� | _t| j� n t d|j d����| j � |� dS ) z-
Process an incoming frame.
Nrs rm Fz incomplete fragmented messageTzunexpected opcode: �02x) �opcoder r rC r rk �lenrc r r r r ro r r rP rH r1 r. rI rJ r- r4 r' r� r� rQ rR r� rN �append)rT r� �
pong_framer) r) r* r� � sF
�
zProtocol.recv_framec C s6 | j r
| j� d|� | j�|j| jtu | jd�� d S )Nz> %s)r� rF )rB r3 rO r� ry r4 r( rF )rT r� r) r) r* ro � s ��zProtocol.send_framec C s2 | j rJ �d| _ | jr| j�d� | j�t� d S )NTz> EOF)rL rB r3 rO r� r rW r) r) r* r� � s
zProtocol.send_eof)
r4 r r1 r r2 r5 r3 r6 r7 r8 )r7 r )r1 r r7 r8 )r7 r5 )r7 r[ )r7 r )rc rd r7 r8 )r7 r8 )rc rd rk rl r7 r8 )T)Nr\ )rY r5 r] rw r7 r8 )r\ )rY r� r] rw r7 r8 )r7 r� )r7 r� )r7 rl )r7 r� )r� r r7 r8 ) r" r# r$ r% r- rU �propertyr1 �setterrZ r^ rb rg rj rq rt rv r| r~ r� r� r� r� r� rP r� r� ro r� r) r) r) r* r I sB �D
%
$
1
-
H
R
r )2�
__future__r �enumr< r9 �collections.abcr �typingr �
exceptionsr r r r r
r rF r �framesr
r r r r r r r r r �http11r r �streamsr r r r �__all__�Event�IntEnumr r' r( r r, r- r. r/ r r r) r) r) r* �<module> s4 0