zecfoundation
/
zcashd
mirror of https://github.com/ZcashFoundation/zcashd.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #6458 from sellout/safer-note-selection 

Make pool selection order more flexible
This commit is contained in:
Kris Nuttycombe 2023-03-08 13:53:11 -07:00 committed by GitHub
parent fcdfb5e780 8c41fbcc85
commit 62a0e78e4a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/wallet/wallet.cpp
View File

@ -8063,9 +8063,6 @@ bool SpendableInputs::LimitToAmount(
// Fully shielded.
selectionOrder = {
OutputPool::Orchard,
// Pools below here are erased.
OutputPool::Transparent,
OutputPool::Sapling,
};
} else if (
recipientPools.count(OutputPool::Transparent) &&
@ -8075,9 +8072,6 @@ bool SpendableInputs::LimitToAmount(
// Fewer pools.
selectionOrder = {
OutputPool::Orchard,
// Pools below here are erased.
OutputPool::Transparent,
OutputPool::Sapling,
};
} else if (wouldSuffice(availableSapling + availableOrchard)) {
// Hide sender.
@ -8087,8 +8081,6 @@ bool SpendableInputs::LimitToAmount(
selectionOrder = {
OutputPool::Sapling,
OutputPool::Orchard,
// Pools below here are erased.
OutputPool::Transparent,
};
} else {
// Opportunistic shielding.
@ -8100,14 +8092,25 @@ bool SpendableInputs::LimitToAmount(
opportunisticShielding = true;
}
// Ensure we provided a total selection order (so that all unselected notes
// and coins are erased).
// Erase all notes and coins from pools that aren’t used in selection.
for (auto pool : available) {
bool poolIsPresent = false;
for (auto entry : selectionOrder) {
poolIsPresent |= entry == pool;
}
assert(poolIsPresent);
if (!poolIsPresent) {
switch (pool) {
case OutputPool::Transparent:
utxos.clear();
break;
case OutputPool::Sapling:
saplingNoteEntries.clear();
break;
case OutputPool::Orchard:
orchardNoteMetadata.clear();
break;
}
}
}
// Finally, select the remaining notes and coins based on this order.