From f5b8c0c89f669afe44ed3e78de21a4030c830ea8 Mon Sep 17 00:00:00 2001
From: acidicoala <67734819+acidicoala@users.noreply.github.com>
Date: Mon, 22 Sep 2025 22:47:53 +0500
Subject: [PATCH] Added support for self-injection via `winhttp.dll`

---
 CMakeLists.txt   | 49 ++++++++++++++++++++++++++++--------------------
 src/main_win.cpp |  4 ++--
 2 files changed, 31 insertions(+), 22 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 09e1630..4175ed8 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -109,26 +109,35 @@ if(WIN32)
         INPUT_SOURCES_DIR ""
         DLL_FILES_GLOB "${CMAKE_CURRENT_SOURCE_DIR}/res/steamworks/*/binaries/${STEAM_API_MODULE}.dll"
     )
-    configure_linker_exports(
-        TARGET SmokeAPI
-        HEADER_NAME "linker_exports_for_version.h"
-        FORWARDED_DLL "C:/Windows/System32/version.dll"
-        INPUT_SOURCES_DIR ""
-        DLL_FILES_GLOB "C:/Windows/System32/version.dll"
-    )
-    configure_linker_exports(
-        TARGET SmokeAPI
-        HEADER_NAME "linker_exports_for_winmm.h"
-        FORWARDED_DLL "C:/Windows/System32/winmm.dll"
-        INPUT_SOURCES_DIR ""
-        DLL_FILES_GLOB "C:/Windows/System32/winmm.dll"
+
+    set(HEADER_CONTENT "#pragma once\n\n")
+
+    foreach(WIN_DLL version winhttp winmm)
+        set(HEADER_NAME "linker_exports_for_${WIN_DLL}.h")
+        configure_linker_exports(
+            TARGET SmokeAPI
+            HEADER_NAME "${HEADER_NAME}"
+            FORWARDED_DLL "C:/Windows/System32/${WIN_DLL}.dll"
+            INPUT_SOURCES_DIR ""
+            DLL_FILES_GLOB "C:/Windows/System32/${WIN_DLL}.dll"
+        )
+
+        set(HEADER_CONTENT "${HEADER_CONTENT}#include <${HEADER_NAME}>\n")
+    endforeach()
+
+    file(GENERATE
+        OUTPUT "${CMAKE_CURRENT_BINARY_DIR}/generated/linker_exports_for_windows_dlls.h"
+        CONTENT "${HEADER_CONTENT}"
     )
+
+    # Ignore linker warnings regarding COM-related private exports
+    set_target_properties(SmokeAPI PROPERTIES LINK_FLAGS "/ignore:4104")
 else()
-#    configure_linker_exports(
-#        TARGET SmokeAPI
-#        HEADER_NAME "libsteam_api_exports.cpp"
-#        FORWARDED_DLL "${STEAM_API_MODULE}_o.so"
-#        DLL_FILES_GLOB "${CMAKE_CURRENT_SOURCE_DIR}/res/steamworks/*/binaries/${LINUX_DIR}/${STEAM_API_MODULE}.so"
-#        INPUT_SOURCES_DIR ""
-#    )
+    # configure_linker_exports(
+    #     TARGET SmokeAPI
+    #     HEADER_NAME "libsteam_api_exports.cpp"
+    #     FORWARDED_DLL "${STEAM_API_MODULE}_o.so"
+    #     DLL_FILES_GLOB "${CMAKE_CURRENT_SOURCE_DIR}/res/steamworks/*/binaries/${LINUX_DIR}/${STEAM_API_MODULE}.so"
+    #     INPUT_SOURCES_DIR ""
+    # )
 endif()
diff --git a/src/main_win.cpp b/src/main_win.cpp
index d31e38d..2b07b10 100644
--- a/src/main_win.cpp
+++ b/src/main_win.cpp
@@ -1,11 +1,11 @@
+// ReSharper disable CppUnusedIncludeDirective
 #include <koalabox/win.hpp>
 
 #include "smoke_api/smoke_api.hpp"
 
 // These headers will be populated at build time
 #include "linker_exports_for_steam_api.h"
-#include "linker_exports_for_version.h"
-#include "linker_exports_for_winmm.h"
+#include "linker_exports_for_windows_dlls.h"
 
 DLL_MAIN(void* handle, const uint32_t reason, void*) {
     if(reason == DLL_PROCESS_ATTACH) {