测试来自 C# 的 AES-NI 指令

Question

我想知道是否有一种方法可以通过 C#.NET 测试主机系统 CPU 中是否存在 AES-NI。

让我先说一下，这个问题不是询问如何使用来自 .NET 的 AES-NI。事实证明，简单地使用AESCryptoServiceProvider 将使用 AES-NI（如果可用）。该结果基于我将AESCryptoServiceProvider 的性能与 TrueCrypt 中提供的基准进行比较的独立基准，TrueCrypt 确实支持 AES-NI。在使用和不使用 AES-NI 的两台机器上，结果惊人地相似。

我希望能够对其进行测试的原因是能够向用户表明他们的计算机支持 AES-NI。这将是相关的，因为它可以减少涉及诸如“但我的朋友也有 Core i5 但他的速度要快得多！”之类的问题的支持事件。如果程序的用户界面可以向用户表明他们的系统支持或不支持 AES-NI，那么它也可能表明“性能较慢是正常的，因为该系统不支持 AES-NI。”

（我们要感谢英特尔对不同处理器步进的所有困惑！:-)）

有没有办法检测这些信息，也许是通过 WMI？

Answer 1

似乎在 SO:Inline Assembly Code to Get CPU ID 上有类似的问题，答案很好。

但是这个答案需要根据你的需要进行一些调整。

首先，据我了解，AES-NI 只能存在于 64 位处理器上，对吗？那么您可以忽略上述答案中的所有 32 位代码。

其次，您需要 ECX 寄存器，或者更确切地说是它的第 25 位，因此您必须稍微更改代码：

private static bool IsAESNIPresent()
{
    byte[] sn = new byte[16]; // !!! Here were 8 bytes

    if (!ExecuteCode(ref sn))
        return false;

    var ecx = BitConverter.ToUInt32(sn, 8);
    return (ecx & (1 << 25)) != 0;
}

最后，您需要将 ECX 寄存器存储在数组中：

byte[] code_x64 = new byte[] {
    0x53,                                     /* push rbx */
    0x48, 0xc7, 0xc0, 0x01, 0x00, 0x00, 0x00, /* mov rax, 0x1 */
    0x0f, 0xa2,                               /* cpuid */
    0x41, 0x89, 0x00,                         /* mov [r8], eax */
    0x41, 0x89, 0x50, 0x04,                   /* mov [r8+0x4], ebx !!! changed */
    0x41, 0x89, 0x50, 0x08,                   /* mov [r8+0x8], ecx !!! added */
    0x41, 0x89, 0x50, 0x0C,                   /* mov [r8+0xC], edx !!! added*/
    0x5b,                                     /* pop rbx */
    0xc3,                                     /* ret */
};

据我所知，这就是所有变化。

Answer 2

上面 Mark 的回答非常棒，对我来说效果很好，但是我确实注意到，如果应用程序在 32 位模式下运行，则不会在 x86 代码中提取 ecx 寄存器，从而导致无法检测到AES-NI。

我添加了一行并更改了另一行，基本上是将 Mark 对 x64 代码所做的更改应用到 x86 代码。这也允许您从 32 位模式查看 AES-NI 位。不确定它是否会帮助某人，但我想我会发布它。

编辑：当我进行一些测试时，我注意到 x64 代码返回的寄存器不正确。 EDX 在偏移量 0x4、0x8 和 0xC 处返回，此外，ECX 和 EDX 寄存器在 x86 代码中的偏移量不同，因此您需要更频繁地检查 IntPtr.Size 以保持在这两种环境中的工作。为了简化操作，我将 ECX 寄存器放置在 0x4 并将 EDX 放置在 0x8，这样数据就可以正确排列。

如果有人要求，我可以发布整个课程，这是我从这篇帖子和其他帖子中学到的一个工作示例。

public static bool ExecuteCode(ref byte[] result) {
    byte[] code_x86 = new byte[] {
        0x55,                      /* push ebp */
        0x89, 0xE5,                /* mov  ebp, esp */
        0x57,                      /* push edi */
        0x8b, 0x7D, 0x10,          /* mov  edi, [ebp+0x10] */
        0x6A, 0x01,                /* push 0x1 */
        0x58,                      /* pop  eax */
        0x53,                      /* push ebx */
        0x0F, 0xA2,                /* cpuid    */
        0x89, 0x07,                /* mov  [edi], eax */
        0x89, 0x4F, 0x04,          /* mov  [edi+0x4], ecx Changed */
        0x89, 0x57, 0x08,          /* mov  [edi+0x8], edx Changed */
        0x5B,                      /* pop  ebx */
        0x5F,                      /* pop  edi */
        0x89, 0xEC,                /* mov  esp, ebp */
        0x5D,                      /* pop  ebp */
        0xC2, 0x10, 0x00,          /* ret  0x10 */
    };
    byte[] code_x64 = new byte[] {
        0x53,                                     /* push rbx */
        0x48, 0xC7, 0xC0, 0x01, 0x00, 0x00, 0x00, /* mov rax, 0x1 */
        0x0f, 0xA2,                               /* cpuid */
        0x41, 0x89, 0x00,                         /* mov [r8], eax */
        0x41, 0x89, 0x48, 0x04,                   /* mov [r8+0x4], ecx Changed */
        0x41, 0x89, 0x50, 0x08,                   /* mov [r8+0x8], edx Changed*/
        0x5B,                                     /* pop rbx */
        0xC3,                                     /* ret */
    };

    int num;
    byte[] code = (IntPtr.Size == 4) ? code_x86 : code_x64;
    IntPtr ptr = new IntPtr(code.Length);

    if (!VirtualProtect(code, ptr, PAGE_EXECUTE_READWRITE, out num))
        Marshal.ThrowExceptionForHR(Marshal.GetHRForLastWin32Error());

    ptr = new IntPtr(result.Length);

    return (ExecuteNativeCode(code, IntPtr.Zero, 0, result, ptr) != IntPtr.Zero);