luct-core 0.2.0

Core types and parsers for certificate transparency
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
116
117
118

use crate::{store::Hashable, tree::HashOutput};
use serde::{Deserialize, Serialize};
use sha2::{Digest, Sha256};
use std::cmp::Ordering;

#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Hash)]

pub struct NodeKey {
    pub(crate) start: u64,
    pub(crate) end: u64,
}

impl NodeKey {
    pub fn leaf(idx: u64) -> Self {
        Self {
            start: idx,
            end: idx + 1,
        }
    }

    pub(crate) fn full_range(end: u64) -> Self {
        Self { start: 0, end }
    }

    pub(crate) fn size(&self) -> u64 {
        self.end - self.start
    }

    pub(crate) fn split_idx(&self) -> u64 {
        let size = self.size();
        size.next_power_of_two() >> 1
    }

    pub fn split(&self) -> (Self, Self) {
        let split = self.split_idx();
        let split = self.start + split;
        (
            Self {
                start: self.start,
                end: split,
            },
            Self {
                start: split,
                end: self.end,
            },
        )
    }

    pub(crate) fn merge(&self, other: &Self) -> Option<Self> {
        if self.end == other.start {
            Some(Self {
                start: self.start,
                end: other.end,
            })
        } else {
            None
        }
    }

    pub fn is_balanced(&self) -> bool {
        let diff = self.end - self.start;
        diff.is_power_of_two()
    }

    pub(crate) fn is_leaf(&self) -> bool {
        self.start + 1 == self.end
    }
}

impl PartialOrd for NodeKey {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

impl Ord for NodeKey {
    fn cmp(&self, other: &Self) -> Ordering {
        match self.end.cmp(&other.end) {
            Ordering::Equal => {}
            ord => return ord,
        }
        self.start.cmp(&other.start)
    }
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Node {
    pub left: HashOutput,
    pub right: HashOutput,
}

impl Hashable for Node {
    fn hash(&self) -> HashOutput {
        let mut hash = Sha256::new();
        hash.update([1]);
        hash.update(self.left);
        hash.update(self.right);
        hash.finalize().into()
    }
}

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

    #[test]
    fn node_key_split() {
        let node_key = NodeKey { start: 0, end: 6 };
        let (left, right) = node_key.split();
        assert_eq!(left, NodeKey { start: 0, end: 4 });
        assert_eq!(right, NodeKey { start: 4, end: 6 });

        let node_key = NodeKey { start: 0, end: 8 };
        let (left, right) = node_key.split();
        assert_eq!(left, NodeKey { start: 0, end: 4 });
        assert_eq!(right, NodeKey { start: 4, end: 8 });
    }
}