earthbucks_lib 0.8.5

EarthBucks library for data structures and algorithms
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

use crate::merkle_proof::MerkleProof;
use crate::tx::Tx;

pub struct MerkleTxs {
    pub txs: Vec<Tx>,
    pub root: [u8; 32],
    pub proofs: Vec<MerkleProof>,
}

impl MerkleTxs {
    pub fn new(txs: Vec<Tx>) -> Self {
        let hashed_datas: Vec<[u8; 32]> = txs.iter().map(|tx| tx.id()).collect::<Vec<_>>();
        let (root, proofs) = MerkleProof::generate_proofs_and_root(hashed_datas);
        Self { txs, root, proofs }
    }

    pub fn get_iterator(&self) -> impl Iterator<Item = (&Tx, &MerkleProof)> {
        self.txs.iter().zip(self.proofs.iter())
    }

    pub fn verify(&self) -> bool {
        for i in 0..self.txs.len() {
            let tx = &self.txs[i];
            let proof = &self.proofs[i];
            if !MerkleProof::verify_proof(&tx.id(), proof, &self.root) {
                return false;
            }
        }
        true
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn verify_with_1_tx() {
        let tx1 = Tx::new(0, vec![], vec![], 0);
        let merkle_txs = MerkleTxs::new(vec![tx1]);
        let verified = merkle_txs.verify();
        assert!(verified);
    }

    #[test]
    fn verify_with_2_txs() {
        let tx1 = Tx::new(0, vec![], vec![], 0);
        let tx2 = Tx::new(0, vec![], vec![], 0);
        let merkle_txs = MerkleTxs::new(vec![tx1, tx2]);
        let verified = merkle_txs.verify();
        assert!(verified);
    }

    #[test]
    fn verify_with_3_txs() {
        let tx1 = Tx::new(0, vec![], vec![], 0);
        let tx2 = Tx::new(0, vec![], vec![], 0);
        let tx3 = Tx::new(0, vec![], vec![], 0);
        let merkle_txs = MerkleTxs::new(vec![tx1, tx2, tx3]);
        let verified = merkle_txs.verify();
        assert!(verified);
    }
}