o &gN @stdZddlZddlmZmZddlmZmZmZGdddeZ GdddZ Gd d d e eZ Gd d d eZ dS) z Python WebSocket server base Copyright 2011 Joel Martin Copyright 2016-2018 Pierre Ossman Licensed under LGPL version 3 (see docs/LICENSE.LGPL-3) N)BaseHTTPRequestHandler HTTPServer) WebSocketWebSocketWantReadErrorWebSocketWantWriteErrorcs:eZdZdZfddZd ddZddZd d ZZS) HttpWebSocketz?Class to glue websocket and http request functionality togethercst||_dSN)super__init__request_handler)selfr  __class__Y/root/parts/websockify/install/lib/python3.10/site-packages/websockify/websocketserver.pyr s  zHttpWebSocket.__init__NcC|j||dSr)r send_response)r codemessagerrrrzHttpWebSocket.send_responsecCrr)r send_header)r keywordvaluerrrrrzHttpWebSocket.send_headercCs|jdSr)r end_headersr rrrrszHttpWebSocket.end_headersr) __name__ __module__ __qualname____doc__r rrr __classcell__rrr rrs   rcs<eZdZdZeZfddZddZddZdd Z Z S) WebSocketRequestHandlerMixInaWebSocket request handler mix-in class This class modifies and existing request handler to handle WebSocket requests. The request handler will continue to function as before, except that WebSocket requests are intercepted and the methods handle_upgrade() and handle_websocket() are called. The standard do_GET() will be called for normal requests. The class instance SocketClass can be overridden with the class to use for the WebSocket connection. cs4|j|_|j|_z tW|j|_dS|j|_w)zExtended request handler This is where WebSocketRequestHandler redirects requests to the new methods. Any sub-classes must call this method in order for the calls to function. N)do_GET _real_do_GET_websocket_do_GETr handle_one_requestrr rrr$/s  z/WebSocketRequestHandlerMixIn.handle_one_requestcCs@|j|_|jdr|jddkr|dS|dS)Nupgrade websocket)r"r!headersgetlowerhandle_upgraderrrrr#=s    z.WebSocketRequestHandlerMixIn._websocket_do_GETcCsh||}z ||j|jWnty'td}|dt|YdSw||_d|_ | dS)a1Initial handler for a WebSocket request This method is called when a WebSocket is requested. By default it will create a WebSocket object and perform the negotiation. The WebSocket object will then replace the request object and handle_websocket() will be called. iNT) SocketClassacceptrequestr' Exceptionsysexc_info send_errorstrclose_connectionhandle_websocket)r r&excrrrr*Gs    z+WebSocketRequestHandlerMixIn.handle_upgradecCsdS)zHandle a WebSocket connection. This is called when the WebSocket is ready to be used. A sub-class should perform the necessary communication here and return once done. Nrrrrrr5^sz-WebSocketRequestHandlerMixIn.handle_websocket) rrrrrr,r$r#r*r5rrrr rr s  r c@ eZdZdS)WebSocketRequestHandlerNrrrrrrrr8isr8c@r7)WebSocketServerNr9rrrrr:msr:) rr0 http.serverrrwebsockify.websocketrrrrr r8r:rrrrs I