File: //usr/local/lib/python3.10/dist-packages/aiohttp/_websocket/__pycache__/helpers.cpython-310.pyc
o
���g� � @ s� U d Z ddlZddlZddlmZ ddlmZmZmZm Z m
Z
m Z ddl m
Z
ddlmZ ed �jZed
�jZed �jZed �jZed
�jZed
�jZed�jZdZee e d< dZ ee e d< dZ ee e d< ej dee fdd��Z!de de"ddfdd�Z#es�e
r�e#Z$nz
dd l%m&Z& e&Z$W n e'y� e#Z$Y nw e�(d �Z)ee
e* e d < e�(d �Z+ee
e* e d < d+d"e e* d#e,de ee,f fd$d%�Z- !d,d'ed#e,d(e,de*fd)d*�Z.dS )-z1Helpers for WebSocket protocol versions 13 and 8.� N)�Struct)�
TYPE_CHECKING�Final�List�Optional�Pattern�Tuple� )�
NO_EXTENSIONS� )�WSHandshakeErrorz!Qz!Hz!BBz!BBHz!BBQz!Li @ �MSG_SIZE� �MASK_LENs$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11�WS_KEY�returnc C s dd� t d�D �S )Nc s& g | ]� t � fd d�td�D ���qS )c 3 s � | ]}|� A V qd S �N� )�.0�a��br �E/usr/local/lib/python3.10/dist-packages/aiohttp/_websocket/helpers.py� <genexpr> � � z(_xor_table.<locals>.<listcomp>.<genexpr>� )�bytes�range)r r r r �
<listcomp> s & z _xor_table.<locals>.<listcomp>r )r r r r r �
_xor_table s r �mask�datac s� t |t�s J |��t| �dksJ | ��|rct� � � fdd�| D �\}}}}|ddd� �|�|ddd�< |ddd� �|�|ddd�< |ddd� �|�|ddd�< |ddd� �|�|ddd�< dS dS )aw Websocket masking function.
`mask` is a `bytes` object of length 4; `data` is a `bytearray`
object of any length. The contents of `data` are masked with `mask`,
as specified in section 5.3 of RFC 6455.
Note that this function mutates the `data` argument.
This pure-python implementation may be replaced by an optimized
version when available.
r c 3 s � | ]}� | V qd S r r )r �n��
_XOR_TABLEr r r 0 r z)_websocket_mask_python.<locals>.<genexpr>Nr r � )�
isinstance� bytearray�lenr � translate)r r! r r �c�dr r# r �_websocket_mask_python s
"�r, )�_websocket_mask_cythonz�^(?:;\s*(?:(server_no_context_takeover)|(client_no_context_takeover)|(server_max_window_bits(?:=(\d+))?)|(client_max_window_bits(?:=(\d+))?)))*$�
_WS_EXT_REzpermessage-deflate([^,]+)?�_WS_EXT_RE_SPLITF�extstr�isserverc C s� | sdS d}d}t �| �D ]m}|�d�}|s d} ||fS t�|�}|rod}|rK|�d�r?t|�d��}|dks<|dk r?d}q
|�d�rFd} ||fS |�d �rct|�d ��}|dks_|dk rctd
��|�d �rjd} ||fS |sztd |�d� ��q
||fS )
N)r Fr Fr � r � T� zInvalid window sizer z#Extension for deflate not supported)r/ �finditer�groupr. �match�intr )r0 r1 �compress�
notakeover�ext�defextr7 r r r � ws_ext_parseM sF
%
�
�
��r= r2 r9 �server_notakeoverc C s^ | dk s| dkr t d��dg}|s|�d� | dk r#|�dt| � � |r*|�d� d�|�S ) Nr3 r2 zCCompress wbits must between 9 and 15, zlib does not support wbits=8zpermessage-deflate�client_max_window_bitszserver_max_window_bits=�server_no_context_takeoverz; )�
ValueError�append�str�join)r9 r1 r> �
enabledextr r r �
ws_ext_gen� s �
rF )F)r2 FF)/�__doc__� functools�re�structr �typingr r r r r r �helpersr
�modelsr � unpack_from� UNPACK_LEN3�unpack�UNPACK_CLOSE_CODE�pack� PACK_LEN1� PACK_LEN2� PACK_LEN3�PACK_CLOSE_CODE�
PACK_RANDBITSr
r8 �__annotations__r r r � lru_cacher r' r, �websocket_maskr r- � ImportError�compiler. rC r/ �boolr= rF r r r r �<module> sV
��$4�����