forked from Hay1tsme/segatools
102 lines
2.9 KiB
C
102 lines
2.9 KiB
C
#include <windows.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include "util/dprintf.h"
|
|
#include "platform/opensslpatch.h"
|
|
|
|
int ChecknPatch(void) {
|
|
char* cpuname = GetCpuName();
|
|
if (cpuname == NULL) {
|
|
dprintf("Error: Unable to detect CPU.\n");
|
|
return 1;
|
|
}
|
|
|
|
//dprintf("CPU Detected: %s\n", cpuname);
|
|
|
|
if (CheckCpu(cpuname)) {
|
|
OpenSSLPatch();
|
|
dprintf("OpenSSL Patch applied successfully.\n");
|
|
} else {
|
|
dprintf("Info: OpenSSL Patch is not required (AMD or Intel < 10th gen or older CPU detected).\n");
|
|
}
|
|
|
|
free(cpuname);
|
|
return 0;
|
|
}
|
|
|
|
char* GetCpuName() {
|
|
FILE* fp;
|
|
char buffer[128];
|
|
char* cpu_info = NULL;
|
|
|
|
fp = _popen("wmic cpu get Name", "r");
|
|
|
|
if (fp == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
fgets(buffer, sizeof(buffer), fp);
|
|
|
|
if (fgets(buffer, sizeof(buffer), fp) != NULL) {
|
|
cpu_info = (char*)malloc(strlen(buffer) + 1);
|
|
strcpy(cpu_info, buffer);
|
|
}
|
|
_pclose(fp);
|
|
|
|
if (cpu_info != NULL) {
|
|
cpu_info[strcspn(cpu_info, "\r\n")] = 0;
|
|
}
|
|
|
|
return cpu_info;
|
|
}
|
|
|
|
int CheckCpu(char* cpuname) {
|
|
if (strstr(cpuname, "Core 2 Duo") || strstr(cpuname, "Core 2 Quad") ||
|
|
(strstr(cpuname, "Pentium") && !strstr(cpuname, "G")) || strstr(cpuname, "Celeron")) {
|
|
//dprintf("Trash detected. No patch needed.\n");
|
|
return 0;
|
|
}
|
|
|
|
if (strstr(cpuname, "Intel")) {
|
|
char* part = strtok(cpuname, " ");
|
|
while (part != NULL) {
|
|
if (part[0] == 'i' && strlen(part) >= 4) {
|
|
int gen = atoi(part + 1);
|
|
if (gen >= 10) {
|
|
dprintf("Intel Gen 10+ CPU Detected: %s\n", cpuname);
|
|
return 1;
|
|
}
|
|
} else if (part[0] == 'G' && strlen(part) >= 4) {
|
|
int pentium = atoi(part + 1);
|
|
if (pentium / 1000 >= 6) {
|
|
dprintf("Intel Gen 10+ CPU Detected: %s\n", cpuname);
|
|
return 1;
|
|
}
|
|
}
|
|
part = strtok(NULL, " ");
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void OpenSSLPatch(void) {
|
|
const char* variablename = "OPENSSL_ia32cap";
|
|
const char* variablevalue = "~0x20000000";
|
|
|
|
HKEY hKey;
|
|
if (RegOpenKeyExA(HKEY_CURRENT_USER, "Environment", 0, KEY_SET_VALUE, &hKey) == ERROR_SUCCESS) {
|
|
if (RegSetValueExA(hKey, variablename, 0, REG_SZ, (const BYTE*)variablevalue, strlen(variablevalue) + 1) == ERROR_SUCCESS) {
|
|
dprintf("Successfully set the user environment variable %s to %s\n", variablename, variablevalue);
|
|
} else {
|
|
dprintf("Error: Failed to set the user environment variable.\n");
|
|
}
|
|
|
|
RegCloseKey(hKey);
|
|
|
|
SendMessageTimeoutA(HWND_BROADCAST, WM_SETTINGCHANGE, 0, (LPARAM)"Environment", SMTO_ABORTIFHUNG, 5000, NULL);
|
|
} else {
|
|
dprintf("Error: Failed to open the user environment registry key.\n");
|
|
}
|
|
}
|