From a8a9af80060e6ad4614ce483cbdefbe09a89f266 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o?= <joaovictorti08@gmail.com>
Date: Tue, 30 Jul 2024 22:08:32 -0300
Subject: [PATCH] Update code lib

---
 driver/src/lib.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/driver/src/lib.rs b/driver/src/lib.rs
index 71da5cc..9ac17de 100644
--- a/driver/src/lib.rs
+++ b/driver/src/lib.rs
@@ -31,6 +31,7 @@ mod process;
 mod thread;
 mod module;
 mod injection;
+mod memory;
 mod utils;
 
 /// The name of the device in the device namespace.
@@ -183,6 +184,9 @@ pub unsafe extern "system" fn shadow_entry(
 /// - `IOCTL_KEYLOGGER`: Start / Stop Keylogger.
 /// - `IOCTL_ENUMERATE_CALLBACK`: Lists callbacks.
 /// - `IOCTL_REMOVE_CALLBACK`: Remove a callback.
+/// - `IOCTL_REGISTRY_PROTECTION_VALUE`:
+/// - `IOCTL_REGISTRY_PROTECTION_KEY`:
+/// - `IOCTL_INJECTION`:
 ///
 pub unsafe extern "C" fn device_control(_device: *mut DEVICE_OBJECT, irp: *mut IRP) -> NTSTATUS {
     let stack = (*irp).Tail.Overlay.__bindgen_anon_2.__bindgen_anon_1.CurrentStackLocation;