File: //lib/python3/dist-packages/websocket/__pycache__/_abnf.cpython-310.pyc
o
_7�a/3 �
@ s d Z ddlZddlZddlZddlZddlT ddlmZ ddlm Z z ddl
m Z dd � Z W n e
y? ejZd
d � Z Y nw g d �Zd Zd
ZdZdZdZdZdZdZdZdZdZdZ dZ eeeeeeeeee f
Z G dd� d�Z G dd � d �Z G d d � d �Z!dS ) z
� N� )�*)�
validate_utf8)�Lock)�XorMaskerSimplec C s t | ��|�S �N)r �process)�_m�_d� r �1/usr/lib/python3/dist-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)�
mask_value�
data_value�datalenr 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� c @ s� e Zd ZdZdZdZdZdZdZdZ eeeeee fZ
eded ed
ed ed e d
iZ dZ dZ
dZddddeddfdd�Zd$dd�Zedd� �Zdd� Zed%dd ��Zd d � Zd d � Zed!d"� �Zd#S )&r 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 � c C sB || _ || _|| _|| _|| _|| _|du rd}|| _tj| _ dS )zG
Constructor for ABNF. Please check RFC for arguments.
Nr2 )
�fin�rsv1�rsv2�rsv3�opcode�mask�data�os�urandom� get_mask_key)�selfr3 r4 r5 r6 r7 r8 r9 r r r �__init__� s z
ABNF.__init__Fc C s� | j s | js | jr
td��| jtjvrtd| j��| jtjkr&| js&td��| jtj krit
| j �}|s5dS |dks=|dkrAtd��|dkrT|sTt | j dd� �sTtd��d | j d
| j d }| �
|�sktd ��dS dS ) z�
Validate the ABNF frame.
Parameters
----------
skip_utf8_validation: skip utf8 validation.
zrsv is not implemented, yetzInvalid opcode %rzInvalid ping frame.Nr r1 zInvalid close frame.r' � r zInvalid close opcode.)r4 r5 r6 �WebSocketProtocolExceptionr7 r �OPCODES� OPCODE_PINGr3 � OPCODE_CLOSEr r9 r �_is_valid_close_status)r= �skip_utf8_validation�l�coder r r �validate� s&
�
z
ABNF.validatec C s | t v pd| ko
dk S S )Ni� i� )�VALID_CLOSE_STATUS)rG r r r rD � s zABNF._is_valid_close_statusc C s* dt | j� d t | j� d t | j� S )Nzfin=z opcode=z data=)�strr3 r7 r9 �r= 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: <type>
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: <type>
operation code. please see OPCODE_XXX.
fin: <type>
fin flag. if set to 0, create continue fragmentation.
�utf-8r r )r � OPCODE_TEXT�
isinstancerJ �encode)r9 r7 r3 r r r � create_frame� s
zABNF.create_framec C sF 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�|| 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-1r1 �!H� �!Q)�anyr3 r4 r5 r6 �
ValueErrorr7 r rA r r9 � LENGTH_63�chrrP �LENGTH_7r8 � LENGTH_16�struct�packr< � _get_masked)r= �length� frame_header�mask_keyr r r �format� s: "
�����
z ABNF.formatc C s* t �|| j�}t|t�r|�d�}|| S )NrM )r r8 r9 rO rJ rP )r= rg �sr r r rd � 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: <type>
4 byte string.
data: <type>
data to mask/unmask.
Nr2 rX �B)rO rJ rP r
�array)rg r9 r r r r8 � s
z ABNF.maskN)F)r )�__name__�
__module__� __qualname__�__doc__� OPCODE_CONTrN �
OPCODE_BINARYrC rB � OPCODE_PONGrA �
OPCODE_MAPr` ra r^ r> rH � staticmethodrD rL rQ rh rd r8 r r r r r c sD ��
�
r c @ sl e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )r rW rV c C s& || _ || _g | _| �� t� | _d S r )�recvrE � recv_buffer�clearr �lock)r= �recv_fnrE r r r r> s
zframe_buffer.__init__c C s d | _ d | _d | _d S r )�headerre r8 rK r r r rw s
zframe_buffer.clearc C �
| j d u S r )rz rK 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 rU r rV rW r � rZ )� recv_strictrz ) r= rz �b1r3 r4 r5 r6 r7 �b2�has_mask� length_bitsr r r � recv_header$ s
zframe_buffer.recv_headerc C s | j sdS | j tj S )NF)rz r �_HEADER_MASK_INDEXrK r r r r� 2 s zframe_buffer.has_maskc C r{ r )re rK r r r �has_received_length7 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 )NrZ r1 r' rY r r( r[ )rz r �_HEADER_LENGTH_INDEXr rb �unpackre )r= �bitsr� �vr r r � recv_length: s
zframe_buffer.recv_lengthc C r{ r )r8 rK r r r �has_received_maskF r} z frame_buffer.has_received_maskc C s" | � � r | �d�| _d S d| _d S )Nr r2 )r� r r8 rK r r r � recv_maskI 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 )rx r| r� rz r� r� re r� r� r8 r r rw rH rE ) r= r3 r4 r5 r6 r7 r� �_re r8 �payload�framer r r �
recv_frameL s(
��zframe_buffer.recv_framec C s� |t tt| j�� }|dkr&| �td|��}| j�|� |t|�8 }|dksd�| j�}|dkr5g | _|S ||d � g| _|d |� S )Nr i @ � )�sum�mapr rv ru �min�append�join)r= �bufsize�shortage�bytes_�unifiedr r r r k s � zframe_buffer.recv_strictN)rl rm rn r� r� r> rw r| r� r� r� r� r� r� r� r r r r r r s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) r c C s || _ || _d | _d | _d S r )�fire_cont_framerE � cont_data�recving_frames)r= r� rE r r r r> � s
zcontinuous_frame.__init__c C sB | j s
|jtjkr
td��| j r |jtjtjfv r td��d S d S )Nz
Illegal frame)r� r7 r rp r@ rN rq �r= r� r r r rH � 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� r9 r7 r rN rq r� r3 r� r r r �add� s
�zcontinuous_frame.addc C s |j p| jS r )r3 r� r� r r r �is_fire� s zcontinuous_frame.is_firec C sX | j }d | _ |d |_| js&|d tjkr&| js&t|j�s&tdt|j� ��|d |gS )Nr r zcannot decode: ) r� r9 r� r rN rE r �WebSocketPayloadException�repr)r= r� r9 r r r �extract� s
$ � zcontinuous_frame.extractN)rl rm rn r> rH r� r� r� r r r r r � s r )"ro rk r: rb �sys� _exceptions�_utilsr � threadingr �wsaccel.xormaskr r
� ImportError� byteorderr �__all__r r r r r r r r! r" r# r$ r% r& rI r r r r r r r �<module> sZ � � -s