U uc m@sddlZddlZddlmZddlZddlZddlZddlZddl Zddl Zddl Z ddl m Z ddlmZddlmZmZddlZddlmZmZdd lmZmZGd d d eZGd d d eZdS)N)pformat)ApplicationError)ApplicationSession)unpack_uint256 pack_uint256)hlhlval)sign_eip712_channel_closerecover_eip712_channel_closec@s$eZdZddZddZddZdS) Transactionc Cs4||_||_||_||_||_||_||_||_dSNchanneldelegatepubkeykey_id channel_seqamountbalance signature) selfrrrrrrrrrR/var/www/html/VBACK/school/venv/lib/python3.8/site-packages/autobahn/xbr/_buyer.py__init__3szTransaction.__init__c Cs*|j|j|j|j|j|j|j|jd}|S)Nrr)rresrrrmarshal=s zTransaction.marshalcCs t|Sr)rrrrrr__str__JszTransaction.__str__N)__name__ __module__ __qualname__rrrrrrrr 1s  r c@seZdZdZdZddZddZddZd d Zdd d Z d d dZ ddZ ddZ ddZ d!ddZddZddZddZdS)" SimpleBuyerz Simple XBR buyer component. This component can be used by a XBR buyer delegate to handle the automated buying of data encryption keys from the XBR market maker. NcCst|tkrt|dks&td|t|tkr>t|dksLtd|t|tkr`|dksdtt|_||_ d|_ ||_ t j ||_d|_|jj|_d|_tjj|_||_d|_d|_d|_i|_d|_d|_d|_ i|_!g|_"dS) a :param market_maker_adr: :type market_maker_adr: :param buyer_key: Consumer delegate (buyer) private Ethereum key. :type buyer_key: bytes :param max_price: Maximum price we are willing to buy per key. :type max_price: int z0market_maker_adr must be bytes[20], but got "{}" z.buyer delegate must be bytes[32], but got "{}"rNFT)#typebyteslenAssertionErrorformatinttxaioZ make_loggerlog_market_maker_adr _xbrmm_config _pkey_raweth_keyskeysZ PrivateKey_pkeyZ_acct public_keyZto_canonical_address_addr_caddrnaclpublicZgenerate _receive_key _max_price_channel_balance_seq_keys_session_running_auto_close_channel_transaction_idx _transactions)rZmarket_maker_adrZ buyer_keyZ max_pricerrrrUs, && zSimpleBuyer.__init__cst|tstt|tkst|jr(t||_d|_|jjdt |j t |j jdddz|dIdH|_t|jtkrt|jdkst|d|jIdH|_|jstd |jd }t|tkrt|d ksttj|d |_|d |jjIdH}Wn|YnXt|dtkr@t|d|d<|ddksVtd|d|_|d|_|jjdt |jt|jt|jd|jS)a Start buying keys to decrypt XBR data by calling ``unwrap()``. :param session: WAMP session over which to communicate with the XBR market maker. :type session: :class:`autobahn.wamp.protocol.ApplicationSession` :param consumer_id: XBR consumer ID. :type consumer_id: str :return: Current remaining balance in payment channel. :rtype: int TzSStart buying from consumer delegate address {address} (public key 0x{public_key}..)N )addressr4zxbr.marketmaker.get_configr$z*xbr.marketmaker.get_active_payment_channelzno active payment channel found channel_oidr'+xbr.marketmaker.get_payment_channel_balance remainingrz1no off-chain balance remaining on payment channelseqzvOk, buyer delegate started [active payment channel {channel_oid} with remaining balance {remaining} at sequence {seq}])rFrJrK) isinstancerr)r&strr@r?r-debugr r6binasciib2a_hexr3r4decodecallr/r5r'r(r; ExceptionuuidUUIDZ _channel_oidZleaverr<r=infor )rsessionZ consumer_idrFpayment_balancerrrstartsD      zSimpleBuyer.startcs |js td|_|jddS)z# Stop buying keys. FzOk, buyer delegate stopped.N)r@r)r-rVrrrrstops zSimpleBuyer.stopcs2|jr|jst|jd|jdIdH}|S)a Return current balance of payment channel: * ``amount``: The initial amount with which the payment channel was opened. * ``remaining``: The remaining amount of XBR in the payment channel that can be spent. * ``inflight``: The amount of XBR allocated to buy transactions that are currently processed. :return: Current payment balance. :rtype: dict rIrFN)r? is_attachedr)rRr;)rrXrrrrs zSimpleBuyer.balancecsV|jr|jsttd}|jd||j||IdH}|d|d|dd}|S)z :param amount: :type amount: :param details: :type details: :return: :rtype: @z$xbr.marketmaker.open_payment_channelNrrJinflight)rrJr])r?r[r)osurandomrRr5)rZ buyer_addrrdetailsrpayment_channelrrrr open_channels   zSimpleBuyer.open_channelcsdS)z[ Requests to close the currently active payment channel. :return: Nr)rr`rrr close_channelszSimpleBuyer.close_channelc$ st|tkrt|dkstt|tkr0|dks4tt|tksDt|jd}|jd}d}|jd}t|jddd }||j kr|j j d t t j|d d n|j j d t t j|d d d|j |<|jd|Id H} t| d} |j j dt t j|d t t| dd| |jkrXtdd|jjt j|d t| dt|jd|j| } | dkrh|jr4d } |} | r| d} | r4|j}|j}d}t|j|||||||| }|j j dt j|d |t|d|d|jd||||t|||| Id Htdd|jjt j|d |t|dtdd|jjt j|d t|jdt| d|jj j!t"j#j$d}|jd}d}t|j||||||| | }|%||j&|||| | |z0|jd |j&||||t| t| | Id H}Wntk r*}z|j'dkr|(|W5d }~XYnHt)k rp}z(|j 'd!|j *||j |<|W5d }~XYnX|d"}|d#}t|d$}t|d%}t|d&}t+|||||||d| }||j,kr|j j-d'|jjt t.|/t t.|j,/d(td)d*|jj|jd|krBtd+d,|jj|j|d#|j| |krttd+d-|jj|j| |d%||_||_|0||j,|||||||d.}t"j12|j}z|3|}WnHt"j4j5k r}z$||j |<td/d0|jj|W5d }~XYnXt"j67||j |<|8} |j j9d1|jjt d2d3d4t t j|d t tt|dd5d3d4t t.|d6/t t|dt t|dt t.|/| d7 d}!|j |dkr|!d8r|j j d9|jjt t j|d d:|!d7}!t:;d;Id Hqt<|j |t)r&|j |}|z|j |3|}"Wn>t"j4j5k rx}ztd/d<|jj|W5d }~XYnXzt=>|"}#Wn>t=j?j@k r}ztd=d>|jj|W5d }~XYnX|#S)?aO Decrypt XBR data. This functions will potentially make the buyer call the XBR market maker to buy data encryption keys from the XBR provider. :param key_id: ID of the data encryption used for decryption of application payload. :type key_id: bytes :param serializer: Application payload serializer. :type serializer: str :param ciphertext: Ciphertext of encrypted application payload to decrypt. :type ciphertext: bytes :return: Decrypted application payload. :rtype: object rG)Zcbor market_oidrFrverifying_chain_idverifying_contract_adrrNz>Key {key_id} already in key store (or currently being bought).rH)rz1Key {key_id} not yet in key store - buying key ..Fzxbr.marketmaker.get_quoteZpricez-Key {key_id} has current price quote {amount}lNZo)rrzxbr.error.max_price_exceededzR{}.unwrap() - key {} needed cannot be bought: price {} exceeds maximum price of {}rTzauto-closing payment channel {channel_oid} [close_seq={close_seq}, close_balance={close_balance}, close_is_final={close_is_final}])rF close_seq close_balanceclose_is_finalzxbr.marketmaker.close_channelzxbr.error.channel_closedzu{}.unwrap() - key {} cannot be bought: payment channel {} ran empty and we initiated close at remaining balance of {}zxbr.error.insufficient_balancez_{}.unwrap() - key {} cannot be bought: insufficient balance {} in payment channel for amount {})encoderzxbr.marketmaker.buy_keyz8Encountered error while calling market maker to buy key!rr amount_paidrJr]zv{klass}.unwrap()::XBRSIG[8/8] - EIP712 signature invalid: signer_address={signer_address}, delegate_adr={delegate_adr})klasssigner_address delegate_adrzxbr.error.invalid_signaturezQ{}.unwrap()::XBRSIG[8/8] - EIP712 signature invalid or not signed by market makerzxbr.error.invalid_transactionz^{}.buy_key(): invalid transaction (channel sequence number mismatch - expected {}, but got {})z_{}.buy_key(): invalid transaction (channel remaining amount mismatch - expected {}, but got {}) sealed_keyzxbr.error.decryption_failedz6{}.unwrap() - could not unseal data encryption key: {}z{klass}.unwrap() - {tx_type} key {key_id} bought for {amount_paid} [payment_channel={payment_channel}, remaining={remaining}, inflight={inflight}, buyer_pubkey={buyer_pubkey}, transactions={transactions}]z XBR BUY Zmagenta)Zcolorz XBRra) rlZtx_typerrkrarJr] buyer_pubkeyZ transactionsdzG{klass}.unwrap() - waiting for key "{key_id}" currently being bought ..)rlrg?z1{}.unwrap() - failed to unwrap encrypted data: {}z xbr.error.deserialization_failedz;{}.unwrap() - failed to deserialize application payload: {})Ar&r'r(r)rMr;r/rOa2b_hexr>r-rNr rTrUr?rRrr+r:rr* __class__r r<rApast_transactionsr=r r0rr9r4encoder7encodingZ RawEncoder_save_transaction_phase1r5errorrZrSZfailurer r.warnrPrQ_save_transaction_phase2r8Z SealedBoxZdecrypt exceptionsZ CryptoErrorZsecretZ SecretBoxcount_transactionsrVr,sleeprLcbor2loadsdecoderZCBORDecodeError)$rrZ serializerZ ciphertextrdrFZcurrent_block_numberrerfquoterrZlast_txZtxnsrgrhrirrprZis_finalZreceipteZmarketmaker_signatureZmarketmaker_channel_seqZmarketmaker_amount_paidZmarketmaker_remainingZmarketmaker_inflightrmroZ unseal_boxkeyZtransactions_countZ log_countermessagepayloadrrrunwrapsv        ,                          (    ((zSimpleBuyer.unwrapc Cs\||jkrtdt|t||||||||} t|j} |j| dg| |j|<dS) :param channel_oid: :param delegate_adr: :param buyer_pubkey: :param key_id: :param channel_seq: :param amount: :param balance: :param signature: :return: z;save_transaction_phase1: duplicate transaction for key 0x{}N) rB RuntimeErrorr*rOrPr r(rCappend) rrFrnrprrrrrtx1key_idxrrrrws  z$SimpleBuyer._save_transaction_phase1c Cs||jkrtdt||j|} |j| drJtdt||j| d} t||||||||} | j| jks~t| j | j kst| j | j kst| j | j kst| j | j kst| j | j kst| j| jkst| |j| d<dS)rz;save_transaction_phase2: transaction for key 0x{} not foundrz;save_transaction_phase2: duplicate transaction for key 0x{}rN)rBrr*rOrPrCr rr)rrrrrr) rrFrnrprrrrrrrtx2rrrrzs"  z$SimpleBuyer._save_transaction_phase2TrcCst|tkstt|tkr$|dks(td}g}||kr~t|j|kr~|j| }|rf|dr||dr||||d7}q0q~q0|S)zQ :param filter_complete: :param limit: :return: rr)r&boolr)r+r(rCr)rZfilter_completelimitnrZtxrrrrtBs   zSimpleBuyer.past_transactionscCsHddd}|jD]2\}}|r2|r2|dd7<q|dd7<q|S)z :return: r)completependingrrr)rC)rrrrrrrr|XszSimpleBuyer.count_transactionscCs |j|d}|r|j|SdS)2 :param key_id: :return: NrBgetrC)rridxrrrget_transactionhszSimpleBuyer.get_transactioncCs,|j|d}|r(|j|\}}|o&|SdS)rNFr)rrrrrrrr is_completers zSimpleBuyer.is_complete)N)N)Tr)r r!r"__doc__r-rrYrZrrbrcrrwrzrtr|rrrrrrr#Ns E:   r&  r#)rTrOpprintrr^r~Z nacl.secretr7Z nacl.utilsZnacl.exceptionsZ nacl.publicr,Zautobahn.wamp.exceptionrZautobahn.wamp.protocolrZ_utilrrr1utilr r Z_eip712_channel_closer r objectr r#rrrrs"