brine-tree 0.6.2

Merkle tree implementation for Solana programs
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use bytemuck::{Pod, Zeroable};

pub const HASH_BYTES: usize = 32;

#[repr(C)]
#[derive(Clone, Copy, PartialEq, Debug, Default, Pod, Zeroable)]
pub struct Hash {
    pub(crate) value: [u8; 32],
}

#[repr(C)]
#[derive(Clone, Copy, PartialEq, Debug, Pod, Zeroable)]
pub struct Leaf(Hash);

impl From<Hash> for [u8; HASH_BYTES] {
    fn from(from: Hash) -> Self {
        from.value
    }
}

impl From<[u8; HASH_BYTES]> for Hash {
    fn from(from: [u8; 32]) -> Self {
        Self { value: from }
    }
}

impl From<[u8; HASH_BYTES]> for Leaf {
    fn from(from: [u8; 32]) -> Self {
        Self(Hash { value: from })
    }
}

impl AsRef<[u8]> for Hash {
    fn as_ref(&self) -> &[u8] {
        &self.value
    }
}

impl AsRef<[u8]> for Leaf {
    fn as_ref(&self) -> &[u8] {
        &self.0.value
    }
}

impl From<Leaf> for Hash {
    fn from(leaf: Leaf) -> Self {
        leaf.0
    }
}

impl Hash {
    pub const LEN: usize = HASH_BYTES;

    pub fn new(hash_slice: &[u8]) -> Self {
        Hash {
            value: <[u8; HASH_BYTES]>::try_from(hash_slice).unwrap(),
        }
    }

    pub const fn new_from_array(hash_array: [u8; HASH_BYTES]) -> Self {
        Self { value: hash_array }
    }

    pub fn to_bytes(self) -> [u8; HASH_BYTES] {
        self.value
    }

    pub fn as_leaf(self) -> Leaf {
        Leaf(self)
    }
}

impl Leaf {
    pub fn new(data: &[&[u8]]) -> Self {
        let mut inputs = vec![b"LEAF".as_ref()];
        inputs.extend(data);
        Leaf(hashv(&inputs))
    }

    pub fn to_bytes(self) -> [u8; HASH_BYTES] {
        self.0.value
    }
}

#[cfg(feature = "solana")]
#[inline(always)]
pub fn hashv(data: &[&[u8]]) -> Hash {
    let res = solana_program::blake3::hashv(data);
    Hash::new_from_array(res.to_bytes())
}

#[cfg(not(feature = "solana"))]
#[inline(always)]
pub fn hashv(data: &[&[u8]]) -> Hash {
    let mut hasher = blake3::Hasher::new();
    for d in data {
        hasher.update(d);
    }
    Hash::new_from_array(hasher.finalize().into())
}

#[cfg(feature = "solana")]
#[inline(always)]
pub fn hash(data: &[u8]) -> Hash {
    let res = solana_program::blake3::hash(data);
    Hash::new_from_array(res.to_bytes())
}

#[cfg(not(feature = "solana"))]
#[inline(always)]
pub fn hash(data: &[u8]) -> Hash {
    let mut hasher = blake3::Hasher::new();
    hasher.update(data);
    Hash::new_from_array(hasher.finalize().into())
}