Really check for Steam DRM now

Jskdjfjdjf · Jul 6, 2018 · 809a1d1 · 809a1d1
1 parent 1077876
commit 809a1d1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/source/dllmain.cpp b/source/dllmain.cpp
@@ -433,16 +433,20 @@ void LoadEverything()
 static LONG RestoredOnce = 0;
 void LoadPluginsAndRestoreIAT(uintptr_t retaddr)
 {
-	if ( _InterlockedCompareExchange( &RestoredOnce, 1, 0 ) != 0 ) return;
+	bool calledFromBind = false;
 
 	//steam drm check
 	GetSections([&](PIMAGE_SECTION_HEADER pSection, size_t dwLoadOffset, DWORD dwPhysSize) {
 		auto dwStart = static_cast<uintptr_t>(dwLoadOffset + pSection->VirtualAddress);
 		auto dwEnd = dwStart + dwPhysSize;
 		if (retaddr >= dwStart && retaddr <= dwEnd)
-			return;
+			calledFromBind = true;
 	}, ".bind");
 
+	if ( calledFromBind ) return;
+
+	if ( _InterlockedCompareExchange( &RestoredOnce, 1, 0 ) != 0 ) return;
+
 	LoadEverything();
 
 	for (size_t i = 0; i < Kernel32ExportsNamesCount; i++)