feat(driver): commit primer

1 files changed, 119 insertions, 0 deletions

diff --git a/crates/windows-kernel-rs/src/lib.rs b/crates/windows-kernel-rs/src/lib.rs
new file mode 100644
index 0000000..a0ddf9a
--- /dev/null
+++ b/crates/windows-kernel-rs/src/lib.rs
@@ -0,0 +1,119 @@
+#![no_std]
+#![feature(alloc_error_handler)]
+
+extern crate alloc;
+
+pub mod affinity;
+pub mod allocator;
+pub mod device;
+pub mod driver;
+pub mod error;
+pub mod intrin;
+pub mod io;
+pub mod ioctl;
+pub mod mdl;
+pub mod memory;
+pub mod process;
+pub mod request;
+pub mod section;
+pub mod string;
+pub mod symbolic_link;
+pub mod sync;
+pub mod user_ptr;
+pub mod version;
+
+pub use widestring::U16CString;
+pub use windows_kernel_sys::base::{
+  DRIVER_OBJECT,
+  IRP_MJ_MAXIMUM_FUNCTION,
+  NTSTATUS,
+  STATUS_SUCCESS,
+  UNICODE_STRING,
+};
+
+pub use crate::{
+  affinity::{get_cpu_count, get_current_cpu_num, run_on_cpu, run_on_each_cpu},
+  device::{
+    dispatch_device,
+    Access,
+    Completion,
+    Device,
+    DeviceDoFlags,
+    DeviceFlags,
+    DeviceOperations,
+    DeviceType,
+    RequestError,
+  },
+  driver::Driver,
+  error::Error,
+  ioctl::{ControlCode, RequiredAccess, TransferMethod},
+  request::{IoControlRequest, IoRequest, ReadRequest, WriteRequest},
+  symbolic_link::SymbolicLink,
+  user_ptr::UserPtr,
+};
+
+#[cfg(feature = "alloc")]
+#[global_allocator]
+static ALLOCATOR: allocator::KernelAllocator =
+  allocator::KernelAllocator::new(u32::from_ne_bytes(*b"rust"));
+
+#[panic_handler]
+fn panic(_info: &core::panic::PanicInfo) -> ! { loop {} }
+
+#[used]
+#[no_mangle]
+pub static _fltused: i32 = 0;
+
+#[no_mangle]
+pub extern "system" fn __CxxFrameHandler3() -> i32 { 0 }
+
+#[macro_export]
+macro_rules! kernel_module {
+  ($module:ty) => {
+    static mut __MOD: Option<$module> = None;
+
+    #[no_mangle]
+    pub extern "system" fn driver_entry(
+      driver: &mut $crate::DRIVER_OBJECT,
+      registry_path: &$crate::UNICODE_STRING,
+    ) -> $crate::NTSTATUS {
+      unsafe {
+        driver.DriverUnload = Some(driver_exit);
+
+        for i in 0..$crate::IRP_MJ_MAXIMUM_FUNCTION {
+          driver.MajorFunction[i as usize] = Some($crate::dispatch_device);
+        }
+      }
+
+      let driver = unsafe { Driver::from_raw(driver) };
+
+      let registry_path = unsafe { $crate::U16CString::from_ptr_str(registry_path.Buffer) };
+      let registry_path = registry_path.to_string_lossy();
+
+      match <$module as $crate::KernelModule>::init(driver, registry_path.as_str()) {
+        Ok(m) => {
+          unsafe {
+            __MOD = Some(m);
+          }
+
+          $crate::STATUS_SUCCESS
+        }
+        Err(e) => e.to_ntstatus(),
+      }
+    }
+
+    pub unsafe extern "C" fn driver_exit(driver: *mut $crate::DRIVER_OBJECT) {
+      let driver = unsafe { Driver::from_raw(driver) };
+
+      match __MOD.take() {
+        Some(mut m) => m.cleanup(driver),
+        _ => (),
+      }
+    }
+  };
+}
+
+pub trait KernelModule: Sized + Sync {
+  fn init(driver: Driver, registry_path: &str) -> Result<Self, Error>;
+  fn cleanup(&mut self, _driver: Driver) {}
+}