ZU
@ 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_EXTENSIONSnoop) ws_loggerWS_CLOSED_MESSAGEWS_CLOSING_MESSAGEWS_KEYWebSocketReaderWebSocketWriter WSMessageWebSocketError 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_AWAYPROTOCOL_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).0ir 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 CONTINUATIONTEXTBINARYPINGPONGCLOSECLOSINGCLOSEDERRORtextbinarypingpongcloseclosingclosederrorr r r r r & s" s$ 258EAFA5-E914-47DA-95CA-C5AB0DC85B11z!Hz!Qz!BBz!BBHz!BBQr _WSMessageBasetypedataextrac @ 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)codesuper__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 )bytesrange)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 bytearrayAssertionErrorlen translate)maskr6 rB rC cdr 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_SPLITfinditergroup
_WS_EXT_REmatchr rA )Zextstrisservercompress
notakeoverextZdefextr] 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; )
ValueErrorappendstrjoin)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_HEADERREAD_PAYLOAD_LENGTHREAD_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 rt | 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 rt
t j d j
| | t j k r| j } d | _ | j j | d j | j }
| r| j j |
t }
| j j | t j k ry; |
j d } | j j t t j | d t | Wqt k
r} z t
t j d | WYd d } ~ XqXq | j j t t j |
d t |
q Wd S)Nwbitsr 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_CODEALLOWED_CLOSE_CODESr r r formatdecodeUnicodeDecodeErrorr r
rn r r$ r% r" r# rr r! rp rd rf
decompress_WS_DEFLATE_TRAILINGclear) r9 r6 finopcodepayload
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| rd n d | _
n | rt 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 rt 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_masklengthr 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_maskFlimitrandomr_ 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 rd | _ | 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 bigi ) r r warningr_ r compressobjr flushr` Z_FULL_FLUSH Z_SYNC_FLUSHendswithr 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_bitheaderrQ 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 resysr 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