use assert! to verify length number

snapview · Jan 8, 2025 · 6e61fc9 · 6e61fc9
1 parent cce13fa
commit 6e61fc9
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/protocol/frame/frame.rs b/src/protocol/frame/frame.rs
@@ -161,13 +161,9 @@ impl FrameHeader {
             let length_length = LengthFormat::for_byte(length_byte).extra_bytes();
             if length_length > 0 {
                 const SIZE: usize = mem::size_of::<u64>();
+                assert!(length_length <= SIZE, "length exceeded size of u64");
+                let start = SIZE - length_length;
                 let mut buffer = [0; SIZE];
-                let start = match SIZE.checked_sub(length_length) {
-                    Some(start) => start,
-                    None => {
-                        panic!("the integer can fit {} bytes, but {} is given", SIZE, length_length)
-                    }
-                };
                 match cursor.read_exact(&mut buffer[start..]) {
                     Err(ref err) if err.kind() == ErrorKind::UnexpectedEof => return Ok(None),
                     Err(err) => return Err(err.into()),