tfJddlmZmZddlmZddlmZddlZejddl 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/m0Z0m1Z1dZ2dZ3dZ4dZ5dZ6Gdde)Z7Gdde j8j'j9Z:eGdde:e'j;Z;eGdd e:e'j<Z<Gd!d"e=Z>eGd#d$e>e'j?e j8j'j@Z?eGd%d&e>e'jAe j8j'jBZAeeGd'd(e=ZCGd)d*eCe;ZDGd+d,eCe<ZEGd-d.e?ZFGd/d0eAZGed>d2ZHed?d5ZIeGd6d7ejJe;ZJeGd8d9ejKe?ZKeGd:d;ejLe<ZLeGd<d=ejMeAZMdS)@) 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 connectWSWampWebSocketServerProtocolWampWebSocketServerFactoryWampWebSocketClientProtocolWampWebSocketClientFactoryc t|S)zB :returns: an instance implementing IWebSocketClientAgent )_TwistedWebSocketClientAgentreactors Y/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/autobahn/twisted/websocket.pyr!r!Vs ( 0 00ct|ts/tdt ||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_configs r7check_transport_configrA]s & , ,  ; B B4HXCYCY Z Z    & &w / /  & &x 0 0  D    4r8ct|tstdgd}|D](}||vr"td|)dS)z5 raises a ValueError if `options` is invalid z'options' must be a dict)origin protocols useragentheadersproxyz'options' may not contain '{}'N)r:dictr<keysr=)options valid_keysactual_ks r7check_client_optionsrMqs gt $ $  &   JLLNN : % %077AA  &r8ch|jr ddlm}||j}|j4||_tj||jd|jd}nO|jr(ddlm}tj ||j|j |}n tj||j|j }|S)Nrsslhostport) isSecuretwisted.internetrPoptionsForClientTLSrQrGcontextFactoryrHostnameEndpointSSL4ClientEndpointrR)r6factoryr@rJrPcontext_factoryendpoints r7_endpoint_from_configr\s@((((((11',??} !0-  M& ! M& !      , , , , , , 3   HH!1  H Or8c eZdZdZdZddZdS)r4z6 This agent creates connections using Twisted c||_dSN)_reactor)selfr6s r7__init__z%_TwistedWebSocketClientAgent.__init__s  r8NcD t|t|td||jd|}|tn||_t |j|||}t ||} fd fd}| | 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). )urlr6Nc2|dSr_)errback)frtn_ds r7failedz1_TwistedWebSocketClientAgent.open..faileds MM!     r8cLfd}j|S)Nc2|Sr_)callback)argprotorhs r7handshake_completedzQ_TwistedWebSocketClientAgent.open..got_proto..handshake_completedsu%%% r8)is_open addCallbacks)rnrorirhs` r7 got_protoz4_TwistedWebSocketClientAgent.open..got_protosB       M & &':F C C CLr8) rArMr'r`r$rr\r connectrq) rar@rJprotocol_classrYr[proto_drrrirhs @@r7openz!_TwistedWebSocketClientAgent.opens" /000W%%%(  M     7E6L22R`)AQSZ[[ ""7++            Y/// r8r_)__name__ __module__ __qualname____doc__rbrwrsr8r7r4r4sA   ,,,,,,r8r4ceZdZUdZejZdZee e d<dZ ee e d<dZ efdefdZdefd Zdd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZ 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_serverct|j|j|_tj|j_|jj|_ |j|j n#YnxYw| |j dt|jt|jdS)Nz/{func} connection established for peer="{peer}"funcr})r transportr~_transport_detailsrCHANNEL_FRAMING_WEBSOCKETchannel_framingr} setTcpNoDelay tcpNoDelay_connectionMadelogdebugrconnectionMaderras r7rz'WebSocketAdapterProtocol.connectionMade s #;4>4>"Z"Z2B2\/+0   N ( ( 9 9 9 9  D  H"4#677!$),,  . . . . .s A((A,reasoncd}t|jtr%|jd|jd}n)t |jr-|jt|jnt|jtr"|jd|jnt|jtrXt|j}t|jdr |jj }|jd|j|n:|jd |jt|j|j |||rC|jd t!|jt%|j dS|jd t!|jt%|j|dS)NFz,Connection to/from {peer} was closed cleanly)r}Tz-Connection to/from {peer} was aborted locallymessagezDConnection to/from {peer} was lost in a non-clean fashion: {message})r}rz7Connection to/from {peer} lost ({error_type}): {error}))r} error_typeerrorz8{func} connection lost for peer="{peer}", closed cleanlyrzD{func} connection lost for peer="{peer}", closed with error {reason})rr}r)r:valuer rrr}rrrr r r;hasattrrr>_connectionLostrconnectionLostr)rar was_cleanrs r7rz'WebSocketAdapterProtocol.connectionLost's flN 3 3 ^ HNNI $   + + +II 6< ( ( ^ HNN,V\:: ; ; ; ;  &7 8 8 ^ HNNJ $   + + + + n 5 5 ^&,''Gv|Y// / ,. HNNVY      HNNT $ d6<6H6HPVP\  ^ ^ ^ V$$$  * HNNU &t': ; ; %di 0 0  2 2 2 2 2 HNNa &t': ; ; %di 0 0"(  * * * * *r8datac |jdt|jt |jt t |||dS)Nz2{func} received {data_len} bytes for peer="{peer}")rr}data_len)rrr dataReceivedrr}len _dataReceivedrars r7rz%WebSocketAdapterProtocol.dataReceivedUsj K"4#455!$),, %c$ii 0 0  2 2 2 4     r8Fc|r0t|jdr|jdS|jdS)NabortConnection)rrrloseConnection)raaborts r7_closeConnectionz)WebSocketAdapterProtocol._closeConnection^sT  ,WT^->?? , N * * , , , , , N ) ) + + + + +r8c|jjr.dt|j|jjdi}||j_|dS)Nz tls-unique)r is_securerrr~ channel_idonOpen)rars r7_onOpenz WebSocketAdapterProtocol._onOpenesR  " , <24>4CZCdfrssJ2,*,*W,*,*,*,*\!!!!!,,,,   &&&))))))!!!%%%***--- = = =,,,,,r8r"c4eZdZdZejZdZdS)r#z Base class for Twisted-based WebSocket server protocols. Implements :class:`autobahn.websocket.interfaces.IWebSocketChannel`. TN)rxryrzr{rrrr~rsr8r7r#r#s/ %   CIIIr8r#cFeZdZdZejZdZdefdZ dZ dS)r$z Base class for Twisted-based WebSocket client protocols. Implements :class:`autobahn.websocket.interfaces.IWebSocketChannel`. Fresponsec|jdt|j|||S)Nz{meth}(response={response}))methr)rrr _onConnect onConnect)rars r7rz"WebSocketClientProtocol._onConnects: 46$/;R;R]efff~~h'''r8c|jd|j|jjdS)NzStarting TLS upgrade)rrrstartTLSrYrVrs r7rz WebSocketClientProtocol.startTLSs8 -...  ;<<<<`_ 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. ct|tr`|}|jD]}||jjvr|dk|_|cS t t jd|jjt|tri|}|j |jjvrB| tj j d|jj|j dk|_dStd)Nbase64z2this server only speaks {0} WebSocket subprotocolsz2this client only speaks {0} WebSocket subprotocolsz logic error)r:rrDrY _subprotocols _binaryModerNOT_ACCEPTABLEr=rr_fail_connectionWebSocketProtocol CLOSE_STATUS_CODE_PROTOCOL_ERROR Exception)rarequestOrResponserequestprs r7rz"WrappingWebSocketAdapter.onConnect$sL '): ; ; +'G&   222()X D$HHH3!!>@t@{@{}A}I}WAXAXYY Y )+= > > +(H  (BBB%%h&@&adXd_d_`d`l`zd{d{||| ( 1X =D   M** *r8c8|jdSr_)_protorrs r7rzWrappingWebSocketAdapter.onOpen6s ""$$$$$r8cb||jkr'|tjjddS|s` t |}nO#t $rB}|tjjd|Yd}~nd}~wwxYw|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)rarres r7rz"WrappingWebSocketAdapter.onMessage9s t' ' '  ! !("<"_bb c c c c c ``'00GG ```))(*D*fiUi\i\]^i_i_````````` K $ $W - - - - -sA B8B  Bc:|jddSr_)rrrs r7rz WrappingWebSocketAdapter.onCloseEs ""4(((((r8ct|tksJ|jr||ddSt |}||ddS)NT)rF)r>rr sendMessagerrs r7writezWrappingWebSocketAdapter.writeHsmDzzU""""   3   TD  1 1 1 1 1T??D   TE  2 2 2 2 2r8c:|D]}||dSr_)r)rards r7 writeSequencez&WrappingWebSocketAdapter.writeSequenceQs,  A JJqMMMM  r8c.|dSr_) sendClosers r7rz'WrappingWebSocketAdapter.loseConnectionVs r8c4|jSr_)rgetPeerrs r7r z WrappingWebSocketAdapter.getPeerZ~%%'''r8c4|jSr_)rgetHostrs r7r z WrappingWebSocketAdapter.getHost^r r8N) rxryrzr{rrrrrrrr r rsr8r7r(r(s&+++$%%% . . .)))333 ((((((((r8r(ceZdZdZdS)r)zD Server protocol for stream-based transport over WebSocket. Nrrsr8r7r)r)crr8r)ceZdZdZdS)r*zD Client protocol for stream-based transport over WebSocket. Nrrsr8r7r*r*irr8r*c4eZdZdZ d dZdZdZdZdS) r+zL Wrapping server factory for stream-based transport over WebSocket. NTrc6||_ddg|_|r|j|t||||j|||d|rd}||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 binaryrrdr6rDautoFragmentSizeF failByDropc\|D](}t|trt|cS)dSr_)r:rr)offersoffers r7acceptz7WrappingWebSocketServerFactory.__init__..acceptsM#CCE!%)?@@C;EBBBBBCCCr8perMessageCompressionAcceptN)_factoryrappendr&rbsetProtocolOptions)rarYrdr6enableCompressionr subprotocolrs r7rbz'WrappingWebSocketServerFactory.__init__ts  &1  3   % %k 2 2 2'',/07262D ( F F F 1ABBB 5111  H C C C  # # # G G G G G H Hr8ct}||_|j||_||j_|Sr_)r)rYr buildProtocolrrraaddrrns r7r$z,WrappingWebSocketServerFactory.buildProtocol:/11 }22488 !&  r8cl|jt|dSr_)r startFactoryr&rs r7r)z+WrappingWebSocketServerFactory.startFactorys0 ""$$$++D11111r8cl|jt|dSr_)r stopFactoryr&rs r7r+z*WrappingWebSocketServerFactory.stopFactorys0 !!###**400000r8NTrN)rxryrzr{rbr$r)r+rsr8r7r+r+osr#'"#! (H(H(H(HT22211111r8r+c(eZdZdZ ddZdZdS)r,zL Wrapping client factory for stream-based transport over WebSocket. NTrc||_ddg|_|r|j|t||||j|||d|r@t g}||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)perMessageCompressionOfferscNt|trt|SdSr_)r:rr )rs r7rz7WrappingWebSocketClientFactory.__init__..accepts0h(ABBE:8DDDEEr8rN)rrrr'rbr r) rarYrdr6r!rr"rrs r7rbz'WrappingWebSocketClientFactory.__init__s  &1  3   % %k 2 2 2'',/07262D ( F F F 1ABBB 5111  H-../F  # # # G G G E E E  # # # G G G G G H Hr8ct}||_|j||_||j_|Sr_)r*rYrr$rrr%s r7r$z,WrappingWebSocketClientFactory.buildProtocolr'r8r,)rxryrzr{rbr$rsr8r7r,r,sT#'"#! +H+H+H+HZr8r,ct|dr|j}nddlm}|jr|ddlm}|}|j7||_||jd|jd|||}nO|jr%| |j |j ||||}n#||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 `_. r6rr5NrOrQrR) rr6rTrSrPClientContextFactoryrGrV connectTCP connectSSLrQrR)rYrVtimeout bindAddressr6rPconns r7r.r.s,w ""-/,,,,,,8  ! , , , , , , 5577N} !/!!'-"7v9NPWY`bmnn   a%%glGL'>[bdoppDD%%glGL'7T_``D Kr82ct|dr|j}nddlm}|jr0|t d||j||||}n||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 `_. r6rr5NzCSecure WebSocket listen requested, but no SSL context factory given)rr6rTrSr listenSSLrR listenTCP)rYrVbacklog interfacer6listeners r7r-r-s,w ""-/,,,,,,P  !abb b$$W\7NGU^__$$W\7GYOO Or8ceZdZdZdS)r/z Twisted-based WAMP-over-WebSocket server protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` Nrrsr8r7r/r/8r8r/ceZdZdZeZdZdS)r0zD Twisted-based WAMP-over-WebSocket server protocol factory. c|dd}tj||||j|d<t j|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` serializersNrD)rrr0rb _protocolsr&rarYrrrFs r7rbz#WampWebSocketServerFactory.__init__Ksgjj55 ,55dG[QQQ"o{ '>t>>>v>>>>>r8N)rxryrzr{r/rrbrsr8r7r0r0Cs4+H?????r8r0ceZdZdZdS)r1z Twisted-based WAMP-over-WebSocket client protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` Nrrsr8r7r1r1brCr8r1ceZdZdZeZdZdS)r2zD Twisted-based WAMP-over-WebSocket client protocol factory. c|dd}tj||||j|d<t j|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` rFNrDF)rrr2rbrGr'noisyrHs r7rbz#WampWebSocketClientFactory.__init__usljj55 ,55dG[QQQ"o{'>t>>>v>>> r8N)rxryrzr{r1rrbrsr8r7r2r2ms4+Hr8r2)Nr2N)Nr:r;)Nrrrtypingrzope.interfacerr use_twistedtwisted.internet.protocoltwistedrTrtwisted.internet.interfacesrtwisted.internet.errorr r 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!rArMr\r4internetProtocolr"r#r$objectr%r& ServerFactoryr' ClientFactoryr(r)r*r+r,r.r-r/r0r1r2rsr8r7rds6('''''''&&&&&&  &&&&&&222222++++++******444444//////////::::::::######000000ZZZZZZZZZZ''''''??????PPPPPPPP$$$$$$$$$$$$  4111(2&&&R44444#8444ns,s,s,s,s,w/8As,s,s,l68X=====68X===&f HHHHH4h6UW^WgWpW~HHH6 E E E E E4h6UW^WgWpW~ E E ER ZP(P(P(P(P(vP(P(P(f&>@W &>@W <1<1<1<1<1%;<1<1<1~77777%;777t((((V    F)"GI`?????!EG]???<)"GI`!EG]r8