69 lines
1.6 KiB
C++
69 lines
1.6 KiB
C++
|
// 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);
|
||
|
|
}
|