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
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

use core::ops::{Deref, DerefMut};

use wdk::println;
use wdk_sys::{
    ntddk::{ObReferenceObjectByHandle, ObfDereferenceObject},
    PsThreadType, NT_SUCCESS, PETHREAD, PHANDLE, THREAD_ALL_ACCESS,
    _MODE::KernelMode,
};

pub struct Ethread {
    raw: PETHREAD,
}

impl Ethread {
    pub fn from_raw(raw: PETHREAD) -> Self {
        Self { raw }
    }

    /// # Safety
    ///
    /// function will dereference phandle
    pub unsafe fn from_handle(h: PHANDLE) -> Self {
        let mut r = Self {
            raw: core::ptr::null_mut(),
        };

        unsafe {
            let status = ObReferenceObjectByHandle(
                *h as _,
                THREAD_ALL_ACCESS,
                *PsThreadType,
                KernelMode as _,
                r.as_mut_raw_ptr() as _,
                core::ptr::null_mut(),
            );

            if !NT_SUCCESS(status) {
                println!("ObReferenceObjectByHandle thread error:{}", status);
            }
        }

        r
    }

    pub fn as_mut_raw_ptr(&mut self) -> *mut PETHREAD {
        &mut self.raw
    }

    pub fn as_mut_raw(&mut self) -> PETHREAD {
        self.raw
    }
}

impl Default for Ethread {
    fn default() -> Self {
        Self {
            raw: core::ptr::null_mut(),
        }
    }
}

impl Deref for Ethread {
    type Target = PETHREAD;

    fn deref(&self) -> &Self::Target {
        &self.raw
    }
}

impl DerefMut for Ethread {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.raw
    }
}

impl Drop for Ethread {
    fn drop(&mut self) {
        if !self.raw.is_null() {
            unsafe {
                ObfDereferenceObject(*self.as_mut_raw_ptr() as _);
            }
        }
    }
}