tasm-lib 9.0.0

use std::collections::HashMap;
use std::collections::HashSet;

use itertools::Itertools;
use num_traits::One;

use crate::data_type::DataType;
use crate::twenty_first::bfe;
use crate::twenty_first::prelude::*;
use crate::twenty_first::util_types::mmr::mmr_accumulator::MmrAccumulator;
use crate::twenty_first::util_types::mmr::shared_advanced::get_peak_heights;
use crate::twenty_first::util_types::mmr::shared_basic::leaf_index_to_mt_index_and_peak_index;

const ROOT_MT_INDEX: u64 = 1;

pub(super) fn indexed_leaf_element_type() -> DataType {
    DataType::Tuple(vec![DataType::U64, DataType::Digest])
}

/// A witness to facilitate the proving of the authenticity of a Merkle
/// authentication struct.
#[derive(Debug, Clone)]
pub struct AuthStructIntegrityProof {
    // All indices are Merkle tree node indices
    pub nd_auth_struct_indices: Vec<u64>,
    pub nd_sibling_indices: Vec<(u64, u64)>,
    pub nd_siblings: Vec<(Digest, Digest)>,
}

/// An authentication structure that can be used to prove membership of a list
/// of leaves in a Merkle tree, along with the indexed leaves in question, and
/// the witness necessary to prove membership in a ZK program.
#[derive(Debug, Clone)]
pub struct AuthenticatedMerkleAuthStruct {
    pub auth_struct: Vec<Digest>,
    pub indexed_leafs: Vec<(u64, Digest)>,
    pub witness: AuthStructIntegrityProof,
}

impl AuthStructIntegrityProof {
    /// Return the Merkle tree node indices of the digests required to prove
    /// membership for the specified leaf indices, as well as the node indices
    /// that can be derived from the leaf indices and their authentication
    /// path.
    fn auth_struct_and_nd_indices(
        num_leafs: u64,
        leaf_indices: &[u64],
    ) -> (Vec<u64>, Vec<(u64, u64)>) {
        // The set of indices of nodes that need to be included in the authentications
        // structure. In principle, every node of every authentication path is needed.
        // The root is never needed. Hence, it is not considered below.
        let mut node_is_needed = HashSet::new();

        // The set of indices of nodes that can be computed from other nodes in the
        // authentication structure or the leafs that are explicitly supplied during
        // verification. Every node on the direct path from the leaf to the root can
        // be computed by the very nature of “authentication path”.
        let mut node_can_be_computed = HashSet::new();

        for &leaf_index in leaf_indices {
            assert!(
                num_leafs > leaf_index,
                "Leaf index must be less than number of leafs. Got leaf_index = {leaf_index}; num_leafs = {num_leafs}"
            );

            let mut node_index = leaf_index + num_leafs;
            while node_index > ROOT_MT_INDEX {
                let sibling_index = node_index ^ 1;
                node_can_be_computed.insert(node_index);
                node_is_needed.insert(sibling_index);
                node_index /= 2;
            }
        }

        let set_difference = node_is_needed.difference(&node_can_be_computed).copied();
        let set_union = node_is_needed
            .union(&node_can_be_computed)
            .sorted_unstable()
            .rev();

        let mut set_union = set_union.peekable();

        let mut set_union_as_ordered_pairs = Vec::new();
        while set_union.peek().is_some() {
            let right_index = *set_union.next().unwrap();

            // Crashes on odd-length of input list, which is what we want, as
            // this acts as a sanity check.
            let left_index = *set_union.next().unwrap();
            set_union_as_ordered_pairs.push((left_index, right_index));
        }

        (
            set_difference.sorted_unstable().rev().collect(),
            set_union_as_ordered_pairs,
        )
    }

