From 3d4145b6a45ea2f38ede0f9b1b2dd8a55833f83e Mon Sep 17 00:00:00 2001
From: momo5502 <mauriceheumann@gmail.com>
Date: Sun, 17 Apr 2022 10:26:26 +0200
Subject: [PATCH] Small fixes

---
 src/driver/irp.cpp  | 2 --
 src/runner/main.cpp | 5 ++++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/driver/irp.cpp b/src/driver/irp.cpp
index 81856e6..572e93b 100644
--- a/src/driver/irp.cpp
+++ b/src/driver/irp.cpp
@@ -74,7 +74,6 @@ namespace
 
 			debug_log("Level: %d\n", static_cast<int>(KeGetCurrentIrql()));
 
-
 			process::scoped_process_attacher attacher{process_handle};
 			translation_hints = vmx::ept::generate_translation_hints(r.target_address, r.source_data_size);
 		});
@@ -112,7 +111,6 @@ namespace
 				debug_log("Hello from the Driver!\n");
 				break;
 			case HOOK_DRV_IOCTL:
-
 				apply_hook(static_cast<hook_request*>(irp_sp->Parameters.DeviceIoControl.Type3InputBuffer));
 				break;
 			default:
diff --git a/src/runner/main.cpp b/src/runner/main.cpp
index 8ad51cf..be87b1c 100644
--- a/src/runner/main.cpp
+++ b/src/runner/main.cpp
@@ -1,4 +1,5 @@
 #include <iostream>
+#include <conio.h>
 
 #include "std_include.hpp"
 #include "finally.hpp"
@@ -79,7 +80,9 @@ void unsafe_main(const int /*argc*/, char* /*argv*/[])
 
 	(void)driver_device.send(HOOK_DRV_IOCTL, input);
 
-	MessageBoxA(0, "Press ok to exit!", 0, 0);
+	//MessageBoxA(0, "Press ok to exit!", 0, 0);
+	printf("Press any key to exit!\n");
+	_getch();
 }
 
 int main(const int argc, char* argv[])