o i @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_idZ_priceZ _interval_count_count_current _on_rotate_id_key_box_archive)selfapi_idpriceintervalcount on_rotater%ZC:\Users\psant\Downloads\PAF backend\school\venv\Lib\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 N)rrr%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 NrrZcbor)cbor2dumpsrr_rotaterencryptr)rpayloaddata ciphertextr%r%r&r-gs    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 )encoderN) rrrrnaclpublicZ SealedBoxZ PublicKeyencodingZ RawEncoderr-)rr) buyer_pubkeykey_Z sendkey_boxZ encrypted_keyr%r%r& encrypt_key}szKeySeries.encrypt_keycCtNNotImplementedErrorr(r%r%r&startzKeySeries.startcCr:r;r<r(r%r%r&stopr?zKeySeries.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 )osurandomrr3utilsrandomsecretZ SecretBoxZKEY_SIZErrrlogdebugr uuidUUIDrrr(r%r%r&r,s  zKeySeries._rotateNNN) __name__ __module__ __qualname____doc__r'propertyr)r-r9r>r@r,r%r%r%r&r1s %   rc@seZdZddZdS) PayingChannelcCsbt|tkr t|dksJt|tkr|dksJt|tkr$|dks&J||_||_||_dS)Nrr)rrrrZ_adr_seq_balance)rZadrseqbalancer%r%r&r's  zPayingChannel.__init__N)rPrQrRr'r%r%r%r&rUs rUc@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 "{}"r1z/seller delegate must be bytes[32], but got "{}"Nz0provider_id must be None or string, but got "{}"r) rrrformatstrtxaioZ make_loggerrKrZ STATE_NONE_state_market_maker_adr _xbrmm_config _pkey_raweth_keyskeysZ PrivateKey_pkeyZ_acct public_keyZto_canonical_address_addr_caddr _provider_id _channels_channelrWrV_keys _keys_map_session _session_regs)rmarket_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 N)rhrir(r%r%r&ri 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|tkVqdSr;rr_).0kr%r%r& +z#SimpleSeller.add..csrur;rv)rwvr%r%r&ry+rzzZinvalid 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)Zprivkeyr! categoriesexpiresZcopiesrtzh{tx_type} key "{key_id}" offered for {price} [api_id={api_id}, prefix="{prefix}", delegate="{delegate}"]z XBR OFFER rArBrDNZor XBR)rEr)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)rpr rjrFrGrlrqcallr rKrLr rMrNr_rbinasciib2a_hexdecodererrorwarnfailurerasynciosleep) key_seriesr)rZ valid_fromr signaturertrer r~rr!rr%r&r$-sf     &        z#SimpleSeller.add..on_rotate)r"r#r$z#Created new key series {key_series})r) rrrrordictrgvaluesrrKrL) rr rr!r"r#r~r$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 )addressriz)xbr.marketmaker.get_active_paying_channelzno active paying channel found channel_oidrrDzxbr.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 channelrXzvOk, seller delegate started [active paying channel {channel_oid} with remaining balance {remaining} at sequence {seq}])rrrX)0 isinstancerr^rbrZra STATE_STOPPEDSTATE_STARTINGrqrrrKrLr rkrrrhrirrrjrn ExceptionrrrrMrNZ _channel_oidrlregistersellrappendr close_channelrorr>rdrrUrm STATE_STARTEDrWrVinfor )rsessionrrregrpaying_balancer%r%r&r>psT         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.)rbrZrSTATE_STOPPINGrorr@rrrrq is_attached unregisterr`gatherrKrrr)rdlrdrr%r%r&r@s0     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)rbrZr RuntimeErrorrqrrrn)rrr%r%r&rYs 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)rrrror_r-)rr urir.Z keyseriesr) serializerr0r%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. r]z.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 {}r1z-channel_balance must be bytes[32], but was {}z)channel_is_final must be bool, but was {}r}z3marketmaker_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)rrXrYis_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 rArBrr)rrEchannel_balance channel_seqrcaller caller_authid)!rrrr^rboolrrrcr __class__rPrrrrnrqleaverVrrWrrKrr rrerjr rLr_rr) rrsrrrZchannel_is_finalZmarketmaker_signaturerrseller_signaturereceiptr%r%r&rsd**** & *     zSimpleSeller.close_channelc  Cst|tkr t|dksJdt|tkrt|dks Jdt|tkr,t|dks0Jdt|tkrr@rYrrrr%r%r%r&rZs& B  ZA! BrZ)!rrrFrMZautobahn.wamp.typesrrZautobahn.wamp.exceptionrrZautobahn.wamp.protocolrZ_utilrr r`r r*rfZ nacl.secretr3Z nacl.utilsZ nacl.publicutilr r Z_eip712_channel_closerrobjectrrUrZr%r%r%r&s*