#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");
    }
}