File: //opt/alt/python37/lib64/python3.7/site-packages/aiohttp/__pycache__/http_websocket.cpython-37.pyc
B
�k�ac � @ s U d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
mZmZmZmZmZmZmZmZ ddlmZ ddlmZ dd lmZ dd
l m Z d Z G d d
� d
e �Z dd� e D �Z e ee! e"d< G dd� de �Z#dZ$e e% e"d< 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/dZ0e e! e"d< d Z1e e! e"d < e�2d d d d!g�Z3G d"d#� d#e3�Z4e4e#j5dd�Z6e4e#j7dd�Z8G d$d%� d%e9�Z:G d&d'� d'e9�Z;ej<Z=e e> e"d(< d)d*� e?d+�D �Z@e ee% e"d,< e%eAdd-�d.d/�ZBe�reBZCn0ydd0lDmEZE eEZCW n eFk
�rH eBZCY nX e%ddd1d1g�ZGe e% e"d2< e�Hd3�ZIe ee> e"d4< e�Hd5�ZJe ee> e"d6< dEee> eKee!eKf d8�d9d:�ZLdFe!eKeKe>d<�d=d>�ZMG d?d@� d@e �ZNG dAdB� dB�ZOG dCdD� dD�ZPdS )Gz%WebSocket protocol versions 13 and 8.� N)�IntEnum)�Struct) �Any�Callable�List�Optional�Pattern�Set�Tuple�Union�cast� )� BaseProtocol)�
NO_EXTENSIONS)� DataQueue)�Final) �WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketReader�WebSocketWriter� WSMessage�WebSocketError� WSMsgType� WSCloseCodec @ s@ e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdS )r i� i� i� i� i� i� i� i� i� i� i� i� i� N)�__name__�
__module__� __qualname__ZOKZ
GOING_AWAY�PROTOCOL_ERRORZUNSUPPORTED_DATAZABNORMAL_CLOSURE� INVALID_TEXTZPOLICY_VIOLATION�MESSAGE_TOO_BIGZMANDATORY_EXTENSIONZINTERNAL_ERRORZSERVICE_RESTARTZTRY_AGAIN_LATERZ BAD_GATEWAY� r! r! �I/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/http_websocket.pyr s r c C s h | ] }t |��qS r! )�int)�.0�ir! r! r" � <setcomp>0 s r& �ALLOWED_CLOSE_CODESc @ 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 3 s" r s$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11r z!Hz!Qz!BBz!BBHz!BBQi @ �MSG_SIZEi �
DEFAULT_LIMIT�_WSMessageBase�type�dataZextrac @ s. e Zd Zejd�eegef ed�dd�ZdS )r )�loads)rB �returnc C s
|| j �S )zAReturn parsed JSON data.
.. versionadded:: 0.22
)rA )�selfrB r! r! r" �json] s zWSMessage.jsonN)r r r rE rB r r r! r! r! r" r \ s r c s8 e Zd ZdZeedd�� fdd� Zed�dd�Z� ZS ) r z WebSocket protocol parser error.N)�code�messagerC c s || _ t� �||� d S )N)rF �super�__init__)rD rF rG )� __class__r! r" rI l s zWebSocketError.__init__)rC c C s t t| jd �S )Nr
)r �str�args)rD r! r! r" �__str__p s zWebSocketError.__str__) r r r �__doc__r# rK rI rM �
__classcell__r! r! )rJ r" r i s r c @ s e Zd ZdZdS )�WSHandshakeErrorz#WebSocket protocol handshake error.N)r r r rN r! r! r! r" rP t s rP �native_byteorderc s& g | ] � t � fd d�td�D ���qS )c 3 s | ]}|� A V qd S )Nr! )r$ �a)�br! r" � <genexpr>| s z<listcomp>.<genexpr>r, )�bytes�range)r$ r! )rS r" �
<listcomp>| s rW r, �
_XOR_TABLE)�maskrA rC 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 )N)rX )r$ �nr! r! r" rT � s z)_websocket_mask_python.<locals>.<genexpr>Nr
r( � )�
isinstance� bytearray�AssertionError�len� translate)rY rA rR rS �c�dr! r! r" �_websocket_mask_python s
rd )�_websocket_mask_cython� �_WS_DEFLATE_TRAILINGz�^(?:;\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�isserverrC c C s� | sdS d}d}x�t �| �D ]�}|�d�}|s4d}P t�|�}|r�d}|r�|�d�rxt|�d��}|dksr|dk rxd}q |�d�r�d}P q�|�d �r�t|�d ��}|dks�|dk r�td
��|�d �r�d}P q |s td |�d� ��q W ||fS )
N)r Fr Fr
� rZ r) T� zInvalid window sizer( z#Extension for deflate not supported)ri �finditer�grouprh �matchr# rP )rj rk �compress�
notakeoverZextZdefextrp r! r! r" � ws_ext_parse� s>
rs rl )rq rk �server_notakeoverrC c C s^ | dk s| dkrt d��dg}|s,|�d� | dk rF|�dt| � � |rT|�d� d�|�S ) Nr) rl 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�appendrK �join)rq rk 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 rC 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)rD r~ r rq r! r! r" rI s" zWebSocketReader.__init__)rC c C s | j �� d S )N)r~ �feed_eof)rD r! r! r" r� s zWebSocketReader.feed_eof)rA rC c
C sR | j rd|fS y
| �|�S tk
rL } z|| _ | j�|� dS d }~X Y nX d S )NT)Tr� )r� �
_feed_data� Exceptionr~ Z
set_exception)rD rA �excr! r! r" � feed_data s
zWebSocketReader.feed_datac
C s� �x�| � |�D �]x\}}}}|r8| js8tjtj d�| _|tjk�r"t|�dkr�t|d d� �d }|dk r�|t kr�t
t j d|� ���y|dd � �
d�}W n0 tk
r� } zt
t jd�|�W d d }~X Y nX ttj||�} n4|�rt
t j d|� d |� d |����nttjdd
�} | j�| d� q|tjk�rL| j�ttj|d
�t|�� q|tjk�rv| 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�rR| jd k �sFt �| 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�rly*|
�
d�} | j�ttj| d
�t| �� W n2 tk
�rh } zt
t jd�|�W d d }~X Y nX q| j�ttj|
d
�t|
�� qW dS )N)�wbitsr( r i� zInvalid close code: zutf-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 r2 r` �UNPACK_CLOSE_CODEr' r r r �decode�UnicodeDecodeErrorr r r~ r� r0 r1 r. r/ r� r- r� �extendr� r �formatr_ rg Z
decompressZunconsumed_tailrU �clear)
rD rA �fin�opcode�payloadZ
compressedZ
close_codeZ
close_messager� �msgZpayload_merged�leftr5 r! r! r" r� & s�
zWebSocketReader._feed_data)�bufrC c C s� g }| j r | j | d }| _ d}t|�}�xD| jtjk�rr|| dk�rp|||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�rttjd��| j �s"| j
dk�r4| �r,dnd| _
n| �rFttjd
��t |�| _ | | _ t |
�| _
|| _tj| _nP | jtjk�r^| j}|dk�r�|| dk�r�|||d � }|d7 }t|�d }|| _| j
�r�tjntj| _nP nz|dk�rB|| dk�r>|||d � }|d7 }t|�d }|| _| j
�r4tjntj| _nP n || _| j
�rVtjntj| _| jtjk�r�|| dk�r�|||d � | _|d7 }tj| _nP | jtjkr,| j}| j}|| }||k�r�|| | _|�||d� � |}n$d| _|�|||| � � || }| jdk�rj| j
�r@| jdk �s4t�t| j|� |�| j | j || j
f� t� | _tj| _q,P q,W ||d� | _ |S )z&Return the next frame from the socket.r� r r( � r
rm � rZ rl z*Received frame with non-zero reserved bitsz!Received fragmented control frame� �} z5Control frame payload cannot be larger than 125 bytesNTF�~ r+ ) r� r` 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^ )rD r� ZframesZ start_posZ
buf_lengthrA Z
first_byteZ second_byter� Zrsv1Zrsv2Zrsv3r� Zhas_mask�lengthr� Z chunk_lenr! r! r" r� � s�
zWebSocketReader.parse_frame)T)r r r r r r# r� rI r� rU 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 rC c C sB || _ || _|| _|j| _|| _|| _d| _|| _d| _d | _ d S )NFr )
r� r� r� � randrangerq rr �_closing�_limit� _output_size� _compressobj)rD r� r� r� r� r� rq rr r! r! r" rI > s zWebSocketWriter.__init__)rG r� rq rC 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 )z<Send a frame over the websocket with message as its payload.z!Cannot write to closing transportr r+ )�levelr� N����@ � r� i r� l �� rZ Zbig) r� r r2 �ConnectionResetErrorrq r� � compressobjZ Z_BEST_SPEEDr� �flushrr Z Z_FULL_FLUSHZ Z_SYNC_FLUSH�endswithrg r` r� � PACK_LEN1� PACK_LEN2� PACK_LEN3r� �to_bytesr^ r� �_writer� r= r� r� Z
_drain_helper) rD rG r� rq Zrsvr� Z
msg_lengthr� Zmask_bit�headerrY r! r! r" � _send_frameT sR
$
zWebSocketWriter._send_frame)rA rC c C s, | j d ks| j �� r td��| j �|� d S )Nz!Cannot write to closing transport)r� Z
is_closingr� �write)rD rA r! r! r" r� � s zWebSocketWriter._writer� )rG rC c � s, t |t�r|�d�}| �|tj�I dH dS )zSend pong message.zutf-8N)r] rK �encoder� r r1 )rD rG r! r! r" r8 � s
zWebSocketWriter.pongc � s, t |t�r|�d�}| �|tj�I dH dS )zSend ping message.zutf-8N)r] rK r� r� r r0 )rD rG r! r! r" r7 � s
zWebSocketWriter.ping)rG r6 rq rC c � sJ t |t�r|�d�}|r0| �|tj|�I dH n| �|tj|�I dH dS )z<Send a frame over the websocket with message as its payload.zutf-8N)r] rK r� r� r r/ r. )rD rG r6 rq r! r! r" �send� s
zWebSocketWriter.send�� )rF rG rC c � sD t |t�r|�d�}z"| jt|�| tjd�I dH W dd| _X dS )z<Close the websocket, sending the specified code and message.zutf-8)r� NT)r] rK r� r� �PACK_CLOSE_CODEr r2 r� )rD rF rG r! r! r" r9 � s
zWebSocketWriter.close)N)r� )r� )FN)r� r� )r r r r> r� ZRandomr �asyncioZ Transportr� r# r rI rU r r� r� r8 r7 r rK r� r9 r! r! r! r" r = s
? r )F)rl FF)QrN r� � collectionsrE r� �re�sysr� �enumr �structr �typingr r r r r r r
r r Z
base_protocolr Zhelpersr Zstreamsr Ztypedefsr �__all__r r' r# �__annotations__r r rU � unpack_fromr� r� �unpackr� �packr� r� r� r� r= r> �
namedtupler? r r4 r r3 r r� r rP � byteorderrQ rK rV rX r^ rd r� Z
_websocketre � ImportErrorrg �compilerh ri r� rs rx ry r r r! r! r! r" �<module> sp ,
"
4 @