From f94a885f94819c12a2ac0a1696082e481c65461f Mon Sep 17 00:00:00 2001 From: Fuwn Date: Mon, 3 Jan 2022 13:11:07 -0800 Subject: fix(windows-kernel-rs): mark unsafe functions --- crates/windows-kernel-rs/src/device.rs | 4 +++- crates/windows-kernel-rs/src/mdl.rs | 4 +++- 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 { + /// # Safety + /// `unsafe` + pub unsafe fn new(addr: *mut core::ffi::c_void, size: usize) -> Result { use windows_kernel_sys::ntoskrnl::IoAllocateMdl; let raw = unsafe { -- cgit v1.2.3