o KK&hxX@sDddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZddlmZddlmZmZddlmZmZmZdd lmZdd lmZdd lmZmZmZdd l m!Z!m"Z"dd l#m$Z$m%Z%ddl&m'Z'ddl m(Z(m)Z)m*Z*ddl+m,Z,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2ddl3m4Z4gdZ5Gddde2Z6dddddddde!dddddddddddddedDdEZ7  dfdgdGdHZ8z%ddIl9m:Z:ddJl;m,Z<e:j=e:j=e:j>e:j>dKZ?ddLddLdKZ@dhdSdTZAWneBydhdUdTZAYnw didjdWdXZCdkdZd[ZDdddd\dld]d^ZEe d_ZFe d`e daeFfdbZGGdcddddZHdS)m) annotationsN)Sequence)AnyCallableLiteralTypeVarcast)ClientProtocol)Headers HeadersLike)InvalidProxyMessageInvalidProxyStatus ProxyError)ClientExtensionFactory) enable_client_permessage_deflate)build_authorization_basic build_hostvalidate_subprotocols) USER_AGENTResponse) CONNECTINGEvent) StreamReader) LoggerLikeOrigin Subprotocol)Proxy WebSocketURI get_proxy parse_proxy parse_uri) Connection)Deadline)connect unix_connectClientConnectioncs\eZdZdZdddddd"fddZdedfd#ddZd$fdd Zd%fd d! ZZ S)&r'a1 :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 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. 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.r/*int | None | tuple[int | None, int | None]returnNonecs*|t|_tj||||||ddS)Nr+) threadingr response_rcvdsuper__init__)selfr0r2r,r-r.r/ __class__I/var/www/html/venv/lib/python3.10/site-packages/websockets/sync/client.pyr:6s  zClientConnection.__init__Nadditional_headersHeadersLike | Noneuser_agent_header str | NonetimeoutcCs|jtd,|j|_|dur|jj||dur$|jjd||j|jWdn1s5wY|j |sDt d|jj durN|jj dS)z1 Perform the opening handshake. )expected_stateN User-Agentz.timed out while waiting for handshake response) send_contextrr2r%requestheadersupdate setdefault send_requestr8wait TimeoutError handshake_exc)r;r@rBrDr>r>r? handshakeKs   zClientConnection.handshakeeventrcs<|jdurt|ts J||_|jdSt|dS)z. Process one incoming event. N)response isinstancerr8setr9 process_event)r;rQr<r>r?rUgs zClientConnection.process_eventcs(z tW|jdS|jw)zI Read incoming data from the socket and process events. N)r9 recv_eventsr8rT)r;r<r>r?rVus zClientConnection.recv_events)r0r1r2r r,r3r-r3r.r3r/r4r5r6)r@rArBrCrDr3r5r6)rQrr5r6)r5r6) __name__ __module__ __qualname____doc__r:rrPrUrV __classcell__r>r>r<r?r's r'deflateTr)r(ir*)socksslserver_hostnameorigin extensions subprotocols compressionr@rBproxy proxy_sslproxy_server_hostname open_timeoutr,r-r.max_sizer/loggercreate_connectionuristrr]socket.socket | Noner^ssl_module.SSLContext | Noner_rCr` Origin | Nonera'Sequence[ClientExtensionFactory] | NonerbSequence[Subprotocol] | Nonercr@rArBrdstr | Literal[True] | Nonererfrgr3r,r-r.rh int | Noner/r4riLoggerLike | Nonerjtype[ClientConnection] | Nonekwargsrr5cKs.|durd|vr|d}tdtt|}|js"|dur"td|dd}|dd}|rH|dur<|dur|jdddkr|jdkr| durtdt|||f| | | d|}ntd|d|t j|j|jffi|}|d|s|t jt jd |jrK|durt}|dur%|j}||| dur9|j ||d}n t!|||d}t"t j |}|dt#||||||d}|||||||d}Wnt$yr|durq|%wz |&|| |Wnt$y|'|j()w|*|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. 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:// proxyrBr^r_zunsupported proxyrDr_)r`rarbrhrir+)+popwarningswarnDeprecationWarningr!secure ValueErrorrrrr$r'r0AF_UNIX SOCK_STREAM settimeoutrDr%r schemeconnect_socks_proxyconnect_http_proxyAssertionErrorrKrjhostport setsockopt IPPROTO_TCP TCP_NODELAY ssl_modulecreate_default_context wrap_socket SSLSSLSocketrr ExceptioncloserP close_socketrecv_events_threadjoinstart_keepalive)rkr]r^r_r`rarbrcr@rBrdrerfrgr,r-r.rhr/rirjrvws_urirxrydeadline proxy_parsedsock_2r2 connectionr>r>r?r%sn                    r%rycKsD|dur|ddur|ddurd}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/``. Nr^rwzws://localhost/zwss://localhost/T)rkrxryr>)getr%)ryrkrvr>r>r?r&s r&) ProxyType)r)socks5hsocks5socks4asocks4Frrrrr$r1c Kstt|j|j|j|j|jt|j}|d| z |j |j|jfi|WSt t t j fy4tyD}ztd|d}~ww)z0Connect via a SOCKS proxy and return the socket.rDz failed to connect to SOCKS proxyN) SocksProxySOCKS_PROXY_TYPESrrrusernamepasswordSOCKS_PROXY_RDNSrKrDr%OSErrorrNr0rr)rdrrrv socks_proxyexcr>r>r?rs" rcKstd)Nz-python-socks is required to use a SOCKS proxy) ImportError)rdrrrvr>r>r?rsbytescCst|j|j|jdd}t}t|j|j|j|d<|dur!||d<|jdur6|jdus-Jt|j|j|d<d|d| S)NT)always_include_portHostrFzProxy-AuthorizationzCONNECT z HTTP/1.1 ) rrrrr rrrencode serialize)rdrrBrrIr>r>r?prepare_connect_requests rrc Cst}tj|j|j|jdd}zsz |||d}|r&| |n| t |qt yg}z-t |jts>J|j}d|jkrLdkr_nt||WYd}~W|dSt|d}~wtjyrtdty}ztd|d}~ww|dw) NF) include_bodyTii,z(timed out while connecting to HTTP proxyz0did not receive a valid HTTP response from proxy)rrparse read_line read_exact read_to_eofrrDrecv feed_datafeed_eofnext StopIterationrSvalue status_coderr0rNrr )r]rreaderparserdatarrRr>r>r?read_connect_responsesH    rrcKs|d|tj|j|jffi|}|jdkr<|dur"t}|dur)|j}| ||j ||d}| d| t |||zt ||W|StyY|w)NrDrr)rKrDr0rjrrrrrrrsendallrrrr)rdrrrBr^r_rvr]r>r>r?rs$     rTF.)boundc@sTeZdZdZdZ d#d$d d Zd%ddZd&ddZd'ddZd(ddZ d)d!d"Z dS)*rzr Socket-like object providing TLS-in-TLS. Only methods that are used by websockets are implemented. iNr]r1rwssl_module.SSLContextr_rCr5r6cCsBt|_t|_||_|j|j|j|d|_||jjdS)Nr) r MemoryBIOincomingoutgoing ssl_socketwrap_bio ssl_objectrun_io do_handshake)r;r]rwr_r>r>r?r:Es  zSSLSSLSocket.__init__funcCallable[..., T]argsrrcGs d}d}z||}Wntjyd}Yn tjy"d}Ynw|j}|r0|j||rH|j|j}|rB|j |n|j q|rKq|S)NTF) rSSLWantReadErrorSSLWantWriteErrorrreadrrr recv_bufsizerwrite write_eof)r;rr want_read want_writeresultrr>r>r?rUs,    zSSLSSLSocket.run_iobuflenintrcCs*z ||jj|WStjyYdSw)N)rrrr SSLEOFError)r;rr>r>r?rss zSSLSSLSocket.recvrcCs||jj|SN)rrr)r;rr>r>r?sendyszSSLSSLSocket.sendc Csd}t|=}|d}t|}||kr$||||d7}||ksWdn1s.wYWddSWddS1sFwYdS)NrB) memoryviewrlenr)r;rcountview byte_viewamountr>r>r?r|sPzSSLSSLSocket.sendallnamerlcCs t|j|Sr)getattrr)r;rr>r>r? __getattr__s zSSLSSLSocket.__getattr__r)r]r1rwrr_rCr5r6)rrrrr5r)rrr5r)rrr5r)rrr5r6)rrlr5r) rWrXrYrZrr:rrrrrr>r>r>r?r;s     r).rkrlr]rmr^rnr_rCr`rorarprbrqrcrCr@rArBrCrdrrrernrfrCrgr3r,r3r-r3r.r3rhrsr/r4rirtrjrurvrr5r')NN)ryrCrkrCrvrr5r') rdrrrrr$rvrr5r1r)rdrrrrBrCr5r)r]r1rr$r5r)rdrrrrr$rBrCr^rnr_rCrvrr5r1)I __future__rr0r^rr7rcollections.abcrtypingrrrrrclientr datastructuresr r exceptionsr rrextensions.baserextensions.permessage_deflaterrIrrrhttp11rrr2rrstreamsrrrrrkrrrr r!rr#utilsr$__all__r'r%r& python_socksrpython_socks.syncrSOCKS5SOCKS4rrrrrrrrrrr>r>r>r?s        i      '&