/* CpuArch.h -- CPU specific code
2018-02-18 : Igor Pavlov : Public domain */

#pragma once

#include "7zTypes.h"

EXTERN_C_BEGIN

/*
MY_CPU_LE means that CPU is LITTLE ENDIAN.
MY_CPU_BE means that CPU is BIG ENDIAN.
If MY_CPU_LE and MY_CPU_BE are not defined, we don't know about ENDIANNESS of platform.

MY_CPU_LE_UNALIGN means that CPU is LITTLE ENDIAN and CPU supports unaligned memory accesses.
*/

#if  defined (_M_X64) \
  || defined (_M_AMD64) \
  || defined (__x86_64__) \
  || defined (__AMD64__) \
  || defined (__amd64__)
#define MY_CPU_AMD64
  #ifdef __ILP32__
#define MY_CPU_NAME  "x32"
  #else
#define MY_CPU_NAME  "x64"
  #endif
#define MY_CPU_64BIT
#endif

#if  defined (_M_IX86) \
  || defined (__i386__)
#define MY_CPU_X86
#define MY_CPU_NAME  "x86"
#define MY_CPU_32BIT
#endif

#if  defined (_M_ARM64) \
  || defined (__AARCH64EL__) \
  || defined (__AARCH64EB__) \
  || defined (__aarch64__)
#define MY_CPU_ARM64
#define MY_CPU_NAME  "arm64"
#define MY_CPU_64BIT
#endif

#if  defined (_M_ARM) \
  || defined (_M_ARM_NT) \
  || defined (_M_ARMT) \
  || defined (__arm__) \
  || defined (__thumb__) \
  || defined (__ARMEL__) \
  || defined (__ARMEB__) \
  || defined (__THUMBEL__) \
  || defined (__THUMBEB__)
#define MY_CPU_ARM
#define MY_CPU_NAME  "arm"
#define MY_CPU_32BIT
#endif

#if  defined (_M_IA64) \
  || defined (__ia64__)
#define MY_CPU_IA64
#define MY_CPU_NAME  "ia64"
#define MY_CPU_64BIT
#endif

#if  defined (__mips64) \
  || defined (__mips64__) \
  || (defined (__mips) && (__mips == 64 || __mips == 4 || __mips == 3))
#define MY_CPU_NAME  "mips64"
#define MY_CPU_64BIT
#elif defined (__mips__)
#define MY_CPU_NAME  "mips"
/* #define MY_CPU_32BIT */
#endif

#if  defined (__ppc64__) \
  || defined (__powerpc64__)
  #ifdef __ILP32__
#define MY_CPU_NAME  "ppc64-32"
  #else
#define MY_CPU_NAME  "ppc64"
  #endif
#define MY_CPU_64BIT
#elif defined (__ppc__) \
  || defined (__powerpc__)
#define MY_CPU_NAME  "ppc"
#define MY_CPU_32BIT
#endif

#if  defined (__sparc64__)
#define MY_CPU_NAME  "sparc64"
#define MY_CPU_64BIT
#elif defined (__sparc__)
#define MY_CPU_NAME  "sparc"
/* #define MY_CPU_32BIT */
#endif

#if defined (MY_CPU_X86) || defined (MY_CPU_AMD64)
#define MY_CPU_X86_OR_AMD64
#endif

#ifdef _WIN32

  #ifdef MY_CPU_ARM
#define MY_CPU_ARM_LE
  #endif

  #ifdef MY_CPU_ARM64
#define MY_CPU_ARM64_LE
  #endif

  #ifdef _M_IA64
#define MY_CPU_IA64_LE
  #endif

#endif

#if defined (MY_CPU_X86_OR_AMD64) \
  || defined (MY_CPU_ARM_LE) \
  || defined (MY_CPU_ARM64_LE) \
  || defined (MY_CPU_IA64_LE) \
  || defined (__LITTLE_ENDIAN__) \
  || defined (__ARMEL__) \
  || defined (__THUMBEL__) \
  || defined (__AARCH64EL__) \
  || defined (__MIPSEL__) \
  || defined (__MIPSEL) \
  || defined (_MIPSEL) \
  || defined (__BFIN__) \
  || (defined (__BYTE_ORDER__) && (__BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__))
