­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 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@_WSMessageBasetypedataextrac@s.eZdZejdeegefedddZdS)r)loads)r?returncCs ||jS)zAReturn parsed JSON data. .. versionadded:: 0.22 )r=)selfr?rrrjsonZszWSMessage.jsonN)rrrrBr?rrrrrrrYsrcs8eZdZdZeeddfdd ZedddZZS) rz WebSocket protocol parser error.Ncodemessager@cs||_t||dSN)rDsuper__init__rArDrE __class__rrrHiszWebSocketError.__init__r@cCs |jdS)Nr )argsrArrr__str__mszWebSocketError.__str__) rrr__doc__r strrHrO __classcell__rrrJrrfsrc@seZdZdZdS)WSHandshakeErrorz#WebSocket protocol handshake error.N)rrrrPrrrrrSqsrScs&g|]tfddtdDqS)c3s|]}|AVqdSrFr)r!abrr ysz.r()bytesrange)r!rrUr ysrZr()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|VqdSrF) _XOR_TABLE)r!nrrrrWsz)_websocket_mask_python..Nr r$) isinstance bytearrayAssertionErrorlen translate)r[r=rTrVcdrrr_websocket_mask_python|s rg)_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 rS)rjrkcompress notakeoverextZdefextrrrrr ws_ext_parses>      rvrl)rsrkserver_notakeoverr@cCs^|dks|dkrtddg}|s,|d|dkrF|dt||rT|dd|S) Nr%rlzCCompress wbits must between 9 and 15, zlib does not support wbits=8zpermessage-deflateZclient_max_window_bitszserver_max_window_bits=Zserver_no_context_takeoverz; ) ValueErrorappendrQjoin)rsrkrwZ enabledextrrr ws_ext_gens  r{c@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_sizersr@cCsp||_||_d|_t|_tj|_d|_d|_ d|_ t|_ d|_ d|_ d|_d|_d|_d|_d|_||_dS)NFr)r _max_msg_size_excra_partialr|r}_state_opcode _frame_fin _frame_opcode_frame_payload_tail _has_mask _frame_mask_payload_length_payload_length_flag _compressed_decompressobj _compress)rArrrsrrrrHs"zWebSocketReader.__init__rLcCs|jdSrF)rfeed_eofrNrrrrszWebSocketReader.feed_eofr=r@c CsZ|jrd|fSz ||WStk rT}z||_|j|WYdSd}~XYnXdS)NT)Tr)r _feed_data Exceptionr set_exception)rAr=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 @