o i'@sdZddlZddlZddlZddlmZddlZddlmZmZm Z ddl m Z ddl m Z mZddlmZz ddlmZmZWneyQdd lmZmZYnwdd lmZdd lmZd d lmZddZddZdZGdddZGdddee Z Gddde eZ!dS)z/Tornado handlers for WebSocket <-> ZMQ sockets.N)urlparse)genioloopweb)StreamClosedError)WebSocketHandlerWebSocketClosedError)Session) json_default extract_dates) date_defaultr ) cast_unicode) maybe_future)IPythonHandlercCs|}t|d}tj|tdd}|d|t|}d|dg}|ddD] }| |dt|q,t j d d |d|g|R}|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. buffersdefaultutf8rrN!I) copylistpopjsondumpsr encodeinsertlenappendstructpackjoin)msgrbmsgnbufsoffsetsbuf offsets_bufr,T/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/base/zmqhandlers.pyserialize_binary_messages    r.c Cstd|ddd}ttdd||dd|d}|dg}t|dd|ddD] \}}||||q4t|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!iNrrrrrrrheader parent_headerr) r#unpackrr"ziprloadsdecoder )r'r(r)bufsstartstopr&r,r,r-deserialize_binary_message;s( "r8i0ucsjeZdZdZdZdZdZdZeddZ eddZ ddd Z d d Z fd d Z ddZddZZS)WebSocketMixinz"Mixin for common websocket optionsNrcCs|jdtS)ziThe interval for websocket keep-alive pings. Set ws_ping_interval = 0 to disable pings. ws_ping_interval)settingsgetWS_PING_INTERVALselfr,r,r- ping_intervalaszWebSocketMixin.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. ws_ping_timeout)r;r<maxr@r=r>r,r,r- ping_timeoutiszWebSocketMixin.ping_timeoutcCs|jdkst|dr|rdS|jjd}|dur|}|dus'|dur)dS|}t|j }||kr8dS|jrA|j|k}n|j rMt |j |}nd}|sY|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_originTHostNFz>Blocking Cross Origin WebSocket Attempt. Origin: %s, Host: %s) allow_originhasattrrFrequestheadersr< get_originlowerrnetlocallow_origin_patboolmatchlogwarning)r?originhost origin_hostallowr,r,r- check_originss2   zWebSocketMixin.check_origincOdS)zmeaningless for websocketsNr,r?argskwargsr,r,r- clear_cookieszWebSocketMixin.clear_cookiecsf|jd|jj|jdkr*tj}||_ |j |_ t |j |j|_ |j tj|i|S)NzOpening websocket %sr)rRdebugrJpathr@rIOLoopcurrenttime last_ping last_pongPeriodicCallback send_ping ping_callbackr6superopen)r?r[r\loop __class__r,r-ris    zWebSocketMixin.openc Cs|jdur|jdur|jdStj}d||j}d||j}|d|j kr?||j kr?|j d|| dSz|dWnttfyW|jYdSw||_dS)z'send a ping to keep the websocket aliveNg@@z#WebSocket ping timeout after %i ms.r) ws_connectionrgr7rr`rarbrdrcr@rDrRrSclosepingrr)r?nowsince_last_pongsince_last_pingr,r,r-rfs"   zWebSocketMixin.send_pingcCstj|_dSN)rr`rarbrd)r?datar,r,r-on_pongszWebSocketMixin.on_pongrt)__name__ __module__ __qualname____doc__rgrcrdstreampropertyr@rDrXr]rirfrv __classcell__r,r,rkr-r9Zs   ' r9cs:eZdZejdkr fddZd ddZddZZS) ZMQStreamHandler)rrcs2|jdurtt|j|i|dS|jdSrt)r{rhr send_errorrorZrkr,r-rs zZMQStreamHandler.send_errorNcCsbt|tr|}n|j|\}}|j|}|r||d<|dr&t|}|Stj|td}t |S)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. channelrr) isinstancedictsessionfeed_identities deserializer.rrr r )r? msg_or_listrr&identsmsg_listr*smsgr,r,r-_reserialize_replys  z#ZMQStreamHandler._reserialize_replyc Cs|jdus |r|jd|dSt|dd}z |j||d}Wnty9|jjd|ddYdSwz |j |t |t dWdSt t fy]|jd|YdSw)Nz%zmq message arrived on closed channelr)rzMalformed message: T)exc_info)binary)rnclosedrRrSrogetattrr Exceptioncritical write_messagerbytesrr)r?r{rrr&r,r,r- _on_zmq_replys$    zZMQStreamHandler._on_zmq_replyrt) rwrxrytornado version_inforrrr}r,r,rkr-r~s    r~csBeZdZddZddZejfddZddZd d Z Z S) AuthenticatedZMQStreamHandlercCrY)zhUndo the set_default_headers in IPythonHandler which doesn't make sense for websockets Nr,r>r,r,r-set_default_headers sz1AuthenticatedZMQStreamHandler.set_default_headerscCsT|dur|jdtd|ddr"t|d|j_dS|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 connectioni session_idFzNo session ID specified)get_current_userrRrSr HTTPError get_argumentr rr>r,r,r-pre_gets    z%AuthenticatedZMQStreamHandler.pre_getc/s4|}t|Vtj|i|}t|VdSrt)rrrhr<)r?r[r\resrkr,r-r< s  z!AuthenticatedZMQStreamHandler.getcCs$|jd|jjt|jd|_dS)Nz$Initializing websocket connection %s)config)rRr^rJr_r rrr>r,r,r- initialize)sz(AuthenticatedZMQStreamHandler.initializecCs|jddS)Nwebsocket_compression_options)r;r<r>r,r,r-get_compression_options-sz5AuthenticatedZMQStreamHandler.get_compression_options) rwrxryrrr coroutiner<rrr}r,r,rkr-rsr)"rzrr#sys urllib.parserrrrrtornado.iostreamrtornado.websocketrrjupyter_client.sessionr jupyter_client.jsonutilr r ImportErrorr ipython_genutils.py3compatr notebook.utilsrhandlersrr.r8r=r9r~rr,r,r,r-s0       n?