author | Sunil Nimmagadda <sunil@nimmagadda.net> |
Tue, 30 Jan 2024 16:25:28 +0530 | |
changeset 23 | 1bfa6fd5b5f9 |
parent 22 | 809a5a9e5dd9 |
child 25 | 72c4d898c478 |
permissions | -rw-r--r-- |
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
1 |
use std::io::Read; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
2 |
use std::io::{self, Cursor}; |
7
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
3 |
use std::net::Ipv4Addr; |
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
4 |
|
16 | 5 |
/// A VRRP version 2 packet. |
6 |
/// |
|
7 |
/// Packet format |
|
8 |
/// |
|
9 |
/// 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 |
|
10 |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
11 |
/// |Version| Type | Virtual Rtr ID| Priority | Count IP Addrs| |
|
12 |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
13 |
/// | Auth Type | Adver Int | Checksum | |
|
14 |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
15 |
/// | IP Address (1) | |
|
16 |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
17 |
/// | . | |
|
18 |
/// | . | |
|
19 |
/// | . | |
|
20 |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
21 |
/// | IP Address (n) | |
|
22 |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
23 |
/// | Authentication Data (1) | |
|
24 |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
25 |
/// | Authentication Data (2) | |
|
26 |
/// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
|
27 |
#[derive(Debug, PartialEq)] |
|
28 |
pub struct VRRPv2 { |
|
29 |
pub virtual_router_id: u8, |
|
30 |
pub priority: u8, |
|
31 |
pub count_ip_addrs: u8, |
|
32 |
pub auth_type: VRRPv2AuthType, |
|
33 |
pub advertisement_interval: u8, |
|
34 |
pub checksum: u16, |
|
35 |
pub ip_addrs: Vec<Ipv4Addr>, |
|
36 |
} |
|
7
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
37 |
|
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
38 |
#[derive(Debug, Clone, PartialEq)] |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
39 |
pub enum VRRPv2Error { |
16 | 40 |
InvalidAuthType, |
41 |
InvalidChecksum, |
|
42 |
InvalidType, |
|
43 |
InvalidVersion, |
|
44 |
ParseError, |
|
7
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
45 |
} |
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
46 |
|
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
47 |
#[derive(Debug, PartialEq)] |
7
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
48 |
pub enum VRRPv2AuthType { |
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
49 |
VRRPv2AuthNoAuth = 0x00, |
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
50 |
VRRPv2AuthReserved1 = 0x01, |
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
51 |
VRRPv2AuthReserved2 = 0x02, |
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
52 |
} |
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
53 |
|
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
54 |
trait ByteReader { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
55 |
fn read_u8(&mut self) -> io::Result<u8>; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
56 |
fn read_u16(&mut self) -> io::Result<u16>; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
57 |
fn read_u32(&mut self) -> io::Result<u32>; |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
58 |
} |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
59 |
|
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
60 |
impl<T: AsRef<[u8]>> ByteReader for Cursor<T> { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
61 |
fn read_u8(&mut self) -> io::Result<u8> { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
62 |
let mut buffer = [0; 1]; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
63 |
self.read_exact(&mut buffer)?; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
64 |
Ok(u8::from_be_bytes(buffer)) |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
65 |
} |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
66 |
|
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
67 |
fn read_u16(&mut self) -> io::Result<u16> { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
68 |
let mut buffer = [0; 2]; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
69 |
self.read_exact(&mut buffer)?; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
70 |
Ok(u16::from_be_bytes(buffer)) |
16 | 71 |
} |
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
72 |
|
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
73 |
fn read_u32(&mut self) -> io::Result<u32> { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
74 |
let mut buffer = [0; 4]; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
75 |
self.read_exact(&mut buffer)?; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
76 |
Ok(u32::from_be_bytes(buffer)) |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
77 |
} |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
78 |
} |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
79 |
|
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
80 |
fn parse(bytes: &[u8]) -> Result<VRRPv2, VRRPv2Error> { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
81 |
let mut rdr = Cursor::new(bytes); |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
82 |
let Ok(vertype) = rdr.read_u8() else { |
16 | 83 |
return Err(VRRPv2Error::ParseError); |
84 |
}; |
|
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
85 |
if (vertype & 0xF) != 1 { |
21
17afb5ba9c2b
Still confused by lsb/msb.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
20
diff
changeset
|
86 |
return Err(VRRPv2Error::InvalidType); |
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
87 |
} |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
88 |
if (vertype >> 4) != 2 { |
21
17afb5ba9c2b
Still confused by lsb/msb.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
20
diff
changeset
|
89 |
return Err(VRRPv2Error::InvalidVersion); |
16 | 90 |
} |
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
91 |
let Ok(virtual_router_id) = rdr.read_u8() else { |
16 | 92 |
return Err(VRRPv2Error::ParseError); |
93 |
}; |
|
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
94 |
let Ok(priority) = rdr.read_u8() else { |
16 | 95 |
return Err(VRRPv2Error::ParseError); |
96 |
}; |
|
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
97 |
let Ok(count_ip_addrs) = rdr.read_u8() else { |
16 | 98 |
return Err(VRRPv2Error::ParseError); |
99 |
}; |
|
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
100 |
let Ok(auth_type) = rdr.read_u8() else { |
16 | 101 |
return Err(VRRPv2Error::ParseError); |
102 |
}; |
|
103 |
let auth_type = match auth_type { |
|
104 |
0 => VRRPv2AuthType::VRRPv2AuthNoAuth, |
|
105 |
1 => VRRPv2AuthType::VRRPv2AuthReserved1, |
|
106 |
2 => VRRPv2AuthType::VRRPv2AuthReserved2, |
|
107 |
_ => return Err(VRRPv2Error::InvalidAuthType), |
|
108 |
}; |
|
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
109 |
let Ok(advertisement_interval) = rdr.read_u8() else { |
16 | 110 |
return Err(VRRPv2Error::ParseError); |
111 |
}; |
|
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
112 |
let Ok(checksum) = rdr.read_u16() else { |
16 | 113 |
return Err(VRRPv2Error::ParseError); |
114 |
}; |
|
23
1bfa6fd5b5f9
Use with_capacity as it's known by now.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
22
diff
changeset
|
115 |
let mut ip_addrs = Vec::with_capacity(count_ip_addrs as usize); |
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
116 |
for _i in 0..count_ip_addrs { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
117 |
let Ok(b) = rdr.read_u32() else { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
118 |
return Err(VRRPv2Error::ParseError); |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
119 |
}; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
120 |
ip_addrs.push(Ipv4Addr::from(b)); |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
121 |
} |
16 | 122 |
Ok(VRRPv2 { |
123 |
virtual_router_id, |
|
124 |
priority, |
|
125 |
count_ip_addrs, |
|
126 |
auth_type, |
|
127 |
advertisement_interval, |
|
128 |
checksum, |
|
129 |
ip_addrs, |
|
130 |
}) |
|
7
e3058ceb81cc
Module to parse(unimplemented) VRRPv2 packet.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff
changeset
|
131 |
} |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
132 |
|
19
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
133 |
/// Parse and validate a byte array to construct a VRRPv2 struct. |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
134 |
/// |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
135 |
/// # Examples |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
136 |
/// |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
137 |
/// ``` |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
138 |
/// use vrrpd::vrrpv2::VRRPv2; |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
139 |
/// use vrrpd::vrrpv2::VRRPv2AuthType; |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
140 |
/// use vrrpd::vrrpv2::from_bytes; |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
141 |
/// use std::net::Ipv4Addr; |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
142 |
/// |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
143 |
/// let bytes = [ |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
144 |
/// 0x21, 0x01, 0x64, 0x01, 0x00, 0x01, 0xba, 0x52, 0xc0, 0xa8, 0x00, 0x01, 0x00, 0x00, 0x00, |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
145 |
/// 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
146 |
/// ]; |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
147 |
/// let expected = VRRPv2 { |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
148 |
/// virtual_router_id: 1, |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
149 |
/// priority: 100, |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
150 |
/// count_ip_addrs: 1, |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
151 |
/// auth_type: VRRPv2AuthType::VRRPv2AuthNoAuth, |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
152 |
/// checksum: 47698, |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
153 |
/// advertisement_interval: 1, |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
154 |
/// ip_addrs: vec![Ipv4Addr::from([192, 168, 0, 1])], |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
155 |
/// }; |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
156 |
/// assert_eq!(from_bytes(&bytes), Ok(expected)); |
1567e8a5a0cb
Recover a lost doc test.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
18
diff
changeset
|
157 |
/// ``` |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
158 |
pub fn from_bytes(bytes: &[u8]) -> Result<VRRPv2, VRRPv2Error> { |
16 | 159 |
let vrrpv2 = parse(bytes)?; |
17
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
160 |
if checksum(bytes) != 0 { |
16 | 161 |
return Err(VRRPv2Error::InvalidChecksum); |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
162 |
} |
16 | 163 |
Ok(vrrpv2) |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
164 |
} |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
165 |
|
22
809a5a9e5dd9
Note about as_chunks API.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
21
diff
changeset
|
166 |
/// TODO: as_chunks is nicer but not stabilised yet. |
20
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
167 |
fn checksum(bytes: &[u8]) -> u16 { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
168 |
let mut sum: u32 = 0; |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
169 |
for chunk in bytes.chunks(2) { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
170 |
// Left over byte if any |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
171 |
if chunk.len() == 1 { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
172 |
sum += u32::from(chunk[0]); |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
173 |
} else { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
174 |
sum += u32::from(u16::from_be_bytes(chunk.try_into().unwrap())); |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
175 |
} |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
176 |
} |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
177 |
while (sum >> 16) > 0 { |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
178 |
sum = (sum & 0xffff) + (sum >> 16); |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
179 |
} |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
180 |
!(sum as u16) |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
181 |
} |
fd3474f42783
Replace nom with std::io::Cursor
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
19
diff
changeset
|
182 |
|
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
183 |
#[test] |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
184 |
fn test_incomplete_bytes() { |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
185 |
let bytes = [0x21, 0x01]; |
16 | 186 |
assert_eq!(from_bytes(&bytes), Err(VRRPv2Error::ParseError)); |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
187 |
} |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
188 |
|
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
189 |
#[test] |
16 | 190 |
fn test_invalid_version() { |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
191 |
let bytes = [ |
21
17afb5ba9c2b
Still confused by lsb/msb.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
20
diff
changeset
|
192 |
0x31, 0x1, 0x2a, 0x0, 0x0, 0x1, 0xb5, 0xfd, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, |
16 | 193 |
0x0, 0x0, 0x0, |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
194 |
]; |
18
2470a15711b1
Fix clippy warnings.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
17
diff
changeset
|
195 |
assert_eq!(from_bytes(&bytes), Err(VRRPv2Error::InvalidVersion)); |
16 | 196 |
} |
197 |
||
198 |
#[test] |
|
199 |
fn test_invalid_type() { |
|
200 |
let bytes = [ |
|
21
17afb5ba9c2b
Still confused by lsb/msb.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
20
diff
changeset
|
201 |
0x20, 0x2a, 0x64, 0x1, 0x0, 0x1, 0xaa, 0x29, 0xc0, 0xa8, 0x0, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, |
16 | 202 |
0x0, 0x0, 0x0, |
203 |
]; |
|
18
2470a15711b1
Fix clippy warnings.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
17
diff
changeset
|
204 |
assert_eq!(from_bytes(&bytes), Err(VRRPv2Error::InvalidType)); |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
205 |
} |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
206 |
|
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
207 |
#[test] |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
208 |
fn test_invalid_auth_type() { |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
209 |
let bytes = [ |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
210 |
0x21, 0x01, 0x64, 0x01, 0x03, 0x01, 0xba, 0x52, 0xc0, 0xa8, 0x00, 0x01, 0x00, 0x00, 0x00, |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
211 |
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
212 |
]; |
16 | 213 |
assert_eq!(from_bytes(&bytes), Err(VRRPv2Error::InvalidAuthType)); |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
214 |
} |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
215 |
|
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
216 |
#[test] |
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
217 |
fn test_invalid_checksum() { |
10 | 218 |
let bytes = [ |
219 |
0x21, 0x01, 0x64, 0x01, 0x00, 0x01, 0xbb, 0x52, 0xc0, 0xa8, 0x00, 0x01, 0x00, 0x00, 0x00, |
|
220 |
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, |
|
221 |
]; |
|
16 | 222 |
assert_eq!(from_bytes(&bytes), Err(VRRPv2Error::InvalidChecksum)); |
8
e4cad4eba60b
Implement a VRRPv2 packet parser.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
7
diff
changeset
|
223 |
} |
17
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
224 |
|
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
225 |
#[test] |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
226 |
fn test_checksum() { |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
227 |
let bytes = [0x00, 0x01, 0xf2, 0x03, 0xf4, 0xf5, 0xf6, 0xf7]; |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
228 |
assert_eq!(checksum(&bytes), 0x220d); |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
229 |
} |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
230 |
|
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
231 |
#[test] |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
232 |
fn test_checksum_singlebyte() { |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
233 |
let bytes = [0; 1]; |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
234 |
assert_eq!(checksum(&bytes), 0xffff); |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
235 |
} |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
236 |
|
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
237 |
#[test] |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
238 |
fn test_checksum_twobytes() { |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
239 |
let bytes = [0x00, 0xff]; |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
240 |
assert_eq!(checksum(&bytes), 0xff00); |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
241 |
} |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
242 |
|
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
243 |
#[test] |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
244 |
fn test_checksum_another() { |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
245 |
let bytes = [0xe3, 0x4f, 0x23, 0x96, 0x44, 0x27, 0x99, 0xf3]; |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
246 |
assert_eq!(checksum(&bytes), 0x1aff); |
90d097c9ea62
Rectify checksum calculation.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
16
diff
changeset
|
247 |
} |