]> git.proxmox.com Git - mirror_edk2.git/commitdiff
UefiCpuPkg: Enable Processor Trace feature.
authorEric Dong <eric.dong@intel.com>
Fri, 4 Aug 2017 01:59:08 +0000 (09:59 +0800)
committerEric Dong <eric.dong@intel.com>
Fri, 4 Aug 2017 04:26:44 +0000 (12:26 +0800)
Cc: Jeff Fan <jeff.fan@intel.com>
Cc: Ruiyu Ni <ruiyu.ni@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.1
Signed-off-by: Eric Dong <eric.dong@intel.com>
Reviewed-by: Jeff Fan <jeff.fan@intel.com>
UefiCpuPkg/Library/CpuCommonFeaturesLib/CpuCommonFeatures.h
UefiCpuPkg/Library/CpuCommonFeaturesLib/CpuCommonFeaturesLib.c
UefiCpuPkg/Library/CpuCommonFeaturesLib/CpuCommonFeaturesLib.inf
UefiCpuPkg/Library/CpuCommonFeaturesLib/ProcTrace.c [new file with mode: 0644]

index b8269b00f3f3f4c724eb205e5d9258a3d1a5401e..d04e2142e32329bf98b81726e03e3a3bb65e9bed 100644 (file)
@@ -961,4 +961,70 @@ LmceInitialize (
   IN BOOLEAN                           State\r
   );\r
 \r
+/**\r
+  Prepares for the data used by CPU feature detection and initialization.\r
+\r
+  @param[in]  NumberOfProcessors  The number of CPUs in the platform.\r
+\r
+  @return  Pointer to a buffer of CPU related configuration data.\r
+\r
+  @note This service could be called by BSP only.\r
+**/\r
+VOID *\r
+EFIAPI\r
+ProcTraceGetConfigData (\r
+  IN UINTN  NumberOfProcessors\r
+  );\r
+\r
+/**\r
+  Detects if Intel Processor Trace feature supported on current processor.\r
+\r
+  @param[in]  ProcessorNumber  The index of the CPU executing this function.\r
+  @param[in]  CpuInfo          A pointer to the REGISTER_CPU_FEATURE_INFORMATION\r
+                               structure for the CPU executing this function.\r
+  @param[in]  ConfigData       A pointer to the configuration buffer returned\r
+                               by CPU_FEATURE_GET_CONFIG_DATA.  NULL if\r
+                               CPU_FEATURE_GET_CONFIG_DATA was not provided in\r
+                               RegisterCpuFeature().\r
+\r
+  @retval TRUE     Processor Trace feature is supported.\r
+  @retval FALSE    Processor Trace feature is not supported.\r
+\r
+  @note This service could be called by BSP/APs.\r
+**/\r
+BOOLEAN\r
+EFIAPI\r
+ProcTraceSupport (\r
+  IN UINTN                             ProcessorNumber,\r
+  IN REGISTER_CPU_FEATURE_INFORMATION  *CpuInfo,\r
+  IN VOID                              *ConfigData  OPTIONAL\r
+  );\r
+\r
+/**\r
+  Initializes Intel Processor Trace feature to specific state.\r
+\r
+  @param[in]  ProcessorNumber  The index of the CPU executing this function.\r
+  @param[in]  CpuInfo          A pointer to the REGISTER_CPU_FEATURE_INFORMATION\r
+                               structure for the CPU executing this function.\r
+  @param[in]  ConfigData       A pointer to the configuration buffer returned\r
+                               by CPU_FEATURE_GET_CONFIG_DATA.  NULL if\r
+                               CPU_FEATURE_GET_CONFIG_DATA was not provided in\r
+                               RegisterCpuFeature().\r
+  @param[in]  State            If TRUE, then the Processor Trace feature must be\r
+                               enabled.\r
+                               If FALSE, then the Processor Trace feature must be\r
+                               disabled.\r
+\r
+  @retval RETURN_SUCCESS       Intel Processor Trace feature is initialized.\r
+\r
+**/\r
+RETURN_STATUS\r
+EFIAPI\r
+ProcTraceInitialize (\r
+  IN UINTN                             ProcessorNumber,\r
+  IN REGISTER_CPU_FEATURE_INFORMATION  *CpuInfo,\r
+  IN VOID                              *ConfigData,  OPTIONAL\r
+  IN BOOLEAN                           State\r
+  );\r
+\r
 #endif\r
