taskstats 0.1.1

high-level encapsulation of Linux's per-task statistics interface
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use super::raw::{genlmsghdr, GENL_HDRLEN};
use crate::netlink::Serialize;
use thiserror::Error;
use zerocopy::LayoutVerified;

#[derive(Debug)]
pub struct GenericNetlinkMessage<P> {
    pub cmd: u8,
    pub version: u8,
    pub payload: P,
}

impl<P: Serialize> Serialize for GenericNetlinkMessage<P> {
    fn len(&self) -> u32 {
        GENL_HDRLEN + self.payload.len()
    }

    fn serialize(&self, buf: &mut [u8]) {
        let header_len = GENL_HDRLEN as usize;
        let (header, payload) = buf.split_at_mut(header_len);
        let mut header = LayoutVerified::<_, genlmsghdr>::new(header).expect("invalid buffer");
        header.cmd = self.cmd;
        header.version = self.version;
        header.reserved = 0;
        self.payload.serialize(payload);
    }
}

impl<'a> GenericNetlinkMessage<&'a [u8]> {
    pub fn deserialize(buf: &'a [u8]) -> Result<Self, InvalidBuffer> {
        let header_len = GENL_HDRLEN as usize;
        if buf.len() < header_len {
            return Err(InvalidBuffer::Header(buf.len()));
        }
        let (header, payload) = buf.split_at(header_len);
        let header = LayoutVerified::<_, genlmsghdr>::new(header).expect("invalid buffer");
        Ok(Self {
            cmd: header.cmd,
            version: header.version,
            payload,
        })
    }
}

#[derive(Debug, Error)]
pub enum InvalidBuffer {
    #[error("insufficient buffer for generic netlink header, got buffer size {0}")]
    Header(usize),
}