third_party/lzma_sdk/CpuArch.c

   1 /* CpuArch.c -- CPU specific code
   2 2010-10-26: Igor Pavlov : Public domain */
   3
   4 #include "CpuArch.h"
   5
   6 #ifdef MY_CPU_X86_OR_AMD64
   7
   8 #if (defined(_MSC_VER) && !defined(MY_CPU_AMD64)) || defined(__GNUC__)
   9 #define USE_ASM
  10 #endif
  11
  12 #if defined(USE_ASM) && !defined(MY_CPU_AMD64)
  13 static UInt32 CheckFlag(UInt32 flag)
  14 {
  15   #ifdef _MSC_VER
  16   __asm pushfd;
  17   __asm pop EAX;
  18   __asm mov EDX, EAX;
  19   __asm xor EAX, flag;
  20   __asm push EAX;
  21   __asm popfd;
  22   __asm pushfd;
  23   __asm pop EAX;
  24   __asm xor EAX, EDX;
  25   __asm push EDX;
  26   __asm popfd;
  27   __asm and flag, EAX;
  28   #else
  29   __asm__ __volatile__ (
  30     "pushf\n\t"
  31     "pop  %%EAX\n\t"
  32     "movl %%EAX,%%EDX\n\t"
  33     "xorl %0,%%EAX\n\t"
  34     "push %%EAX\n\t"
  35     "popf\n\t"
  36     "pushf\n\t"
  37     "pop  %%EAX\n\t"
  38     "xorl %%EDX,%%EAX\n\t"
  39     "push %%EDX\n\t"
  40     "popf\n\t"
  41     "andl %%EAX, %0\n\t":
  42     "=c" (flag) : "c" (flag));
  43   #endif
  44   return flag;
  45 }
  46 #define CHECK_CPUID_IS_SUPPORTED if (CheckFlag(1 << 18) == 0 || CheckFlag(1 << 21) == 0) return False;
  47 #else
  48 #define CHECK_CPUID_IS_SUPPORTED
  49 #endif
  50
  51 static void MyCPUID(UInt32 function, UInt32 *a, UInt32 *b, UInt32 *c, UInt32 *d)
  52 {
  53   #ifdef USE_ASM
  54
  55   #ifdef _MSC_VER
  56
  57   UInt32 a2, b2, c2, d2;
  58   __asm xor EBX, EBX;
  59   __asm xor ECX, ECX;
  60   __asm xor EDX, EDX;
  61   __asm mov EAX, function;
  62   __asm cpuid;
  63   __asm mov a2, EAX;
  64   __asm mov b2, EBX;
  65   __asm mov c2, ECX;
  66   __asm mov d2, EDX;
  67
  68   *a = a2;
  69   *b = b2;
  70   *c = c2;
  71   *d = d2;
  72
  73   #else
  74
  75   __asm__ __volatile__ (
  76     "mov %%ebx, %%edi\n"
  77     "cpuid\n"
  78     "xchg %%edi, %%ebx\n"
  79     : "=a" (*a) ,
  80       "=D" (*b) ,
  81       "=c" (*c) ,
  82       "=d" (*d)
  83     : "0" (function)) ;
  84
  85   #endif
  86
  87   #else
  88
  89   int CPUInfo[4];
  90   __cpuid(CPUInfo, function);
  91   *a = CPUInfo[0];
  92   *b = CPUInfo[1];
  93   *c = CPUInfo[2];
  94   *d = CPUInfo[3];
  95
  96   #endif
  97 }
  98
  99 Bool x86cpuid_CheckAndRead(Cx86cpuid *p)
 100 {
 101   CHECK_CPUID_IS_SUPPORTED
 102   MyCPUID(0, &p->maxFunc, &p->vendor[0], &p->vendor[2], &p->vendor[1]);
 103   MyCPUID(1, &p->ver, &p->b, &p->c, &p->d);
 104   return True;
 105 }
 106
 107 static UInt32 kVendors[][3] =
 108 {
 109   { 0x756E6547, 0x49656E69, 0x6C65746E},
 110   { 0x68747541, 0x69746E65, 0x444D4163},
 111   { 0x746E6543, 0x48727561, 0x736C7561}
 112 };
 113
 114 int x86cpuid_GetFirm(const Cx86cpuid *p)
 115 {
 116   unsigned i;
 117   for (i = 0; i < sizeof(kVendors) / sizeof(kVendors[i]); i++)
 118   {
 119     const UInt32 *v = kVendors[i];
 120     if (v[0] == p->vendor[0] &&
 121         v[1] == p->vendor[1] &&
 122         v[2] == p->vendor[2])
 123       return (int)i;
 124   }
 125   return -1;
 126 }
 127
 128 Bool CPU_Is_InOrder()
 129 {
 130   Cx86cpuid p;
 131   int firm;
 132   UInt32 family, model;
 133   if (!x86cpuid_CheckAndRead(&p))
 134     return True;
 135   family = x86cpuid_GetFamily(&p);
 136   model = x86cpuid_GetModel(&p);
 137   firm = x86cpuid_GetFirm(&p);
 138   switch (firm)
 139   {
 140     case CPU_FIRM_INTEL: return (family < 6 || (family == 6 && model == 0x100C));
 141     case CPU_FIRM_AMD: return (family < 5 || (family == 5 && (model < 6 || model == 0xA)));
 142     case CPU_FIRM_VIA: return (family < 6 || (family == 6 && model < 0xF));
 143   }
 144   return True;
 145 }
 146
 147 #if !defined(MY_CPU_AMD64) && defined(_WIN32)
 148 static Bool CPU_Sys_Is_SSE_Supported()
 149 {
 150   OSVERSIONINFO vi;
 151   vi.dwOSVersionInfoSize = sizeof(vi);
 152   if (!GetVersionEx(&vi))
 153     return False;
 154   return (vi.dwMajorVersion >= 5);
 155 }
 156 #define CHECK_SYS_SSE_SUPPORT if (!CPU_Sys_Is_SSE_Supported()) return False;
 157 #else
 158 #define CHECK_SYS_SSE_SUPPORT
 159 #endif
 160
 161 Bool CPU_Is_Aes_Supported()
 162 {
 163   Cx86cpuid p;
 164   CHECK_SYS_SSE_SUPPORT
 165   if (!x86cpuid_CheckAndRead(&p))
 166     return False;
 167   return (p.c >> 25) & 1;
 168 }
 169
 170 #endif