index 4c78209a18bc48ced1a34c0e522d2fd68350166c..bbaffc538d8b507b89e6e2a24b033e254b07aac8 100644 (file)
@@ -229,6 +229,17 @@ CpuCommonFeaturesLibConstructor (
                );\r
     ASSERT_EFI_ERROR (Status);\r
   }\r
+  if (IsCpuFeatureSupported (CPU_FEATURE_PROC_TRACE)) {\r
+    Status = RegisterCpuFeature (\r
+               "Proc Trace",\r
+               ProcTraceGetConfigData,\r
+               ProcTraceSupport,\r
+               ProcTraceInitialize,\r
+               CPU_FEATURE_PROC_TRACE,\r
+               CPU_FEATURE_END\r
+               );\r
+    ASSERT_EFI_ERROR (Status);\r
+  }\r
 \r
   return RETURN_SUCCESS;\r
 }\r
index 2cca58e7fc0353534c7b9ebb2b70c819780f04ec..9fc3a9c86d69720c378d0572b724f1dbf76c7ee1 100644 (file)
@@ -49,6 +49,7 @@
   X2Apic.c\r
   Ppin.c\r
   Lmce.c\r
+  ProcTrace.c\r
 \r
 [Packages]\r
   MdePkg/MdePkg.dec\r
@@ -67,4 +68,5 @@
   gUefiCpuPkgTokenSpaceGuid.PcdCpuFeaturesSupport            ## CONSUMES\r
   gUefiCpuPkgTokenSpaceGuid.PcdCpuClockModulationDutyCycle   ## SOMETIMES_CONSUMES\r
   gUefiCpuPkgTokenSpaceGuid.PcdIsPowerOnReset                ## SOMETIMES_CONSUMES\r
