o &æÑgôrã@sÔdZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZzddlZWneyNddlZe d¡dZYnwGdd„dejƒZGd d „d ejƒZGd d „d ƒZdS) 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@ó eZdZdS)ÚWebSocketWantReadErrorN©Ú__name__Ú __module__Ú __qualname__©r r úS/root/parts/websockify/install/lib/python3.10/site-packages/websockify/websocket.pyr"órc@r)ÚWebSocketWantWriteErrorNrr r r r r$r rc@s$eZdZdZdZdd„Zdd„Zdgfdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„ZdGd%d&„ZdGd'd(„ZdHd*d+„ZdHd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4d5„Zd6d7„Zd8d9„Zd:d;„Zdd?„Z!d@dA„Z"dIdCdD„Z#dEdF„Z$dS)JÚ 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-C5AB0DC85B11cCsFd|_d|_d|_g|_d|_d|_d|_d|_d|_d|_ d|_ dS)z Creates an unconnected WebSocketÚnewóNF) Ú_stateÚ _partial_msgÚ _recv_bufferÚ _recv_queueÚ _send_bufferÚ_previous_sendmsgÚ _sent_closeÚ_received_closeÚ close_codeÚ close_reasonÚsocket©Úselfr r r Ú__init__Gs zWebSocket.__init__cCs6|dvr|jdus J‚t|j|ƒStd|jj|fƒ‚)N)ÚfilenoÚ getpeernameÚ getsocknameÚ getsockoptÚ setsockoptÚ gettimeoutÚ settimeoutÚ setblockingz!%s instance has no attribute '%s')rÚgetattrÚAttributeErrorÚ __class__r)rÚnamer r r Ú __getattr__\s   ÿzWebSocket.__getattr__Nc Cs\d|_t|ƒ}|j}|jdvr|sd}n|jdvr|sd}ntd|jƒ‚|jdkrMt |j|f¡|_|jdvrJt   ¡}|j |j|jd|_d |_nd |_|jd krZ|j  ¡d |_|jd krÎd |_ td ƒD]}|j tt d ¡ƒ7_ qft|j  d¡ƒ d¡|_ |j}|sˆd}| d|¡| d|j¡| dd¡| dd¡| d|j ¡| dd¡|dur¸| d|¡t|ƒdkrÇ| dd |¡¡| ¡d|_|jdkrÚ| ¡d |_|jd krª| ¡sètd!ƒ‚|j d"¡d#kròt‚|j  d$d%¡\}|_| d¡}|  ¡} t| ƒd&ks| dd'krtd(ƒ‚| d%d)kr+td*d+ | d%d…¡ƒ‚|j  d"d%¡\} |_|  d¡d,} t! "| ¡} |  #dd ¡ $¡dkrVt%t&| ƒƒtd-ƒ‚|  #d.¡} | durdtd/ƒ‚t'|j |j( d¡ƒ )¡} t| ƒ d¡} |` | | krƒtd0ƒ‚|  #d¡|_*t|ƒdkr›|j*durštd1ƒ‚n |j*|vr¥td2ƒ‚d3|_dStd4ƒ‚)5azEstablishes 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)ÚwsÚhttpéP)ÚwssÚhttpsi»zUnknown scheme '%s'r)Úserver_hostnameÚ ssl_handshakeÚheadersÚééúlatin-1Úasciiú/ÚGETÚHostÚUpgradeÚ websocketÚ ConnectionÚupgradeúSec-WebSocket-KeyúSec-WebSocket-Versioné NÚOriginrúSec-WebSocket-Protocolz, Ú send_headersÚresponsezSocket closed unexpectedlys éÿÿÿÿs éézHTTP/1.1zInvalid responseÚ101zWebSocket request denied: %sú ú ú#Missing or incorrect upgrade headerúSec-WebSocket-Acceptz#Missing Sec-WebSocket-Accept headerz#Invalid Sec-WebSocket-Accept headerz(Unexpected Sec-WebSocket-Protocol headerz!Invalid protocol chosen by serverÚdoneú WebSocket is in an invalid state)+ÚclientrÚportÚschemeÚ ExceptionrrÚcreate_connectionÚhostnameÚsslÚcreate_default_contextÚ wrap_socketÚ do_handshakeÚ_keyÚrangeÚchrÚrandomÚ randrangerÚencodeÚdecodeÚpathÚ send_requestÚ send_headerÚlenÚjoinÚ end_headersÚ_flushÚ_recvrÚfindrÚsplitÚemailÚmessage_from_stringÚgetÚlowerÚprintÚtyperÚGUIDÚdigestÚprotocol) rÚuriÚoriginÚ protocolsrSÚcontextÚircÚrequestÚwordsr4ÚacceptÚexpectedr r r Úconnectis¦ € €  ÿ                    ÿ zWebSocket.connectcCs`|jdkržd|_||_| dd¡ ¡dkrtdƒ‚| d¡}|dur&td ƒ‚|d vr2d t|ƒ|_ntd |ƒ‚| d ¡}|durEtdƒ‚t||j   d¡ƒ  ¡}t |ƒ  d¡}d|_| dd¡ d¡}|ru| |¡|_|j|vrutdƒ‚| dd¡| dd¡| dd¡| d|¡|jr—| d|j¡| ¡d|_|jdkr¬| ¡d|_dStdƒ‚)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. rFr@r5r>rNrBNz$Missing Sec-WebSocket-Version header)Ú7Ú8Ú13z hybi-%02dzUnsupported protocol version %srAz Missing Sec-WebSocket-Key headerr9rEú,zInvalid protocol selectedéezSwitching Protocolsr=r?rOÚflushrPrQ)rrRrrorprUÚintÚversionrrsrartrrbrurlÚselect_subprotocolÚ send_responsererhri)rrr4ÚverÚkeyr}rxr r r r}ésF           zWebSocket.acceptcCódS)aReturns which sub-protocol should be used. This method does not select any sub-protocol by default and is meant to be overridden by an implementation that wishes to make use of sub-protocols. It will be called during handling of accept(). r5r )rrxr r r rˆ4szWebSocket.select_subprotocolcCs| |¡dS)z®Called when a WebSocket ping message is received. This will be called whilst processing recv()/recvmsg(). The default implementation sends a pong reply back.N)Úpong©rÚdatar r r Ú handle_ping>szWebSocket.handle_pingcCrŒ)z£Called when a WebSocket pong message is received. This will be called whilst processing recv()/recvmsg(). The default implementation does nothing.Nr rŽr r r Ú handle_pongEszWebSocket.handle_pongcCs| ¡S)a¹Read data from the WebSocket. This will return any available data on the socket (which may be the empty string if the peer sent an empty message or messages). If the socket is closed then None will be returned. The reason for the close is found in the 'close_code' and 'close_reason' properties. Unlike recvmsg() this method may return data from more than one WebSocket message. It is however not guaranteed to return all buffered data. Callers should continue calling recv() whilst pending() returns True. Both WebSocketWantReadError and WebSocketWantWriteError can be raised when calling recv(). )Úrecvmsgrr r r ÚrecvLszWebSocket.recvcCs6|jr | ¡dS| ¡r| ¡S| ¡sdS| ¡S)aŠRead a single message from the WebSocket. This will return a single WebSocket message from the socket (which will be the empty string if the peer sent an empty message). If the socket is closed then None will be returned. The reason for the close is found in the 'close_code' and 'close_reason' properties. Unlike recv() this method will not return data from more than one WebSocket message. Callers should continue calling recvmsg() whilst pending() returns True. Both WebSocketWantReadError and WebSocketWantWriteError can be raised when calling recvmsg(). N)rriÚpendingÚ_recvmsgÚ _recv_framesrr r r r’_szWebSocket.recvmsgcCst|jƒdkS)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)rfrrr r r r”ƒs zWebSocket.pendingcCst|ƒdkrdS| |¡S)aÀWrite 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)rfÚsendmsg)rÚbytesr r r Úsend’s  zWebSocket.sendcCstt|tƒst‚|jr dS|jdur#|j|krt‚| ¡d|_t|ƒSz | d|¡Wt|ƒSt y9||_‚w)a­Write 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. rNrJ) Ú isinstancer˜Ú TypeErrorrrÚ ValueErrorrirfÚ_sendmsgr)rÚmsgr r r r—¢s"    üþzWebSocket.sendmsgcCó| d||f¡dS)NzHTTP/1.1 %d %s ©Ú _queue_str)rÚcodeÚmessager r r r‰ÄózWebSocket.send_responsecCrŸ)Nz%s: %s r )rÚkeywordÚvaluer r r reÇr¤zWebSocket.send_headercCs| d¡dS)NrMr rr r r rhÊszWebSocket.end_headerscCs| d| ¡|f¡dS)Nz%s %s HTTP/1.1 )r¡Úupper)rrrrcr r r rdÍszWebSocket.send_requestrcCóbt|tƒst‚|jdur|j|krt‚| ¡d|_dSz | d|¡WdSty0||_‚w)zùWrite 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é ©ršr˜r›rrœrirrrŽr r r ÚpingÐó    þzWebSocket.pingcCr¨)zùWrite 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é rªrŽr r r rér¬zWebSocket.pongéècCsl|jr | ¡dS|jsd|_d|_d|_d}|dur.|t d|¡7}|dur.|| d¡7}| d|¡dS) 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. Nr®zLocally initiated closeTrú>HúUTF-8é) rrirrrÚstructÚpackrar)rÚhowr¢Úreasonržr r r ÚshutdownszWebSocket.shutdowncCs| tj||¡| ¡dS)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)r¶rÚ SHUT_RDWRÚ_close)rr¢rµr r r Úclose's zWebSocket.closec Cs|jdusJ‚ z|j d¡}Wnty%}z |jtjkr t‚‚d}~wwt|ƒdkr.dS|j|7_t|jdƒrD|j  ¡sC dSn dSq)NTirFr”) rr“ÚOSErrorÚerrnoÚ EWOULDBLOCKrrfrÚhasattrr”)rrÚexcr r r rj5s* €ý   ûìzWebSocket._recvcCst| ¡s|jdurd|_d|_d|_|_| ¡dS | |j¡}|dur) dS|j|dd…|_|j  |¡q)NiîzConnection closed abnormallyTFÚlength) rjrrrrr¸Ú _decode_hybirrÚappend)rÚframer r r r–Os   ý ûzWebSocket._recv_framescCs¢|jrO|j d¡}|js|ds| tjdd¡q|jr*|dr*| tjdd¡q|ddkrR|js<| tjdd¡q|j|d7_|d rQ|j}d |_|Snú|dd kra| tjd d ¡në|ddkr|jrs| tjdd¡q|d r{|dS|d|_nË|ddkr|jrŒqd|_|jr˜|  ¡dS|d s¥| tjd d¡qd}d}t |dƒdkrèt   d|ddd…¡d}t |dƒdkrè|ddd…}z|  d¡}Wntyç| tjdd¡Yqw|duród|_d|_n ||_|durý||_| d||¡dS|ddkr#|d s| tjd d¡q| |d¡n)|ddkr@|d s8| tjd d¡q| |d¡n | tjd d|d¡|jst‚)NrÚmaskediêz Procotol error: Frame not maskedzProcotol error: Frame maskedÚopcodez-Procotol error: Unexpected continuation frameÚpayloadÚfinrrIiëz*Unsupported: Text frames are not supportedrJz$Procotol error: Unexpected new framer±TzUnsupported: Fragmented closer¯r°z&Procotol error: Invalid UTF-8 in closeiíz&No close status code specified by peerr©zUnsupported: Fragmented pingr­zUnsupported: Fragmented pongz"Unsupported: Unknown opcode 0x%02x)rÚpoprRr¶rr·rrrr¸rfr²ÚunpackrbÚUnicodeDecodeErrorrrrr‘r)rrÂržr¢rµr r r r•bsŒ  ý    þ  ¯SzWebSocket._recvmsgc Cs|jsdS|jdus J‚z |j |j¡}Wnty*}z |jtjkr%t‚‚d}~ww|j|d…|_|jr8t‚|jrD|jrF|  ¡dSdSdS©N) rrr™rºr»r¼rrrr¸)rÚsentr¾r r r ri¹s" €ý  ÿzWebSocket._flushcCs|j|7_| ¡dSrÊ)rrirŽr r r Ú_sendÑs zWebSocket._sendcCs|j| d¡7_dS)Nr8)rra)rÚstringr r r r¡ÖszWebSocket._queue_strcCsP|jrd}tdƒD] }|t d¡ ¡7}q | |||¡}n| ||¡}| |¡S)Nrér7)rRr]r_r`Úto_bytesÚ _encode_hybirÌ)rrÄržÚmaskrzrÂr r r rÛs   zWebSocket._sendmsgcCs|j ¡d|_dSrÊ)rr¹rr r r r¸çs  zWebSocket._closecCs | ||¡SrÊ)Ú_unmask)rÚbufrÑr r r Ú_maskìs zWebSocket._maskc Cs0trut|ƒ}d}|}d}}|dkr=t d¡}tjdkr!| d¡}tj||dd}tj||t|dƒd} t | |¡  ¡}|drqt d ¡}tjdkrP| d¡}tj|||dd}tj||||d|dd } t | |¡  ¡}||St   d ¡} |   |¡t t| ƒƒD]} | | || dN<q…|   ¡S) NrrrÎzrI)ÚcountÚB)Úoffsetr×) ÚnumpyrfÚdtypeÚsysÚ byteorderÚ newbyteorderÚ frombufferr†Ú bitwise_xorÚtobytesÚarrayÚ frombytesr]) rrÓrÑÚplenÚpstartÚpendÚbÚcrÛrrzr r r rÒðs6      ÿ  zWebSocket._unmaskTc Cs²|d@}|r |dO}d}|durd}| ||¡}t|ƒ}|dkr*t d|||B¡}n!|dkr=|dkr=t d|d |B|¡}n|dkrKt d |d |B|¡}|durU|||S||S) zó Encode a HyBi style WebSocket frame. Optional opcode: 0x0 - continuation 0x1 - text frame 0x2 - binary frame 0x8 - connection close 0x9 - ping 0xA - pong éé€rNé}ú>BBiz>BBHé~z>BBQé)rÔrfr²r³) rrÄrÓÚmask_keyrÆÚb1Úmask_bitÚ payload_lenÚheaderr r r rÐs"   zWebSocket._encode_hybic Csbddddddœ}t|ƒ}d}||krdSt d|dd…¡\}}|d@|d<|d @ |d <|d @ |d <|d rD|d 7}||krDdS|d @}|dkrb|d7}||krVdSt d|dd …¡\}n|d kr{|d7}||krpdSt d|dd…¡\}|||d<||dkr‰dS|d r¥||d |…}| ||||…|¡|d<|S||||…|d<|S)a Decode HyBi style WebSocket packets. Returns: {'fin' : boolean, 'opcode' : number, 'masked' : boolean, 'length' : encoded_length, 'payload' : decoded_buffer} rFN)rÆrÄrÃr¿rÅrJrìrérÄrêrÆrÃrÎrîrír¯r±z>Qr­r¿rÅ)rfr²rÈrÒ) rrÓÚfÚblenÚhlenrðÚb2r¿rïr r r rÀ2sL ü   þzWebSocket._decode_hybi)r)r®N)NT)%rr r Ú__doc__rsrr,rr}rˆrr‘r“r’r”r™r—r‰rerhrdr«rr¶r¹rjr–r•rirÌr¡rr¸rÔrÒrÐrÀr r r r r'sH K $"    %W  ! !r)rørÜrârmr»r_rrXr²Úbase64rÚhashlibrÚ urllib.parserrÚÚ ImportErrorÚwarningsÚwarnÚSSLWantReadErrorrÚSSLWantWriteErrorrrr r r r Ús,       ý