OvmfPkg/SmmRelocationLib: Add library instance for OVMF
There are below 2 differences between AMD & OVMF according existing implementation: 1.The mode of the CPU check is different between the AMD & OVMF. OVMF: CpuSaveState->x86.SMMRevId & 0Xffff AMD: LMAValue = (UINT32)AsmReadMsr64 (EFER_ADDRESS) & LMA 2.Existing SmBase configuration is different between the AMD & OVMF. OVMF: if ((CpuSaveState->x86.SMMRevId & 0xFFFF) == 0) { CpuSaveState->x86.SMBASE = mSmBaseForAllCpus[CpuIndex]; } else { CpuSaveState->x64.SMBASE = mSmBaseForAllCpus[CpuIndex]; } AMD: AmdCpuState->x64.SMBASE = mSmBaseForAllCpus[CpuIndex]; This patch provides the SmmRelocationLib library instance for OVMF to handle the logic difference, and it won't change the existing implementation code logic. Cc: Ray Ni <ray.ni@intel.com> Cc: Zeng Star <star.zeng@intel.com> Cc: Ard Biesheuvel <ardb+tianocore@kernel.org> Cc: Jiewen Yao <jiewen.yao@intel.com> Cc: Gerd Hoffmann <kraxel@redhat.com> Cc: Rahul Kumar <rahul1.kumar@intel.com> Signed-off-by: Jiaxin Wu <jiaxin.wu@intel.com> Tested-by: Gerd Hoffmann <kraxel@redhat.com> Acked-by: Gerd Hoffmann <kraxel@redhat.com> Acked-by: Jiewen Yao <Jiewen.yao@intel.com>
This commit is contained in:
100
OvmfPkg/Library/SmmRelocationLib/SmramSaveStateConfig.c
Normal file
100
OvmfPkg/Library/SmmRelocationLib/SmramSaveStateConfig.c
Normal file
@@ -0,0 +1,100 @@
|
||||
/** @file
|
||||
Config SMRAM Save State for SmmBases Relocation.
|
||||
|
||||
Copyright (c) 2024, Intel Corporation. All rights reserved.<BR>
|
||||
SPDX-License-Identifier: BSD-2-Clause-Patent
|
||||
|
||||
**/
|
||||
#include "InternalSmmRelocationLib.h"
|
||||
#include <Register/Amd/SmramSaveStateMap.h>
|
||||
|
||||
/**
|
||||
This function configures the SmBase on the currently executing CPU.
|
||||
|
||||
@param[in] SmBase The SmBase on the currently executing CPU.
|
||||
|
||||
**/
|
||||
VOID
|
||||
EFIAPI
|
||||
ConfigureSmBase (
|
||||
IN UINT64 SmBase
|
||||
)
|
||||
{
|
||||
AMD_SMRAM_SAVE_STATE_MAP *CpuSaveState;
|
||||
|
||||
CpuSaveState = (AMD_SMRAM_SAVE_STATE_MAP *)(UINTN)(SMM_DEFAULT_SMBASE + SMRAM_SAVE_STATE_MAP_OFFSET);
|
||||
|
||||
if ((CpuSaveState->x86.SMMRevId & 0xFFFF) == 0) {
|
||||
CpuSaveState->x86.SMBASE = (UINT32)SmBase;
|
||||
} else {
|
||||
CpuSaveState->x64.SMBASE = (UINT32)SmBase;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
This function updates the SMRAM save state on the currently executing CPU
|
||||
to resume execution at a specific address after an RSM instruction. This
|
||||
function must evaluate the SMRAM save state to determine the execution mode
|
||||
the RSM instruction resumes and update the resume execution address with
|
||||
either NewInstructionPointer32 or NewInstructionPoint. The auto HALT restart
|
||||
flag in the SMRAM save state must always be cleared. This function returns
|
||||
the value of the instruction pointer from the SMRAM save state that was
|
||||
replaced. If this function returns 0, then the SMRAM save state was not
|
||||
modified.
|
||||
|
||||
This function is called during the very first SMI on each CPU after
|
||||
SmmCpuFeaturesInitializeProcessor() to set a flag in normal execution mode
|
||||
to signal that the SMBASE of each CPU has been updated before the default
|
||||
SMBASE address is used for the first SMI to the next CPU.
|
||||
|
||||
@param[in,out] CpuState Pointer to SMRAM Save State Map for the
|
||||
currently executing CPU.
|
||||
@param[in] NewInstructionPointer32 Instruction pointer to use if resuming to
|
||||
32-bit mode from 64-bit SMM.
|
||||
@param[in] NewInstructionPointer Instruction pointer to use if resuming to
|
||||
same mode as SMM.
|
||||
|
||||
@retval The value of the original instruction pointer before it was hooked.
|
||||
|
||||
**/
|
||||
UINT64
|
||||
EFIAPI
|
||||
HookReturnFromSmm (
|
||||
IN OUT SMRAM_SAVE_STATE_MAP *CpuState,
|
||||
IN UINT64 NewInstructionPointer32,
|
||||
IN UINT64 NewInstructionPointer
|
||||
)
|
||||
{
|
||||
UINT64 OriginalInstructionPointer;
|
||||
AMD_SMRAM_SAVE_STATE_MAP *CpuSaveState;
|
||||
|
||||
CpuSaveState = (AMD_SMRAM_SAVE_STATE_MAP *)CpuState;
|
||||
if ((CpuSaveState->x86.SMMRevId & 0xFFFF) == 0) {
|
||||
OriginalInstructionPointer = (UINT64)CpuSaveState->x86._EIP;
|
||||
CpuSaveState->x86._EIP = (UINT32)NewInstructionPointer;
|
||||
//
|
||||
// Clear the auto HALT restart flag so the RSM instruction returns
|
||||
// program control to the instruction following the HLT instruction.
|
||||
//
|
||||
if ((CpuSaveState->x86.AutoHALTRestart & BIT0) != 0) {
|
||||
CpuSaveState->x86.AutoHALTRestart &= ~BIT0;
|
||||
}
|
||||
} else {
|
||||
OriginalInstructionPointer = CpuSaveState->x64._RIP;
|
||||
if ((CpuSaveState->x64.EFER & LMA) == 0) {
|
||||
CpuSaveState->x64._RIP = (UINT32)NewInstructionPointer32;
|
||||
} else {
|
||||
CpuSaveState->x64._RIP = (UINT32)NewInstructionPointer;
|
||||
}
|
||||
|
||||
//
|
||||
// Clear the auto HALT restart flag so the RSM instruction returns
|
||||
// program control to the instruction following the HLT instruction.
|
||||
//
|
||||
if ((CpuSaveState->x64.AutoHALTRestart & BIT0) != 0) {
|
||||
CpuSaveState->x64.AutoHALTRestart &= ~BIT0;
|
||||
}
|
||||
}
|
||||
|
||||
return OriginalInstructionPointer;
|
||||
}
|
Reference in New Issue
Block a user