196 lines
8.1 KiB
C#
196 lines
8.1 KiB
C#
|
using NBitcoin;
|
||
|
|
using PalladiumWallet.Core.Chain;
|
||
|
|
using PalladiumWallet.Core.Crypto;
|
||
|
|
using PalladiumWallet.Core.Net;
|
||
|
|
using PalladiumWallet.Core.Storage;
|
||
|
|
|
||
|
|
namespace PalladiumWallet.Core.Spv;
|
||
|
|
|
||
|
|
/// <summary>Indirizzo derivato e tracciato durante la sincronizzazione.</summary>
|
||
|
|
public sealed record TrackedAddress(
|
||
|
|
BitcoinAddress Address, string ScriptHash, bool IsChange, int Index)
|
||
|
|
{
|
||
|
|
public Script ScriptPubKey => Address.ScriptPubKey;
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>Esito di una passata di sincronizzazione.</summary>
|
||
|
|
public sealed class SyncResult
|
||
|
|
{
|
||
|
|
public required int TipHeight { get; init; }
|
||
|
|
public required long ConfirmedSats { get; init; }
|
||
|
|
public required long UnconfirmedSats { get; init; }
|
||
|
|
public required int NextReceiveIndex { get; init; }
|
||
|
|
public required int NextChangeIndex { get; init; }
|
||
|
|
public required IReadOnlyList<CachedTx> History { get; init; }
|
||
|
|
public required IReadOnlyList<CachedUtxo> Utxos { get; init; }
|
||
|
|
public required IReadOnlyList<TrackedAddress> Addresses { get; init; }
|
||
|
|
public required IReadOnlyDictionary<string, Transaction> Transactions { get; init; }
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Sincronizzazione del wallet (blueprint §7.4): per ogni indirizzo calcola lo
|
||
|
|
/// scripthash e si sottoscrive; scarica storico e transazioni; verifica ogni tx
|
||
|
|
/// confermata con la prova di Merkle contro l'header del suo blocco (le risposte
|
||
|
|
/// del server non sono fidate, §17); ricostruisce localmente UTXO e saldo;
|
||
|
|
/// estende la scansione fino al gap limit (§5).
|
||
|
|
/// </summary>
|
||
|
|
public sealed class WalletSynchronizer(HdAccount account, ElectrumClient client, int gapLimit = 20)
|
||
|
|
{
|
||
|
|
/// <summary>Avanzamento leggibile (per CLI e barra di stato GUI).</summary>
|
||
|
|
public event Action<string>? Progress;
|
||
|
|
|
||
|
|
public async Task<SyncResult> SyncOnceAsync(CancellationToken ct = default)
|
||
|
|
{
|
||
|
|
var tip = await client.SubscribeHeadersAsync(ct);
|
||
|
|
Progress?.Invoke($"tip della catena: {tip.Height}");
|
||
|
|
|
||
|
|
// 1-2. Scansione indirizzi con gap limit, per catena receiving e change.
|
||
|
|
var tracked = new List<TrackedAddress>();
|
||
|
|
var historyByAddress = new Dictionary<string, IReadOnlyList<HistoryItem>>();
|
||
|
|
var nextReceive = await ScanChainAsync(isChange: false, tracked, historyByAddress, ct);
|
||
|
|
var nextChange = await ScanChainAsync(isChange: true, tracked, historyByAddress, ct);
|
||
|
|
|
||
|
|
// 3. Storico unico (txid → altezza massima riportata).
|
||
|
|
var txHeights = new Dictionary<string, int>();
|
||
|
|
foreach (var item in historyByAddress.Values.SelectMany(h => h))
|
||
|
|
txHeights[item.TxHash] = item.Height;
|
||
|
|
|
||
|
|
// 4. Scarica le transazioni.
|
||
|
|
Progress?.Invoke($"scarico {txHeights.Count} transazioni…");
|
||
|
|
var network = PalladiumNetworks.For(account.Profile.Kind);
|
||
|
|
var transactions = new Dictionary<string, Transaction>();
|
||
|
|
foreach (var txid in txHeights.Keys)
|
||
|
|
{
|
||
|
|
var hex = await client.GetTransactionAsync(txid, ct);
|
||
|
|
transactions[txid] = Transaction.Parse(hex, network);
|
||
|
|
}
|
||
|
|
|
||
|
|
// 5. Verifica Merkle delle confermate (§7.4 punto 4).
|
||
|
|
var verified = new Dictionary<string, bool>();
|
||
|
|
foreach (var (txid, height) in txHeights)
|
||
|
|
{
|
||
|
|
if (height <= 0)
|
||
|
|
{
|
||
|
|
verified[txid] = false; // in mempool: nessuna prova possibile
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
var proof = await client.GetMerkleAsync(txid, height, ct);
|
||
|
|
var header = BlockHeaderInfo.Parse(await client.GetBlockHeaderAsync(height, ct));
|
||
|
|
var ok = MerkleProof.Verify(
|
||
|
|
uint256.Parse(txid),
|
||
|
|
proof.Pos,
|
||
|
|
proof.Merkle.Select(uint256.Parse),
|
||
|
|
header.MerkleRoot);
|
||
|
|
if (!ok)
|
||
|
|
throw new SpvVerificationException(
|
||
|
|
$"Prova di Merkle non valida per {txid} (blocco {height}): server non affidabile.");
|
||
|
|
verified[txid] = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
// 6. Ricostruzione locale degli UTXO: accrediti = output verso nostri
|
||
|
|
// script; spesi = outpoint consumati da una qualunque tx del wallet.
|
||
|
|
var byScript = tracked.ToDictionary(t => t.ScriptPubKey, t => t);
|
||
|
|
var spent = transactions.Values
|
||
|
|
.SelectMany(tx => tx.Inputs)
|
||
|
|
.Select(i => i.PrevOut)
|
||
|
|
.ToHashSet();
|
||
|
|
|
||
|
|
var utxos = new List<CachedUtxo>();
|
||
|
|
foreach (var (txid, tx) in transactions)
|
||
|
|
{
|
||
|
|
for (var vout = 0; vout < tx.Outputs.Count; vout++)
|
||
|
|
{
|
||
|
|
var output = tx.Outputs[vout];
|
||
|
|
if (!byScript.TryGetValue(output.ScriptPubKey, out var addr))
|
||
|
|
continue;
|
||
|
|
if (spent.Contains(new OutPoint(tx, vout)))
|
||
|
|
continue;
|
||
|
|
utxos.Add(new CachedUtxo
|
||
|
|
{
|
||
|
|
Txid = txid,
|
||
|
|
Vout = vout,
|
||
|
|
ValueSats = output.Value.Satoshi,
|
||
|
|
Address = addr.Address.ToString(),
|
||
|
|
IsChange = addr.IsChange,
|
||
|
|
AddressIndex = addr.Index,
|
||
|
|
Height = txHeights[txid],
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 7. Delta per voce di storico (entrate - uscite del wallet).
|
||
|
|
var history = new List<CachedTx>();
|
||
|
|
foreach (var (txid, tx) in transactions)
|
||
|
|
{
|
||
|
|
var received = tx.Outputs
|
||
|
|
.Where(o => byScript.ContainsKey(o.ScriptPubKey))
|
||
|
|
.Sum(o => o.Value.Satoshi);
|
||
|
|
var sentSats = tx.Inputs
|
||
|
|
.Where(i => transactions.TryGetValue(i.PrevOut.Hash.ToString(), out var prev)
|
||
|
|
&& byScript.ContainsKey(prev.Outputs[i.PrevOut.N].ScriptPubKey))
|
||
|
|
.Sum(i => transactions[i.PrevOut.Hash.ToString()].Outputs[i.PrevOut.N].Value.Satoshi);
|
||
|
|
history.Add(new CachedTx
|
||
|
|
{
|
||
|
|
Txid = txid,
|
||
|
|
Height = txHeights[txid],
|
||
|
|
DeltaSats = received - sentSats,
|
||
|
|
Verified = verified[txid],
|
||
|
|
});
|
||
|
|
}
|
||
|
|
history.Sort((a, b) =>
|
||
|
|
{
|
||
|
|
// Non confermate (height<=0) in cima, poi per altezza decrescente.
|
||
|
|
var ha = a.Height <= 0 ? int.MaxValue : a.Height;
|
||
|
|
var hb = b.Height <= 0 ? int.MaxValue : b.Height;
|
||
|
|
return hb.CompareTo(ha);
|
||
|
|
});
|
||
|
|
|
||
|
|
return new SyncResult
|
||
|
|
{
|
||
|
|
TipHeight = tip.Height,
|
||
|
|
ConfirmedSats = utxos.Where(u => u.Height > 0).Sum(u => u.ValueSats),
|
||
|
|
UnconfirmedSats = utxos.Where(u => u.Height <= 0).Sum(u => u.ValueSats),
|
||
|
|
NextReceiveIndex = nextReceive,
|
||
|
|
NextChangeIndex = nextChange,
|
||
|
|
History = history,
|
||
|
|
Utxos = utxos,
|
||
|
|
Addresses = tracked,
|
||
|
|
Transactions = transactions,
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Scansiona una catena (receiving o change) finché trova gapLimit indirizzi
|
||
|
|
/// vuoti consecutivi (§5). Ritorna il primo indice non usato.
|
||
|
|
/// </summary>
|
||
|
|
private async Task<int> ScanChainAsync(bool isChange, List<TrackedAddress> tracked,
|
||
|
|
Dictionary<string, IReadOnlyList<HistoryItem>> historyByAddress, CancellationToken ct)
|
||
|
|
{
|
||
|
|
var consecutiveEmpty = 0;
|
||
|
|
var index = 0;
|
||
|
|
var firstUnused = 0;
|
||
|
|
for (; consecutiveEmpty < gapLimit; index++)
|
||
|
|
{
|
||
|
|
var address = account.GetAddress(isChange, index);
|
||
|
|
var scripthash = Scripthash.FromAddress(address);
|
||
|
|
tracked.Add(new TrackedAddress(address, scripthash, isChange, index));
|
||
|
|
|
||
|
|
// La subscribe registra anche la notifica push per i cambi futuri.
|
||
|
|
var status = await client.SubscribeScripthashAsync(scripthash, ct);
|
||
|
|
if (status is null)
|
||
|
|
{
|
||
|
|
consecutiveEmpty++;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
historyByAddress[scripthash] = await client.GetHistoryAsync(scripthash, ct);
|
||
|
|
consecutiveEmpty = 0;
|
||
|
|
firstUnused = index + 1;
|
||
|
|
}
|
||
|
|
return firstUnused;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>La verifica SPV è fallita: i dati del server contraddicono le prove (§17).</summary>
|
||
|
|
public sealed class SpvVerificationException(string message) : Exception(message);
|