|
#pragma once |
|
#include <stdint.h> |
|
|
|
#if defined(__GNUC__) |
|
#include <cpuid.h> |
|
#elif defined(_MSC_VER) |
|
#include <intrin.h> |
|
#endif |
|
|
|
#if CPUINFO_MOCK |
|
#include <cpuinfo-mock.h> |
|
#endif |
|
#include <x86/api.h> |
|
|
|
|
|
#if defined(__GNUC__) || defined(_MSC_VER) |
|
static inline struct cpuid_regs cpuid(uint32_t eax) { |
|
#if CPUINFO_MOCK |
|
uint32_t regs_array[4]; |
|
cpuinfo_mock_get_cpuid(eax, regs_array); |
|
return (struct cpuid_regs) { |
|
.eax = regs_array[0], |
|
.ebx = regs_array[1], |
|
.ecx = regs_array[2], |
|
.edx = regs_array[3], |
|
}; |
|
#else |
|
struct cpuid_regs regs; |
|
#if defined(__GNUC__) |
|
__cpuid(eax, regs.eax, regs.ebx, regs.ecx, regs.edx); |
|
#else |
|
int regs_array[4]; |
|
__cpuid(regs_array, (int) eax); |
|
regs.eax = regs_array[0]; |
|
regs.ebx = regs_array[1]; |
|
regs.ecx = regs_array[2]; |
|
regs.edx = regs_array[3]; |
|
#endif |
|
return regs; |
|
#endif |
|
} |
|
|
|
static inline struct cpuid_regs cpuidex(uint32_t eax, uint32_t ecx) { |
|
#if CPUINFO_MOCK |
|
uint32_t regs_array[4]; |
|
cpuinfo_mock_get_cpuidex(eax, ecx, regs_array); |
|
return (struct cpuid_regs) { |
|
.eax = regs_array[0], |
|
.ebx = regs_array[1], |
|
.ecx = regs_array[2], |
|
.edx = regs_array[3], |
|
}; |
|
#else |
|
struct cpuid_regs regs; |
|
#if defined(__GNUC__) |
|
__cpuid_count(eax, ecx, regs.eax, regs.ebx, regs.ecx, regs.edx); |
|
#else |
|
int regs_array[4]; |
|
__cpuidex(regs_array, (int) eax, (int) ecx); |
|
regs.eax = regs_array[0]; |
|
regs.ebx = regs_array[1]; |
|
regs.ecx = regs_array[2]; |
|
regs.edx = regs_array[3]; |
|
#endif |
|
return regs; |
|
#endif |
|
} |
|
#endif |
|
|
|
static inline uint64_t xgetbv(uint32_t ext_ctrl_reg) { |
|
#ifdef _MSC_VER |
|
return (uint64_t)_xgetbv((unsigned int)ext_ctrl_reg); |
|
#else |
|
uint32_t lo, hi; |
|
__asm__(".byte 0x0F, 0x01, 0xD0" : "=a" (lo), "=d" (hi) : "c" (ext_ctrl_reg)); |
|
return ((uint64_t) hi << 32) | (uint64_t) lo; |
|
#endif |
|
} |
|
|
|
|