netlink_packet_route/rtnl/nsid/
nlas.rs

1// SPDX-License-Identifier: MIT
2
3use anyhow::Context;
4use byteorder::{ByteOrder, NativeEndian};
5
6use crate::constants::*;
7
8use netlink_packet_utils::{
9    nla::{self, DefaultNla, NlaBuffer},
10    parsers::{parse_i32, parse_u32},
11    traits::Parseable,
12    DecodeError,
13};
14
15#[derive(Debug, PartialEq, Eq, Clone)]
16#[non_exhaustive]
17pub enum Nla {
18    Unspec(Vec<u8>),
19    Id(i32),
20    Pid(u32),
21    Fd(u32),
22    Other(DefaultNla),
23}
24
25impl nla::Nla for Nla {
26    fn value_len(&self) -> usize {
27        use self::Nla::*;
28        match *self {
29            Unspec(ref bytes) => bytes.len(),
30            Id(_) | Pid(_) | Fd(_) => 4,
31            Other(ref attr) => attr.value_len(),
32        }
33    }
34
35    fn emit_value(&self, buffer: &mut [u8]) {
36        use self::Nla::*;
37        match *self {
38            Unspec(ref bytes) => buffer.copy_from_slice(bytes.as_slice()),
39            Fd(ref value) | Pid(ref value) => {
40                NativeEndian::write_u32(buffer, *value)
41            }
42            Id(ref value) => NativeEndian::write_i32(buffer, *value),
43            Other(ref attr) => attr.emit_value(buffer),
44        }
45    }
46
47    fn kind(&self) -> u16 {
48        use self::Nla::*;
49        match *self {
50            Unspec(_) => NETNSA_NONE,
51            Id(_) => NETNSA_NSID,
52            Pid(_) => NETNSA_PID,
53            Fd(_) => NETNSA_FD,
54            Other(ref attr) => attr.kind(),
55        }
56    }
57}
58
59impl<'a, T: AsRef<[u8]> + ?Sized> Parseable<NlaBuffer<&'a T>> for Nla {
60    fn parse(buf: &NlaBuffer<&'a T>) -> Result<Self, DecodeError> {
61        use self::Nla::*;
62        let payload = buf.value();
63        Ok(match buf.kind() {
64            NETNSA_NONE => Unspec(payload.to_vec()),
65            NETNSA_NSID => {
66                Id(parse_i32(payload).context("invalid NETNSA_NSID")?)
67            }
68            NETNSA_PID => {
69                Pid(parse_u32(payload).context("invalid NETNSA_PID")?)
70            }
71            NETNSA_FD => Fd(parse_u32(payload).context("invalid NETNSA_FD")?),
72            kind => Other(
73                DefaultNla::parse(buf)
74                    .context(format!("unknown NLA type {kind}"))?,
75            ),
76        })
77    }
78}