­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ U '7` b@sjdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZmZmZddlmZddlmZdd lmZd ZGd d d e Zd deDZGddde ZdZe djZe djZ e dj!Z"e dj#Z$e dj#Z%e dj#Z&e dj#Z'dZ(dZ)e*ddddgZ+Gddde+Z,e,ej-ddZ.e,ej/ddZ0Gdd d e1Z2Gd!d"d"e1Z3ej4Z5d#d$e6d%DZ7e8e9dd&d'd(Z:ere:Z;n0zdd)lk re:Z;YnXe8ddd*d*gZ?e@d+ZAe@d,ZBd;eeCeDeeEeDfd.d/d0ZFd-sr#c@sPeZdZdZdZdZdZdZdZdZ dZ d Z eZ eZ eZeZeZe Ze Ze Zd S) rrr  iiN)rrr CONTINUATIONTEXTBINARYPINGPONGCLOSECLOSINGCLOSEDERRORtextbinarypingpongcloseclosingclosederrorrrrrr0s"rs$258EAFA5-E914-47DA-95CA-C5AB0DC85B11z!Hz!Qz!BBz!BBHz!BBQi@_WSMessageBasetypedataZextrac@s.eZdZejdeegefedddZdS)r)loads)r>returncCs ||jS)zAReturn parsed JSON data. .. versionadded:: 0.22 )r=)selfr>rrrjsonZszWSMessage.jsonN)rrrrAr>rrrrrrrYsrcs8eZdZdZeeddfdd ZedddZZS) rz WebSocket protocol parser error.Ncodemessager?cs||_t||dSN)rCsuper__init__r@rCrD __class__rrrGiszWebSocketError.__init__r?cCs |jdS)Nr )argsr@rrr__str__mszWebSocketError.__str__) rrr__doc__r strrGrN __classcell__rrrIrrfsrc@seZdZdZdS)WSHandshakeErrorz#WebSocket protocol handshake error.N)rrrrOrrrrrRqsrRcs&g|]tfddtdDqS)c3s|]}|AVqdSrEr)r!abrr ysz.r()bytesrange)r!rrTr ysrYr()maskr=r?cCst|tst|t|dks&t||rdd|D\}}}}|ddd||ddd<|ddd||ddd<|ddd||ddd<|ddd||ddd<dS)awWebsocket 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. css|]}t|VqdSrE) _XOR_TABLE)r!nrrrrVsz)_websocket_mask_python..Nr r$) isinstance bytearrayAssertionErrorlen translate)rZr=rSrUcdrrr_websocket_mask_python|s rf)_websocket_mask_cythonz^(?:;\s*(?:(server_no_context_takeover)|(client_no_context_takeover)|(server_max_window_bits(?:=(\d+))?)|(client_max_window_bits(?:=(\d+))?)))*$zpermessage-deflate([^,]+)?F)extstrisserverr?cCs|sdSd}d}t|D]}|d}|s4d}qt|}|rd}|r|drxt|d}|dksr|dkrxd}q|drd}qq|d rt|d }|dks|dkrtd |d rd}qq|std |dq||fS) N)rFrFr r[r%TzInvalid window sizer$z#Extension for deflate not supported)_WS_EXT_RE_SPLITfinditergroup _WS_EXT_REmatchr rR)rirjcompress notakeoverextZdefextrqrrr ws_ext_parses>      rurk)rrrjserver_notakeoverr?cCs^|dks|dkrtddg}|s,|d|dkrF|dt||rT|dd|S) Nr%rkzCCompress wbits must between 9 and 15, zlib does not support wbits=8zpermessage-deflateZclient_max_window_bitszserver_max_window_bits=Zserver_no_context_takeoverz; ) ValueErrorappendrPjoin)rrrjrvZ enabledextrrr ws_ext_gens  rzc@seZdZdZdZdZdZdS) WSParserStater r$r^r[N)rrr READ_HEADERREAD_PAYLOAD_LENGTHREAD_PAYLOAD_MASK READ_PAYLOADrrrrr{sr{c@seZdZdeeeeddddZddddZe e ee fd d d Z e e ee fd d d Z e e e eeeeeefdddZdS)rTN)queue max_msg_sizerrr?cCsp||_||_d|_t|_tj|_d|_d|_ d|_ t|_ d|_ d|_ d|_d|_d|_d|_d|_||_dS)NFr)r _max_msg_size_excr`_partialr{r|_state_opcode _frame_fin _frame_opcode_frame_payload_tail _has_mask _frame_mask_payload_length_payload_length_flag _compressed_decompressobj _compress)r@rrrrrrrrGs"zWebSocketReader.__init__rKcCs|jdSrE)rfeed_eofrMrrrrszWebSocketReader.feed_eofr=r?c CsZ|jrd|fSz ||WStk rT}z||_|j|WYdSd}~XYnXdS)NT)Tr)r _feed_data ExceptionrZ set_exception)r@r=excrrr feed_datas  zWebSocketReader.feed_datac Cs||D]v\}}}}|r4|js4tjtj d|_|tjkrt|dkrt|ddd}|dkr|t krt t j d|z|dd d}Wn0tk r}zt t jd|W5d}~XYnXttj||} n2|rt t j d|d |d |nttjdd } |j| dq |tjkrF|jttj|d t|q |tjkrp|jttj|d t|q |tjtjfkr|jdkrt t j d |q |s|tjkr||_|j||jrt|j|jkrt t jd t|j|jq |jr$|tjkr$t t j d ||tjkrL|jdk s@t|j}d|_|j||jrt|j|jkrt t jd t|j|j|r|jt|j |j|j} |jj!rt|jj!} t t jd|j| |jn t"|j} |j#|tjkrfz*| d} |jttj| d t| Wn2tk rb}zt t jd|W5d}~XYnXq |jttj| d t| q dS)N)wbitsr$ri zInvalid close code: utf-8zInvalid UTF-8 text messagezInvalid close frame:  zUnexpected opcode=z Message size {} exceeds limit {}zs~                    4 @