tunsafe-clang15/tunsafe_cpu.cpp

69 lines
1.6 KiB
C++
Raw Normal View History

// SPDX-License-Identifier: AGPL-1.0-only
// Copyright (C) 2018 Ludvig Strigeus <info@tunsafe.com>. All Rights Reserved.
#include "stdafx.h"
#include "tunsafe_cpu.h"
#include "tunsafe_types.h"
#if defined(COMPILER_MSVC)
#include <intrin.h>
#endif
#include <string.h>
uint32 x86_pcap[3];
#if !defined(COMPILER_MSVC)
static inline void __cpuid(int info[4], int func) {
__asm__ __volatile__(
"cpuid"
: "=a"(info[0]), "=b"(info[1]), "=c"(info[2]), "=d"(info[3])
: "a"(func), "c"(0)
);
}
#endif
void InitCpuFeatures() {
unsigned nIds, nExIds;
{
int info[4];
__cpuid(info, 0);
nIds = info[0];
__cpuid(info, 0x80000000);
nExIds = info[0];
}
if (nIds >= 0x00000001) {
int info[4];
__cpuid(info, 0x00000001);
x86_pcap[0] = info[3];
x86_pcap[1] = info[2];
}
if (nIds >= 0x00000007) {
int info[4];
__cpuid(info, 0x00000007);
x86_pcap[2] = info[1];
}
}
static char *strcpy_e(char *dst, char *end, const char *copy) {
size_t len = strlen(copy);
if (len >= (size_t)(end - dst)) return end;
memcpy(dst, copy, len + 1);
return dst + len;
}
void PrintCpuFeatures() {
char capbuf[2048], *end = capbuf + 2048, *s = capbuf;
if (X86_PCAP_AVX) s = strcpy_e(s, end, " avx");
if (X86_PCAP_SSSE3) s = strcpy_e(s, end, " ssse3");
if (X86_PCAP_AVX2) s = strcpy_e(s, end, " avx2");
if (X86_PCAP_MOVBE) s = strcpy_e(s, end, " movbe");
if (X86_PCAP_AES) s = strcpy_e(s, end, " aes");
if (X86_PCAP_PCLMULQDQ) s = strcpy_e(s, end, " pclmuldqd");
if (X86_PCAP_AVX512F) s = strcpy_e(s, end, " avx512f");
if (X86_PCAP_AVX512VL) s = strcpy_e(s, end, " avx512vl");
RINFO("Using:%s", capbuf);
}