4e904348e3
SmmServicesTableLib instance only supports DXE_SMM_DRIVER type drivers that will be loaded into SMM range. InSmm() could return TRUE directly. Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Jeff Fan <jeff.fan@intel.com> Reviewed-by: Jiewen Yao <Jiewen.Yao@intel.com> git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@18495 6f19259b-4bc3-4df7-8a09-765794883524
86 lines
2.5 KiB
C
86 lines
2.5 KiB
C
/** @file
|
|
SMM Services Table Library.
|
|
|
|
Copyright (c) 2009 - 2015, 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 <PiSmm.h>
|
|
#include <Protocol/SmmBase2.h>
|
|
#include <Library/SmmServicesTableLib.h>
|
|
#include <Library/DebugLib.h>
|
|
|
|
EFI_SMM_SYSTEM_TABLE2 *gSmst = NULL;
|
|
|
|
/**
|
|
The constructor function caches the pointer of SMM Services Table.
|
|
|
|
@param ImageHandle The firmware allocated handle for the EFI image.
|
|
@param SystemTable A pointer to the EFI System Table.
|
|
|
|
@retval EFI_SUCCESS The constructor always returns EFI_SUCCESS.
|
|
|
|
**/
|
|
EFI_STATUS
|
|
EFIAPI
|
|
SmmServicesTableLibConstructor (
|
|
IN EFI_HANDLE ImageHandle,
|
|
IN EFI_SYSTEM_TABLE *SystemTable
|
|
)
|
|
{
|
|
EFI_STATUS Status;
|
|
EFI_SMM_BASE2_PROTOCOL *InternalSmmBase2;
|
|
|
|
InternalSmmBase2 = NULL;
|
|
//
|
|
// Retrieve SMM Base2 Protocol, Do not use gBS from UefiBootServicesTableLib on purpose
|
|
// to prevent inclusion of gBS, gST, and gImageHandle from SMM Drivers unless the
|
|
// SMM driver explicity declares that dependency.
|
|
//
|
|
Status = SystemTable->BootServices->LocateProtocol (
|
|
&gEfiSmmBase2ProtocolGuid,
|
|
NULL,
|
|
(VOID **)&InternalSmmBase2
|
|
);
|
|
ASSERT_EFI_ERROR (Status);
|
|
ASSERT (InternalSmmBase2 != NULL);
|
|
|
|
//
|
|
// We are in SMM, retrieve the pointer to SMM System Table
|
|
//
|
|
InternalSmmBase2->GetSmstLocation (InternalSmmBase2, &gSmst);
|
|
ASSERT (gSmst != NULL);
|
|
|
|
return EFI_SUCCESS;
|
|
}
|
|
|
|
/**
|
|
This function allows the caller to determine if the driver is executing in
|
|
System Management Mode(SMM).
|
|
|
|
This function returns TRUE if the driver is executing in SMM and FALSE if the
|
|
driver is not executing in SMM.
|
|
|
|
@retval TRUE The driver is executing in System Management Mode (SMM).
|
|
@retval FALSE The driver is not executing in System Management Mode (SMM).
|
|
|
|
**/
|
|
BOOLEAN
|
|
EFIAPI
|
|
InSmm (
|
|
VOID
|
|
)
|
|
{
|
|
//
|
|
// We are already in SMM
|
|
//
|
|
return TRUE;
|
|
}
|