GT3X: Harden against unexpected EOF

childmindresearch · Mar 30, 2024 · 213de2a · 213de2a
1 parent 29f5e50
commit 213de2a
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/actigraph/mod.rs b/src/actigraph/mod.rs
@@ -111,9 +111,14 @@ impl<R: Read> LogRecordIterator<R> {
 
                 let mut data = &mut data[0..record_header.record_size as usize + 1];
 
-                self.buffer.read_exact(&mut data).unwrap();
-
-                Some((record_header, data))
+                match self.buffer.read_exact(&mut data) {
+                    Ok(_) => Some((record_header, data)),
+                    Err(_) => {
+                        // Todo: Raise proper Python warning so users can catch this
+                        println!("Warning: Unexpected end of file");
+                        None
+                    },
+                }
             }
             Err(_) => None,
         }