RPfHddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z mZmZmZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZm Z m!Z!dd l"m#Z#ddl m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*gdZ+Gdde(Z,GddZ- dddddddddddeddddddd ddZ. d d!dZ/y)") annotationsN) TracebackType)AnyCallableOptionalSequenceType) CloseCode)ServerExtensionFactory) enable_server_permessage_deflate)validate_subprotocols) USER_AGENT)RequestResponse) CONNECTINGOPENEvent)ServerProtocol) LoggerLikeOrigin Subprotocol) Connection)Deadline)serve unix_serveServerConnectionWebSocketServercteZdZdZdd d fdZddedf d dZd fd Zd fd ZxZ S) ra Threaded implementation of a WebSocket server connection. :class:`ServerConnection` provides :meth:`recv` and :meth:`send` methods for receiving and sending messages. It supports iteration to receive messages:: for message in websocket: process(message) The iterator exits normally when the connection is closed with close code 1000 (OK) or 1001 (going away) or without a close code. It raises a :exc:`~websockets.exceptions.ConnectionClosedError` when the connection is closed with any other code. Args: socket: Socket connected to a WebSocket client. protocol: Sans-I/O connection. close_timeout: Timeout for closing the connection in seconds.  close_timeoutc`|tj|_t||||y)Nr") threadingr request_rcvdsuper__init__)selfsocketprotocolr# __class__s P/var/www/fastapitest/venv/lib/python3.12/site-packages/websockets/sync/server.pyr(zServerConnection.__init__7s2 %OO-   '  Nc*|jj|s5|j|jj t d|j 5|j|jj td|jt5d|_ | |||j |_ |j*|jj)|j |_ |||jj*d <|) |||j |j}|||_ |jj-|jddd|jj.t0urO|jj |j2|j|jj |jj|jjy#t$rr}||j_ |jjdd|jj!t"j$j&d|_ Yd}~d}~wwxYw#t$rr}||j_ |jjdd|jj!t"j$j&d|_ Yd}~d}~wwxYw#1swYxYw) z1 Perform the opening handshake. ztimed out during handshakeNz"connection closed during handshake)expected_statezopening handshake failedTexc_infozLFailed to open a WebSocket connection. See server log for more information. Server)r&wait close_socketrecv_events_threadjoin TimeoutErrorrequestConnectionError send_contextrresponse Exceptionr+ handshake_excloggererrorrejecthttp HTTPStatusINTERNAL_SERVER_ERRORacceptheaders send_responsestaterr#)r)process_requestprocess_response server_headertimeoutexcr<s r- handshakezServerConnection.handshakeFs{*  %%g.      # # ( ( *;< < <<       # # ( ( *!"FG G   j  9( 7 DM* $3D$,,$GDM}}$ $ 4 4T\\ B (2? %%h/+1/dllDMMRH +(0 MM ' ' 6Q( 7T ==  d *  # # ( ();); <      # # ( ( * == & & 2---- - 3S! 25DMM/KK%%&@4%P$(MM$8$8==E%DMM (! 25DMM/KK%%&@4%P$(MM$8$8==E%DMM 5( 7( 7sc( L3H  ALJ =/L JA'J<LJL LA'L:LLLLc|j4t|tsJ||_|jj yt ||y)z. Process one incoming event. N)r9 isinstancerr&setr' process_event)r)eventr,s r-rRzServerConnection.process_eventsG << eW- -- DL    ! ! # G !% (r.c t||jjy#|jjwxYw)zI Read incoming data from the socket and process events. N)r' recv_eventsr&rQ)r)r,s r-rUzServerConnection.recv_eventss:  $ G  !    ! ! #D   ! ! #s ,A)r* socket.socketr+rr#Optional[float]returnNone) rICOptional[Callable[[ServerConnection, Request], Optional[Response]]]rJMOptional[Callable[[ServerConnection, Request, Response], Optional[Response]]]rK Optional[str]rLrWrXrY)rSrrXrYrXrY) __name__ __module__ __qualname____doc__r(rrNrRrU __classcell__)r,s@r-rrs8*,    !  '    ,  '1#'O. O. O.%O.!O. !O.b ) $ $r.rc^eZdZdZ d d dZd dZd dZd dZd dZ ddZ y)ra WebSocket server returned by :func:`serve`. This class mirrors the API of :class:`~socketserver.BaseServer`, notably the :meth:`~socketserver.BaseServer.serve_forever` and :meth:`~socketserver.BaseServer.shutdown` methods, as well as the context manager protocol. Args: socket: Server socket listening for new connections. handler: Handler for one connection. Receives the socket and address returned by :meth:`~socket.socket.accept`. logger: Logger for this server. Nc||_||_|tjd}||_t j dk7r"tj\|_ |_ yy)Nzwebsockets.serverwin32) r*handlerlogging getLoggerr?sysplatformospipeshutdown_watchershutdown_notifier)r)r*rfr?s r-r(zWebSocketServer.__init__sV   >&&':;F <<7 "<>GGI 9D !4#9 #r.ctj}|j|jtjt j dk7r*|j|jtj |j |jj\}}tj|j||f}|jb#t$rYywxYw)a See :meth:`socketserver.BaseServer.serve_forever`. This method doesn't return. Calling :meth:`shutdown` from another thread stops the server. Typical use:: with serve(...) as server: server.serve_forever() re)targetargsN) selectorsDefaultSelectorregisterr* EVENT_READrirjrmselectrEOSErrorr%Threadrfstart)r)pollersockaddrthreads r- serve_foreverzWebSocketServer.serve_forevers**, Y%9%9: <<7 " OOD1193G3G H MMO "[[//1 d%%T\\t MF LLN   sC C+*C+c|jjtjdk7r!t j |j dyy)z@ See :meth:`socketserver.BaseServer.shutdown`. rexN)r*closerirjrkwriternr)s r-shutdownzWebSocketServer.shutdowns9  <<7 " HHT++T 2 #r.c6|jjS)z> See :meth:`socketserver.BaseServer.fileno`. )r*filenors r-rzWebSocketServer.filenos {{!!##r.c|SNrs r- __enter__zWebSocketServer.__enter__s r.c$|jyr)r)r)exc_type exc_value tracebacks r-__exit__zWebSocketServer.__exit__s r.r)r*rVrfz$Callable[[socket.socket, Any], None]r?Optional[LoggerLike]r])rXint)rXr)rzOptional[Type[BaseException]]rzOptional[BaseException]rzOptional[TracebackType]rXrY) r^r_r`rar(r~rrrrrr.r-rrsp((, F F6 F% F:3$/++   r.rFdeflater!i)r{ ssl_contextunixpathorigins extensions subprotocolsselect_subprotocolrIrJrK compression open_timeoutr#max_sizer?create_connectionc  t |dk(r tn|td|t|Mr3| t dt j |t j}n%t j ||f}n | t dj|dd}d   fd }t||S) aH Create a WebSocket server listening on ``host`` and ``port``. Whenever a client connects, the server creates a :class:`ServerConnection`, performs the opening handshake, and delegates to the ``handler``. The handler receives a :class:`ServerConnection` instance, which you can use to send and receive messages. Once the handler completes, either normally or with an exception, the server performs the closing handshake and closes the connection. :class:`WebSocketServer` mirrors the API of :class:`~socketserver.BaseServer`. Treat it as a context manager to ensure that it will be closed and call the :meth:`~WebSocketServer.serve_forever` method to serve requests:: def handler(websocket): ... with websockets.sync.server.serve(handler, ...) as server: server.serve_forever() Args: handler: Connection handler. It receives the WebSocket connection, which is a :class:`ServerConnection`, in argument. host: Network interfaces the server binds to. See :func:`~socket.create_server` for details. port: TCP port the server listens on. See :func:`~socket.create_server` for details. sock: Preexisting TCP socket. ``sock`` replaces ``host`` and ``port``. You may call :func:`socket.create_server` to create a suitable TCP socket. ssl_context: Configuration for enabling TLS on the connection. origins: Acceptable values of the ``Origin`` header, for defending against Cross-Site WebSocket Hijacking attacks. Include :obj:`None` in the list if the lack of an origin is acceptable. extensions: List of supported extensions, in order in which they should be negotiated and run. subprotocols: List of supported subprotocols, in order of decreasing preference. select_subprotocol: Callback for selecting a subprotocol among those supported by the client and the server. It receives a :class:`ServerConnection` (not a :class:`~websockets.server.ServerProtocol`!) instance and a list of subprotocols offered by the client. Other than the first argument, it has the same behavior as the :meth:`ServerProtocol.select_subprotocol ` method. process_request: Intercept the request during the opening handshake. Return an HTTP response to force the response or :obj:`None` to continue normally. When you force an HTTP 101 Continue response, the handshake is successful. Else, the connection is aborted. process_response: Intercept the response during the opening handshake. Return an HTTP response to force the response or :obj:`None` to continue normally. When you force an HTTP 101 Continue response, the handshake is successful. Else, the connection is aborted. server_header: Value of the ``Server`` response header. It defaults to ``"Python/x.y.z websockets/X.Y"``. Setting it to :obj:`None` removes the header. compression: The "permessage-deflate" extension is enabled by default. Set ``compression`` to :obj:`None` to disable it. See the :doc:`compression guide <../../topics/compression>` for details. open_timeout: Timeout for opening connections in seconds. :obj:`None` disables the timeout. close_timeout: Timeout for closing connections in seconds. :obj:`None` disables the timeout. max_size: Maximum size of incoming messages in bytes. :obj:`None` disables the limit. logger: Logger for this server. It defaults to ``logging.getLogger("websockets.server")``. See the :doc:`logging guide <../../topics/logging>` for details. create_connection: Factory for the :class:`ServerConnection` managing the connection. Set it to a wrapper or a subclass to customize connection handling. rzunsupported compression: zmissing path argument)familyz(path and sock arguments are incompatibleTF) server_sidedo_handshake_on_connectc t } s/|jtjtjd\|j |j t|tjsJ|j|j dd} dfd }t |t }J||j|j   jy#t$r|jYywxYw#t$r?|j j#ddjt$j&YywxYw)NTc>J|jusJ|Sr)r+)r+r connectionrs r-protocol_select_subprotocolz@serve..conn_handler..protocol_select_subprotocols2 .999#z':'::::-j,GGr.)rrrrrHrr?r"zconnection handler failedr1)r+rrzSequence[Subprotocol]rXzOptional[Subprotocol])r setsockoptr* IPPROTO_TCP TCP_NODELAY settimeoutrLrPssl SSLSocket do_handshakerrrNr=rr?r@r INTERNAL_ERROR)r{r|deadlinerr+rr#rrrfr?rrrrIrJrrKrrrs @r- conn_handlerzserve..conn_handlersL)B  2 2F4F4FM& 0 0 23!$ 666!!#% ("-H,H"7H+H&%)#> !H%0 00*+J     "    J       JJL    7 OO ! !"= ! M   Y55 6 7s%C'D7D/D,+D,/AE76E7)r{rVr|rrXrY) rr ValueErrorr TypeErrorr* create_serverAF_UNIX wrap_socketr)rfhostportr{rrrrrrrrIrJrKrrr#rr?rrs` `` ``````` ````` r-rr svl+i5jA  4[MBCC , | | 788''V^^DD''t 5D  FG G&& $) ' PPPh 4v 66r.c  t|f|dd|S)a Create a WebSocket server listening on a Unix socket. This function is identical to :func:`serve`, except the ``host`` and ``port`` arguments are replaced by ``path``. It's only available on Unix. It's useful for deploying a server behind a reverse proxy such as nginx. Args: handler: Connection handler. It receives the WebSocket connection, which is a :class:`ServerConnection`, in argument. path: File system path to the Unix socket. T)rr)r)rfrkwargss r-rrs&  9t$ 9& 99r.)NN)*rfz"Callable[[ServerConnection], None]rr\r Optional[int]r{zOptional[socket.socket]rzOptional[ssl.SSLContext]rboolrr\rz$Optional[Sequence[Optional[Origin]]]rz*Optional[Sequence[ServerExtensionFactory]]rzOptional[Sequence[Subprotocol]]rzTOptional[Callable[[ServerConnection, Sequence[Subprotocol]], Optional[Subprotocol]]]rIrZrJr[rKr\rr\rrWr#rWrrr?rrz Optional[Type[ServerConnection]]rXrr)rfz!Callable[[ServerConnection], Any]rr\rrrXr)0 __future__rrBrgrkrrr*rrir%typesrtypingrrrrr websockets.framesr extensions.baser extensions.permessage_deflater rFrrhttp11rrr+rrrserverrrrrrrutilsr__all__rrrrrr.r-rs"   ::'4L+&..#44" IO$zO$dUUts7 %),048=A48 #-!*$&%'##':>Us7 /s7 s7 s7 " s7*s7 s7 s72s7;s72s7s7()s745s7@!As7BCs7F"Gs7H#Is7LMs7P !Qs7T8Us7VWs7p: .: :: :r.