B @¦CZP,ã@sˆddlZddlZddlZddlZddlZddlmZmZddlm Z dZ dd„Z ddd „Z dd d „Z Gd d „d eƒZGdd„deƒZdS)éN)Ú b64decodeÚ b64encode)Úsha1s$258EAFA5-E914-47DA-95CA-C5AB0DC85B11cCs4t|tƒr| d¡}nt|tƒs(t|ƒ}d|}|S)znPack the message inside ``00`` and ``FF`` As per the dataframing section (5.3) for the websocket spec zutf-8u%sÿ)Ú isinstanceZunicodeÚencodeÚstr)ÚmessageÚpacked©r ú1lib/python3.7/site-packages/werkzeug/websocket.pyÚ pack_messages    r FcCs„|r t|ƒ}d|d@B}t|ƒ}|dkr8t d||¡}n:|dkrZ|dkrZt d|d|¡}n|dkrrt d|d |¡}||t|ƒd fS) zõ Encode a HyBi style WebSocket frame. Optional opcode: 0x0 - continuation 0x1 - text frame (base64 encode buf) 0x2 - binary frame (use raw buf) 0x8 - connection close 0x9 - ping 0xA - pong é€éé}z>BBiz>BBHé~z>BBQér)rÚlenÚstructÚpack)ÚbufÚopcodeÚbase64Úb1Z payload_lenÚheaderr r r Ú encode_hybis  rc Csddddddddddœ }t|ƒ}||d<||dkr8|St d|¡\}}|d@|d <|d @d ?|d <|d @d ?}|d @|d<|ddkr´d|d<||dkr |St d|¡\|d<n6|dd krêd|d<||dkrØ|St d|¡\|d<|d|d|d}||kr|S|||d<|r|||d|dd…|d<d}} |ddkrêt d||d|dd…¡d} |dd} d}x`tdt|ddƒƒD]F} t d|| d| | d| d…¡d} |t d| | A¡7}q W|ddrn|dd}|d} ||}d} xTtd|ƒD]F} t d|| | ¡d} t d||| ¡d} | t| | Aƒ7} q$W|| |d<n||d|d|…|d<|r¼|d dkr¼t |dƒ|d<|d dkr|ddkrìt d|d¡|d<|ddkr|ddd…|d <|S)!z$Decode HyBi style WebSocket packets.réN) ÚfinrÚmaskÚhlenÚlengthÚpayloadÚleftÚ close_codeÚ close_reasonr!rz>BBrrr érrrréz>xxHé z>xxQrÚzHr"ér#) rrZ unpack_fromZunpackZxrangeÚintrÚrangeÚchrr)rrÚfZblenrZb2Zhas_maskZfull_lenÚbÚcÚdataZof1ÚirÚlZof2r r r Ú decode_hybi1sz          $ , r6c@s$eZdZdd„Zdd„Zdd„ZdS)Ú WebSocketWSGIcCs ||_dS)N)Úhandler)Úselfr8r r r Ú__init__†szWebSocketWSGI.__init__cCsdS)Nr )r9Úwsr r r Ú verify_client‰szWebSocketWSGI.verify_clientc Csì| d¡ d¡dkr$|d ¡dks4|ddgƒgS|d}t||ƒ}d }|d }| d ¡}|rxt|ƒ}t|ƒd kr†|ddgƒgSg} | d ¡} g} | rÊx,|  d¡D]} |  ¡} | | kr¨|  | ¡q¨W| rà|dd  | ¡7}g} | d¡}g}|r*x0| d¡D]"}| ¡}|| kr| |¡qW|rB|dd  |¡7}|d| d¡| d¡||j t t |t ƒ ¡ƒf7}n|d| d¡| d¡|f7}| |¡y| |¡Wn8tjk ræ}z|dtjkrÖ‚Wdd}~XYnXtS)NZHTTP_CONNECTIONZUpgradeéÿÿÿÿZ HTTP_UPGRADEZ websocketz400 Bad Request)Z ConnectionÚclosezgunicorn.socketzKHTTP/1.1 101 Switching Protocols Upgrade: websocket Connection: Upgrade Z PATH_INFOZHTTP_SEC_WEBSOCKET_KEYéZHTTP_SEC_WEBSOCKET_PROTOCOLú,zSec-WebSocket-Protocol: %s z, ZHTTP_SEC_WEBSOCKET_EXTENSIONSzSec-WebSocket-Extensions: %s ztSec-WebSocket-Origin: %s Sec-WebSocket-Location: ws://%s%s Sec-WebSocket-Version: %s Sec-WebSocket-Accept: %s Z HTTP_ORIGINZ HTTP_HOSTz7WebSocket-Origin: %s WebSocket-Location: ws://%s%s r)ÚgetÚfindÚlowerÚ WebSocketrrÚsplitÚstripÚappendÚjoinÚversionrrÚWS_KEYZdigestÚsendallr8ÚsocketÚerrorÚerrnoZEPIPEZALREADY_HANDLED)r9ÚenvironZstart_responseÚsockr;Zhandshake_replyÚpathÚkeyZws_keyZ protocolsZ subprotocolsZ ws_protocolsÚsZextsÚ extensionsZ ws_extensionsZextÚer r r Ú__call__Œsj          zWebSocketWSGI.__call__N)Ú__name__Ú __module__Ú __qualname__r:r<rVr r r r r7„sr7c@s@eZdZddd„Zdd„Zdd„Zdd „Zdd d „Zd d„ZdS)rDéLc Cs\||_yt| d¡ƒ}Wnttfk r4d}YnX||_d|_d|_d|_t   ¡|_ dS)NZHTTP_SEC_WEBSOCKET_VERSIONrZFó) Ú_socketr-rAÚ ValueErrorÚ TypeErrorrIÚclosedZacceptedÚ_bufÚ collectionsÚdequeÚ_msgs)r9rPrOrIr r r r:Þs zWebSocket.__init__cCs"g}d}|j}x|r|jdkr„t|dd}|ddkr>PnD|ddkrTd |_Pn.| |d¡|d r~||d  d…}nd }qt|dƒ}|dkrØ| d ¡}|d kr¬P| |d|… dd¡¡||dd…}q|dkrt|dƒdksþtd|ƒ‚d |_Pqt d|ƒ‚qW||_|S)a0 Parses for messages in the buffer *buf*. It is assumed that the buffer contains the start character for a message, but that it may contain only part of the rest of the message. Returns an array of messages, and the buffer remainder that didn't contain any full messages.r)r$r+é F)rr Nrr+Tr!r[õÿr=r(zutf-8Úreplaceéÿz Unexpected closing handshake: %rz6Don't understand how to parse this type of message: %r) r`rIr6r_rGÚordrBÚdecodeÚAssertionErrorr])r9ÚmsgsZend_idxrÚframeZ frame_typer r r Ú_parse_messagesês:        zWebSocket._parse_messagescCs8|jdkr t|ddd\}}}nt|ƒ}|j |¡dS)aSend a message to the browser. *message* should be convertable to a string; unicode objects should be encodable as utf-8. Raises socket.error with errno of 32 (broken pipe) if the socket has already been closed by the client. )r$r+rdr(F)rrN)rIrr r\rK)r9rr ZlenheadZlentailr r r Úsends  zWebSocket.sendcCsZxN|jsN|jrdS|j d¡}|dkr*dS|j|7_| ¡}|j |¡qW|j ¡S)a!Waits for and deserializes messages. Returns a single message; the oldest not yet processed. If the client has already closed the connection, returns None. This is different from normal socket behavior because the empty string is a valid websocket message.Ni r')rcr_r\Zrecvr`rmÚextendÚpopleft)r9Zdeltarkr r r Úwait$s zWebSocket.waitFcCs„|jdkr<|js)r9r r r r>Ls zWebSocket.closeN)rZ)F) rWrXrYr:rmrnrqrur>r r r r rDÜs  -  rD)F)F)ÚrerNrLrrarrrZhashlibrrJr rr6Úobjectr7rDr r r r Ús   SX