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
119
120
121
122
123
124
125
126
127

//! Supporting types for checkpoint-based block verification

use std::cmp::Ordering;

use zebra_chain::block;

use Progress::*;
use TargetHeight::*;

/// A `CheckpointVerifier`'s current progress verifying the chain.
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum Progress<HeightOrHash> {
    /// We have not verified any blocks yet.
    BeforeGenesis,

    /// We have verified up to and including this initial tip.
    ///
    /// Initial tips might not be one of our hard-coded checkpoints, because we
    /// might have:
    ///   - changed the checkpoint spacing, or
    ///   - added new checkpoints above the initial tip.
    InitialTip(HeightOrHash),

    /// We have verified up to and including this checkpoint.
    PreviousCheckpoint(HeightOrHash),

    /// We have finished verifying.
    ///
    /// The final checkpoint is not included in this variant. The verifier has
    /// finished, so the checkpoints aren't particularly useful.
    /// To get the value of the final checkpoint, use `checkpoint_list.max_height()`.
    FinalCheckpoint,
}

/// Block height progress, in chain order.
impl Ord for Progress<block::Height> {
    fn cmp(&self, other: &Self) -> Ordering {
        if self == other {
            return Ordering::Equal;
        }
        match (self, other) {
            (BeforeGenesis, _) => Ordering::Less,
            (_, BeforeGenesis) => Ordering::Greater,
            (FinalCheckpoint, _) => Ordering::Greater,
            (_, FinalCheckpoint) => Ordering::Less,
            (InitialTip(self_height), InitialTip(other_height))
            | (InitialTip(self_height), PreviousCheckpoint(other_height))
            | (PreviousCheckpoint(self_height), InitialTip(other_height))
            | (PreviousCheckpoint(self_height), PreviousCheckpoint(other_height)) => {
                self_height.cmp(other_height)
            }
        }
    }
}

/// Partial order for block height progress.
///
/// The partial order must match the total order.
impl PartialOrd for Progress<block::Height> {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

impl Progress<block::Height> {
    /// Returns the contained height, or `None` if the progress has finished, or has not started.
    pub fn height(&self) -> Option<block::Height> {
        match self {
            BeforeGenesis => None,
            InitialTip(height) => Some(*height),
            PreviousCheckpoint(height) => Some(*height),
            FinalCheckpoint => None,
        }
    }
}

impl<HeightOrHash> Progress<HeightOrHash> {
    /// Returns `true` if the progress is before the genesis block.
    #[allow(dead_code)]
    pub fn is_before_genesis(&self) -> bool {
        matches!(self, BeforeGenesis)
    }

    /// Returns `true` if the progress is at or after the final checkpoint block.
    pub fn is_final_checkpoint(&self) -> bool {
        matches!(self, FinalCheckpoint)
    }
}

/// A `CheckpointVerifier`'s target checkpoint height, based on the current
/// queue.
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum TargetHeight {
    /// We need more blocks before we can choose a target checkpoint.
    WaitingForBlocks,
    /// We want to verify this checkpoint.
    ///
    /// The target checkpoint can be multiple checkpoints ahead of the previous
    /// checkpoint.
    Checkpoint(block::Height),
    /// We have finished verifying, there will be no more targets.
    FinishedVerifying,
}

/// Block height target, in chain order.
///
/// `WaitingForBlocks` is incomparable with itself and `Checkpoint(_)`.
impl PartialOrd for TargetHeight {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        match (self, other) {
            // FinishedVerifying is the final state
            (FinishedVerifying, FinishedVerifying) => Some(Ordering::Equal),
            (FinishedVerifying, _) => Some(Ordering::Greater),
            (_, FinishedVerifying) => Some(Ordering::Less),
            // Checkpoints are comparable with each other by height
            (Checkpoint(self_height), Checkpoint(other_height)) => {
                self_height.partial_cmp(other_height)
            }
            // We can wait for blocks before or after any target checkpoint,
            // so there is no ordering between checkpoint and waiting.
            (WaitingForBlocks, Checkpoint(_)) => None,
            (Checkpoint(_), WaitingForBlocks) => None,
            // However, we consider waiting equal to itself.
            (WaitingForBlocks, WaitingForBlocks) => Some(Ordering::Equal),
        }
    }
}