    /// Calculate a root from an authentication structure, indexed leafs, and
    /// additional witness data on `self`. Crashes if the witness data is
    /// incorrect and if the list of indexed leafs is empty.
    pub fn root_from_authentication_struct(
        &self,
        tree_height: u32,
        auth_struct: Vec<Digest>,
        indexed_leafs: Vec<(u64, Digest)>,
    ) -> Digest {
        fn digest_to_xfe(digest: Digest, challenge: XFieldElement) -> XFieldElement {
            let leaf_xfe_lo = XFieldElement::new([digest.0[0], digest.0[1], digest.0[2]]);
            let leaf_xfe_hi =
                challenge * XFieldElement::new([digest.0[3], digest.0[4], BFieldElement::one()]);

            leaf_xfe_lo + leaf_xfe_hi
        }

        fn node_index_to_bfe(node_index: u64) -> BFieldElement {
            BFieldElement::new(node_index)
        }

        // Sanity check
        assert_eq!(
            self.nd_auth_struct_indices.len(),
            auth_struct.len(),
            "Provided auth struct length must match that specified in receiver"
        );

        // Get challenges
        let (alpha, beta, gamma) = {
            let mut sponge = Tip5::init();
            sponge.pad_and_absorb_all(&indexed_leafs.encode());
            sponge.pad_and_absorb_all(&auth_struct.encode());
            let challenges = sponge.sample_scalars(3);
            (challenges[0], challenges[1], challenges[2])
        };

        // Accumulate `p` from public data
        let mut p = XFieldElement::one();
        for i in (0..indexed_leafs.len()).rev() {
            let node_index_as_bfe = node_index_to_bfe((1 << tree_height) ^ indexed_leafs[i].0);
            let leaf_as_xfe = digest_to_xfe(indexed_leafs[i].1, alpha);
            let fact = leaf_as_xfe - beta + gamma * node_index_as_bfe;
            p *= fact;
        }

        let mut prev = 0;
        for i in (0..auth_struct.len()).rev() {
            let auth_struct_index = self.nd_auth_struct_indices[i];

            // `auth_struct` must be sorted high-to-low by node-index. But since
            // we're traversing in reverse order, the inequality is flipped.
            assert!(auth_struct_index > prev);
            prev = auth_struct_index;

            let auth_struct_index_as_bfe = node_index_to_bfe(auth_struct_index);

            let auth_str_elem_as_xfe = digest_to_xfe(auth_struct[i], alpha);
            let fact = auth_str_elem_as_xfe - beta + gamma * auth_struct_index_as_bfe;
            p *= fact;
        }

        // Use secret data to invert `p` back and to calculate the root
        let mut t = indexed_leafs.first().unwrap().1;
        let mut t_xfe = digest_to_xfe(t, alpha);
        let mut parent_index_bfe = BFieldElement::one();
        for ((l, r), (left_index, right_index)) in self
            .nd_siblings
            .iter()
            .zip_eq(self.nd_sibling_indices.clone())
        {
            let left_index_bfe = node_index_to_bfe(left_index);
            let right_index_bfe = node_index_to_bfe(right_index);
            assert_eq!(left_index + 1, right_index);
            parent_index_bfe = left_index_bfe / bfe!(2);

            t = Tip5::hash_pair(*l, *r);

            let l_xfe = digest_to_xfe(*l, alpha);
            let r_xfe = digest_to_xfe(*r, alpha);
            t_xfe = digest_to_xfe(t, alpha);

            let fact_left = l_xfe - beta + gamma * left_index_bfe;
            let fact_right = r_xfe - beta + gamma * right_index_bfe;
            let fact_parent = t_xfe - beta + gamma * parent_index_bfe;

            p *= fact_left.inverse() * fact_right.inverse() * fact_parent;
        }

        assert_eq!(t_xfe - beta + gamma, p);

        assert!(parent_index_bfe.is_one());

        t
    }

