add debug build cmake preset

Author: dumbasPL

CMakeLists.txt

@@ -15,5 +15,6 @@ 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")
+set(FUMO_DEBUG OFF CACHE BOOL "Disable creating new executables")
 
 add_subdirectory(src)

CMakePresets.json

@@ -13,6 +13,18 @@
       "architecture": {
         "value": "x64",
         "strategy": "set"
+      },
+      "cacheVariables": {
+        "FUMO_DEBUG": "OFF",
+        "FUMO_DRIVER_DEBUG": "OFF"
+      }
+    },
+    {
+      "name": "x64-windows-debug",
+      "inherits": "x64-windows",
+      "cacheVariables": {
+        "FUMO_DEBUG": "ON",
+        "FUMO_DRIVER_DEBUG": "ON"
       }
     }
   ],
@@ -21,6 +33,11 @@
       "name": "Release",
       "configurePreset": "x64-windows",
       "configuration": "Release"
+    },
+    {
+      "name": "Debug",
+      "configurePreset": "x64-windows-debug",
+      "configuration": "Release"
     }
   ]
 }

src/CMakeLists.txt

@@ -42,6 +42,10 @@ find_package(WDK REQUIRED)
 
 CPMAddPackage("gh:dumbasPL/[email protected]")
 
+if (FUMO_DEBUG)
+    add_compile_definitions(FUMO_DEBUG)
+endif()
+
 add_subdirectory(driver)
 add_subdirectory(driver_interface)
 add_subdirectory(rsource_generator)

src/initial_loader/initial_loader.cpp

@@ -212,6 +212,7 @@ extern "C" void initial_loader(ULONG_PTR xorKey) {
         }
     }
 
+#ifndef FUMO_DEBUG
     // allocate memory for new executable
     auto new_image_base = (ULONG_PTR)fnVirtualAlloc(nullptr, nt_headers->OptionalHeader.SizeOfImage, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
     if (!new_image_base)
@@ -339,20 +340,25 @@ extern "C" void initial_loader(ULONG_PTR xorKey) {
     }
 
     fnVirtualFree((PVOID)new_image_base, 0, MEM_RELEASE);
+#endif
 
     auto error = map_encrypted_sections(base);
     if (error != ERR_SUCCESS) {
+#ifndef FUMO_DEBUG
         fnCloseHandle(file_handle);
+#endif
         fnExitProcess(error);
         return;
     }
 
+#ifndef FUMO_DEBUG
     disposition_info.DeleteFileW = FALSE;
     if (!fnSetFileInformationByHandle(file_handle, FileDispositionInfo, &disposition_info, sizeof(disposition_info))) {
         fnCloseHandle(file_handle);
         EXIT_WITH_ERROR(ERR_FAILED_TO_SET_DELETE_FILE_INFO, "Failed to set delete file info (2)");
     }
 
     fnCloseHandle(file_handle);
+#endif
     fnExitProcess(0);
 }

src/stage2/stage2.cpp

@@ -18,9 +18,11 @@ int main(HANDLE loader_process) {
         return fumo::error(ERR_STAGE2_FAILED_TO_WAIT_FOR_PROCESS, L"Failed to wait for loader process({}): {}", loader_process_name, wait_result);
     CloseHandle(loader_process);
 
+#ifndef FUMO_DEBUG
     // delete the loader executable
     if (!DeleteFileW(loader_process_name.c_str()))
         return fumo::error(ERR_STAGE2_FAILED_TO_DELETE_LOADER, L"Failed to delete loader executable: {}", loader_process_name);
+#endif
 
     auto driver = fumo::DriverInterface::Open(FUMO_HOOKED_DRIVER_NAME_USER);
     if (!driver)