File: //usr/local/CyberCP/lib64/python3.10/site-packages/websocket/__pycache__/_abnf.cpython-310.pyc
o
�h18 � @ sJ d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZm Z ddl
m Z m Z ddl
mZ zd dlmZ de j d e j d
efd d �ZW n ey] ejZde j d e j d
efd
d �ZY nw g d�ZdZdZdZdZdZdZ dZ dZ dZ dZ dZ!dZ"dZ#d Z$d Z%eeeee e e e e!e"e#e$f Z&G d d � d �Z'G d d!� d!�Z(G d"d#� d#�Z)dS )$� N)�Lock)�Callable�Optional�Union� )�WebSocketPayloadException�WebSocketProtocolException)�
validate_utf8)�XorMaskerSimple�
mask_value�
data_value�returnc C s t | ��|�}|S �N)r
�process)r r � mask_result� r �B/usr/local/CyberCP/lib/python3.10/site-packages/websocket/_abnf.py�_mask% s r c C sH t |�}t�|t�}t�| |d | d |d � t�}||A �|t�S )N� )�len�int�
from_bytes�native_byteorder�to_bytes)r r �datalen�int_data_value�int_mask_valuer r r r - s �)�ABNF�continuous_frame� frame_buffer�
STATUS_NORMAL�STATUS_GOING_AWAY�STATUS_PROTOCOL_ERROR� STATUS_UNSUPPORTED_DATA_TYPE�STATUS_STATUS_NOT_AVAILABLE�STATUS_ABNORMAL_CLOSED�STATUS_INVALID_PAYLOAD�STATUS_POLICY_VIOLATION�STATUS_MESSAGE_TOO_BIG�STATUS_INVALID_EXTENSION�STATUS_UNEXPECTED_CONDITION�STATUS_BAD_GATEWAY�STATUS_TLS_HANDSHAKE_ERRORi� i� i� i� i� i� i� i� i� i� i� i� i� i� i� c @ sV e Zd ZdZdZdZdZdZdZdZ eeeeee fZ
eded ed
ed ed e d
iZ dZ dZ
dZddddeddfdededededededeeedf ddfdd �Zd/d eddfd d �Zed!edefd"d#��Zdefd$d%�Zed0deeef dededd fd&d'��Zdefd(d)�Zd*eeef defd+d,�Zed*eeef deeef defd-d.��Z dS )1r z}
ABNF frame class.
See http://tools.ietf.org/html/rfc5234
and http://tools.ietf.org/html/rfc6455#section-5.2
r r � � � �
�cont�text�binary�close�ping�pong�~ i l � �fin�rsv1�rsv2�rsv3�opcoder �dataNr
c C sB || _ || _|| _|| _|| _|| _|du rd}|| _tj| _ dS )zG
Constructor for ABNF. Please check RFC for arguments.
Nr8 )
r9 r: r; r< r= r r> �os�urandom� get_mask_key)�selfr9 r: r; r< r= r r> r r r �__init__� s
z
ABNF.__init__F�skip_utf8_validationc C s� | j s | js | jr
td��| jtjvrtd| j��| jtjkr&| js&td��| jtj krnt
| j �}|s5dS |dks=|dkrAtd��|dkrT|sTt | j dd� �sTtd��d t
| j d
� t
| j d � }| �|�sptd |��dS dS ) z�
Validate the ABNF frame.
Parameters
----------
skip_utf8_validation: skip utf8 validation.
zrsv is not implemented, yetzInvalid opcode %rzInvalid ping frame.Nr r7 zInvalid close frame.r- � r zInvalid close opcode %r)r: r; r< r r= r �OPCODES� OPCODE_PINGr9 � OPCODE_CLOSEr r> r r �_is_valid_close_status)rB rD �l�coder r r �validate� s&
�
z
ABNF.validaterK c C s | t v pd| ko
dk S S )Ni� i� )�VALID_CLOSE_STATUS)rK r r r rI � s zABNF._is_valid_close_statusc C s d| j � d| j� d| j� �S )Nzfin=z opcode=z data=)r9 r= r> �rB r r r �__str__� s z ABNF.__str__c C s2 |t jkrt| t�r| �d�} t |ddd|d| �S )a�
Create frame to send text, binary and other data.
Parameters
----------
data: str
data to send. This is string value(byte array).
If opcode is OPCODE_TEXT and this value is unicode,
data value is converted into unicode string, automatically.
opcode: int
operation code. please see OPCODE_MAP.
fin: int
fin flag. if set to 0, create continue fragmentation.
�utf-8r r )r � OPCODE_TEXT�
isinstance�str�encode)r> r= r9 r r r � create_frame� s
zABNF.create_framec C s` t dd� | j| j| j| jfD ��rtd��| jtjvr td��t | j
�}|tj kr-td��t | jd> | jd> B | jd> B | jd > B | jB ��
d
�}|tjk r]|t | jd> |B ��
d
�7 }n2|tjk ry|t | jd> d B ��
d
�7 }|t�d |�7 }n|t | jd> d
B ��
d
�7 }|t�d|�7 }| js�t| j
t�r�| j
�
d�| _
|| j
S | �d �}|| �|� S )zR
Format this object to string(byte array) to send data to server.
c s s � | ]}|d vV qdS ))r r Nr )�.0�xr r r � <genexpr>� s � z ABNF.format.<locals>.<genexpr>z
not 0 or 1zInvalid OPCODEzdata is too long� � � r �latin-1r7 �!H� �!QrP )�anyr9 r: r; r< �
ValueErrorr= r rF r r> � LENGTH_63�chrrT �LENGTH_7r � LENGTH_16�struct�packrR rS rA � _get_masked)rB �length� frame_header�mask_keyr r r �format� sB "
������
z ABNF.formatrk c C s* t �|| j�}t|t�r|�d�}|| S )NrP )r �maskr> rR rS rT )rB rk �sr r r rh s
zABNF._get_maskedc C sN |du rd}t | t�r| �d�} t |t�r|�d�}tt�d| �t�d|��S )z�
Mask or unmask data. Just do xor for each byte
Parameters
----------
mask_key: bytes or str
4 byte mask.
data: bytes or str
data to mask/unmask.
Nr8 r\ �B)rR rS rT r �array)rk r> r r r rm s
z ABNF.mask)F)r ) �__name__�
__module__� __qualname__�__doc__� OPCODE_CONTrQ �
OPCODE_BINARYrH rG � OPCODE_PONGrF �
OPCODE_MAPrd re rb r r rS �bytesrC �boolrL � staticmethodrI rO rU rl rh rm r r r r r j sv � �
������� �
� &#,r c @ s� e Zd ZdZdZdeegef deddfdd�Zd d d
�Z defd d �Z
d d
d�Z de eef fdd�Z
defdd�Zd dd�Zdefdd�Zd dd�Zdefdd�Zdedefd d �ZdS ) r r[ rZ �recv_fnrD r
Nc C s& || _ || _g | _| �� t� | _d S r )�recvrD � recv_buffer�clearr �lock)rB r| rD r r r rC + s
zframe_buffer.__init__c C s d | _ d | _d | _d S r )�headerri r rN r r r r 6 s
zframe_buffer.clearc C �
| j d u S r )r� rN r r r �has_received_header; �
z frame_buffer.has_received_headerc C s~ | � d�}|d }|d? d@ }|d? d@ }|d? d@ }|d? d@ }|d@ }|d }|d? d@ } |d @ }
|||||| |
f| _d S )
Nr- r rY r rZ r[ r � r^ )� recv_strictr� ) rB r� �b1r9 r: r; r< r= �b2�has_mask� length_bitsr r r � recv_header> s
zframe_buffer.recv_headerc C s | j sdS | j tj }|S )NF)r� r �_HEADER_MASK_INDEX)rB �
header_valr r r r� L s zframe_buffer.has_maskc C r� r )ri rN r r r �has_received_lengthR r� z frame_buffer.has_received_lengthc C sn | j tj }|d@ }|dkr | �d�}t�d|�d | _d S |dkr2| �d�}t�d|�d | _d S || _d S )Nr^ r7 r- r] r r. r_ )r� r �_HEADER_LENGTH_INDEXr� rf �unpackri )rB �bitsr� �vr r r � recv_lengthU s
zframe_buffer.recv_lengthc C r� r )r rN r r r �has_received_maska r� z frame_buffer.has_received_maskc C s" | � � r | �d�| _d S d| _d S )Nr r8 )r� r� r rN r r r � recv_maskd s "zframe_buffer.recv_maskc C s� | j �R | �� r | �� | j\}}}}}}}| �� r | �� | j}| �� r)| �� | j } | �
|�}
|r9t � | |
�}
| �
� t |||||||
�} | �| j� W d � | S 1 sXw Y | S r )r� r� r� r� r� r� ri r� r� r r� r rm r rL rD ) rB r9 r: r; r< r= r� �_ri r �payload�framer r r �
recv_frameg s(
��zframe_buffer.recv_frame�bufsizec C s� |t tt| j�� }|dkr&| �td|��}| j�|� |t|�8 }|dksd�| j�}|dkr5g | _|S ||d � g| _|d |� S )Nr i @ � )�sum�mapr r~ r} �min�append�join)rB r� �shortage�bytes_�unifiedr r r r� � s � zframe_buffer.recv_strict)r
N)rq rr rs r� r� r r rz rC r r� r� r r� r� r� r� r� r r� ry r� r r r r r ' s( ��
�
r c @ sr e Zd Zdededdfdd�Zdeddfdd �Zdeddfd
d �Zdedeee f fd d
�Z
dede fdd�Z dS )r �fire_cont_framerD r
Nc C s || _ || _d | _d | _d S r )r� rD � cont_data�recving_frames)rB r� rD r r r rC � s
zcontinuous_frame.__init__r� c C sB | j s
|jtjkr
td��| j r |jtjtjfv r td��d S d S )Nz
Illegal frame)r� r= r ru r rQ rv �rB r� r r r rL � s
��zcontinuous_frame.validatec C sX | j r| j d |j7 < n|jtjtjfv r|j| _|j|jg| _ |jr*d | _d S d S )Nr )r� r> r= r rQ rv r� r9 r� r r r �add� s
�zcontinuous_frame.addc C s |j p| jS r )r9 r� r� r r r �is_fire� s zcontinuous_frame.is_firec C sZ | j }d | _ |d |_| js'|d tjkr'| js't|j�s'tdt|j�� ���|d |fS )Nr r zcannot decode: ) r� r> r� r rQ rD r r �repr)rB r� r> r r r �extract� s
��� zcontinuous_frame.extract)
rq rr rs rz rC r rL r� r r r� �tupler� r r r r r � s r )*rp r? rf �sys� threadingr �typingr r r � _exceptionsr r �_utilsr �wsaccel.xormaskr
ry r � ImportError� byteorderr �__all__r r! r" r# r$ r% r& r' r( r) r* �STATUS_SERVICE_RESTART�STATUS_TRY_AGAIN_LATERr+ r, rM r r r r r r r �<module> sb �
� >u