zec
/
zcash-test-vectors
mirror of https://github.com/zcash/zcash-test-vectors.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Reduce run times of unified_{address, incoming_viewing_keys, full_viewing_keys}.py by over half 

by avoiding duplicated work in deriving levels of the key hierarchy above the account level.

Signed-off-by: Daira Hopwood <daira@jacaranda.org>
This commit is contained in:
Daira Hopwood 2022-02-11 16:12:16 +00:00
parent f8a462d859
commit 189cadb357
3 changed files with 62 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
zcash_test_vectors/unified_address.py
View File

@ -25,6 +25,18 @@ def main():
rand = Rand(randbytes(rng))
seed = bytes(range(32))
t_root_key = bip_0032.ExtendedSecretKey.master(seed)
t_purpose_key = t_root_key.child(hardened(44))
t_coin_key = t_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
s_root_key = sapling_zip32.ExtendedSpendingKey.master(seed)
s_purpose_key = s_root_key.child(hardened(32))
s_coin_key = s_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
o_root_key = orchard_key_components.ExtendedSpendingKey.master(seed)
o_purpose_key = o_root_key.child(hardened(32))
o_coin_key = o_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
test_vectors = []
for account in range(0, 20):
has_t_addr = rand.bool()
@ -39,25 +51,18 @@ def main():
j = 0
has_s_addr = rand.bool()
if has_s_addr:
root_key = sapling_zip32.ExtendedSpendingKey.master(seed)
purpose_key = root_key.child(hardened(32))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
j = account_key.find_j(0)
sapling_d = account_key.diversifier(j)
sapling_pk_d = account_key.pk_d(j)
s_account_key = s_coin_key.child(hardened(account))
j = s_account_key.find_j(0)
sapling_d = s_account_key.diversifier(j)
sapling_pk_d = s_account_key.pk_d(j)
sapling_raw_addr = sapling_d + bytes(sapling_pk_d)
else:
sapling_raw_addr = None
has_o_addr = (not has_s_addr) or rand.bool()
if has_o_addr:
root_key = orchard_key_components.ExtendedSpendingKey.master(seed)
purpose_key = root_key.child(hardened(32))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
orchard_fvk = orchard_key_components.FullViewingKey.from_spending_key(account_key)
o_account_key = o_coin_key.child(hardened(account))
orchard_fvk = orchard_key_components.FullViewingKey.from_spending_key(o_account_key)
orchard_d = orchard_fvk.diversifier(j)
orchard_pk_d = orchard_fvk.pk_d(j)
orchard_raw_addr = orchard_d + bytes(orchard_pk_d)
@ -66,14 +71,11 @@ def main():
is_p2pkh = rand.bool()
if has_t_addr and is_p2pkh:
root_key = bip_0032.ExtendedSecretKey.master(seed)
purpose_key = root_key.child(hardened(44))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
external_key = account_key.child(0)
index_key = account_key.child(j)
index_pubkey = index_key.public_key()
t_addr = index_pubkey.address()
t_account_key = t_coin_key.child(hardened(account))
t_external_key = t_account_key.child(0)
t_index_key = t_account_key.child(j)
t_index_pubkey = t_index_key.public_key()
t_addr = t_index_pubkey.address()
# include an unknown item 1/4 of the time
has_unknown_item = rand.bool() and rand.bool()

33
zcash_test_vectors/unified_full_viewing_keys.py
View File

