1use core::num::NonZeroU8;
8use core::time::Duration;
9
10use net_types::ip::{Ipv6, Ipv6Addr};
11use zerocopy::byteorder::network_endian::{U16, U32};
12use zerocopy::{FromBytes, Immutable, IntoBytes, KnownLayout, SplitByteSlice, Unaligned};
13
14use crate::icmp::{IcmpIpExt, IcmpPacket, IcmpPacketRaw, IcmpZeroCode};
15use crate::utils::NonZeroDuration;
16
17#[allow(missing_docs)]
19#[derive(Debug)]
20pub enum NdpPacket<B: SplitByteSlice> {
21 RouterSolicitation(IcmpPacket<Ipv6, B, RouterSolicitation>),
22 RouterAdvertisement(IcmpPacket<Ipv6, B, RouterAdvertisement>),
23 NeighborSolicitation(IcmpPacket<Ipv6, B, NeighborSolicitation>),
24 NeighborAdvertisement(IcmpPacket<Ipv6, B, NeighborAdvertisement>),
25 Redirect(IcmpPacket<Ipv6, B, Redirect>),
26}
27
28#[allow(missing_docs)]
30#[derive(Debug)]
31pub enum NdpPacketRaw<B: SplitByteSlice> {
32 RouterSolicitation(IcmpPacketRaw<Ipv6, B, RouterSolicitation>),
33 RouterAdvertisement(IcmpPacketRaw<Ipv6, B, RouterAdvertisement>),
34 NeighborSolicitation(IcmpPacketRaw<Ipv6, B, NeighborSolicitation>),
35 NeighborAdvertisement(IcmpPacketRaw<Ipv6, B, NeighborAdvertisement>),
36 Redirect(IcmpPacketRaw<Ipv6, B, Redirect>),
37}
38
39#[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
41pub enum NonZeroNdpLifetime {
42 Finite(NonZeroDuration),
50
51 Infinite,
53}
54
55impl NonZeroNdpLifetime {
56 pub fn from_u32_with_infinite(lifetime: u32) -> Option<NonZeroNdpLifetime> {
59 match lifetime {
77 u32::MAX => Some(NonZeroNdpLifetime::Infinite),
78 finite => NonZeroDuration::new(Duration::from_secs(finite.into()))
79 .map(NonZeroNdpLifetime::Finite),
80 }
81 }
82
83 pub fn min_finite_duration(self, other: NonZeroDuration) -> NonZeroDuration {
85 match self {
86 NonZeroNdpLifetime::Finite(lifetime) => core::cmp::min(lifetime, other),
87 NonZeroNdpLifetime::Infinite => other,
88 }
89 }
90}
91
92pub type Options<B> = packet::records::options::Options<B, options::NdpOptionsImpl>;
98
99pub type OptionSequenceBuilder<'a, I> =
105 packet::records::options::OptionSequenceBuilder<options::NdpOptionBuilder<'a>, I>;
106
107#[derive(
109 Copy,
110 Clone,
111 Default,
112 Debug,
113 KnownLayout,
114 FromBytes,
115 IntoBytes,
116 Immutable,
117 Unaligned,
118 PartialEq,
119 Eq,
120)]
121#[repr(C)]
122pub struct RouterSolicitation {
123 _reserved: [u8; 4],
124}
125
126impl_icmp_message!(Ipv6, RouterSolicitation, RouterSolicitation, IcmpZeroCode, Options<B>);
127
128#[allow(missing_docs)]
132#[derive(Copy, Clone, Debug, PartialEq, Eq)]
133pub enum RoutePreference {
134 High,
140 Medium,
141 Low,
142}
143
144impl Default for RoutePreference {
145 fn default() -> RoutePreference {
146 RoutePreference::Medium
156 }
157}
158
159impl From<RoutePreference> for u8 {
160 fn from(v: RoutePreference) -> u8 {
161 match v {
171 RoutePreference::High => 0b01,
172 RoutePreference::Medium => 0b00,
173 RoutePreference::Low => 0b11,
174 }
175 }
176}
177
178impl TryFrom<u8> for RoutePreference {
179 type Error = ();
180
181 fn try_from(v: u8) -> Result<Self, Self::Error> {
182 match v {
192 0b01 => Ok(RoutePreference::High),
193 0b00 => Ok(RoutePreference::Medium),
194 0b11 => Ok(RoutePreference::Low),
195 _ => Err(()),
196 }
197 }
198}
199
200#[derive(
202 Copy, Clone, Debug, KnownLayout, FromBytes, IntoBytes, Immutable, Unaligned, PartialEq, Eq,
203)]
204#[repr(C)]
205pub struct RouterAdvertisement {
206 current_hop_limit: u8,
207 configuration_mo: u8,
208 router_lifetime: U16,
209 reachable_time: U32,
210 retransmit_timer: U32,
211}
212
213impl_icmp_message!(Ipv6, RouterAdvertisement, RouterAdvertisement, IcmpZeroCode, Options<B>);
214
215impl RouterAdvertisement {
216 const MANAGED_FLAG: u8 = 0x80;
224
225 const OTHER_CONFIGURATION_FLAG: u8 = 0x40;
231
232 const DEFAULT_ROUTER_PREFERENCE_SHIFT: u8 = 3;
256 const DEFAULT_ROUTER_PREFERENCE_MASK: u8 = 0b11 << Self::DEFAULT_ROUTER_PREFERENCE_SHIFT;
257
258 pub fn new(
262 current_hop_limit: u8,
263 managed_flag: bool,
264 other_config_flag: bool,
265 router_lifetime: u16,
266 reachable_time: u32,
267 retransmit_timer: u32,
268 ) -> Self {
269 Self::with_prf(
270 current_hop_limit,
271 managed_flag,
272 other_config_flag,
273 RoutePreference::default(),
274 router_lifetime,
275 reachable_time,
276 retransmit_timer,
277 )
278 }
279
280 pub fn with_prf(
282 current_hop_limit: u8,
283 managed_flag: bool,
284 other_config_flag: bool,
285 preference: RoutePreference,
286 router_lifetime: u16,
287 reachable_time: u32,
288 retransmit_timer: u32,
289 ) -> Self {
290 let mut configuration_mo = 0;
291
292 if managed_flag {
293 configuration_mo |= Self::MANAGED_FLAG;
294 }
295
296 if other_config_flag {
297 configuration_mo |= Self::OTHER_CONFIGURATION_FLAG;
298 }
299
300 configuration_mo |= (u8::from(preference) << Self::DEFAULT_ROUTER_PREFERENCE_SHIFT)
301 & Self::DEFAULT_ROUTER_PREFERENCE_MASK;
302
303 Self {
304 current_hop_limit,
305 configuration_mo,
306 router_lifetime: U16::new(router_lifetime),
307 reachable_time: U32::new(reachable_time),
308 retransmit_timer: U32::new(retransmit_timer),
309 }
310 }
311
312 pub fn current_hop_limit(&self) -> Option<NonZeroU8> {
316 NonZeroU8::new(self.current_hop_limit)
317 }
318
319 pub fn router_lifetime(&self) -> Option<NonZeroDuration> {
324 NonZeroDuration::new(Duration::from_secs(self.router_lifetime.get().into()))
327 }
328
329 pub fn reachable_time(&self) -> Option<NonZeroDuration> {
333 NonZeroDuration::new(Duration::from_millis(self.reachable_time.get().into()))
336 }
337
338 pub fn retransmit_timer(&self) -> Option<NonZeroDuration> {
342 NonZeroDuration::new(Duration::from_millis(self.retransmit_timer.get().into()))
345 }
346}
347
348#[derive(
350 Copy, Clone, Debug, KnownLayout, FromBytes, IntoBytes, Immutable, Unaligned, PartialEq, Eq,
351)]
352#[repr(C)]
353pub struct NeighborSolicitation {
354 _reserved: [u8; 4],
355 target_address: Ipv6Addr,
356}
357
358impl_icmp_message!(Ipv6, NeighborSolicitation, NeighborSolicitation, IcmpZeroCode, Options<B>);
359
360impl NeighborSolicitation {
361 pub fn new(target_address: Ipv6Addr) -> Self {
364 Self { _reserved: [0; 4], target_address }
365 }
366
367 pub fn target_address(&self) -> &Ipv6Addr {
369 &self.target_address
370 }
371}
372
373#[derive(
375 Copy, Clone, Debug, KnownLayout, FromBytes, IntoBytes, Immutable, Unaligned, PartialEq, Eq,
376)]
377#[repr(C)]
378pub struct NeighborAdvertisement {
379 flags_rso: u8,
380 _reserved: [u8; 3],
381 target_address: Ipv6Addr,
382}
383
384impl_icmp_message!(Ipv6, NeighborAdvertisement, NeighborAdvertisement, IcmpZeroCode, Options<B>);
385
386impl NeighborAdvertisement {
387 const FLAG_ROUTER: u8 = 0x80;
393
394 const FLAG_SOLICITED: u8 = 0x40;
402
403 const FLAG_OVERRIDE: u8 = 0x20;
414
415 pub fn new(
418 router_flag: bool,
419 solicited_flag: bool,
420 override_flag: bool,
421 target_address: Ipv6Addr,
422 ) -> Self {
423 let mut flags_rso = 0;
424
425 if router_flag {
426 flags_rso |= Self::FLAG_ROUTER;
427 }
428
429 if solicited_flag {
430 flags_rso |= Self::FLAG_SOLICITED;
431 }
432
433 if override_flag {
434 flags_rso |= Self::FLAG_OVERRIDE;
435 }
436
437 Self { flags_rso, _reserved: [0; 3], target_address }
438 }
439
440 pub fn target_address(&self) -> &Ipv6Addr {
442 &self.target_address
443 }
444
445 pub fn router_flag(&self) -> bool {
447 (self.flags_rso & Self::FLAG_ROUTER) != 0
448 }
449
450 pub fn solicited_flag(&self) -> bool {
452 (self.flags_rso & Self::FLAG_SOLICITED) != 0
453 }
454
455 pub fn override_flag(&self) -> bool {
457 (self.flags_rso & Self::FLAG_OVERRIDE) != 0
458 }
459}
460
461#[derive(
463 Copy, Clone, Debug, KnownLayout, FromBytes, IntoBytes, Immutable, Unaligned, PartialEq, Eq,
464)]
465#[repr(C)]
466pub struct Redirect {
467 _reserved: [u8; 4],
468 target_address: Ipv6Addr,
469 destination_address: Ipv6Addr,
470}
471
472impl_icmp_message!(Ipv6, Redirect, Redirect, IcmpZeroCode, Options<B>);
473
474pub mod options {
476 use core::num::NonZeroUsize;
477 use core::time::Duration;
478
479 use byteorder::{ByteOrder, NetworkEndian};
480 use net_types::ip::{IpAddress as _, Ipv6Addr, Subnet, SubnetError};
481 use net_types::UnicastAddress;
482 use packet::records::options::{
483 LengthEncoding, OptionBuilder, OptionLayout, OptionParseErr, OptionParseLayout, OptionsImpl,
484 };
485 use packet::BufferView as _;
486 use zerocopy::byteorder::network_endian::U32;
487 use zerocopy::{FromBytes, Immutable, IntoBytes, KnownLayout, Ref, SplitByteSlice, Unaligned};
488
489 use super::NonZeroNdpLifetime;
490 use crate::utils::NonZeroDuration;
491
492 pub const INFINITE_LIFETIME_SECONDS: u32 = u32::MAX;
494
495 pub const INFINITE_LIFETIME: NonZeroDuration =
498 NonZeroDuration::from_secs(INFINITE_LIFETIME_SECONDS as u64).unwrap();
499
500 const REDIRECTED_HEADER_OPTION_RESERVED_BYTES_LENGTH: usize = 6;
507
508 const MTU_OPTION_LENGTH: usize = 6;
514
515 const MTU_OPTION_RESERVED_BYTES_LENGTH: usize = 2;
522
523 pub const MIN_NONCE_LENGTH: usize = 6;
529
530 const MIN_RECURSIVE_DNS_SERVER_OPTION_LENGTH: usize = 22;
540
541 const RECURSIVE_DNS_SERVER_OPTION_RESERVED_BYTES_LENGTH: usize = 2;
548
549 const ROUTE_INFORMATION_PREFERENCE_RESERVED_BITS_RIGHT: u8 = 3;
555
556 const ROUTE_INFORMATION_PREFERENCE_MASK: u8 = 0x18;
562
563 const OPTION_BYTES_PER_LENGTH_UNIT: usize = 8;
569
570 #[derive(Debug, PartialEq, Eq, Clone)]
576 pub struct RecursiveDnsServer<'a> {
577 lifetime: u32,
578 addresses: &'a [Ipv6Addr],
579 }
580
581 impl<'a> RecursiveDnsServer<'a> {
582 pub const INFINITE_LIFETIME: u32 = INFINITE_LIFETIME_SECONDS;
584
585 pub fn new(lifetime: u32, addresses: &'a [Ipv6Addr]) -> RecursiveDnsServer<'a> {
587 RecursiveDnsServer { lifetime, addresses }
588 }
589
590 pub fn lifetime(&self) -> Option<NonZeroDuration> {
596 NonZeroDuration::new(Duration::from_secs(self.lifetime.into()))
597 }
598
599 pub fn iter_addresses(&self) -> &'a [Ipv6Addr] {
601 self.addresses
602 }
603
604 pub fn parse(data: &'a [u8]) -> Result<Self, OptionParseErr> {
607 if data.len() < MIN_RECURSIVE_DNS_SERVER_OPTION_LENGTH {
608 return Err(OptionParseErr);
609 }
610
611 let (_, data) = data.split_at(RECURSIVE_DNS_SERVER_OPTION_RESERVED_BYTES_LENGTH);
614
615 let (lifetime, data) = Ref::<_, U32>::from_prefix(data).map_err(|_| OptionParseErr)?;
618
619 let addresses = Ref::into_ref(
622 Ref::<_, [Ipv6Addr]>::from_bytes(data)
623 .map_err(Into::into)
624 .map_err(|_: zerocopy::SizeError<_, _>| OptionParseErr)?,
625 );
626
627 if !addresses.iter().all(UnicastAddress::is_unicast) {
629 return Err(OptionParseErr);
630 }
631
632 Ok(Self::new(lifetime.get(), addresses))
633 }
634 }
635
636 #[derive(KnownLayout, FromBytes, IntoBytes, Immutable, Unaligned)]
659 #[repr(C)]
660 struct RouteInformationHeader {
661 prefix_length: u8,
662 flags: u8,
663 route_lifetime: U32,
664 }
665
666 impl RouteInformationHeader {
667 const PREFERENCE_SHIFT: u8 = 3;
683 const PREFERENCE_MASK: u8 = 0b11 << Self::PREFERENCE_SHIFT;
684
685 fn set_preference(&mut self, preference: super::RoutePreference) {
686 let preference: u8 = preference.into();
687
688 self.flags &= !Self::PREFERENCE_MASK;
689 self.flags |= (preference << Self::PREFERENCE_SHIFT) & Self::PREFERENCE_MASK;
690 }
691 }
692
693 #[derive(Debug, PartialEq, Eq)]
699 pub struct RouteInformation {
700 prefix: Subnet<Ipv6Addr>,
701 route_lifetime_seconds: u32,
702 preference: super::RoutePreference,
703 }
704
705 impl RouteInformation {
706 pub fn new(
708 prefix: Subnet<Ipv6Addr>,
709 route_lifetime_seconds: u32,
710 preference: super::RoutePreference,
711 ) -> Self {
712 Self { prefix, route_lifetime_seconds, preference }
713 }
714
715 pub fn prefix(&self) -> &Subnet<Ipv6Addr> {
717 &self.prefix
718 }
719
720 pub fn preference(&self) -> super::RoutePreference {
722 self.preference
723 }
724
725 pub fn route_lifetime(&self) -> Option<NonZeroNdpLifetime> {
727 NonZeroNdpLifetime::from_u32_with_infinite(self.route_lifetime_seconds)
728 }
729
730 fn prefix_bytes_len(&self) -> usize {
731 let RouteInformation { prefix, route_lifetime_seconds: _, preference: _ } = self;
732
733 let prefix_length = prefix.prefix();
734 if prefix_length == 0 {
747 0
748 } else if prefix_length <= 64 {
749 core::mem::size_of::<Ipv6Addr>() / 2
750 } else {
751 core::mem::size_of::<Ipv6Addr>()
752 }
753 }
754
755 fn serialized_len(&self) -> usize {
756 core::mem::size_of::<RouteInformationHeader>() + self.prefix_bytes_len()
757 }
758
759 fn serialize(&self, buffer: &mut [u8]) {
760 let (mut hdr, buffer) = Ref::<_, RouteInformationHeader>::from_prefix(buffer)
761 .expect("expected buffer to hold enough bytes for serialization");
762
763 let prefix_bytes_len = self.prefix_bytes_len();
764 let RouteInformation { prefix, route_lifetime_seconds, preference } = self;
765
766 hdr.prefix_length = prefix.prefix();
767 hdr.set_preference(*preference);
768 hdr.route_lifetime.set(*route_lifetime_seconds);
769 buffer[..prefix_bytes_len]
770 .copy_from_slice(&prefix.network().bytes()[..prefix_bytes_len])
771 }
772 }
773
774 const PREFIX_INFORMATION_OPTION_LENGTH: usize = 30;
781
782 #[derive(
788 Debug, KnownLayout, FromBytes, IntoBytes, Immutable, Unaligned, PartialEq, Eq, Clone,
789 )]
790 #[repr(C)]
791 pub struct PrefixInformation {
792 prefix_length: u8,
793 flags_la: u8,
794 valid_lifetime: U32,
795 preferred_lifetime: U32,
796 _reserved: [u8; 4],
797 prefix: Ipv6Addr,
798 }
799
800 impl PrefixInformation {
801 const ON_LINK_FLAG: u8 = 0x80;
807
808 const AUTONOMOUS_ADDRESS_CONFIGURATION_FLAG: u8 = 0x40;
815
816 pub fn new(
818 prefix_length: u8,
819 on_link_flag: bool,
820 autonomous_address_configuration_flag: bool,
821 valid_lifetime: u32,
822 preferred_lifetime: u32,
823 prefix: Ipv6Addr,
824 ) -> Self {
825 let mut flags_la = 0;
826
827 if on_link_flag {
828 flags_la |= Self::ON_LINK_FLAG;
829 }
830
831 if autonomous_address_configuration_flag {
832 flags_la |= Self::AUTONOMOUS_ADDRESS_CONFIGURATION_FLAG;
833 }
834
835 Self {
836 prefix_length,
837 flags_la,
838 valid_lifetime: U32::new(valid_lifetime),
839 preferred_lifetime: U32::new(preferred_lifetime),
840 _reserved: [0; 4],
841 prefix,
842 }
843 }
844
845 pub fn prefix_length(&self) -> u8 {
847 self.prefix_length
848 }
849
850 pub fn on_link_flag(&self) -> bool {
857 (self.flags_la & Self::ON_LINK_FLAG) != 0
858 }
859
860 pub fn autonomous_address_configuration_flag(&self) -> bool {
862 (self.flags_la & Self::AUTONOMOUS_ADDRESS_CONFIGURATION_FLAG) != 0
863 }
864
865 pub fn valid_lifetime(&self) -> Option<NonZeroNdpLifetime> {
871 NonZeroNdpLifetime::from_u32_with_infinite(self.valid_lifetime.get())
872 }
873
874 pub fn preferred_lifetime(&self) -> Option<NonZeroNdpLifetime> {
880 NonZeroNdpLifetime::from_u32_with_infinite(self.preferred_lifetime.get())
881 }
882
883 pub fn prefix(&self) -> &Ipv6Addr {
890 &self.prefix
891 }
892
893 pub fn subnet(&self) -> Result<Subnet<Ipv6Addr>, SubnetError> {
895 Subnet::new(self.prefix, self.prefix_length)
896 }
897 }
898
899 create_protocol_enum!(
900 #[allow(missing_docs)]
902 pub enum NdpOptionType: u8 {
903 SourceLinkLayerAddress, 1, "Source Link-Layer Address";
904 TargetLinkLayerAddress, 2, "Target Link-Layer Address";
905 PrefixInformation, 3, "Prefix Information";
906 RedirectedHeader, 4, "Redirected Header";
907 Mtu, 5, "MTU";
908 Nonce, 14, "Nonce";
909 RouteInformation, 24, "Route Information";
910 RecursiveDnsServer, 25, "Recursive DNS Server";
911 }
912 );
913
914 #[derive(Debug, PartialEq, Eq, Copy, Clone, PartialOrd, Ord)]
921 pub struct NdpNonce<B: SplitByteSlice> {
922 nonce: B,
923 }
924
925 impl<B: SplitByteSlice> NdpNonce<B> {
926 pub fn bytes(&self) -> &[u8] {
928 let Self { nonce } = self;
929 nonce.deref()
930 }
931
932 pub fn new(value: B) -> Result<Self, InvalidNonceError> {
935 let bytes = value.deref();
936 let nonce_option_length_bytes = bytes.len() + 2;
940 if nonce_option_length_bytes % 8 != 0 {
941 return Err(InvalidNonceError::ResultsInNonMultipleOf8);
942 }
943
944 let nonce_option_length_in_groups_of_8_bytes = nonce_option_length_bytes / 8;
945
946 match u8::try_from(nonce_option_length_in_groups_of_8_bytes) {
949 Ok(_) => (),
950 Err(_) => return Err(InvalidNonceError::TooLong),
951 };
952
953 Ok(Self { nonce: value })
954 }
955 }
956
957 impl<B: SplitByteSlice> AsRef<[u8]> for NdpNonce<B> {
958 fn as_ref(&self) -> &[u8] {
959 self.bytes()
960 }
961 }
962
963 impl<'a> From<&'a [u8; MIN_NONCE_LENGTH]> for NdpNonce<&'a [u8]> {
966 fn from(value: &'a [u8; MIN_NONCE_LENGTH]) -> Self {
967 Self { nonce: &value[..] }
968 }
969 }
970
971 #[derive(Debug, PartialEq, Eq, Copy, Clone)]
973 pub enum InvalidNonceError {
974 ResultsInNonMultipleOf8,
977 TooLong,
979 }
980
981 #[allow(missing_docs)]
983 #[derive(Debug, PartialEq, Eq)]
984 pub enum NdpOption<'a> {
985 SourceLinkLayerAddress(&'a [u8]),
986 TargetLinkLayerAddress(&'a [u8]),
987 PrefixInformation(&'a PrefixInformation),
988
989 RedirectedHeader { original_packet: &'a [u8] },
990
991 Mtu(u32),
992 Nonce(NdpNonce<&'a [u8]>),
993
994 RecursiveDnsServer(RecursiveDnsServer<'a>),
995 RouteInformation(RouteInformation),
996 }
997
998 impl<'a> NdpOption<'a> {
999 pub fn nonce(self) -> Option<NdpNonce<&'a [u8]>> {
1001 match self {
1002 NdpOption::Nonce(nonce) => Some(nonce),
1003 _ => None,
1004 }
1005 }
1006
1007 pub fn source_link_layer_address(self) -> Option<&'a [u8]> {
1009 match self {
1010 NdpOption::SourceLinkLayerAddress(a) => Some(a),
1011 _ => None,
1012 }
1013 }
1014
1015 pub fn target_link_layer_address(self) -> Option<&'a [u8]> {
1017 match self {
1018 NdpOption::TargetLinkLayerAddress(a) => Some(a),
1019 _ => None,
1020 }
1021 }
1022 }
1023
1024 #[derive(Debug)]
1026 pub struct NdpOptionsImpl;
1027
1028 impl<'a> OptionLayout for NdpOptionsImpl {
1029 type KindLenField = u8;
1030
1031 const LENGTH_ENCODING: LengthEncoding = LengthEncoding::TypeLengthValue {
1033 option_len_multiplier: NonZeroUsize::new(8).unwrap(),
1034 };
1035 }
1036
1037 impl OptionParseLayout for NdpOptionsImpl {
1038 type Error = OptionParseErr;
1040
1041 const END_OF_OPTIONS: Option<u8> = None;
1043 const NOP: Option<u8> = None;
1044 }
1045
1046 impl OptionsImpl for NdpOptionsImpl {
1047 type Option<'a> = NdpOption<'a>;
1048
1049 fn parse<'a>(
1050 kind: u8,
1051 mut data: &'a [u8],
1052 ) -> Result<Option<NdpOption<'a>>, OptionParseErr> {
1053 let kind = if let Ok(k) = NdpOptionType::try_from(kind) {
1054 k
1055 } else {
1056 return Ok(None);
1057 };
1058
1059 let opt = match kind {
1060 NdpOptionType::SourceLinkLayerAddress => NdpOption::SourceLinkLayerAddress(data),
1061 NdpOptionType::TargetLinkLayerAddress => NdpOption::TargetLinkLayerAddress(data),
1062 NdpOptionType::PrefixInformation => {
1063 let data = Ref::<_, PrefixInformation>::from_bytes(data)
1064 .map_err(|_| OptionParseErr)?;
1065 NdpOption::PrefixInformation(Ref::into_ref(data))
1066 }
1067 NdpOptionType::RedirectedHeader => NdpOption::RedirectedHeader {
1068 original_packet: &data[REDIRECTED_HEADER_OPTION_RESERVED_BYTES_LENGTH..],
1069 },
1070 NdpOptionType::Mtu => NdpOption::Mtu(NetworkEndian::read_u32(
1071 &data[MTU_OPTION_RESERVED_BYTES_LENGTH..],
1072 )),
1073 NdpOptionType::Nonce => NdpOption::Nonce(
1074 NdpNonce::new(data).map_err(|_: InvalidNonceError| OptionParseErr)?,
1075 ),
1076 NdpOptionType::RecursiveDnsServer => {
1077 NdpOption::RecursiveDnsServer(RecursiveDnsServer::parse(data)?)
1078 }
1079 NdpOptionType::RouteInformation => {
1080 #[derive(KnownLayout, FromBytes, Immutable, Unaligned)]
1083 #[repr(C)]
1084 struct RouteInfoFixed {
1085 prefix_length: u8,
1086 preference_raw: u8,
1087 route_lifetime_seconds: U32,
1088 }
1089
1090 let mut buf = &mut data;
1091
1092 let fixed = buf.take_obj_front::<RouteInfoFixed>().ok_or(OptionParseErr)?;
1093
1094 let preference = super::RoutePreference::try_from(
1096 (fixed.preference_raw & ROUTE_INFORMATION_PREFERENCE_MASK)
1097 >> ROUTE_INFORMATION_PREFERENCE_RESERVED_BITS_RIGHT,
1098 )
1099 .map_err(|()| OptionParseErr)?;
1100
1101 let buf_len = buf.len();
1113 if buf_len % OPTION_BYTES_PER_LENGTH_UNIT != 0 {
1114 return Err(OptionParseErr);
1115 }
1116 let length = buf_len / OPTION_BYTES_PER_LENGTH_UNIT;
1117 match (fixed.prefix_length, length) {
1118 (65..=128, 2) => {}
1119 (1..=64, 1 | 2) => {}
1120 (0, 0 | 1 | 2) => {}
1121 _ => return Err(OptionParseErr),
1122 }
1123
1124 let mut prefix_buf = [0; 16];
1125 prefix_buf[..buf_len].copy_from_slice(&buf);
1127 let prefix = Ipv6Addr::from_bytes(prefix_buf);
1128
1129 NdpOption::RouteInformation(RouteInformation::new(
1130 Subnet::new(prefix, fixed.prefix_length).map_err(|_| OptionParseErr)?,
1131 fixed.route_lifetime_seconds.get(),
1132 preference,
1133 ))
1134 }
1135 };
1136
1137 Ok(Some(opt))
1138 }
1139 }
1140
1141 #[allow(missing_docs)]
1143 #[derive(Debug)]
1144 pub enum NdpOptionBuilder<'a> {
1145 SourceLinkLayerAddress(&'a [u8]),
1146 TargetLinkLayerAddress(&'a [u8]),
1147 PrefixInformation(PrefixInformation),
1148
1149 RedirectedHeader { original_packet: &'a [u8] },
1150
1151 Mtu(u32),
1152 Nonce(NdpNonce<&'a [u8]>),
1153
1154 RouteInformation(RouteInformation),
1155 RecursiveDnsServer(RecursiveDnsServer<'a>),
1156 }
1157
1158 impl<'a> From<&NdpOptionBuilder<'a>> for NdpOptionType {
1159 fn from(v: &NdpOptionBuilder<'a>) -> Self {
1160 match v {
1161 NdpOptionBuilder::SourceLinkLayerAddress(_) => {
1162 NdpOptionType::SourceLinkLayerAddress
1163 }
1164 NdpOptionBuilder::TargetLinkLayerAddress(_) => {
1165 NdpOptionType::TargetLinkLayerAddress
1166 }
1167 NdpOptionBuilder::PrefixInformation(_) => NdpOptionType::PrefixInformation,
1168 NdpOptionBuilder::RedirectedHeader { .. } => NdpOptionType::RedirectedHeader,
1169 NdpOptionBuilder::Mtu { .. } => NdpOptionType::Mtu,
1170 NdpOptionBuilder::Nonce(_) => NdpOptionType::Nonce,
1171 NdpOptionBuilder::RouteInformation(_) => NdpOptionType::RouteInformation,
1172 NdpOptionBuilder::RecursiveDnsServer(_) => NdpOptionType::RecursiveDnsServer,
1173 }
1174 }
1175 }
1176
1177 impl<'a> OptionBuilder for NdpOptionBuilder<'a> {
1178 type Layout = NdpOptionsImpl;
1179
1180 fn serialized_len(&self) -> usize {
1181 match self {
1182 NdpOptionBuilder::SourceLinkLayerAddress(data)
1183 | NdpOptionBuilder::TargetLinkLayerAddress(data) => data.len(),
1184 NdpOptionBuilder::PrefixInformation(_) => PREFIX_INFORMATION_OPTION_LENGTH,
1185 NdpOptionBuilder::RedirectedHeader { original_packet } => {
1186 REDIRECTED_HEADER_OPTION_RESERVED_BYTES_LENGTH + original_packet.len()
1187 }
1188 NdpOptionBuilder::Mtu(_) => MTU_OPTION_LENGTH,
1189 NdpOptionBuilder::Nonce(NdpNonce { nonce }) => nonce.len(),
1190 NdpOptionBuilder::RouteInformation(o) => o.serialized_len(),
1191 NdpOptionBuilder::RecursiveDnsServer(RecursiveDnsServer {
1192 lifetime,
1193 addresses,
1194 }) => {
1195 RECURSIVE_DNS_SERVER_OPTION_RESERVED_BYTES_LENGTH
1196 + core::mem::size_of_val(lifetime)
1197 + core::mem::size_of_val(*addresses)
1198 }
1199 }
1200 }
1201
1202 fn option_kind(&self) -> u8 {
1203 NdpOptionType::from(self).into()
1204 }
1205
1206 fn serialize_into(&self, buffer: &mut [u8]) {
1207 match self {
1208 NdpOptionBuilder::SourceLinkLayerAddress(data)
1209 | NdpOptionBuilder::TargetLinkLayerAddress(data) => buffer.copy_from_slice(data),
1210 NdpOptionBuilder::PrefixInformation(pfx_info) => {
1211 buffer.copy_from_slice(pfx_info.as_bytes());
1212 }
1213 NdpOptionBuilder::RedirectedHeader { original_packet } => {
1214 let (reserved_bytes, original_packet_bytes) =
1218 buffer.split_at_mut(REDIRECTED_HEADER_OPTION_RESERVED_BYTES_LENGTH);
1219 reserved_bytes
1220 .copy_from_slice(&[0; REDIRECTED_HEADER_OPTION_RESERVED_BYTES_LENGTH]);
1221 original_packet_bytes.copy_from_slice(original_packet);
1222 }
1223 NdpOptionBuilder::Mtu(mtu) => {
1224 let (reserved_bytes, mtu_bytes) =
1227 buffer.split_at_mut(MTU_OPTION_RESERVED_BYTES_LENGTH);
1228 reserved_bytes.copy_from_slice(&[0; MTU_OPTION_RESERVED_BYTES_LENGTH]);
1229 mtu_bytes.copy_from_slice(U32::new(*mtu).as_bytes());
1230 }
1231 NdpOptionBuilder::Nonce(NdpNonce { nonce }) => {
1232 buffer.copy_from_slice(nonce);
1233 }
1234 NdpOptionBuilder::RouteInformation(p) => p.serialize(buffer),
1235 NdpOptionBuilder::RecursiveDnsServer(RecursiveDnsServer {
1236 lifetime,
1237 addresses,
1238 }) => {
1239 let (reserved_bytes, buffer) =
1242 buffer.split_at_mut(RECURSIVE_DNS_SERVER_OPTION_RESERVED_BYTES_LENGTH);
1243 reserved_bytes
1244 .copy_from_slice(&[0; RECURSIVE_DNS_SERVER_OPTION_RESERVED_BYTES_LENGTH]);
1245
1246 let (lifetime_bytes, addresses_bytes) =
1250 buffer.split_at_mut(core::mem::size_of_val(lifetime));
1251 lifetime_bytes.copy_from_slice(U32::new(*lifetime).as_bytes());
1252 addresses_bytes.copy_from_slice(addresses.as_bytes());
1253 }
1254 }
1255 }
1256 }
1257}
1258
1259#[cfg(test)]
1260mod tests {
1261 use byteorder::{ByteOrder, NetworkEndian};
1262 use net_types::ip::{Ip, IpAddress, Subnet};
1263 use packet::serialize::Serializer;
1264 use packet::{InnerPacketBuilder, ParseBuffer};
1265 use test_case::test_case;
1266 use zerocopy::Ref;
1267
1268 use super::*;
1269 use crate::icmp::{IcmpPacketBuilder, IcmpParseArgs};
1270 use crate::ipv6::{Ipv6Header, Ipv6Packet};
1271
1272 #[test]
1273 fn parse_serialize_redirected_header() {
1274 let expected_packet = [1, 2, 3, 4, 5, 6, 7, 8];
1275 let options =
1276 &[options::NdpOptionBuilder::RedirectedHeader { original_packet: &expected_packet }];
1277 let serialized = OptionSequenceBuilder::new(options.iter())
1278 .into_serializer()
1279 .serialize_vec_outer()
1280 .unwrap();
1281 let mut expected = [0; 16];
1283 (&mut expected[..2]).copy_from_slice(&[4, 2]);
1288 (&mut expected[8..]).copy_from_slice(&expected_packet);
1289 assert_eq!(serialized.as_ref(), expected);
1290
1291 let parsed = Options::parse(&expected[..]).unwrap();
1292 let parsed = parsed.iter().collect::<Vec<options::NdpOption<'_>>>();
1293 assert_eq!(parsed.len(), 1);
1294 assert_eq!(
1295 options::NdpOption::RedirectedHeader { original_packet: &expected_packet },
1296 parsed[0]
1297 );
1298 }
1299
1300 #[test]
1301 fn parse_serialize_mtu_option() {
1302 let expected_mtu = 5781;
1303 let options = &[options::NdpOptionBuilder::Mtu(expected_mtu)];
1304 let serialized = OptionSequenceBuilder::new(options.iter())
1305 .into_serializer()
1306 .serialize_vec_outer()
1307 .unwrap();
1308 let mut expected = [5, 1, 0, 0, 0, 0, 0, 0];
1313 NetworkEndian::write_u32(&mut expected[4..], expected_mtu);
1314 assert_eq!(serialized.as_ref(), expected);
1315
1316 let parsed = Options::parse(&expected[..]).unwrap();
1317 let parsed = parsed.iter().collect::<Vec<options::NdpOption<'_>>>();
1318 assert_eq!(parsed.len(), 1);
1319 assert_eq!(options::NdpOption::Mtu(expected_mtu), parsed[0]);
1320 }
1321
1322 #[test_case(
1323 options::MIN_NONCE_LENGTH - 1 =>
1324 matches Err(options::InvalidNonceError::ResultsInNonMultipleOf8);
1325 "resulting nonce option length must be multiple of 8")]
1326 #[test_case(
1327 options::MIN_NONCE_LENGTH => matches Ok(_);
1328 "MIN_NONCE_LENGTH must validate successfully")]
1329 #[test_case(
1330 usize::from(u8::MAX) * 8 - 2 => matches Ok(_);
1331 "maximum possible nonce length must validate successfully")]
1332 #[test_case(
1333 usize::from(u8::MAX) * 8 - 2 + 8 =>
1334 matches Err(options::InvalidNonceError::TooLong);
1335 "nonce option's length must fit in u8")]
1336 fn nonce_length_validation(
1337 length: usize,
1338 ) -> Result<options::NdpNonce<&'static [u8]>, options::InvalidNonceError> {
1339 const LEN: usize = (u8::MAX as usize + 1) * 8;
1340 const BYTES: [u8; LEN] = [0u8; LEN];
1341 options::NdpNonce::new(&BYTES[..length])
1342 }
1343
1344 #[test]
1345 fn parse_serialize_nonce_option() {
1346 let expected_nonce: [u8; 6] = [1, 2, 3, 4, 5, 6];
1347 let nonce = options::NdpNonce::new(&expected_nonce[..]).expect("should be valid nonce");
1348 let options = &[options::NdpOptionBuilder::Nonce(nonce)];
1349 let serialized = OptionSequenceBuilder::new(options.iter())
1350 .into_serializer()
1351 .serialize_vec_outer()
1352 .unwrap();
1353
1354 let mut expected_bytes: [u8; 8] = [14, 1, 0, 0, 0, 0, 0, 0];
1357 expected_bytes[2..].copy_from_slice(&expected_nonce);
1358
1359 assert_eq!(serialized.as_ref(), expected_bytes);
1360
1361 let parsed = Options::parse(&expected_bytes[..]).unwrap();
1362 let parsed = parsed.iter().collect::<Vec<options::NdpOption<'_>>>();
1363 assert_eq!(parsed.len(), 1);
1364 assert_eq!(parsed[0], options::NdpOption::Nonce(nonce));
1365 }
1366
1367 #[test]
1368 fn parse_serialize_prefix_option() {
1369 let expected_prefix_info = options::PrefixInformation::new(
1370 120,
1371 true,
1372 false,
1373 100,
1374 100,
1375 Ipv6Addr::from([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 192, 168, 0, 0]),
1376 );
1377 let options = &[options::NdpOptionBuilder::PrefixInformation(expected_prefix_info.clone())];
1378 let serialized = OptionSequenceBuilder::new(options.iter())
1379 .into_serializer()
1380 .serialize_vec_outer()
1381 .unwrap();
1382 let mut expected = [0; 32];
1387 expected[0] = 3;
1388 expected[1] = 4;
1389 (&mut expected[2..]).copy_from_slice(expected_prefix_info.as_bytes());
1390 assert_eq!(serialized.as_ref(), expected);
1391
1392 let parsed = Options::parse(&expected[..]).unwrap();
1393 let parsed = parsed.iter().collect::<Vec<options::NdpOption<'_>>>();
1394 assert_eq!(parsed.len(), 1);
1395 assert_eq!(options::NdpOption::PrefixInformation(&expected_prefix_info), parsed[0]);
1396 }
1397
1398 #[test]
1399 fn parse_serialize_rdnss_option() {
1400 let test = |addrs: &[Ipv6Addr]| {
1401 let lifetime = 120;
1402 let expected_rdnss = options::RecursiveDnsServer::new(lifetime, addrs);
1403 let options = &[options::NdpOptionBuilder::RecursiveDnsServer(expected_rdnss.clone())];
1404 let serialized = OptionSequenceBuilder::new(options.iter())
1405 .into_serializer()
1406 .serialize_vec_outer()
1407 .unwrap();
1408 let mut expected = vec![0; 8 + addrs.len() * usize::from(Ipv6Addr::BYTES)];
1411 (&mut expected[..4]).copy_from_slice(&[
1416 25,
1417 1 + u8::try_from(addrs.len()).unwrap() * 2,
1418 0,
1419 0,
1420 ]);
1421 NetworkEndian::write_u32(&mut expected[4..8], lifetime);
1423 (&mut expected[8..]).copy_from_slice(addrs.as_bytes());
1425 assert_eq!(serialized.as_ref(), expected.as_slice());
1426
1427 let parsed = Options::parse(&expected[..])
1428 .expect("should have parsed a valid recursive dns erver option");
1429 let parsed = parsed.iter().collect::<Vec<options::NdpOption<'_>>>();
1430 assert_eq!(parsed.len(), 1);
1431
1432 assert_eq!(
1434 options::RecursiveDnsServer::parse(&expected[2..]).expect("parsing should succeed"),
1435 expected_rdnss
1436 );
1437
1438 assert_eq!(options::NdpOption::RecursiveDnsServer(expected_rdnss), parsed[0]);
1439 };
1440 test(&[Ipv6Addr::from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16])]);
1441 test(&[
1442 Ipv6Addr::from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]),
1443 Ipv6Addr::from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 17]),
1444 ]);
1445 }
1446
1447 #[test]
1448 fn parse_serialize_rdnss_option_error() {
1449 let addrs = [
1450 Ipv6Addr::from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]),
1451 Ipv6Addr::from([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 17]),
1452 ];
1453 let lifetime = 120;
1454 let mut buf = vec![0; 8 + addrs.len() * usize::from(Ipv6Addr::BYTES)];
1457 (&mut buf[..4]).copy_from_slice(&[25, 1 + u8::try_from(addrs.len()).unwrap() * 2, 0, 0]);
1462 NetworkEndian::write_u32(&mut buf[4..8], lifetime);
1464 (&mut buf[8..]).copy_from_slice(addrs.as_bytes());
1466
1467 let _parsed = Options::parse(&buf[..])
1469 .expect("should have parsed a valid recursive dns erver option");
1470
1471 let _err = Options::parse(&buf[..8]).expect_err(
1473 "should not have parsed a recursive dns server option that has no addresses",
1474 );
1475
1476 let _err = Options::parse(&buf[..buf.len()-1])
1478 .expect_err("should not have parsed a recursive dns server option that cuts off in the middle of an address");
1479
1480 (&mut buf[8..8 + usize::from(Ipv6Addr::BYTES)])
1482 .copy_from_slice(Ipv6::UNSPECIFIED_ADDRESS.as_bytes());
1483 let _parsed = Options::parse(&buf[..]).expect_err(
1484 "should not have parsed a recursive dns erver option with an unspecified address",
1485 );
1486
1487 (&mut buf[8..8 + usize::from(Ipv6Addr::BYTES)])
1489 .copy_from_slice(Ipv6::ALL_NODES_LINK_LOCAL_MULTICAST_ADDRESS.as_bytes());
1490 let _parsed = Options::parse(&buf[..]).expect_err(
1491 "should not have parsed a recursive dns erver option with a multicast address",
1492 );
1493 }
1494
1495 #[test]
1496 fn parse_neighbor_solicitation() {
1497 use crate::icmp::testdata::ndp_neighbor::*;
1498 let mut buf = SOLICITATION_IP_PACKET_BYTES;
1499 let ip = buf.parse::<Ipv6Packet<_>>().unwrap();
1500 let ipv6_builder = ip.builder();
1501 let (src_ip, dst_ip) = (ip.src_ip(), ip.dst_ip());
1502 let icmp = buf
1503 .parse_with::<_, IcmpPacket<_, _, NeighborSolicitation>>(IcmpParseArgs::new(
1504 src_ip, dst_ip,
1505 ))
1506 .unwrap();
1507
1508 assert_eq!(icmp.message().target_address.ipv6_bytes(), TARGET_ADDRESS);
1509 let collected = icmp.ndp_options().iter().collect::<Vec<options::NdpOption<'_>>>();
1510 for option in collected.iter() {
1511 match option {
1512 options::NdpOption::SourceLinkLayerAddress(address) => {
1513 assert_eq!(address, &SOURCE_LINK_LAYER_ADDRESS);
1514 }
1515 o => panic!("Found unexpected option: {:?}", o),
1516 }
1517 }
1518 let option_builders =
1519 [options::NdpOptionBuilder::SourceLinkLayerAddress(&SOURCE_LINK_LAYER_ADDRESS)];
1520 let serialized = OptionSequenceBuilder::new(option_builders.iter())
1521 .into_serializer()
1522 .encapsulate(IcmpPacketBuilder::<Ipv6, _>::new(
1523 src_ip,
1524 dst_ip,
1525 IcmpZeroCode,
1526 *icmp.message(),
1527 ))
1528 .encapsulate(ipv6_builder)
1529 .serialize_vec_outer()
1530 .unwrap()
1531 .as_ref()
1532 .to_vec();
1533 assert_eq!(&serialized, &SOLICITATION_IP_PACKET_BYTES)
1534 }
1535
1536 #[test]
1537 fn parse_neighbor_advertisement() {
1538 use crate::icmp::testdata::ndp_neighbor::*;
1539 let mut buf = ADVERTISEMENT_IP_PACKET_BYTES;
1540 let ip = buf.parse::<Ipv6Packet<_>>().unwrap();
1541 let ipv6_builder = ip.builder();
1542 let (src_ip, dst_ip) = (ip.src_ip(), ip.dst_ip());
1543 let icmp = buf
1544 .parse_with::<_, IcmpPacket<_, _, NeighborAdvertisement>>(IcmpParseArgs::new(
1545 src_ip, dst_ip,
1546 ))
1547 .unwrap();
1548 assert_eq!(icmp.message().target_address.ipv6_bytes(), TARGET_ADDRESS);
1549 assert_eq!(icmp.ndp_options().iter().count(), 0);
1550
1551 let serialized = []
1552 .into_serializer()
1553 .encapsulate(IcmpPacketBuilder::<Ipv6, _>::new(
1554 src_ip,
1555 dst_ip,
1556 IcmpZeroCode,
1557 *icmp.message(),
1558 ))
1559 .encapsulate(ipv6_builder)
1560 .serialize_vec_outer()
1561 .unwrap()
1562 .as_ref()
1563 .to_vec();
1564 assert_eq!(&serialized, &ADVERTISEMENT_IP_PACKET_BYTES);
1565 }
1566
1567 #[test]
1568 fn parse_router_advertisement() {
1569 use crate::icmp::ndp::options::RouteInformation;
1570 use crate::icmp::testdata::ndp_router::*;
1571
1572 let mut buf = ADVERTISEMENT_IP_PACKET_BYTES;
1573 let ip = buf.parse::<Ipv6Packet<_>>().unwrap();
1574 let ipv6_builder = ip.builder();
1575 let (src_ip, dst_ip) = (ip.src_ip(), ip.dst_ip());
1576 let icmp = buf
1577 .parse_with::<_, IcmpPacket<_, _, RouterAdvertisement>>(IcmpParseArgs::new(
1578 src_ip, dst_ip,
1579 ))
1580 .unwrap();
1581 assert_eq!(icmp.message().current_hop_limit(), HOP_LIMIT);
1582 assert_eq!(icmp.message().router_lifetime(), LIFETIME);
1583 assert_eq!(icmp.message().reachable_time(), REACHABLE_TIME);
1584 assert_eq!(icmp.message().retransmit_timer(), RETRANS_TIMER);
1585
1586 assert_eq!(icmp.ndp_options().iter().count(), 5);
1587
1588 let collected = icmp.ndp_options().iter().collect::<Vec<options::NdpOption<'_>>>();
1589 for option in collected.iter() {
1590 match option {
1591 options::NdpOption::SourceLinkLayerAddress(address) => {
1592 assert_eq!(address, &SOURCE_LINK_LAYER_ADDRESS);
1593 }
1594 options::NdpOption::PrefixInformation(info) => {
1595 assert_eq!(info.on_link_flag(), PREFIX_INFO_ON_LINK_FLAG);
1596 assert_eq!(
1597 info.autonomous_address_configuration_flag(),
1598 PREFIX_INFO_AUTONOMOUS_ADDRESS_CONFIGURATION_FLAG
1599 );
1600 assert_eq!(
1601 info.valid_lifetime(),
1602 NonZeroNdpLifetime::from_u32_with_infinite(
1603 PREFIX_INFO_VALID_LIFETIME_SECONDS
1604 )
1605 );
1606 assert_eq!(
1607 info.preferred_lifetime(),
1608 NonZeroNdpLifetime::from_u32_with_infinite(
1609 PREFIX_INFO_PREFERRED_LIFETIME_SECONDS
1610 )
1611 );
1612 assert_eq!(info.prefix_length(), PREFIX_INFO_PREFIX.prefix());
1613 assert_eq!(info.prefix(), &PREFIX_INFO_PREFIX.network());
1614 }
1615 options::NdpOption::RouteInformation(_) => {
1616 }
1618 o => panic!("Found unexpected option: {:?}", o),
1619 }
1620 }
1621
1622 let mut route_information_options = collected
1623 .iter()
1624 .filter_map(|o| match o {
1625 options::NdpOption::RouteInformation(info) => Some(info),
1626 _ => None,
1627 })
1628 .collect::<Vec<&RouteInformation>>();
1629 route_information_options.sort_by_key(|o| o.prefix().prefix());
1634 assert_eq!(
1635 route_information_options,
1636 [
1637 &options::RouteInformation::new(
1638 ROUTE_INFO_LOW_PREF_PREFIX,
1639 ROUTE_INFO_LOW_PREF_VALID_LIFETIME_SECONDS,
1640 ROUTE_INFO_LOW_PREF,
1641 ),
1642 &options::RouteInformation::new(
1643 ROUTE_INFO_MEDIUM_PREF_PREFIX,
1644 ROUTE_INFO_MEDIUM_PREF_VALID_LIFETIME_SECONDS,
1645 ROUTE_INFO_MEDIUM_PREF,
1646 ),
1647 &options::RouteInformation::new(
1648 ROUTE_INFO_HIGH_PREF_PREFIX,
1649 ROUTE_INFO_HIGH_PREF_VALID_LIFETIME_SECONDS,
1650 ROUTE_INFO_HIGH_PREF,
1651 )
1652 ]
1653 );
1654
1655 let option_builders = [
1656 options::NdpOptionBuilder::SourceLinkLayerAddress(&SOURCE_LINK_LAYER_ADDRESS),
1657 options::NdpOptionBuilder::PrefixInformation(options::PrefixInformation::new(
1658 PREFIX_INFO_PREFIX.prefix(),
1659 PREFIX_INFO_ON_LINK_FLAG,
1660 PREFIX_INFO_AUTONOMOUS_ADDRESS_CONFIGURATION_FLAG,
1661 PREFIX_INFO_VALID_LIFETIME_SECONDS,
1662 PREFIX_INFO_PREFERRED_LIFETIME_SECONDS,
1663 PREFIX_INFO_PREFIX.network(),
1664 )),
1665 options::NdpOptionBuilder::RouteInformation(options::RouteInformation::new(
1666 ROUTE_INFO_HIGH_PREF_PREFIX,
1667 ROUTE_INFO_HIGH_PREF_VALID_LIFETIME_SECONDS,
1668 ROUTE_INFO_HIGH_PREF,
1669 )),
1670 options::NdpOptionBuilder::RouteInformation(options::RouteInformation::new(
1671 ROUTE_INFO_MEDIUM_PREF_PREFIX,
1672 ROUTE_INFO_MEDIUM_PREF_VALID_LIFETIME_SECONDS,
1673 ROUTE_INFO_MEDIUM_PREF,
1674 )),
1675 options::NdpOptionBuilder::RouteInformation(options::RouteInformation::new(
1676 ROUTE_INFO_LOW_PREF_PREFIX,
1677 ROUTE_INFO_LOW_PREF_VALID_LIFETIME_SECONDS,
1678 ROUTE_INFO_LOW_PREF,
1679 )),
1680 ];
1681 let serialized = OptionSequenceBuilder::new(option_builders.iter())
1682 .into_serializer()
1683 .encapsulate(IcmpPacketBuilder::<Ipv6, _>::new(
1684 src_ip,
1685 dst_ip,
1686 IcmpZeroCode,
1687 *icmp.message(),
1688 ))
1689 .encapsulate(ipv6_builder)
1690 .serialize_vec_outer()
1691 .unwrap()
1692 .as_ref()
1693 .to_vec();
1694 assert_eq!(&serialized, &ADVERTISEMENT_IP_PACKET_BYTES);
1695 }
1696
1697 struct SerializeRATest {
1698 hop_limit: u8,
1699 managed_flag: bool,
1700 other_config_flag: bool,
1701 preference: RoutePreference,
1702 router_lifetime_seconds: u16,
1703 reachable_time_seconds: u32,
1704 retransmit_timer_seconds: u32,
1705 }
1706
1707 #[test_case(
1708 SerializeRATest{
1709 hop_limit: 1,
1710 managed_flag: true,
1711 other_config_flag: false,
1712 preference: RoutePreference::High,
1713 router_lifetime_seconds: 1_000,
1714 reachable_time_seconds: 1_000_000,
1715 retransmit_timer_seconds: 5,
1716 }; "test_1")]
1717 #[test_case(
1718 SerializeRATest{
1719 hop_limit: 64,
1720 managed_flag: false,
1721 other_config_flag: true,
1722 preference: RoutePreference::Low,
1723 router_lifetime_seconds: 5,
1724 reachable_time_seconds: 23425621,
1725 retransmit_timer_seconds: 13252521,
1726 }; "test_2")]
1727 fn serialize_router_advertisement(test: SerializeRATest) {
1728 let SerializeRATest {
1729 hop_limit,
1730 managed_flag,
1731 other_config_flag,
1732 preference,
1733 router_lifetime_seconds,
1734 reachable_time_seconds,
1735 retransmit_timer_seconds,
1736 } = test;
1737
1738 const SRC_IP: Ipv6Addr =
1739 Ipv6Addr::from_bytes([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16]);
1740 const DST_IP: Ipv6Addr =
1741 Ipv6Addr::from_bytes([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 17]);
1742 let serialized = packet::EmptyBuf
1743 .encapsulate(IcmpPacketBuilder::<Ipv6, _>::new(
1744 SRC_IP,
1745 DST_IP,
1746 IcmpZeroCode,
1747 RouterAdvertisement::with_prf(
1748 hop_limit,
1749 managed_flag,
1750 other_config_flag,
1751 preference,
1752 router_lifetime_seconds,
1753 reachable_time_seconds,
1754 retransmit_timer_seconds,
1755 ),
1756 ))
1757 .serialize_vec_outer()
1758 .unwrap();
1759
1760 const RA_LEN: u32 = 16;
1782 let mut expected = [0; RA_LEN as usize];
1783 expected[0] = 134;
1784 expected[4] = hop_limit;
1785 if managed_flag {
1786 expected[5] |= 1 << 7;
1787 }
1788 if other_config_flag {
1789 expected[5] |= 1 << 6;
1790 }
1791 expected[5] |= u8::from(preference) << 3;
1792 let (mut router_lifetime, _rest) = Ref::<_, U16>::from_prefix(&mut expected[6..]).unwrap();
1793 router_lifetime.set(router_lifetime_seconds);
1794 let (mut reachable_time, _rest) = Ref::<_, U32>::from_prefix(&mut expected[8..]).unwrap();
1795 reachable_time.set(reachable_time_seconds);
1796 let (mut retransmit_timer, _rest) =
1797 Ref::<_, U32>::from_prefix(&mut expected[12..]).unwrap();
1798 retransmit_timer.set(retransmit_timer_seconds);
1799
1800 let mut c = internet_checksum::Checksum::new();
1801 c.add_bytes(SRC_IP.bytes());
1803 c.add_bytes(DST_IP.bytes());
1804 c.add_bytes(U32::new(RA_LEN).as_bytes());
1805 c.add_bytes(&[0, crate::ip::Ipv6Proto::Icmpv6.into()]);
1806 c.add_bytes(&expected[..]);
1808 expected[2..4].copy_from_slice(&c.checksum()[..]);
1809
1810 assert_eq!(serialized.as_ref(), &expected[..]);
1811 }
1812
1813 struct SerializeRioTest {
1814 prefix_length: u8,
1815 route_lifetime_seconds: u32,
1816 preference: RoutePreference,
1817 expected_option_length: u8,
1818 }
1819
1820 #[test_case(
1830 SerializeRioTest{
1831 prefix_length: 0,
1832 route_lifetime_seconds: 1,
1833 preference: RoutePreference::High,
1834 expected_option_length: 8,
1835 }; "prefix_length_0")]
1836 #[test_case(
1837 SerializeRioTest{
1838 prefix_length: 1,
1839 route_lifetime_seconds: 1000,
1840 preference: RoutePreference::Medium,
1841 expected_option_length: 16,
1842 }; "prefix_length_1")]
1843 #[test_case(
1844 SerializeRioTest{
1845 prefix_length: 64,
1846 route_lifetime_seconds: 100000,
1847 preference: RoutePreference::Low,
1848 expected_option_length: 16,
1849 }; "prefix_length_64")]
1850 #[test_case(
1851 SerializeRioTest{
1852 prefix_length: 65,
1853 route_lifetime_seconds: 1000000,
1854 preference: RoutePreference::Medium,
1855 expected_option_length: 24,
1856 }; "prefix_length_65")]
1857 #[test_case(
1858 SerializeRioTest{
1859 prefix_length: 128,
1860 route_lifetime_seconds: 10000000,
1861 preference: RoutePreference::Medium,
1862 expected_option_length: 24,
1863 }; "prefix_length_128")]
1864 fn serialize_route_information_option(test: SerializeRioTest) {
1865 const IPV6ADDR: Ipv6Addr =
1866 Ipv6Addr::new([0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff]);
1867
1868 let SerializeRioTest {
1869 prefix_length,
1870 route_lifetime_seconds,
1871 preference,
1872 expected_option_length,
1873 } = test;
1874 let prefix = IPV6ADDR.mask(prefix_length);
1875
1876 let option_builders =
1877 [options::NdpOptionBuilder::RouteInformation(options::RouteInformation::new(
1878 Subnet::new(prefix, prefix_length).unwrap(),
1879 route_lifetime_seconds,
1880 preference,
1881 ))];
1882
1883 let serialized = OptionSequenceBuilder::new(option_builders.iter())
1884 .into_serializer()
1885 .serialize_vec_outer()
1886 .unwrap();
1887
1888 let mut expected = [0; 24];
1916 expected[0] = 24;
1917 expected[1] = expected_option_length / 8;
1918 expected[2] = prefix_length;
1919 expected[3] = u8::from(preference) << 3;
1920 let (mut lifetime_seconds, _rest) = Ref::<_, U32>::from_prefix(&mut expected[4..]).unwrap();
1921 lifetime_seconds.set(route_lifetime_seconds);
1922 expected[8..].copy_from_slice(prefix.bytes());
1923
1924 assert_eq!(serialized.as_ref(), &expected[..expected_option_length.into()]);
1925 }
1926
1927 #[test_case(0, None)]
1928 #[test_case(
1929 1,
1930 Some(NonZeroNdpLifetime::Finite(NonZeroDuration::new(
1931 Duration::from_secs(1),
1932 ).unwrap()))
1933 )]
1934 #[test_case(
1935 u32::MAX - 1,
1936 Some(NonZeroNdpLifetime::Finite(NonZeroDuration::new(
1937 Duration::from_secs(u64::from(u32::MAX) - 1),
1938 ).unwrap()))
1939 )]
1940 #[test_case(u32::MAX, Some(NonZeroNdpLifetime::Infinite))]
1941 fn non_zero_ndp_lifetime_non_zero_or_max_u32_from_u32_with_infinite(
1942 t: u32,
1943 expected: Option<NonZeroNdpLifetime>,
1944 ) {
1945 assert_eq!(NonZeroNdpLifetime::from_u32_with_infinite(t), expected)
1946 }
1947
1948 const MIN_NON_ZERO_DURATION: Duration = Duration::new(0, 1);
1949 #[test_case(
1950 NonZeroNdpLifetime::Infinite,
1951 NonZeroDuration::new(MIN_NON_ZERO_DURATION).unwrap(),
1952 NonZeroDuration::new(MIN_NON_ZERO_DURATION).unwrap()
1953 )]
1954 #[test_case(
1955 NonZeroNdpLifetime::Infinite,
1956 NonZeroDuration::new(Duration::MAX).unwrap(),
1957 NonZeroDuration::new(Duration::MAX).unwrap()
1958 )]
1959 #[test_case(
1960 NonZeroNdpLifetime::Finite(NonZeroDuration::new(
1961 Duration::from_secs(2)).unwrap()
1962 ),
1963 NonZeroDuration::new(Duration::from_secs(1)).unwrap(),
1964 NonZeroDuration::new(Duration::from_secs(1)).unwrap()
1965 )]
1966 #[test_case(
1967 NonZeroNdpLifetime::Finite(NonZeroDuration::new(
1968 Duration::from_secs(3)).unwrap()
1969 ),
1970 NonZeroDuration::new(Duration::from_secs(4)).unwrap(),
1971 NonZeroDuration::new(Duration::from_secs(3)).unwrap()
1972 )]
1973 fn non_zero_ndp_lifetime_min_finite_duration(
1974 lifetime: NonZeroNdpLifetime,
1975 duration: NonZeroDuration,
1976 expected: NonZeroDuration,
1977 ) {
1978 assert_eq!(lifetime.min_finite_duration(duration), expected)
1979 }
1980}