import 'dart:async'; import 'dart:convert'; import 'dart:io'; import 'dart:isolate'; import 'dart:math'; import 'package:cw_core/encryption_file_utils.dart'; import 'package:cw_core/unspent_coins_info.dart'; import 'package:hive/hive.dart'; import 'package:cw_bitcoin/electrum_wallet_addresses.dart'; import 'package:mobx/mobx.dart'; import 'package:rxdart/subjects.dart'; import 'package:flutter/foundation.dart'; import 'package:bitcoin_flutter/bitcoin_flutter.dart' as bitcoin; import 'package:collection/collection.dart'; import 'package:cw_bitcoin/address_to_output_script.dart'; import 'package:cw_bitcoin/bitcoin_address_record.dart'; import 'package:cw_bitcoin/bitcoin_transaction_credentials.dart'; import 'package:cw_bitcoin/bitcoin_transaction_no_inputs_exception.dart'; import 'package:cw_bitcoin/bitcoin_transaction_priority.dart'; import 'package:cw_bitcoin/bitcoin_transaction_wrong_balance_exception.dart'; import 'package:cw_bitcoin/bitcoin_unspent.dart'; import 'package:cw_bitcoin/bitcoin_wallet_keys.dart'; import 'package:cw_bitcoin/electrum.dart'; import 'package:cw_bitcoin/electrum_balance.dart'; import 'package:cw_bitcoin/electrum_transaction_history.dart'; import 'package:cw_bitcoin/electrum_transaction_info.dart'; import 'package:cw_bitcoin/electrum_wallet_addresses.dart'; import 'package:cw_bitcoin/pending_bitcoin_transaction.dart'; import 'package:cw_bitcoin/script_hash.dart'; import 'package:cw_bitcoin/utils.dart'; import 'package:cw_core/crypto_currency.dart'; import 'package:collection/collection.dart'; import 'package:cw_core/node.dart'; import 'package:cw_core/pathForWallet.dart'; import 'package:cw_core/pending_transaction.dart'; import 'package:cw_core/sync_status.dart'; import 'package:cw_core/transaction_direction.dart'; import 'package:cw_core/transaction_priority.dart'; import 'package:cw_core/unspent_coins_info.dart'; import 'package:cw_core/wallet_base.dart'; import 'package:cw_core/wallet_info.dart'; import 'package:flutter/foundation.dart'; import 'package:hex/hex.dart'; import 'package:hive/hive.dart'; import 'package:mobx/mobx.dart'; import 'package:rxdart/subjects.dart'; part 'electrum_wallet.g.dart'; class ElectrumWallet = ElectrumWalletBase with _$ElectrumWallet; abstract class ElectrumWalletBase extends WalletBase with Store { ElectrumWalletBase( {required String password, required WalletInfo walletInfo, required Box unspentCoinsInfo, required this.networkType, required this.mnemonic, required Uint8List seedBytes, required this.encryptionFileUtils, List? initialAddresses, ElectrumClient? electrumClient, ElectrumBalance? initialBalance, CryptoCurrency? currency}) : hd = currency == CryptoCurrency.bch ? bitcoinCashHDWallet(seedBytes) : bitcoin.HDWallet.fromSeed(seedBytes, network: networkType).derivePath("m/0'/0"), syncStatus = NotConnectedSyncStatus(), _password = password, _feeRates = [], _isTransactionUpdating = false, unspentCoins = [], _scripthashesUpdateSubject = {}, balance = ObservableMap.of(currency != null ? { currency: initialBalance ?? const ElectrumBalance(confirmed: 0, unconfirmed: 0, frozen: 0) } : {}), this.unspentCoinsInfo = unspentCoinsInfo, super(walletInfo) { this.electrumClient = electrumClient ?? ElectrumClient(); this.walletInfo = walletInfo; transactionHistory = ElectrumTransactionHistory( walletInfo: walletInfo, password: password, encryptionFileUtils: encryptionFileUtils); } static bitcoin.HDWallet bitcoinCashHDWallet(Uint8List seedBytes) => bitcoin.HDWallet.fromSeed(seedBytes).derivePath("m/44'/145'/0'/0"); static int estimatedTransactionSize(int inputsCount, int outputsCounts) => inputsCount * 146 + outputsCounts * 33 + 8; final bitcoin.HDWallet hd; final String mnemonic; final EncryptionFileUtils encryptionFileUtils; late ElectrumClient electrumClient; Box unspentCoinsInfo; @override late ElectrumWalletAddresses walletAddresses; @override @observable late ObservableMap balance; @override @observable SyncStatus syncStatus; List get scriptHashes => walletAddresses.addresses .map((addr) => scriptHash(addr.address, networkType: networkType)) .toList(); List get publicScriptHashes => walletAddresses.addresses .where((addr) => !addr.isHidden) .map((addr) => scriptHash(addr.address, networkType: networkType)) .toList(); String get xpub => hd.base58!; @override String get seed => mnemonic; @override String get password => _password; bitcoin.NetworkType networkType; @override BitcoinWalletKeys get keys => BitcoinWalletKeys(wif: hd.wif!, privateKey: hd.privKey!, publicKey: hd.pubKey!); String _password; List unspentCoins; List _feeRates; Map?> _scripthashesUpdateSubject; BehaviorSubject? _chainTipUpdateSubject; bool _isTransactionUpdating; int _initialSyncHeight = 0; Future? _isolate; void Function(FlutterErrorDetails)? _onError; Future init() async { await walletAddresses.init(); await transactionHistory.init(); await save(); } @action @override Future startSync() async { try { await _setInitialHeight(); } catch (_) {} try { rescan(height: walletInfo.restoreHeight); await walletAddresses.discoverAddresses(); await updateTransactions(); _subscribeForUpdates(); await updateUnspent(); await updateBalance(); _feeRates = await electrumClient.feeRates(); Timer.periodic( const Duration(minutes: 1), (timer) async => _feeRates = await electrumClient.feeRates()); } catch (e, stacktrace) { print(stacktrace); print(e.toString()); syncStatus = FailedSyncStatus(); } } @action @override Future connectToNode({required Node node}) async { try { syncStatus = ConnectingSyncStatus(); await electrumClient.connectToUri(node.uri); electrumClient.onConnectionStatusChange = (bool isConnected) { if (!isConnected) { syncStatus = LostConnectionSyncStatus(); } }; syncStatus = ConnectedSyncStatus(); } catch (e) { print(e.toString()); syncStatus = FailedSyncStatus(); } } @override Future createTransaction(Object credentials) async { try { const minAmount = 546; final transactionCredentials = credentials as BitcoinTransactionCredentials; final inputs = []; final outputs = transactionCredentials.outputs; final hasMultiDestination = outputs.length > 1; var allInputsAmount = 0; if (unspentCoins.isEmpty) { await updateUnspent(); } for (int i = 0; i < unspentCoins.length; i++) { final utx = unspentCoins[i]; if (utx.isSending) { allInputsAmount += utx.value; inputs.add(utx); } } if (inputs.isEmpty) { throw BitcoinTransactionNoInputsException(); } final allAmountFee = 222; final allAmount = allInputsAmount - allAmountFee; var credentialsAmount = 0; var amount = 0; var fee = 0; if (hasMultiDestination) { if (outputs.any((item) => item.sendAll || item.formattedCryptoAmount! <= 0)) { throw BitcoinTransactionWrongBalanceException(currency); } credentialsAmount = outputs.fold(0, (acc, value) => acc + value.formattedCryptoAmount!); if (allAmount - credentialsAmount < minAmount) { throw BitcoinTransactionWrongBalanceException(currency); } amount = credentialsAmount; if (transactionCredentials.feeRate != null) { fee = calculateEstimatedFeeWithFeeRate(transactionCredentials.feeRate!, amount, outputsCount: outputs.length + 1); } else { fee = calculateEstimatedFee(transactionCredentials.priority, amount, outputsCount: outputs.length + 1); } } else { final output = outputs.first; credentialsAmount = !output.sendAll ? output.formattedCryptoAmount! : 0; if (credentialsAmount > allAmount) { throw BitcoinTransactionWrongBalanceException(currency); } amount = output.sendAll || allAmount - credentialsAmount < minAmount ? allAmount : credentialsAmount; if (output.sendAll || amount == allAmount) { fee = allAmountFee; } else if (transactionCredentials.feeRate != null) { fee = calculateEstimatedFeeWithFeeRate(transactionCredentials.feeRate!, amount); } else { fee = calculateEstimatedFee(transactionCredentials.priority, amount); } } if (fee == 0 && networkType == bitcoin.bitcoin) { // throw BitcoinTransactionWrongBalanceException(currency); } final totalAmount = amount + fee; if (totalAmount > balance[currency]!.confirmed || totalAmount > allInputsAmount) { // throw BitcoinTransactionWrongBalanceException(currency); } final changeAddress = await walletAddresses.getChangeAddress(); var leftAmount = totalAmount; var totalInputAmount = 0; final txb = bitcoin.TransactionBuilder(network: networkType, version: 1); List inputPrivKeys = []; List outpoints = []; for (int i = 0; i < inputs.length; i++) { final utx = inputs[i]; leftAmount = utx.value - leftAmount; totalInputAmount += utx.value; if (leftAmount <= 0) { break; } final isSilentPayment = utx.bitcoinAddressRecord.silentPaymentTweak != null; outpoints.add(bitcoin.Outpoint(txid: utx.hash, index: utx.vout)); if (isSilentPayment) { // https://github.com/bitcoin/bips/blob/c55f80c53c98642357712c1839cfdc0551d531c4/bip-0352.mediawiki#user-content-Spending final d = bitcoin.PrivateKey.fromHex(bitcoin.getSecp256k1(), walletAddresses.silentAddress!.spendPrivkey.toCompressedHex()) .tweakAdd(utx.bitcoinAddressRecord.silentPaymentTweak!.bigint)!; inputPrivKeys.add(bitcoin.PrivateKeyInfo(d, true)); final point = bitcoin.ECPublic.fromHex(d.publicKey.toHex()).toTapPoint(); final p2tr = bitcoin.P2trAddress(program: point); bitcoin.ECPair keyPair = bitcoin.ECPair.fromPrivateKey(d.toCompressedHex().fromHex, compressed: true, network: networkType); txb.addInput( utx.hash, utx.vout, null, p2tr.toScriptPubKey().toBytes(), keyPair, utx.value); continue; } inputPrivKeys.add(bitcoin.PrivateKeyInfo( bitcoin.PrivateKey.fromHex( bitcoin.getSecp256k1(), generateKeyPair( hd: utx.bitcoinAddressRecord.isHidden ? walletAddresses.sideHd : walletAddresses.mainHd, index: utx.bitcoinAddressRecord.index, network: networkType) .privateKey! .hex), false)); bitcoin.ECPair keyPair = generateKeyPair( hd: utx.bitcoinAddressRecord.isHidden ? walletAddresses.sideHd : walletAddresses.mainHd, index: utx.bitcoinAddressRecord.index, network: networkType); if (utx.isP2wpkh) { final p2wpkh = bitcoin .P2WPKH( data: generatePaymentData( hd: utx.bitcoinAddressRecord.isHidden ? walletAddresses.sideHd : walletAddresses.mainHd, index: utx.bitcoinAddressRecord.index), network: networkType) .data; txb.addInput(utx.hash, utx.vout, null, p2wpkh.output, keyPair, utx.value); continue; } txb.addInput(utx.hash, utx.vout, null, null, keyPair, utx.value); } if (txb.inputs.isEmpty) { throw BitcoinTransactionNoInputsException(); } if (amount <= 0 || totalInputAmount < totalAmount) { // throw BitcoinTransactionWrongBalanceException(currency); } List silentPaymentDestinations = []; outputs.forEach((item) { final outputAmount = hasMultiDestination ? item.formattedCryptoAmount : amount; final outputAddress = item.isParsedAddress ? item.extractedAddress! : item.address; if (outputAddress.startsWith('tsp1')) { silentPaymentDestinations .add(bitcoin.SilentPaymentDestination.fromAddress(outputAddress, outputAmount!)); } else { txb.addOutput(addressToOutputScript(outputAddress, networkType), outputAmount!); } }); if (silentPaymentDestinations.isNotEmpty) { final outpointsHash = bitcoin.SilentPayment.hashOutpoints(outpoints); final aSum = bitcoin.SilentPayment.getSumInputPrivKeys(inputPrivKeys); final generatedOutputs = bitcoin.SilentPayment.generateMultipleRecipientPubkeys( aSum, outpointsHash, silentPaymentDestinations); generatedOutputs.forEach((recipientSilentAddress, generatedOutput) { generatedOutput.forEach((output) { final generatedPubkey = output.$1.toHex(); // TODO: DRY code: pubkeyToOutputScript (?) final point = bitcoin.ECPublic.fromHex(generatedPubkey).toTapPoint(); final p2tr = bitcoin.P2trAddress(program: point); txb.addOutput(p2tr.toScriptPubKey().toBytes(), amount); }); }); } final estimatedSize = estimatedTransactionSize(inputs.length, outputs.length + 1); var feeAmount = 222; // if (transactionCredentials.feeRate != null) { // feeAmount = transactionCredentials.feeRate! * estimatedSize; // } else { // feeAmount = feeRate(transactionCredentials.priority!) * estimatedSize; // } final changeValue = totalInputAmount - amount - feeAmount; if (changeValue > minAmount) { txb.addOutput(changeAddress, changeValue); } final amounts = txb.inputs.map((utx) => utx.value!).toList(); final scriptPubKeys = txb.inputs.map((utx) => utx.prevOutScript!).toList(); for (var i = 0; i < inputs.length; i++) { txb.sign(vin: i, amounts: amounts, scriptPubKeys: scriptPubKeys, inputs: inputs); } return PendingBitcoinTransaction(txb.build(), type, electrumClient: electrumClient, amount: amount, fee: fee) ..addListener((transaction) async { transactionHistory.addOne(transaction); await updateBalance(); }); } catch (e, stacktrace) { print(stacktrace); print(e.toString()); rethrow; } } String toJSON() => json.encode({ 'mnemonic': mnemonic, 'account_index': walletAddresses.currentReceiveAddressIndex.toString(), 'change_address_index': walletAddresses.currentChangeAddressIndex.toString(), 'addresses': walletAddresses.addresses.map((addr) => addr.toJSON()).toList(), 'balance': balance[currency]?.toJSON(), 'network_type': networkType.toString() }); int feeRate(TransactionPriority priority) { try { if (priority is BitcoinTransactionPriority) { return _feeRates[priority.raw]; } return 0; } catch (_) { return 0; } } int feeAmountForPriority( BitcoinTransactionPriority priority, int inputsCount, int outputsCount) => feeRate(priority) * estimatedTransactionSize(inputsCount, outputsCount); int feeAmountWithFeeRate(int feeRate, int inputsCount, int outputsCount) => feeRate * estimatedTransactionSize(inputsCount, outputsCount); @override int calculateEstimatedFee(TransactionPriority? priority, int? amount, {int? outputsCount}) { if (priority is BitcoinTransactionPriority) { return calculateEstimatedFeeWithFeeRate(feeRate(priority), amount, outputsCount: outputsCount); } return 0; } int calculateEstimatedFeeWithFeeRate(int feeRate, int? amount, {int? outputsCount}) { int inputsCount = 0; if (amount != null) { int totalValue = 0; for (final input in unspentCoins) { if (totalValue >= amount) { break; } if (input.isSending) { totalValue += input.value; inputsCount += 1; } } if (totalValue < amount) return 0; } else { for (final input in unspentCoins) { if (input.isSending) { inputsCount += 1; } } } // If send all, then we have no change value final _outputsCount = outputsCount ?? (amount != null ? 2 : 1); return feeAmountWithFeeRate(feeRate, inputsCount, _outputsCount); } @override Future save() async { final path = await makePath(); await encryptionFileUtils.write(path: path, password: _password, data: toJSON()); await transactionHistory.save(); } @override Future renameWalletFiles(String newWalletName) async { final currentWalletPath = await pathForWallet(name: walletInfo.name, type: type); final currentWalletFile = File(currentWalletPath); final currentDirPath = await pathForWalletDir(name: walletInfo.name, type: type); final currentTransactionsFile = File('$currentDirPath/$transactionsHistoryFileName'); // Copies current wallet files into new wallet name's dir and files if (currentWalletFile.existsSync()) { final newWalletPath = await pathForWallet(name: newWalletName, type: type); await currentWalletFile.copy(newWalletPath); } if (currentTransactionsFile.existsSync()) { final newDirPath = await pathForWalletDir(name: newWalletName, type: type); await currentTransactionsFile.copy('$newDirPath/$transactionsHistoryFileName'); } // Delete old name's dir and files await Directory(currentDirPath).delete(recursive: true); } @override Future changePassword(String password) async { _password = password; await save(); await transactionHistory.changePassword(password); } bitcoin.ECPair keyPairFor({required int index}) => generateKeyPair(hd: hd, index: index, network: networkType); @action @override Future rescan({required int height, int? chainTip, ScanData? scanData}) async { syncStatus = AttemptingSyncStatus(); walletInfo.restoreHeight = height; await walletInfo.save(); final currentChainTip = chainTip ?? await electrumClient.getCurrentBlockChainTip() ?? 0; if (_isolate != null) { final runningIsolate = await _isolate!; runningIsolate.kill(priority: Isolate.immediate); } if (currentChainTip <= height) { syncStatus = SyncedSyncStatus(); return; } final receivePort = ReceivePort(); _isolate = Isolate.spawn( startRefresh, ScanData( sendPort: receivePort.sendPort, silentAddress: walletAddresses.silentAddress!.toString(), scanPrivkeyCompressed: walletAddresses.silentAddress!.scanPrivkey.toCompressedHex().fromHex, spendPubkeyCompressed: walletAddresses.silentAddress!.spendPubkey.toCompressedHex().fromHex, networkType: networkType, height: walletInfo.restoreHeight, chainTip: currentChainTip, initialSyncHeight: _initialSyncHeight, electrumClient: ElectrumClient(), transactionHistoryIds: transactionHistory.transactions.keys.toList(), node: electrumClient.uri.toString())); await for (var message in receivePort) { if (message is BitcoinUnspent) { final myNewUnspent = message; final hasUnspent = unspentCoins.any((element) { if (element.address == message.address) { unspentCoins.remove(element); unspentCoins.add(myNewUnspent); return true; } return false; }); if (!hasUnspent) { unspentCoins.add(myNewUnspent); } final myNewAddress = message.bitcoinAddressRecord; final hasAddress = walletAddresses.addresses.any((element) { if (element.address == message.address) { walletAddresses.addresses.remove(element); walletAddresses.addresses.add(myNewAddress); return true; } return false; }); if (!hasAddress) { walletAddresses.addresses.add(myNewAddress); } await save(); await updateUnspent(); await updateBalance(); await updateTransactions(); _subscribeForUpdates(); } // check if is a SyncStatus type since "is SyncStatus" doesn't work here if (message is SyncResponse) { syncStatus = message.syncStatus; walletInfo.restoreHeight = message.height; await walletInfo.save(); } } } @override Future close() async { try { await electrumClient.close(); } catch (_) {} } Future makePath() async => pathForWallet(name: walletInfo.name, type: walletInfo.type); Future updateUnspent() async { final unspent = await Future.wait(walletAddresses.addresses.map((address) => electrumClient .getListUnspentWithAddress(address.address, networkType) .then((unspent) => unspent.map((unspent) { try { return BitcoinUnspent.fromJSON(address, unspent); } catch (_) { return null; } }).whereNotNull()))); unspent.expand((e) => e).forEach((newUnspent) { try { if (!unspentCoins.any((currentUnspent) => currentUnspent.address.contains(newUnspent.address) && currentUnspent.hash.contains(newUnspent.hash))) { unspentCoins.add(newUnspent); } } catch (_) {} }); if (unspentCoinsInfo.isEmpty) { unspentCoins.forEach((coin) => _addCoinInfo(coin)); return; } if (unspentCoins.isNotEmpty) { unspentCoins.forEach((coin) { final coinInfoList = unspentCoinsInfo.values.where((element) => element.walletId.contains(id) && element.hash.contains(coin.hash) && element.address.contains(coin.address)); if (coinInfoList.isNotEmpty) { final coinInfo = coinInfoList.first; coin.isFrozen = coinInfo.isFrozen; coin.isSending = coinInfo.isSending; coin.note = coinInfo.note; } else { _addCoinInfo(coin); } }); } await _refreshUnspentCoinsInfo(); } Future _addCoinInfo(BitcoinUnspent coin) async { final newInfo = UnspentCoinsInfo( walletId: id, hash: coin.hash, isFrozen: coin.isFrozen, isSending: coin.isSending, noteRaw: coin.note, address: coin.bitcoinAddressRecord.address, value: coin.value, vout: coin.vout, isChange: coin.isChange, ); await unspentCoinsInfo.add(newInfo); } Future _refreshUnspentCoinsInfo() async { try { final List keys = []; final currentWalletUnspentCoins = unspentCoinsInfo.values.where((element) => element.walletId.contains(id)); if (currentWalletUnspentCoins.isNotEmpty) { currentWalletUnspentCoins.forEach((element) { final existUnspentCoins = unspentCoins.where((coin) => element.hash.contains(coin.hash)); if (existUnspentCoins.isEmpty) { keys.add(element.key); } }); } if (keys.isNotEmpty) { await unspentCoinsInfo.deleteAll(keys); } } catch (e) { print(e.toString()); } } @override Future> fetchTransactions() async { final addressHashes = {}; final normalizedHistories = >[]; walletAddresses.addresses.forEach((addressRecord) { if (addressRecord.address == "tb1pch9qmsq87wy4my4akd60x2r2yt784zfmfwqeuk7w7g7u45za4ktq9pdnmf") {} final sh = scriptHash(addressRecord.address, networkType: networkType); addressHashes[sh] = addressRecord; }); final histories = addressHashes.keys.map((scriptHash) => electrumClient.getHistory(scriptHash).then((history) => {scriptHash: history})); final historyResults = await Future.wait(histories); historyResults.forEach((history) { history.entries.forEach((historyItem) { if (historyItem.value.isNotEmpty) { final address = addressHashes[historyItem.key]; address?.setAsUsed(); normalizedHistories.addAll(historyItem.value); } }); }); final historiesWithDetails = await Future.wait(normalizedHistories.map((transaction) { try { return fetchTransactionInfo( hash: transaction['tx_hash'] as String, height: transaction['height'] as int, electrumClient: electrumClient, addressRecords: walletAddresses.addresses, walletInfo: walletInfo, networkType: networkType); } catch (_) { return Future.value(null); } })); return historiesWithDetails .fold>({}, (acc, tx) { if (tx == null) { return acc; } acc[tx.id] = acc[tx.id]?.updated(tx) ?? tx; return acc; }); } Future updateTransactions() async { try { if (_isTransactionUpdating) { return; } _isTransactionUpdating = true; final transactions = await fetchTransactions(); transactionHistory.addMany(transactions); walletAddresses.updateReceiveAddresses(); await transactionHistory.save(); _isTransactionUpdating = false; } catch (e, stacktrace) { print(stacktrace); print(e); _isTransactionUpdating = false; } } void _subscribeForUpdates() async { scriptHashes.forEach((sh) async { await _scripthashesUpdateSubject[sh]?.close(); _scripthashesUpdateSubject[sh] = electrumClient.scripthashUpdate(sh); _scripthashesUpdateSubject[sh]?.listen((event) async { try { await updateUnspent(); await updateBalance(); await updateTransactions(); final currentHeight = await electrumClient.getCurrentBlockChainTip(); if (currentHeight != null) walletInfo.restoreHeight = currentHeight; rescan(height: walletInfo.restoreHeight); } catch (e, s) { print(e.toString()); _onError?.call(FlutterErrorDetails( exception: e, stack: s, library: this.runtimeType.toString(), )); } }); }); } Future _fetchBalances() async { final addresses = walletAddresses.addresses.toList(); final balanceFutures = >>[]; for (var i = 0; i < addresses.length; i++) { final addressRecord = addresses[i]; final sh = scriptHash(addressRecord.address, networkType: networkType); final balanceFuture = electrumClient.getBalance(sh); balanceFutures.add(balanceFuture); } var totalFrozen = 0; unspentCoinsInfo.values.forEach((info) { unspentCoins.forEach((element) { if (element.hash == info.hash && info.isFrozen && element.bitcoinAddressRecord.address == info.address && element.value == info.value) { totalFrozen += element.value; } }); }); final balances = await Future.wait(balanceFutures); var totalConfirmed = 0; var totalUnconfirmed = 0; for (var i = 0; i < balances.length; i++) { final addressRecord = addresses[i]; final balance = balances[i]; final confirmed = balance['confirmed'] as int? ?? 0; final unconfirmed = balance['unconfirmed'] as int? ?? 0; totalConfirmed += confirmed; totalUnconfirmed += unconfirmed; if (confirmed > 0 || unconfirmed > 0) { addressRecord.setAsUsed(); } } return ElectrumBalance( confirmed: totalConfirmed, unconfirmed: totalUnconfirmed, frozen: totalFrozen); } Future updateBalance() async { balance[currency] = await _fetchBalances(); await save(); } String getChangeAddress() { const minCountOfHiddenAddresses = 5; final random = Random(); var addresses = walletAddresses.addresses.where((addr) => addr.isHidden).toList(); if (addresses.length < minCountOfHiddenAddresses) { addresses = walletAddresses.addresses.toList(); } return addresses[random.nextInt(addresses.length)].address; } @override void setExceptionHandler(void Function(FlutterErrorDetails) onError) => _onError = onError; @override String signMessage(String message, {String? address = null}) { final index = address != null ? walletAddresses.addresses.firstWhere((element) => element.address == address).index : null; return index == null ? base64Encode(hd.sign(message)) : base64Encode(hd.derive(index).sign(message)); } Future _setInitialHeight() async { if (walletInfo.isRecovery) { return; } if (walletInfo.restoreHeight == 0) { final currentHeight = await electrumClient.getCurrentBlockChainTip(); if (currentHeight != null) walletInfo.restoreHeight = currentHeight; } _initialSyncHeight = walletInfo.restoreHeight; } } Future fetchTransactionInfo( {required String hash, required int height, required ElectrumClient electrumClient, required Iterable addressRecords, required WalletInfo walletInfo, required bitcoin.NetworkType networkType}) async { try { final tx = await getTransactionExpanded( hash: hash, height: height, electrumClient: electrumClient, networkType: networkType); final addresses = addressRecords.map((addr) => addr.address).toSet(); return ElectrumTransactionInfo.fromElectrumBundle(tx, walletInfo.type, networkType, addresses: addresses, height: height); } catch (_) { return null; } } Future getTransactionExpanded( {required String hash, required int height, required ElectrumClient electrumClient, required bitcoin.NetworkType networkType}) async { final verboseTransaction = await electrumClient.getTransactionRaw(hash: hash, networkType: networkType); String transactionHex; int? time; int confirmations = 0; if (networkType.bech32 == bitcoin.testnet.bech32) { transactionHex = verboseTransaction as String; confirmations = 1; } else { transactionHex = verboseTransaction['hex'] as String; time = verboseTransaction['time'] as int?; confirmations = verboseTransaction['confirmations'] as int? ?? 0; } final original = bitcoin.Transaction.fromHex(transactionHex); final ins = []; for (final vin in original.ins) { final id = HEX.encode(vin.hash!.reversed.toList()); final txHex = await electrumClient.getTransactionHex(hash: id); final tx = bitcoin.Transaction.fromHex(txHex); ins.add(tx); } return ElectrumTransactionBundle(original, ins: ins, time: time, confirmations: confirmations); } class ScanData { final SendPort sendPort; final Uint8List scanPrivkeyCompressed; final Uint8List spendPubkeyCompressed; final String silentAddress; final int height; final String node; final bitcoin.NetworkType networkType; final int chainTip; final int initialSyncHeight; final ElectrumClient electrumClient; final List transactionHistoryIds; final int? checkpointTxPos; ScanData( {required this.sendPort, required this.scanPrivkeyCompressed, required this.spendPubkeyCompressed, required this.silentAddress, required this.height, required this.node, required this.networkType, required this.chainTip, required this.initialSyncHeight, required this.electrumClient, required this.transactionHistoryIds, this.checkpointTxPos}); factory ScanData.withCheckpoint(ScanData scanData, int newHeight, int? checkpointTx) { return ScanData( sendPort: scanData.sendPort, scanPrivkeyCompressed: scanData.scanPrivkeyCompressed, spendPubkeyCompressed: scanData.spendPubkeyCompressed, silentAddress: scanData.silentAddress, height: newHeight, node: scanData.node, networkType: scanData.networkType, chainTip: scanData.chainTip, initialSyncHeight: scanData.initialSyncHeight, electrumClient: scanData.electrumClient, transactionHistoryIds: scanData.transactionHistoryIds, checkpointTxPos: checkpointTx); } } class SyncResponse { final int height; final SyncStatus syncStatus; SyncResponse(this.height, this.syncStatus); } Future startRefresh(ScanData scanData) async { final currentChainTip = scanData.chainTip; if (scanData.height >= currentChainTip) { scanData.sendPort.send(SyncResponse(scanData.height, SyncedSyncStatus())); return; } var checkpointTxPos = scanData.checkpointTxPos; final height = scanData.height; print(["HEIGHT:", height]); try { final track = currentChainTip - height; final diff = (currentChainTip - scanData.initialSyncHeight) - track; final ptc = diff <= 0 ? 0.0 : diff / track; scanData.sendPort.send(SyncResponse(height, SyncingSyncStatus(track, ptc))); final electrumClient = scanData.electrumClient; if (!electrumClient.isConnected) { final node = scanData.node; await electrumClient.connectToUri(Uri.parse(node)); } // tx pos always begin from 1 --> we know pos 0 is coinbase tx int pos = checkpointTxPos ?? 1; while (true) { try { final txid = await electrumClient.getTxidFromPos(height: height, pos: pos); print(["scanning tx:", txid]); // TODO: if already tx already scanned & stored skip // if (scanData.transactionHistoryIds.contains(txid)) { // // already scanned tx, continue to next tx // checkpointTxPos = pos; // pos++; // continue; // } List pubkeys = []; List outpoints = []; try { final txBundle = await getTransactionExpanded( hash: txid, height: height, electrumClient: electrumClient, networkType: scanData.networkType); bool skip = false; txBundle.originalTransaction.ins.forEach((input) { if (input.witness == null) { skip = true; return; } if (input.witness!.length != 2) { skip = true; return; } final pubkey = input.witness![1].hex; pubkeys.add(pubkey); outpoints.add(bitcoin.Outpoint( txid: HEX.encode(input.hash!.reversed.toList()), index: input.index!)); }); if (skip) { // skipped tx, save checkpoint in case of issues and continue to next tx checkpointTxPos = pos; pos++; continue; } Map outpointsByP2TRpubkey = {}; int i = 0; txBundle.originalTransaction.outs.forEach((output) { if (bitcoin.classifyOutput(output.script!) != "taproot") { return; } final address = bitcoin.P2trAddress(program: output.script!.sublist(2).hex).toAddress( scanData.networkType.bech32 == bitcoin.testnet.bech32 ? bitcoin.NetworkInfo.TESTNET : bitcoin.NetworkInfo.BITCOIN); print(["verifying taproot address:", address]); outpointsByP2TRpubkey[address] = bitcoin.Outpoint(txid: txid, index: i, value: output.value!); i++; }); if (pubkeys.isEmpty || outpoints.isEmpty || outpointsByP2TRpubkey.isEmpty) { // skipped tx, save checkpoint in case of issues and continue to next tx checkpointTxPos = pos; pos++; continue; } Uint8List sumOfInputPublicKeys = bitcoin.getSumInputPubKeys(pubkeys).toCompressedHex().fromHex; final outpointHash = bitcoin.SilentPayment.hashOutpoints(outpoints); final result = bitcoin.scanOutputs( scanData.scanPrivkeyCompressed, scanData.spendPubkeyCompressed, sumOfInputPublicKeys, outpointHash, outpointsByP2TRpubkey.keys.toList()); if (result.isEmpty) { // no results tx, save checkpoint in case of issues and continue to next tx checkpointTxPos = pos; pos++; continue; } print("UNSPENT COIN FOUND!"); print(result); result.forEach((key, value) { final outpoint = outpointsByP2TRpubkey[key]; if (outpoint == null) { return; } // found utxo for tx scanData.sendPort.send(BitcoinUnspent( BitcoinAddressRecord( key, index: 0, isHidden: false, isUsed: true, silentAddressLabel: null, silentPaymentTweak: value, ), outpoint.txid, outpoint.value!, outpoint.index, silentPaymentTweak: value, )); }); } catch (e, stacktrace) { print(stacktrace); print(e.toString()); } pos++; } catch (e, stacktrace) { print(stacktrace); print(e.toString()); // last position, no more txs for the given block height break; } } final newHeight = height + 1; if (newHeight < currentChainTip) { // recursive, scan next block again until at the current tip return startRefresh(ScanData.withCheckpoint(scanData, newHeight, 1)); } // otherwise, finished scanning scanData.sendPort.send(SyncResponse(currentChainTip, SyncedSyncStatus())); } catch (e, stacktrace) { print(stacktrace); print(e.toString()); startRefresh(ScanData.withCheckpoint(scanData, height, checkpointTxPos)); } }