U ucJ@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 )_TwistedWebSocketClientAgentreactorr5[/var/www/html/HOTSTAR/school/venv/lib/python3.8/site-packages/autobahn/twisted/websocket.pyr Vsr cCs<t|tstdt||ds8|ds8tddS)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>cCsHt|tstddddddg}|D]}||kr(td|q(dS) z5 raises a ValueError if `options` is invalid z'options' must be a dictorigin protocolsZ useragentheadersproxyz'options' may not contain '{}'N)r7dictr9keysr:)options valid_keysZactual_kr5r5r6check_client_optionsqs  rGcCs|jrddlm}||j}|jdk rJ||_t||jd|jd}n:|jrrddlm}t ||j|j |}nt||j|j }|S)Nrsslhostport) isSecuretwisted.internetrIoptionsForClientTLSrJrBcontextFactoryrZHostnameEndpointZSSL4ClientEndpointrK)r4factoryr=rErIZcontext_factoryendpointr5r5r6_endpoint_from_configs0    rRc@s"eZdZdZddZdddZdS)r2z6 This agent creates connections using Twisted cCs ||_dSN)_reactor)selfr4r5r5r6__init__sz%_TwistedWebSocketClientAgent.__init__Ncst|t|tf||jd|}|dkr2tn||_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|dSrS)Zerrback)f)rtn_dr5r6failedsz1_TwistedWebSocketClientAgent.open..failedcs fdd}j|S)Ncs|SrS)callback)arg)protorYr5r6handshake_completeds zQ_TwistedWebSocketClientAgent.open..got_proto..handshake_completed)Zis_open addCallbacks)r]r^rZrY)r]r6 got_protosz4_TwistedWebSocketClientAgent.open..got_proto) r>rGr&rTr#rrRr connectr_)rUr=rEZprotocol_classrPrQZproto_drar5r`r6opens    z!_TwistedWebSocketClientAgent.open)N)__name__ __module__ __qualname____doc__rVrcr5r5r5r6r2sr2c@seZdZUdZeZdZee e d<dZ ee e d<ddZ efeddd Zed d 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_servercCsnt|j|j|_tj|j_|jj|_z|j|j Wn YnX| |j j dt |jt|jddS)Nz/{func} connection established for peer="{peer}"funcrh)r transportri_transport_detailsrZCHANNEL_FRAMING_WEBSOCKETZchannel_framingrhZ setTcpNoDelayZ tcpNoDelayZ_connectionMadelogdebugrconnectionMaderrUr5r5r6rp s  z'WebSocketAdapterProtocol.connectionMade)reasoncCsd}t|jtr(|jjd|jdd}nt|jrF|jt|jn~t|jt rf|jjd|jdn^t|jt rt |j}t |jdr|jj }|jjd|j|dn|jjd |jt|j|jd |||r|jjd t|jt|jd n |jjd t|jt|j|ddS)NFz,Connection to/from {peer} was closed cleanly)rhTz-Connection to/from {peer} was aborted locallymessagezDConnection to/from {peer} was lost in a non-clean fashion: {message})rhrsz7Connection to/from {peer} lost ({error_type}): {error}))rhZ error_typeerrorz8{func} connection lost for peer="{peer}", closed cleanlyrjzD{func} connection lost for peer="{peer}", closed with error {reason})rkrhrr)r7valuerrnrorhrrtrr r r8hasattrrsr;Z_connectionLostrconnectionLostr)rUrrZ was_cleanrsr5r5r6rw'sJ       z'WebSocketAdapterProtocol.connectionLost)datacCs6|jjdt|jt|jtt|d||dS)Nz2{func} received {data_len} bytes for peer="{peer}")rkrhdata_len)rnror dataReceivedrrhlenZ _dataReceivedrUrxr5r5r6rzUs  z%WebSocketAdapterProtocol.dataReceivedFcCs*|rt|jdr|jn |jdS)NabortConnection)rvrlr}loseConnection)rUabortr5r5r6_closeConnection^s z)WebSocketAdapterProtocol._closeConnectioncCs2|jjr&dt|j|jjdi}||j_|dS)Nz tls-unique)rmZ is_securerrlri channel_idonOpen)rUrr5r5r6_onOpenes z WebSocketAdapterProtocol._onOpencCs||dSrS)ZonMessageBegin)rUisBinaryr5r5r6_onMessageBeginpsz(WebSocketAdapterProtocol._onMessageBegincCs||dSrS)ZonMessageFrameBegin)rUlengthr5r5r6_onMessageFrameBeginssz-WebSocketAdapterProtocol._onMessageFrameBegincCs||dSrS)ZonMessageFrameDatarUpayloadr5r5r6_onMessageFrameDatavsz,WebSocketAdapterProtocol._onMessageFrameDatacCs |dSrS)ZonMessageFrameEndrqr5r5r6_onMessageFrameEndysz+WebSocketAdapterProtocol._onMessageFrameEndcCs||dSrS)ZonMessageFramerr5r5r6_onMessageFrame|sz(WebSocketAdapterProtocol._onMessageFramecCs |dSrS)Z onMessageEndrqr5r5r6 _onMessageEndsz&WebSocketAdapterProtocol._onMessageEndcCs|||dSrS) onMessage)rUrrr5r5r6 _onMessagesz#WebSocketAdapterProtocol._onMessagecCs||dSrS)ZonPingrr5r5r6_onPingsz WebSocketAdapterProtocol._onPingcCs||dSrS)ZonPongrr5r5r6_onPongsz WebSocketAdapterProtocol._onPongcCs||||dSrS)onCloserUZwasCleancoderrr5r5r6_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)rlregisterProducer)rUZproducerZ streamingr5r5r6rs z)WebSocketAdapterProtocol.registerProducercCs|jdS)zA Unregister Twisted producer with this protocol. N)rlunregisterProducerrqr5r5r6rsz+WebSocketAdapterProtocol.unregisterProducer)F)!rdrerfrgtxaio make_loggerrnrhrr8__annotations__riboolrpr r rwbytesrzrrrrrrrrrrrrrrr5r5r5r6r!s* .   r!c@seZdZdZeZdZdS)r"z Base class for Twisted-based WebSocket server protocols. Implements :class:`autobahn.websocket.interfaces.IWebSocketChannel`. TN)rdrerfrgrrrnrir5r5r5r6r"sr"c@s2eZdZdZeZdZedddZ 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)rnror _onConnect onConnect)rUrr5r5r6rsz"WebSocketClientProtocol._onConnectcCs |jd|j|jjdS)NzStarting TLS upgrade)rnrorlstartTLSrPrOrqr5r5r6rs z WebSocketClientProtocol.startTLSN) rdrerfrgrrrnrirrrr5r5r5r6r#s r#c@seZdZdZdS)r$zP Adapter class for Twisted-based WebSocket client and server factories. Nrdrerfrgr5r5r5r6r$sr$c@s eZdZdZeZddZdS)r%z Base class for Twisted-based WebSocket server factories. Implements :class:`autobahn.websocket.interfaces.IWebSocketServerChannelFactory` cOs>|dd}|dkr ddlm}||_tjj|f||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)poprMr4rr%rV)rUargskwargsr4r5r5r6rVs  zWebSocketServerFactory.__init__NrdrerfrgrrrnrVr5r5r5r6r%sr%c@s eZdZdZeZddZdS)r&z Base class for Twisted-based WebSocket client factories. Implements :class:`autobahn.websocket.interfaces.IWebSocketClientChannelFactory` cOsh|dd}|dkr ddlm}||_tjj|f|||jrd|jdk rdddlm}| |j |_ dS)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. r4Nrr3rH) rrMr4rr&rVrLrBrIrNrJrO)rUrrr4rIr5r5r6rVs   zWebSocketClientFactory.__init__Nrr5r5r5r6r&sr&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|trP|}|jD]"}||jjkr|dk|_|Sqttjd|jjnLt|t r|}|j |jjkr| t j j d|jj|j dk|_ntddS)Nbase64z2this server only speaks {0} WebSocket subprotocolsz2this client only speaks {0} WebSocket subprotocolsz logic error)r7rr@rP _subprotocols _binaryModerZNOT_ACCEPTABLEr:rr_fail_connectionWebSocketProtocolZ CLOSE_STATUS_CODE_PROTOCOL_ERROR Exception)rUZrequestOrResponserequestprr5r5r6r$s      z"WrappingWebSocketAdapter.onConnectcCs|jdSrS)_protorprqr5r5r6r6szWrappingWebSocketAdapter.onOpenc Csv||jkr|tjjdnV|sfz t|}Wn8tk rd}z|tjjd|W5d}~XYnX|j |dS)Nz>message payload type does not match the negotiated subprotocolz*message payload base64 decoding error: {0}) rrrrZ"CLOSE_STATUS_CODE_UNSUPPORTED_DATArrZ!CLOSE_STATUS_CODE_INVALID_PAYLOADr:rrz)rUrrer5r5r6r9s  (z"WrappingWebSocketAdapter.onMessagecCs|jddSrS)rrwrr5r5r6rEsz WrappingWebSocketAdapter.onClosecCs@t|tkst|jr&|j|ddnt|}|j|dddS)NT)rF)r;rAssertionErrorrZ sendMessagerr|r5r5r6writeHs zWrappingWebSocketAdapter.writecCs|D]}||qdSrS)r)rUrxdr5r5r6 writeSequenceQsz&WrappingWebSocketAdapter.writeSequencecCs |dSrS)Z sendCloserqr5r5r6r~Vsz'WrappingWebSocketAdapter.loseConnectioncCs |jSrS)rlgetPeerrqr5r5r6rZsz WrappingWebSocketAdapter.getPeercCs |jSrS)rlgetHostrqr5r5r6r^sz WrappingWebSocketAdapter.getHostN) rdrerfrgrrrrrrr~rrr5r5r5r6r's  r'c@seZdZdZdS)r(zD Server protocol for stream-based transport over WebSocket. Nrr5r5r5r6r(csr(c@seZdZdZdS)r)zD Client protocol for stream-based transport over WebSocket. Nrr5r5r5r6r)isr)c@s2eZdZdZd ddZddZd d Zd d ZdS)r*zL Wrapping server factory for stream-based transport over WebSocket. NTrcCsh||_ddg|_|r |j|tj||||jd|j|d|jdd|rddd}|j|d d 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 binaryrrWr4r@autoFragmentSizeFZ failByDropcSs$|D]}t|trt|SqdSrS)r7rr)offersZofferr5r5r6accepts z7WrappingWebSocketServerFactory.__init__..acceptZperMessageCompressionAcceptN)_factoryrappendr%rVsetProtocolOptions)rUrPrWr4enableCompressionr subprotocolrr5r5r6rVts    z'WrappingWebSocketServerFactory.__init__cCs&t}||_|j||_||j_|SrS)r(rPr buildProtocolrrlrUaddrr]r5r5r6rs z,WrappingWebSocketServerFactory.buildProtocolcCs|jt|dSrS)r startFactoryr%rqr5r5r6rs z+WrappingWebSocketServerFactory.startFactorycCs|jt|dSrS)r stopFactoryr%rqr5r5r6rs z*WrappingWebSocketServerFactory.stopFactory)NTrN)rdrerfrgrVrrrr5r5r5r6r*os *r*c@s"eZdZdZd ddZddZdS) r+zL Wrapping client factory for stream-based transport over WebSocket. NTrc Cs|||_ddg|_|r |j|tj||||jd|j|d|jdd|rxtg}|j|ddd }|j|d d 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)ZperMessageCompressionOfferscSst|trt|SdSrS)r7rrrr5r5r6rs z7WrappingWebSocketClientFactory.__init__..acceptrN)rrrr&rVrr) rUrPrWr4rrrrrr5r5r6rVs      z'WrappingWebSocketClientFactory.__init__cCs&t}||_|j||_||j_|SrS)r)rPrrrrlrr5r5r6rs z,WrappingWebSocketClientFactory.buildProtocol)NTrN)rdrerfrgrVrr5r5r5r6r+s -r+cCst|dr|j}n ddlm}|jr@|dkr@ddlm}|}|jdk rp||_||jd|jd|||}n6|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 `_. r4rr3NrHrJrK) rvr4rMrLrIZClientContextFactoryrBrOZ connectTCPZ connectSSLrJrK)rPrOtimeoutZ bindAddressr4rIconnr5r5r6r-s     r-2cCs`t|dr|j}n ddlm}|jrJ|dkr4td||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 `_. r4rr3NzCSecure WebSocket listen requested, but no SSL context factory given)rvr4rMrLrZ listenSSLrKZ listenTCP)rPrObacklogZ interfacer4Zlistenerr5r5r6r,s  r,c@seZdZdZdS)r.z Twisted-based WAMP-over-WebSocket server protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` Nrr5r5r5r6r.8sr.c@seZdZdZeZddZdS)r/zD Twisted-based WAMP-over-WebSocket server protocol factory. cOs<|dd}tj||||j|d<tj|f||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/rV _protocolsr%rUrPrrrr5r5r6rVKs  z#WampWebSocketServerFactory.__init__N)rdrerfrgr.rrVr5r5r5r6r/Csr/c@seZdZdZdS)r0z Twisted-based WAMP-over-WebSocket client protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` Nrr5r5r5r6r0bsr0c@seZdZdZeZddZdS)r1zD Twisted-based WAMP-over-WebSocket client protocol factory. cOsB|dd}tj||||j|d<tj|f||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)rrr1rVrr&Znoisyrr5r5r6rVus  z#WampWebSocketClientFactory.__init__N)rdrerfrgr0rrVr5r5r5r6r1msr1)NrN)Nrr)NrrrtypingrZzope.interfacerrZ use_twistedZtwisted.internet.protocoltwistedrMrZtwisted.internet.interfacesrZtwisted.internet.errorrr r Ztwisted.internet.deferr Ztwisted.python.failurer r Z autobahn.utilrrrrrZ autobahn.wamprZautobahn.wamp.typesrZautobahn.websocket.typesrrrZautobahn.websocketrZautobahn.websocket.interfacesrZautobahn.twisted.utilrrZautobahn.websocket.compressrrrr__all__r r>rGrRr2ZinternetProtocolr!r"r#objectr$r%Z ServerFactoryr&Z ClientFactoryr'r(r)r*r+r-r,r.r/r0r1r5r5r5r6sn           )77)S?: + #