add FUMO_DRIVER_DEBUG cmake flag

dumbasPL · Jun 9, 2024 · 6ee5530 · 6ee5530
1 parent 068354d
commit 6ee5530
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 8 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -14,4 +14,6 @@ endif()
 set(CPM_SOURCE_CACHE "${CMAKE_SOURCE_DIR}/.cache/cpm")
 include(cmake/CPM.cmake)
 
+set(FUMO_DRIVER_DEBUG OFF CACHE BOOL "Enable driver debug logs/force reload")
+
 add_subdirectory(src)
diff --git a/src/driver/CMakeLists.txt b/src/driver/CMakeLists.txt
@@ -2,5 +2,6 @@ wdk_add_driver(fumo_drv WINVER 0x4A61 driver.c memory.c thread.c)
 target_include_directories(fumo_drv PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/../include)
 set_target_properties(fumo_drv PROPERTIES RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
 
-# FIXME: add a cmake option to enable debug mode
-target_compile_definitions(fumo_drv PRIVATE -DNO_FUMO_DRIVER_DEBUG)
+if (FUMO_DRIVER_DEBUG)
+    target_compile_definitions(fumo_drv PRIVATE -DFUMO_DRIVER_DEBUG)
+endif()
diff --git a/src/driver/driver.h b/src/driver/driver.h
@@ -1,10 +1,6 @@
 #pragma once
 #include "imports.h"
 
-#ifndef NO_FUMO_DRIVER_DEBUG
-#define FUMO_DRIVER_DEBUG
-#endif
-
 #ifdef FUMO_DRIVER_DEBUG
 
 // FUMO

diff --git a/src/stage1/CMakeLists.txt b/src/stage1/CMakeLists.txt
@@ -21,4 +21,8 @@ target_compile_features(stage1 PUBLIC c_std_17 cxx_std_20)
 target_compile_definitions(stage1 PRIVATE UNICODE _UNICODE)
 target_compile_options(stage1 PRIVATE /O1 /GS- /sdl- /guard:cf- /Zc:threadSafeInit-)
 target_link_libraries(stage1 PUBLIC ntdll libKDU driver_interface lazy_importer)
-set_target_properties(stage1 PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})
+set_target_properties(stage1 PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${CMAKE_RUNTIME_OUTPUT_DIRECTORY})
+
+if (FUMO_DRIVER_DEBUG)
+    target_compile_definitions(stage1 PRIVATE -DFUMO_DRIVER_DEBUG)
+endif()
diff --git a/src/stage1/stage1.cpp b/src/stage1/stage1.cpp
@@ -2,6 +2,12 @@
 #include <filesystem>
 #include <fstream>
 
+#ifdef FUMO_DRIVER_DEBUG
+#define FORCE_RELOAD_DRIVER true
+#else
+#define FORCE_RELOAD_DRIVER false
+#endif
+
 int stage1(PFUMO_EMBEDDED_DATA embedded_data) {
     std::vector<BYTE> fumo_data;
     std::wstring fumo_file_path;
@@ -68,7 +74,7 @@ int stage1(PFUMO_EMBEDDED_DATA embedded_data) {
     if (!driver)
         return fumo::error(ERR_STAGE1_FAILED_TO_OPEN_DRIVER, L"Failed to open driver");
 
-    auto error = init_driver(driver.get(), osv.dwBuildNumber);
+    auto error = init_driver(driver.get(), osv.dwBuildNumber, FORCE_RELOAD_DRIVER);
     if (error != ERR_STAGE1_SUCCESS)
         return error;