netlink_packet_route/rtnl/nsid/
nlas.rs

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