MdeModulePkg DxeCore/PiSmmCore: Add UEFI memory and SMRAM profile support.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Star Zeng <star.zeng@intel.com>
Reviewed-by: Jiewen Yao <jiewen.yao@intel.com>

git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@16335 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 197 insertions, 0 deletions

diff --git a/MdeModulePkg/Core/Dxe/DxeMain.h b/MdeModulePkg/Core/Dxe/DxeMain.h
index 75542b3432..bb2bfab370 100644
--- a/MdeModulePkg/Core/Dxe/DxeMain.h
+++ b/MdeModulePkg/Core/Dxe/DxeMain.h
@@ -69,6 +69,8 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Guid/IdleLoopEvent.h>

 #include <Guid/VectorHandoffTable.h>

 #include <Ppi/VectorHandoffInfo.h>

+#include <Guid/ZeroGuid.h>

+#include <Guid/MemoryProfile.h>

 

 #include <Library/DxeCoreEntryPoint.h>

 #include <Library/DebugLib.h>

@@ -191,6 +193,56 @@ typedef struct {
   EFI_HANDLE            DeviceHandle;

 } EFI_GCD_MAP_ENTRY;

 

+

+#define LOADED_IMAGE_PRIVATE_DATA_SIGNATURE   SIGNATURE_32('l','d','r','i')

+

+typedef struct {

+  UINTN                       Signature;

+  /// Image handle

+  EFI_HANDLE                  Handle;   

+  /// Image type

+  UINTN                       Type;           

+  /// If entrypoint has been called

+  BOOLEAN                     Started;        

+  /// The image's entry point

+  EFI_IMAGE_ENTRY_POINT       EntryPoint;     

+  /// loaded image protocol

+  EFI_LOADED_IMAGE_PROTOCOL   Info;           

+  /// Location in memory

+  EFI_PHYSICAL_ADDRESS        ImageBasePage;  

+  /// Number of pages

+  UINTN                       NumberOfPages;  

+  /// Original fixup data

+  CHAR8                       *FixupData;     

+  /// Tpl of started image

+  EFI_TPL                     Tpl;            

+  /// Status returned by started image

+  EFI_STATUS                  Status;         

+  /// Size of ExitData from started image

+  UINTN                       ExitDataSize;   

+  /// Pointer to exit data from started image

+  VOID                        *ExitData;      

+  /// Pointer to pool allocation for context save/retore

+  VOID                        *JumpBuffer;    

+  /// Pointer to buffer for context save/retore

+  BASE_LIBRARY_JUMP_BUFFER    *JumpContext;  

+  /// Machine type from PE image

+  UINT16                      Machine;        

+  /// EBC Protocol pointer

+  EFI_EBC_PROTOCOL            *Ebc;           

+  /// Runtime image list

+  EFI_RUNTIME_IMAGE_ENTRY     *RuntimeData;   

+  /// Pointer to Loaded Image Device Path Protocl

+  EFI_DEVICE_PATH_PROTOCOL    *LoadedImageDevicePath;  

+  /// PeCoffLoader ImageContext

+  PE_COFF_LOADER_IMAGE_CONTEXT  ImageContext; 

+  /// Status returned by LoadImage() service.

+  EFI_STATUS                  LoadImageStatus;

+} LOADED_IMAGE_PRIVATE_DATA;

+

+#define LOADED_IMAGE_PRIVATE_DATA_FROM_THIS(a) \

+          CR(a, LOADED_IMAGE_PRIVATE_DATA, Info, LOADED_IMAGE_PRIVATE_DATA_SIGNATURE)

+

 //

 // DXE Core Global Variables

 //

@@ -1192,7 +1244,32 @@ CoreAllocatePages (
   IN OUT EFI_PHYSICAL_ADDRESS  *Memory

   );

 

+/**

+  Allocates pages from the memory map.

+

+  @param  Type                   The type of allocation to perform

+  @param  MemoryType             The type of memory to turn the allocated pages

+                                 into

+  @param  NumberOfPages          The number of pages to allocate

+  @param  Memory                 A pointer to receive the base allocated memory

+                                 address

+

+  @return Status. On success, Memory is filled in with the base address allocated

+  @retval EFI_INVALID_PARAMETER  Parameters violate checking rules defined in

+                                 spec.

+  @retval EFI_NOT_FOUND          Could not allocate pages match the requirement.

+  @retval EFI_OUT_OF_RESOURCES   No enough pages to allocate.

+  @retval EFI_SUCCESS            Pages successfully allocated.

 

+**/

+EFI_STATUS

+EFIAPI

