/*
 * Copyright (c) 2024, Dan Klishch <danilklishch@gmail.com>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <AK/CPUFeatures.h>

namespace AK {

#ifndef KERNEL
#    if ARCH(I386) || ARCH(X86_64)
struct CPUIDResult {
    u32 eax { 0 };
    u32 ebx { 0 };
    u32 ecx { 0 };
    u32 edx { 0 };
};

static CPUIDResult cpuid(u32 leaf, u32 subleaf)
{
    CPUIDResult result;
    asm("cpuid"
        : "=a"(result.eax), "=b"(result.ebx), "=c"(result.ecx), "=d"(result.edx)
        : "0"(leaf), "2"(subleaf));
    return result;
}
#    endif

CPUFeatures Detail::detect_cpu_features_uncached()
{
    CPUFeatures result = CPUFeatures::None;

#    if ARCH(I386) || ARCH(X86_64)
    u32 max_leaf = cpuid(0, 0).eax;
    [[maybe_unused]] auto cpuid1 = max_leaf >= 1 ? cpuid(1, 0) : CPUIDResult {};
    [[maybe_unused]] auto cpuid7 = max_leaf >= 7 ? cpuid(7, 0) : CPUIDResult {};

#        if AK_CAN_CODEGEN_FOR_X86_SSE42
    if (cpuid1.ecx >> 20 & 1)
        result |= CPUFeatures::X86_SSE42;
#        endif
#        if AK_CAN_CODEGEN_FOR_X86_SHA
    if (cpuid7.ebx >> 29 & 1)
        result |= CPUFeatures::X86_SHA;
#        endif
#        if AK_CAN_CODEGEN_FOR_X86_AES
    if (cpuid1.ecx >> 25 & 1)
        result |= CPUFeatures::X86_AES;
#        endif
#    endif

    return result;
}
#endif

}