-\r
+  gUefiCpuPkgTokenSpaceGuid.PcdCpuProcTraceOutputScheme      ## SOMETIMES_CONSUMES\r
+  gUefiCpuPkgTokenSpaceGuid.PcdCpuProcTraceMemSize           ## SOMETIMES_CONSUMES
\ No newline at end of file
diff --git a/UefiCpuPkg/Library/CpuCommonFeaturesLib/ProcTrace.c b/UefiCpuPkg/Library/CpuCommonFeaturesLib/ProcTrace.c
new file mode 100644 (file)
index 0000000..40e6321
--- /dev/null
@@ -0,0 +1,463 @@
+/** @file\r
+  Intel Processor Trace feature.\r
+\r
+  Copyright (c) 2017, Intel Corporation. All rights reserved.<BR>\r
+  This program and the accompanying materials\r
+  are licensed and made available under the terms and conditions of the BSD License\r
+  which accompanies this distribution.  The full text of the license may be found at\r
+  http://opensource.org/licenses/bsd-license.php\r
+\r
+  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
+  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+\r
+**/\r
+\r
+#include "CpuCommonFeatures.h"\r
+\r
+#define MAX_TOPA_ENTRY_COUNT 2\r
+\r
+///\r
+/// Processor trace buffer size selection.\r
+///\r
+typedef enum {\r
+  Enum4K    = 0,\r
+  Enum8K,\r
+  Enum16K,\r
+  Enum32K,\r
+  Enum64K,\r
+  Enum128K,\r
+  Enum256K,\r
+  Enum512K,\r
+  Enum1M,\r
+  Enum2M,\r
+  Enum4M,\r
+  Enum8M,\r
+  Enum16M,\r
+  Enum32M,\r
+  Enum64M,\r
+  Enum128M,\r
+  EnumProcTraceMemDisable\r
+} PROC_TRACE_MEM_SIZE;\r
+\r
+///\r
+/// Processor trace output scheme selection.\r
+///\r
+typedef enum {\r
+  OutputSchemeSingleRange = 0,\r
+  OutputSchemeToPA,\r
+  OutputSchemeInvalid\r
+} PROC_TRACE_OUTPUT_SCHEME;\r
+\r
+typedef struct  {\r
+  BOOLEAN  ProcTraceSupported;\r
+  BOOLEAN  TopaSupported;\r
+  BOOLEAN  SingleRangeSupported;\r
+} PROC_TRACE_PROCESSOR_DATA;\r
+\r
+typedef struct  {\r
+  UINT32                      NumberOfProcessors;\r
+\r
+  UINT8                       ProcTraceOutputScheme;  \r
+  UINT32                      ProcTraceMemSize;\r
+\r
+  UINTN                       *ThreadMemRegionTable;\r
+  UINTN                       AllocatedThreads;\r
+\r
+  UINTN                       *TopaMemArray;\r
+  UINTN                       TopaMemArrayCount;\r
+\r
+  PROC_TRACE_PROCESSOR_DATA   *ProcessorData;\r
+} PROC_TRACE_DATA;\r
+\r
+typedef struct {\r
+  UINT64   TopaEntry[MAX_TOPA_ENTRY_COUNT];\r
+} PROC_TRACE_TOPA_TABLE;\r
+\r
+/**\r
+  Prepares for the data used by CPU feature detection and initialization.\r
+\r
+  @param[in]  NumberOfProcessors  The number of CPUs in the platform.\r
+\r
+  @return  Pointer to a buffer of CPU related configuration data.\r
+\r
+  @note This service could be called by BSP only.\r
+**/\r
+VOID *\r
+EFIAPI\r
+ProcTraceGetConfigData (\r
+  IN UINTN  NumberOfProcessors\r
+  )\r
+{\r
+  PROC_TRACE_DATA  *ConfigData;\r
+\r
+  ConfigData = AllocateZeroPool (sizeof (PROC_TRACE_DATA) + sizeof (PROC_TRACE_PROCESSOR_DATA) * NumberOfProcessors);\r
+  ASSERT (ConfigData != NULL);\r
+  ConfigData->ProcessorData = (PROC_TRACE_PROCESSOR_DATA *) ((UINT8*) ConfigData + sizeof (PROC_TRACE_DATA));\r
+\r
+  ConfigData->NumberOfProcessors = (UINT32) NumberOfProcessors;\r
+  ConfigData->ProcTraceMemSize = PcdGet32 (PcdCpuProcTraceMemSize);\r
+  ConfigData->ProcTraceOutputScheme = PcdGet8 (PcdCpuProcTraceOutputScheme);\r
+\r
+  return ConfigData;\r
+}\r
+\r
+/**\r
+  Detects if Intel Processor Trace feature supported on current \r
+  processor.\r
+\r
+  @param[in]  ProcessorNumber  The index of the CPU executing this function.\r
+  @param[in]  CpuInfo          A pointer to the REGISTER_CPU_FEATURE_INFORMATION\r
+                               structure for the CPU executing this function.\r
+  @param[in]  ConfigData       A pointer to the configuration buffer returned\r
+                               by CPU_FEATURE_GET_CONFIG_DATA.  NULL if\r
+                               CPU_FEATURE_GET_CONFIG_DATA was not provided in\r
+                               RegisterCpuFeature().\r
+\r
+  @retval TRUE     Processor Trace feature is supported.\r
+  @retval FALSE    Processor Trace feature is not supported.\r
+\r
+  @note This service could be called by BSP/APs.\r
+**/\r
+BOOLEAN\r
+EFIAPI\r
+ProcTraceSupport (\r
+  IN UINTN                             ProcessorNumber,\r
+  IN REGISTER_CPU_FEATURE_INFORMATION  *CpuInfo,\r
+  IN VOID                              *ConfigData  OPTIONAL\r
+  )\r
+{\r
+  PROC_TRACE_DATA                             *ProcTraceData;\r
+  CPUID_STRUCTURED_EXTENDED_FEATURE_FLAGS_EBX Ebx;\r
+  CPUID_INTEL_PROCESSOR_TRACE_MAIN_LEAF_ECX   Ecx;\r
+\r
+  //\r
+  // Check if ProcTraceMemorySize option is enabled (0xFF means disable by user)\r
+  //\r
+  ProcTraceData = (PROC_TRACE_DATA *) ConfigData;\r
+  if ((ProcTraceData->ProcTraceMemSize >= EnumProcTraceMemDisable) ||\r
+      (ProcTraceData->ProcTraceOutputScheme >= OutputSchemeInvalid)) {\r
+    return FALSE;\r
+  }\r
+\r
+  //\r
+  // Check if Processor Trace is supported\r
+  //\r
+  AsmCpuidEx (CPUID_STRUCTURED_EXTENDED_FEATURE_FLAGS, 0, NULL, &Ebx.Uint32, NULL, NULL);\r
+  ProcTraceData->ProcessorData[ProcessorNumber].ProcTraceSupported = (BOOLEAN) (Ebx.Bits.IntelProcessorTrace == 1);\r
+  if (!ProcTraceData->ProcessorData[ProcessorNumber].ProcTraceSupported) {\r
+    return FALSE;\r
+  }\r
+\r
+  AsmCpuidEx (CPUID_INTEL_PROCESSOR_TRACE, CPUID_INTEL_PROCESSOR_TRACE_MAIN_LEAF, NULL, NULL, &Ecx.Uint32, NULL);\r
+  ProcTraceData->ProcessorData[ProcessorNumber].TopaSupported = (BOOLEAN) (Ecx.Bits.RTIT == 1);\r
+  ProcTraceData->ProcessorData[ProcessorNumber].SingleRangeSupported = (BOOLEAN) (Ecx.Bits.SingleRangeOutput == 1);\r
+  if (ProcTraceData->ProcessorData[ProcessorNumber].TopaSupported || \r
+      ProcTraceData->ProcessorData[ProcessorNumber].SingleRangeSupported) {\r
+    return TRUE;\r
+  }\r
+\r
+  return FALSE;\r
+}\r
+\r
+/**\r
+  Initializes Intel Processor Trace feature to specific state.\r
+\r
+  @param[in]  ProcessorNumber  The index of the CPU executing this function.\r
+  @param[in]  CpuInfo          A pointer to the REGISTER_CPU_FEATURE_INFORMATION\r
+                               structure for the CPU executing this function.\r
+  @param[in]  ConfigData       A pointer to the configuration buffer returned\r
+                               by CPU_FEATURE_GET_CONFIG_DATA.  NULL if\r
+                               CPU_FEATURE_GET_CONFIG_DATA was not provided in\r
+                               RegisterCpuFeature().\r
+  @param[in]  State            If TRUE, then the Processor Trace feature must be\r
+                               enabled.\r
+                               If FALSE, then the Processor Trace feature must be\r
+                               disabled.\r
+\r
+  @retval RETURN_SUCCESS       Intel Processor Trace feature is initialized.\r
+\r
+**/\r
+RETURN_STATUS\r
+EFIAPI\r
+ProcTraceInitialize (\r
+  IN UINTN                             ProcessorNumber,\r
+  IN REGISTER_CPU_FEATURE_INFORMATION  *CpuInfo,\r
+  IN VOID                              *ConfigData,  OPTIONAL\r
+  IN BOOLEAN                           State\r
+  )\r
+{\r
+  UINT64                               MsrValue;\r
+  UINT32                               MemRegionSize;\r
+  UINTN                                Pages;\r
+  UINTN                                Alignment;\r
+  UINTN                                MemRegionBaseAddr;\r
+  UINTN                                *ThreadMemRegionTable;\r
+  UINTN                                Index;\r
+  UINTN                                TopaTableBaseAddr;\r
+  UINTN                                AlignedAddress;\r
+  UINTN                                *TopaMemArray;\r
+  PROC_TRACE_TOPA_TABLE                *TopaTable;\r
+  PROC_TRACE_DATA                      *ProcTraceData;\r
+  BOOLEAN                              FirstIn;\r
+\r
+  ProcTraceData = (PROC_TRACE_DATA *) ConfigData;\r
+\r
+  MemRegionBaseAddr = 0;\r
+  FirstIn = FALSE;\r
+\r
+  if (ProcTraceData->ThreadMemRegionTable == NULL) {\r
+    FirstIn = TRUE;\r
+    DEBUG ((DEBUG_INFO, "Initialize Processor Trace\n"));\r
+  }\r
+\r
+  ///\r
+  /// Refer to PROC_TRACE_MEM_SIZE Table for Size Encoding\r
+  ///\r
+  MemRegionSize = (UINT32) (1 << (ProcTraceData->ProcTraceMemSize + 12));\r
+  if (FirstIn) {\r
+    DEBUG ((DEBUG_INFO, "ProcTrace: MemSize requested: 0x%X \n", MemRegionSize));\r
+  }\r
+\r
+  //\r
+  // Clear MSR_IA32_RTIT_CTL[0] and IA32_RTIT_STS only if MSR_IA32_RTIT_CTL[0]==1b\r
+  //\r
+  MsrValue = AsmReadMsr64 (MSR_IA32_RTIT_CTL);\r
+  if ((MsrValue & BIT0) != 0) {\r
+    ///\r
+    /// Clear bit 0 in MSR IA32_RTIT_CTL (570)\r
+    ///\r
+    MsrValue &= (UINT64) ~BIT0;\r
+    CPU_REGISTER_TABLE_WRITE64 (\r
+      ProcessorNumber,\r
+      Msr,\r
+      MSR_IA32_RTIT_CTL,\r
+      MsrValue\r
+      );\r
+\r
+    ///\r
+    /// Clear MSR IA32_RTIT_STS (571h) to all zeros\r
+    ///\r
+    MsrValue = AsmReadMsr64 (MSR_IA32_RTIT_STATUS);\r
+    MsrValue &= 0x0;\r
+    CPU_REGISTER_TABLE_WRITE64 (\r
+      ProcessorNumber,\r
+      Msr,\r
+      MSR_IA32_RTIT_STATUS,\r
+      MsrValue\r
+      );\r
+  }\r
+\r
+  if (FirstIn) {\r
+    //\r
+    //   Let BSP allocate and create the necessary memory region (Aligned to the size of\r
+    //   the memory region from setup option(ProcTraceMemSize) which is an integral multiple of 4kB)\r
+    //   for the all the enabled threads for storing Processor Trace debug data. Then Configure the trace\r
+    //   address base in MSR, IA32_RTIT_OUTPUT_BASE (560h) bits 47:12. Note that all regions must be\r
+    //   aligned based on their size, not just 4K. Thus a 2M region must have bits 20:12 clear.\r
+    //\r
+    ThreadMemRegionTable = (UINTN *) AllocatePool (ProcTraceData->NumberOfProcessors * sizeof (UINTN *));\r
+    if (ThreadMemRegionTable == NULL) {\r
+      DEBUG ((DEBUG_ERROR, "Allocate ProcTrace ThreadMemRegionTable Failed\n"));\r
+      return RETURN_OUT_OF_RESOURCES;\r
+    }\r
+    ProcTraceData->ThreadMemRegionTable = ThreadMemRegionTable;\r
+\r
+    for (Index = 0; Index < ProcTraceData->NumberOfProcessors; Index++, ProcTraceData->AllocatedThreads++) {\r
+      Pages = EFI_SIZE_TO_PAGES (MemRegionSize);\r
+      Alignment = MemRegionSize;\r
+      AlignedAddress = (UINTN) AllocateAlignedReservedPages (Pages, Alignment);\r
+      if (AlignedAddress == 0) {\r
+        DEBUG ((DEBUG_ERROR, "ProcTrace: Out of mem, allocated only for %d threads\n", ProcTraceData->AllocatedThreads));\r
+        if (Index == 0) {\r
+          //\r
+          // Could not allocate for BSP even\r
+          //\r
+          FreePool ((VOID *) ThreadMemRegionTable);\r
+          ThreadMemRegionTable = NULL;\r
+          return RETURN_OUT_OF_RESOURCES;\r
+        }\r
+        break;\r
+      }\r
+\r
+      ThreadMemRegionTable[Index] = AlignedAddress;\r
+      DEBUG ((DEBUG_INFO, "ProcTrace: PT MemRegionBaseAddr(aligned) for thread %d: 0x%llX \n", Index, (UINT64) ThreadMemRegionTable[Index]));\r
+    }\r
+\r
+    DEBUG ((DEBUG_INFO, "ProcTrace: Allocated PT mem for %d thread \n", ProcTraceData->AllocatedThreads));\r
+    MemRegionBaseAddr = ThreadMemRegionTable[0];\r
+  } else {\r
+    if (ProcessorNumber < ProcTraceData->AllocatedThreads) {\r
+      MemRegionBaseAddr = ProcTraceData->ThreadMemRegionTable[ProcessorNumber];\r
+    } else {\r
+      return RETURN_SUCCESS;\r
+    }\r
+  }\r
+\r
+  ///\r
+  /// Check Processor Trace output scheme: Single Range output or ToPA table\r
+  ///\r
+\r
+  //\r
+  //  Single Range output scheme\r
+  //\r
+  if (ProcTraceData->ProcessorData[ProcessorNumber].SingleRangeSupported && \r
+      (ProcTraceData->ProcTraceOutputScheme == OutputSchemeSingleRange)) {\r
+    if (FirstIn) {\r
+      DEBUG ((DEBUG_INFO, "ProcTrace: Enabling Single Range Output scheme \n"));\r
+    }\r
+\r
+    //\r
+    // Clear MSR IA32_RTIT_CTL (0x570) ToPA (Bit 8)\r
+    //\r
+    MsrValue = AsmReadMsr64 (MSR_IA32_RTIT_CTL);\r
+    MsrValue &= (UINT64) ~BIT8;\r
+    CPU_REGISTER_TABLE_WRITE64 (\r
+      ProcessorNumber,\r
+      Msr,\r
+      MSR_IA32_RTIT_CTL,\r
+      MsrValue\r
+      );\r
+\r
+    //\r
+    // Program MSR IA32_RTIT_OUTPUT_BASE (0x560) bits[47:12] with the allocated Memory Region\r
+    //\r
+    MsrValue = (UINT64) MemRegionBaseAddr;\r
+    CPU_REGISTER_TABLE_WRITE64 (\r
+      ProcessorNumber,\r
+      Msr,\r
+      MSR_IA32_RTIT_OUTPUT_BASE,\r
+      MsrValue\r
+      );\r
+\r
+    //\r
+    // Program the Mask bits for the Memory Region to MSR IA32_RTIT_OUTPUT_MASK_PTRS (561h)\r
+    //\r
+    MsrValue = (UINT64) MemRegionSize - 1;\r
+    CPU_REGISTER_TABLE_WRITE64 (\r
+      ProcessorNumber,\r
+      Msr,\r
+      MSR_IA32_RTIT_OUTPUT_MASK_PTRS,\r
+      MsrValue\r
+      );\r
+\r
+  }\r
+\r
+  //\r
+  //  ToPA(Table of physical address) scheme\r
+  //\r
+  if (ProcTraceData->ProcessorData[ProcessorNumber].TopaSupported && \r
+      (ProcTraceData->ProcTraceOutputScheme == OutputSchemeToPA)) {\r
+    //\r
+    //  Create ToPA structure aligned at 4KB for each logical thread\r
+    //  with at least 2 entries by 8 bytes size each. The first entry\r
+    //  should have the trace output base address in bits 47:12, 6:9\r
+    //  for Size, bits 4,2 and 0 must be cleared. The second entry\r
+    //  should have the base address of the table location in bits\r
+    //  47:12, bits 4 and 2 must be cleared and bit 0 must be set.\r
+    //\r
+    if (FirstIn) {\r
+      DEBUG ((DEBUG_INFO, "ProcTrace: Enabling ToPA scheme \n"));\r
+      //\r
+      // Let BSP allocate ToPA table mem for all threads\r
+      //\r
+      TopaMemArray = (UINTN *) AllocatePool (ProcTraceData->AllocatedThreads * sizeof (UINTN *));\r
+      if (TopaMemArray == NULL) {\r
+        DEBUG ((DEBUG_ERROR, "ProcTrace: Allocate mem for ToPA Failed\n"));\r
+        return RETURN_OUT_OF_RESOURCES;\r
+      }\r
+      ProcTraceData->TopaMemArray = TopaMemArray;\r
+\r
+      for (Index = 0; Index < ProcTraceData->AllocatedThreads; Index++) {\r
+        Pages = EFI_SIZE_TO_PAGES (sizeof (PROC_TRACE_TOPA_TABLE));\r
+        Alignment = 0x1000;\r
+        AlignedAddress = (UINTN) AllocateAlignedReservedPages (Pages, Alignment);\r
+        if (AlignedAddress == 0) {\r
+          if (Index < ProcTraceData->AllocatedThreads) {\r
+            ProcTraceData->AllocatedThreads = Index;\r
+          }\r
+          DEBUG ((DEBUG_ERROR, "ProcTrace:  Out of mem, allocating ToPA mem only for %d threads\n", ProcTraceData->AllocatedThreads));\r
+          if (Index == 0) {\r
+            //\r
+            // Could not allocate for BSP\r
+            //\r
+            FreePool ((VOID *) TopaMemArray);\r
+            TopaMemArray = NULL;\r
+            return RETURN_OUT_OF_RESOURCES;\r
+          }\r
+          break;\r
+        }\r
+\r
+        TopaMemArray[Index] = AlignedAddress;\r
+        DEBUG ((DEBUG_INFO, "ProcTrace: Topa table address(aligned) for thread %d is 0x%llX \n", Index,  (UINT64) TopaMemArray[Index]));\r
+      }\r
+\r
+      DEBUG ((DEBUG_INFO, "ProcTrace: Allocated ToPA mem for %d thread \n", ProcTraceData->AllocatedThreads));\r
+      //\r
+      // BSP gets the first block\r
+      //\r
+      TopaTableBaseAddr = TopaMemArray[0];\r
+    } else {\r
+      //\r
+      // Count for currently executing AP.\r
+      //\r
+      if (ProcessorNumber < ProcTraceData->AllocatedThreads) {\r
+        TopaTableBaseAddr = ProcTraceData->TopaMemArray[ProcessorNumber];\r
+      } else {\r
+        return RETURN_SUCCESS;\r
+      }\r
+    }\r
+\r
+    TopaTable = (PROC_TRACE_TOPA_TABLE *) TopaTableBaseAddr;\r
+    TopaTable->TopaEntry[0] = (UINT64) (MemRegionBaseAddr | ((ProcTraceData->ProcTraceMemSize) << 6)) & ~BIT0;\r
+    TopaTable->TopaEntry[1] = (UINT64) TopaTableBaseAddr | BIT0;\r
+\r
+    //\r
+    // Program the MSR IA32_RTIT_OUTPUT_BASE (0x560) bits[47:12] with ToPA base\r
+    //\r
+    MsrValue = (UINT64) TopaTableBaseAddr;\r
+    CPU_REGISTER_TABLE_WRITE64 (\r
+      ProcessorNumber,\r
+      Msr,\r
+      MSR_IA32_RTIT_OUTPUT_BASE,\r
+      MsrValue\r
+      );\r
+\r
+    //\r
+    // Set the MSR IA32_RTIT_OUTPUT_MASK (0x561) bits[63:7] to 0\r
+    //\r
+    CPU_REGISTER_TABLE_WRITE64 (\r
+      ProcessorNumber,\r
+      Msr,\r
+      MSR_IA32_RTIT_OUTPUT_MASK_PTRS,\r
+      0x7F\r
+      );\r
+    //\r
+    // Enable ToPA output scheme by enabling MSR IA32_RTIT_CTL (0x570) ToPA (Bit 8)\r
+    //\r
+    MsrValue = AsmReadMsr64 (MSR_IA32_RTIT_CTL);\r
+    MsrValue |= BIT8;\r
+    CPU_REGISTER_TABLE_WRITE64 (\r
+      ProcessorNumber,\r
+      Msr,\r
+      MSR_IA32_RTIT_CTL,\r
+      MsrValue\r
+      );\r
+  }\r
+\r
+  ///\r
+  /// Enable the Processor Trace feature from MSR IA32_RTIT_CTL (570h)\r
+  ///\r
+  MsrValue = AsmReadMsr64 (MSR_IA32_RTIT_CTL);\r
+  MsrValue |= (UINT64) BIT0 + BIT2 + BIT3 + BIT13;\r
+  if (!State) {\r
+    MsrValue &= (UINT64) ~BIT0;\r
+  }\r
+  CPU_REGISTER_TABLE_WRITE64 (\r
+    ProcessorNumber,\r
+    Msr,\r
+    MSR_IA32_RTIT_CTL,\r
+    MsrValue\r
+    );\r
+\r
+  return RETURN_SUCCESS;\r
+}\r