feat: avoid error out on frame with invalid manufacturer code

maebli · Nov 7, 2024 · fe3a15a · fe3a15a
1 parent 1c73aff
commit fe3a15a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 15 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -195,7 +195,10 @@ fn parse_to_table(input: &str) -> std::string::String {
                     }) => {
                         table.add_row(row![
                             fixed_data_header.identification_number,
-                            fixed_data_header.manufacturer,
+                            fixed_data_header
+                                .manufacturer
+                                .map(|i| i.to_string())
+                                .unwrap_or_else(|_| "invalid".to_string()),
                             fixed_data_header.access_number,
                             fixed_data_header.status,
                             fixed_data_header.signature,

diff --git a/src/user_data/mod.rs b/src/user_data/mod.rs
@@ -507,7 +507,7 @@ impl fmt::Display for Medium {
 #[derive(Debug, PartialEq)]
 pub struct FixedDataHeader {
     pub identification_number: IdentificationNumber,
-    pub manufacturer: ManufacturerCode,
+    pub manufacturer: Result<ManufacturerCode, ApplicationLayerError>,
     pub version: u8,
     pub medium: Medium,
     pub access_number: u8,
@@ -613,7 +613,7 @@ impl<'a> TryFrom<&'a [u8]> for UserDataBlock<'a> {
                         manufacturer: ManufacturerCode::from_id(u16::from_le_bytes([
                             *iter.next().ok_or(ApplicationLayerError::InsufficientData)?,
                             *iter.next().ok_or(ApplicationLayerError::InsufficientData)?,
-                        ]))?,
+                        ])),
                         version: *iter.next().ok_or(ApplicationLayerError::InsufficientData)?,
                         medium: MeasuredMedium::new(
                             *iter.next().ok_or(ApplicationLayerError::InsufficientData)?,

diff --git a/tests/test.rs b/tests/test.rs
@@ -138,18 +138,10 @@ mod tests {
                         .manufacturer
                         .unwrap()
                         .into_bytes();
-                    assert_eq!(
-                        fixed_data_header.manufacturer.code[0],
-                        expected_manufacturer[0] as char
-                    );
-                    assert_eq!(
-                        fixed_data_header.manufacturer.code[1],
-                        expected_manufacturer[1] as char
-                    );
-                    assert_eq!(
-                        fixed_data_header.manufacturer.code[2],
-                        expected_manufacturer[2] as char
-                    );
+                    let manufacturer = fixed_data_header.manufacturer.unwrap();
+                    assert_eq!(manufacturer.code[0], expected_manufacturer[0] as char);
+                    assert_eq!(manufacturer.code[1], expected_manufacturer[1] as char);
+                    assert_eq!(manufacturer.code[2], expected_manufacturer[2] as char);
                     assert_eq!(
                         fixed_data_header.access_number,
                         mbus_data.slave_information.access_number as u8