U uc @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|tkrt|dkstt|tkr0|dks4t|dksTt|tkrP|dksTt|dkstt|tkrp|dkstt|dkr|dk s|dk r|dkst|dkst|st||_||_||_||_ 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)typebyteslenAssertionErrorintcallable_api_idZ_priceZ _interval_count_count_current _on_rotate_id_key_box_archive)selfapi_idpriceintervalcount on_rotater&W/var/www/html/VPFITNESS/school/venv/lib/python3.8/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_idcsZt|}|jdk rB|jd7_|j|jkrB|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 NrrZcbor)cbor2dumpsrr_rotaterencryptr)r payloaddata ciphertextr&r&r'r.gs    zKeySeries.encryptcCsxt|tkrt|dkstt|tkr4t|dks8t|j|\}}tjtjj|tj j d}|j |tj j 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) rrrrrnaclpublicZ SealedBoxZ PublicKeyencodingZ RawEncoderr.)r r* buyer_pubkeykey_Z sendkey_boxZ encrypted_keyr&r&r' encrypt_key}szKeySeries.encrypt_keycCs tdSNNotImplementedErrorr)r&r&r'startszKeySeries.startcCs tdSr;r<r)r&r&r'stopszKeySeries.stopcstd|_tjtjjj|_ tj|j |_ |j |j f|j |j<|j j dtdddttj|jdttj|jdd|jr||IdHdS)Nrz4{tx_type} key "{key_id}" rotated [api_id="{api_id}"]z XBR ROTATEmagentaZcolorr)tx_typer*r!)osurandomrr4utilsrandomZsecretZ SecretBoxZKEY_SIZErrrlogdebugr uuidUUIDrrr)r&r&r'r-s  zKeySeries._rotate)NNN) __name__ __module__ __qualname____doc__r(propertyr*r.r:r>r?r-r&r&r&r'r1s % rc@seZdZddZdS) PayingChannelcCsbt|tkrt|dkstt|tkr0|dks4tt|tkrH|dksLt||_||_||_dS)Nrr)rrrrrZ_adr_seq_balance)r Zadrseqbalancer&r&r'r(s zPayingChannel.__init__N)rLrMrNr(r&r&r&r'rQsrQc@s|eZdZdZdZdZdZdZdZdZ dddZ e d d Z dd d Z d dZddZddZddZdddZdddZdS) SimpleSellerNrrr cCst|tkrt|dks&td|t|tkr>t|dksLtd||dksnt|tksntd|t|_t j |_ ||_ d|_ ||_tj||_d|_|jj|_d|_|pt|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 "{}"r2z/seller delegate must be bytes[32], but got "{}"Nz0provider_id must be None or string, but got "{}"r)!rrrrformatstrtxaioZ make_loggerrHrV STATE_NONE_state_market_maker_adr _xbrmm_config _pkey_raweth_keyskeysZ PrivateKey_pkeyZ_acct public_keyZto_canonical_address_addr_caddr _provider_id _channels_channelrSrR_keys _keys_map_session _session_regs)r market_maker_adrZ 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 )rdrer)r&r&r're szSimpleSeller.public_keyc s(ttkr"tdkr"jks&tttkr:dks>t|dks^t|tkrZ|dks^t|dks~t|tkrz|dks~t|dkr|dk s|dk r|dkstdksttkrddDrddDstdfdd }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 rrNcss|]}t|tkVqdSr;rr[).0kr&r&r' +sz#SimpleSeller.add..css|]}t|tkVqdSr;rq)rrvr&r&r'rt+szZinvalid categories type (must be dict) or category key or value type (must both be string)c s|j}|j|<d}|rztd}j}td}j}jjd||||ddk rbt nddd|d IdH}j j dt ddd t t j|d t t jd t tdk rtd nd d dd t t|t dj j d|dWqWntk rR}z2|jdkr.j dnj WYqW5d}~XYn<tk rxj dYqYnj YnX|d8}j jd|dtdIdHqdS)Nld( Azxbr.marketmaker.place_offer)Zprivkeyr" categoriesZexpiresZcopiesrpzh{tx_type} key "{key_id}" offered for {price} [api_id={api_id}, prefix="{prefix}", delegate="{delegate}"]z XBR OFFER r@rArBNZor XBR)rCr*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*rlr rfrDrErhrmcallr rHrIr rJrKr[rbinasciib2a_hexdecodererrorwarnfailurerasynciosleep) key_seriesr*r~Z valid_fromr{ signaturerpr}er!rxr|r"r r&r'r%-s^     &      z#SimpleSeller.add..on_rotate)r#r$r%z#Created new key series {key_series})r) rrrrkrrdictrcvaluesrrHrI) r r!r|r"r#r$rxr%rr&rr'adds&  $@= zSimpleSeller.addcsdt|tstd||jtjtjfks2tdtj|_||_ g|_ |j j dt |jt|jjddd|d|jIdH|_|jstd|jd }t|tkrt|d ksttj|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]}|'IdHqn|dIdH|_(|d|jjIdH}t|dtkrt)|d|d<|ddkstdt*||d|d|j+|<tj,|_|d|_-t|j-tkr,t)|j-|_-|d|_.|j j/dt |jt0|j-t0|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 )addressrez)xbr.marketmaker.get_active_paying_channelzno active paying channel found channel_oidrrBzxbr.provider.{}.selldetails)Z 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 channelrTzvOk, seller delegate started [active paying channel {channel_oid} with remaining balance {remaining} at sequence {seq}])rrrT)1 isinstancerrrZr^rVr] STATE_STOPPEDSTATE_STARTINGrmrnrHrIr rgrrrdrerrrfrj ExceptionrrrrJrKZ _channel_oidrhregistersellrappendr close_channelrkrr>r`rrQri STATE_STARTEDrSrRinfor )r sessionrrregrpaying_balancer&r&r'r>psV        zSimpleSeller.startcs|jtjfkstdtj|_g}|jD]}|}||q,|j r~|j rx|j rx|j D]}| }||q`d|_ t |}z*z|IdHWn|jYnXW5tj|_d|_ X|jddS)zF Stop rotating/offering keys to the XBR market maker. seller not runningNzOk, seller delegate stopped.)r^rVrrSTATE_STOPPINGrkrr?rrnrm is_attached unregisterr\gatherrrHrr)r Zdlrdrr&r&r'r?s(    zSimpleSeller.stopcsL|jtjfkrtd|jr&|js.td|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)r^rVr RuntimeErrorrmrrrj)r rr&r&r'rUs zSimpleSeller.balancecslt|tkr"t|dkr"||jks&tt|tks6t|dk sBt|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)rrrrkrr[r.)r r!Zurir/Z keyseriesr*Z serializerr1r&r&r'wraps &  zSimpleSeller.wrapc Cst|tkrt|dks*tdt|t|tkrBt|dksTtdt|t|tksrtdt|t|tkrt|dkstdt|t|tkstdt|t|tkrt|d kstd t||d kst|tstd ||j kr8t d d|j j t |j t |||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|j t|j!d"| S)#zI Called by a XBR Market Maker to close a paying channel. rYz.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 {}r2z-channel_balance must be bytes[32], but was {}z)channel_is_final must be bool, but was {}rwz3marketmaker_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_addressZ delegate_adrxbr.error.invalid_signatureO{}.sell()::XBRSIG[4/8] - EIP712 signature invalid or not signed by market maker)r{rTrUZis_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 r@rAryrz)rrCchannel_balance channel_seqrcaller caller_authid)"rrrrrZrboolrrr_r __class__rLrrrrjrmleaverRrrSrrHrr rrarfr rIr[rr) r rorrrZchannel_is_finalZmarketmaker_signaturerrseller_signaturereceiptr&r&r'rsd**** & *     zSimpleSeller.close_channelc  Cst|tkrt|dks tdt|tkr8t|dks@tdt|tkrXt|dks`tdt|tkrxt|dkstdt|tkstdt|tkrt|dkstd t|t|tkrt|dkstd t|t|tkrt|d ks td | d ks*t| ts*tdt|}t|}||j krtt dd|j j t |j t |||jkrt dd|j j ||j|} ||jdkr|jt dd|j j t |jdt |||jdkrt dd|j j |jd|||j|krHt dd|j j |j||d} |jd} t |jddd } |jd}t| | | ||||d| }||kr|jjd |j j tt |tt |d!t d"d#|j j |jd7_|j|8_| ||}t|tkr&t|d$ks:td%|j j |t|j| | | |||j|jd }||j|||j||j|d&}|jj d'|j j td(d)d*tt!j"|d+tt#t|d,d-d)d*tt#t|jd,d-d)d*t| j$t| j%tt |d. |S)/al Called by a XBR Market Maker to buy a data encyption key. The XBR Market Maker here is acting for (triggered by) the XBR buyer delegate. :param market_maker_adr: The market maker Ethereum address. The technical buyer is usually the XBR market maker (== the XBR delegate of the XBR market operator). :type market_maker_adr: bytes of length 20 :param buyer_pubkey: The buyer delegate Ed25519 public key. :type buyer_pubkey: bytes of length 32 :param key_id: The UUID of the data encryption key to buy. :type key_id: bytes of length 16 :param channel_oid: The on-chain channel contract address. :type channel_oid: bytes of length 16 :param channel_seq: Paying channel sequence off-chain transaction number. :type channel_seq: int :param amount: The amount paid by the XBR Buyer via the XBR Market Maker. :type amount: bytes :param balance: Balance remaining in the payment channel (from the market maker to the seller) after successfully buying the key. :type balance: bytes :param signature: Signature over the supplied buying information, using the Ethereum private key of the market maker (which is the delegate of the marker operator). :type signature: bytes of length 65 :param details: Caller details. The call will come from the XBR Market Maker. :type details: :class:`autobahn.wamp.types.CallDetails` :return: The data encryption key, itself encrypted to the public key of the original buyer. :rtype: bytes rYzdelegate_adr must be bytes[20]r2zbuyer_pubkey must be bytes[32]rzkey_id must be bytes[16]zchannel_oid must be bytes[16]zchannel_seq must be intz)amount_paid must be bytes[32], but was {}z*post_balance must be bytes[32], but was {}rwzsignature must be bytes[65]Nrz$xbr.error.unexpected_marketmaker_adrzH{}.sell() - unexpected market maker address: expected 0x{}, but got 0x{}zcrossbar.error.no_such_objectz{}.sell() - no key with ID "{}"rrrrrrrrverifying_chain_idverifying_contract_adrr market_oidFrrrrPzC{}.sell() - unexpected sealed key computed (expected bytes[80]): {})r*r{r7 sealed_keyramountrUrz{klass}.sell() - {tx_type} key "{key_id}" sold for {amount_earned} - balance is {balance} [caller={caller}, caller_authid="{caller_authid}", buyer_pubkey="{buyer_pubkey}"]z XBR SELL r@rArBryrz)rrCr*Z amount_earnedrUrrr7)&rrrrrrZrrrr_rrrLrrrrlrjrmrrRrSr`a2b_hexrrHrr r:rrarfrrJrKr[rr)r ror7r*rrrrUrrrZcurrent_block_numberrrrrrrrr&r&r'rBs&    **$ &   *    0   zSimpleSeller.sell)N)NNN)N)N)rLrMrNrHrr]rrrrr(rPrerr>r?rUrrrr&r&r&r'rVs$ B ZA! BrV)!rrrDrJZautobahn.wamp.typesrrZautobahn.wamp.exceptionrrZautobahn.wamp.protocolrZ_utilrr r\r r+rbZ nacl.secretr4Z nacl.utilsZ nacl.publicutilr r Z_eip712_channel_closerrobjectrrQrVr&r&r&r's(