Website : rimsha.abasa.com
backdoor
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
proc
/
self
/
root
/
lib
/
python3
/
dist-packages
/
websocket
/
__pycache__
/
Filename :
_abnf.cpython-310.pyc
back
Copy
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