Hardware/CPU/CPUID.cs
author moel.mich
Fri, 30 Apr 2010 20:53:40 +0000
changeset 105 f46d163ffba2
parent 102 9620449d2620
child 112 69d29f1a2307
permissions -rw-r--r--
Fixed an overflow problem with the log function in the CPUID constructor.
moel@90
     1
/*
moel@90
     2
  
moel@90
     3
  Version: MPL 1.1/GPL 2.0/LGPL 2.1
moel@90
     4
moel@90
     5
  The contents of this file are subject to the Mozilla Public License Version
moel@90
     6
  1.1 (the "License"); you may not use this file except in compliance with
moel@90
     7
  the License. You may obtain a copy of the License at
moel@90
     8
 
moel@90
     9
  http://www.mozilla.org/MPL/
moel@90
    10
moel@90
    11
  Software distributed under the License is distributed on an "AS IS" basis,
moel@90
    12
  WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
moel@90
    13
  for the specific language governing rights and limitations under the License.
moel@90
    14
moel@90
    15
  The Original Code is the Open Hardware Monitor code.
moel@90
    16
moel@90
    17
  The Initial Developer of the Original Code is 
moel@90
    18
  Michael Möller <m.moeller@gmx.ch>.
moel@90
    19
  Portions created by the Initial Developer are Copyright (C) 2009-2010
moel@90
    20
  the Initial Developer. All Rights Reserved.
moel@90
    21
moel@90
    22
  Contributor(s):
moel@90
    23
moel@90
    24
  Alternatively, the contents of this file may be used under the terms of
moel@90
    25
  either the GNU General Public License Version 2 or later (the "GPL"), or
moel@90
    26
  the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
moel@90
    27
  in which case the provisions of the GPL or the LGPL are applicable instead
moel@90
    28
  of those above. If you wish to allow use of your version of this file only
moel@90
    29
  under the terms of either the GPL or the LGPL, and not to allow others to
moel@90
    30
  use your version of this file under the terms of the MPL, indicate your
moel@90
    31
  decision by deleting the provisions above and replace them with the notice
moel@90
    32
  and other provisions required by the GPL or the LGPL. If you do not delete
moel@90
    33
  the provisions above, a recipient may use your version of this file under
moel@90
    34
  the terms of any one of the MPL, the GPL or the LGPL.
moel@90
    35
 
moel@90
    36
*/
moel@90
    37
moel@90
    38
using System;
moel@90
    39
using System.Collections.Generic;
moel@90
    40
using System.Text;
moel@90
    41
moel@90
    42
