moon-driver-utils 0.1.0

Windows Kernel Utils
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

use core::ffi::c_void;

use wdk_sys::ntddk::{ExAcquireRundownProtection, ExReleaseRundownProtection};

#[derive(Debug)]
pub struct RunDownProtectionError {}
pub struct RunDownProtection {
    raw: *mut c_void,
}

pub struct RunDownProtectionGuard<'a> {
    run_down_protection: &'a RunDownProtection,
}

impl RunDownProtection {
    pub fn from_raw(raw: *mut c_void) -> Self {
        Self { raw }
    }

    pub fn lock(&self) -> Result<RunDownProtectionGuard, RunDownProtectionError> {
        let r = unsafe { ExAcquireRundownProtection(&self.raw as *const _ as *mut _) };
        if r == 0 {
            return Err(RunDownProtectionError {});
        }
        Ok(RunDownProtectionGuard {
            run_down_protection: self,
        })
    }
}

impl Drop for RunDownProtectionGuard<'_> {
    fn drop(&mut self) {
        unsafe { ExReleaseRundownProtection(&self.run_down_protection.raw as *const _ as *mut _) };
    }
}