o @BiC@sddlmZzddlZWn eyddlZYnwddlZddlZddlZddlm Z m Z ddl m Z m Z mZddlmZmZddlZdZedZdZdZd ZGd d d ejZGd d d eZdZddddddZGdddeZGdddeZ GdddeZ!Gddde"Z#Gddde"Z$e Gddde#e$e!Z%e Gd d!d!e#e$e Z&Gd"d#d#e"Z'e Gd$d%d%e'Z(e Gd&d'd'e'Z)dS)()absolute_importN)public_LazyHexFormatter) ProtocolErrorSerializationError TransportLost)peer2strget_serializers)WampRawSocketServerProtocolWampRawSocketClientProtocolWampRawSocketServerFactoryWampRawSocketClientFactoryc@s~eZdZdZeeZdZeZe Z ddZ e ddZddZd d Zd d Zd dZddZddZddZddZdS)PrefixProtocolz!LcCsF||_|d}t||_|jjd|jdd|_d|_t |_ dS)Npeernamez3RawSocker Asyncio: Connection made with peer {peer})peer) transportget_extra_inforrlogdebug_buffer_headertxaio create_future _wait_closed)selfrrr Z/var/www/html/Trade-python/venv/lib/python3.10/site-packages/autobahn/asyncio/rawsocket.pyconnection_madeCs  zPrefixProtocol.connection_madecCs&t|dr|jSt}|d|S)NrT)hasattrrrr set_result)rfr r r! is_closedLs  zPrefixProtocol.is_closedcCs,|jdd|_|jd||dS)Nz"RawSocker Asyncio: Connection lostT)rrrrr$_on_connection_lostrexcr r r!connection_lostUs  zPrefixProtocol.connection_lostcCsdSNr r(r r r!r'[z"PrefixProtocol._on_connection_lostcCs|j||jdSr+)rerrorrclose)rmsgr r r!protocol_error^s zPrefixProtocol.protocol_errorcCsHt|}||jkr tdt|jt|}|j||j|dS)Nz Data too big)lenmax_length_send ValueErrorstructpack prefix_formatrwrite)rdatalheaderr r r! sendStringbs   zPrefixProtocol.sendStringcCtr+NotImplementedErrorrr8r r r!pingjzPrefixProtocol.pingcCr<r+r=r?r r r!pongmrAzPrefixProtocol.pongcCs\|j|7_d}t|j}||jkr|jr|j\}}n:|j|||j}t|ddd@}|tkr;|ddSt|j d|ddd}||j krV|ddS||j|krd|_||j7}||j8}|j|||}||7}||8}|t kr| |n|t kr||n|tkr||n||f|_n||jks|j|d|_dS)NrrzInvalid frame typez Frame too big)rr1 prefix_lengthrordFRAME_TYPE_PONGr0r4unpackr6 max_lengthFRAME_TYPE_DATAstringReceivedFRAME_TYPE_PINGr@rB)rr8pos remaining frame_type frame_lengthr:r r r! data_receivedpsB             "zPrefixProtocol.data_receivedcCr<r+r=r?r r r!rKrAzPrefixProtocol.stringReceivedN)__name__ __module__ __qualname__r6r4calcsizerErIr2r make_loggerrr"propertyr&r*r'r0r;r@rBrQrKr r r r!r;s"   (rc@s<eZdZddZddZddZddZd d Zd d Zd S)RawSocketProtocolcCsZd}|r%ttt|dd}|dkrtdd|d|_||_dSd|_d|_dS)Nr zMaximum length is 16Mr)intmathceilrr3rI _length_exp)rmax_sizeexpr r r!__init__s  zRawSocketProtocol.__init__cCst||d|_dS)NF)rr"_handshake_done)rrr r r!r"s  z!RawSocketProtocol.connection_madecCr<r+r=rr r r!_on_handshake_completerAz(RawSocketProtocol._on_handshake_completecCstt|jdd}|dtkrtd|dd@}|dd?}d|d|_|ddks2|ddkr6td ||fS) NrzInvalid magic byte in handshakerrZrrYzReserved bytes must be zero) bytearrayr MAGIC_BYTEHandshakeErrorr2)rbufserlexpr r r!parse_handshakes   z!RawSocketProtocol.parse_handshakecCr<r+r=rcr r r!process_handshakerAz#RawSocketProtocol.process_handshakec Cs|jjdt|d|jrt||S|j|7_t|jdkr_z|Wnt yC}z| dj |dWYd}~dSd}~wwd|_| |jdd}d|_|rat||dSdSdS)Nz'RawSocker Asyncio: data received {data})r8rezHandshake error : {err}errTr) rrrrbrrQrr1rnrir0formatrd)rr8er r r!rQs(   zRawSocketProtocol.data_receivedN) rRrSrTrar"rdrmrnrQr r r r!rXs  rXzillegal (must not be used)zserializer unsupportedz#maximum message length unacceptablez*use of reserved bits (unsupported feature)z maximum connection count reached)rrrrfrec@seZdZdddZdS)rircCs,t||s |dS|dt|dS)Nz : %s) ExceptionraERRMAPget)rr/coder r r!ras,zHandshakeError.__init__N)r)rRrSrTrar r r r!risric@s0eZdZddZddZeddZddZd S) RawSocketClientProtocolcCdS)NTr rser_idr r r!check_serializerr,z(RawSocketClientProtocol.check_serializercCs>|\}}|dkrtd||j|krtd||jdS)NrzServer returned handshake errorz;Server returned different serializer {0} then requested {1})rmri serializer_idrq)rrzrpr r r!rns    z)RawSocketClientProtocol.process_handshakecCr<r+r=rcr r r!r|sz%RawSocketClientProtocol.serializer_idcCsFt||ttt|jd>|jBddg}|||j ddS)Nrerz(RawSocket Asyncio: Client handshake sent) rXr"bytesrgrhr^r|r7rr)rrhsr r r!r"s  z'RawSocketClientProtocol.connection_madeN)rRrSrTr{rnrWr|r"r r r r!rws   rwc@seZdZddZddZdS)RawSocketServerProtocolcCr<r+r=ryr r r!supports_serializerrAz+RawSocketServerProtocol.supports_serializercsLfdd}\}}|s|tdtdj|d|j|dS)Ncs0|d>|d@B}jttt|ddgdS)NrerZr)rr7r}rgrh)rlrzb2rcr r! send_responses z@RawSocketServerProtocol.process_handshake..send_responserz!Serializer unsupported : {ser_id})rz)rmrERR_SERIALIZER_UNSUPPORTEDrirqr^)rrrz_lexpr rcr!rns    z)RawSocketServerProtocol.process_handshakeN)rRrSrTrrnr r r r!rs rc@s,eZdZddZddZddZddZd S) WampRawSocketMixinGeneralc Csx|jdz|j|_|j|Wnty3}z|jjd|d|WYd}~dSd}~ww|j ddS)Nz)WampRawSocketProtocol: Handshake completezMWampRawSocketProtocol: ApplicationSession constructor / onOpen raised ({err})rozApplicationSession started.) rrfactory_factory_sessiononOpenrswarnabortinfo)rrrr r r!rds  z0WampRawSocketMixinGeneral._on_handshake_completec Cs|jjdt|dz|j|D]}|jjd|d|j|qWdStyC}z|jjd|d| WYd}~dSd}~wt ya}z|jjd|d| WYd}~dSd}~ww)Nz*WampRawSocketProtocol: RX octets: {octets}octetsz-WampRawSocketProtocol: RX WAMP message: {msg}r/zHWampRawSocketProtocol: WAMP Protocol Error ({err}) - aborting connectionrozHWampRawSocketProtocol: WAMP Internal Error ({err}) - aborting connection) rrr _serializer unserializer onMessagerrrrs)rpayloadr/rrr r r!rKsz(WampRawSocketMixinGeneral.stringReceivedc Cs||r;|jjd|dz |j|\}}Wnty)}ztd|d}~ww|||jjdt |ddSt )zM Implements :func:`autobahn.wamp.interfaces.ITransport.send` z-WampRawSocketProtocol: TX WAMP message: {msg}rzIWampRawSocketProtocol: unable to serialize WAMP application payload ({0})Nz*WampRawSocketProtocol: TX octets: {octets}r) isOpenrrr serializersrrqr;rr)rr/r_rrr r r!send-s zWampRawSocketMixinGeneral.sendcCst|do |jduS)zO Implements :func:`autobahn.wamp.interfaces.ITransport.isOpen` rN)r#rrcr r r!r?sz WampRawSocketMixinGeneral.isOpenN)rRrSrTrdrKrrr r r r!rs   rc@s(eZdZdZddZddZddZdS) WampRawSocketMixinAsynciozE Base class for asyncio-based WAMP-over-RawSocket protocols. c CsXz |du}|j|Wnty&}z|jjd|dWYd}~nd}~wwd|_dS)Nz@WampRawSocketProtocol: ApplicationSession.onClose raised ({err})ro)ronClosersrr)rr)wasCleanrrr r r!r'Ls z-WampRawSocketMixinAsyncio._on_connection_lostcCs|r |jdSt)zN Implements :func:`autobahn.wamp.interfaces.ITransport.close` N)rrr.rrcr r r!r.UszWampRawSocketMixinAsyncio.closecCs6|rt|jdr|jdS|jdSt)zN Implements :func:`autobahn.wamp.interfaces.ITransport.abort` rN)rr#rrr.rrcr r r!r^s  zWampRawSocketMixinAsyncio.abortN)rRrSrT__doc__r'r.rr r r r!rGs  rc@s"eZdZdZddZdddZdS) r z asyncio-based WAMP-over-RawSocket server protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` cCsZ||jjvrt|jj||_|jjd|ddS|jjd||jjd|dS)NzDWampRawSocketProtocol: client wants to use serializer '{serializer}') serializerTzWampRawSocketProtocol: opening handshake - no suitable serializer found (client requested {serializer}, and we have {serializers})r serializersF)r _serializerscopyrrrkeysrryr r r!rvs  z/WampRawSocketServerProtocol.supports_serializer tls-uniquecCrxzW Implements :func:`autobahn.wamp.interfaces.ITransport.get_channel_id` Nr rchannel_id_typer r r!get_channel_idz*WampRawSocketServerProtocol.get_channel_idNr)rRrSrTrrrr r r r!r lsr c@s&eZdZdZeddZdddZdS) r z asyncio-based WAMP-over-RawSocket client protocol. Implements: * :class:`autobahn.wamp.interfaces.ITransport` cCs"t|ds t|jj|_|jjS)Nr)r#rrrRAWSOCKET_SERIALIZER_IDrcr r r!r|s z)WampRawSocketClientProtocol.serializer_idrcCrxrr rr r r!rrz*WampRawSocketClientProtocol.get_channel_idNr)rRrSrTrrWr|rr r r r!r s  r c@s$eZdZdZeZeddZdS)WampRawSocketFactoryzm Adapter class for asyncio-based WebSocket client and server factories.def dataReceived(self, data): cCs|}||_|Sr+)protocolr)rprotor r r!__call__szWampRawSocketFactory.__call__N) rRrSrTrrrVrrrr r r r!rs rc@eZdZdZeZdddZdS)r zD asyncio-based WAMP-over-RawSocket server protocol factory. NcsLtr|_nfdd|_|durt}|stddd|D|_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` cSr+r r rr r!z5WampRawSocketServerFactory.__init__..Nz%could not import any WAMP serializerscSsi|]}|j|qSr )r).0rkr r r! sz7WampRawSocketServerFactory.__init__..)callablerr rsr)rrrr rr!ras z#WampRawSocketServerFactory.__init__r+)rRrSrTrr rrar r r r!r r c@r)r z; asyncio-based WAMP-over-RawSocket client factory. NcsTtr|_nfdd|_|durt}|r|d}|dur%td||_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` crr+r r rr r!rrz5WampRawSocketClientFactory.__init__..Nrz$could not import any WAMP serializer)rrr rsr)rrrrr rr!ras   z#WampRawSocketClientFactory.__init__r+)rRrSrTrr rrar r r r!r rr )* __future__rasyncio ImportErrortrolliusr4r\r autobahn.utilrrautobahn.wamp.exceptionrrrautobahn.asyncio.utilrr r__all__ use_asynciorJrLrGrhProtocolrrXrrtrsrirwrobjectrrr r rr r r r r r!sT    a8 7%""