#define MY_CPU_LE
#endif

#if defined (__BIG_ENDIAN__) \
  || defined (__ARMEB__) \
  || defined (__THUMBEB__) \
  || defined (__AARCH64EB__) \
  || defined (__MIPSEB__) \
  || defined (__MIPSEB) \
  || defined (_MIPSEB) \
  || defined (__m68k__) \
  || defined (__s390__) \
  || defined (__s390x__) \
  || defined (__zarch__) \
  || (defined (__BYTE_ORDER__) && (__BYTE_ORDER__ == __ORDER_BIG_ENDIAN__))
#define MY_CPU_BE
#endif

#if defined (MY_CPU_LE) && defined (MY_CPU_BE)
  #error Stop_Compiling_Bad_Endian
#endif

#if defined (MY_CPU_32BIT) && defined (MY_CPU_64BIT)
  #error Stop_Compiling_Bad_32_64_BIT
#endif

#ifndef MY_CPU_NAME
  #ifdef MY_CPU_LE
#define MY_CPU_NAME  "LE"
  #elif defined (MY_CPU_BE)
#define MY_CPU_NAME  "BE"
  #else

/*
#define MY_CPU_NAME ""
*/
  #endif
#endif

#ifdef MY_CPU_LE
  #if defined (MY_CPU_X86_OR_AMD64) \
  || defined (MY_CPU_ARM64) \
  || defined (__ARM_FEATURE_UNALIGNED)
#define MY_CPU_LE_UNALIGN
  #endif
#endif

#ifdef MY_CPU_LE_UNALIGN

#define GetUi16(p)  (*(const UInt16 *)(const void *)(p))
#define GetUi32(p)  (*(const UInt32 *)(const void *)(p))
#define GetUi64(p)  (*(const UInt64 *)(const void *)(p))

#define SetUi16(p, v)  { *(UInt16 *)(p) = (v); }
#define SetUi32(p, v)  { *(UInt32 *)(p) = (v); }
#define SetUi64(p, v)  { *(UInt64 *)(p) = (v); }

#else

#define GetUi16(p)  ( (UInt16) (\
             ((const Byte *)(p))[0] | \
    ((UInt16)((const Byte *)(p))[1] << 8) ))

#define GetUi32(p)  (\
             ((const Byte *)(p))[0]        | \
    ((UInt32)((const Byte *)(p))[1] <<  8) | \
    ((UInt32)((const Byte *)(p))[2] << 16) | \
    ((UInt32)((const Byte *)(p))[3] << 24))

#define GetUi64(p)  (GetUi32(p) | ((UInt64)GetUi32(((const Byte *)(p)) + 4) << 32))

#define SetUi16(p, v)  { Byte *_ppp_ = (Byte *)(p); UInt32 _vvv_ = (v);\
    _ppp_[0] = (Byte)_vvv_; \
    _ppp_[1] = (Byte)(_vvv_ >> 8); }

#define SetUi32(p, v)  { Byte *_ppp_ = (Byte *)(p); UInt32 _vvv_ = (v);\
    _ppp_[0] = (Byte)_vvv_; \
    _ppp_[1] = (Byte)(_vvv_ >> 8); \
    _ppp_[2] = (Byte)(_vvv_ >> 16); \
    _ppp_[3] = (Byte)(_vvv_ >> 24); }

#define SetUi64(p, v)  { Byte *_ppp2_ = (Byte *)(p); UInt64 _vvv2_ = (v);\
    SetUi32(_ppp2_    , (UInt32)_vvv2_); \
    SetUi32(_ppp2_ + 4, (UInt32)(_vvv2_ >> 32)); }

#endif

