fix(windows-kernel-rs): mark unsafe functions

2 files changed, 6 insertions, 2 deletions

diff --git a/crates/windows-kernel-rs/src/device.rs b/crates/windows-kernel-rs/src/device.rs
index 4dabfba..6bebc36 100644
--- a/crates/windows-kernel-rs/src/device.rs
+++ b/crates/windows-kernel-rs/src/device.rs
@@ -434,7 +434,9 @@ pub struct DeviceExtension {
   pub(crate) device_type: DeviceType,
 }
 
-pub extern "C" fn dispatch_device(device: *mut DEVICE_OBJECT, irp: *mut IRP) -> NTSTATUS {
+/// # Safety
+/// `unsafe`
+pub unsafe extern "C" fn dispatch_device(device: *mut DEVICE_OBJECT, irp: *mut IRP) -> NTSTATUS {
   let stack_location = unsafe { &*IoGetCurrentIrpStackLocation(irp) };
   let device = unsafe { Device::from_raw(device) };
   let vtable = device.vtable();
diff --git a/crates/windows-kernel-rs/src/mdl.rs b/crates/windows-kernel-rs/src/mdl.rs
index 7d002f0..df92865 100644
--- a/crates/windows-kernel-rs/src/mdl.rs
+++ b/crates/windows-kernel-rs/src/mdl.rs
@@ -15,7 +15,9 @@ unsafe impl Send for MemoryDescriptorList {}
 unsafe impl Sync for MemoryDescriptorList {}
 
 impl MemoryDescriptorList {
-  pub fn new(addr: *mut core::ffi::c_void, size: usize) -> Result<Self, Error> {
+  /// # Safety
+  /// `unsafe`
+  pub unsafe fn new(addr: *mut core::ffi::c_void, size: usize) -> Result<Self, Error> {
     use windows_kernel_sys::ntoskrnl::IoAllocateMdl;
 
     let raw = unsafe {