o >hJ@sddlmZmZddlmZddlmZddlZeddl Z ddl m Z ddl mZddlmZmZmZddlmZdd lmZdd l mZdd lmZmZmZdd lmZmZdd lmZddl m!Z!ddl"m#Z#m$Z$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+m,Z,ddl-m.Z.m/Z/m0Z0m1Z1dZ2ddZ3ddZ4ddZ5ddZ6Gddde)Z7Gdd d e j8j'j9Z:eGd!d"d"e:e'j;Z;eGd#d$d$e:e'j<ZeGd'd(d(e>e'j?e j8j'j@Z?eGd)d*d*e>e'jAe j8j'jBZAeeGd+d,d,e=ZCGd-d.d.eCe;ZDGd/d0d0eCe<ZEGd1d2d2e?ZFGd3d4d4eAZGedDd6d7ZHedEd:d;ZIeGdd?d?ejKe?ZKeGd@dAdAejLe<ZLeGdBdCdCejMeAZMdS)F) b64encode b64decode)Optional) implementerN) endpoints) ITransport)ConnectionDoneConnectionAbortedConnectionLost)Deferred)Failure)connectionDone)publichltypehlval) _is_tls_error_maybe_tls_reason) websocket)TransportDetails)ConnectionRequestConnectionResponseConnectionDeny)protocol)IWebSocketClientAgent)create_transport_detailstransport_channel_id)PerMessageDeflateOfferPerMessageDeflateOfferAcceptPerMessageDeflateResponsePerMessageDeflateResponseAccept)create_client_agentWebSocketAdapterProtocolWebSocketServerProtocolWebSocketClientProtocolWebSocketAdapterFactoryWebSocketServerFactoryWebSocketClientFactoryWrappingWebSocketAdapterWrappingWebSocketServerProtocolWrappingWebSocketClientProtocolWrappingWebSocketServerFactoryWrappingWebSocketClientFactorylistenWS connectWSWampWebSocketServerProtocolWampWebSocketServerFactoryWampWebSocketClientProtocolWampWebSocketClientFactorycCst|S)zB :returns: an instance implementing IWebSocketClientAgent )_TwistedWebSocketClientAgentreactorr5z/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/autobahn/twisted/websocket.pyr Vsr cCs<t|tstdt||ds|dstddS)z> raises a ValueError if `transport_config` is invalid z-'transport_config' must be a string, found {}zws://zwss://z6'transport_config' must start with 'ws://' or 'wss://'N) isinstancestr ValueErrorformattype startswith)transport_configr5r5r6check_transport_config]s   r>cCsBt|ts tdgd}|D] }||vrtd|qdS)z5 raises a ValueError if `options` is invalid z'options' must be a dict)origin protocols useragentheadersproxyz'options' may not contain '{}'N)r7dictr9keysr:)options valid_keysactual_kr5r5r6check_client_optionsqs  rIcCs|jrddlm}||j}|jdur&||_t||jd|jd}|S|jr;ddlm}t ||j|j |}|St||j|j }|S)Nrsslhostport) isSecuretwisted.internetrKoptionsForClientTLSrLrCcontextFactoryrHostnameEndpointSSL4ClientEndpointrM)r4factoryr=rFrKcontext_factoryendpointr5r5r6_endpoint_from_configs4    rWc@s"eZdZdZddZdddZdS)r2z6 This agent creates connections using Twisted cCs ||_dSN)_reactor)selfr4r5r5r6__init__s z%_TwistedWebSocketClientAgent.__init__Ncst|t|td||jd|}|durtn||_t|j|||}t||}fddfdd}| |S)a- Open a new connection. :param dict transport_config: valid transport configuration :param dict options: additional options for the factory :param protocol_class: a callable that returns an instance of the protocol (WebSocketClientProtocol if the default None is passed in) :returns: a Deferred that fires with an instance of `protocol_class` (or WebSocketClientProtocol by default) that has successfully shaken hands (completed the handshake). )urlr4Ncs|dSrX)errback)f)rtn_dr5r6failedz1_TwistedWebSocketClientAgent.open..failedcs fdd}j|S)Ncs|SrX)callback)arg)protor_r5r6handshake_completeds zQ_TwistedWebSocketClientAgent.open..got_proto..handshake_completed)is_open addCallbacks)rdrer`r_)rdr6 got_protosz4_TwistedWebSocketClientAgent.open..got_protor5) r>rIr&rYr#rrWr connectrg)rZr=rFprotocol_classrTrVproto_drir5rhr6opens    z!_TwistedWebSocketClientAgent.openrX)__name__ __module__ __qualname____doc__r[rmr5r5r5r6r2sr2c@seZdZUdZeZdZee e d<dZ ee e d<ddZ efdefdd Zd efd d Zd*ddZddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Z dS)+r!a Adapter class for Twisted WebSocket client and server protocols. Called from Twisted: * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol.connectionMade` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol.connectionLost` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol.dataReceived` Called from Network-independent Code (WebSocket implementation): * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onOpen` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onMessageBegin` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onMessageFrameData` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onMessageFrameEnd` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onMessageEnd` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onMessage` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onPing` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onPong` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._onClose` FIXME: * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._closeConnection` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol._create_transport_details` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol.registerProducer` * :meth:`autobahn.twisted.websocket.WebSocketAdapterProtocol.unregisterProducer` Npeer is_servercCsjt|j|j|_tj|j_|jj|_z |j|j WnY| |j j dt |jt|jddS)Nz/{func} connection established for peer="{peer}"funcrr)r transportrs_transport_detailsrCHANNEL_FRAMING_WEBSOCKETchannel_framingrr setTcpNoDelay tcpNoDelay_connectionMadelogdebugrconnectionMaderrZr5r5r6r s   z'WebSocketAdapterProtocol.connectionMadereasoncCsd}t|jtr|jjd|jdd}nNt|jr#|jt|jn?t|jt r3|jjd|jdn/t|jt rSt |j}t |jdrH|jj }|jjd|j|dn|jjd |jt|j|jd |||rz|jjd t|jt|jd dS|jjd t|jt|j|ddS)NFz,Connection to/from {peer} was closed cleanly)rrTz-Connection to/from {peer} was aborted locallymessagezDConnection to/from {peer} was lost in a non-clean fashion: {message})rrrz7Connection to/from {peer} lost ({error_type}): {error}))rr error_typeerrorz8{func} connection lost for peer="{peer}", closed cleanlyrtzD{func} connection lost for peer="{peer}", closed with error {reason})rurrr)r7valuerr}r~rrrrrr r r8hasattrrr;_connectionLostrconnectionLostr)rZr was_cleanrr5r5r6r'sF         z'WebSocketAdapterProtocol.connectionLostdatacCs6|jjdt|jt|jtt|d||dS)Nz2{func} received {data_len} bytes for peer="{peer}")rurrdata_len)r}r~r dataReceivedrrrlen _dataReceivedrZrr5r5r6rUs  z%WebSocketAdapterProtocol.dataReceivedFcCs,|rt|jdr|jdS|jdS)NabortConnection)rrvrloseConnection)rZabortr5r5r6_closeConnection^sz)WebSocketAdapterProtocol._closeConnectioncCs2|jjrdt|j|jjdi}||j_|dS)Nz tls-unique)rw is_securerrvrs channel_idonOpen)rZrr5r5r6_onOpenes  z WebSocketAdapterProtocol._onOpencC||dSrX)onMessageBegin)rZisBinaryr5r5r6_onMessageBeginpraz(WebSocketAdapterProtocol._onMessageBegincCrrX)onMessageFrameBegin)rZlengthr5r5r6_onMessageFrameBeginsraz-WebSocketAdapterProtocol._onMessageFrameBegincCrrX)onMessageFrameDatarZpayloadr5r5r6_onMessageFrameDatavraz,WebSocketAdapterProtocol._onMessageFrameDatacC |dSrX)onMessageFrameEndrr5r5r6_onMessageFrameEndy z+WebSocketAdapterProtocol._onMessageFrameEndcCrrX)onMessageFramerr5r5r6_onMessageFrame|raz(WebSocketAdapterProtocol._onMessageFramecCrrX) onMessageEndrr5r5r6 _onMessageEndrz&WebSocketAdapterProtocol._onMessageEndcCs|||dSrX) onMessage)rZrrr5r5r6 _onMessagez#WebSocketAdapterProtocol._onMessagecCrrX)onPingrr5r5r6_onPingraz WebSocketAdapterProtocol._onPingcCrrX)onPongrr5r5r6_onPongraz WebSocketAdapterProtocol._onPongcCs||||dSrX)onCloserZwasCleancoderr5r5r6_onClosesz!WebSocketAdapterProtocol._onClosecCs|j||dS)z Register a Twisted producer with this protocol. :param producer: A Twisted push or pull producer. :type producer: object :param streaming: Producer type. :type streaming: bool N)rvregisterProducer)rZproducer streamingr5r5r6rs z)WebSocketAdapterProtocol.registerProducercCs|jdS)zA Unregister Twisted producer with this protocol. N)rvunregisterProducerrr5r5r6rsz+WebSocketAdapterProtocol.unregisterProducer)F)!rnrorprqtxaio make_loggerr}rrrr8__annotations__rsboolrr r rbytesrrrrrrrrrrrrrrrr5r5r5r6r!s, .   r!c@seZdZdZeZdZdS)r"z Base class for Twisted-based WebSocket server protocols. Implements :class:`autobahn.websocket.interfaces.IWebSocketChannel`. TN)rnrorprqrrr}rsr5r5r5r6r"sr"c@s2eZdZdZeZdZdefddZ ddZ dS) r#z Base class for Twisted-based WebSocket client protocols. Implements :class:`autobahn.websocket.interfaces.IWebSocketChannel`. FresponsecCs"|jjdt|j|d||S)Nz{meth}(response={response}))methr)r}r~r _onConnect onConnect)rZrr5r5r6rs z"WebSocketClientProtocol._onConnectcCs |jd|j|jjdS)NzStarting TLS upgrade)r}r~rvstartTLSrTrQrr5r5r6rs z WebSocketClientProtocol.startTLSN) rnrorprqrrr}rsrrrr5r5r5r6r#s  r#c@eZdZdZdS)r$zP Adapter class for Twisted-based WebSocket client and server factories. Nrnrorprqr5r5r5r6r$r$c@ eZdZdZeZddZdS)r%z Base class for Twisted-based WebSocket server factories. Implements :class:`autobahn.websocket.interfaces.IWebSocketServerChannelFactory` cOsD|dd}|durddlm}||_tjj|g|Ri|dS)a .. note:: In addition to all arguments to the constructor of :meth:`autobahn.websocket.interfaces.IWebSocketServerChannelFactory`, you can supply a ``reactor`` keyword argument to specify the Twisted reactor to be used. r4Nrr3)poprOr4rr%r[)rZargskwargsr4r5r5r6r[s  zWebSocketServerFactory.__init__Nrnrorprqrrr}r[r5r5r5r6r% r%c@r)r&z Base class for Twisted-based WebSocket client factories. Implements :class:`autobahn.websocket.interfaces.IWebSocketClientChannelFactory` cOsv|dd}|durddlm}||_tjj|g|Ri||jr7|jdur9ddlm}| |j |_ dSdSdS)a .. note:: In addition to all arguments to the constructor of :func:`autobahn.websocket.interfaces.IWebSocketClientChannelFactory`, you can supply a ``reactor`` keyword argument to specify the Twisted reactor to be used. r4Nrr3rJ) rrOr4rr&r[rNrCrKrPrLrQ)rZrrr4rKr5r5r6r[s   zWebSocketClientFactory.__init__Nrr5r5r5r6r&rr&c@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)r'a An adapter for stream-based transport over WebSocket. This follows `websockify `_ and should be compatible with that. It uses WebSocket subprotocol negotiation and supports the following WebSocket subprotocols: - ``binary`` (or a compatible subprotocol) - ``base64`` Octets are either transmitted as the payload of WebSocket binary messages when using the ``binary`` subprotocol (or an alternative binary compatible subprotocol), or encoded with Base64 and then transmitted as the payload of WebSocket text messages when using the ``base64`` subprotocol. cCst|tr'|}|jD]}||jjvr|dk|_|Sq ttjd|jjt|t rJ|}|j |jjvrB| t j j d|jj|j dk|_dStd)Nbase64z2this server only speaks {0} WebSocket subprotocolsz2this client only speaks {0} WebSocket subprotocolsz logic error)r7rr@rT _subprotocols _binaryModerNOT_ACCEPTABLEr:rr_fail_connectionWebSocketProtocol CLOSE_STATUS_CODE_PROTOCOL_ERROR Exception)rZrequestOrResponserequestprr5r5r6r$s     z"WrappingWebSocketAdapter.onConnectcCs|jdSrX)_protorrr5r5r6r6razWrappingWebSocketAdapter.onOpenc Csz||jkr|tjjddS|s5zt|}Wnty4}z|tjjd|WYd}~nd}~ww|j |dS)Nz>message payload type does not match the negotiated subprotocolz*message payload base64 decoding error: {0}) rrrr"CLOSE_STATUS_CODE_UNSUPPORTED_DATArr!CLOSE_STATUS_CODE_INVALID_PAYLOADr:rr)rZrrer5r5r6r9s  "z"WrappingWebSocketAdapter.onMessagecCs|jddSrX)rrrr5r5r6rErz WrappingWebSocketAdapter.onClosecCsBt|tksJ|jr|j|dddSt|}|j|dddS)NT)rF)r;rr sendMessagerrr5r5r6writeHs zWrappingWebSocketAdapter.writecCs|D]}||qdSrX)r)rZrdr5r5r6 writeSequenceQs z&WrappingWebSocketAdapter.writeSequencecCrrX) sendCloserr5r5r6rVs z'WrappingWebSocketAdapter.loseConnectioncC |jSrX)rvgetPeerrr5r5r6rZ z WrappingWebSocketAdapter.getPeercCrrX)rvgetHostrr5r5r6r^rz WrappingWebSocketAdapter.getHostN) rnrorprqrrrrrrrrrr5r5r5r6r's   r'c@r)r(zD Server protocol for stream-based transport over WebSocket. Nrr5r5r5r6r(crr(c@r)r)zD Client protocol for stream-based transport over WebSocket. Nrr5r5r5r6r)irr)c@s:eZdZdZ    d ddZddZd d Zd d ZdS)r*zL Wrapping server factory for stream-based transport over WebSocket. NTrcCsl||_ddg|_|r|j|tj||||jd|j|d|jdd|r4dd}|j|d d Sd S) z :param factory: Stream-based factory to be wrapped. :type factory: A subclass of ``twisted.internet.protocol.Factory`` :param url: WebSocket URL of the server this server factory will work for. :type url: unicode binaryrr\r4r@autoFragmentSizeF failByDropcSs$|D] }t|trt|SqdSrX)r7rr)offersofferr5r5r6accepts   z7WrappingWebSocketServerFactory.__init__..acceptperMessageCompressionAcceptN)_factoryrappendr%r[setProtocolOptions)rZrTr\r4enableCompressionr subprotocolrr5r5r6r[ts    z'WrappingWebSocketServerFactory.__init__cC&t}||_|j||_||j_|SrX)r(rTr buildProtocolrrvrZaddrrdr5r5r6r z,WrappingWebSocketServerFactory.buildProtocolcC|jt|dSrX)r startFactoryr%rr5r5r6r z+WrappingWebSocketServerFactory.startFactorycCrrX)r stopFactoryr%rr5r5r6rrz*WrappingWebSocketServerFactory.stopFactoryNTrN)rnrorprqr[rrrr5r5r5r6r*os * r*c@s*eZdZdZ    d ddZddZdS) r+zL Wrapping client factory for stream-based transport over WebSocket. NTrc Cs||_ddg|_|r|j|tj||||jd|j|d|jdd|r>tg}|j|ddd }|j|d d Sd S) a :param factory: Stream-based factory to be wrapped. :type factory: A subclass of ``twisted.internet.protocol.Factory`` :param url: WebSocket URL of the server this client factory will connect to. :type url: unicode rrrrFr)perMessageCompressionOfferscSst|tr t|SdSrX)r7rr)rr5r5r6rs z7WrappingWebSocketClientFactory.__init__..acceptrN)rrrr&r[rr) rZrTr\r4rrrrrr5r5r6r[s"     z'WrappingWebSocketClientFactory.__init__cCrrX)r)rTrrrrvrr5r5r6rrz,WrappingWebSocketClientFactory.buildProtocolr)rnrorprqr[rr5r5r5r6r+s -r+cCst|dr |j}nddlm}|jr |dur ddlm}|}|jdur9||_||jd|jd|||}|S|jrJ| |j |j ||||}|S||j |j |||}|S)a Establish WebSocket connection to a server. The connection parameters like target host, port, resource and others are provided via the factory. :param factory: The WebSocket protocol factory to be used for creating client protocol instances. :type factory: An :class:`autobahn.websocket.WebSocketClientFactory` instance. :param contextFactory: SSL context factory, required for secure WebSocket connections ("wss"). :type contextFactory: A `twisted.internet.ssl.ClientContextFactory `_ instance. :param timeout: Number of seconds to wait before assuming the connection has failed. :type timeout: int :param bindAddress: A (host, port) tuple of local address to bind to, or None. :type bindAddress: tuple :returns: The connector. :rtype: An object which implements `twisted.interface.IConnector `_. r4rr3NrJrLrM) rr4rOrNrKClientContextFactoryrCrQ connectTCP connectSSLrLrM)rTrQtimeout bindAddressr4rKconnr5r5r6r-s    r-2cCsbt|dr |j}nddlm}|jr&|durtd||j||||}|S||j|||}|S)a Listen for incoming WebSocket connections from clients. The connection parameters like listening port and others are provided via the factory. :param factory: The WebSocket protocol factory to be used for creating server protocol instances. :type factory: An :class:`autobahn.websocket.WebSocketServerFactory` instance. :param contextFactory: SSL context factory, required for secure WebSocket connections ("wss"). :type contextFactory: A twisted.internet.ssl.ContextFactory. :param backlog: Size of the listen queue. :type backlog: int :param interface: The interface (derived from hostname given) to bind to, defaults to '' (all). :type interface: str :returns: The listening port. :rtype: An object that implements `twisted.interface.IListeningPort `_. r4rr3NzCSecure WebSocket listen requested, but no SSL context factory given)rr4rOrNr listenSSLrM listenTCP)rTrQbacklog interfacer4listenerr5r5r6r,s  r,c@r)r.z Twisted-based WAMP-over-WebSocket server protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` Nrr5r5r5r6r.8r.c@eZdZdZeZddZdS)r/zD Twisted-based WAMP-over-WebSocket server protocol factory. cOsB|dd}tj||||j|d<tj|g|Ri|dS)a :param factory: A callable that produces instances that implement :class:`autobahn.wamp.interfaces.ITransportHandler` :type factory: callable :param serializers: A list of WAMP serializers to use (or ``None`` for all available serializers). :type serializers: list of objects implementing :class:`autobahn.wamp.interfaces.ISerializer` serializersNr@)rrr/r[ _protocolsr%rZrTrrrr5r5r6r[Ks  z#WampWebSocketServerFactory.__init__N)rnrorprqr.rr[r5r5r5r6r/C r/c@r)r0z Twisted-based WAMP-over-WebSocket client protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` Nrr5r5r5r6r0brr0c@r)r1zD Twisted-based WAMP-over-WebSocket client protocol factory. cOsH|dd}tj||||j|d<tj|g|Ri|d|_dS)a :param factory: A callable that produces instances that implement :class:`autobahn.wamp.interfaces.ITransportHandler` :type factory: callable :param serializer: The WAMP serializer to use (or ``None`` for "best" serializer, chosen as the first serializer available from this list: CBOR, MessagePack, UBJSON, JSON). :type serializer: object implementing :class:`autobahn.wamp.interfaces.ISerializer` rNr@F)rrr1r[rr&noisyrr5r5r6r[us   z#WampWebSocketClientFactory.__init__N)rnrorprqr0rr[r5r5r5r6r1mrr1)NrN)Nrr )Nrrrtypingrzope.interfacerr use_twistedtwisted.internet.protocoltwistedrOrtwisted.internet.interfacesrtwisted.internet.errorrr r twisted.internet.deferr twisted.python.failurer r autobahn.utilrrrrr autobahn.wamprautobahn.wamp.typesrautobahn.websocket.typesrrrautobahn.websocketrautobahn.websocket.interfacesrautobahn.twisted.utilrrautobahn.websocket.compressrrrr__all__r r>rIrWr2internetProtocolr!r"r#objectr$r% ServerFactoryr& ClientFactoryr'r(r)r*r+r-r,r.r/r0r1r5r5r5r6sp           )77)S?: + #