#&aJq&dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZ ddlZGddej&ZGd d ej*ZGd d eZy#e$rddlZej$ddZYUwxYw) al Python WebSocket library Copyright 2011 Joel Martin Copyright 2016 Pierre Ossman Licensed under LGPL version 3 (see docs/LICENSE.LGPL-3) Supports following protocol versions: - http://tools.ietf.org/html/draft-ietf-hybi-thewebsocketprotocol-07 - http://tools.ietf.org/html/draft-ietf-hybi-thewebsocketprotocol-10 - http://tools.ietf.org/html/rfc6455 N) b64encode)sha1)urlparsez/no 'numpy' module, HyBi protocol will be slowerc eZdZy)WebSocketWantReadErrorN__name__ __module__ __qualname__6/usr/lib/python3/dist-packages/websockify/websocket.pyrr"r rc eZdZy)WebSocketWantWriteErrorNrr r rrr$rr rceZdZdZdZdZdZdgfdZdZdZ d Z d Z d Z d Z d ZdZdZd dZd dZd!dZd!dZdZdZdZdZdZdZdZdZdZdZd"dZdZ y)# WebSocketaoWebSocket protocol socket like class. This provides access to the WebSocket protocol by behaving much like a real socket would. It shares many similarities with ssl.SSLSocket. The WebSocket protocols requires extra data to be sent and received compared to the application level data. This means that a socket that is ready to be read may not hold enough data to decode any application data, and a socket that is ready to be written to may not have enough space for an entire WebSocket frame. This is handled by the exceptions WebSocketWantReadError and WebSocketWantWriteError. When these are raised the caller must wait for the socket to become ready again and call the relevant function again. A connection is established by using either connect() or accept(), depending on if a client or server session is desired. See the respective functions for details. The following methods are passed on to the underlying socket: - fileno - getpeername, getsockname - getsockopt, setsockopt - gettimeout, settimeout - setblocking z$258EAFA5-E914-47DA-95CA-C5AB0DC85B11cd|_d|_d|_g|_d|_d|_d|_d|_d|_d|_ d|_ y)z Creates an unconnected WebSocketnewr NF) _state _partial_msg _recv_buffer _recv_queue _send_buffer_previous_sendmsg _sent_close_received_close close_code close_reasonsocketselfs r__init__zWebSocket.__init__GsY !% $  r c|dvr$|jJt|j|St|jjd|d)N)fileno getpeername getsockname getsockopt setsockopt gettimeout settimeout setblockingz instance has no attribute '')r getattrAttributeError __class__r )r"names r __getattr__zWebSocket.__getattr__\sV # # ;;* **4;;- - "&.."9"94"AB Br Nc d|_t|}|j}|jdvr|s.d}n+|jdvr|sd}nt d|jz|j dk(rgt j|j|f|_|jdvr,tj|j |_d|_nd |_|j dk(r!|j jd |_|j d k(rrd |_ td D]3}|xjttj d z c_ 5t#|jj%d j'd|_ |j(}|sd}|j+d|z|j+d|jz|j+d|j+d|j+d|jz|j+d||j+d|zt-|dkDr#|j+ddj/|z|j+dd|_|j dk(r|j1d|_|j dk(rA|j3s t d|j4j7dd k(rt8|j4j;d!d"\}|_|j'd }|j;}t-|d#ks|dd$k7r t d%|d"d&k7r t d'd(j/|d"dz|j4j;dd"\} |_| j'd dz} t=j>| } | jAd)d jCd*k7rtEtG| t d+| jAd,} | t d-tI|j|jJzj%djM} t#| j'd} |` | | k7r t d.| jAd/|_'t-|dk(r|jN$t d0|jN|vr t d1d2|_yt d3)4azEstablishes a new connection to a WebSocket server. This method connects to the host specified by uri and negotiates a WebSocket connection. origin should be specified in accordance with RFC 6454 if known. A list of valid sub-protocols can be specified in the protocols argument. The data will be sent in the clear if the "ws" scheme is used, and encrypted if the "wss" scheme is used. Both WebSocketWantReadError and WebSocketWantWriteError can be raised whilst negotiating the connection. Repeated calls to connect() must retain the same arguments. T)wshttpP)wsshttpsizUnknown scheme '%s'r ssl_handshakeheaderslatin-1ascii/zGET %s HTTP/1.1 z Host: %s Upgrade: websocket zConnection: upgrade zSec-WebSocket-Key: %s zSec-WebSocket-Version: 13 Nz Origin: %s rSec-WebSocket-Protocol: %s z,  send_headersresponsezSocket closed unexpectedlys s zHTTP/1.1zInvalid response101zWebSocket request denied: %s Upgrade websocket#Missing or incorrect upgrade headerzSec-WebSocket-Acceptz#Missing Sec-WebSocket-Accept headerz#Invalid Sec-WebSocket-Accept headerSec-WebSocket-Protocolz(Unexpected Sec-WebSocket-Protocol headerz!Invalid protocol chosen by serverdone WebSocket is in an invalid state)(clientrportscheme Exceptionrr create_connectionhostnamessl wrap_socket do_handshake_keyrangechrrandom randrangerencodedecodepath _queue_strlenjoin_flush_recvrfindrsplitemailmessage_from_stringgetlowerprinttyperGUIDdigestprotocol) r"uriorigin protocolsrRirarequestwordsr:acceptexpecteds rconnectzWebSocket.connectis>  smxx :: ' ZZ+ +1CJJ>? ? ;;%  22CLL$3GHDKzz--!oodkk: - ' ;;/ ) KK $ $ &#DK ;;) #DI2Y 8 S!1!1#!677  8!$))"2"29"=>EEgNDI88D OO1D8 9 OONS\\9 : OO4 5 OO5 6 OO7$))C D OO; <! 06 9:9~! @499YCW WX OOF #(DK ;;. ( KKM$DK ;;* $::< <==  %%k2b8,,+/+<+<+B+B7A+N (Wd'nnY/GMMOEE QE!H $: 233Qx5  >%PQPR)AT TUU+/+<+<+B+B;PQ+R (Wd'nnY/&8G//8G{{9b)//1[@d7m$ EFF[[!78F~ EFFTYY2::7CDKKMH *11':H ! EFF#KK(@ADM9~"==,#$NOO== 1#$GHH DK :;;r c$|jdk(rd|_||_|jddj dk7r t d|jd}| t d |d vrd t |z|_nt d |z|jd }| t dt||jzjdj}t|jd}d|_|jddjd}|r/|j!||_|j|vr t d|j#d|j#d|j#d|j#d|z|jr|j#d|jz|j#dd|_|jdk(r|j%d|_yt d)afEstablishes a new WebSocket session with a client. This method negotiates a WebSocket connection with an incoming client. The caller must provide the client socket and the headers from the HTTP request. A server can identify that a client is requesting a WebSocket connection by looking at the "Upgrade" header. It will include the value "websocket" in such cases. WebSocketWantWriteError can be raised if the response cannot be sent right away. accept() must be called again once more space is available using the same arguments. rFupgrader;rLrMzSec-WebSocket-VersionNz$Missing Sec-WebSocket-Version header)7813z hybi-%02dzUnsupported protocol version %szSec-WebSocket-Keyz Missing Sec-WebSocket-Key headerr?rN,zInvalid protocol selectedz"HTTP/1.1 101 Switching Protocols rAzConnection: Upgrade zSec-WebSocket-Accept: %s rBrCflushrOrP)rrQr rkrlrTintversionrror_rprr`rqrhselect_subprotocolrbre)r"r r:verkeyrxrts rrxzWebSocket.accepts& ;;% DK DK{{9b)//1[@ EFF++56C{ FGG &&*SX5  AC GHH++12C{ BCC3?227;<CCEFv&--g6FDM $==? "  "}}r c2t|jdkDS)aCheck if any WebSocket data is pending. This method will return True as long as there are WebSocket frames that have yet been processed. A single recv() from the underlying socket may return multiple WebSocket frames and it is therefore important that a caller continues calling recv() or recvmsg() as long as pending() returns True. Note that this function merely tells if there are raw WebSocket frames pending. Those frames may not contain any application data. r)rcrr!s rrzWebSocket.pendings4##$q((r cBt|dk(ry|j|S)aWrite data to the WebSocket This will queue the given data and attempt to send it to the peer. Unlike sendmsg() this method might coalesce the data with data from other calls, or split it over multiple messages. WebSocketWantWriteError can be raised if there is insufficient space in the underlying socket. send() must be called again once more space is available using the same arguments. r)rcsendmsg)r"bytess rsendzWebSocket.sends! u:?||E""r c4t|tst|jry|j7|j|k7rt |j d|_t|S |jd|t|S#t$r ||_wxYw)aWrite a single message to the WebSocket This will queue the given message and attempt to send it to the peer. Unlike send() this method will preserve the data as a single WebSocket message. WebSocketWantWriteError can be raised if there is insufficient space in the underlying socket. sendmsg() must be called again once more space is available using the same arguments. rNrH) isinstancer TypeErrorrr ValueErrorrerc_sendmsgr)r"msgs rrzWebSocket.sendmsgs#u%O     ! ! -%%,  KKM%)D "s8O  MM#s # 3x ' %(D "  s (BBct|tst|j-|j|k7rt|j d|_y |j d|y#t$r ||_wxYw)zWrite a ping message to the WebSocket WebSocketWantWriteError can be raised if there is insufficient space in the underlying socket. ping() must be called again once more space is available using the same arguments. N rrrrrrerrrs rpingzWebSocket.pingt$&O  ! ! -%%-  KKM%)D "   MM#t $& %)D "   A$$A6ct|tst|j-|j|k7rt|j d|_y |j d|y#t$r ||_wxYw)zWrite a pong message to the WebSocket WebSocketWantWriteError can be raised if there is insufficient space in the underlying socket. pong() must be called again once more space is available using the same arguments. N rrs rrzWebSocket.pongrrc |jr|jy|jsd|_d|_d|_d}|/|t j d|z }|||jdz }|jd|y) asGracefully terminate the WebSocket connection. This will start the process to terminate the WebSocket connection. The caller must continue to calling recv() or recvmsg() after this function in order to wait for the peer to acknowledge the close. Calls to send() and sendmsg() will be ignored. WebSocketWantWriteError can be raised if there is insufficient space in the underlying socket for the close message. shutdown() must be called again once more space is available using the same arguments. The how argument is currently ignored. NzLocally initiated closeTr >HUTF-8) rrerrrstructpackr_r)r"howcodereasonrs rshutdownzWebSocket.shutdowns$    KKM ##"DO 9D    6;;tT* *C!v}}W-- c3r cf|jtj|||jy)aTerminate the WebSocket connection immediately. This will close the WebSocket connection directly after sending a close message to the peer. WebSocketWantWriteError can be raised if there is insufficient space in the underlying socket for the close message. close() must be called again once more space is available using the same arguments. N)rr SHUT_RDWR_close)r"rrs rclosezWebSocket.closes# f&&f5 r ct|jJ |jjd}t |dk(ry|xj|z c_t|jdr|jjs y yu#t$r)}|jtjk(rt d}~wwxYw)NTirFr) r rOSErrorerrno EWOULDBLOCKrrcrhasattrr)r"rexcs rrfzWebSocket._recv's{{&&& {{''- 4yA~    % t{{I.{{**,) 99 1 1100 sB B7$B22B7c:|js9|jd|_d|_dx|_|_|j y |j |j}| y|j|dd|_|jj|R)NizConnection closed abnormallyTFlength) rfrrrrr _decode_hybirrappend)r"frames rrzWebSocket._recv_framesAszz|&"&$B!:>> 4#7 KKM%%d&7&78E}!% 1 1%/2B CD     # #E * r c|jr|jjd}|js'|ds"|jtj dd[|jr'|dr"|jtj dd|ddk(ra|j s"|jtj dd|xj |dz c_|d r6|j }d |_|S|dd k(r#|jtj d d n|ddk(rE|j r#|jtj ddY|d r|dS|d|_n|ddk(r|jrd|_|jr|jy|d s#|jtj d dd}d}t|ddk\rJtjd|dddd}t|ddkDr|ddd} |jd}|dx|_}d|_n||_|||_|jd||y|ddk(r=|d s#|jtj d d|j#|dnl|ddk(r=|d s#|jtj d d|j%|dn'|jtj d d|dz|jrt&#t$r%|jtj ddYYwxYw)Nrmaskediz Procotol error: Frame not maskedzProcotol error: Frame maskedopcodez-Procotol error: Unexpected continuation framepayloadfinr rGiz*Unsupported: Text frames are not supportedrHz$Procotol error: Unexpected new framerTzUnsupported: Fragmented closerrz&Procotol error: Invalid UTF-8 in closeiz&No close status code specified by peerrzUnsupported: Fragmented pingrzUnsupported: Fragmented pongz"Unsupported: Unknown opcode 0x%02x)rpoprQrr rrrrrrcrunpackr`UnicodeDecodeErrorrrrrr)r"rrrrs rrzWebSocket._recvmsgTs3$$((+E;;uX f..6XY{{uX f..6TUX#%((MM&"2"2D:ij!!U9%55!<++C(+D%JxC' f..6bcxC'$$MM&"2"2D:`a< ++(-i(8D%xC''''+$##KKMU|MM&"2"2D:YZuY'(A-!==uY/?/CDQGD5+,q0!&y!1!"!5%%+]]7%;F <-11DOd(PD%&*DO),2) dD&1xC'U|MM&"2"2D:XY  y!12xC'U|MM&"2"2D:XY  y!12 f..6Z]bck]l6lmcf%$; 2% MM&*:*:DBjk$%s L--*MMc|jsy|jJ |jj|j}|j|d|_|jrt |jr|jr|jyyy#t$r)}|jtj k(rt d}~wwxYwN) rr rrrrrrrr)r"sentrs rrezWebSocket._flushs  {{&&& ;;##D$5$56D !--de4   ) )   D$4$4 KKM%5  yyE-----  s%B C$CCcN|xj|z c_|jyr)rrers r_sendzWebSocket._sends T! r cL|xj|jdz c_y)Nr>)rr_)r"strings rrbzWebSocket._queue_strs V]]955r c|jr>d}tdD]}|tjdz }|j |||}n|j ||}|j |S)Nr r=)rQr[r]r^ _encode_hybir)r"rrmaskrurs rrzWebSocket._sendmsgsk ;;D1X .((-- .%%fc48E%%fc2Ezz%  r cF|jjd|_yr)r rr!s rrzWebSocket._closes  r c&|j||Sr)_unmask)r"bufrs r_maskzWebSocket._masks||C&&r c~trUt|}d}|}dx}}|dk\rtjd}tjdk(r|j d}tj ||d}tj ||t|dz } tj| |j}|dzrtjd }tjdk(r|j d}tj |||dz}tj ||||dzz |dz } tj| |j}||zStjd } | j|tt| D]} | | xx|| dzzcc<| jS) Nrr rzrG)countB)offsetr) numpyrcdtypesys byteorder newbyteorder frombufferr bitwise_xortobytesarray frombytesr[) r"rrplenpstartpendbcrrrus rrzWebSocket._unmasks s8DFDKAqykk%(==E)!..s3E''e1=''U#dQh-H%%dD199;axkk#&==E)!..s3E''eD1HF''U#tax0D%%dD199;q5L;;s#D NN3 3t9% 'Q4A;& '<<> !r cJ|dz}|r|dz}d}|d}|j||}t|}|dkrtjd|||z}nF|dkDr!|dkrtjd|d|z|}n |dk\rtjd |d |z|}||z|zS|zS) z Encode a HyBi style WebSocket frame. Optional opcode: 0x0 - continuation 0x1 - text frame 0x2 - binary frame 0x8 - connection close 0x9 - ping 0xA - pong r}>BBiz>BBH~z>BBQ)rrcrr) r"rrmask_keyrb1mask_bit payload_lenheaders rrzWebSocket._encode_hybisd]  $JB  H**S(+C#h # [[K(,BCF 3 ;#6[[S8^[IF E ![[S8^[IF  H$s* *C< r cdddddd}t|}d}||krytjd|dd\}}|dz|d<|d z |d <|d z |d <|d r |d z }||kry|d z}|dk(r'|dz }||krytjd|dd \}n+|d k(r&|dz }||krytjd|dd\}||z|d<||dkry|d r%||d z |}|j||||z||d<|S||||z|d<|S)a Decode HyBi style WebSocket packets. Returns: {'fin' : boolean, 'opcode' : number, 'masked' : boolean, 'length' : encoded_length, 'payload' : decoded_buffer} rFN)rrrrrrHrrrrrrrrrrrz>Qrrr)rcrrr) r"rfblenhlenrb2rrs rrzWebSocket._decode_hybi$sz #"  $ 3x $;uc"1g.B4i( I&%!Dy/)( X; AIDd{d S= AIDd{mmD#a(3GF s] AIDd{mmD#a)4GFVm( !H+  X;46$'H<<D$v+(?JAiLtT&[2AiLr )r )rN)NT)!r r r __doc__ror#r2rzrxrrrrrrrrrrrrrfrrrerrbrrrrrrr r rrr's: 2D* B#'"|<|Ir s  !  S11  c33 v v HMMCD EsA00BB