logos-blockchain/lssa
1
0
Fork 0
mirror of https://github.com/logos-blockchain/lssa.git synced 2026-07-29 02:53:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(wallet): locate shared-account spends by their update nullifier

Browse Source
This commit is contained in:
Artem Gureev 2026-07-28 17:37:50 +00:00 committed by agureev
parent b5d5d862af
commit 131ed6d025
2 changed files with 24 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lez/wallet/src/lib.rs
View File

@ -29,7 +29,7 @@ use lee_core::{
BlockId, Commitment, CommitmentSetDigest, MembershipProof, SharedSecretKey, account::Nonce,
program::InstructionData,
};
use log::info;
use log::{info, warn};
use sequencer_service_rpc::{RpcClient as _, SequencerClient};
use storage::Storage;
use tokio::io::AsyncWriteExt as _;
@ -691,6 +691,9 @@ impl WalletCore {
.key_chain()
.locate_spend(*acc_account_id, &tx.message)
else {
warn!(
"No note located for {acc_account_id}; cached state stays stale until the next sync"
);
continue;
};
let (kind, res_acc) =

26
lez/wallet/src/storage/key_chain.rs
View File

@ -451,15 +451,29 @@ impl UserKeyChain {
Some(new_nullifier)
}
/// Constructs the next nullifier based on current account state
/// of the ID.
fn next_update_nullifier(&self, account_id: AccountId) -> Option<Nullifier> {
if let Some(entry) = self.shared_private_account(account_id) {
let keys = self.derive_shared_account_keys(entry)?;
return Some(NullifierIndex::next_update_nullifier(
account_id,
&entry.account,
&keys.nullifier_secret_key,
));
}
let acc = self.private_account(account_id)?;
Some(NullifierIndex::next_update_nullifier(
account_id,
acc.account,
&acc.key_chain.private_key_holder.nullifier_secret_key,
))
}
#[must_use]
pub fn locate_spend(&self, account_id: AccountId, message: &Message) -> Option<usize> {
let init = Nullifier::for_account_initialization(&account_id);
let update = self.private_account(account_id).map(|acc| {
Nullifier::for_account_update(
&Commitment::new(&account_id, acc.account),
&acc.key_chain.private_key_holder.nullifier_secret_key,
)
});
let update = self.next_update_nullifier(account_id);
message
.new_nullifiers
.iter()