// 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>

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;
}


#if defined(ARCH_CPU_X86_FAMILY)

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];
  }
}

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

#endif  // defined(ARCH_CPU_X86_FAMILY)

#if defined(ARCH_CPU_ARM_FAMILY)

uint32 arm_pcap[1];

void InitCpuFeatures() {
  arm_pcap[0] = 0xffffffff;
}

void PrintCpuFeatures() {
  char capbuf[2048], *end = capbuf + 2048, *s = capbuf;

  if (ARM_PCAP_NEON) s = strcpy_e(s, end, " neon");

  RINFO("Using:%s", capbuf);
}
#endif  // defined(ARCH_CPU_ARM_FAMILY)