With correct model CPU, current checking logic will always execute AsmReadMsr64 operation and then check ECX.AESNI[bit 25] = 1. Update checking logic to check ECX.AESNI[bit 25] = 1 first and then do AsmReadMsr64 operation. Cc: Eric Dong <eric.dong@intel.com> Cc: Laszlo Ersek <lersek@redhat.com> Contributed-under: TianoCore Contribution Agreement 1.1 Signed-off-by: Bell Song <binx.song@intel.com> Reviewed-by: Eric Dong <eric.dong@intel.com>
		
			
				
	
	
		
			132 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			132 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /** @file
 | |
|   AESNI feature.
 | |
| 
 | |
|   Copyright (c) 2017, Intel Corporation. All rights reserved.<BR>
 | |
|   This program and the accompanying materials
 | |
|   are licensed and made available under the terms and conditions of the BSD License
 | |
|   which accompanies this distribution.  The full text of the license may be found at
 | |
|   http://opensource.org/licenses/bsd-license.php
 | |
| 
 | |
|   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
 | |
|   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 | |
| 
 | |
| **/
 | |
| 
 | |
| #include "CpuCommonFeatures.h"
 | |
| 
 | |
| /**
 | |
|   Prepares for the data used by CPU feature detection and initialization.
 | |
| 
 | |
|   @param[in]  NumberOfProcessors  The number of CPUs in the platform.
 | |
| 
 | |
|   @return  Pointer to a buffer of CPU related configuration data.
 | |
| 
 | |
|   @note This service could be called by BSP only.
 | |
| **/
 | |
| VOID *
 | |
| EFIAPI
 | |
| AesniGetConfigData (
 | |
|   IN UINTN  NumberOfProcessors
 | |
|   )
 | |
| {
 | |
|   UINT64                            *ConfigData;
 | |
| 
 | |
|   ConfigData = AllocateZeroPool (sizeof (UINT64) * NumberOfProcessors);
 | |
|   ASSERT (ConfigData != NULL);
 | |
|   return ConfigData;
 | |
| }
 | |
| 
 | |
| /**
 | |
|   Detects if AESNI feature supported on current processor.
 | |
| 
 | |
|   @param[in]  ProcessorNumber  The index of the CPU executing this function.
 | |
|   @param[in]  CpuInfo          A pointer to the REGISTER_CPU_FEATURE_INFORMATION
 | |
|                                structure for the CPU executing this function.
 | |
|   @param[in]  ConfigData       A pointer to the configuration buffer returned
 | |
|                                by CPU_FEATURE_GET_CONFIG_DATA.  NULL if
 | |
|                                CPU_FEATURE_GET_CONFIG_DATA was not provided in
 | |
|                                RegisterCpuFeature().
 | |
| 
 | |
|   @retval TRUE     AESNI feature is supported.
 | |
|   @retval FALSE    AESNI feature is not supported.
 | |
| 
 | |
|   @note This service could be called by BSP/APs.
 | |
| **/
 | |
| BOOLEAN
 | |
| EFIAPI
 | |
| AesniSupport (
 | |
|   IN UINTN                             ProcessorNumber,
 | |
|   IN REGISTER_CPU_FEATURE_INFORMATION  *CpuInfo,
 | |
|   IN VOID                              *ConfigData  OPTIONAL
 | |
|   )
 | |
