Your IP : 216.73.216.213
���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 �A/opt/alt/python35/lib64/python3.5/site-packages/aiohttp/web_ws.py�__bool__ s zWebSocketReady.__bool__N)
�__name__�
__module__� __qualname__�attr�ib�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/ r0 )� __class__r r r4 & s,
zWebSocketResponse.__init__c C sN | j d k r% | j j � d | _ | j d k rJ | j j � d | _ d S)N)rG �cancelrE )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)rK rD r �_send_heartbeatr= rE )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)
rD r9 r7 �pingrG rJ r �_pong_not_receivedrF r= )r r r r rL 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� transportr9 r< �asyncio� TimeoutErrorr? �close)r r r r rO 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_startr3 �prepare� _post_start�drain)r �requestr) �writerZpayload_writer)rI r r rV 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".�upgradez 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}�chunkedFZisserverTr0 Zserver_notakeover)rc rd re )&�headers�methodr ZMETH_GETr �getZUPGRADE�lowerr^ r �formatZ
CONNECTIONZSEC_WEBSOCKET_PROTOCOL�splitr5 r �warningZSEC_WEBSOCKET_VERSIONZSEC_WEBSOCKET_KEY�len�base64� b64decode�binascii�Error� b64encode�hashlib�sha1�encoder �digest�decoder ZTRANSFER_ENCODINGZSEC_WEBSOCKET_ACCEPTrH ZSEC_WEBSOCKET_EXTENSIONSr r )r rY rh r) Z
req_protocolsr` �version�keyZ
accept_valZresponse_headers�
notakeoverr0 �
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)Nr2 r0 r| ) �loopr= r~ rM Z
set_statusrh �updateZ force_closerH r � _protocolrP )r rY rh r) r0 r| rZ r r r rU � 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� rf r r0 Fi ) r6 r7 r r� r= r8 r) Z
set_parserr rH Z
keep_alive)r rY r) rZ r r r rW � 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)r7 � RuntimeErrorr~ r r )r rY �_r) r r r � can_prepare� s
z WebSocketResponse.can_preparec C s | j S)N)r9 )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)r6 )r r r r � ws_protocol� s z WebSocketResponse.ws_protocolc C s | j S)N)rH )r r r r r0 � 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)r7 r� rN )r �messager r r rN � s zWebSocketResponse.pingc � s4 | j d k r t d � � | j j | � Id Hd S)NzCall .prepare() first)r7 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)�binaryFr0 )r7 r� �
isinstancer( � TypeErrorr �send)r �datar0 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� Tr0 ) r7 r� r� �bytes� bytearray�
memoryviewr� r r� )r r� r0 r r r �
send_bytes� s z WebSocketResponse.send_bytes�dumpsc � s" | j | | � d | �Id Hd S)Nr0 )r� )r r� r0 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_sentrT r� rS )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� r F)r7 r� rK r> r9 r8 Z feed_datar rS rT rX rQ �CancelledErrorrR r< � Exceptionr? r: �
async_timeoutr* r@ r= �readr r �CLOSEr� )r r� r� �exc�msgr r r rS 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.r Ti� r� )&r8 r� r> r9 r; �THRESHOLD_CONNLOST_ACCESSr r: r r= Z
create_futurer� r* rA r� rM r rQ r� rR r< r
r� rS r
r �ERRORr� r? r r� r� rB �CLOSINGZPINGrC 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 �TEXTr� rl 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� rl 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$ r4 rK rM rL rO rV r~ rU rW r� �propertyr� r� r� r0 r� rN r� r� r� �jsonr� r� r� rS r� r� r� r� r� r� r� r� r r )rI r r $ s@ F
19 )r r r )(rQ rp rr ru r� r� r% Z multidictr r[ r �helpersr r �httpr 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 +