    /// Return the authentication structure authenticity witness,
    /// authentication structure, and the (leaf-index, leaf-digest) pairs
    /// from a list of MMR membership proofs. All MMR membership proofs must
    /// belong under the same peak, i.e., be part of the same Merkle tree in
    /// the list of Merkle trees that the MMR contains.
    ///
    /// Panics if the input list of MMR-membership proofs is empty, or if they
    /// do not all belong under the same peak.
    pub fn new_from_mmr_membership_proofs(
        mmra: &MmrAccumulator,
        indexed_mmr_mps: Vec<(u64, Digest, MmrMembershipProof)>,
    ) -> HashMap<u32, AuthenticatedMerkleAuthStruct> {
        #[derive(Clone, Debug)]
        struct IndexedAuthenticatedMmrLeaf {
            merkle_tree_node_index: u64,
            merkle_tree_leaf_index: u64,
            leaf_digest: Digest,
            membership_proof: MmrMembershipProof,
        }

        // Split indexed MMR-mps into a hashmap with one entry for each
        // referenced peak in the MMR.
        let num_mmr_leafs = mmra.num_leafs();
        let mut peak_index_to_indexed_mmr_mp: HashMap<u32, Vec<IndexedAuthenticatedMmrLeaf>> =
            HashMap::default();
        let peak_heights = get_peak_heights(num_mmr_leafs);
        for (mmr_leaf_index, leaf, mmr_mp) in indexed_mmr_mps {
            let (mt_index, peak_index) =
                leaf_index_to_mt_index_and_peak_index(mmr_leaf_index, num_mmr_leafs);
            let peak_index_as_usize: usize = peak_index.try_into().unwrap();
            let num_leafs_local_mt = 1 << peak_heights[peak_index_as_usize];
            let mt_leaf_index = mt_index - num_leafs_local_mt;
            peak_index_to_indexed_mmr_mp
                .entry(peak_index)
                .or_default()
                .push(IndexedAuthenticatedMmrLeaf {
                    merkle_tree_node_index: mt_index,
                    merkle_tree_leaf_index: mt_leaf_index,
                    leaf_digest: leaf,
                    membership_proof: mmr_mp,
                });
        }

        // Loop over all peaks and collect an authentication witness struct
        // for each peak.
        let mut peak_index_to_authenticated_auth_struct = HashMap::default();
        for (peak_index, indexed_mmr_mp_structs) in peak_index_to_indexed_mmr_mp {
            let peak_index_as_usize: usize = peak_index.try_into().unwrap();
            let num_leafs_in_local_mt = 1 << peak_heights[peak_index_as_usize];
            let local_mt_leaf_indices = indexed_mmr_mp_structs
                .iter()
                .map(|x| x.merkle_tree_leaf_index)
                .collect_vec();

            let (nd_auth_struct_indices, nd_sibling_indices) =
                Self::auth_struct_and_nd_indices(num_leafs_in_local_mt, &local_mt_leaf_indices);
            let peak = mmra.peaks()[peak_index_as_usize];

            let mut node_digests: HashMap<u64, Digest> = HashMap::default();
            node_digests.insert(ROOT_MT_INDEX, peak);

            // Loop over all indexed leafs for this peak
            for indexed_mmr_mp in indexed_mmr_mp_structs.iter() {
                let mut mt_node_index = indexed_mmr_mp.merkle_tree_node_index;
                let mut node = indexed_mmr_mp.leaf_digest;

                // Loop over all authentication path elements for this indexed leaf
                for ap_elem in indexed_mmr_mp.membership_proof.authentication_path.iter() {
                    node_digests.insert(mt_node_index, node);
                    node_digests.insert(mt_node_index ^ 1, *ap_elem);
                    node = if mt_node_index & 1 == 0 {
                        Tip5::hash_pair(node, *ap_elem)
                    } else {
                        Tip5::hash_pair(*ap_elem, node)
                    };

                    mt_node_index /= 2;
                }

                // Sanity check that MMR-MPs are valid
                assert_eq!(peak, node, "Derived peak must match provided peak");
            }
            let nd_siblings = nd_sibling_indices
                .iter()
                .map(|(left_idx, right_idx)| (node_digests[left_idx], node_digests[right_idx]))
                .collect_vec();
            let auth_struct = nd_auth_struct_indices
                .iter()
                .map(|idx| node_digests[idx])
                .collect_vec();
            let indexed_leafs = indexed_mmr_mp_structs
                .into_iter()
                .map(|indexed_mmr_mp| {
                    (
                        indexed_mmr_mp.merkle_tree_leaf_index,
                        indexed_mmr_mp.leaf_digest,
                    )
                })
                .collect_vec();

            let witness = Self {
                nd_auth_struct_indices,
                nd_sibling_indices,
                nd_siblings,
            };

            peak_index_to_authenticated_auth_struct.insert(
                peak_index,
                AuthenticatedMerkleAuthStruct {
                    auth_struct,
                    indexed_leafs,
                    witness,
                },
            );
        }

        peak_index_to_authenticated_auth_struct
    }

