Use size_of instead of hard coded numbers.

1 files changed, 3 insertions, 3 deletions

diff --git a/src/vrrpv2.rs b/src/vrrpv2.rs
index 63c58c8..7e4a6cd 100644
--- a/src/vrrpv2.rs
+++ b/src/vrrpv2.rs
@@ -115,19 +115,19 @@ impl VRRPv2 {
 }
 
 fn read_u8(cursor: &mut Cursor<&[u8]>) -> io::Result<u8> {
-    let mut buffer = [0; 1];
+    let mut buffer = [0; size_of::<u8>()];
     cursor.read_exact(&mut buffer)?;
     Ok(u8::from_be_bytes(buffer))
 }
 
 fn read_u16(cursor: &mut Cursor<&[u8]>) -> io::Result<u16> {
-    let mut buffer = [0; 2];
+    let mut buffer = [0; size_of::<u16>()];
     cursor.read_exact(&mut buffer)?;
     Ok(u16::from_be_bytes(buffer))
 }
 
 fn read_u32(cursor: &mut Cursor<&[u8]>) -> io::Result<u32> {
-    let mut buffer = [0; 4];
+    let mut buffer = [0; size_of::<u32>()];
     cursor.read_exact(&mut buffer)?;
     Ok(u32::from_be_bytes(buffer))
 }