Enhancement: Preparing for cpu features check

Ceiridge · Aug 1, 2021 · c4c9ef9 · c4c9ef9
1 parent 9b1e614
commit c4c9ef9
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 6 deletions.
diff --git a/ChromePatcherDll/patches.cpp b/ChromePatcherDll/patches.cpp
@@ -141,10 +141,15 @@ namespace ChromePatch {
 	}
 
 
-	auto simpleSearcher = std::make_unique<SimplePatternSearcher>();
-
-	// TODO: Externalize this function in different implementations (traditional and with SIMD support) and add multithreading
+	// TODO: Externalize this function in different implementations (traditional and with SIMD support) and add multi-threading
 	int Patches::ApplyPatches() {
+		std::unique_ptr<PatternSearcher> patternSearcher;
+		if(SimdPatternSearcher::IsCpuSupported()) {
+			patternSearcher = std::make_unique<SimdPatternSearcher>();
+		} else {
+			patternSearcher = std::make_unique<SimplePatternSearcher>();
+		}
+
 		int successfulPatches = 0;
 		std::cout << "Applying patches, please wait..." << std::endl;
 		HANDLE proc = GetCurrentProcess();
@@ -163,7 +168,8 @@ namespace ChromePatch {
 							continue;
 						}
 
-						byte* searchResult = simpleSearcher->SearchBytePattern(patch, static_cast<byte*>(mbi.BaseAddress), mbi.RegionSize);
+						// TODO: Check for cpu features and use the respective searcher & multi-threading
+						byte* searchResult = patternSearcher->SearchBytePattern(patch, static_cast<byte*>(mbi.BaseAddress), mbi.RegionSize);
 
 						if(searchResult == nullptr) {
 							std::cerr << "Pattern not found for patch " << patch << std::endl;

diff --git a/ChromePatcherDll/simdpatternsearcher.cpp b/ChromePatcherDll/simdpatternsearcher.cpp
@@ -7,4 +7,9 @@ namespace ChromePatch {
 		// TODO
 		return nullptr;
 	}
+
+	bool SimdPatternSearcher::IsCpuSupported() {
+		// TODO
+		return IsProcessorFeaturePresent(PF_AVX2_INSTRUCTIONS_AVAILABLE);
+	}
 }
diff --git a/ChromePatcherDll/simdpatternsearcher.hpp b/ChromePatcherDll/simdpatternsearcher.hpp
@@ -5,8 +5,9 @@ namespace ChromePatch {
 #ifdef _DEBUG
 		__declspec(dllexport)
 #endif
-	SimdPatternSearcher : PatternSearcher {
+	SimdPatternSearcher : public PatternSearcher {
 	public:
 		byte* SearchBytePattern(Patch& patch, byte* startAddr, size_t length) override;
+		static bool IsCpuSupported();
 	};
 }
diff --git a/ChromePatcherDll/simplepatternsearcher.hpp b/ChromePatcherDll/simplepatternsearcher.hpp
@@ -5,7 +5,7 @@ namespace ChromePatch {
 #ifdef _DEBUG
 	__declspec(dllexport)
 #endif
-	SimplePatternSearcher : PatternSearcher {
+	SimplePatternSearcher : public PatternSearcher {
 	public:
 		byte* SearchBytePattern(Patch& patch, byte* startAddr, size_t length) override;
 	};