netlink_packet_route/route/
message.rs

1// SPDX-License-Identifier: MIT
2
3use super::super::AddressFamily;
4use super::attribute::RTA_ENCAP_TYPE;
5use super::error::RouteError;
6use super::{RouteAttribute, RouteHeader, RouteLwEnCapType, RouteMessageBuffer, RouteType};
7use netlink_packet_utils::traits::{Emitable, Parseable, ParseableParametrized};
8
9#[derive(Debug, PartialEq, Eq, Clone, Default)]
10#[non_exhaustive]
11pub struct RouteMessage {
12    pub header: RouteHeader,
13    pub attributes: Vec<RouteAttribute>,
14}
15
16impl Emitable for RouteMessage {
17    fn buffer_len(&self) -> usize {
18        self.header.buffer_len() + self.attributes.as_slice().buffer_len()
19    }
20
21    fn emit(&self, buffer: &mut [u8]) {
22        self.header.emit(buffer);
23        self.attributes.as_slice().emit(&mut buffer[self.header.buffer_len()..]);
24    }
25}
26
27impl<'a, T: AsRef<[u8]> + 'a> Parseable<RouteMessageBuffer<&'a T>> for RouteMessage {
28    type Error = RouteError;
29    fn parse(buf: &RouteMessageBuffer<&'a T>) -> Result<Self, RouteError> {
30        // unwrap: RouteHeader can't fail.
31        let header = RouteHeader::parse(buf).unwrap();
32        let address_family = header.address_family;
33        let route_type = header.kind;
34        Ok(RouteMessage {
35            header,
36            attributes: Vec::<RouteAttribute>::parse_with_param(buf, (address_family, route_type))?,
37        })
38    }
39}
40
41impl<'a, T: AsRef<[u8]> + 'a>
42    ParseableParametrized<RouteMessageBuffer<&'a T>, (AddressFamily, RouteType)>
43    for Vec<RouteAttribute>
44{
45    type Error = RouteError;
46    fn parse_with_param(
47        buf: &RouteMessageBuffer<&'a T>,
48        (address_family, route_type): (AddressFamily, RouteType),
49    ) -> Result<Self, RouteError> {
50        let mut attributes = vec![];
51        let mut encap_type = RouteLwEnCapType::None;
52        // The RTA_ENCAP_TYPE is provided __after__ RTA_ENCAP, we should find
53        // RTA_ENCAP_TYPE first.
54        for nla_buf in buf.attributes() {
55            let nla = match nla_buf {
56                Ok(n) => n,
57                Err(_) => continue,
58            };
59            if nla.kind() == RTA_ENCAP_TYPE {
60                if let Ok(RouteAttribute::EncapType(v)) =
61                    RouteAttribute::parse_with_param(&nla, (address_family, route_type, encap_type))
62                {
63                    encap_type = v;
64                    break;
65                }
66            }
67        }
68        for nla_buf in buf.attributes() {
69            attributes.push(RouteAttribute::parse_with_param(
70                &nla_buf?,
71                (address_family, route_type, encap_type),
72            )?);
73        }
74        Ok(attributes)
75    }
76}