use core::fmt; use crate::{Error, Result}; use crate::wire::IpProtocol as Protocol; use crate::wire::Ipv6Address as Address; enum_with_unknown! { /// IPv6 Extension Routing Header Routing Type pub doc enum Type(u8) { /// Source Route (DEPRECATED) /// /// See https://tools.ietf.org/html/rfc5095 for details. Type0 = 0, /// Nimrod (DEPRECATED 2009-05-06) Nimrod = 1, /// Type 2 Routing Header for Mobile IPv6 /// /// See https://tools.ietf.org/html/rfc6275#section-6.4 for details. Type2 = 2, /// RPL Source Routing Header /// /// See https://tools.ietf.org/html/rfc6554 for details. Rpl = 3, /// RFC3692-style Experiment 1 /// /// See https://tools.ietf.org/html/rfc4727 for details. Experiment1 = 253, /// RFC3692-style Experiment 2 /// /// See https://tools.ietf.org/html/rfc4727 for details. Experiment2 = 254, /// Reserved for future use Reserved = 252 } } impl fmt::Display for Type { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { match *self { Type::Type0 => write!(f, "Type0"), Type::Nimrod => write!(f, "Nimrod"), Type::Type2 => write!(f, "Type2"), Type::Rpl => write!(f, "Rpl"), Type::Experiment1 => write!(f, "Experiment1"), Type::Experiment2 => write!(f, "Experiment2"), Type::Reserved => write!(f, "Reserved"), Type::Unknown(id) => write!(f, "{}", id) } } } /// A read/write wrapper around an IPv6 Routing Header buffer. #[derive(Debug, PartialEq)] pub struct Header> { buffer: T } // Format of the Routing Header // // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | Next Header | Hdr Ext Len | Routing Type | Segments Left | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | | // . . // . type-specific data . // . . // | | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // // // See https://tools.ietf.org/html/rfc8200#section-4.4 for details. mod field { #![allow(non_snake_case)] use crate::wire::field::*; // Minimum size of the header. pub const MIN_HEADER_SIZE: usize = 4; // 8-bit identifier of the header immediately following this header. pub const NXT_HDR: usize = 0; // 8-bit unsigned integer. Length of the DATA field in 8-octet units, // not including the first 8 octets. pub const LENGTH: usize = 1; // 8-bit identifier of a particular Routing header variant. pub const TYPE: usize = 2; // 8-bit unsigned integer. The number of route segments remaining. pub const SEG_LEFT: usize = 3; // Variable-length field. Routing-Type-specific data. // // Length of the header is in 8-octet units, not including the first 8 octets. The first four // octets are the next header type, the header length, routing type and segments left. pub fn DATA(length_field: u8) -> Field { let bytes = length_field * 8 + 8; 4..bytes as usize } // The Type 2 Routing Header has the following format: // // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | Next Header | Hdr Ext Len=2 | Routing Type=2|Segments Left=1| // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | Reserved | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | | // + + // | | // + Home Address + // | | // + + // | | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // 16-byte field containing the home address of the destination mobile node. pub const HOME_ADDRESS: Field = 8..24; // The RPL Source Routing Header has the following format: // // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | Next Header | Hdr Ext Len | Routing Type | Segments Left | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | CmprI | CmprE | Pad | Reserved | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // | | // . . // . Addresses[1..n] . // . . // | | // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ // 8-bit field containing the CmprI and CmprE values. pub const CMPR: usize = 4; // 8-bit field containing the Pad value. pub const PAD: usize = 5; // Variable length field containing addresses pub fn ADDRESSES(length_field: u8) -> Field { let data = DATA(length_field); 8..data.end } } /// Core getter methods relevant to any routing type. impl> Header { /// Create a raw octet buffer with an IPv6 Routing Header structure. pub fn new(buffer: T) -> Header { Header { buffer } } /// Shorthand for a combination of [new_unchecked] and [check_len]. /// /// [new_unchecked]: #method.new_unchecked /// [check_len]: #method.check_len pub fn new_checked(buffer: T) -> Result> { let header = Self::new(buffer); header.check_len()?; Ok(header) } /// Ensure that no accessor method will panic if called. /// Returns `Err(Error::Truncated)` if the buffer is too short. /// /// The result of this check is invalidated by calling [set_header_len]. /// /// [set_header_len]: #method.set_header_len pub fn check_len(&self) -> Result<()> { let len = self.buffer.as_ref().len(); if len < field::MIN_HEADER_SIZE { return Err(Error::Truncated); } if len < field::DATA(self.header_len()).end as usize { return Err(Error::Truncated); } Ok(()) } /// Consume the header, returning the underlying buffer. pub fn into_inner(self) -> T { self.buffer } /// Return the next header field. #[inline] pub fn next_header(&self) -> Protocol { let data = self.buffer.as_ref(); Protocol::from(data[field::NXT_HDR]) } /// Return the header length field. Length of the Routing header in 8-octet units, /// not including the first 8 octets. #[inline] pub fn header_len(&self) -> u8 { let data = self.buffer.as_ref(); data[field::LENGTH] } /// Return the routing type field. #[inline] pub fn routing_type(&self) -> Type { let data = self.buffer.as_ref(); Type::from(data[field::TYPE]) } /// Return the segments left field. #[inline] pub fn segments_left(&self) -> u8 { let data = self.buffer.as_ref(); data[field::SEG_LEFT] } } /// Getter methods for the Type 2 Routing Header routing type. impl> Header { /// Return the IPv6 Home Address /// /// # Panics /// This function may panic if this header is not the Type2 Routing Header routing type. pub fn home_address(&self) -> Address { let data = self.buffer.as_ref(); Address::from_bytes(&data[field::HOME_ADDRESS]) } } /// Getter methods for the RPL Source Routing Header routing type. impl> Header { /// Return the number of prefix octects elided from addresses[1..n-1]. /// /// # Panics /// This function may panic if this header is not the RPL Source Routing Header routing type. pub fn cmpr_i(&self) -> u8 { let data = self.buffer.as_ref(); data[field::CMPR] >> 4 } /// Return the number of prefix octects elided from the last address (`addresses[n]`). /// /// # Panics /// This function may panic if this header is not the RPL Source Routing Header routing type. pub fn cmpr_e(&self) -> u8 { let data = self.buffer.as_ref(); data[field::CMPR] & 0xf } /// Return the number of octects used for padding after `addresses[n]`. /// /// # Panics /// This function may panic if this header is not the RPL Source Routing Header routing type. pub fn pad(&self) -> u8 { let data = self.buffer.as_ref(); data[field::PAD] >> 4 } /// Return the address vector in bytes /// /// # Panics /// This function may panic if this header is not the RPL Source Routing Header routing type. pub fn addresses(&self) -> &[u8] { let data = self.buffer.as_ref(); &data[field::ADDRESSES(data[field::LENGTH])] } } /// Core setter methods relevant to any routing type. impl + AsMut<[u8]>> Header { /// Set the next header field. #[inline] pub fn set_next_header(&mut self, value: Protocol) { let data = self.buffer.as_mut(); data[field::NXT_HDR] = value.into(); } /// Set the option data length. Length of the Routing header in 8-octet units. #[inline] pub fn set_header_len(&mut self, value: u8) { let data = self.buffer.as_mut(); data[field::LENGTH] = value; } /// Set the routing type. #[inline] pub fn set_routing_type(&mut self, value: Type) { let data = self.buffer.as_mut(); data[field::TYPE] = value.into(); } /// Set the segments left field. #[inline] pub fn set_segments_left(&mut self, value: u8) { let data = self.buffer.as_mut(); data[field::SEG_LEFT] = value; } /// Initialize reserved fields to 0. /// /// # Panics /// This function may panic if the routing type is not set. #[inline] pub fn clear_reserved(&mut self) { let routing_type = self.routing_type(); let data = self.buffer.as_mut(); match routing_type { Type::Type2 => { data[4] = 0; data[5] = 0; data[6] = 0; data[7] = 0; } Type::Rpl => { // Retain the higher order 4 bits of the padding field data[field::PAD] = data[field::PAD] & 0xF0; data[6] = 0; data[7] = 0; } _ => panic!("Unrecognized routing type when clearing reserved fields.") } } } /// Setter methods for the RPL Source Routing Header routing type. impl + AsMut<[u8]>> Header { /// Set the Ipv6 Home Address /// /// # Panics /// This function may panic if this header is not the Type 2 Routing Header routing type. pub fn set_home_address(&mut self, value: Address) { let data = self.buffer.as_mut(); data[field::HOME_ADDRESS].copy_from_slice(value.as_bytes()); } } /// Setter methods for the RPL Source Routing Header routing type. impl + AsMut<[u8]>> Header { /// Set the number of prefix octects elided from addresses[1..n-1]. /// /// # Panics /// This function may panic if this header is not the RPL Source Routing Header routing type. pub fn set_cmpr_i(&mut self, value: u8) { let data = self.buffer.as_mut(); let raw = (value << 4) | (data[field::CMPR] & 0xF); data[field::CMPR] = raw; } /// Set the number of prefix octects elided from the last address (`addresses[n]`). /// /// # Panics /// This function may panic if this header is not the RPL Source Routing Header routing type. pub fn set_cmpr_e(&mut self, value: u8) { let data = self.buffer.as_mut(); let raw = (value & 0xF) | (data[field::CMPR] & 0xF0); data[field::CMPR] = raw; } /// Set the number of octects used for padding after `addresses[n]`. /// /// # Panics /// This function may panic if this header is not the RPL Source Routing Header routing type. pub fn set_pad(&mut self, value: u8) { let data = self.buffer.as_mut(); data[field::PAD] = value << 4; } /// Set address data /// /// # Panics /// This function may panic if this header is not the RPL Source Routing Header routing type. pub fn set_addresses(&mut self, value: &[u8]) { let data = self.buffer.as_mut(); let len = data[field::LENGTH]; let addresses = &mut data[field::ADDRESSES(len)]; addresses.copy_from_slice(value); } } impl<'a, T: AsRef<[u8]> + ?Sized> fmt::Display for Header<&'a T> { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { match Repr::parse(self) { Ok(repr) => write!(f, "{}", repr), Err(err) => { write!(f, "IPv6 Routing ({})", err)?; Ok(()) } } } } /// A high-level representation of an IPv6 Routing Header. #[derive(Debug, PartialEq, Eq, Clone, Copy)] pub enum Repr<'a> { Type2 { /// The type of header immediately following the Routing header. next_header: Protocol, /// Length of the Routing header in 8-octet units, not including the first 8 octets. length: u8, /// Number of route segments remaining. segments_left: u8, /// The home address of the destination mobile node. home_address: Address, }, Rpl { /// The type of header immediately following the Routing header. next_header: Protocol, /// Length of the Routing header in 8-octet units, not including the first 8 octets. length: u8, /// Number of route segments remaining. segments_left: u8, /// Number of prefix octets from each segment, except the last segment, that are elided. cmpr_i: u8, /// Number of prefix octets from the last segment that are elided. cmpr_e: u8, /// Number of octets that are used for padding after `address[n]` at the end of the /// RPL Source Route Header. pad: u8, /// Vector of addresses, numbered 1 to `n`. addresses: &'a[u8], }, #[doc(hidden)] __Nonexhaustive } impl<'a> Repr<'a> { /// Parse an IPv6 Routing Header and return a high-level representation. pub fn parse(header: &'a Header<&'a T>) -> Result> where T: AsRef<[u8]> + ?Sized { match header.routing_type() { Type::Type2 => { Ok(Repr::Type2 { next_header: header.next_header(), length: header.header_len(), segments_left: header.segments_left(), home_address: header.home_address(), }) } Type::Rpl => { Ok(Repr::Rpl { next_header: header.next_header(), length: header.header_len(), segments_left: header.segments_left(), cmpr_i: header.cmpr_i(), cmpr_e: header.cmpr_e(), pad: header.pad(), addresses: header.addresses(), }) } _ => Err(Error::Unrecognized) } } /// Return the length, in bytes, of a header that will be emitted from this high-level /// representation. pub fn buffer_len(&self) -> usize { match self { &Repr::Rpl { length, .. } | &Repr::Type2 { length, .. } => { field::DATA(length).end } &Repr::__Nonexhaustive => unreachable!() } } /// Emit a high-level representation into an IPv6 Routing Header. pub fn emit + AsMut<[u8]> + ?Sized>(&self, header: &mut Header<&mut T>) { match *self { Repr::Type2 { next_header, length, segments_left, home_address } => { header.set_next_header(next_header); header.set_header_len(length); header.set_routing_type(Type::Type2); header.set_segments_left(segments_left); header.clear_reserved(); header.set_home_address(home_address); } Repr::Rpl { next_header, length, segments_left, cmpr_i, cmpr_e, pad, addresses } => { header.set_next_header(next_header); header.set_header_len(length); header.set_routing_type(Type::Rpl); header.set_segments_left(segments_left); header.set_cmpr_i(cmpr_i); header.set_cmpr_e(cmpr_e); header.set_pad(pad); header.clear_reserved(); header.set_addresses(addresses); } Repr::__Nonexhaustive => unreachable!(), } } } impl<'a> fmt::Display for Repr<'a> { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { match *self { Repr::Type2 { next_header, length, segments_left, home_address } => { write!(f, "IPv6 Routing next_hdr={} length={} type={} seg_left={} home_address={}", next_header, length, Type::Type2, segments_left, home_address) } Repr::Rpl { next_header, length, segments_left, cmpr_i, cmpr_e, pad, .. } => { write!(f, "IPv6 Routing next_hdr={} length={} type={} seg_left={} cmpr_i={} cmpr_e={} pad={}", next_header, length, Type::Rpl, segments_left, cmpr_i, cmpr_e, pad) } Repr::__Nonexhaustive => unreachable!(), } } } #[cfg(test)] mod test { use super::*; // A Type 2 Routing Header static BYTES_TYPE2: [u8; 24] = [0x6, 0x2, 0x2, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1]; // A representation of a Type 2 Routing header static REPR_TYPE2: Repr = Repr::Type2 { next_header: Protocol::Tcp, length: 2, segments_left: 1, home_address: Address::LOOPBACK, }; // A Source Routing Header with full IPv6 addresses in bytes static BYTES_SRH_FULL: [u8; 40] = [0x6, 0x4, 0x3, 0x2, 0x0, 0x0, 0x0, 0x0, 0xfd, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x2, 0xfd, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x3, 0x1]; // A representation of a Source Routing Header with full IPv6 addresses static REPR_SRH_FULL: Repr = Repr::Rpl { next_header: Protocol::Tcp, length: 4, segments_left: 2, cmpr_i: 0, cmpr_e: 0, pad: 0, addresses: &[0xfd, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x2, 0xfd, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x3, 0x1] }; // A Source Routing Header with elided IPv6 addresses in bytes static BYTES_SRH_ELIDED: [u8; 16] = [0x6, 0x1, 0x3, 0x2, 0xfe, 0x50, 0x0, 0x0, 0x2, 0x3, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0]; // A representation of a Source Routing Header with elided IPv6 addresses static REPR_SRH_ELIDED: Repr = Repr::Rpl { next_header: Protocol::Tcp, length: 1, segments_left: 2, cmpr_i: 15, cmpr_e: 14, pad: 5, addresses: &[0x2, 0x3, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0] }; #[test] fn test_check_len() { // less than min header size assert_eq!(Err(Error::Truncated), Header::new(&BYTES_TYPE2[..3]).check_len()); assert_eq!(Err(Error::Truncated), Header::new(&BYTES_SRH_FULL[..3]).check_len()); assert_eq!(Err(Error::Truncated), Header::new(&BYTES_SRH_ELIDED[..3]).check_len()); // less than specfied length field assert_eq!(Err(Error::Truncated), Header::new(&BYTES_TYPE2[..23]).check_len()); assert_eq!(Err(Error::Truncated), Header::new(&BYTES_SRH_FULL[..39]).check_len()); assert_eq!(Err(Error::Truncated), Header::new(&BYTES_SRH_ELIDED[..11]).check_len()); // valid assert_eq!(Ok(()), Header::new(&BYTES_TYPE2[..]).check_len()); assert_eq!(Ok(()), Header::new(&BYTES_SRH_FULL[..]).check_len()); assert_eq!(Ok(()), Header::new(&BYTES_SRH_ELIDED[..]).check_len()); } #[test] fn test_header_deconstruct() { let header = Header::new(&BYTES_TYPE2[..]); assert_eq!(header.next_header(), Protocol::Tcp); assert_eq!(header.header_len(), 2); assert_eq!(header.routing_type(), Type::Type2); assert_eq!(header.segments_left(), 1); assert_eq!(header.home_address(), Address::LOOPBACK); let header = Header::new(&BYTES_SRH_FULL[..]); assert_eq!(header.next_header(), Protocol::Tcp); assert_eq!(header.header_len(), 4); assert_eq!(header.routing_type(), Type::Rpl); assert_eq!(header.segments_left(), 2); assert_eq!(header.addresses(), &BYTES_SRH_FULL[8..]); let header = Header::new(&BYTES_SRH_ELIDED[..]); assert_eq!(header.next_header(), Protocol::Tcp); assert_eq!(header.header_len(), 1); assert_eq!(header.routing_type(), Type::Rpl); assert_eq!(header.segments_left(), 2); assert_eq!(header.addresses(), &BYTES_SRH_ELIDED[8..]); } #[test] fn test_repr_parse_valid() { let header = Header::new_checked(&BYTES_TYPE2[..]).unwrap(); let repr = Repr::parse(&header).unwrap(); assert_eq!(repr, REPR_TYPE2); let header = Header::new_checked(&BYTES_SRH_FULL[..]).unwrap(); let repr = Repr::parse(&header).unwrap(); assert_eq!(repr, REPR_SRH_FULL); let header = Header::new_checked(&BYTES_SRH_ELIDED[..]).unwrap(); let repr = Repr::parse(&header).unwrap(); assert_eq!(repr, REPR_SRH_ELIDED); } #[test] fn test_repr_emit() { let mut bytes = [0u8; 24]; let mut header = Header::new(&mut bytes[..]); REPR_TYPE2.emit(&mut header); assert_eq!(header.into_inner(), &BYTES_TYPE2[..]); let mut bytes = [0u8; 40]; let mut header = Header::new(&mut bytes[..]); REPR_SRH_FULL.emit(&mut header); assert_eq!(header.into_inner(), &BYTES_SRH_FULL[..]); let mut bytes = [0u8; 16]; let mut header = Header::new(&mut bytes[..]); REPR_SRH_ELIDED.emit(&mut header); assert_eq!(header.into_inner(), &BYTES_SRH_ELIDED[..]); } #[test] fn test_buffer_len() { assert_eq!(REPR_TYPE2.buffer_len(), 24); assert_eq!(REPR_SRH_FULL.buffer_len(), 40); assert_eq!(REPR_SRH_ELIDED.buffer_len(), 16); } }