Windows Kernel-Mode library

1 files changed, 34 insertions, 0 deletions

diff --git a/src/macros.rs b/src/macros.rs
new file mode 100644
index 0000000..ba3b24d
--- /dev/null
+++ b/src/macros.rs
@@ -0,0 +1,34 @@
+//! Macros for Kernel-Mode drivers.
+
+/// Macro to send a message to the kernel debugger.
+///
+/// # Example
+///
+/// ```no_run
+/// KdPrint!("NTSTATUS is 0x%X\n", status);
+/// ```
+#[macro_export]
+macro_rules! KdPrint {
+	($msg:expr $(, $arg:expr)*) => { unsafe { $crate::debug::DbgPrint( concat!($msg, "\0").as_ptr() $(, $arg )* )} };
+}
+
+/// Macro to send a message to the kernel debugger for unsafe blocks.
+///
+/// Used in `unsafe {}` blocks.
+#[macro_export]
+macro_rules! KdPrint_u {
+	($msg:expr $(, $arg:expr)*) => { $crate::debug::DbgPrint( concat!($msg, "\0").as_ptr() $(, $arg )* ) };
+}
+
+#[macro_export]
+macro_rules! check_unsafe {
+	($expr:expr) => {{
+		let st: $crate::status::Status = unsafe { $expr };
+		if st.is_err() {
+			KdPrint!("[km] error: status 0x%X\n", st);
+			return st;
+		} else {
+			st
+		}
+	}}
+}