    /// Return the authentication structure witness, authentication structure,
    /// and the (leaf-index, leaf-digest) pairs.
    pub fn new_from_merkle_tree(
        tree: &MerkleTree,
        mut revealed_leaf_indices: Vec<u64>,
    ) -> AuthenticatedMerkleAuthStruct {
        revealed_leaf_indices.sort_unstable();
        revealed_leaf_indices.dedup();
        revealed_leaf_indices.reverse();
        let num_leafs: u64 = tree.num_leafs() as u64;

        let (nd_auth_struct_indices, nd_sibling_indices) =
            Self::auth_struct_and_nd_indices(num_leafs, &revealed_leaf_indices);

        let nd_siblings = nd_sibling_indices
            .iter()
            .map(|&(l, r)| {
                let l: usize = l.try_into().unwrap();
                let r: usize = r.try_into().unwrap();
                (tree.node(l).unwrap(), tree.node(r).unwrap())
            })
            .collect_vec();

        let revealed_leafs = revealed_leaf_indices
            .iter()
            .map(|j| tree.node((*j + num_leafs) as usize).unwrap())
            .collect_vec();
        let indexed_leafs = revealed_leaf_indices
            .clone()
            .into_iter()
            .zip_eq(revealed_leafs)
            .collect_vec();

        let auth_struct = nd_auth_struct_indices
            .iter()
            .map(|node_index| tree.node(*node_index as usize).unwrap())
            .collect_vec();

        let witness = Self {
            nd_auth_struct_indices,
            nd_sibling_indices,
            nd_siblings,
        };

        AuthenticatedMerkleAuthStruct {
            auth_struct,
            indexed_leafs,
            witness,
        }
    }
}

#[cfg(test)]
mod tests {
    use proptest::collection::vec;
    use proptest::prop_assert_eq;
    use rand::random;
    use test_strategy::proptest;

    use super::*;
    use crate::twenty_first::math::other::random_elements;
    use crate::twenty_first::util_types::mmr::mmr_accumulator::util::mmra_with_mps;

    #[proptest(cases = 20)]
    fn root_from_authentication_struct_mmr_prop_test(
        #[strategy(0..u64::MAX / 2)] mmr_leaf_count: u64,
        #[strategy(0usize..20)] _num_revealed_leafs: usize,
        #[strategy(vec(0u64..#mmr_leaf_count, #_num_revealed_leafs))]
        mmr_revealed_leaf_indices: Vec<u64>,
    ) {
        let indexed_leafs_input: Vec<(u64, Digest)> = mmr_revealed_leaf_indices
            .iter()
            .map(|idx| (*idx, random()))
            .collect_vec();
        let (mmra, mmr_mps) = mmra_with_mps(mmr_leaf_count, indexed_leafs_input.clone());
        let indexed_mmr_mps = mmr_mps
            .into_iter()
            .zip_eq(indexed_leafs_input)
            .map(|(mmr_mp, (idx, leaf))| (idx, leaf, mmr_mp))
            .collect_vec();

        let authenticated_auth_structs =
            AuthStructIntegrityProof::new_from_mmr_membership_proofs(&mmra, indexed_mmr_mps);

        let peak_heights = get_peak_heights(mmr_leaf_count);
        for (peak_index, authentication_auth_struct) in authenticated_auth_structs {
            let AuthenticatedMerkleAuthStruct {
                auth_struct,
                indexed_leafs,
                witness,
            } = &authentication_auth_struct;
            let tree_height: u32 = peak_heights[peak_index as usize];
            let computed_root = witness.root_from_authentication_struct(
                tree_height,
                auth_struct.to_owned(),
                indexed_leafs.to_owned(),
            );

            prop_assert_eq!(mmra.peaks()[peak_index as usize], computed_root);
        }
    }

    #[test]
    fn auth_struct_on_empty_mmr() {
        let empty_mmra = MmrAccumulator::init(vec![], 0);
        let authenticated_auth_structs =
            AuthStructIntegrityProof::new_from_mmr_membership_proofs(&empty_mmra, vec![]);
        assert!(authenticated_auth_structs.is_empty());
    }

    #[test]
    fn auth_struct_non_empty_mmr_empty_leaf_list() {
        let mmra_10_leafs = MmrAccumulator::new_from_leafs(vec![Digest::default(); 10]);
        let authenticated_auth_structs =
            AuthStructIntegrityProof::new_from_mmr_membership_proofs(&mmra_10_leafs, vec![]);
        assert!(authenticated_auth_structs.is_empty());
    }

    #[should_panic]
    #[test]
    fn panics_on_missing_nd_digests() {
        let tree_height = 3u32;
        let num_leafs = 1u64 << tree_height;
        let leafs: Vec<Digest> = random_elements(num_leafs.try_into().unwrap());
        let tree = MerkleTree::par_new(&leafs).unwrap();

        let authenticated_auth_struct =
            AuthStructIntegrityProof::new_from_merkle_tree(&tree, vec![0]);
        let AuthenticatedMerkleAuthStruct {
            auth_struct,
            indexed_leafs,
            mut witness,
        } = authenticated_auth_struct;
        witness.nd_sibling_indices.clear();
        witness.nd_siblings.clear();

        witness.root_from_authentication_struct(tree_height, auth_struct, indexed_leafs);
    }

