79 lines
2.5 KiB
Rust
79 lines
2.5 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use zcash_primitives::{
|
|
consensus::{self, BlockHeight},
|
|
memo::MemoBytes,
|
|
sapling::{
|
|
note_encryption::{try_sapling_note_decryption, try_sapling_output_recovery},
|
|
Note, PaymentAddress,
|
|
},
|
|
transaction::Transaction,
|
|
zip32::AccountId,
|
|
};
|
|
|
|
use crate::keys::UnifiedFullViewingKey;
|
|
|
|
/// A decrypted shielded output.
|
|
pub struct DecryptedOutput {
|
|
/// The index of the output within [`shielded_outputs`].
|
|
///
|
|
/// [`shielded_outputs`]: zcash_primitives::transaction::TransactionData
|
|
pub index: usize,
|
|
/// The note within the output.
|
|
pub note: Note,
|
|
/// The account that decrypted the note.
|
|
pub account: AccountId,
|
|
/// The address the note was sent to.
|
|
pub to: PaymentAddress,
|
|
/// The memo bytes included with the note.
|
|
pub memo: MemoBytes,
|
|
/// True if this output was recovered using an [`OutgoingViewingKey`], meaning that
|
|
/// this is a logical output of the transaction.
|
|
///
|
|
/// [`OutgoingViewingKey`]: zcash_primitives::keys::OutgoingViewingKey
|
|
pub outgoing: bool,
|
|
}
|
|
|
|
/// Scans a [`Transaction`] for any information that can be decrypted by the set of
|
|
/// [`UnifiedFullViewingKey`]s.
|
|
pub fn decrypt_transaction<P: consensus::Parameters>(
|
|
params: &P,
|
|
height: BlockHeight,
|
|
tx: &Transaction,
|
|
ufvks: &HashMap<AccountId, UnifiedFullViewingKey>,
|
|
) -> Vec<DecryptedOutput> {
|
|
let mut decrypted = vec![];
|
|
|
|
if let Some(bundle) = tx.sapling_bundle() {
|
|
for (account, ufvk) in ufvks.iter() {
|
|
if let Some(dfvk) = ufvk.sapling() {
|
|
let ivk = dfvk.fvk().vk.ivk();
|
|
let ovk = dfvk.fvk().ovk;
|
|
|
|
for (index, output) in bundle.shielded_outputs.iter().enumerate() {
|
|
let ((note, to, memo), outgoing) =
|
|
match try_sapling_note_decryption(params, height, &ivk, output) {
|
|
Some(ret) => (ret, false),
|
|
None => match try_sapling_output_recovery(params, height, &ovk, output)
|
|
{
|
|
Some(ret) => (ret, true),
|
|
None => continue,
|
|
},
|
|
};
|
|
|
|
decrypted.push(DecryptedOutput {
|
|
index,
|
|
note,
|
|
account: *account,
|
|
to,
|
|
memo,
|
|
outgoing,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
decrypted
|
|
}
|