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
36
37
38
39

use wdk_sys::{
    ntddk::{KeAcquireSpinLockRaiseToDpc, KeGetCurrentIrql, KeReleaseSpinLock},
    KIRQL, KSPIN_LOCK,
};

pub struct SpinLock {
    raw: KSPIN_LOCK,
    old_irql: KIRQL,
}

pub struct SpinLockGuard<'a> {
    spinlock: &'a SpinLock,
}

impl SpinLock {
    pub fn new() -> Self {
        let raw = KSPIN_LOCK::default();
        Self {
            raw,
            old_irql: unsafe { KeGetCurrentIrql() },
        }
    }

    pub fn lock(&mut self) -> SpinLockGuard {
        self.old_irql = unsafe { KeAcquireSpinLockRaiseToDpc(&self.raw as *const _ as *mut _) };
        SpinLockGuard { spinlock: self }
    }
}

impl Drop for SpinLockGuard<'_> {
    fn drop(&mut self) {
        unsafe {
            KeReleaseSpinLock(
                &self.spinlock.raw as *const _ as *mut _,
                self.spinlock.old_irql,
            )
        };
    }
}