@ -20,6 +20,18 @@ def main():
rand = Rand(randbytes(rng))
seed = bytes(range(32))
t_root_key = bip_0032.ExtendedSecretKey.master(seed)
t_purpose_key = t_root_key.child(hardened(44))
t_coin_key = t_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
s_root_key = sapling_zip32.ExtendedSpendingKey.master(seed)
s_purpose_key = s_root_key.child(hardened(32))
s_coin_key = s_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
o_root_key = orchard_key_components.ExtendedSpendingKey.master(seed)
o_purpose_key = o_root_key.child(hardened(32))
o_coin_key = o_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
test_vectors = []
for account in range(0, 20):
has_t_key = rand.bool()
@ -30,21 +42,15 @@ def main():
# "However, the [Transparent P2PKH] FVK uses the key at the Account level, i.e.
# at path m/44'/coin_type'/account', while the IVK uses the external (non-change)
# child key at the Change level, i.e. at path m/44'/coin_type'/account'/0."
root_key = bip_0032.ExtendedSecretKey.master(seed)
purpose_key = root_key.child(hardened(44))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
t_key_bytes = bytes(account_key.public_key())
t_account_key = t_coin_key.child(hardened(account))
t_key_bytes = bytes(t_account_key.public_key())
else:
t_key_bytes = None
has_s_key = rand.bool()
if has_s_key:
root_key = sapling_zip32.ExtendedSpendingKey.master(seed)
purpose_key = root_key.child(hardened(32))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
sapling_fvk = account_key.to_extended_fvk()
s_account_key = s_coin_key.child(hardened(account))
sapling_fvk = s_account_key.to_extended_fvk()
sapling_fvk_bytes = b"".join([
bytes(sapling_fvk.ak()),
bytes(sapling_fvk.nk()),
@ -56,11 +62,8 @@ def main():
has_o_key = (not has_s_key) or rand.bool()
if has_o_key:
root_key = orchard_key_components.ExtendedSpendingKey.master(seed)
purpose_key = root_key.child(hardened(32))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
orchard_fvk = orchard_key_components.FullViewingKey.from_spending_key(account_key)
o_account_key = o_coin_key.child(hardened(account))
orchard_fvk = orchard_key_components.FullViewingKey.from_spending_key(o_account_key)
orchard_fvk_bytes = b"".join([
bytes(orchard_fvk.ak),
bytes(orchard_fvk.nk),

38
zcash_test_vectors/unified_incoming_viewing_keys.py
View File

@ -20,6 +20,18 @@ def main():
rand = Rand(randbytes(rng))
seed = bytes(range(32))
t_root_key = bip_0032.ExtendedSecretKey.master(seed)
t_purpose_key = t_root_key.child(hardened(44))
t_coin_key = t_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
s_root_key = sapling_zip32.ExtendedSpendingKey.master(seed)
s_purpose_key = s_root_key.child(hardened(32))
s_coin_key = s_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
o_root_key = orchard_key_components.ExtendedSpendingKey.master(seed)
o_purpose_key = o_root_key.child(hardened(32))
o_coin_key = o_purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
test_vectors = []
for account in range(0, 20):
has_t_key = rand.bool()
@ -30,34 +42,26 @@ def main():
# "However, the [Transparent P2PKH] FVK uses the key at the Account level, i.e.
# at path m/44'/coin_type'/account', while the IVK uses the external (non-change)
# child key at the Change level, i.e. at path m/44'/coin_type'/account'/0."
root_key = bip_0032.ExtendedSecretKey.master(seed)
purpose_key = root_key.child(hardened(44))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
external_key = account_key.child(0)
t_key_bytes = bytes(external_key.public_key())
t_account_key = t_coin_key.child(hardened(account))
t_external_key = t_account_key.child(0)
t_key_bytes = bytes(t_external_key.public_key())
else:
t_key_bytes = None
has_s_key = rand.bool()
if has_s_key:
root_key = sapling_zip32.ExtendedSpendingKey.master(seed)
purpose_key = root_key.child(hardened(32))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
sapling_dk = account_key.to_extended_fvk().dk()
sapling_ivk = account_key.ivk()
s_account_key = s_coin_key.child(hardened(account))
sapling_fvk = s_account_key.to_extended_fvk()
sapling_dk = sapling_fvk.dk()
sapling_ivk = sapling_fvk.ivk()
sapling_ivk_bytes = bytes(sapling_dk) + bytes(sapling_ivk)
else:
sapling_ivk_bytes = None
has_o_key = (not has_s_key) or rand.bool()
if has_o_key:
root_key = orchard_key_components.ExtendedSpendingKey.master(seed)
purpose_key = root_key.child(hardened(32))
coin_key = purpose_key.child(hardened(ZCASH_MAIN_COINTYPE))
account_key = coin_key.child(hardened(account))
orchard_fvk = orchard_key_components.FullViewingKey.from_spending_key(account_key)
o_account_key = o_coin_key.child(hardened(account))
orchard_fvk = orchard_key_components.FullViewingKey.from_spending_key(o_account_key)
orchard_dk = orchard_fvk.dk
orchard_ivk = orchard_fvk.ivk()
orchard_ivk_bytes = bytes(orchard_dk) + bytes(orchard_ivk)