namespace OpenHardwareMonitor.Hardware.CPU {
moel@90
    43
  
moel@90
    44
  public enum Vendor {
moel@90
    45
    Unknown,
moel@90
    46
    Intel,
moel@90
    47
    AMD,
moel@90
    48
  }
moel@90
    49
  
moel@90
    50
  public class CPUID {
moel@90
    51
moel@90
    52
    private int thread;
moel@90
    53
moel@90
    54
    private uint maxCpuid = 0;
moel@90
    55
    private uint maxCpuidExt = 0;
moel@90
    56
moel@90
    57
    private Vendor vendor;
moel@90
    58
moel@90
    59
    private string cpuBrandString;
moel@90
    60
    private string name;
moel@90
    61
moel@90
    62
    private uint[,] cpuidData;
moel@90
    63
    private uint[,] cpuidExtData;
moel@90
    64
moel@90
    65
    private uint family;
moel@90
    66
    private uint model;
moel@90
    67
    private uint stepping;
moel@90
    68
moel@90
    69
    private uint apicId;
moel@90
    70
moel@90
    71
    private uint threadMaskWith;
moel@90
    72
    private uint coreMaskWith;
moel@90
    73
moel@90
    74
    private uint processorId;
moel@90
    75
    private uint coreId;
moel@90
    76
    private uint threadId;
moel@90
    77
moel@90
    78
    public static uint CPUID_0 = 0;
moel@90
    79
    public static uint CPUID_EXT = 0x80000000;
moel@90
    80
moel@90
    81
    private void AppendRegister(StringBuilder b, uint value) {
moel@90
    82
      b.Append((char)((value) & 0xff));
moel@90
    83
      b.Append((char)((value >> 8) & 0xff));
moel@90
    84
      b.Append((char)((value >> 16) & 0xff));
moel@90
    85
      b.Append((char)((value >> 24) & 0xff));
moel@90
    86
    }
moel@90
    87
moel@105
    88
    private uint NextLog2(long x) {
moel@105
    89
      if (x <= 0)
moel@105
    90
        return 0;
moel@105
    91
moel@105
    92
      x--;
moel@105
    93
      uint count = 0;
moel@105
    94
      while (x > 0) {
moel@105
    95
        x >>= 1;
moel@105
    96
        count++;
moel@105
    97
      }
moel@105
    98
moel@105
    99
      return count;
moel@105
   100
    }
moel@105
   101
moel@90
   102
    public CPUID(int thread) {
moel@90
   103
      this.thread = thread;
moel@90
   104
moel@90
   105
      uint eax, ebx, ecx, edx;
moel@90
   106
moel@90
   107
      UIntPtr mask = (UIntPtr)(1L << thread);
moel@90
   108
moel@90
   109
      if (WinRing0.CpuidTx(CPUID_0, 0,
moel@90
   110
          out eax, out ebx, out ecx, out edx, mask)) {
moel@90
   111
        maxCpuid = eax;
moel@90
   112
        StringBuilder vendorBuilder = new StringBuilder();
moel@90
   113
        AppendRegister(vendorBuilder, ebx);
moel@90
   114
        AppendRegister(vendorBuilder, edx);
moel@90
   115
        AppendRegister(vendorBuilder, ecx);
moel@90
   116
        string cpuVendor = vendorBuilder.ToString();
moel@90
   117
        switch (cpuVendor) {
moel@90
   118
          case "GenuineIntel":
moel@90
   119
            vendor = Vendor.Intel;
moel@90
   120
            break;
moel@90
   121
          case "AuthenticAMD":
moel@90
   122
            vendor = Vendor.AMD;
moel@90
   123
            break;
moel@90
   124
          default:
moel@90
   125
            vendor = Vendor.Unknown;
moel@90
   126
            break;
moel@90
   127
        }
moel@90
   128
        eax = ebx = ecx = edx = 0;
moel@90
   129
        if (WinRing0.CpuidTx(CPUID_EXT, 0,
moel@90
   130
          out eax, out ebx, out ecx, out edx, mask))
moel@90
   131
          maxCpuidExt = eax - CPUID_EXT;
moel@90
   132
      } else {
moel@90
   133
        throw new ArgumentException();
moel@90
   134
      }
moel@90
   135
moel@90
   136
      if (maxCpuid == 0 || maxCpuidExt == 0)
moel@90
   137
        return;
moel@90
   138
moel@90
   139
      cpuidData = new uint[maxCpuid + 1, 4];
moel@90
   140
      for (uint i = 0; i < (maxCpuid + 1); i++)
moel@90
   141
        WinRing0.CpuidTx(CPUID_0 + i, 0, 
moel@90
   142
          out cpuidData[i, 0], out cpuidData[i, 1],
moel@90
   143
          out cpuidData[i, 2], out cpuidData[i, 3], mask);
moel@90
   144
moel@95
   145
      cpuidExtData = new uint[maxCpuidExt + 1, 4];
moel@95
   146
      for (uint i = 0; i < (maxCpuidExt + 1); i++)
moel@90
   147
        WinRing0.CpuidTx(CPUID_EXT + i, 0, 
moel@90
   148
          out cpuidExtData[i, 0], out cpuidExtData[i, 1], 
moel@90
   149
          out cpuidExtData[i, 2], out cpuidExtData[i, 3], mask);
moel@90
   150
moel@90
   151
      StringBuilder nameBuilder = new StringBuilder();
moel@90
   152
      for (uint i = 2; i <= 4; i++) {
moel@90
   153
        if (WinRing0.CpuidTx(CPUID_EXT + i, 0, 
moel@90
   154
          out eax, out ebx, out ecx, out edx, mask)) 
moel@90
   155
        {
moel@90
   156
          AppendRegister(nameBuilder, eax);
moel@90
   157
          AppendRegister(nameBuilder, ebx);
moel@90
   158
          AppendRegister(nameBuilder, ecx);
moel@90
   159
          AppendRegister(nameBuilder, edx);
moel@90
   160
        }
moel@90
   161
      }
moel@90
   162
      nameBuilder.Replace('\0', ' ');
moel@90
   163
      cpuBrandString = nameBuilder.ToString().Trim();
moel@90
   164
      nameBuilder.Replace("(R)", " ");
moel@90
   165
      nameBuilder.Replace("(TM)", " ");
moel@90
   166
      nameBuilder.Replace("(tm)", " ");
moel@90
   167
      nameBuilder.Replace("CPU", "");
moel@90
   168
      for (int i = 0; i < 10; i++) nameBuilder.Replace("  ", " ");
moel@90
   169
      name = nameBuilder.ToString();
moel@90
   170
      if (name.Contains("@"))
moel@90
   171
        name = name.Remove(name.LastIndexOf('@'));
moel@95
   172
      name = name.Trim();      
moel@90
   173
moel@90
   174
      this.family = ((cpuidData[1, 0] & 0x0FF00000) >> 20) +
moel@90
   175
        ((cpuidData[1, 0] & 0x0F00) >> 8);
moel@90
   176
      this.model = ((cpuidData[1, 0] & 0x0F0000) >> 12) +
moel@90
   177
        ((cpuidData[1, 0] & 0xF0) >> 4);
moel@90
   178
      this.stepping = (cpuidData[1, 0] & 0x0F);
moel@90
   179
moel@90
   180
      this.apicId = (cpuidData[1, 1] >> 24) & 0xFF;
moel@90
   181
moel@90
   182
      switch (vendor) {
moel@90
   183
        case Vendor.Intel:
moel@90
   184
          uint maxCoreAndThreadIdPerPackage = (cpuidData[1, 1] >> 16) & 0xFF;
moel@95
   185
          uint maxCoreIdPerPackage;
moel@95
   186
          if (maxCpuid >= 4)
moel@95
   187
            maxCoreIdPerPackage = ((cpuidData[4, 0] >> 26) & 0x3F) + 1;
moel@95
   188
          else
moel@95
   189
            maxCoreIdPerPackage = 1;
moel@105
   190
          threadMaskWith = 
moel@105
   191
            NextLog2(maxCoreAndThreadIdPerPackage / maxCoreIdPerPackage);
moel@105
   192
          coreMaskWith = NextLog2(maxCoreIdPerPackage);
moel@90
   193
          break;
moel@90
   194
        case Vendor.AMD:
moel@95
   195
          uint corePerPackage;
moel@95
   196
          if (maxCpuidExt >= 8)
moel@95
   197
            corePerPackage = (cpuidExtData[8, 2] & 0xFF) + 1;
moel@95
   198
          else
moel@95
   199
            corePerPackage = 1;
moel@90
   200
          threadMaskWith = 0;
moel@105
   201
          coreMaskWith = NextLog2(corePerPackage);
moel@90
   202
          break;
moel@90
   203
        default:
moel@90
   204
          threadMaskWith = 0;
moel@90
   205
          coreMaskWith = 0;
moel@90
   206
          break;
moel@90
   207
      }
moel@90
   208
moel@90
   209
      processorId = (uint)(apicId >> (int)(coreMaskWith + threadMaskWith));
moel@95
   210
      coreId = (uint)((apicId >> (int)(threadMaskWith)) 
moel@102
   211
        - (processorId << (int)(coreMaskWith)));
moel@95
   212
      threadId = apicId
moel@95
   213
        - (processorId << (int)(coreMaskWith + threadMaskWith))
moel@95
   214
        - (coreId << (int)(threadMaskWith)); 
moel@90
   215
    }
moel@90
   216
moel@90
   217
    public string Name {
moel@90
   218
      get { return name; }
moel@90
   219
    }
moel@90
   220
moel@90
   221
    public string BrandString {
moel@90
   222
      get { return cpuBrandString; }
moel@90
   223
    }
moel@90
   224
moel@90
   225
    public int Thread {
moel@90
   226
      get { return thread; }
moel@90
   227
    }
moel@90
   228
moel@90
   229
    public uint MaxCPUID {
moel@90
   230
      get { return maxCpuid; }
moel@90
   231
    }
moel@90
   232
moel@90
   233
    public uint MaxCpuidExt {
moel@90
   234
      get { return maxCpuidExt; }
moel@90
   235
    }
moel@90
   236
moel@90
   237
    public Vendor Vendor {
moel@90
   238
      get { return vendor; }
moel@90
   239
    }
moel@90
   240
moel@90
   241
    public uint Family {
moel@90
   242
      get { return family; }
moel@90
   243
    }
moel@90
   244
moel@90
   245
    public uint Model {
moel@90
   246
      get { return model; }
moel@90
   247
    }
moel@90
   248
moel@90
   249
    public uint Stepping {
moel@90
   250
      get { return stepping; }
moel@90
   251
    }
moel@90
   252
moel@90
   253
    public uint ApicId {
moel@90
   254
      get { return apicId; }
moel@90
   255
    }
moel@90
   256
moel@90
   257
    public uint ProcessorId {
moel@90
   258
      get { return processorId; }
moel@90
   259
    }
moel@90
   260
moel@90
   261
    public uint CoreId {
moel@90
   262
      get { return coreId; }
moel@90
   263
    }
moel@90
   264
moel@90
   265
    public uint ThreadId {
moel@90
   266
      get { return threadId; }
moel@90
   267
    }
moel@90
   268
moel@90
   269
    public uint[,] Data {
moel@90
   270
      get { return cpuidData; }
moel@90
   271
    }
moel@90
   272
moel@90
   273
    public uint[,] ExtData {
moel@90
   274
      get { return cpuidExtData; }
moel@90
   275
    }
moel@90
   276
  }
moel@90
   277
}