o @Biu@sddlmZddlmZmZddlZddlmZddlZe ddl Z ddl m Z ddlmZmZddlmZmZmZddlmZdd lmZdd lmZmZdd lmZdd lmZmZm Z m!Z!dd l"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-dZ.ddZ/ddZ0ddZ1ddZ2Gddde%Z3Gddde j4j#j5Z6eGddde6e#j7Z7eGd d!d!e6e#j8Z8Gd"d#d#e9Z:eGd$d%d%e:e#j;e j4j#j<Z;eGd&d'd'e:e#j=e j4j#j>Z=eeGd(d)d)e9Z?Gd*d+d+e?e7Z@Gd,d-d-e?e8ZAGd.d/d/e;ZBGd0d1d1e=ZCedAd3d4ZDedBd7d8ZEeGd9d:d:ejFe7ZFeGd;d<dd>ejHe8ZHeGd?d@d@ejIe=ZIdS)C)absolute_import) b64encode b64decodeN) implementer) endpoints) ITransport ISSLTransport)ConnectionDoneConnectionAbortedConnectionLost)Deferred)public) _is_tls_error_maybe_tls_reason) websocket)ConnectionRequestConnectionResponseConnectionDenyTransportDetails)protocol)IWebSocketClientAgent)peer2strtransport_channel_id)PerMessageDeflateOfferPerMessageDeflateOfferAcceptPerMessageDeflateResponsePerMessageDeflateResponseAccept)create_client_agentWebSocketAdapterProtocolWebSocketServerProtocolWebSocketClientProtocolWebSocketAdapterFactoryWebSocketServerFactoryWebSocketClientFactoryWrappingWebSocketAdapterWrappingWebSocketServerProtocolWrappingWebSocketClientProtocolWrappingWebSocketServerFactoryWrappingWebSocketClientFactorylistenWS connectWSWampWebSocketServerProtocolWampWebSocketServerFactoryWampWebSocketClientProtocolWampWebSocketClientFactorycCst|S)zB :returns: an instance implementing IWebSocketClientAgent )_TwistedWebSocketClientAgentreactorr2Z/var/www/html/Trade-python/venv/lib/python3.10/site-packages/autobahn/twisted/websocket.pyrVsrcCs>t|tjstdt||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) isinstancesix text_type ValueErrorformattype startswith)transport_configr2r2r3check_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)r4dictr7keysr8)options valid_keysactual_kr2r2r3check_client_optionsqs  rGcCs|jrddlm}||j}|jdur&||_t||jd|jd}|S|jr;ddlm}t ||j|j |}|St||j|j }|S)Nrsslhostport) isSecuretwisted.internetrIoptionsForClientTLSrJrAcontextFactoryrHostnameEndpointSSL4ClientEndpointrK)r1factoryr;rDrIcontext_factoryendpointr2r2r3_endpoint_from_configs4    rUc@s"eZdZdZddZdddZdS)r/z6 This agent creates connections using Twisted cCs ||_dSN)_reactor)selfr1r2r2r3__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). )urlr1Ncs|dSrV)errback)f)rtn_dr2r3failedz1_TwistedWebSocketClientAgent.open..failedcs fdd}j|S)Ncs|SrV)callback)arg)protor]r2r3handshake_completeds zQ_TwistedWebSocketClientAgent.open..got_proto..handshake_completed)is_open addCallbacks)rbrcr^r])rbr3 got_protosz4_TwistedWebSocketClientAgent.open..got_protor2) r<rGr#rWr rrUr connectre)rXr;rDprotocol_classrRrTproto_drgr2rfr3opens    z!_TwistedWebSocketClientAgent.openrV)__name__ __module__ __qualname____doc__rYrkr2r2r2r3r/sr/c@seZdZdZdZeZddZddZ ddZ d'd d Z d d Z ddZ ddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&S)(rzJ Adapter class for Twisted WebSocket client and server protocols. zcCstz t|j|_Wntyd|jj|_Ynw||jj d|jdz |j |j WdSYdS)Nz process:{}zConnection made to {peer}peer) r transportgetPeerrqAttributeErrorr8pid_connectionMadelogdebug setTcpNoDelay tcpNoDelayrXr2r2r3connectionMades z'WebSocketAdapterProtocol.connectionMadecCst|jtr|jjd|jdnNt|jr|jt|jn?t|jt r/|jjd|jdn/t|jt rOt |j}t |jdrD|jj }|jjd|j|dn|jjd|jt|j|jd||dS) Nz,Connection to/from {peer} was closed cleanlyrpz-Connection to/from {peer} was aborted locallymessagezDConnection to/from {peer} was lost in a non-clean fashion: {message})rqr}z7Connection to/from {peer} lost ({error_type}): {error}))rq error_typeerror)r4valuer rwrxrqrrrr r strhasattrr}r9_connectionLost)rXreasonr}r2r2r3connectionLosts.      z'WebSocketAdapterProtocol.connectionLostcC||dSrV) _dataReceivedrXdatar2r2r3 dataReceived"r_z%WebSocketAdapterProtocol.dataReceivedFcCs,|rt|jdr|jdS|jdS)NabortConnection)rrrrloseConnection)rXabortr2r2r3_closeConnection%sz)WebSocketAdapterProtocol._closeConnectioncC |dSrV)onOpenr{r2r2r3_onOpen, z WebSocketAdapterProtocol._onOpencCrrV)onMessageBegin)rXisBinaryr2r2r3_onMessageBegin/r_z(WebSocketAdapterProtocol._onMessageBegincCrrV)onMessageFrameBegin)rXlengthr2r2r3_onMessageFrameBegin2r_z-WebSocketAdapterProtocol._onMessageFrameBegincCrrV)onMessageFrameDatarXpayloadr2r2r3_onMessageFrameData5r_z,WebSocketAdapterProtocol._onMessageFrameDatacCrrV)onMessageFrameEndr{r2r2r3_onMessageFrameEnd8rz+WebSocketAdapterProtocol._onMessageFrameEndcCrrV)onMessageFramerr2r2r3_onMessageFrame;r_z(WebSocketAdapterProtocol._onMessageFramecCrrV) onMessageEndr{r2r2r3 _onMessageEnd>rz&WebSocketAdapterProtocol._onMessageEndcCs|||dSrV) onMessage)rXrrr2r2r3 _onMessageAz#WebSocketAdapterProtocol._onMessagecCrrV)onPingrr2r2r3_onPingDr_z WebSocketAdapterProtocol._onPingcCrrV)onPongrr2r2r3_onPongGr_z WebSocketAdapterProtocol._onPongcCs||||dSrV)onCloserXwasCleancoderr2r2r3_onCloseJsz!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)rrregisterProducer)rXproducer streamingr2r2r3rMs z)WebSocketAdapterProtocol.registerProducercCs|jdS)zA Unregister Twisted producer with this protocol. N)rrunregisterProducerr{r2r2r3rXz+WebSocketAdapterProtocol.unregisterProducerN)F)rlrmrnrorqtxaio make_loggerrwr|rrrrrrrrrrrrrrrrr2r2r2r3rs*  rc@s"eZdZdZeZdddZdS)rz Base class for Twisted-based WebSocket server protocols. Implements :class:`autobahn.websocket.interfaces.IWebSocketChannel`. tls-uniquecCt|jd|S)W Implements :func:`autobahn.wamp.interfaces.ITransport.get_channel_id` TrrrrXchannel_id_typer2r2r3get_channel_idirz&WebSocketServerProtocol.get_channel_idNr)rlrmrnrorrrwrr2r2r2r3r_src@s:eZdZdZeZddZddZd ddZ d d Z d S) r z Base class for Twisted-based WebSocket client protocols. Implements :class:`autobahn.websocket.interfaces.IWebSocketChannel`. cCrrV) onConnect)rXresponser2r2r3 _onConnectzr_z"WebSocketClientProtocol._onConnectcCs |jd|j|jjdS)NzStarting TLS upgrade)rwrxrrstartTLSrRrOr{r2r2r3r}s z WebSocketClientProtocol.startTLSrcCr)rFrrr2r2r3rrz&WebSocketClientProtocol.get_channel_idcCs8t|j}|rdt|jddi}ni}t|j||dS)z] Internal helper. Base class calls this to create a TransportDetails rF)rq is_securesecure_channel_id)r providedByrrrrrq)rXrrr2r2r3_create_transport_detailss z1WebSocketClientProtocol._create_transport_detailsNr) rlrmrnrorrrwrrrrr2r2r2r3r ps  r c@eZdZdZdS)r!zP Adapter class for Twisted-based WebSocket client and server factories. Nrlrmrnror2r2r2r3r!r!c@eZdZdZddZdS)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. r1Nrr0)poprMr1rr"rY)rXargskwargsr1r2r2r3rYs  zWebSocketServerFactory.__init__NrlrmrnrorYr2r2r2r3r" 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. r1Nrr0rH) rrMr1rr#rYrLrArIrNrJrO)rXrrr1rIr2r2r3rYs   zWebSocketClientFactory.__init__Nrr2r2r2r3r#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)r4rr>rR _subprotocols _binaryModerNOT_ACCEPTABLEr8rr_fail_connectionWebSocketProtocol CLOSE_STATUS_CODE_PROTOCOL_ERROR Exception)rXrequestOrResponserequestprr2r2r3rs     z"WrappingWebSocketAdapter.onConnectcCs|jdSrV)_protor|r{r2r2r3r r_zWrappingWebSocketAdapter.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_PAYLOADr8rr)rXrrer2r2r3rs  "z"WrappingWebSocketAdapter.onMessagecCs|jddSrV)rrrr2r2r3rrz WrappingWebSocketAdapter.onClosecCsBt|tksJ|jr|j|dddSt|}|j|dddS)NT)rF)r9bytesr sendMessagerrr2r2r3writes zWrappingWebSocketAdapter.writecCs|D]}||qdSrV)r)rXrdr2r2r3 writeSequence&s z&WrappingWebSocketAdapter.writeSequencecCrrV) sendCloser{r2r2r3r+s z'WrappingWebSocketAdapter.loseConnectioncC |jSrV)rrrsr{r2r2r3rs/ z WrappingWebSocketAdapter.getPeercCrrV)rrgetHostr{r2r2r3r3rz WrappingWebSocketAdapter.getHostN) rlrmrnrorrrrrrrrsrr2r2r2r3r$s   r$c@r)r%zD Server protocol for stream-based transport over WebSocket. Nrr2r2r2r3r%8rr%c@r)r&zD Client protocol for stream-based transport over WebSocket. Nrr2r2r2r3r&>rr&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 binaryrrZr1r>autoFragmentSizeF failByDropcSs$|D] }t|trt|SqdSrV)r4rr)offersofferr2r2r3acceptls   z7WrappingWebSocketServerFactory.__init__..acceptperMessageCompressionAcceptN)_factoryrappendr"rYsetProtocolOptions)rXrRrZr1enableCompressionr subprotocolrr2r2r3rYIs    z'WrappingWebSocketServerFactory.__init__cC&t}||_|j||_||j_|SrV)r%rRr buildProtocolrrrrXaddrrbr2r2r3rs z,WrappingWebSocketServerFactory.buildProtocolcC|jt|dSrV)r startFactoryr"r{r2r2r3rz z+WrappingWebSocketServerFactory.startFactorycCrrV)r stopFactoryr"r{r2r2r3r~rz*WrappingWebSocketServerFactory.stopFactoryNTrN)rlrmrnrorYrrrr2r2r2r3r'Ds * 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|SdSrV)r4rr)rr2r2r3rs z7WrappingWebSocketClientFactory.__init__..acceptrN)rrrr#rYrr) rXrRrZr1rrrrrr2r2r3rYs"     z'WrappingWebSocketClientFactory.__init__cCrrV)r&rRrrrrrrr2r2r3rrz,WrappingWebSocketClientFactory.buildProtocolr)rlrmrnrorYrr2r2r2r3r(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 `_. r1rr0NrHrJrK) rr1rMrLrIClientContextFactoryrArO connectTCP connectSSLrJrK)rRrOtimeout bindAddressr1rIconnr2r2r3r*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 `_. r1rr0NzCSecure WebSocket listen requested, but no SSL context factory given)rr1rMrLr listenSSLrK listenTCP)rRrObacklog interfacer1listenerr2r2r3r)s  r)c@r)r+z Twisted-based WAMP-over-WebSocket server protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` Nrr2r2r2r3r+ r+c@eZdZdZeZddZdS)r,zD Twisted-based WAMP-over-WebSocket server protocol factory. cOB|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,rY _protocolsr"rXrRrrrr2r2r3rY s  z#WampWebSocketServerFactory.__init__N)rlrmrnror+rrYr2r2r2r3r, r,c@r)r-z Twisted-based WAMP-over-WebSocket client protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` Nrr2r2r2r3r-7rr-c@r)r.zD Twisted-based WAMP-over-WebSocket client protocol factory. cOr)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>)rrr.rYrr#rr2r2r3rYJs  z#WampWebSocketClientFactory.__init__N)rlrmrnror-rrYr2r2r2r3r.Brr.)NrN)Nr r )J __future__rrrrr5zope.interfacerr use_twistedtwisted.internet.protocoltwistedrMrtwisted.internet.interfacesrrtwisted.internet.errorr r r twisted.internet.deferr autobahn.utilr rr autobahn.wamprautobahn.websocket.typesrrrrautobahn.websocketrautobahn.websocket.interfacesrautobahn.twisted.utilrrautobahn.websocket.compressrrrr__all__rr<rGrUr/internetProtocolrrr objectr!r" ServerFactoryr# ClientFactoryr$r%r&r'r(r*r)r+r,r-r.r2r2r2r3sj        )7u+'S?: + #