#ifdef __has_builtin
#define MY__has_builtin(x)  __has_builtin(x)
#else
#define MY__has_builtin(x)  0
#endif

#if defined (MY_CPU_LE_UNALIGN) && /* defined(_WIN64) && */ (_MSC_VER >= 1300)

/* Note: we use bswap instruction, that is unsupported in 386 cpu */

  #include <stdlib.h>

  #pragma intrinsic(_byteswap_ushort)
  #pragma intrinsic(_byteswap_ulong)
  #pragma intrinsic(_byteswap_uint64)

/* #define GetBe16(p) _byteswap_ushort(*(const UInt16 *)(const Byte *)(p)) */
#define GetBe32(p)  _byteswap_ulong(*(const UInt32 *)(const Byte *)(p))
#define GetBe64(p)  _byteswap_uint64(*(const UInt64 *)(const Byte *)(p))

#define SetBe32(p, v)  (*(UInt32 *)(void *)(p)) = _byteswap_ulong(v)

#elif defined (MY_CPU_LE_UNALIGN) && ( \
                                       (defined (__GNUC__) && (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 3))) \
  || (defined (__clang__) && MY__has_builtin (__builtin_bswap16)))

/* #define GetBe16(p) __builtin_bswap16(*(const UInt16 *)(const Byte *)(p)) */
#define GetBe32(p)  __builtin_bswap32(*(const UInt32 *)(const Byte *)(p))
#define GetBe64(p)  __builtin_bswap64(*(const UInt64 *)(const Byte *)(p))

#define SetBe32(p, v)  (*(UInt32 *)(void *)(p)) = __builtin_bswap32(v)

#else

#define GetBe32(p)  (\
    ((UInt32)((const Byte *)(p))[0] << 24) | \
    ((UInt32)((const Byte *)(p))[1] << 16) | \
    ((UInt32)((const Byte *)(p))[2] <<  8) | \
             ((const Byte *)(p))[3] )

#define GetBe64(p)  (((UInt64)GetBe32(p) << 32) | GetBe32(((const Byte *)(p)) + 4))

#define SetBe32(p, v)  { Byte *_ppp_ = (Byte *)(p); UInt32 _vvv_ = (v);\
    _ppp_[0] = (Byte)(_vvv_ >> 24); \
    _ppp_[1] = (Byte)(_vvv_ >> 16); \
    _ppp_[2] = (Byte)(_vvv_ >> 8); \
    _ppp_[3] = (Byte)_vvv_; }

#endif

#ifndef GetBe16

#define GetBe16(p)  ( (UInt16) (\
    ((UInt16)((const Byte *)(p))[0] << 8) | \
             ((const Byte *)(p))[1] ))

#endif

#ifdef MY_CPU_X86_OR_AMD64

typedef struct {
  UInt32    maxFunc;
  UInt32    vendor[3];
  UInt32    ver;
  UInt32    b;
  UInt32    c;
  UInt32    d;
} Cx86cpuid;

enum {
  CPU_FIRM_INTEL,
  CPU_FIRM_AMD,
  CPU_FIRM_VIA
};

void
MyCPUID (
  UInt32  function,
  UInt32  *a,
  UInt32  *b,
  UInt32  *c,
  UInt32  *d
  );

BoolInt
x86cpuid_CheckAndRead (
  Cx86cpuid  *p
  );

int
x86cpuid_GetFirm (
  const Cx86cpuid  *p
  );

#define x86cpuid_GetFamily(ver)    (((ver >> 16) & 0xFF0) | ((ver >> 8) & 0xF))
#define x86cpuid_GetModel(ver)     (((ver >> 12) &  0xF0) | ((ver >> 4) & 0xF))
#define x86cpuid_GetStepping(ver)  (ver & 0xF)

BoolInt
CPU_Is_InOrder (
  );

BoolInt
CPU_Is_Aes_Supported (
  );

BoolInt
CPU_IsSupported_PageGB (
  );

#endif

EXTERN_C_END