o @Bihz@sddlZddlZddlZddlZddlmZmZddlmZm Z ddl m Z ddl m Z mZddlmZddlZddlZddlZddlZddlZddlZddl mZmZmZGd d d eZGd d d eZGd ddeZdS)N)RegisterOptions CallDetails)ApplicationError TransportLost)ApplicationSession)unpack_uint256 pack_uint256)time_ns)hlrecover_eip712_signersign_eip712_datac@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|dksJt|tkr$|dks&J|dus0t|s0J||_||_||_||_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: Key rotation interval in seconds. :type interval: int :param on_rotate: Optional user callback fired after key was rotated. :type on_rotate: callable rN) typebyteslenintcallable_api_id_price _interval _on_rotate_id_key_box_archive)selfapi_idpriceinterval on_rotater"T/var/www/html/Trade-python/venv/lib/python3.10/site-packages/autobahn/xbr/_seller.py__init__8s 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 )rrr"r"r#key_idVs zKeySeries.key_idcCs"t|}|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 cbor)cbor2dumpsrencryptr)rpayloaddata ciphertextr"r"r#r*as  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) rrrrnaclpublic SealedBox PublicKeyencoding RawEncoderr*)rr& buyer_pubkeykey_ sendkey_box encrypted_keyr"r"r# encrypt_keypszKeySeries.encrypt_keycCtNNotImplementedErrorr%r"r"r#startzKeySeries.startcCr<r=r>r%r"r"r#stoprAzKeySeries.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)osurandomrr0utilsrandomsecret SecretBoxKEY_SIZErrrloginfor uuidUUIDrrr%r"r"r#_rotates  zKeySeries._rotater=) __name__ __module__ __qualname____doc__r$propertyr&r*r;r@rBrSr"r"r"r#r2s    rc@seZdZddZdS) PayingChannelcCsbt|tkr t|dksJt|tkr|dksJt|tkr$|dks&J||_||_||_dS)Nr)rrrr_adr_seq_balance)radrseqbalancer"r"r#r$s  zPayingChannel.__init__N)rTrUrVr$r"r"r"r#rYs rYc@s~eZdZdZdZdZdZdZdZdZ dddZ e d d Z dd d Z d dZddZddZddZ dddZdddZdS) SimpleSellerNrrcCst|tkr t|dksJd|t|tkrt|dks&Jd||dus7t|tks7Jd|t|_tj |_ ||_ ||_ t j||_d|_|jj|_d|_|pat|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 rZz0market_maker_adr must be bytes[20], but got "{}"r.z/seller delegate must be bytes[32], but got "{}"Nz0provider_id must be None or string, but got "{}"r)rrrformatstrtxaio make_loggerrOra STATE_NONE_state_market_maker_adr _pkey_raweth_keyskeys PrivateKey_pkey_acct public_keyto_canonical_address_addr_caddr _provider_id _channels_channelr]r\_keys _keys_map_session _session_regs)rmarket_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 )rprrr%r"r"r#rrszSimpleSeller.public_keycsttkrtdkrjvsJttkrdksJt|tkr)|dks+JdusKttkrGddDrGddDsKJdfdd }||}|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) in which to auto-rotate the encryption key. :type interval: int rrNcs|] }t|tkVqdSr=rrf).0kr"r"r# z#SimpleSeller.add..csrr=r)rvr"r"r#rrzZinvalid 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 jd|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 rCrDrFNZor XBR)rGr&rrdelegateprefixz 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&rzr rtrHrIrvr{callr rOrPr rQrRrfrbinasciib2a_hexdecodedebugrerrorwarnfailurerasynciosleep) key_seriesr&r valid_fromr signaturerrerrrrrr"r#r!sf     &        z#SimpleSeller.add..on_rotatez#Created new key series {key_series})r) rrrryrdictrnvaluesrrOrP)rrrrr rr!rr"rr#adds& @= zSimpleSeller.addcst|ts Jd||jtjtjfvsJdtj|_||_g|_ |j j dt |j t|jjdddd|j}|j|j|tdd d IdH}|j ||j jd t |jd d |j}|j|j|tdd d IdH}|j ||j jd t |jd |jD] }|IdHq|d|jIdH}|d|dIdH}t |dt!krt"|d|d<|j j dt dt|ddt#|d|d|d|j$|d<tj%|_||_&|d|_'t |j't!krt"|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 )addressrrzxbr.provider.{}.selldetails) details_arg)optionsz"Registered procedure "{procedure}") procedurezxbr.provider.{}.close_channelz)xbr.marketmaker.get_active_paying_channelz*xbr.marketmaker.get_paying_channel_balancechannel remainingzIDelegate has currently active paying channel address {paying_channel_adr}0x)paying_channel_adrr_)) isinstancerrerjrari STATE_STOPPEDSTATE_STARTINGr{r|rOrPr rurrrprrrrvregistersellrappendrr close_channelryrr@rrtrrrrYrw STATE_STARTEDrxr]r\)rsessionrregrrpaying_balancer"r"r#r@]sF    "   zSimpleSeller.startcs|jtjfvs Jdtj|_g}|jD] }|}||q|jr@|j r=|j r=|jD] }| }||q1d|_t |}z%z|IdHWn|jYWtj|_d|_ dSWtj|_d|_ dStj|_d|_ w)zF Stop rotating/offering keys to the XBR market maker. seller not runningN)rjrarSTATE_STOPPINGryrrBrr|r{ is_attached unregisterrggatherrOrr)rdlrdrr"r"r#rBs2      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 attachedz+xbr.marketmaker.get_payment_channel_balancerN)rjrar RuntimeErrorr{rrrx)rrr"r"r#r`s zSimpleSeller.balancecsht|tkrt|dkr||jvsJt|tksJ|dus"J|j|}||\}}}|||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)rrrryrfr*)rrurir+ keyseriesr& serializerr-r"r"r#wraps&   zSimpleSeller.wrapc Cst|tkr t|dksJdt|t|tkr!t|dks*Jdt|t|tks9Jdt|t|tkrEt|dksNJdt|t|tks]Jdt|t|tkrit|dksrJd 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|dd ddt|tt |t|jt|j d!| S)"zI Called by a XBR Market Maker to close a paying channel. rZz.market_maker_adr must be bytes[20], but was {}z)channel_adr must be bytes[20], but was {}z#channel_seq must be int, but was {}r.z-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_adrJ{}.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)rr_r`is_finalrz{klass}.close_channel() - {tx_type} closing channel {channel_adr}, closing balance {channel_balance}, closing sequence {channel_seq} [caller={caller}, caller_authid="{caller_authid}"]z XBR CLOSE rCrDrr)rrGchannel_balance channel_seq channel_adrcaller caller_authid)!rrrrerboolrrrkr __class__rTrrrrxr{leaver\rr]r rOrr r rlrtr rPrfrr) rr}rrrchannel_is_finalmarketmaker_signaturerrseller_signaturereceiptr"r"r#rsd**** & *     zSimpleSeller.close_channelc  Cst|tkr t|dksJdt|tkrt|dks Jdt|tkr,t|dks0Jdt|tkrs&  v