| {
 | |
|   MSR_SANDY_BRIDGE_FEATURE_CONFIG_REGISTER   *MsrFeatureConfig;
 | |
| 
 | |
|   if (CpuInfo->CpuIdVersionInfoEcx.Bits.AESNI == 1) {
 | |
|     if (IS_SANDY_BRIDGE_PROCESSOR (CpuInfo->DisplayFamily, CpuInfo->DisplayModel) ||
 | |
|         IS_SILVERMONT_PROCESSOR (CpuInfo->DisplayFamily, CpuInfo->DisplayModel) ||
 | |
|         IS_XEON_5600_PROCESSOR (CpuInfo->DisplayFamily, CpuInfo->DisplayModel) ||
 | |
|         IS_XEON_E7_PROCESSOR (CpuInfo->DisplayFamily, CpuInfo->DisplayModel) ||
 | |
|         IS_XEON_PHI_PROCESSOR (CpuInfo->DisplayFamily, CpuInfo->DisplayModel)) {
 | |
|       MsrFeatureConfig = (MSR_SANDY_BRIDGE_FEATURE_CONFIG_REGISTER *) ConfigData;
 | |
|       ASSERT (MsrFeatureConfig != NULL);
 | |
|       MsrFeatureConfig[ProcessorNumber].Uint64 = AsmReadMsr64 (MSR_SANDY_BRIDGE_FEATURE_CONFIG);
 | |
|     }
 | |
|     return TRUE;
 | |
|   }
 | |
|   return FALSE;
 | |
| }
 | |
| 
 | |
| /**
 | |
|   Initializes AESNI feature to specific state.
 | |
| 
 | |
|   @param[in]  ProcessorNumber  The index of the CPU executing this function.
 | |
|   @param[in]  CpuInfo          A pointer to the REGISTER_CPU_FEATURE_INFORMATION
 | |
|                                structure for the CPU executing this function.
 | |
|   @param[in]  ConfigData       A pointer to the configuration buffer returned
 | |
|                                by CPU_FEATURE_GET_CONFIG_DATA.  NULL if
 | |
|                                CPU_FEATURE_GET_CONFIG_DATA was not provided in
 | |
|                                RegisterCpuFeature().
 | |
|   @param[in]  State            If TRUE, then the AESNI feature must be enabled.
 | |
|                                If FALSE, then the AESNI feature must be disabled.
 | |
| 
 | |
|   @retval RETURN_SUCCESS       AESNI feature is initialized.
 | |
| 
 | |
|   @note This service could be called by BSP only.
 | |
| **/
 | |
| RETURN_STATUS
 | |
| EFIAPI
 | |
| AesniInitialize (
 | |
|   IN UINTN                             ProcessorNumber,
 | |
|   IN REGISTER_CPU_FEATURE_INFORMATION  *CpuInfo,
 | |
|   IN VOID                              *ConfigData,  OPTIONAL
 | |
|   IN BOOLEAN                           State
 | |
|   )
 | |
| {
 | |
|   MSR_SANDY_BRIDGE_FEATURE_CONFIG_REGISTER   *MsrFeatureConfig;
 | |
| 
 | |
|   //
 | |
|   // SANDY_BRIDGE, SILVERMONT, XEON_5600, XEON_7, and XEON_PHI have the same MSR index,
 | |
|   // Simply use MSR_SANDY_BRIDGE_FEATURE_CONFIG here
 | |
|   //
 | |
|   // The scope of the MSR_SANDY_BRIDGE_FEATURE_CONFIG is Core, only program MSR_FEATURE_CONFIG for thread 0
 | |
|   // of each core. Otherwise, once a thread in the core disabled AES, the other thread will cause GP when
 | |
|   // programming it.
 | |
|   //
 | |
|   if (CpuInfo->ProcessorInfo.Location.Thread == 0) {
 | |
|     MsrFeatureConfig = (MSR_SANDY_BRIDGE_FEATURE_CONFIG_REGISTER *) ConfigData;
 | |
|     ASSERT (MsrFeatureConfig != NULL);
 | |
|     if ((MsrFeatureConfig[ProcessorNumber].Bits.AESConfiguration & BIT0) == 0) {
 | |
|       CPU_REGISTER_TABLE_WRITE_FIELD (
 | |
|         ProcessorNumber,
 | |
|         Msr,
 | |
|         MSR_SANDY_BRIDGE_FEATURE_CONFIG,
 | |
|         MSR_SANDY_BRIDGE_FEATURE_CONFIG_REGISTER,
 | |
|         Bits.AESConfiguration,
 | |
|         BIT1 | ((State) ? 0 : BIT0)
 | |
|         );
 | |
|     }
 | |
|   }
 | |
|   return RETURN_SUCCESS;
 | |
| }
 |