tni\VddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZddlmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZmZdd l m!Z!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)ddl m*Z*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4gdZ5Gdde2Z6ddddddddeddddddddddddd]dDZ7 d^d_dFZ8 ddGl9m:Z:ddHl;m$Z<e:j=e:j=e:j>e:j>dIZ?ddJddJdIZ@d`dQZAn#eB$rd`dRZAYnwxYwdadTZCddddUdbdVZDe dWZEe dXe dYeEfZZFGd[d\ZGdS)c) annotationsN)Sequence)AnyCallableLiteralTypeVarcast)ClientProtocol) HeadersLike)InvalidProxyMessageInvalidProxyStatus ProxyError)ClientExtensionFactory) enable_client_permessage_deflate)validate_subprotocols) USER_AGENTResponse) CONNECTINGEvent)Proxy get_proxy parse_proxyprepare_connect_request) StreamReader) BytesLike LoggerLikeOrigin Subprotocol) WebSocketURI parse_uri) Connection)Deadline)connect unix_connectClientConnectioncXeZdZdZddddddfdZdedfddZd fd Zd!fd ZxZ S)"r'a+ :mod:`threading` implementation of a WebSocket client connection. :class:`ClientConnection` 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 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. The ``ping_interval``, ``ping_timeout``, ``close_timeout``, and ``max_queue`` arguments have the same meaning as in :func:`connect`. Args: socket: Socket connected to a WebSocket server. protocol: Sans-I/O connection.   ping_interval ping_timeout close_timeout max_queuesocket socket.socketprotocolr r- float | Noner.r/r0*int | None | tuple[int | None, int | None]returnNonec|tj|_t||||||dS)Nr,) threadingr response_rcvdsuper__init__)selfr1r3r-r.r/r0 __class__s S/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/websockets/sync/client.pyr<zClientConnection.__init__7sX %&_..   '%'      Nadditional_headersHeadersLike | Noneuser_agent_header str | Nonetimeoutc|t5|j|_||jj|| |jjd||j|jdddn #1swxYwY|j |std|jj |jj dS)z1 Perform the opening handshake. )expected_stateNz User-Agentz.timed out while waiting for handshake response) send_contextrr3r%requestheadersupdate setdefault send_requestr:wait TimeoutError handshake_exc)r=rArCrEs r? handshakezClientConnection.handshakeLs*  j  9 9 5 5=0022DL!- $++,>??? , $// >OPPP M & &t| 4 4 4  5 5 5 5 5 5 5 5 5 5 5 5 5 5 5!&&w// QOPP P = & 2-- - 3 2sBB))B-0B-eventrc|j9t|tsJ||_|jdSt |dS)z. Process one incoming event. N)response isinstancerr:setr; process_event)r=rRr>s r?rWzClientConnection.process_eventhse = eX.. . . .!DM   " " $ $ $ $ $ GG ! !% ( ( ( ( (r@c t|jdS#|jwxYw)zI Read incoming data from the socket and process events. N)r; recv_eventsr:rV)r=r>s r?rYzClientConnection.recv_eventsvsV  % GG   ! ! !   " " $ $ $ $ $D  " " $ $ $ $s >A)r1r2r3r r-r4r.r4r/r4r0r5r6r7)rArBrCrDrEr4r6r7)rRrr6r7)r6r7) __name__ __module__ __qualname____doc__r<rrQrWrY __classcell__)r>s@r?r'r's<')%'&(@B        .26(2 $ .....8 ) ) ) ) ) ) % % % % % % % % % %r@r'deflateTr*r)ir+)socksslserver_hostnameorigin extensions subprotocols compressionrArCproxy proxy_sslproxy_server_hostname open_timeoutr-r.r/max_sizer0loggercreate_connectionuristrr`socket.socket | Nonerassl_module.SSLContext | NonerbrDrc Origin | Nonerd'Sequence[ClientExtensionFactory] | NonereSequence[Subprotocol] | NonerfrArBrCrgstr | Literal[True] | Nonerhrirjr4r-r.r/rkr5r0rlLoggerLike | Nonermtype[ClientConnection] | Nonekwargsrr6c J|3d|vr/|d}tjdtt |}|js|t d|dd}|dd}|r&||t d||t d |t||d krt|}n|t d ||rd} |d} | d urt|} t| }|t} |e|rjtj tj tj}|||J||n| t%| }|jdd dkr(t)||||dd}n|jdddkr0|jdkr| t dt+|||f| | | d|}nUt-d|d|tj|j|jffi|}|d|s+|tjtjd |jr|t=j}||j}||| | ||}n,tC|||}tEtj |}|dtG||||||}|||||||}n%#tH$r||%wxYw |&|| |n<#tH$r/|'|j()wxYw|*|S)a Connect to the WebSocket server at ``uri``. This function returns a :class:`ClientConnection` instance, which you can use to send and receive messages. :func:`connect` may be used as a context manager:: from websockets.sync.client import connect with connect(...) as websocket: ... The connection is closed automatically when exiting the context. Args: uri: URI of the WebSocket server. sock: Preexisting TCP socket. ``sock`` overrides the host and port from ``uri``. You may call :func:`socket.create_connection` to create a suitable TCP socket. ssl: Configuration for enabling TLS on the connection. server_hostname: Host name for the TLS handshake. ``server_hostname`` overrides the host name from ``uri``. origin: Value of the ``Origin`` header, for servers that require it. 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. 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. additional_headers (HeadersLike | None): Arbitrary HTTP headers to add to the handshake request. user_agent_header: Value of the ``User-Agent`` request header. It defaults to ``"Python/x.y.z websockets/X.Y"``. Setting it to :obj:`None` removes the header. proxy: If a proxy is configured, it is used by default. Set ``proxy`` to :obj:`None` to disable the proxy or to the address of a proxy to override the system configuration. See the :doc:`proxy docs <../../topics/proxies>` for details. proxy_ssl: Configuration for enabling TLS on the proxy connection. proxy_server_hostname: Host name for the TLS handshake with the proxy. ``proxy_server_hostname`` overrides the host name from ``proxy``. open_timeout: Timeout for opening the connection in seconds. :obj:`None` disables the timeout. ping_interval: Interval between keepalive pings in seconds. :obj:`None` disables keepalive. ping_timeout: Timeout for keepalive pings in seconds. :obj:`None` disables timeouts. close_timeout: Timeout for closing the connection in seconds. :obj:`None` disables the timeout. max_size: Maximum size of incoming messages in bytes. :obj:`None` disables the limit. You may pass a ``(max_message_size, max_fragment_size)`` tuple to set different limits for messages and fragments when you expect long messages sent in short fragments. max_queue: High-water mark of the buffer where frames are received. It defaults to 16 frames. The low-water mark defaults to ``max_queue // 4``. You may pass a ``(high, low)`` tuple to set the high-water and low-water marks. If you want to disable flow control entirely, you may set it to ``None``, although that's a bad idea. logger: Logger for this client. It defaults to ``logging.getLogger("websockets.client")``. See the :doc:`logging guide <../../topics/logging>` for details. create_connection: Factory for the :class:`ClientConnection` managing the connection. Set it to a wrapper or a subclass to customize connection handling. Any other keyword arguments are passed to :func:`~socket.create_connection`. Raises: InvalidURI: If ``uri`` isn't a valid WebSocket URI. OSError: If the TCP connection fails. InvalidHandshake: If the opening handshake fails. TimeoutError: If the opening handshake times out. N ssl_contextzssl_context was renamed to sslz-ssl argument is incompatible with a ws:// URIunixFpathzmissing path argumentz(path and sock arguments are incompatibler_zunsupported compression: Tsockssource_address) local_addrhttphttpsz8proxy_ssl argument is incompatible with an http:// proxyrCrarbzunsupported proxyrErb)rcrdrerkrlr,)+popwarningswarnDeprecationWarningr!secure ValueErrorrrrr$r'r1AF_UNIX SOCK_STREAM settimeoutrEr%rschemeconnect_socks_proxyconnect_http_proxyAssertionErrorrLrmhostport setsockopt IPPROTO_TCP TCP_NODELAY ssl_modulecreate_default_context wrap_socket SSLSSLSocketr r ExceptioncloserQ close_socketrecv_events_threadjoinstart_keepalive)rnr`rarbrcrdrerfrArCrgrhrirjr-r.r/rkr0rlrmrxws_urir{r|deadline proxy_parsedsock_2r3 connections r?r%r%s` {}..jj'' ,    s^^F =JS_HIIIFE**Dzz&$//D I #E#E DD"(!,66#*g55):O(V.$ +<%(= !DD))<===!!)X-=-=-?-?@@@/[&+. OOD ! ! ! J OOF.0BD I I I = "{ 799&"(+ OOH,,.. / / / t_MM%dCQQQFM622 OOD ! ! !" !%    '&  '%'        JJLLL             !!!%**,,,     s I"N"N%))O9P r|c |/|d|dd}nd}td|d|d|S) a Connect to a WebSocket server listening on a Unix socket. This function accepts the same keyword arguments as :func:`connect`. It's only available on Unix. It's mainly useful for debugging servers listening on Unix sockets. Args: path: File system path to the Unix socket. uri: URI of the WebSocket server. ``uri`` defaults to ``ws://localhost/`` or, when a ``ssl`` is provided, to ``wss://localhost/``. Nrarzzws://localhost/zwss://localhost/T)rnr{r|)getr%)r|rnrxs r?r&r&sU* { ::e   $M)B)B)J#CC$C  ;sD ; ;F ; ;;r@) ProxyType)r)socks5hsocks5socks4asocks4Frrr rr$r2c tt|j|j|j|j|jt|j}|d|  |j |j|jfi|S#tttj f$rt$r}td|d}~wwxYw)z0Connect via a SOCKS proxy and return the socket.rEz failed to connect to SOCKS proxyN) SocksProxySOCKS_PROXY_TYPESrrrusernamepasswordSOCKS_PROXY_RDNSrLrEr%OSErrorrOr1rr)rgrrrx socks_proxyexcs r?rrs! el + J J N N U\ *    )X%5%5%7%7888 J&;&v{FKJJ6JJ Jv~6     J J J?@@c I Js2B &C1CCc  td)Nz6connecting through a SOCKS proxy requires python-socks) ImportError)rgrrrxs r?rrs RSSSr@rct}tj|j|j|jd} |||d}|r| |n| t|x#t$rh}t|jtsJ|j}d|jcxkrdkrnn|cYd}~|dSt!|d}~wt"j$rt%dt&$r}t)d|d}~wwxYw#|dwxYw)NT)rgii,z(timed out while connecting to HTTP proxyz0did not receive a valid HTTP response from proxy)rrparse read_line read_exact read_to_eofrrErecv feed_datafeed_eofnext StopIterationrUvalue status_coderr1rOrr )r`rreaderparserdatarrTs r?read_connect_responsers ^^F ^ F   OOH,,.. / / /99T??D "  &&&&!!! LLL  ///#)X.....9 (& , , , , , , , , ,OOOOO %X.. . >GGGEFFF ! >    sBA9B00 E:9D3E4ED*EEEEE6rc <|d|tj|j|jffi|}|jdkrq|tj}||j}| || ||}| d| t||| t||n##t$r|wxYw|S)NrErr)rLrEr1rmrrrrrrrsendallrrrr)rgrrrCrarbrxr`s r?rrs' i!1!1!3!3444  #UZ$< G G G GD |w ;355C  "#jO ((**+++t_EE  LL(8IJJKKKdH----    Ks (C99 DTF.)boundcJeZdZdZdZ ddd Zd dZd!dZd"dZd#dZ d$dZ dS)%rzr Socket-like object providing TLS-in-TLS. Only methods that are used by websockets are implemented. iNr`r2rzssl_module.SSLContextrbrDr6r7ctj|_tj|_||_||j|j||_||jjdS)Nr) r MemoryBIOincomingoutgoing ssl_socketwrap_bio ssl_objectrun_io do_handshake)r=r`rzrbs r?r<zSSLSSLSocket.__init__6ss #,.. ",.. %.. M M+/   DO011111r@funcCallable[..., T]argsrrc d}d} ||}n)#tj$rd}Yntj$rd}YnwxYw|j}|r|j||rV|j|j}|r|j |n|j |r|S)NTF) rSSLWantReadErrorSSLWantWriteErrorrreadrrr recv_bufsizerwrite write_eof)r=rr want_read want_writeresultrs r?rzSSLSSLSocket.run_ioFs IJ "t. ! ! ! / " " "!  "=%%''D .''--- ++D,=>>.M''----M++--- Ms 333buflenintbytescp ||jj|S#tj$rYdSwxYw)Nr@)rrrr SSLEOFError)r=rs r?rzSSLSSLSocket.recvdsE ;;t3V<< <%   33 s "55rrcB||jj|SN)rrr)r=rs r?sendzSSLSSLSocket.sendjs{{4?0$777r@c*d}t|5}|d5}t|}||kr&||||dz }||k&dddn #1swxYwYddddS#1swxYwYdS)NrB) memoryviewr lenr)r=rcountview byte_viewamounts r?rzSSLSSLSocket.sendallms    6tyy~~ 6^^F&..9UVV#4555&.. 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s4Br s"""""" $$$$$$88888888888888######((((((LLLLLLLLLL444444LLLLLL++++++))))))))((((((((JJJJJJJJJJJJ""""""????????????))))))))"""""" : 9 9b%b%b%b%b%zb%b%b%R"&(,"& :>15'-1$.(,.2(,!#"$!#"$;@<> $7;;FFFFFFT<<<<<<4J&&&&&&555555#"#"  JJJJJ3TTTTTTTTTTfN%)(,"&######L GCLL GCxQ'(((M.M.M.M.M.M.M.M.M.M.s C<< DD