winaudit 0.1.3

Advanced Windows auditing and security assessment Crate in Rust
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

//! This module offer experimental checks for Network Adapaters
//!
//! Note:
//! You need enable feature "experimental" in Cargo.toml.
//!

use crate::WinAuditError;
use crate::hardwarechecks::windows_defs::*;

use windows::Win32::Foundation::*;
use windows::Win32::Storage::FileSystem::*;
use windows::Win32::System::IO::*;
use windows::core::PCWSTR;

use std::ptr;

/// Check is Network Adapter Device support **SRIOV**
///
/// # Example Usage:
/// ```
/// use winaudit::is_sriov_enabled;
///
/// match is_sriov_enabled("\\\\.\\PhysicalNetAdapter") {
///     Ok(enabled) => {
///         if enabled {
///             println!("SRIOV is enabled.");
///         } else {
///             println!("SRIOV is not enabled.");
///         }
///     }
///     Err(e) => {
///         eprintln!("Error: {}", e);
///     }
/// }
/// ```
#[allow(dead_code)]
pub(crate) fn is_sriov_enabled(adapter_device_path: &str) -> Result<bool, WinAuditError> {
    unsafe {
        let wide: Vec<u16> = adapter_device_path.encode_utf16().chain(Some(0)).collect();

        let handle = CreateFileW(
            PCWSTR(wide.as_ptr()),
            FILE_GENERIC_READ.0,
            FILE_SHARE_READ | FILE_SHARE_WRITE,
            Some(ptr::null()),
            OPEN_EXISTING,
            FILE_ATTRIBUTE_NORMAL,
            Some(HANDLE::default()),
        );

        if handle == Ok(INVALID_HANDLE_VALUE) {
            return Err(WinAuditError::from(
                "Failed to open network adapter device path",
            ));
        }

        let mut sriov_capable: u32 = 0;
        let mut bytes = 0u32;

        let handle = HANDLE::default();

        let result = DeviceIoControl(
            handle,
            IOCTL_NDIS_QUERY_GLOBAL_STATS,
            Some(&OID_GEN_SRIOV_CAPABLE as *const _ as *mut _),
            std::mem::size_of::<u32>() as u32,
            Some(&mut sriov_capable as *mut _ as *mut _),
            std::mem::size_of::<u32>() as u32,
            Some(&mut bytes),
            Some(ptr::null_mut()),
        );

        CloseHandle(handle)?;

        match result {
            Ok(_) => return Ok(sriov_capable != 0),
            Err(_) => {
                return Err(WinAuditError::from(
                    "DeviceIoControl failed for SRIOV capability",
                ));
            }
        }
    }
}