#!/usr/bin/env python # -*- coding: utf-8 -*- # Copyright (c) 2020-2023 tecnovert # Distributed under the MIT software license, see the accompanying # file LICENSE or http://www.opensource.org/licenses/mit-license.php. import json import logging import basicswap.contrib.ed25519_fast as edf import basicswap.ed25519_fast_util as edu import basicswap.util_xmr as xmr_util from coincurve.ed25519 import ( ed25519_add, ed25519_get_pubkey, ed25519_scalar_add, ) from coincurve.keys import PrivateKey from coincurve.dleag import ( dleag_prove, dleag_verify, dleag_proof_len, verify_ed25519_point, ) from basicswap.interface import ( Curves) from basicswap.util import ( i2b, dumpj, ensure, make_int, TemporaryError) from basicswap.rpc_xmr import ( make_xmr_rpc_func, make_xmr_rpc2_func) from basicswap.util import ( b2i, b2h) from basicswap.chainparams import XMR_COIN, CoinInterface, Coins class XMRInterface(CoinInterface): @staticmethod def curve_type(): return Curves.ed25519 @staticmethod def coin_type(): return Coins.XMR @staticmethod def COIN(): return XMR_COIN @staticmethod def exp() -> int: return 12 @staticmethod def nbk() -> int: return 32 @staticmethod def nbK() -> int: # No. of bytes requires to encode a public key return 32 @staticmethod def depth_spendable() -> int: return 10 @staticmethod def xmr_swap_a_lock_spend_tx_vsize() -> int: raise ValueError('Not possible') @staticmethod def xmr_swap_b_lock_spend_tx_vsize() -> int: # TODO: Estimate with ringsize return 1507 def __init__(self, coin_settings, network, swap_client=None): super().__init__(network) daemon_login = None if coin_settings.get('rpcuser', '') != '': daemon_login = (coin_settings.get('rpcuser', ''), coin_settings.get('rpcpassword', '')) self.rpc_cb = make_xmr_rpc_func(coin_settings['rpcport'], daemon_login, host=coin_settings.get('rpchost', '127.0.0.1')) self.rpc_cb2 = make_xmr_rpc2_func(coin_settings['rpcport'], daemon_login, host=coin_settings.get('rpchost', '127.0.0.1')) # non-json endpoint self.rpc_wallet_cb = make_xmr_rpc_func(coin_settings['walletrpcport'], coin_settings['walletrpcauth'], host=coin_settings.get('walletrpchost', '127.0.0.1')) self.blocks_confirmed = coin_settings['blocks_confirmed'] self._restore_height = coin_settings.get('restore_height', 0) self.setFeePriority(coin_settings.get('fee_priority', 0)) self._sc = swap_client self._log = self._sc.log if self._sc and self._sc.log else logging self._wallet_password = None self._have_checked_seed = False def setFeePriority(self, new_priority): ensure(new_priority >= 0 and new_priority < 4, 'Invalid fee_priority value') self._fee_priority = new_priority def setWalletFilename(self, wallet_filename): self._wallet_filename = wallet_filename def createWallet(self, params): if self._wallet_password is not None: params['password'] = self._wallet_password rv = self.rpc_wallet_cb('generate_from_keys', params) self._log.info('generate_from_keys %s', dumpj(rv)) def openWallet(self, filename): params = {'filename': filename} if self._wallet_password is not None: params['password'] = self._wallet_password try: # Can't reopen the same wallet in windows, !is_keys_file_locked() self.rpc_wallet_cb('close_wallet') except Exception: pass self.rpc_wallet_cb('open_wallet', params) def initialiseWallet(self, key_view, key_spend, restore_height=None): with self._mx_wallet: try: self.openWallet(self._wallet_filename) # TODO: Check address return # Wallet exists except Exception as e: pass Kbv = self.getPubkey(key_view) Kbs = self.getPubkey(key_spend) address_b58 = xmr_util.encode_address(Kbv, Kbs) params = { 'filename': self._wallet_filename, 'address': address_b58, 'viewkey': b2h(key_view[::-1]), 'spendkey': b2h(key_spend[::-1]), 'restore_height': self._restore_height, } self.createWallet(params) self.openWallet(self._wallet_filename) def ensureWalletExists(self) -> None: with self._mx_wallet: self.openWallet(self._wallet_filename) def testDaemonRPC(self, with_wallet=True): self.rpc_wallet_cb('get_languages') def getDaemonVersion(self): return self.rpc_wallet_cb('get_version')['version'] def getBlockchainInfo(self): get_height = self.rpc_cb2('get_height', timeout=30) rv = { 'blocks': get_height['height'], 'verificationprogress': 0.0, } try: # get_block_count.block_count is how many blocks are in the longest chain known to the node. # get_block_count returns "Internal error" if bootstrap-daemon is active if get_height['untrusted'] is True: rv['bootstrapping'] = True get_info = self.rpc_cb2('get_info', timeout=30) if 'height_without_bootstrap' in get_info: rv['blocks'] = get_info['height_without_bootstrap'] rv['known_block_count'] = get_info['height'] if rv['known_block_count'] > rv['blocks']: rv['verificationprogress'] = rv['blocks'] / rv['known_block_count'] else: rv['known_block_count'] = self.rpc_cb('get_block_count', timeout=30)['count'] rv['verificationprogress'] = rv['blocks'] / rv['known_block_count'] except Exception as e: self._log.warning('XMR get_block_count failed with: %s', str(e)) rv['verificationprogress'] = 0.0 return rv def getChainHeight(self): return self.rpc_cb2('get_height', timeout=30)['height'] def getWalletInfo(self): with self._mx_wallet: try: self.openWallet(self._wallet_filename) except Exception as e: if 'Failed to open wallet' in str(e): rv = {'encrypted': True, 'locked': True, 'balance': 0, 'unconfirmed_balance': 0} return rv raise e rv = {} self.rpc_wallet_cb('refresh') balance_info = self.rpc_wallet_cb('get_balance') rv['balance'] = self.format_amount(balance_info['unlocked_balance']) rv['unconfirmed_balance'] = self.format_amount(balance_info['balance'] - balance_info['unlocked_balance']) rv['encrypted'] = False if self._wallet_password is None else True rv['locked'] = False return rv def walletRestoreHeight(self): return self._restore_height def getMainWalletAddress(self) -> str: with self._mx_wallet: self.openWallet(self._wallet_filename) return self.rpc_wallet_cb('get_address')['address'] def getNewAddress(self, placeholder) -> str: with self._mx_wallet: self.openWallet(self._wallet_filename) return self.rpc_wallet_cb('create_address', {'account_index': 0})['address'] def get_fee_rate(self, conf_target: int = 2): self._log.warning('TODO - estimate fee rate?') return 0.0, 'unused' def getNewSecretKey(self) -> bytes: # Note: Returned bytes are in big endian order return i2b(edu.get_secret()) def pubkey(self, key: bytes) -> bytes: return edf.scalarmult_B(key) def encodeKey(self, vk: bytes) -> str: return vk[::-1].hex() def decodeKey(self, k_hex: str) -> bytes: return bytes.fromhex(k_hex)[::-1] def encodePubkey(self, pk: bytes) -> str: return edu.encodepoint(pk) def decodePubkey(self, pke): return edf.decodepoint(pke) def getPubkey(self, privkey): return ed25519_get_pubkey(privkey) def getAddressFromKeys(self, key_view: bytes, key_spend: bytes) -> str: pk_view = self.getPubkey(key_view) pk_spend = self.getPubkey(key_spend) return xmr_util.encode_address(pk_view, pk_spend) def verifyKey(self, k: int) -> bool: i = b2i(k) return (i < edf.l and i > 8) def verifyPubkey(self, pubkey_bytes): # Calls ed25519_decode_check_point() in secp256k1 # Checks for small order return verify_ed25519_point(pubkey_bytes) def proveDLEAG(self, key: bytes) -> bytes: privkey = PrivateKey(key) return dleag_prove(privkey) def verifyDLEAG(self, dleag_bytes: bytes) -> bool: return dleag_verify(dleag_bytes) def lengthDLEAG(self) -> int: return dleag_proof_len() def sumKeys(self, ka: bytes, kb: bytes) -> bytes: return ed25519_scalar_add(ka, kb) def sumPubkeys(self, Ka: bytes, Kb: bytes) -> bytes: return ed25519_add(Ka, Kb) def encodeSharedAddress(self, Kbv: bytes, Kbs: bytes) -> str: return xmr_util.encode_address(Kbv, Kbs) def publishBLockTx(self, kbv: bytes, Kbs: bytes, output_amount: int, feerate: int, delay_for: int = 10, unlock_time: int = 0) -> bytes: with self._mx_wallet: self.openWallet(self._wallet_filename) self.rpc_wallet_cb('refresh') Kbv = self.getPubkey(kbv) shared_addr = xmr_util.encode_address(Kbv, Kbs) params = {'destinations': [{'amount': output_amount, 'address': shared_addr}], 'unlock_time': unlock_time} if self._fee_priority > 0: params['priority'] = self._fee_priority rv = self.rpc_wallet_cb('transfer', params) self._log.info('publishBLockTx %s to address_b58 %s', rv['tx_hash'], shared_addr) tx_hash = bytes.fromhex(rv['tx_hash']) if self._sc.debug: i = 0 while not self._sc.delay_event.is_set(): gt_params = {'out': True, 'pending': True, 'failed': True, 'pool': True, } rv = self.rpc_wallet_cb('get_transfers', gt_params) self._log.debug('get_transfers {}'.format(dumpj(rv))) if 'pending' not in rv: break if i >= delay_for: break self._sc.delay_event.wait(1.0) return tx_hash def findTxB(self, kbv, Kbs, cb_swap_value, cb_block_confirmed, restore_height, bid_sender): with self._mx_wallet: Kbv = self.getPubkey(kbv) address_b58 = xmr_util.encode_address(Kbv, Kbs) kbv_le = kbv[::-1] params = { 'restore_height': restore_height, 'filename': address_b58, 'address': address_b58, 'viewkey': b2h(kbv_le), } try: self.openWallet(address_b58) except Exception as e: self.createWallet(params) self.openWallet(address_b58) self.rpc_wallet_cb('refresh', timeout=600) ''' # Debug try: current_height = self.rpc_wallet_cb('get_height')['height'] self._log.info('findTxB XMR current_height %d\nAddress: %s', current_height, address_b58) except Exception as e: self._log.info('rpc_cb failed %s', str(e)) current_height = None # If the transfer is available it will be deep enough # and (current_height is None or current_height - transfer['block_height'] > cb_block_confirmed): ''' params = {'transfer_type': 'available'} transfers = self.rpc_wallet_cb('incoming_transfers', params) rv = None if 'transfers' in transfers: for transfer in transfers['transfers']: # unlocked <- wallet->is_transfer_unlocked() checks unlock_time and CRYPTONOTE_DEFAULT_TX_SPENDABLE_AGE if not transfer['unlocked']: full_tx = self.rpc_wallet_cb('get_transfer_by_txid', {'txid': transfer['tx_hash']}) unlock_time = full_tx['transfer']['unlock_time'] if unlock_time != 0: self._log.warning('Coin b lock txn is locked: {}, unlock_time {}'.format(transfer['tx_hash'], unlock_time)) rv = -1 continue if transfer['amount'] == cb_swap_value: return {'txid': transfer['tx_hash'], 'amount': transfer['amount'], 'height': 0 if 'block_height' not in transfer else transfer['block_height']} else: self._log.warning('Incorrect amount detected for coin b lock txn: {}'.format(transfer['tx_hash'])) rv = -1 return rv def findTxnByHash(self, txid): with self._mx_wallet: self.openWallet(self._wallet_filename) self.rpc_wallet_cb('refresh', timeout=600) try: current_height = self.rpc_cb2('get_height', timeout=30)['height'] self._log.info('findTxnByHash XMR current_height %d\nhash: %s', current_height, txid) except Exception as e: self._log.info('rpc_cb failed %s', str(e)) current_height = None # If the transfer is available it will be deep enough params = {'transfer_type': 'available'} rv = self.rpc_wallet_cb('incoming_transfers', params) if 'transfers' in rv: for transfer in rv['transfers']: if transfer['tx_hash'] == txid \ and (current_height is None or current_height - transfer['block_height'] > self.blocks_confirmed): return {'txid': transfer['tx_hash'], 'amount': transfer['amount'], 'height': transfer['block_height']} return None def spendBLockTx(self, chain_b_lock_txid: bytes, address_to: str, kbv: bytes, kbs: bytes, cb_swap_value: int, b_fee_rate: int, restore_height: int, spend_actual_balance: bool = False) -> bytes: ''' Notes: "Error: No unlocked balance in the specified subaddress(es)" can mean not enough funds after tx fee. ''' with self._mx_wallet: Kbv = self.getPubkey(kbv) Kbs = self.getPubkey(kbs) address_b58 = xmr_util.encode_address(Kbv, Kbs) wallet_filename = address_b58 + '_spend' params = { 'filename': wallet_filename, 'address': address_b58, 'viewkey': b2h(kbv[::-1]), 'spendkey': b2h(kbs[::-1]), 'restore_height': restore_height, } try: self.openWallet(wallet_filename) except Exception as e: self.createWallet(params) self.openWallet(wallet_filename) self.rpc_wallet_cb('refresh') rv = self.rpc_wallet_cb('get_balance') if rv['balance'] < cb_swap_value: self._log.warning('Balance is too low, checking for existing spend.') txns = self.rpc_wallet_cb('get_transfers', {'out': True}) if 'out' in txns: txns = txns['out'] if len(txns) > 0: txid = txns[0]['txid'] self._log.warning(f'spendBLockTx detected spending tx: {txid}.') if txns[0]['address'] == address_b58: return bytes.fromhex(txid) self._log.error('wallet {} balance {}, expected {}'.format(wallet_filename, rv['balance'], cb_swap_value)) if not spend_actual_balance: raise TemporaryError('Invalid balance') if spend_actual_balance and rv['balance'] != cb_swap_value: self._log.warning('Spending actual balance {}, not swap value {}.'.format(rv['balance'], cb_swap_value)) cb_swap_value = rv['balance'] if rv['unlocked_balance'] < cb_swap_value: self._log.error('wallet {} balance {}, expected {}, blocks_to_unlock {}'.format(wallet_filename, rv['unlocked_balance'], cb_swap_value, rv['blocks_to_unlock'])) raise TemporaryError('Invalid unlocked_balance') params = {'address': address_to} if self._fee_priority > 0: params['priority'] = self._fee_priority rv = self.rpc_wallet_cb('sweep_all', params) self._log.debug('sweep_all {}'.format(json.dumps(rv))) return bytes.fromhex(rv['tx_hash_list'][0]) def withdrawCoin(self, value: int, addr_to: str, subfee: bool) -> str: with self._mx_wallet: value_sats = make_int(value, self.exp()) self.openWallet(self._wallet_filename) self.rpc_wallet_cb('refresh') if subfee: balance = self.rpc_wallet_cb('get_balance') diff = balance['unlocked_balance'] - value_sats if diff >= 0 and diff <= 10: self._log.info('subfee enabled and value close to total, using sweep_all.') params = {'address': addr_to} if self._fee_priority > 0: params['priority'] = self._fee_priority rv = self.rpc_wallet_cb('sweep_all', params) return rv['tx_hash_list'][0] raise ValueError('Withdraw value must be close to total to use subfee/sweep_all.') params = {'destinations': [{'amount': value_sats, 'address': addr_to}]} if self._fee_priority > 0: params['priority'] = self._fee_priority rv = self.rpc_wallet_cb('transfer', params) return rv['tx_hash'] def showLockTransfers(self, kbv, Kbs, restore_height): with self._mx_wallet: try: Kbv = self.getPubkey(kbv) address_b58 = xmr_util.encode_address(Kbv, Kbs) wallet_file = address_b58 + '_spend' try: self.openWallet(wallet_file) except Exception: wallet_file = address_b58 try: self.openWallet(wallet_file) except Exception: self._log.info(f'showLockTransfers trying to create wallet for address {address_b58}.') kbv_le = kbv[::-1] params = { 'restore_height': restore_height, 'filename': address_b58, 'address': address_b58, 'viewkey': b2h(kbv_le), } self.createWallet(params) self.openWallet(address_b58) self.rpc_wallet_cb('refresh') rv = self.rpc_wallet_cb('get_transfers', {'in': True, 'out': True, 'pending': True, 'failed': True}) rv['filename'] = wallet_file return rv except Exception as e: return {'error': str(e)} def getSpendableBalance(self) -> int: with self._mx_wallet: self.openWallet(self._wallet_filename) self.rpc_wallet_cb('refresh') balance_info = self.rpc_wallet_cb('get_balance') return balance_info['unlocked_balance'] def changeWalletPassword(self, old_password, new_password): self._log.info('changeWalletPassword - {}'.format(self.ticker())) orig_password = self._wallet_password if old_password != '': self._wallet_password = old_password try: self.openWallet(self._wallet_filename) self.rpc_wallet_cb('change_wallet_password', {'old_password': old_password, 'new_password': new_password}) except Exception as e: self._wallet_password = orig_password raise e def unlockWallet(self, password: str) -> None: self._log.info('unlockWallet - {}'.format(self.ticker())) self._wallet_password = password if not self._have_checked_seed: self._sc.checkWalletSeed(self.coin_type()) def lockWallet(self) -> None: self._log.info('lockWallet - {}'.format(self.ticker())) self._wallet_password = None def isAddressMine(self, address): # TODO return True def ensureFunds(self, amount: int) -> None: if self.getSpendableBalance() < amount: raise ValueError('Balance too low') def getTransaction(self, txid: bytes): return self.rpc_cb2('get_transactions', {'txs_hashes': [txid.hex(), ]})