2018-10-15 07:51:40 -07:00
|
|
|
//! Helper functions for managing light client key material.
|
2022-01-24 17:02:25 -08:00
|
|
|
use zcash_primitives::{consensus, zip32::AccountId};
|
|
|
|
|
|
|
|
#[cfg(feature = "transparent-inputs")]
|
|
|
|
use zcash_primitives::legacy::keys as legacy;
|
2018-10-15 07:51:40 -07:00
|
|
|
|
2022-01-20 19:01:33 -08:00
|
|
|
pub mod sapling {
|
2022-01-24 17:02:25 -08:00
|
|
|
use zcash_primitives::zip32::{AccountId, ChildIndex};
|
2022-01-21 17:20:12 -08:00
|
|
|
pub use zcash_primitives::zip32::{ExtendedFullViewingKey, ExtendedSpendingKey};
|
2020-12-22 06:10:13 -08:00
|
|
|
|
2022-01-20 19:01:33 -08:00
|
|
|
/// Derives the ZIP 32 [`ExtendedSpendingKey`] for a given coin type and account from the
|
|
|
|
/// given seed.
|
|
|
|
///
|
|
|
|
/// # Panics
|
|
|
|
///
|
|
|
|
/// Panics if `seed` is shorter than 32 bytes.
|
|
|
|
///
|
|
|
|
/// # Examples
|
|
|
|
///
|
|
|
|
/// ```
|
2022-01-24 17:02:25 -08:00
|
|
|
/// use zcash_primitives::{
|
|
|
|
/// constants::testnet::COIN_TYPE,
|
|
|
|
/// zip32::AccountId,
|
|
|
|
/// };
|
2022-01-20 19:01:33 -08:00
|
|
|
/// use zcash_client_backend::{
|
|
|
|
/// keys::sapling,
|
|
|
|
/// };
|
|
|
|
///
|
2022-02-10 08:47:42 -08:00
|
|
|
/// let extsk = sapling::spending_key(&[0; 32][..], COIN_TYPE, AccountId::from(0));
|
2022-01-20 19:01:33 -08:00
|
|
|
/// ```
|
|
|
|
/// [`ExtendedSpendingKey`]: zcash_primitives::zip32::ExtendedSpendingKey
|
|
|
|
pub fn spending_key(seed: &[u8], coin_type: u32, account: AccountId) -> ExtendedSpendingKey {
|
|
|
|
if seed.len() < 32 {
|
|
|
|
panic!("ZIP 32 seeds MUST be at least 32 bytes");
|
|
|
|
}
|
2021-10-04 13:09:02 -07:00
|
|
|
|
2022-01-20 19:01:33 -08:00
|
|
|
ExtendedSpendingKey::from_path(
|
2022-02-01 13:02:07 -08:00
|
|
|
&ExtendedSpendingKey::master(seed),
|
2022-01-20 19:01:33 -08:00
|
|
|
&[
|
|
|
|
ChildIndex::Hardened(32),
|
|
|
|
ChildIndex::Hardened(coin_type),
|
2022-02-10 08:47:42 -08:00
|
|
|
ChildIndex::Hardened(account.into()),
|
2022-01-20 19:01:33 -08:00
|
|
|
],
|
|
|
|
)
|
2019-09-17 10:03:58 -07:00
|
|
|
}
|
2018-10-15 07:51:40 -07:00
|
|
|
}
|
2019-09-17 15:58:14 -07:00
|
|
|
|
2022-01-21 18:01:32 -08:00
|
|
|
#[derive(Debug)]
|
2022-02-01 10:37:43 -08:00
|
|
|
#[doc(hidden)]
|
2022-01-21 12:47:03 -08:00
|
|
|
pub enum DerivationError {
|
|
|
|
#[cfg(feature = "transparent-inputs")]
|
|
|
|
Transparent(hdwallet::error::Error),
|
|
|
|
}
|
|
|
|
|
|
|
|
/// A set of viewing keys that are all associated with a single
|
|
|
|
/// ZIP-0032 account identifier.
|
|
|
|
#[derive(Clone, Debug)]
|
2022-02-01 10:37:43 -08:00
|
|
|
#[doc(hidden)]
|
2022-01-21 12:47:03 -08:00
|
|
|
pub struct UnifiedSpendingKey {
|
|
|
|
account: AccountId,
|
|
|
|
#[cfg(feature = "transparent-inputs")]
|
2022-01-24 17:02:25 -08:00
|
|
|
transparent: legacy::AccountPrivKey,
|
2022-01-21 17:20:12 -08:00
|
|
|
sapling: sapling::ExtendedSpendingKey,
|
2022-01-21 12:47:03 -08:00
|
|
|
}
|
|
|
|
|
2022-02-01 10:37:43 -08:00
|
|
|
#[doc(hidden)]
|
2022-01-21 12:47:03 -08:00
|
|
|
impl UnifiedSpendingKey {
|
|
|
|
pub fn from_seed<P: consensus::Parameters>(
|
|
|
|
params: &P,
|
|
|
|
seed: &[u8],
|
|
|
|
account: AccountId,
|
|
|
|
) -> Result<UnifiedSpendingKey, DerivationError> {
|
|
|
|
if seed.len() < 32 {
|
|
|
|
panic!("ZIP 32 seeds MUST be at least 32 bytes");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(feature = "transparent-inputs")]
|
2022-01-24 17:02:25 -08:00
|
|
|
let transparent = legacy::AccountPrivKey::from_seed(params, seed, account)
|
2022-01-21 12:47:03 -08:00
|
|
|
.map_err(DerivationError::Transparent)?;
|
|
|
|
|
|
|
|
Ok(UnifiedSpendingKey {
|
|
|
|
account,
|
|
|
|
#[cfg(feature = "transparent-inputs")]
|
|
|
|
transparent,
|
|
|
|
sapling: sapling::spending_key(seed, params.coin_type(), account),
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn to_unified_full_viewing_key(&self) -> UnifiedFullViewingKey {
|
|
|
|
UnifiedFullViewingKey {
|
|
|
|
account: self.account,
|
|
|
|
#[cfg(feature = "transparent-inputs")]
|
|
|
|
transparent: Some(self.transparent.to_account_pubkey()),
|
2022-01-21 17:20:12 -08:00
|
|
|
sapling: Some(sapling::ExtendedFullViewingKey::from(&self.sapling)),
|
2022-01-21 12:47:03 -08:00
|
|
|
}
|
|
|
|
}
|
2022-01-21 17:20:12 -08:00
|
|
|
|
|
|
|
pub fn account(&self) -> AccountId {
|
|
|
|
self.account
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Returns the transparent component of the unified key at the
|
|
|
|
/// BIP44 path `m/44'/<coin_type>'/<account>'`.
|
|
|
|
#[cfg(feature = "transparent-inputs")]
|
2022-01-24 17:02:25 -08:00
|
|
|
pub fn transparent(&self) -> &legacy::AccountPrivKey {
|
2022-01-21 18:01:32 -08:00
|
|
|
&self.transparent
|
2022-01-21 17:20:12 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Returns the Sapling extended full viewing key component of this
|
|
|
|
/// unified key.
|
2022-01-21 18:01:32 -08:00
|
|
|
pub fn sapling(&self) -> &sapling::ExtendedSpendingKey {
|
|
|
|
&self.sapling
|
2022-01-21 17:20:12 -08:00
|
|
|
}
|
2022-01-21 12:47:03 -08:00
|
|
|
}
|
|
|
|
|
2021-10-04 13:09:02 -07:00
|
|
|
/// A set of viewing keys that are all associated with a single
|
|
|
|
/// ZIP-0032 account identifier.
|
|
|
|
#[derive(Clone, Debug)]
|
2022-02-01 10:37:43 -08:00
|
|
|
#[doc(hidden)]
|
2021-10-04 13:09:02 -07:00
|
|
|
pub struct UnifiedFullViewingKey {
|
|
|
|
account: AccountId,
|
2022-01-20 19:55:34 -08:00
|
|
|
#[cfg(feature = "transparent-inputs")]
|
2022-01-24 17:02:25 -08:00
|
|
|
transparent: Option<legacy::AccountPubKey>,
|
2022-02-01 10:37:43 -08:00
|
|
|
// TODO: This type is invalid for a UFVK; create a `sapling::DiversifiableFullViewingKey`
|
|
|
|
// to replace it.
|
2022-01-20 19:01:33 -08:00
|
|
|
sapling: Option<sapling::ExtendedFullViewingKey>,
|
2021-10-04 13:09:02 -07:00
|
|
|
}
|
|
|
|
|
2022-02-01 10:37:43 -08:00
|
|
|
#[doc(hidden)]
|
2021-10-04 13:09:02 -07:00
|
|
|
impl UnifiedFullViewingKey {
|
|
|
|
/// Construct a new unified full viewing key, if the required components are present.
|
|
|
|
pub fn new(
|
|
|
|
account: AccountId,
|
2022-01-24 17:02:25 -08:00
|
|
|
#[cfg(feature = "transparent-inputs")] transparent: Option<legacy::AccountPubKey>,
|
2022-01-20 19:01:33 -08:00
|
|
|
sapling: Option<sapling::ExtendedFullViewingKey>,
|
2021-10-04 13:09:02 -07:00
|
|
|
) -> Option<UnifiedFullViewingKey> {
|
|
|
|
if sapling.is_none() {
|
|
|
|
None
|
|
|
|
} else {
|
|
|
|
Some(UnifiedFullViewingKey {
|
|
|
|
account,
|
2022-01-20 19:55:34 -08:00
|
|
|
#[cfg(feature = "transparent-inputs")]
|
2021-10-04 13:09:02 -07:00
|
|
|
transparent,
|
|
|
|
sapling,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Returns the ZIP32 account identifier to which all component
|
|
|
|
/// keys are related.
|
|
|
|
pub fn account(&self) -> AccountId {
|
|
|
|
self.account
|
|
|
|
}
|
|
|
|
|
2022-01-20 19:01:33 -08:00
|
|
|
/// Returns the transparent component of the unified key at the
|
|
|
|
/// BIP44 path `m/44'/<coin_type>'/<account>'`.
|
2022-02-01 10:37:43 -08:00
|
|
|
#[cfg(feature = "transparent-inputs")]
|
2022-01-24 17:02:25 -08:00
|
|
|
pub fn transparent(&self) -> Option<&legacy::AccountPubKey> {
|
2021-10-04 13:09:02 -07:00
|
|
|
self.transparent.as_ref()
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Returns the Sapling extended full viewing key component of this
|
|
|
|
/// unified key.
|
2022-01-20 19:01:33 -08:00
|
|
|
pub fn sapling(&self) -> Option<&sapling::ExtendedFullViewingKey> {
|
2021-10-04 13:09:02 -07:00
|
|
|
self.sapling.as_ref()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-09-17 15:58:14 -07:00
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
2022-01-20 19:01:33 -08:00
|
|
|
use super::sapling;
|
2022-01-24 17:02:25 -08:00
|
|
|
use zcash_primitives::zip32::AccountId;
|
2019-09-17 15:58:14 -07:00
|
|
|
|
2021-02-12 14:54:59 -08:00
|
|
|
#[cfg(feature = "transparent-inputs")]
|
|
|
|
use {
|
2022-01-24 17:02:25 -08:00
|
|
|
crate::encoding::AddressCodec,
|
2022-01-26 12:24:56 -08:00
|
|
|
zcash_primitives::{consensus::MAIN_NETWORK, legacy, legacy::keys::IncomingViewingKey},
|
2021-02-12 14:54:59 -08:00
|
|
|
};
|
|
|
|
|
2021-10-04 13:09:02 -07:00
|
|
|
#[cfg(feature = "transparent-inputs")]
|
2021-03-31 18:04:38 -07:00
|
|
|
fn seed() -> Vec<u8> {
|
|
|
|
let seed_hex = "6ef5f84def6f4b9d38f466586a8380a38593bd47c8cda77f091856176da47f26b5bd1c8d097486e5635df5a66e820d28e1d73346f499801c86228d43f390304f";
|
|
|
|
hex::decode(&seed_hex).unwrap()
|
|
|
|
}
|
|
|
|
|
2019-09-17 15:58:14 -07:00
|
|
|
#[test]
|
|
|
|
#[should_panic]
|
|
|
|
fn spending_key_panics_on_short_seed() {
|
2022-02-10 08:47:42 -08:00
|
|
|
let _ = sapling::spending_key(&[0; 31][..], 0, AccountId::from(0));
|
2019-09-17 15:58:14 -07:00
|
|
|
}
|
2021-02-12 14:54:59 -08:00
|
|
|
|
2021-03-31 18:04:38 -07:00
|
|
|
#[cfg(feature = "transparent-inputs")]
|
|
|
|
#[test]
|
|
|
|
fn pk_to_taddr() {
|
2022-02-10 08:47:42 -08:00
|
|
|
let taddr =
|
|
|
|
legacy::keys::AccountPrivKey::from_seed(&MAIN_NETWORK, &seed(), AccountId::from(0))
|
|
|
|
.unwrap()
|
|
|
|
.to_account_pubkey()
|
|
|
|
.derive_external_ivk()
|
|
|
|
.unwrap()
|
|
|
|
.derive_address(0)
|
|
|
|
.unwrap()
|
|
|
|
.encode(&MAIN_NETWORK);
|
2021-03-31 18:04:38 -07:00
|
|
|
assert_eq!(taddr, "t1PKtYdJJHhc3Pxowmznkg7vdTwnhEsCvR4".to_string());
|
|
|
|
}
|
2019-09-17 15:58:14 -07:00
|
|
|
}
|