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

use super::super::aligned_buffer::{FromBuffer, U32AlignedBuffer};

#[derive(Debug)]
pub(crate) struct NetlinkIterator<'a>(U32AlignedBuffer<'a>);

unsafe impl FromBuffer for libc::nlmsghdr {
    fn len(&self, total: usize) -> u32 {
        // This not only works around a Linux kernel bug in the audit system,
        // it also makes the code simpler by automatically breaking out of loops
        // when a message that should terminate them is received.
        //
        // We don’t care about the audit bug, but we do care about simplcity.
        if self.nlmsg_flags & libc::NLM_F_MULTI as u16 != 0 {
            self.nlmsg_len
        } else {
            total as _
        }
    }
}

impl<'a> NetlinkIterator<'a> {
    pub(crate) fn new(buffer: &'a [u8]) -> NetlinkIterator<'a> {
        Self(U32AlignedBuffer::new(buffer))
    }
}

impl<'a> Iterator for NetlinkIterator<'a> {
    type Item = (libc::nlmsghdr, U32AlignedBuffer<'a>);

    fn next(&mut self) -> Option<Self::Item> {
        self.0.read()
    }
}