affinity option for blacklist

ThirteenAG · Oct 27, 2024 · d733f17 · d733f17
1 parent 4cf3a06
commit d733f17
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/data/SplinterCellBlacklist.FusionMod/src/system/scripts/SplinterCellBlacklist.FusionMod.ini b/data/SplinterCellBlacklist.FusionMod/src/system/scripts/SplinterCellBlacklist.FusionMod.ini
@@ -11,6 +11,7 @@ DisableNightVisionFlash = 1                // Disables the white flash when acti
 DisablePerfectionistChecks = 1             // M&E UI, radar
 DefaultMissionFilter = 1                   // 0: All missions | 1: Active missions | 2: Campaign missions | 3: 4E missions
 SMIMapDisableStartupAnimation = 1
+ForceCPUAffinityToAllCores = 1
 
 [UNLOCKS]
 UnlockDLC = 1                              // Unlocks all DLC content that is not possible to obtain after server shutdown

diff --git a/source/SplinterCellBlacklist.FusionMod/dllmain.cpp b/source/SplinterCellBlacklist.FusionMod/dllmain.cpp
@@ -2,6 +2,23 @@
 #include <LEDEffects.h>
 #include <random>
 
+SafetyHookInline shSetProcessAffinityMask{};
+BOOL WINAPI SetProcessAffinityMaskHook(HANDLE hProcess, DWORD_PTR dwProcessAffinityMask)
+{
+    if (hProcess == GetCurrentProcess())
+    {
+        DWORD_PTR processAffinityMask;
+        DWORD_PTR systemAffinityMask;
+
+        if (GetProcessAffinityMask(GetCurrentProcess(), &processAffinityMask, &systemAffinityMask))
+        {
+            return shSetProcessAffinityMask.stdcall<BOOL>(hProcess, systemAffinityMask);
+        }
+    }
+
+    return shSetProcessAffinityMask.stdcall<BOOL>(hProcess, dwProcessAffinityMask);
+}
+
 HWND WindowHandle;
 
 enum eGameMode
@@ -208,6 +225,7 @@ void Init()
     auto bDisablePerfectionistChecks = iniReader.ReadInteger("MAIN", "DisablePerfectionistChecks", 1) != 0;
     auto nDefaultMissionFilter = std::clamp(iniReader.ReadInteger("MAIN", "DefaultMissionFilter", 1), 0, 3);
     auto bSMIMapDisableStartupAnimation = iniReader.ReadInteger("MAIN", "SMIMapDisableStartupAnimation", 1) != 0;
+    auto bForceCPUAffinityToAllCores = iniReader.ReadInteger("MAIN", "ForceCPUAffinityToAllCores", 1) != 0;
 
     sExtractionWaveConfigs = iniReader.ReadString("EXTRACTION", "ExtractionWaveConfigs", "Default");
     nExtractionWaveEnemyMultiplier = std::clamp(iniReader.ReadInteger("EXTRACTION", "ExtractionWaveEnemyMultiplier", 1), 1, 9999);
@@ -919,6 +937,11 @@ void Init()
         pattern = hook::pattern("0F 8E ? ? ? ? 8D 8E ? ? ? ? E8 ? ? ? ? 80 BE");
         injector::WriteMemory<uint16_t>(pattern.count(2).get(1).get<void>(), 0xE990, true); //jle -> jmp
     }
+
+    if (bForceCPUAffinityToAllCores)
+    {
+        shSetProcessAffinityMask = safetyhook::create_inline(SetProcessAffinityMask, SetProcessAffinityMaskHook);
+    }
 }
 
 std::string currentGogglesLight = "LightGreen";