/** @file Support routines for memory profile for PiSmmCore. Copyright (c) 2016 - 2018, Intel Corporation. All rights reserved.
SPDX-License-Identifier: BSD-2-Clause-Patent **/ #include #include #include #include #include "PiSmmCoreMemoryProfileServices.h" EDKII_MEMORY_PROFILE_PROTOCOL *mLibProfileProtocol; /** Check whether the start address of buffer is within any of the SMRAM ranges. @param[in] Buffer The pointer to the buffer to be checked. @retval TRUE The buffer is in SMRAM ranges. @retval FALSE The buffer is out of SMRAM ranges. **/ BOOLEAN EFIAPI BufferInSmram ( IN VOID *Buffer ); /** The constructor function initializes memory profile for SMM phase. @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 PiSmmCoreMemoryProfileLibConstructor ( IN EFI_HANDLE ImageHandle, IN EFI_SYSTEM_TABLE *SystemTable ) { EFI_STATUS Status; // // Locate Profile Protocol // Status = gBS->LocateProtocol ( &gEdkiiMemoryProfileGuid, NULL, (VOID **)&mLibProfileProtocol ); if (EFI_ERROR (Status)) { mLibProfileProtocol = NULL; } return EFI_SUCCESS; } /** Record memory profile of multilevel caller. @param[in] CallerAddress Address of caller. @param[in] Action Memory profile action. @param[in] MemoryType Memory type. EfiMaxMemoryType means the MemoryType is unknown. @param[in] Buffer Buffer address. @param[in] Size Buffer size. @param[in] ActionString String for memory profile action. Only needed for user defined allocate action. @return EFI_SUCCESS Memory profile is updated. @return EFI_UNSUPPORTED Memory profile is unsupported, or memory profile for the image is not required, or memory profile for the memory type is not required. @return EFI_ACCESS_DENIED It is during memory profile data getting. @return EFI_ABORTED Memory profile recording is not enabled. @return EFI_OUT_OF_RESOURCES No enough resource to update memory profile for allocate action. @return EFI_NOT_FOUND No matched allocate info found for free action. **/ EFI_STATUS EFIAPI MemoryProfileLibRecord ( IN PHYSICAL_ADDRESS CallerAddress, IN MEMORY_PROFILE_ACTION Action, IN EFI_MEMORY_TYPE MemoryType, IN VOID *Buffer, IN UINTN Size, IN CHAR8 *ActionString OPTIONAL ) { if (BufferInSmram (Buffer)) { return SmmCoreUpdateProfile (CallerAddress, Action, MemoryType, Size, Buffer, ActionString); } else { if (mLibProfileProtocol == NULL) { return EFI_UNSUPPORTED; } return mLibProfileProtocol->Record ( mLibProfileProtocol, CallerAddress, Action, MemoryType, Buffer, Size, ActionString ); } }