o iJ@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 N)_TwistedWebSocketClientAgentreactorr5`C:\Users\psant\Downloads\PAF backend\school\venv\Lib\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 protocolsZ useragentheadersproxyz'options' may not contain '{}'N)r7dictr9keysr:)optionsZ valid_keysZactual_kr5r5r6check_client_optionsqs  rFcCs|jrddlm}||j}|jdur&||_t||jd|jd}|S|jr;ddlm}t ||j|j |}|St||j|j }|S)Nrsslhostport) isSecuretwisted.internetrHoptionsForClientTLSrIrBcontextFactoryrZHostnameEndpointZSSL4ClientEndpointrJ)r4factoryr=rErHZcontext_factoryendpointr5r5r6_endpoint_from_configs4    rQc@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|dSrR)Zerrback)f)rtn_dr5r6failedz1_TwistedWebSocketClientAgent.open..failedcs fdd}j|S)Ncs|SrR)callback)arg)protorXr5r6handshake_completeds zQ_TwistedWebSocketClientAgent.open..got_proto..handshake_completed)Zis_open addCallbacks)r]r^rYrX)r]r6 got_protosz4_TwistedWebSocketClientAgent.open..got_protor5) r>rFr&rSr#rrQr connectr_)rTr=rEZprotocol_classrOrPZproto_drar5r`r6opens    z!_TwistedWebSocketClientAgent.openrR)__name__ __module__ __qualname____doc__rUrcr5r5r5r6r2sr2c@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}"funcrh)r transportri_transport_detailsrZCHANNEL_FRAMING_WEBSOCKETZchannel_framingrhZ setTcpNoDelayZ tcpNoDelayZ_connectionMadelogdebugrconnectionMaderrTr5r5r6rp 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)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)rTrrZ was_cleanrsr5r5r6rw'sF         z'WebSocketAdapterProtocol.connectionLostdatacCs6|jjdt|jt|jtt|d||dS)Nz2{func} received {data_len} bytes for peer="{peer}")rkrhdata_len)rnror dataReceivedrrhlenZ _dataReceivedrTrxr5r5r6rzUs  z%WebSocketAdapterProtocol.dataReceivedFcCs,|rt|jdr|jdS|jdS)NabortConnection)rvrlr}loseConnection)rTabortr5r5r6_closeConnection^sz)WebSocketAdapterProtocol._closeConnectioncCs2|jjrdt|j|jjdi}||j_|dS)Nz tls-unique)rmZ is_securerrlri channel_idonOpen)rTrr5r5r6_onOpenes  z WebSocketAdapterProtocol._onOpencC||dSrR)ZonMessageBegin)rTisBinaryr5r5r6_onMessageBeginprZz(WebSocketAdapterProtocol._onMessageBegincCrrR)ZonMessageFrameBegin)rTlengthr5r5r6_onMessageFrameBeginsrZz-WebSocketAdapterProtocol._onMessageFrameBegincCrrR)ZonMessageFrameDatarTpayloadr5r5r6_onMessageFrameDatavrZz,WebSocketAdapterProtocol._onMessageFrameDatacC |dSrR)ZonMessageFrameEndrqr5r5r6_onMessageFrameEndy z+WebSocketAdapterProtocol._onMessageFrameEndcCrrR)ZonMessageFramerr5r5r6_onMessageFrame|rZz(WebSocketAdapterProtocol._onMessageFramecCrrR)Z onMessageEndrqr5r5r6 _onMessageEndrz&WebSocketAdapterProtocol._onMessageEndcCs|||dSrR) onMessage)rTrrr5r5r6 _onMessagez#WebSocketAdapterProtocol._onMessagecCrrR)ZonPingrr5r5r6_onPingrZz WebSocketAdapterProtocol._onPingcCrrR)ZonPongrr5r5r6_onPongrZz WebSocketAdapterProtocol._onPongcCs||||dSrR)onCloserTZwasCleancoderrr5r5r6_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)rTZproducerZ 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@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)rnror _onConnect onConnect)rTrr5r5r6rs z"WebSocketClientProtocol._onConnectcCs |jd|j|jjdS)NzStarting TLS upgrade)rnrorlstartTLSrOrNrqr5r5r6rs z WebSocketClientProtocol.startTLSN) rdrerfrgrrrnrirrrr5r5r5r6r#s  r#c@eZdZdZdS)r$zP Adapter class for Twisted-based WebSocket client and server factories. Nrdrerfrgr5r5r5r6r$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)poprLr4rr%rU)rTargskwargsr4r5r5r6rUs  zWebSocketServerFactory.__init__NrdrerfrgrrrnrUr5r5r5r6r% 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. r4Nrr3rG) rrLr4rr&rUrKrBrHrMrIrN)rTrrr4rHr5r5r6rUs   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@rO _subprotocols _binaryModerNOT_ACCEPTABLEr:rr_fail_connectionWebSocketProtocolZ CLOSE_STATUS_CODE_PROTOCOL_ERROR Exception)rTZrequestOrResponserequestprr5r5r6r$s     z"WrappingWebSocketAdapter.onConnectcCs|jdSrR)_protorprqr5r5r6r6rZzWrappingWebSocketAdapter.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}) rrrrZ"CLOSE_STATUS_CODE_UNSUPPORTED_DATArrZ!CLOSE_STATUS_CODE_INVALID_PAYLOADr:rrz)rTrrer5r5r6r9s  "z"WrappingWebSocketAdapter.onMessagecCs|jddSrR)rrwrr5r5r6rErz WrappingWebSocketAdapter.onClosecCsBt|tksJ|jr|j|dddSt|}|j|dddS)NT)rF)r;rrZ sendMessagerr|r5r5r6writeHs zWrappingWebSocketAdapter.writecCs|D]}||qdSrR)r)rTrxdr5r5r6 writeSequenceQs z&WrappingWebSocketAdapter.writeSequencecCrrR)Z sendCloserqr5r5r6r~Vs z'WrappingWebSocketAdapter.loseConnectioncC |jSrR)rlgetPeerrqr5r5r6rZ z WrappingWebSocketAdapter.getPeercCrrR)rlgetHostrqr5r5r6r^rz WrappingWebSocketAdapter.getHostN) rdrerfrgrrrrrrr~rrr5r5r5r6r'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 binaryrrVr4r@autoFragmentSizeFZ failByDropcSs$|D] }t|trt|SqdSrR)r7rr)offersZofferr5r5r6accepts   z7WrappingWebSocketServerFactory.__init__..acceptZperMessageCompressionAcceptN)_factoryrappendr%rUsetProtocolOptions)rTrOrVr4enableCompressionr subprotocolrr5r5r6rUts    z'WrappingWebSocketServerFactory.__init__cC&t}||_|j||_||j_|SrR)r(rOr buildProtocolrrlrTaddrr]r5r5r6r z,WrappingWebSocketServerFactory.buildProtocolcC|jt|dSrR)r startFactoryr%rqr5r5r6r z+WrappingWebSocketServerFactory.startFactorycCrrR)r stopFactoryr%rqr5r5r6rrz*WrappingWebSocketServerFactory.stopFactoryNTrN)rdrerfrgrUrrrr5r5r5r6r*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)ZperMessageCompressionOfferscSst|tr t|SdSrR)r7rr)rr5r5r6rs z7WrappingWebSocketClientFactory.__init__..acceptrN)rrrr&rUrr) rTrOrVr4rrrrrr5r5r6rUs"     z'WrappingWebSocketClientFactory.__init__cCrrR)r)rOrrrrlrr5r5r6rrz,WrappingWebSocketClientFactory.buildProtocolr)rdrerfrgrUrr5r5r5r6r+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 `_. r4rr3NrGrIrJ) rvr4rLrKrHZClientContextFactoryrBrNZ connectTCPZ connectSSLrIrJ)rOrNtimeoutZ bindAddressr4rHconnr5r5r6r-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)rvr4rLrKrZ listenSSLrJZ listenTCP)rOrNbacklogZ 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/rU _protocolsr%rTrOrrrr5r5r6rUKs  z#WampWebSocketServerFactory.__init__N)rdrerfrgr.rrUr5r5r5r6r/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)rrr1rUrr&Znoisyrr5r5r6rUus   z#WampWebSocketClientFactory.__init__N)rdrerfrgr0rrUr5r5r5r6r1mrr1)NrN)Nrr)NrrrtypingrZzope.interfacerrZ use_twistedZtwisted.internet.protocolZtwistedrLrZtwisted.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>rFrQr2ZinternetProtocolr!r"r#objectr$r%Z ServerFactoryr&Z ClientFactoryr'r(r)r*r+r-r,r.r/r0r1r5r5r5r6sp           )77)S?: + #