Add functions to get measurements out of the log (#228)

oxidecomputer · Feb 28, 2025 · f1ccb0e · f1ccb0e
1 parent a207669
commit f1ccb0e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/attest-data/src/lib.rs b/attest-data/src/lib.rs
@@ -156,6 +156,14 @@ impl<const N: usize> MeasurementLog<N> {
         self.index as usize == N
     }
 
+    pub fn is_empty(&self) -> bool {
+        self.index == 0
+    }
+
+    pub fn len(&self) -> u32 {
+        self.index
+    }
+
     pub fn push(&mut self, measurement: Measurement) -> bool {
         if !self.is_full() {
             self.measurements[self.index as usize] = measurement;
@@ -167,6 +175,13 @@ impl<const N: usize> MeasurementLog<N> {
     }
 }
 
+impl<const N: usize> core::ops::Index<u32> for MeasurementLog<N> {
+    type Output = Measurement;
+    fn index(&self, i: u32) -> &Self::Output {
+        &self.measurements[i as usize]
+    }
+}
+
 impl<const N: usize> Default for MeasurementLog<N> {
     fn default() -> Self {
         Self {