���ZZ9 � @ sm d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m
Z
d d l m Z m
Z
d d l m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z d d l m Z d d l m Z d d l m Z m Z m Z d d
l! m" Z" d Z# d Z$ e j% d d d d � Gd d � d � � Z& Gd d � d e" � Z' d S)� N)� CIMultiDict� )�hdrs)�
call_later�
set_result)
�WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketError�WebSocketReader�WebSocketWriter� WSMessage� WSMsgType�
ws_ext_gen� ws_ext_parse)� ws_logger)�FlowControlDataQueue)�HTTPBadRequest�
HTTPException�HTTPMethodNotAllowed)�StreamResponse�WebSocketResponse�WebSocketReadyr � �frozenT�slotsc @ s@ e Z d Z e j d e � Z e j d e � Z d d � Z d S)r �typec C s | j S)N)�ok)�self� r �
/web_ws.py�__bool__ s zWebSocketReady.__bool__N)
�__name__�
__module__� __qualname__�attrZib�boolr �str�protocolr! r r r r r s c s� e Z d Z d d d d d d d d d d d f d
d � f d d � Z d
d � Z d d � Z d d � Z d d � Z � f d d � Z d d � Z d d � Z
d d � Z d d � Z e
d d � � Z e
d! d" � � Z e
d# d$ � � Z e
d% d& � � Z d' d( � Z d) d* d+ � Z d) d, d- � Z d d. d/ � Z d d0 d1 � Z d d2 e j d3 d4 �Z d5 d6 � Z d7 d8 d9 d: d; d< � Z d d= d> � Z d d d? d@ � Z d d dA dB � Z dC e j d d dD dE � Z dF dG � Z! dH dI � Z" dJ dK � Z# � S)Lr �timeoutg $@�receive_timeoutN� autocloseT�autoping� heartbeat� protocols�compressc s� t � j d d � | | _ d | _ d | _ d | _ d | _ d | _ d | _ d | _ d | _
d | _ d | _ | | _
| | _ | | _ | | _ | | _ d | _ | d k r� | d | _ d | _ | | _ d S)N�status�e Fr 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)r r) r* r+ r, r- r. r/ )� __class__r r r3 & s,
zWebSocketResponse.__init__c C sN | j d k r% | j j � d | _ | j d k rJ | j j � d | _ d S)N)rF �cancelrD )r r r r �_cancel_heartbeatA s
z#WebSocketResponse._cancel_heartbeatc C s; | j � | j d k r7 t | j | j | j � | _ d S)N)rJ rC r �_send_heartbeatr< rD )r r r r �_reset_heartbeatJ s
z"WebSocketResponse._reset_heartbeatc C sd | j d k r` | j r` | j j � | j d k rB | j j � t | j | j | j � | _ d S)N)
rC r8 r6 �pingrF rI r �_pong_not_receivedrE r< )r r r r rK Q s
z!WebSocketResponse._send_heartbeatc C sV | j d k rR | j j d k rR d | _ d | _ t j � | _ | j j j � d S)NTi� )Z_req� transportr8 r; �asyncio� TimeoutErrorr> �close)r r r r rN Z s
! z$WebSocketResponse._pong_not_receivedc � sh | j d k r | j S| j | � \ } } t � j | � Id H} | j | | | � | j � Id H| S)N)�_payload_writer�
_pre_startr2 �prepare� _post_start�drain)r �requestr( �writerZpayload_writer)rH r r rU a s zWebSocketResponse.preparec C s� | j } | j t j k r3 t | j t j g � � d | j t j d � j � j � k r~ t d d j
| j t j � � � � d | j t j d � j � k r� t d d j
| j t j � � � � d } t j | k r8d d � | t j j
d � D� } x: | D] } | | j k r| } PqWt j d
| | j � | j t j d � } | d k rqt d d j
| � � � | j t j � } y> | s�t t j | � � d k r�t d d j
| � � � Wn0 t j k
r�t d d j
| � � d � Yn Xt j t j | j � t � j � � j � } t t j d t j d t j d t j! | i � } d }
| j" } | r�| j t j# � } t$ | d d �\ } }
| r�t% d | d d d |
� }
|
| t j# <| r�| | t j <| | | |
f S)NZ websocket� �textz@No WebSocket UPGRADE hdr: {}
Can "Upgrade" only to "WebSocket".Zupgradez No CONNECTION upgrade hdr: {}c S s" g | ] } t | j � � � q Sr )r' �strip)�.0�protor r r �
<listcomp>~ s z0WebSocketResponse._handshake.<locals>.<listcomp>�,u8 Client protocols %r don’t overlap server-known ones %r�13�8�7zUnsupported version: {}� zHandshake error: {!r}ZchunkedFZisserverTr/ Zserver_notakeover)ra rb rc )&�headers�methodr ZMETH_GETr �getZUPGRADE�lowerr\ r �formatZ
CONNECTIONZSEC_WEBSOCKET_PROTOCOL�splitr4 r �warningZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEY�len�base64Z b64decode�binascii�ErrorZ b64encode�hashlib�sha1�encoder �digest�decoder ZTRANSFER_ENCODINGZSEC_WEBSOCKET_ACCEPTrG ZSEC_WEBSOCKET_EXTENSIONSr r )r rX re r( Z
req_protocolsr^ �version�keyZ
accept_valZresponse_headers�
notakeoverr/ �
extensionsZ
enabledextr r r �
_handshakel sh '!
"(
z WebSocketResponse._handshakec C s� | j | _ | j | � \ } } } } | j � | j d � | j j | � | j � | | _ t | j
| j
j d | d | �} | | f S)Nr1 r/ rw ) �loopr< ry rL Z
set_statusre �updateZ force_closerG r � _protocolrO )r rX re r( r/ rw rY r r r rT � s
z WebSocketResponse._pre_startc C si | | _ | | _ t | j d d d | j �| _ | j j t | j d | j �� | j j
d � d S)N�limit� rd rz r/ Fi ) r5 r6 r r| r< r7 r( Z
set_parserr rG Z
keep_alive)r rX r( rY r r r rV � s z WebSocketResponse._post_startc C sm | j d k r t d � � y | j | � \ } } } } Wn t k
r[ t d d � SYn Xt d | � Sd S)NzAlready startedFT)r6 � RuntimeErrorry r r )r rX �_r( r r r � can_prepare� s
z WebSocketResponse.can_preparec C s | j S)N)r8 )r r r r �closed� s zWebSocketResponse.closedc C s | j S)N)r; )r r r r �
close_code� s z WebSocketResponse.close_codec C s | j S)N)r5 )r r r r � ws_protocol� s z WebSocketResponse.ws_protocolc C s | j S)N)rG )r r r r r/ � s zWebSocketResponse.compressc C s | j S)N)r> )r r r r � exception� s zWebSocketResponse.exception�bc � s4 | j d k r t d � � | j j | � Id Hd S)NzCall .prepare() first)r6 r rM )r �messager r r rM � s zWebSocketResponse.pingc � s4 | j d k r t d � � | j j | � Id Hd S)NzCall .prepare() first)r6 r �pong)r r� r r r r� � s zWebSocketResponse.pongc � se | j d k r t d � � t | t � s@ t d t | � � � | j j | d d d | �Id Hd S)NzCall .prepare() firstz data argument must be str (%r)�binaryFr/ )r6 r �
isinstancer' � TypeErrorr �send)r �datar/ r r r �send_str� s
zWebSocketResponse.send_strc � sn | j d k r t d � � t | t t t f � sI t d t | � � � | j j | d d d | �Id Hd S)NzCall .prepare() firstz#data argument must be byte-ish (%r)r� Tr/ ) r6 r r� �bytes� bytearray�
memoryviewr� r r� )r r� r/ r r r �
send_bytes� s z WebSocketResponse.send_bytes�dumpsc � s" | j | | � d | �Id Hd S)Nr/ )r� )r r� r/ r� r r r � send_json s zWebSocketResponse.send_jsonc � sD | j r
d S| j d k r( t d � � | j � Id Hd | _ d S)Nz Response has not been startedT)Z _eof_sentrS r rR )r r r r � write_eof s zWebSocketResponse.write_eof�codei� r� � c � s� | j d k r t d � � | j � | j d k r] | j r] | j j t d � | j Id H| j s�d | _ y) | j j | | � | j j
� Id HWnd t j t j
f k
r� d | _ � Yn; t k
r� } z d | _ | | _ d SWYd d } ~ Xn X| j r d Sy; t j | j d | j �� | j j � Id H} Wd QRXWn[ t j k
rjd | _ � Yn; t k
r�} z d | _ | | _ d SWYd d } ~ Xn X| j t j k r�| j | _ d Sd | _ t j
� | _ d Sd Sd S)NzCall .prepare() firstr Ti� rz F)r6 r rJ r= r8 r7 Z feed_datar rR rS rW rP �CancelledErrorrQ r; � Exceptionr> r9 �
async_timeoutr) r? r< �readr r �CLOSEr� )r r� r� �exc�msgr r r rR sJ
zWebSocketResponse.closec � s� | j d k r t d � � x�| j d k r9 t d � � | j rp | j d 7_ | j t k rl t d � � t S| j r} t Sy� | j j
� | _ zK t j | p� | j
d | j �� | j j � Id H} Wd QRX| j � Wd | j } t | d � d | _ XWn� t j t j f k
r-d | _ � Yn� t k
r�} z< | j | _ | j d | j � Id Ht t j | d � SWYd d } ~ Xnb t k
r�} zB | | _ d | _ d | _ | j � Id Ht t j | d � SWYd d } ~ Xn X| j t j k r6d | _ | j | _ | j r�| j r�| j � Id Hnr | j t j! k rTd | _ nT | j t j" k r�| j# r�| j$ | j � Id Hq n | j t j% k r�| j# r�q | SWd S) NzCall .prepare() firstz+Concurrent call to receive() is not allowedr z WebSocket connection is closed.rz Ti� r� )&r7 r r= r8 r: �THRESHOLD_CONNLOST_ACCESSr r9 r r< Z
create_futurer� r) r@ r� rL r rP r� rQ r; r
r� rR r
r �ERRORr� r> r r� r� rA �CLOSINGZPINGrB r� ZPONG)r r) r� �waiterr� r r r �receiveC sb
% & zWebSocketResponse.receivec � sK | j | � Id H} | j t j k rD t d j | j | j � � � | j S)Nz.Received message {}:{!r} is not WSMsgType.TEXT)r� r r ZTEXTr� ri r� )r r) r� r r r � receive_str| s z WebSocketResponse.receive_strc � sK | j | � Id H} | j t j k rD t d j | j | j � � � | j S)Nz%Received message {}:{!r} is not bytes)r� r r ZBINARYr� ri r� )r r) r� r r r �
receive_bytes� s z WebSocketResponse.receive_bytes�loadsc � s! | j d | � Id H} | | � S)Nr) )r� )r r� r) r� r r r � receive_json� s z WebSocketResponse.receive_jsonc � 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< | j � Id H} | j t j t j t j f k r8 t � | S)N)r� r r r� r� ZCLOSED�StopAsyncIteration)r r� r r r � __anext__� s zWebSocketResponse.__anext__)$r" r# r$ r3 rJ rL rK rN rU ry rT rV r� �propertyr� r� r� r/ r� rM r� r� r� �jsonr� r� r� rR r� r� r� r� r� r� r� r� r r )rH r r $ s@ F
19 )r r r )(rP rm rn rp r� r� r% Z multidictr rZ r Zhelpersr r Zhttpr r r r
r r r
r r r �logr Zstreamsr Zweb_exceptionsr r r Z web_responser �__all__r� �sr r r r r r �<module> s$ F +