summaryrefslogtreecommitdiff
path: root/MdePkg/Library/BaseLib/X64/TdProbe.c
blob: b893c0a004d59befaa6e2e0d425308a6d4c7e451 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
/** @file

  Copyright (c) 2020-2021, Intel Corporation. All rights reserved.<BR>
  SPDX-License-Identifier: BSD-2-Clause-Patent

**/

#include <Library/BaseLib.h>
#include <Register/Intel/Cpuid.h>

/**
  Probe if TD is enabled.

  @return TRUE    TD is enabled.
  @return FALSE   TD is not enabled.
**/
BOOLEAN
EFIAPI
TdIsEnabled (
  )
{
  UINT32                  Eax;
  UINT32                  Ebx;
  UINT32                  Ecx;
  UINT32                  Edx;
  UINT32                  LargestEax;
  BOOLEAN                 TdEnabled;
  CPUID_VERSION_INFO_ECX  CpuIdVersionInfoEcx;

  TdEnabled = FALSE;

  do {
    AsmCpuid (CPUID_SIGNATURE, &LargestEax, &Ebx, &Ecx, &Edx);

    if (  (Ebx != CPUID_SIGNATURE_GENUINE_INTEL_EBX)
       || (Edx != CPUID_SIGNATURE_GENUINE_INTEL_EDX)
       || (Ecx != CPUID_SIGNATURE_GENUINE_INTEL_ECX))
    {
      break;
    }

    AsmCpuid (CPUID_VERSION_INFO, NULL, NULL, &CpuIdVersionInfoEcx.Uint32, NULL);
    if (CpuIdVersionInfoEcx.Bits.ParaVirtualized == 0) {
      break;
    }

    if (LargestEax < CPUID_GUESTTD_RUNTIME_ENVIRONMENT) {
      break;
    }

    AsmCpuidEx (CPUID_GUESTTD_RUNTIME_ENVIRONMENT, 0, &Eax, &Ebx, &Ecx, &Edx);
    if (  (Ebx != CPUID_GUESTTD_SIGNATURE_GENUINE_INTEL_EBX)
       || (Edx != CPUID_GUESTTD_SIGNATURE_GENUINE_INTEL_EDX)
       || (Ecx != CPUID_GUESTTD_SIGNATURE_GENUINE_INTEL_ECX))
    {
      break;
    }

    TdEnabled = TRUE;
  } while (FALSE);

  return TdEnabled;
}