Your IP : 216.73.216.213
B
�k�akC � @ s\ U d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z m
Z
d dl Z d dl Z d dl
mZ ddlmZ ddlmZ ddlmZmZ ddlmZmZmZmZmZm Z m Z m Z m Z m Z m!Z! dd l"m#Z# dd
l$m%Z%m&Z& dd l'm(Z(m)Z)m*Z* dd l+m,Z,m-Z- dd
l.m/Z/ ddl0m1Z1 dZ2dZ3e(e4 e5d< e j6dddd�G dd� d��Z7G dd� de1�Z8dS )� N)�Any�Iterable�Optional�Tuple�cast)� CIMultiDict� )�hdrs)�AbstractStreamWriter)�
call_later�
set_result) �WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketError�WebSocketReader�WebSocketWriter� WSCloseCode� WSMessage� WSMsgType�
ws_ext_gen� ws_ext_parse)� ws_logger)� EofStream�FlowControlDataQueue)�Final� JSONDecoder� JSONEncoder)�HTTPBadRequest�
HTTPException)� BaseRequest)�StreamResponse)�WebSocketResponse�WebSocketReadyr � �THRESHOLD_CONNLOST_ACCESST)Z auto_attribs�frozen�slotsc @ s0 e Zd ZU eed< ee ed< ed�dd�ZdS )r# �ok�protocol)�returnc C s | j S )N)r( )�self� r, �A/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/web_ws.py�__bool__1 s zWebSocketReady.__bool__N)�__name__�
__module__� __qualname__�bool�__annotations__r �strr. r, r, r, r- r# , s
r# c s� e Zd ZdZddddddddd�eee eeee ee ee dd� � fd d
�Z
dd �d d
�Z dd �dd�Z dd �dd�Z
dd �dd�Zeed�� fdd� Zeedeeef d�dd�Zeeeef d�dd�Zeeedd �d d �Zeed�d d �Zeed �d!d"��Zeee d �d#d$��Zeee d �d%d&��Z eed �d'd(��Z ee d �d)d*�Z dVe dd,�d-d.�Z!dWe dd,�d/d0�Z"dXeee dd1�d2d3�Z#dYe ee dd1�d4d5�Z$dZe%j&d6�e'ee e(dd7�d8d9�Z)dd �d:d;�Z*e+j,d+d<�e e ed=�d>d?�Z-d[ee e.d@�dAdB�Z/ddC�ee ed@�dDdE�Z0ddC�ee e d@�dFdG�Z1e%j2ddH�e3ee e'dI�dJdK�Z4e ddL�dMdN�Z5d d �dOdP�Z6e.d �dQdR�Z7e ddS�dTdU�Z8� Z9S )\r" Fg $@NTr, i @ )�timeout�receive_timeout� autoclose�autoping� heartbeat� protocols�compress� max_msg_size) r5 r6 r7 r8 r9 r: r; r< r* c s� t � jdd� || _d | _d | _d | _d| _d| _d| _d | _ d | _
d | _ d | _ || _
|| _|| _|| _|| _d | _|d k r�|d | _d | _|| _|| _d S )N�e )�statusFr g @)�super�__init__�
_protocols� _ws_protocol�_writer�_reader�_closed�_closing�
_conn_lost� _close_code�_loop�_waiting�
_exception�_timeout�_receive_timeout�
_autoclose� _autoping�
_heartbeat�
_heartbeat_cb�_pong_heartbeat�_pong_response_cb� _compress�
_max_msg_size) r+ r5 r6 r7 r8 r9 r: r; r< )� __class__r, r- r@ 9 s.
zWebSocketResponse.__init__)r* c C s8 | j d k r| j �� d | _ | jd k r4| j�� d | _d S )N)rS �cancelrQ )r+ r, r, r- �_cancel_heartbeat] s
z#WebSocketResponse._cancel_heartbeatc C s8 | � � | jd k r4| jd k s t�t| j| j| j�| _d S )N)rX rP rI �AssertionErrorr �_send_heartbeatrQ )r+ r, r, r- �_reset_heartbeatf s
z"WebSocketResponse._reset_heartbeatc C s\ | j d k rX| jsX| jd k s t�| j�| j�� � | jd k rD| j�� t | j
| j | j�| _d S )N) rP rE rI rY Z create_taskrC �pingrS rW r �_pong_not_receivedrR )r+ r, r, r- rZ o s
z!WebSocketResponse._send_heartbeatc C s> | j d k r:| j jd k r:d| _tj| _t�� | _| j j� � d S )NT)
Z_req� transportrE r �ABNORMAL_CLOSURErH �asyncio� TimeoutErrorrK �close)r+ r, r, r- r] } s
z$WebSocketResponse._pong_not_received)�requestr* c � s\ | j d k r| j S | �|�\}}t� �|�I d H }|d k s<t�| �|||� |�� I d H |S )N)�_payload_writer�
_pre_startr? �preparerY � _post_start�drain)r+ rc r) �writerZpayload_writer)rV r, r- rf � s
zWebSocketResponse.preparezCIMultiDict[str]c C s� |j }d|�tjd��� �� kr8td�|�tj��d��d|�tjd��� krftd�|�tj��d��d }tj |kr�dd� |tj �
d �D �}x,|D ]}|| j kr�|}P q�W t �
d
|| j � |�tjd�}|d kr�td |� �d��|�tj�}y.|�r
tt�|��d
k�rtd|��d��W n* tjk
�rF td|��d�d �Y nX t�t�|�� t ��� ��� }t tjdtjdtj |i�} d}
d} | j �r�|�tj �} t | dd�\} }
| �r�t!| d|
d�}
|
| tj < |�r�|| tj < | || |
fS )NZ websocket� z@No WebSocket UPGRADE hdr: {}
Can "Upgrade" only to "WebSocket".)�textZupgradez No CONNECTION upgrade hdr: {}c S s g | ]}t |�� ��qS r, )r4 �strip)�.0�protor, r, r- �
<listcomp>� s z0WebSocketResponse._handshake.<locals>.<listcomp>�,u8 Client protocols %r don’t overlap server-known ones %r)Z13�8�7zUnsupported version: � zHandshake error: Fr T)�isserver)r; rt Zserver_notakeover)"�headers�getr ZUPGRADE�lowerrl r �formatZ
CONNECTIONZSEC_WEBSOCKET_PROTOCOL�splitrA r �warningZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEY�len�base64� b64decode�binascii�Error� b64encode�hashlib�sha1�encoder �digest�decoder ZSEC_WEBSOCKET_ACCEPTrT ZSEC_WEBSOCKET_EXTENSIONSr r )r+ rc ru r) Z
req_protocolsrn �version�keyZ
accept_valZresponse_headers�
notakeoverr; �
extensionsZ
enabledextr, r, r- �
_handshake� sf
z WebSocketResponse._handshakec C sl |j | _ | �|�\}}}}| �d� | j�|� | �� || _|jj}|d k sRt �t
|j|||d�}||fS )Nr= )r; r� ) rI r� Z
set_statusru �updateZ force_closerT � _protocolr^ rY r )r+ rc ru r) r; r� r^ ri r, r, r- re � s
z WebSocketResponse._pre_start)rc r) ri r* c C sd || _ || _| �� | j}|d k s&t�t|jd|d�| _|j� t
| j| j | j d�� |j�
d� d S )Ni )�loop)r; F)rB rC r[ rI rY r r� rD r) Z
set_parserr rU rT Z
keep_alive)r+ rc r) ri r� r, r, r- rg � s z WebSocketResponse._post_startc C sR | j d k rtd��y| �|�\}}}}W n tk
rB tdd �S X td|�S d S )NzAlready startedFT)rC � RuntimeErrorr� r r# )r+ rc �_r) r, r, r- � can_prepare s
z WebSocketResponse.can_preparec C s | j S )N)rE )r+ r, r, r- �closed s zWebSocketResponse.closedc C s | j S )N)rH )r+ r, r, r- �
close_code s z WebSocketResponse.close_codec C s | j S )N)rB )r+ r, r, r- � ws_protocol s z WebSocketResponse.ws_protocolc C s | j S )N)rT )r+ r, r, r- r; s zWebSocketResponse.compressc C s | j S )N)rK )r+ r, r, r- � exception! s zWebSocketResponse.exception� )�messager* c � s( | j d krtd��| j �|�I d H d S )NzCall .prepare() first)rC r� r\ )r+ r� r, r, r- r\ $ s
zWebSocketResponse.pingc � s( | j d krtd��| j �|�I d H d S )NzCall .prepare() first)rC r� �pong)r+ r� r, r, r- r� ) s
zWebSocketResponse.pong)�datar; r* c � sH | j d krtd��t|t�s,tdt|� ��| j j|d|d�I d H d S )NzCall .prepare() firstz data argument must be str (%r)F)�binaryr; )rC r� �
isinstancer4 � TypeError�type�send)r+ r� r; r, r, r- �send_str/ s
zWebSocketResponse.send_strc � sN | j d krtd��t|tttf�s2tdt|� ��| j j|d|d�I d H d S )NzCall .prepare() firstz#data argument must be byte-ish (%r)T)r� r; ) rC r� r� �bytes� bytearray�
memoryviewr� r� r� )r+ r� r; r, r, r- �
send_bytes6 s
z WebSocketResponse.send_bytes)�dumps)r� r; r� r* c � s | j ||�|d�I d H d S )N)r; )r� )r+ r� r; r� r, r, r- � send_json= s zWebSocketResponse.send_jsonc � s4 | j r
d S | jd kr td��| �� I d H d| _ d S )Nz Response has not been startedT)Z _eof_sentrd r� rb )r+ r, r, r- � write_eofF s
zWebSocketResponse.write_eof)�coder� )r� r� r* c
� s� | j d krtd��| �� | j}|d k s,t�| jd k rT| jsT|�td� | jI d H | j�s�d| _y8| j � ||�I d H | j
}|d k s�t�|� � I d H W nT t j
t jfk
r� tj| _� Y n0 tk
r� } ztj| _|| _dS d }~X Y nX | jr�dS | j}|d k �st�y8t�| j�4 I d H � |�� I d H }W d Q I d H R X W nR t j
k
�rh tj| _� Y n2 tk
�r� } ztj| _|| _dS d }~X Y nX |jtjk�r�|j | _dS tj| _t �� | _dS dS d S )NzCall .prepare() firstr TF) rC r� rX rD rY rJ rE Z feed_datar rb rd rh r` �CancelledErrorra r r_ rH � ExceptionrK rF �
async_timeoutr5 rL �readr� r �CLOSEr� )r+ r� r� �readerri �exc�msgr, r, r- rb O sV
"
zWebSocketResponse.close)r5 r* c
� sn | j d krtd��| j}|d k s$t��xB| jd k r:td��| jrd| jd7 _| jtkr`td��tS | j rnt
S yn|� � | _zFt �
|p�| j�4 I d H � | j �� I d H }W d Q I d H R X | �� W d | j}t|d� d | _X W n� tjtjfk
�r tj| _� Y n� tk
�r8 tj| _| �� I d H tt j d d �S t k
�r� } z*|j | _| j|j d�I d H tt j |d �S d }~X Y nP t!k
�r� } z0|| _"d| _ tj| _| �� I d H tt j |d �S d }~X Y nX |j#t j$k�r d| _ |j%| _| j�sf| j&�rf| �� I d H nZ|j#t j'k�r"d| _ nD|j#t j(k�rN| j)�rN| �*|j%�I d H q(n|j#t j+k�rf| j)�rfq(|S d S )NzCall .prepare() firstz+Concurrent call to receive() is not allowedr z WebSocket connection is closed.T)r� ),rD r� rI rY rJ rE rG r% r
rF r Z
create_futurer� r5 rM r� r[ r r` r� ra r r_ rH r �OKrb r r �CLOSEDr r� �ERRORr� rK r� r� r� rN �CLOSINGZPINGrO r� ZPONG)r+ r5 r� r� �waiterr� r, r, r- �receive� sj
zWebSocketResponse.receive)r5 c � s< | � |�I d H }|jtjkr0td�|j|j���tt|j�S )Nz.Received message {}:{!r} is not WSMsgType.TEXT) r� r� r ZTEXTr� rx r� r r4 )r+ r5 r� r, r, r- � receive_str� s z WebSocketResponse.receive_strc � sB | � |�I d H }|jtjkr6td|j� d|j�d���tt|j�S )NzReceived message �:z
is not bytes)r� r� r ZBINARYr� r� r r� )r+ r5 r� r, r, r- �
receive_bytes� s z WebSocketResponse.receive_bytes)�loadsr5 )r� r5 r* c � s | j |d�I d H }||�S )N)r5 )r� )r+ r� r5 r� r, r, r- � receive_json� s z WebSocketResponse.receive_json)r� r* c � s t d��d S )Nz"Cannot call .write() for websocket)r� )r+ r� r, r, r- �write� s zWebSocketResponse.writec C s | S )Nr, )r+ r, r, r- � __aiter__� s zWebSocketResponse.__aiter__c � s, | � � I d H }|jtjtjtjfkr(t�|S )N)r� r� r r� r� r� �StopAsyncIteration)r+ r� r, r, r- � __anext__� s zWebSocketResponse.__anext__)r� r* c C s | j d k r| j �|� d S )N)rD �
set_exception)r+ r� r, r, r- �_cancel� s
zWebSocketResponse._cancel)r� )r� )N)N)N)N):r/ r0 r1 Z
_length_check�floatr r2 r r4 �intr@ rX r[ rZ r] r r
rf r r� r re rg r# r� �propertyr� r� r� r; �
BaseExceptionr� r� r\ r� r� r� �jsonr� r r r� r� r r� rb r r� r� r� r� r r� r� r� r� r� �
__classcell__r, r, )rV r- r" 5 s` "
S
7=
r" )9r` r| r~ r� r� �typingr r r r r r� �attrZ multidictr rj r �abcr
Zhelpersr r �httpr
r r r r r r r r r r �logr Zstreamsr r Ztypedefsr r r Zweb_exceptionsr r Z web_requestr Z web_responser! �__all__r% r� r3 �sr# r" r, r, r, r- �<module> s. 4