netlink_packet_route/address/
message.rs

1// SPDX-License-Identifier: MIT
2
3use crate::address::{AddressAttribute, AddressError, AddressHeaderFlags, AddressScope};
4use crate::AddressFamily;
5use netlink_packet_utils::nla::{NlaBuffer, NlaError, NlasIterator};
6use netlink_packet_utils::traits::{Emitable, Parseable};
7use netlink_packet_utils::DecodeError;
8
9const ADDRESS_HEADER_LEN: usize = 8;
10
11buffer!(AddressMessageBuffer(ADDRESS_HEADER_LEN) {
12    family: (u8, 0),
13    prefix_len: (u8, 1),
14    flags: (u8, 2),
15    scope: (u8, 3),
16    index: (u32, 4..ADDRESS_HEADER_LEN),
17    payload: (slice, ADDRESS_HEADER_LEN..),
18});
19
20impl<'a, T: AsRef<[u8]> + ?Sized> AddressMessageBuffer<&'a T> {
21    pub fn attributes(&self) -> impl Iterator<Item = Result<NlaBuffer<&'a [u8]>, NlaError>> {
22        NlasIterator::new(self.payload())
23    }
24}
25
26#[derive(Debug, PartialEq, Eq, Clone, Default)]
27#[non_exhaustive]
28pub struct AddressMessage {
29    pub header: AddressHeader,
30    pub attributes: Vec<AddressAttribute>,
31}
32
33#[derive(Debug, PartialEq, Eq, Clone, Default)]
34pub struct AddressHeader {
35    pub family: AddressFamily,
36    pub prefix_len: u8,
37    pub flags: AddressHeaderFlags,
38    pub scope: AddressScope,
39    pub index: u32,
40}
41
42impl Emitable for AddressHeader {
43    fn buffer_len(&self) -> usize {
44        ADDRESS_HEADER_LEN
45    }
46
47    fn emit(&self, buffer: &mut [u8]) {
48        let mut packet = AddressMessageBuffer::new(buffer);
49        packet.set_family(self.family.into());
50        packet.set_prefix_len(self.prefix_len);
51        packet.set_flags(self.flags.bits());
52        packet.set_scope(self.scope.into());
53        packet.set_index(self.index);
54    }
55}
56
57impl Emitable for AddressMessage {
58    fn buffer_len(&self) -> usize {
59        self.header.buffer_len() + self.attributes.as_slice().buffer_len()
60    }
61
62    fn emit(&self, buffer: &mut [u8]) {
63        self.header.emit(buffer);
64        self.attributes.as_slice().emit(&mut buffer[self.header.buffer_len()..]);
65    }
66}
67
68impl<T: AsRef<[u8]>> Parseable<AddressMessageBuffer<T>> for AddressHeader {
69    type Error = ();
70    fn parse(buf: &AddressMessageBuffer<T>) -> Result<Self, ()> {
71        Ok(Self {
72            family: buf.family().into(),
73            prefix_len: buf.prefix_len(),
74            flags: AddressHeaderFlags::from_bits_retain(buf.flags()),
75            scope: buf.scope().into(),
76            index: buf.index(),
77        })
78    }
79}
80
81impl<'a, T: AsRef<[u8]> + 'a> Parseable<AddressMessageBuffer<&'a T>> for AddressMessage {
82    type Error = AddressError;
83    fn parse(buf: &AddressMessageBuffer<&'a T>) -> Result<Self, AddressError> {
84        Ok(AddressMessage {
85            // ok to unwrap, we never fail parsing the header.
86            header: AddressHeader::parse(buf).unwrap(),
87            attributes: Vec::<AddressAttribute>::parse(buf)?,
88        })
89    }
90}
91
92impl<'a, T: AsRef<[u8]> + 'a> Parseable<AddressMessageBuffer<&'a T>> for Vec<AddressAttribute> {
93    type Error = AddressError;
94    fn parse(buf: &AddressMessageBuffer<&'a T>) -> Result<Self, AddressError> {
95        let mut attributes = vec![];
96        for nla_buf in buf.attributes() {
97            attributes.push(AddressAttribute::parse(&nla_buf?)?);
98        }
99        Ok(attributes)
100    }
101}