���Z�U �
@ s
d 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 d d d
d d d
d d d f Z Gd d � d e � Z d d � e D� Z Gd d � d e � Z d 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 e
d � j Z d d Z d d Z e j! d d d d! g � Z" Gd" d
� d
e" � Z# e# e j$ d d � Z% e# e j& d d � Z' Gd# d � d e( � Z) Gd$ d% � d% e( � Z* e j+ Z, d& d' � e- d( � D� Z. d) d* � Z/ e r;e/ Z0 n5 y d d+ l1 m2 Z2 e2 Z0 Wn e3 k
roe/ Z0 Yn Xe4 d d d, d, g � Z5 e j6 d- � Z7 e j6 d. � Z8 d/ d0 d1 � Z9 d2 d/ d/ d3 d4 � Z: Gd5 d6 � d6 e � Z; Gd7 d � d � Z< Gd8 d � d � Z= d S)9z%WebSocket protocol versions 13 and 8.� N)�IntEnum)�Struct� )�
NO_EXTENSIONS�noop)� ws_logger�WS_CLOSED_MESSAGE�WS_CLOSING_MESSAGE�WS_KEY�WebSocketReader�WebSocketWriter� WSMessage�WebSocketError� WSMsgType� WSCloseCodec @ sR e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z d
Z d Z
d S)
r i� i� i� i� i� i� i� i� i� i� i� N)�__name__�
__module__� __qualname__ZOKZ
GOING_AWAY�PROTOCOL_ERRORZUNSUPPORTED_DATA� INVALID_TEXTZPOLICY_VIOLATIONZMESSAGE_TOO_BIGZMANDATORY_EXTENSIONZINTERNAL_ERRORZSERVICE_RESTARTZTRY_AGAIN_LATER� r r �/http_websocket.pyr s c C s h | ] } t | � � q Sr )�int)�.0�ir r r � <setcomp># s r c @ sv e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z e Z e Z
e Z e Z e Z e Z e
Z e Z d
S) r r r � � �
� � i i N)r r r � CONTINUATION�TEXT�BINARY�PING�PONG�CLOSE�CLOSING�CLOSED�ERROR�text�binary�ping�pong�close�closing�closed�errorr r r r r & s" s$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11z!Hz!Qz!BBz!BBHz!BBQr � � �_WSMessageBase�type�data�extrac @ s% e Z d Z d e j d d � Z d S)r
�loadsc C s
| | j � S)zAReturn parsed JSON data.
.. versionadded:: 0.22
)r6 )�selfr8 r r r �jsonR s zWSMessage.jsonN)r r r r: r8 r r r r r
P s c s( e Z d Z d Z � f d d � Z � S)r z WebSocket protocol parser error.c s | | _ t � j | � d S)N)�code�super�__init__)r9 r; �message)� __class__r r r= a s zWebSocketError.__init__)r r r �__doc__r= r r )r? r r ^ s c @ s e Z d Z d Z d S)�WSHandshakeErrorz#WebSocket protocol handshake error.N)r r r r@ r r r r rA f s rA c s5 g | ]+ � t � f d d � t d � D� � � q S)c 3 s | ] } | � AVq d S)Nr )r �a)�br r � <genexpr>n s z<listcomp>.<genexpr>r )�bytes�range)r r )rC r �
<listcomp>n s rG r c C s t | t � s t | � � t | � d k s9 t | � � | rd d � | D� \ } } } } | d d d � j | � | d d d � <| d d d � j | � | d d d � <| d d d � j | � | d d d � <| d d d � j | � | d d d � <d S)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 | Vq d S)N)�
_XOR_TABLE)r �nr r r rD � s z)_websocket_mask_python.<locals>.<genexpr>Nr r � )�
isinstance� bytearray�AssertionError�len� translate)�maskr6 rB rC �c�dr r r �_websocket_mask_pythonq s
)))rT )�_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([^,]+)?Fc C sU | s
d Sd } d } x2t j | � D]!} | j d � } | sH d } Pt j | � } | r(d } | r� | j d � r� t | j d � � } | d k s� | d k r� d } q& | j d � r� d } PqG| j d � rt | j d � � } | d k s| d k rt d � � | j d
� r$d } Pq& | s& t d | j d � � � q& W| | f S)
Nr Fr � rH r T� zInvalid window sizer z#Extension for deflate not supported)r F)�_WS_EXT_RE_SPLIT�finditer�group�
_WS_EXT_RE�matchr rA )Zextstr�isserver�compress�
notakeover�extZdefextr] r r r � ws_ext_parse� s@ rb rW c C s� | d k s | d k r$ t d � � d g } | s@ | j d � | d k rc | j d t | � � | rv | j d � d j | � S) Nr rW 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�append�str�join)r_ r^ Zserver_notakeoverZ
enabledextr r r �
ws_ext_gen� s
rg c @ s( e Z d Z d Z d Z d Z d Z d S)�
WSParserStater r rK rH N)r r r � READ_HEADER�READ_PAYLOAD_LENGTH�READ_PAYLOAD_MASK� READ_PAYLOADr r r r rh � s rh c @ sO e Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
r Tc C s� | | _ d | _ g | _ t j | _ d | _ d | _ d | _ t � | _
d | _ d | _ d | _
d | _ d | _ d | _ d | _ | | _ d S)NF� r )�queue�_exc�_partialrh ri �_state�_opcode�
_frame_fin�
_frame_opcoderM �_frame_payload�_tail� _has_mask� _frame_mask�_payload_length�_payload_length_flag� _compressed�_decompressobj� _compress)r9 rn r_ r r r r= � s zWebSocketReader.__init__c C s | j j � d S)N)rn �feed_eof)r9 r r r r~ s zWebSocketReader.feed_eofc C sm | j r d | f Sy | j | � SWnB t k
rh } z" | | _ | j j | � d SWYd d } ~ Xn Xd S)NTrm )Trm )ro �
_feed_data� Exceptionrn �
set_exception)r9 r6 �excr r r � feed_data s
zWebSocketReader.feed_datac C s� x�| j | � D]�\ } } } } | rK | j rK t j d t j � | _ | t j k r�t | � d k r+t | d d � � d } | d k r� | t k r� t
t j d j
| � � � y | d d � j d � } Wn: t k
r} z t
t j d � | � WYd d } ~ Xn Xt t j | | � } n? | rUt
t j d j
| | | � � � n t t j d d � } | j j | d � q | t j k r�| j j t t j | d � t | � � q | t j k r�| j j t t j | d � t | � � q | t j t j f k r9| j d k r9t
t j d
j
| � � � q | sj| t j k rW| | _ | j j | � q | j r�| t j k r�t
t j d j
| � � � | t j k r�| j } d | _ | j j | � d j | j � }
| r�| j j |
t � }
| j j � | t j k r�y; |
j d � } | j j t t j | d � t | � � Wq�t k
r�} z t
t j d � | � WYd d } ~ Xq�Xq | j j t t j |
d � t |
� � q Wd S)N�wbitsr r i� zInvalid close code: {}zutf-8zInvalid UTF-8 text messagez Invalid close frame: {} {} {!r}� zUnexpected opcode={!r}z<The opcode in non-fin frame is expected to be zero, got {!r}rm F)Frm ) � parse_framer| �zlib�
decompressobj� MAX_WBITSr r& rO �UNPACK_CLOSE_CODE�ALLOWED_CLOSE_CODESr r r �format�decode�UnicodeDecodeErrorr r
rn r� r$ r% r" r# rr r! rp rd rf �
decompress�_WS_DEFLATE_TRAILING�clear) r9 r6 �fin�opcode�payload�
compressedZ
close_codeZ
close_messager� �msgZpayload_mergedr* r r r r s� " " "$
#" zWebSocketReader._feed_datac C sz g } | j r& | j | d } | _ d } t | � } x(| j t j k r�| | d k r�| | | d � } | d 7} | \ } } | d ?d @} | d ?d @} | d ?d @}
| d ?d @} | d @} |
s� | s� | r� | j r� t t j d
� � | d k r!| d k r!t t j d � � | d ?d @}
| d @} | d k rc| d
k rct t j d � � | j s{| j
d k r�| r�d n d | _
n | r�t t j d
� � | | _ | | _ |
| _ | | _
t j | _ n P| j t j k r| j
} | d k rr| | d k rn| | | d � } | d 7} t | � d } | | _ | j r_t j n t j | _ qPn� | d k r�| | d k r�| | | d � } | d 7} t | � d } | | _ | j r�t j n t j | _ qPn' | | _ | j rt j n t j | _ | j t j k rd| | d k rc| | | d � | _ | d 7} t j | _ n P| j t j k r; | j } | j } | | } | | k r�| | | _ | j | | d � � | } n. d | _ | j | | | | � � | | } | j d k r^| j r!t | j | � | j | j | j | | j
f � t � | _ t j | _ q; Pq; W| | d � | _ | S)z&Return the next frame from the socket.rm r r � r rX � rH rW z*Received frame with non-zero reserved bitsz!Received fragmented control frame� �} z5Control frame payload cannot be larger than 125 bytesNTF�~ r )rv rO rq rh ri r} r r r rs r{ rt rw rz rj � UNPACK_LEN2ry rk rl � UNPACK_LEN3rx ru �extend�_websocket_maskrd rM )r9 �bufZframesZ start_posZ
buf_lengthr6 Z
first_byteZ second_byter� Zrsv1Zrsv2Zrsv3r� Zhas_mask�lengthr� Z chunk_lenr r r r� m s�
zWebSocketReader.parse_frameN)r r r r= r~ r� r r� r r r r r � s
[c @ s� e Z d Z d d d e d e j � d d d d d d � Z d
d d � Z d
d d � Z d
d d � Z d d
d d � Z
d d
d d � Z d
S)r �use_maskF�limit�randomr_ r r` c C sa | | _ | | _ | | _ | j | _ | | _ | | _ d | _ | | _ d | _ d | _ d S)NFr )
�protocol� transportr� � randranger_ r` �_closing�_limit� _output_size� _compressobj)r9 r� r� r� r� r� r_ r` r r r r= s zWebSocketWriter.__init__Nc C s� | j r t j d � d } | s+ | j r� | d k r� | rS t j d | � } n+ | j su t j d | j � | _ | j } | j | � } | | j | j r� t j n t j
� } | j t � r� | d d � } | d B} t
| � } | j } | rd } n d } | d k r2t d | B| B| | B� } nK | d k r_t d | B| Bd | B| � } n t d | B| Bd | B| � } | r| j d d
� }
|
j d d � }
t | � } t |
| � | j j | |
| � | j t
| � t
|
� t
| � 7_ nh t
| � t k r9| j j | � | j j | � n | j j | | � | j t
| � t
| � 7_ | j | j k r�d | _ | j j � St � S)z<Send a frame over the websocket with message as its payload.z websocket connection is closing.r r r� NrH �@ � r� r r3 r� l �� �big���i ) r� r �warningr_ r� � compressobjr� �flushr` � Z_FULL_FLUSH� Z_SYNC_FLUSH�endswithr� rO r� � PACK_LEN1� PACK_LEN2� PACK_LEN3r� �to_bytesrM r� r� �writer� �MSG_SIZEr� r� Z
_drain_helperr ) r9 r> r� r_ Zrsvr� Z
msg_lengthr� Zmask_bit�headerrQ r r r � _send_frame sR
!
,
zWebSocketWriter._send_framerm c C s1 t | t � r | j d � } | j | t j � S)zSend pong message.zutf-8)rL re �encoder� r r% )r9 r> r r r r- P s zWebSocketWriter.pongc C s1 t | t � r | j d � } | j | t j � S)zSend ping message.zutf-8)rL re r� r� r r$ )r9 r> r r r r, V s zWebSocketWriter.pingc C sT t | t � r | j d � } | r: | j | t j | � S| j | t j | � Sd S)z<Send a frame over the websocket with message as its payload.zutf-8N)rL re r� r� r r# r" )r9 r> r+ r_ r r r �send\ s
zWebSocketWriter.sendi� c C sS t | t � r | j d � } z$ | j t | � | d t j �SWd d | _ Xd S)z<Close the websocket, sending the specified code and message.zutf-8r� NT)rL re r� r� �PACK_CLOSE_CODEr r& r� )r9 r; r> r r r r. e s zWebSocketWriter.close) r r r �
DEFAULT_LIMITr� �Randomr= r� r- r, r� r. r r r r r s > )>r@ � collectionsr: r� �re�sysr� �enumr �structr Zhelpersr r �logr �__all__r r� r r
� unpack_fromr� r� �unpackr� �packr� r� r� r� r� r� �
namedtupler4 r
r( r r' r r� r rA � byteorderZnative_byteorderrF rI rT r� Z
_websocketrU � ImportErrorrE r� �compiler\ rY rb rg rh r r r r r r �<module> sj
4