moon-driver-core 0.1.3

use core::ffi::c_void;

use alloc::boxed::Box;
use moon_driver_utils::string::{string_to_u16_slice, u16_slice_to_unicode_string};
use wdk_sys::{ntddk::IoCreateDevice, DRIVER_OBJECT, NT_SUCCESS};

use crate::device::{Device, DeviceExtension, DeviceOperations, DeviceOperationsVtable};

pub struct Driver {
    pub raw: *mut DRIVER_OBJECT,
}

impl Driver {
    pub fn create_device<T: DeviceOperations>(
        &mut self,
        name: &str,
        device_type: u32,
        device_characteristics: u32,
        exclusive: u8,
        data: T,
    ) -> Result<Device, &'static str> {
        let data = Box::new(data);
        let name = string_to_u16_slice(name);
        let mut name_ptr = u16_slice_to_unicode_string(name.as_slice());
        let mut device = core::ptr::null_mut();
        let status = unsafe {
            IoCreateDevice(
                self.raw,
                core::mem::size_of::<DeviceExtension>() as u32,
                &mut name_ptr,
                device_type,
                device_characteristics,
                exclusive,
                &mut device,
            )
        };
        if !NT_SUCCESS(status) {
            return Err("CreateDevice error");
        }
        let mut device = Device::from_raw(device);
        let extension = device.extension_mut();
        extension.vtable = &DeviceOperationsVtable::<T>::VTABLE;
        extension.data = Box::into_raw(data) as *mut c_void;
        Ok(device)
    }
    pub fn from_raw(raw: *mut DRIVER_OBJECT) -> Self {
        Self { raw }
    }
    pub fn as_raw(&self) -> *const DRIVER_OBJECT {
        self.raw as *const _
    }
    pub fn as_raw_mut(&self) -> *mut DRIVER_OBJECT {
        self.raw
    }
    pub fn into_raw(mut self) -> *mut DRIVER_OBJECT {
        core::mem::replace(&mut self.raw, core::ptr::null_mut())
    }
}