+CoreInternalAllocatePages (

+  IN EFI_ALLOCATE_TYPE      Type,

+  IN EFI_MEMORY_TYPE        MemoryType,

+  IN UINTN                  NumberOfPages,

+  IN OUT EFI_PHYSICAL_ADDRESS  *Memory

+  );

 

 /**

   Frees previous allocated pages.

@@ -1212,7 +1289,23 @@ CoreFreePages (
   IN UINTN                  NumberOfPages

   );

 

+/**

+  Frees previous allocated pages.

+

+  @param  Memory                 Base address of memory being freed

+  @param  NumberOfPages          The number of pages to free

+

+  @retval EFI_NOT_FOUND          Could not find the entry that covers the range

+  @retval EFI_INVALID_PARAMETER  Address not aligned

+  @return EFI_SUCCESS         -Pages successfully freed.

 

+**/

+EFI_STATUS

+EFIAPI

+CoreInternalFreePages (

+  IN EFI_PHYSICAL_ADDRESS   Memory,

+  IN UINTN                  NumberOfPages

+  );

 

 /**

   This function returns a copy of the current memory map. The map is an array of

@@ -1277,7 +1370,26 @@ CoreAllocatePool (
   OUT VOID            **Buffer

   );

 

+/**

+  Allocate pool of a particular type.

+

+  @param  PoolType               Type of pool to allocate

+  @param  Size                   The amount of pool to allocate

+  @param  Buffer                 The address to return a pointer to the allocated

+                                 pool

 

+  @retval EFI_INVALID_PARAMETER  PoolType not valid or Buffer is NULL

+  @retval EFI_OUT_OF_RESOURCES   Size exceeds max pool size or allocation failed.

+  @retval EFI_SUCCESS            Pool successfully allocated.

+

+**/

+EFI_STATUS

+EFIAPI

+CoreInternalAllocatePool (

+  IN EFI_MEMORY_TYPE  PoolType,

+  IN UINTN            Size,

+  OUT VOID            **Buffer

+  );

 

 /**

   Frees pool.

@@ -1294,7 +1406,20 @@ CoreFreePool (
   IN VOID        *Buffer

   );

 

+/**

+  Frees pool.

+

+  @param  Buffer                 The allocated pool entry to free

 

+  @retval EFI_INVALID_PARAMETER  Buffer is not a valid value.

+  @retval EFI_SUCCESS            Pool successfully freed.

+

+**/

+EFI_STATUS

+EFIAPI

+CoreInternalFreePool (

+  IN VOID        *Buffer

+  );

 

 /**

   Loads an EFI image into memory and returns a handle to the image.

@@ -2619,4 +2744,76 @@ VerifyFvHeaderChecksum (
   IN EFI_FIRMWARE_VOLUME_HEADER *FvHeader

   );

 

+/**

+  Initialize memory profile.

+

+  @param HobStart   The start address of the HOB.

+

+**/

+VOID

+MemoryProfileInit (

+  IN VOID   *HobStart

+  );

+

+/**

+  Install memory profile protocol.

+

+**/

+VOID

+MemoryProfileInstallProtocol (

+  VOID

+  );

+

+/**

+  Register image to memory profile.

+

+  @param DriverEntry    Image info.

+  @param FileType       Image file type.

+

+  @retval TRUE          Register success.

+  @retval FALSE         Register fail.

+

+**/

+BOOLEAN

+RegisterMemoryProfileImage (

+  IN LOADED_IMAGE_PRIVATE_DATA  *DriverEntry,

+  IN EFI_FV_FILETYPE            FileType

+  );

+

+/**

+  Unregister image from memory profile.

+

+  @param DriverEntry    Image info.

+

+  @retval TRUE          Unregister success.

+  @retval FALSE         Unregister fail.

+

+**/

+BOOLEAN

+UnregisterMemoryProfileImage (

+  IN LOADED_IMAGE_PRIVATE_DATA  *DriverEntry

+  );

+

+/**

+  Update memory profile information.

+

+  @param CallerAddress  Address of caller who call Allocate or Free.

+  @param Action         This Allocate or Free action.

+  @param MemoryType     Memory type.

+  @param Size           Buffer size.

+  @param Buffer         Buffer address.

+

+  @retval TRUE          Profile udpate success.

+  @retval FALSE         Profile update fail.

+

+**/

+BOOLEAN

+CoreUpdateProfile (

+  IN EFI_PHYSICAL_ADDRESS   CallerAddress,

+  IN MEMORY_PROFILE_ACTION  Action,

+  IN EFI_MEMORY_TYPE        MemoryType, // Valid for AllocatePages/AllocatePool

+  IN UINTN                  Size,       // Valid for AllocatePages/FreePages/AllocatePool

+  IN VOID                   *Buffer

+  );

+

 #endif