Skip to content

Commit

Permalink
Rpc: Manually add immature coinbase utxos
Browse files Browse the repository at this point in the history
Before this commit, the rpc backend would not notice immature utxos
(`listunspent` does not return them), making the rpc balance different
to other blockchain implementations.

Co-authored-by: Daniela Brozzoni <danielabrozzoni@protonmail.com>
  • Loading branch information
evanlinjin and danielabrozzoni committed Jul 30, 2022
1 parent 9d787bf commit 783b975
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 26 deletions.
44 changes: 25 additions & 19 deletions src/blockchain/rpc.rs
Original file line number Diff line number Diff line change
Expand Up @@ -424,24 +424,29 @@ impl DbState {
// check if tx has an immature coinbase output (add to updated UTXOs)
// this is required because `listunspent` does not include immature coinbase outputs
if tx_res.detail.category == GetTransactionResultDetailCategory::Immature {
// let vout = tx_res.detail.vout;
// let txout = raw_tx.output.get(vout as usize).cloned().ok_or_else(|| {
// Error::Generic(format!(
// "Core RPC returned detail with invalid vout '{}' for tx '{}'",
// vout, tx_res.info.txid,
// ))
// })?;
// println!("got immature detail!");

// if let Some((keychain, _)) = db.get_path_from_script_pubkey(&txout.script_pubkey)? {
// let utxo = LocalUtxo {
// outpoint: OutPoint::new(tx_res.info.txid, d.vout),
// txout,
// keychain,
// is_spent: false,
// };
// self.updated_utxos.insert(utxo);
// }
let txout = raw_tx
.output
.get(tx_res.detail.vout as usize)
.cloned()
.ok_or_else(|| {
Error::Generic(format!(
"Core RPC returned detail with invalid vout '{}' for tx '{}'",
tx_res.detail.vout, tx_res.info.txid,
))
})?;

if let Some((keychain, index)) =
db.get_path_from_script_pubkey(&txout.script_pubkey)?
{
let utxo = LocalUtxo {
outpoint: OutPoint::new(tx_res.info.txid, tx_res.detail.vout),
txout,
keychain,
is_spent: false,
};
self.updated_utxos.insert(utxo);
self._update_last_index(keychain, index);
}
}

// update tx deltas
Expand Down Expand Up @@ -511,7 +516,7 @@ impl DbState {
let new_utxos = core_utxos.difference(&self.utxos).cloned();

// add to updated utxos
self.updated_utxos = spent_utxos.chain(new_utxos).collect();
self.updated_utxos.extend(spent_utxos.chain(new_utxos));

Ok(self)
}
Expand Down Expand Up @@ -604,6 +609,7 @@ impl DbState {
// update utxos
self.updated_utxos
.iter()
.inspect(|&utxo| println!("updating: {:?}", utxo.txout))
.try_for_each(|utxo| batch.set_utxo(utxo))?;

// update last indexes
Expand Down
7 changes: 0 additions & 7 deletions src/testutils/blockchain_tests.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1064,13 +1064,6 @@ macro_rules! bdk_blockchain_tests {

test_client.generate(1, Some(wallet_addr));

#[cfg(feature = "rpc")]
{
// rpc consider coinbase only when mature (100 blocks)
let node_addr = test_client.get_node_address(None);
test_client.generate(100, Some(node_addr));
}

wallet.sync(&blockchain, SyncOptions::default()).unwrap();
assert!(wallet.get_balance().unwrap() > 0, "incorrect balance after receiving coinbase");
}
Expand Down

0 comments on commit 783b975

Please sign in to comment.