B 3\(@sdZddlZddlZddlZddlZddlZyddlmZWn ek r\ddlmZYnXddl Z ddl m Z m Z m Z ddl mZddlmZmZddlmZddlmZmZdd lmZd d lmZd d ZddZdZGdddeZGdddeeZ Gddde eZ!dS)z/Tornado handlers for WebSocket <-> ZMQ sockets.N)urlparse)genioloopweb)StreamClosedError)WebSocketHandlerWebSocketClosedError)Session) date_default extract_dates) cast_unicode)IPythonHandlercCs|}t|d}tjdkr.dd|D}tj|tdd}| d|t |}d|d g}x*|d d D]}| |d t |qrWt j d d |d |f|}| d|d|S)a2serialize a message as a binary blob Header: 4 bytes: number of msg parts (nbufs) as 32b int 4 * nbufs bytes: offset for each buffer as integer as 32b int Offsets are from the start of the buffer, including the header. Returns ------- The message serialized to bytes. buffers)cSsg|] }|qS)tobytes).0xrr8lib/python3.7/site-packages/notebook/base/zmqhandlers.py 1sz,serialize_binary_message..)defaultutf8rrr N!I)copylistpopsys version_infojsondumpsr encodeinsertlenappendstructZpackjoin)msgrbmsgnbufsoffsetsbufZ offsets_bufrrrserialize_binary_messages   r0c Cstd|ddd}ttdd||dd|d}|dg}x8t|dd|ddD]\}}||||qjWt|dd }t|d |d <t|d |d <|dd|d <|S) a'deserialize a message from a binary blog Header: 4 bytes: number of msg parts (nbufs) as 32b int 4 * nbufs bytes: offset for each buffer as integer as 32b int Offsets are from the start of the buffer, including the header. Returns ------- message dictionary z!iNrrrrr rrheaderZ parent_headerr) r)Zunpackrr(zipr#loadsdecoder )r,r-r.Zbufsstartstopr+rrrdeserialize_binary_message=s( $r7i0ucsjeZdZdZdZdZdZdZeddZ eddZ ddd Z d d Z fd d Z ddZddZZS)WebSocketMixinz"Mixin for common websocket optionsNrcCs|jdtS)zqThe interval for websocket keep-alive pings. Set ws_ping_interval = 0 to disable pings. Zws_ping_interval)settingsgetWS_PING_INTERVAL)selfrrr ping_intervalcszWebSocketMixin.ping_intervalcCs|jdtd|jtS)zIf no ping is received in this many milliseconds, close the websocket connection (VPNs, etc. can fail to cleanly close ws connections). Default is max of 3 pings or 30 seconds. Zws_ping_timeoutr)r9r:maxr=r;)r<rrr ping_timeoutkszWebSocketMixin.ping_timeoutcCs|jdkst|dr |r dS|jjd}|dkr>|}|dksN|dkrRdS|}t|j }||krpdS|jr|j|k}n|j rt |j |}nd}|s|j d|||S)zCheck Origin == Host or Access-Control-Allow-Origin. Tornado >= 4 calls this method automatically, raising 403 if it returns False. *skip_check_originTZHostNFz>Blocking Cross Origin WebSocket Attempt. Origin: %s, Host: %s)Z allow_originhasattrrArequestZheadersr:Z get_originlowerrZnetlocZallow_origin_patboolmatchlogwarning)r<originZhostZ origin_hostZallowrrr check_originus*   zWebSocketMixin.check_origincOsdS)zmeaningless for websocketsNr)r<argskwargsrrr clear_cookieszWebSocketMixin.clear_cookiecsf|jd|jj|jdkrTtj}||_ |j |_ t |j |j|_ |j tt|j||S)NzOpening websocket %sr)rGdebugrCpathr=rIOLoopcurrenttime last_ping last_pongZPeriodicCallback send_ping ping_callbackr5superr8open)r<rKrLZloop) __class__rrrXs    zWebSocketMixin.openc Cs|jdkr"|jdk r"|jdStj}d||j}d||j}|d|j kr~||j kr~|j d|| dSy|dWn"ttfk r|jdSX||_dS)z'send a ping to keep the websocket aliveNg@@z#WebSocket ping timeout after %i ms.r) ws_connectionrVr6rrPrQrRrTrSr=r?rGrHcloseZpingrr)r<ZnowZsince_last_pongZsince_last_pingrrrrUs   zWebSocketMixin.send_pingcCstj|_dS)N)rrPrQrRrT)r<datarrron_pongszWebSocketMixin.on_pong)N)__name__ __module__ __qualname____doc__rVrSrTstreampropertyr=r?rJrMrXrUr^ __classcell__rr)rYrr8\s  ' r8cs8eZdZejdkrfddZd ddZddZZS) ZMQStreamHandler)rr cs,|jdkrtt|j||n |jdS)N)rcrWr send_errorr\)r<rKrL)rYrrrgs zZMQStreamHandler.send_errorNcCsft|tr|}n|j|\}}|j|}|r8||d<|drLt|}|Stj|td}t |SdS)aReserialize a reply message using JSON. msg_or_list can be an already-deserialized msg dict or the zmq buffer list. If it is the zmq list, it will be deserialized with self.session. This takes the msg list from the ZMQ socket and serializes the result for the websocket. This method should be used by self._on_zmq_reply to build messages that can be sent back to the browser. channelr)rN) isinstancedictsessionZfeed_identitiesZ deserializer0r#r$r r )r<Z msg_or_listrhr+Zidentsmsg_listr/Zsmsgrrr_reserialize_replys  z#ZMQStreamHandler._reserialize_replyc Cs|jdks|r*|jd|dSt|dd}y|j||d}Wn(tk rp|jjd|dddSXy|j |t |t dWn,t t fk r|jd|dSXdS)Nz%zmq message arrived on closed channelrh)rhzMalformed message: %rT)exc_info)Zbinary)r[closedrGrHr\getattrrm ExceptionZcriticalZ write_messageribytesrr)r<rcrlrhr+rrr _on_zmq_replys    zZMQStreamHandler._on_zmq_reply)N) r_r`ratornador"rgrmrsrerr)rYrrfs  rfcsBeZdZddZddZejfddZddZd d Z Z S) AuthenticatedZMQStreamHandlercCsdS)zpUndo the set_default_headers in IPythonHandler which doesn't make sense for websockets Nr)r<rrrset_default_headers sz1AuthenticatedZMQStreamHandler.set_default_headerscCsR|dkr"|jdtd|ddrBt|d|j_n |jddS)zRun before finishing the GET request Extend this method to add logic that should fire before the websocket finishes completing. Nz*Couldn't authenticate WebSocket connectioniZ session_idFzNo session ID specified)Zget_current_userrGrHrZ HTTPErrorZ get_argumentr rk)r<rrrpre_gets     z%AuthenticatedZMQStreamHandler.pre_getc/s6|}t|Vtt|j||}t|VdS)N)rwrZ maybe_futurerWrur:)r<rKrLZres)rYrrr:"s z!AuthenticatedZMQStreamHandler.getcCs$|jd|jjt|jd|_dS)Nz$Initializing websocket connection %s)config)rGrNrCrOr rxrk)r<rrr initialize+sz(AuthenticatedZMQStreamHandler.initializecCs|jddS)NZwebsocket_compression_options)r9r:)r<rrrget_compression_options/sz5AuthenticatedZMQStreamHandler.get_compression_options) r_r`rarvrwr coroutiner:ryrzrerr)rYrru s  ru)"rbosr#r)warningsr!Z urllib.parser ImportErrorrtrrrZtornado.iostreamrZtornado.websocketrrZjupyter_client.sessionr Zjupyter_client.jsonutilr r Zipython_genutils.py3compatr Zhandlersrr0r7r;objectr8rfrurrrrs.     n?