    #[test]
    fn auth_struct_from_mmr_mps_test_height_5_9_indices() {
        let local_tree_height = 5;
        let mmr_leaf_indices = [0, 1, 2, 16, 17, 18, 27, 29, 31];
        let indexed_leafs_input: Vec<(u64, Digest)> = mmr_leaf_indices
            .iter()
            .map(|idx| (*idx, random()))
            .collect_vec();
        let (mmra, mmr_mps) = mmra_with_mps(1 << local_tree_height, indexed_leafs_input.clone());
        let indexed_mmr_mps = mmr_mps
            .into_iter()
            .zip_eq(indexed_leafs_input)
            .map(|(mmr_mp, (idx, leaf))| (idx, leaf, mmr_mp))
            .collect_vec();

        let authenticity_witnesses =
            AuthStructIntegrityProof::new_from_mmr_membership_proofs(&mmra, indexed_mmr_mps);
        assert!(
            authenticity_witnesses.len().is_one(),
            "All indices belong to first peak"
        );
        let AuthenticatedMerkleAuthStruct {
            auth_struct,
            indexed_leafs,
            witness,
        } = &authenticity_witnesses[&0];

        let tree_height: u32 = local_tree_height.try_into().unwrap();
        let computed_root = witness.root_from_authentication_struct(
            tree_height,
            auth_struct.to_owned(),
            indexed_leafs.to_owned(),
        );

        let peak_index = 0;
        let expected_root = mmra.peaks()[peak_index];
        assert_eq!(expected_root, computed_root);
    }

    #[test]
    fn auth_struct_from_mmr_mps_test_height_4_2_indices() {
        let local_tree_height = 4;
        let mmr_leaf_indices = [0, 1];
        let indexed_leafs_input: Vec<(u64, Digest)> = mmr_leaf_indices
            .iter()
            .map(|idx| (*idx, random()))
            .collect_vec();
        let (mmra, mmr_mps) = mmra_with_mps(1 << local_tree_height, indexed_leafs_input.clone());
        let indexed_mmr_mps = mmr_mps
            .into_iter()
            .zip_eq(indexed_leafs_input)
            .map(|(mmr_mp, (idx, leaf))| (idx, leaf, mmr_mp))
            .collect_vec();

        let authenticity_witnesses =
            AuthStructIntegrityProof::new_from_mmr_membership_proofs(&mmra, indexed_mmr_mps);
        assert!(
            authenticity_witnesses.len().is_one(),
            "All indices belong to first peak"
        );
        let AuthenticatedMerkleAuthStruct {
            auth_struct,
            indexed_leafs,
            witness,
        } = &authenticity_witnesses[&0];

        let tree_height: u32 = local_tree_height.try_into().unwrap();
        let computed_root = witness.root_from_authentication_struct(
            tree_height,
            auth_struct.to_owned(),
            indexed_leafs.to_owned(),
        );

        let peak_index = 0;
        let expected_root = mmra.peaks()[peak_index];
        assert_eq!(expected_root, computed_root);
    }

    #[proptest(cases = 20)]
    fn root_from_authentication_struct_prop_test(
        #[strategy(0..12u64)] tree_height: u64,
        #[strategy(1usize..100)] _num_revealed_leafs: usize,
        #[strategy(vec(0u64..1<<#tree_height, #_num_revealed_leafs))] revealed_leaf_indices: Vec<
            u64,
        >,
    ) {
        let num_leafs = 1u64 << tree_height;
        let leafs: Vec<Digest> = random_elements(num_leafs.try_into().unwrap());
        let tree = MerkleTree::par_new(&leafs).unwrap();

        let authenticated_auth_struct =
            AuthStructIntegrityProof::new_from_merkle_tree(&tree, revealed_leaf_indices);
        let AuthenticatedMerkleAuthStruct {
            auth_struct,
            indexed_leafs,
            witness,
        } = authenticated_auth_struct;

        let tree_height: u32 = tree_height.try_into().unwrap();
        let computed_root =
            witness.root_from_authentication_struct(tree_height, auth_struct, indexed_leafs);
        let expected_root = tree.root();
        prop_assert_eq!(expected_root, computed_root);
    }

