U
'�7`
b � @ sj d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
mZmZmZmZmZ ddlmZ ddlmZ dd lmZ d
ZG d d � d e �Zd
d� eD �ZG dd� de �Z dZ e d�j Z e d�j Z e d�j!Z"e d�j#Z$e d�j#Z%e d�j#Z&e d�j#Z'dZ(dZ)e�*dddd g�Z+G d d � d e+�Z,e,e j-dd�Z.e,e j/dd�Z0G d d � d e1�Z2G d!d"� d"e1�Z3ej4Z5d#d$� e6d%�D �Z7e8e9dd&�d'd(�Z:e�r�e:Z;n0zdd)l<m=Z= e=Z;W n e>k
�r� e:Z;Y nX e8ddd*d*g�Z?e�@d+�ZAe�@d,�ZBd;eeC eDeeEeDf d.�d/d0�ZFd<eEeDeDeCd2�d3d4�ZGG d5d6� d6e �ZHG d7d8� d8�ZIG d9d:� d:�ZJdS )=z%WebSocket protocol versions 13 and 8.� N)�IntEnum)�Struct)�Any�Callable�List�Optional�Tuple�Union� )� BaseProtocol)�
NO_EXTENSIONS)� DataQueue) �WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketReader�WebSocketWriter� WSMessage�WebSocketError� WSMsgType� WSCloseCodec @ s8 e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d S )
r �� i� i� i� i� i� i� i� i� i� i� N)�__name__�
__module__� __qualname__ZOKZ
GOING_AWAY�PROTOCOL_ERRORZUNSUPPORTED_DATA� INVALID_TEXTZPOLICY_VIOLATION�MESSAGE_TOO_BIGZMANDATORY_EXTENSIONZINTERNAL_ERRORZSERVICE_RESTARTZTRY_AGAIN_LATER� r r �I/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/http_websocket.pyr s r c C s h | ] }t |��qS r )�int)�.0�ir r r � <setcomp>- s r# c @ sP e Zd ZdZdZdZdZdZdZdZ dZ
d Z eZ eZ
eZeZeZe Ze
Ze Zd
S ) r r r
� � �
� � i i N)r r r � CONTINUATION�TEXT�BINARY�PING�PONG�CLOSE�CLOSING�CLOSEDZERROR�text�binary�ping�pong�close�closing�closed�errorr r r r r 0 s" r s$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11z!Hz!Qz!BBz!BBHz!BBQi @ � �_WSMessageBase�type�dataZextrac @ s. e Zd Zejd�eegef ed�dd�ZdS )r )�loads)r= �returnc C s
|| j �S )zAReturn parsed JSON data.
.. versionadded:: 0.22
)r< )�selfr= r r r �jsonZ s zWSMessage.jsonN)r r r r@ r= r r r r r r r Y s r c s8 e Zd ZdZeedd�� fdd� Zed�dd�Z� ZS ) r z WebSocket protocol parser error.N��code�messager> c s || _ t� �||� d S �N)rB �super�__init__�r? rB rC �� __class__r r rF i s zWebSocketError.__init__�r> c C s
| j d S )Nr
)�args�r? r r r �__str__m s zWebSocketError.__str__) r r r �__doc__r �strrF rM �
__classcell__r r rH r r f s r c @ s e Zd ZdZdS )�WSHandshakeErrorz#WebSocket protocol handshake error.N)r r r rN r r r r rQ q s rQ c s& g | ] � t � fd d�td�D ���qS )c 3 s | ]}|� A V qd S rD r )r! �a��br r � <genexpr>y s z<listcomp>.<genexpr>r( )�bytes�range)r! r rS r �
<listcomp>y s rX r( )�maskr< r> c C s� t |t�st|��t| �dks&t| ��|r�dd� | D �\}}}}|ddd� �|�|ddd�< |ddd� �|�|ddd�< |ddd� �|�|ddd�< |ddd� �|�|ddd�< 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.
� c s s | ]}t | V qd S rD )�
_XOR_TABLE)r! �nr r r rU � s z)_websocket_mask_python.<locals>.<genexpr>Nr
r$ � )�
isinstance� bytearray�AssertionError�len� translate)rY r< rR rT �c�dr r r �_websocket_mask_python| s
re )�_websocket_mask_cython� z�^(?:;\s*(?:(server_no_context_takeover)|(client_no_context_takeover)|(server_max_window_bits(?:=(\d+))?)|(client_max_window_bits(?:=(\d+))?)))*$zpermessage-deflate([^,]+)?F)�extstr�isserverr> c C s� | sdS d}d}t �| �D ]�}|�d�}|s4d} q�t�|�}|r�d}|r�|�d�rxt|�d��}|dksr|dk rxd}q|�d�r�d} q�q�|�d �r�t|�d ��}|dks�|dk r�td
��|�d �r�d} q�q|std |�d� ��q||fS )
N)r Fr Fr
� rZ r% T� zInvalid window sizer$ z#Extension for deflate not supported)�_WS_EXT_RE_SPLIT�finditer�group�
_WS_EXT_RE�matchr rQ )rh ri �compress�
notakeoverZextZdefextrp r r r � ws_ext_parse� s>
rs rj )rq ri �server_notakeoverr> c C s^ | dk s| dkrt d��dg}|s,|�d� | dk rF|�dt| � � |rT|�d� d�|�S ) Nr% rj zCCompress wbits must between 9 and 15, zlib does not support wbits=8zpermessage-deflateZclient_max_window_bitszserver_max_window_bits=Zserver_no_context_takeoverz; )�
ValueError�appendrO �join)rq ri rt Z
enabledextr r r �
ws_ext_gen� s �
rx c @ s e Zd ZdZdZdZdZdS )�
WSParserStater
r$ r] rZ N)r r r � READ_HEADER�READ_PAYLOAD_LENGTH�READ_PAYLOAD_MASK� READ_PAYLOADr r r r ry � s ry c @ s� e Zd Zdee eedd�dd�Zdd�dd�Ze e
ee f d �d
d �Z e e
ee f d �d d
�Z e e
e
eee eee f d�dd�ZdS )r TN)�queue� max_msg_sizerq r> c C sp || _ || _d | _t� | _tj| _d | _d| _ d | _
t� | _ d| _ d| _
d | _d| _d| _d | _d | _|| _d S )NF� r )r~ �
_max_msg_size�_excr_ �_partialry rz �_state�_opcode�
_frame_fin�
_frame_opcode�_frame_payload�_tail� _has_mask� _frame_mask�_payload_length�_payload_length_flag� _compressed�_decompressobj� _compress)r? r~ r rq r r r rF � s" zWebSocketReader.__init__rJ c C s | j �� d S rD )r~ �feed_eofrL r r r r� s zWebSocketReader.feed_eof�r< r> c
C sZ | j rd|fS z | �|�W S tk
rT } z || _ | j�|� W Y �dS d }~X Y nX d S )NT)Tr� )r� �
_feed_data� Exceptionr~ Z
set_exception)r? r< �excr r r � feed_data s zWebSocketReader.feed_datac
C s� | � |�D �]v\}}}}|r4| js4tjtj d�| _|tjk�r t|�dkr�t|d d� �d }|dk r�|t kr�t
t j d|� ���z|dd � �
d�}W n0 tk
r� } zt
t jd�|�W 5 d }~X Y nX ttj||�} n2|r�t
t j d|� d |� d |����nttjdd
�} | j�| d� q
|tjk�rF| j�ttj|d
�t|�� q
|tjk�rp| j�ttj|d
�t|�� q
|tjtjfk�r�| jd k�r�t
t j d |����q
|�s�|tjk�r�|| _| j�|� | j�r�t| j�| jk�r�t
t j d � t| j�| j���q
| j�r$|tjk�r$t
t j d
� |���|tjk�rL| jd k �s@t �| j}d | _| j�|� | j�r�t| j�| jk�r�t
t j d � t| j�| j���|�r�| j�t � | j� | j| j�}
| jj!�r�t| jj!�} t
t j d� | j| | j���n
t"| j�}
| j�#� |tjk�rfz*|
�
d�} | j�ttj| d
�t| �� W n2 tk
�rb } zt
t jd�|�W 5 d }~X Y nX q
| j�ttj|
d
�t|
�� q
dS )N)�wbitsr$ r i� zInvalid close code: �utf-8zInvalid UTF-8 text messagezInvalid close frame: � � zUnexpected opcode=z Message size {} exceeds limit {}z<The opcode in non-fin frame is expected to be zero, got {!r}z-Decompressed message size {} exceeds limit {})Fr� )$� parse_framer� �zlibZ
decompressobjZ MAX_WBITSr r. ra �UNPACK_CLOSE_CODE�ALLOWED_CLOSE_CODESr r r �decode�UnicodeDecodeErrorr r r~ r� r, r- r* r+ r� r) r� �extendr� r �formatr` �_WS_DEFLATE_TRAILING�
decompressZunconsumed_tailrV �clear)
r? r< �fin�opcode�payloadZ
compressedZ
close_codeZ
close_messager� �msgZpayload_merged�leftr1 r r r r� # s�
� ��� � ��� � �� �� �� �
��
� �� �zWebSocketReader._feed_data)�bufr> c C s� g }| j r | j | d }| _ d}t|�}| jtjk�rp|| dk�rt|||d � }|d7 }|\}}|d? d@ }|d? d@ } |d? d@ }
|d? d@ } |d @ } |
s�| s�| r�| js�ttjd
��| dkr�|dkr�ttjd ��|d? d@ }
|d @ }| dk�r
|d
k�r
ttjd��| j �s | j
dk�r0| �r(dnd| _
n| �rBttjd
��t |�| _ | | _ t |
�| _
|| _tj| _n�qt| jtjk�r`| j}|dk�r�|| dk�rt|||d � }|d7 }t|�d }|| _| j
�r�tjntj| _n�qtn||dk�rD|| dk�rt|||d � }|d7 }t|�d }|| _| j
�r4tjntj| _n�qtn || _| j
�rXtjntj| _| jtjk�r�|| dk�rt|||d � | _|d7 }tj| _n�qt| jtjkr(| j}| j}|| }||k�r�|| | _|�||d� � |}n$d| _|�|||| � � || }| jdk�rt| j
�rD| jdk �s8t�t| j|� |�| j | j || j
f� t� | _tj| _q(�qtq(||d� | _ |S )z&Return the next frame from the socket.r� r r$ � r
rk � rZ rj z*Received frame with non-zero reserved bitsz!Received fragmented control frame� �} z5Control frame payload cannot be larger than 125 bytesNTF�~ r' ) r� ra r� ry rz r� r r r r� r� �boolr� r� r� r{ � UNPACK_LEN2r� r| r} � UNPACK_LEN3r� r� r� r` �_websocket_maskrv r_ )r? r� ZframesZ start_posZ
buf_lengthr< Z
first_byteZ second_byter� Zrsv1Zrsv2Zrsv3r� Zhas_maskZlengthr� Z chunk_lenr r r r� � s� �� ��
�
�
�
��
�
�
zWebSocketReader.parse_frame)T)r r r r
r r r� rF r� rV r r� r� r r r_ r� r r r r r � s � � |�r c
@ s� e Zd Zdee�� ddd�eeje e
e e
e dd�dd�Z de
e
ee
dd�d d
�Ze
dd �d d
�Zd e
dd�dd�Zd e
dd�dd�Zd eee
f e ee
dd�dd�Zd e
e
dd�dd�ZdS ) r Fr )�use_mask�limit�randomrq rr N)�protocol� transportr� r� r� rq rr r> c C sB || _ || _|| _|j| _|| _|| _d| _|| _d| _d | _ d S )NFr )
r� r� r� � randrangerq rr �_closing�_limit� _output_size� _compressobj)r? r� r� r� r� r� rq rr r r r rF ; s zWebSocketWriter.__init__)rC r� rq r> c � s� | j r|tj@ std��d}|s&| jr�|dk r�|rFtjtj| d�}n"| jsbtjtj| j d�| _| j}|�|�}||� | j
r�tj ntj � }|�
t�r�|dd� }|dB }t|�}| j}|r�d}nd}|d k r�td|B |B ||B �} n<|d
k �r td|B |B d |B |�} ntd|B |B d |B |�} |�r�| �dd �}
|
�d
d�}
t|�}t|
|� | �| |
| � | jt| �t|
� t|� 7 _nLt|�tk�r�| �| � | �|� n| �| | � | jt| �t|� 7 _| j| jk�r�d| _| j � � I dH dS )�<Send a frame over the websocket with message as its payload.�!Cannot write to closing transportr r' )�levelr� N����@ � r� r9 r� l �� rZ Zbig) r� r r. �ConnectionResetErrorrq r� � compressobjZ Z_BEST_SPEEDr� �flushrr Z Z_FULL_FLUSHZ Z_SYNC_FLUSH�endswithr� ra r� � PACK_LEN1� PACK_LEN2� PACK_LEN3r� �to_bytesr_ r� �_writer� �MSG_SIZEr� r� Z
_drain_helper) r? rC r� rq Zrsvr� Z
msg_lengthr� Zmask_bit�headerrY r r r � _send_frameQ sX �
�
$
zWebSocketWriter._send_framer� c C s, | j d ks| j �� r td��| j �|� d S )Nr� )r� Z
is_closingr� �write)r? r< r r r r� � s zWebSocketWriter._writer� )rC r> c � s, t |t�r|�d�}| �|tj�I dH dS )zSend pong message.r� N)r^ rO �encoder� r r- �r? rC r r r r4 � s
zWebSocketWriter.pongc � s, t |t�r|�d�}| �|tj�I dH dS )zSend ping message.r� N)r^ rO r� r� r r, r� r r r r3 � s
zWebSocketWriter.ping)rC r2 rq r> c � sJ t |t�r|�d�}|r0| �|tj|�I dH n| �|tj|�I dH dS )r� r� N)r^ rO r� r� r r+ r* )r? rC r2 rq r r r �send� s
zWebSocketWriter.sendr rA c � sD t |t�r|�d�}z"| jt|�| tjd�I dH W 5 d| _X dS )z<Close the websocket, sending the specified code and message.r� T)r� N)r^ rO r� r� r� �PACK_CLOSE_CODEr r. rG r r r r5 � s
�zWebSocketWriter.close)N)r� )r� )FN)r r� )r r r �
DEFAULT_LIMITr� ZRandomr �asyncioZ Transportr� r r rF rV r r� r� r4 r3 r rO r� r5 r r r r r : sD �� � � A �
� r )F)rj FF)KrN r� � collectionsr@ r� �re�sysr� �enumr �structr �typingr r r r r r Z
base_protocolr Zhelpersr Zstreamsr
�__all__r r� r r � unpack_fromr� r� �unpackr� �packr� r� r� r� r� r� �
namedtupler: r r0 r r/ r r� r rQ � byteorderZnative_byteorderrW r[ rV r_ re r� Z
_websocketrf � ImportErrorr� �compilero rl rO r� r rs rx ry r r r r r r �<module> s~
�
4 � � @