o >h @sddlZddlZddlZddlZddlmZmZddlmZm Z ddl m Z ddl m Z mZddlmZddlZddlZddlZddlZddlZddlZdd lmZmZdd lmZmZGd d d eZGd ddeZGdddeZ dS)N)RegisterOptions CallDetails)ApplicationError TransportLost)ApplicationSession)unpack_uint256 pack_uint256)time_ns)hlhlval)sign_eip712_channel_closerecover_eip712_channel_closec@sNeZdZdZdddZeddZddZd d Zd d Z d dZ ddZ dS) KeySeriesz} Data encryption key series with automatic (time-based) key rotation and key offering (to the XBR market maker). NcCst|tkr t|dksJt|tkr|dksJ|dus*t|tkr(|dks*J|dus:t|tkr8|dks:J|durB|dusL|durJ|dusLJ|dusVt|sVJ||_||_||_||_d|_ ||_ d|_ d|_ d|_ i|_dS)a :param api_id: ID of the API for which to generate keys. :type api_id: bytes :param price: Price per key in key series. :type price: int :param interval: Interval in seconds after which to auto-rotate key. :type interval: int :param count: Number of encryption operations after which to auto-rotate key. :type count: int :param on_rotate: Optional user callback fired after key was rotated. :type on_rotate: callable rN)typebyteslenintcallable_api_id_price _interval_count_count_current _on_rotate_id_key_box_archive)selfapi_idpriceintervalcount on_rotater't/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/autobahn/xbr/_seller.py__init__7s   $ zKeySeries.__init__cCs|jS)z Get current XBR data encryption key ID (of the keys being rotated in a series). :return: Current key ID in key series (16 bytes). :rtype: bytes )rr!r'r'r(key_id\s zKeySeries.key_idcs\t|}|jdur"|jd7_|j|jkr"|IdHd|_|j|}|jd|fS)z Encrypt data with the current XBR data encryption key. :param payload: Application payload to encrypt. :type payload: object :return: The ciphertext for the encrypted application payload. :rtype: bytes Nrrcbor)cbor2dumpsrr_rotaterencryptr)r!payloaddata ciphertextr'r'r(r0gs    zKeySeries.encryptcCsxt|tkr t|dksJt|tkrt|dksJ|j|\}}tjtjj|tjj d}|j |tjj d}|S)a Encrypt a (previously used) XBR data encryption key with a buyer public key. :param key_id: ID of the data encryption key to encrypt. :type key_id: bytes :param buyer_pubkey: Buyer WAMP public key (Ed25519) to asymmetrically encrypt the data encryption key (selected by ``key_id``) against. :type buyer_pubkey: bytes :return: The ciphertext for the encrypted data encryption key. :rtype: bytes r )encoder) rrrr naclpublic SealedBox PublicKeyencoding RawEncoderr0)r!r+ buyer_pubkeykey_ sendkey_box encrypted_keyr'r'r( encrypt_key}szKeySeries.encrypt_keycCtNNotImplementedErrorr*r'r'r(startzKeySeries.startcCrBrCrDr*r'r'r(stoprGzKeySeries.stopcstd|_tjtjjj|_ tj|j |_ |j |j f|j |j<|j j dtdddttj|jdttj|jdd|jrL||IdHdSdS)Nrz4{tx_type} key "{key_id}" rotated [api_id="{api_id}"]z XBR ROTATEmagentacolorr)tx_typer+r")osurandomrr6utilsrandomsecret SecretBoxKEY_SIZErrr logdebugr uuidUUIDrrr*r'r'r(r/s  zKeySeries._rotateNNN) __name__ __module__ __qualname____doc__r)propertyr+r0rArFrHr/r'r'r'r(r1s %   rc@seZdZddZdS) PayingChannelcCsbt|tkr t|dksJt|tkr|dksJt|tkr$|dks&J||_||_||_dS)Nrr)rrrr_adr_seq_balance)r!adrseqbalancer'r'r(r)s  zPayingChannel.__init__N)rZr[r\r)r'r'r'r(r_s r_c@s~eZdZdZdZdZdZdZdZdZ dddZ e d d Z dd d Z d dZddZddZddZ dddZdddZdS) SimpleSellerNrrr cCst|tkr t|dksJd|t|tkrt|dks&Jd||dus7t|tks7Jd|t|_tj |_ ||_ d|_ ||_ tj||_d|_|jj|_d|_|pdt|jj|_i|_d|_d|_d|_i|_i|_d|_d|_dS)a :param market_maker_adr: Market maker public Ethereum address (20 bytes). :type market_maker_adr: bytes :param seller_key: Seller (delegate) private Ethereum key (32 bytes). :type seller_key: bytes :param provider_id: Optional explicit data provider ID. When not given, the seller delegate public WAMP key (Ed25519 in Hex) is used as the provider ID. This must be a valid WAMP URI part. :type provider_id: string z0market_maker_adr must be bytes[20], but got "{}"r4z/seller delegate must be bytes[32], but got "{}"Nz0provider_id must be None or string, but got "{}"r) rrrformatstrtxaio make_loggerrUrf STATE_NONE_state_market_maker_adr _xbrmm_config _pkey_raweth_keyskeys PrivateKey_pkey_acct public_keyto_canonical_address_addr_caddr _provider_id _channels_channelrbra_keys _keys_map_session _session_regs)r!market_maker_adr seller_key provider_idr'r'r(r)s*& &"  zSimpleSeller.__init__cCs|jjS)z This seller delegate public Ethereum key. :return: Ethereum public key of this seller delegate. :rtype: bytes )rvrxr*r'r'r(rx szSimpleSeller.public_keyc s(ttkrtdkrjvsJttkrdksJ|dus/t|tkr-|dks/J|dus?t|tkr=|dks?J|durG|dusQ|durO|dusQJdusqttkrmddDrmddDsqJdfdd }j|||d }|j<j j d |d |S) a Add a new (rotating) private encryption key for encrypting data on the given API. :param api_id: API for which to create a new series of rotating encryption keys. :type api_id: bytes :param price: Price in XBR token per key. :type price: int :param interval: Interval (in seconds) after which to auto-rotate the encryption key. :type interval: int :param count: Number of encryption operations after which to auto-rotate the encryption key. :type count: int rrNcs|] }t|tkVqdSrCrrk).0kr'r'r( +z#SimpleSeller.add..csrrCr)rvr'r'r(r+rzZinvalid categories type (must be dict) or category key or value type (must both be string)c s|j}|j|<d}|rzqtd}j}td}j}jjd||||ddur1t nddd|d IdH}j j dt ddd t t j|d t t jd t tdur_td nd d dd t t|t dj j d|dWdSty}z|jdkrj dn j WYd}~dSWYd}~nd}~wtyj dYdSj Y|d8}j jd|dtdIdH|s dSdS)Nld( Azxbr.marketmaker.place_offer)privkeyr# categoriesexpirescopiesrzh{tx_type} key "{key_id}" offered for {price} [api_id={api_id}, prefix="{prefix}", delegate="{delegate}"]z XBR OFFER rIrJrLNZor XBR)rMr+r"r#delegateprefixz offer={offer})offerzwamp.error.no_such_procedurez-xbr.marketmaker.offer: procedure unavailable!z2TransportLost while calling xbr.marketmaker.offer!rz6Failed to place offer for key! Retrying {retries}/5 ..)retries)r+rr rzrNrOr|rcallr rUrVr rWrXrkrbinasciib2a_hexdecodererrorwarnfailurerasynciosleep) key_seriesr+r valid_fromr signaturerrer"rrr#r!r'r(r&-sf     &        z#SimpleSeller.add..on_rotate)r$r%r&z#Created new key series {key_series})r) rrrrrdictrtvaluesrrUrV) r!r"rr#r$r%rr&rr'rr(adds&  $@= zSimpleSeller.addcs`t|ts Jd||jtjtjfvsJdtj|_||_g|_ |j j dt |j t|jjddd|d|jIdH|_|jsMtd|jd }t|tkr^t|d ks`Jtj|d |_d |j}|j|j|t d ddIdH}|j !||j j dt |j"dd|j}|j|j#|t d ddIdH}|j !||j j dt |j"d|j$%D] }|&IdHq|dIdH|_'|d|jjIdH}t|dtkrt(|d|d<|ddkstdt)||d|d|j*|<tj+|_|d|_,t|j,tkrt(|j,|_,|d|_-|j j.dt |jt/|j,t/|j-d|dS)z Start rotating keys and placing key offers with the XBR market maker. :param session: WAMP session over which to communicate with the XBR market maker. :type session: :class:`autobahn.wamp.protocol.ApplicationSession` z/session must be an ApplicationSession, was "{}"zseller already runningzRStart selling from seller delegate address {address} (public key 0x{public_key}..)N )addressrxz)xbr.marketmaker.get_active_paying_channelzno active paying channel found channel_oidrrLzxbr.provider.{}.selldetails) details_arg)optionsz"Registered procedure "{procedure}") procedurezxbr.provider.{}.close_channelzxbr.marketmaker.get_config*xbr.marketmaker.get_paying_channel_balance remainingrz0no off-chain balance remaining on paying channelrdzvOk, seller delegate started [active paying channel {channel_oid} with remaining balance {remaining} at sequence {seq}])rrrd)0 isinstancerrjrorfrn STATE_STOPPEDSTATE_STARTINGrrrUrVr r{rrrvrxrrrzr~ ExceptionrrrrWrX _channel_oidr|registersellrappendr close_channelrrrFrqrr_r} STATE_STARTEDrbrainfor )r!sessionrrregrpaying_balancer'r'r(rFpsT         zSimpleSeller.startcs|jtjfvs Jdtj|_g}|jD] }|}||q|jr@|j r=|j r=|jD] }| }||q1d|_t |}zz|IdHWn |jYWtj|_d|_ ntj|_d|_ w|jddS)zF Stop rotating/offering keys to the XBR market maker. seller not runningNzOk, seller delegate stopped.)rorfrSTATE_STOPPINGrrrHrrr is_attached unregisterrlgatherrUrrr)r!dlrdrr'r'r(rHs0     zSimpleSeller.stopcsN|jtjfvr td|jr|jstd|jd|jdIdH}|S)a Return current (off-chain) balance of paying channel: * ``amount``: The initial amount with which the paying channel was opened. * ``remaining``: The remaining amount of XBR in the paying channel that can be earned. * ``inflight``: The amount of XBR allocated to sell transactions that are currently processed. :return: Current paying balance. :rtype: dict rz!market-maker session not attachedrrN)rorfr RuntimeErrorrrrr~)r!rr'r'r(res zSimpleSeller.balancecsnt|tkrt|dkr||jvsJt|tksJ|dus"J|j|}||IdH\}}}|||fS)aG Encrypt and wrap application payload for a given API and destined for a specific WAMP URI. :param api_id: API for which to encrypt and wrap the application payload for. :type api_id: bytes :param uri: WAMP URI the application payload is destined for (eg the procedure or topic URI). :type uri: str :param payload: Application payload to encrypt and wrap. :type payload: object :return: The encrypted and wrapped application payload: a tuple with ``(key_id, serializer, ciphertext)``. :rtype: tuple rN)rrrrrkr0)r!r"urir1 keyseriesr+ serializerr3r'r'r(wraps&   zSimpleSeller.wrapc Cst|tkr t|dksJdt|t|tkr!t|dks*Jdt|t|tks9Jdt|t|tkrEt|dksNJdt|t|tks]Jdt|t|tkrit|d ksrJd t||d ust|tsJd ||jkrt d d|j j t |jt |||jdkr|jt dd|j j t |jdt |||jkrt dd|j j |jd|t|}||jkrt dd|j j |j|t|||||}||kr|jjd|j j tt |tt |dt dd|j j t|j||||} |j|t||| d} |jjd|j j tdddttt|d d!ddt|tt |t|jt|j d"| S)#zI Called by a XBR Market Maker to close a paying channel. riz.market_maker_adr must be bytes[20], but was {}rz)channel_oid must be bytes[16], but was {}z#channel_seq must be int, but was {}r4z-channel_balance must be bytes[32], but was {}z)channel_is_final must be bool, but was {}rz3marketmaker_signature must be bytes[65], but was {}N/details must be autobahn.wamp.types.CallDetailsz!xbr.error.unexpected_delegate_adrzS{}.sell() - unexpected market maker (delegate) address: expected 0x{}, but got 0x{}r xbr.error.unexpected_channel_oidJ{}.sell() - unexpected paying channel address: expected 0x{}, but got 0x{} xbr.error.unexpected_channel_seqR{}.sell() - unexpected channel (after tx) sequence number: expected {}, but got {}r$xbr.error.unexpected_channel_balanceJ{}.sell() - unexpected channel (after tx) balance: expected {}, but got {}t{klass}.sell()::XBRSIG[4/8] - EIP712 signature invalid: signer_address={signer_address}, delegate_adr={delegate_adr}klasssigner_address delegate_adrxbr.error.invalid_signatureO{}.sell()::XBRSIG[4/8] - EIP712 signature invalid or not signed by market maker)rrdreis_finalrz{klass}.close_channel() - {tx_type} closing channel {channel_oid}, closing balance {channel_balance}, closing sequence {channel_seq} [caller={caller}, caller_authid="{caller_authid}"]z XBR CLOSE rIrJrr)rrMchannel_balance channel_seqrcaller caller_authid)!rrrrjrboolrrrpr __class__rZrrrr~rleaverarrbrrUrr rrrrzr rVrkrr) r!rrrrchannel_is_finalmarketmaker_signaturerrseller_signaturereceiptr'r'r(rsd**** & *     zSimpleSeller.close_channelc  Cst|tkr t|dksJdt|tkrt|dks Jdt|tkr,t|dks0Jdt|tkrs*