netconfig-rs 0.1.6

Crate for managing network interfaces and IP addresses in a cross-platform way
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
50
51
52
53
54
55
56
57
58
59
60
61
62

#![allow(non_snake_case)]
use crate::Error;
use std::marker::PhantomData;
use windows::Win32::NetworkManagement::IpHelper::{
    FreeMibTable, GetIpInterfaceTable, GetUnicastIpAddressTable, MIB_IPINTERFACE_ROW,
    MIB_IPINTERFACE_TABLE, MIB_UNICASTIPADDRESS_ROW, MIB_UNICASTIPADDRESS_TABLE,
};
use windows::Win32::Networking::WinSock::ADDRESS_FAMILY;

pub struct MibTable<'a, T, R> {
    table: *mut T,
    phantom: PhantomData<&'a R>,
}

impl<T, R> Default for MibTable<'_, T, R> {
    fn default() -> Self {
        Self {
            table: std::ptr::null_mut(),
            phantom: Default::default(),
        }
    }
}

impl<'a> MibTable<'a, MIB_UNICASTIPADDRESS_TABLE, MIB_UNICASTIPADDRESS_ROW> {
    pub fn GetUnicastIpAddressTable(family: &ADDRESS_FAMILY) -> Result<Self, Error> {
        let mut result = Self::default();
        unsafe { GetUnicastIpAddressTable(*family, &mut result.table).ok()? }
        Ok(result)
    }

    pub fn as_slice(&self) -> &'a [MIB_UNICASTIPADDRESS_ROW] {
        unsafe {
            let table = self.table.as_ref().unwrap();
            std::slice::from_raw_parts(table.Table.as_ptr(), table.NumEntries as _)
        }
    }
}

impl<'a> MibTable<'a, MIB_IPINTERFACE_TABLE, MIB_IPINTERFACE_ROW> {
    pub fn GetIpInterfaceTable(family: &ADDRESS_FAMILY) -> Result<Self, Error> {
        let mut result = Self::default();
        unsafe { GetIpInterfaceTable(*family, &mut result.table).ok()? }
        Ok(result)
    }

    pub fn as_slice(&self) -> &'a [MIB_IPINTERFACE_ROW] {
        unsafe {
            let table = self.table.as_ref().unwrap();
            std::slice::from_raw_parts(table.Table.as_ptr(), table.NumEntries as _)
        }
    }
}

impl<T, R> Drop for MibTable<'_, T, R> {
    fn drop(&mut self) {
        if !self.table.is_null() {
            unsafe {
                FreeMibTable(self.table as _);
            }
        }
    }
}