feat: add more heartbeats on sapling address generation (#120)

* feat: add more heartbeats on sapling address generation

* chore: bump version

* chore: update snapshots

Author: emmanuelm41

app/Makefile.version

@@ -3,4 +3,4 @@ APPVERSION_M=4
 # This is the minor version
 APPVERSION_N=1
 # This is the patch version
-APPVERSION_P=3
+APPVERSION_P=4

app/rust/src/bolos/blake2b.rs

@@ -241,11 +241,23 @@ pub fn blake2b_expand_v4(
         .hash_length(64)
         .personal(PRF_EXPAND_PERSONALIZATION)
         .to_state();
+    crate::bolos::heartbeat();
+
     blake2b_state.update(in_a);
+    crate::bolos::heartbeat();
+
     blake2b_state.update(in_b);
+    crate::bolos::heartbeat();
+
     blake2b_state.update(in_c);
+    crate::bolos::heartbeat();
+
     blake2b_state.update(in_d);
+    crate::bolos::heartbeat();
+
     blake2b_state.update(in_e);
+    crate::bolos::heartbeat();
+
     let mut hash = [0u8; 64];
     hash.copy_from_slice(blake2b_state.finalize().as_bytes());
     hash

app/rust/src/zip32.rs

@@ -263,6 +263,8 @@ fn zip32_sapling_derive_child(
     path_i: u32,
     key_bundle_i: &mut SaplingKeyBundle,
 ) {
+    crate::bolos::heartbeat();
+
     let hardened = (path_i & 0x8000_0000) != 0;
     let c = path_i & 0x7FFF_FFFF;
 
@@ -321,6 +323,7 @@ fn zip32_sapling_derive_child(
 pub fn zip32_sapling_derive(path: &Zip32Path) -> SaplingKeyBundle {
     // ik as in capital I (https://zips.z.cash/zip-0032#sapling-child-key-derivation)
     let mut ik = zip32_master_key_i();
+    crate::bolos::heartbeat();
 
     let mut key_bundle_i = SaplingKeyBundle::new(
         zip32_sapling_ask_m(&ik.spending_key()),

app/rust/src/zip32_extern.rs

@@ -151,19 +151,25 @@ pub extern "C" fn get_pkd_from_seed(
     pkd_ptr: *mut [u8; 32],
 ) {
     crate::bolos::heartbeat();
+
     let path = [ZIP32_PURPOSE, ZIP32_COIN_TYPE, account];
     let start = unsafe { &mut *start_diversifier };
     let div_out = unsafe { &mut *div_ptr };
 
     let key_bundle = zip32_sapling_derive(&path);
+    crate::bolos::heartbeat();
+
     let dk = key_bundle.dk();
+    crate::bolos::heartbeat();
 
     div_out.copy_from_slice(&zip32::diversifier_find_valid(&dk, start));
     crate::bolos::heartbeat();
 
     let ivk = sapling_asknsk_to_ivk(&key_bundle.ask(), &key_bundle.nsk());
     crate::bolos::heartbeat();
+
     let tmp_pkd = zip32::pkd_default(&ivk, div_out);
+    crate::bolos::heartbeat();
 
     let pkd_out = unsafe { &mut *pkd_ptr };
     pkd_out.copy_from_slice(&tmp_pkd);