Browse Source

Merge pull request #49 from DenioD/fixzdust

improve sietch generation
checkpoints
Denio 4 years ago
committed by GitHub
parent
commit
62d2a10cbd
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      lib/src/lib.rs
  2. 7
      lib/src/lightclient.rs
  3. 44
      lib/src/lightwallet.rs

2
lib/src/lib.rs

@ -16,7 +16,7 @@ pub struct SaplingParams;
pub struct PubCertificate; pub struct PubCertificate;
pub const ANCHOR_OFFSET: u32 = 2; pub const ANCHOR_OFFSET: u32 = 0;
pub mod grpc_client { pub mod grpc_client {
tonic::include_proto!("cash.z.wallet.sdk.rpc"); tonic::include_proto!("cash.z.wallet.sdk.rpc");

7
lib/src/lightclient.rs

@ -902,8 +902,7 @@ impl LightClient {
pub fn do_new_sietchaddress(&self, addr_type: &str) -> Result<JsonValue, String> { pub fn do_new_sietchaddress(&self, addr_type: &str) -> Result<JsonValue, String> {
let zdust_address = {
let new_address = {
let wallet = self.wallet.write().unwrap(); let wallet = self.wallet.write().unwrap();
match addr_type { match addr_type {
@ -917,9 +916,7 @@ impl LightClient {
} }
}; };
self.do_save()?; Ok(array![zdust_address])
Ok(array![new_address])
} }
pub fn clear_state(&self) { pub fn clear_state(&self) {

44
lib/src/lightwallet.rs

@ -38,9 +38,13 @@ use zcash_primitives::{
zip32::{ExtendedFullViewingKey, ExtendedSpendingKey, ChildIndex}, zip32::{ExtendedFullViewingKey, ExtendedSpendingKey, ChildIndex},
JUBJUB, JUBJUB,
primitives::{PaymentAddress}, primitives::{PaymentAddress},
}; };
use crate::lightclient::{LightClientConfig}; use crate::lightclient::{LightClientConfig};
mod data; mod data;
@ -112,6 +116,7 @@ pub struct LightWallet {
extfvks: Arc<RwLock<Vec<ExtendedFullViewingKey>>>, extfvks: Arc<RwLock<Vec<ExtendedFullViewingKey>>>,
pub zaddress: Arc<RwLock<Vec<PaymentAddress<Bls12>>>>, pub zaddress: Arc<RwLock<Vec<PaymentAddress<Bls12>>>>,
// Transparent keys. If the wallet is locked, then the secret keys will be encrypted, // Transparent keys. If the wallet is locked, then the secret keys will be encrypted,
// but the addresses will be present. // but the addresses will be present.
@ -170,6 +175,25 @@ impl LightWallet {
(extsk, extfvk, address) (extsk, extfvk, address)
} }
fn get_sietch_from_bip39seed( bip39_seed: &[u8]) ->
PaymentAddress<Bls12> {
assert_eq!(bip39_seed.len(), 64);
let zdustextsk: ExtendedSpendingKey = ExtendedSpendingKey::from_path(
&ExtendedSpendingKey::master(bip39_seed),
&[
ChildIndex::Hardened(32),
],
);
let zdustextfvk = ExtendedFullViewingKey::from(&zdustextsk);
let zdustaddress = zdustextfvk.default_address().unwrap().1;
(zdustaddress)
}
pub fn is_shielded_address(addr: &String, config: &LightClientConfig) -> bool { pub fn is_shielded_address(addr: &String, config: &LightClientConfig) -> bool {
match address::RecipientAddress::from_str(addr, match address::RecipientAddress::from_str(addr,
config.hrp_sapling_address(), config.hrp_sapling_address(),
@ -469,13 +493,14 @@ impl LightWallet {
zaddr zaddr
} }
// Add a new Sietch Addr. This will derive a new zdust address from manipluated seed // Add a new Sietch Addr. This will derive a new zdust address from manipluated seed
pub fn add_zaddrdust(&self) -> String { pub fn add_zaddrdust(&self) -> String {
let mut seed_bytes = [0u8; 32];
let pos = self.extsks.read().unwrap().len() as u32;
// Use random generator to create a new Sietch seed every time when call. // Use random generator to create a new Sietch seed
let mut rng = rand::thread_rng(); let mut rng = rand::thread_rng();
let letter: String = rng.gen_range(b'A', b'Z').to_string(); let letter: String = rng.gen_range(b'A', b'Z').to_string();
@ -485,16 +510,17 @@ impl LightWallet {
let dust: &str = &my_string; let dust: &str = &my_string;
let mut system_rng = OsRng;
system_rng.fill(&mut seed_bytes);
let bip39_seed = bip39::Seed::new(&Mnemonic::from_entropy(&seed_bytes, Language::English).unwrap(), dust);
let bip39_seed = bip39::Seed::new(&Mnemonic::from_entropy(&self.seed, Language::English).unwrap(), dust); let zdustaddress = LightWallet::get_sietch_from_bip39seed(&bip39_seed.as_bytes());
let (_extsk, _extfvk, address) = let zdust = encode_payment_address("zs", &zdustaddress);
LightWallet::get_zaddr_from_bip39seed(&self.config, &bip39_seed.as_bytes(), pos);
let zaddr = encode_payment_address(self.config.hrp_sapling_address(), &address);
zaddr zdust
} }
/// Add a new t address to the wallet. This will derive a new address from the seed /// Add a new t address to the wallet. This will derive a new address from the seed
@ -597,7 +623,7 @@ impl LightWallet {
) )
} { } {
(Some(min_height), Some(max_height)) => { (Some(min_height), Some(max_height)) => {
let target_height = max_height + 1; let target_height = max_height;
// Select an anchor ANCHOR_OFFSET back from the target block, // Select an anchor ANCHOR_OFFSET back from the target block,
// unless that would be before the earliest block we have. // unless that would be before the earliest block we have.

Loading…
Cancel
Save