    fn prop_from_merkle_tree(
        tree_height: usize,
        leaf_indices: Vec<u64>,
        nd_auth_struct_indices: Vec<u64>,
        nd_sibling_indices: Vec<(u64, u64)>,
    ) {
        let leafs: Vec<Digest> = random_elements(1 << tree_height);
        let tree = MerkleTree::par_new(&leafs).unwrap();

        let auth_struct = nd_auth_struct_indices
            .iter()
            .map(|i| tree.node(*i as usize).unwrap())
            .collect_vec();
        let revealed_leafs = leaf_indices
            .iter()
            .map(|i| tree.leaf(*i as usize).unwrap())
            .collect_vec();
        let revealed_leafs = leaf_indices
            .into_iter()
            .zip_eq(revealed_leafs)
            .collect_vec();
        let nd_siblings = nd_sibling_indices
            .iter()
            .map(|(left_idx, right_idx)| {
                (
                    tree.node(*left_idx as usize).unwrap(),
                    tree.node(*right_idx as usize).unwrap(),
                )
            })
            .collect_vec();

        let mmr_auth_struct = AuthStructIntegrityProof {
            nd_auth_struct_indices,
            nd_sibling_indices,
            nd_siblings,
        };
        let tree_height: u32 = tree_height.try_into().unwrap();
        let calculated_root = mmr_auth_struct.root_from_authentication_struct(
            tree_height,
            auth_struct,
            revealed_leafs,
        );
        assert_eq!(tree.root(), calculated_root);
    }

    #[test]
    fn root_from_authentication_struct_tree_height_0_1_revealed() {
        let tree_height = 0;
        let leaf_indices = vec![0];
        let nd_auth_struct_indices = vec![];
        let nd_sibling_indices = vec![];
        prop_from_merkle_tree(
            tree_height,
            leaf_indices,
            nd_auth_struct_indices,
            nd_sibling_indices,
        )
    }

    #[test]
    fn root_from_authentication_struct_tree_height_1_1_revealed() {
        let tree_height = 1;
        let leaf_indices = vec![0u64];
        let nd_auth_struct_indices = vec![3];
        let nd_sibling_indices = vec![(2u64, 3u64)];
        prop_from_merkle_tree(
            tree_height,
            leaf_indices,
            nd_auth_struct_indices,
            nd_sibling_indices,
        )
    }

    #[test]
    fn root_from_authentication_struct_tree_height_1_2_revealed() {
        let tree_height = 1;
        let leaf_indices = vec![0u64, 1];
        let nd_auth_struct_indices = vec![];
        let nd_sibling_indices = vec![(2u64, 3u64)];
        prop_from_merkle_tree(
            tree_height,
            leaf_indices,
            nd_auth_struct_indices,
            nd_sibling_indices,
        )
    }

    #[test]
    fn root_from_authentication_struct_tree_height_2_2_revealed() {
        let tree_height = 2;
        let leaf_indices = vec![0u64, 1];
        let auth_struct_indices = vec![3];
        let nd_sibling_indices = vec![(4u64, 5u64), (2, 3)];
        prop_from_merkle_tree(
            tree_height,
            leaf_indices,
            auth_struct_indices,
            nd_sibling_indices,
        )
    }

    #[test]
    fn root_from_authentication_struct_tree_height_4_4_revealed() {
        let tree_height = 4;
        let leaf_indices = vec![14u64, 12, 10, 8];
        let num_leafs = 1 << tree_height;
        let auth_struct_indices = vec![
            num_leafs + 15,
            num_leafs + 13,
            num_leafs + 11,
            num_leafs + 9,
            2,
        ];
        let nd_sibling_indices_layer_height_0 = [(14u64, 15u64), (12, 13), (10, 11), (8, 9)]
            .map(|(l, r)| (l + num_leafs, r + num_leafs));
        let nd_sibling_indices_layer_height_1 = [(14u64, 15u64), (12u64, 13u64)];
        let nd_sibling_indices_layer_height_2 = [(6u64, 7u64)];
        let nd_sibling_indices_layer_height_3 = [(2u64, 3u64)];
        let nd_sibling_indices = [
            nd_sibling_indices_layer_height_0.to_vec(),
            nd_sibling_indices_layer_height_1.to_vec(),
            nd_sibling_indices_layer_height_2.to_vec(),
            nd_sibling_indices_layer_height_3.to_vec(),
        ]
        .concat();

        prop_from_merkle_tree(
            tree_height,
            leaf_indices,
            auth_struct_indices,
            nd_sibling_indices,
        )
    }
}