mirror of
				https://github.com/acidanthera/audk.git
				synced 2025-10-31 11:13:53 +01:00 
			
		
		
		
	Cc: Jiewen Yao <jiewen.yao@intel.com> Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Star Zeng <star.zeng@intel.com> Reviewed-by: Jiewen Yao <jiewen.yao@intel.com>
		
			
				
	
	
		
			103 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			103 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /** @file
 | |
|   Support routines for memory profile for Dxe phase drivers.
 | |
| 
 | |
|   Copyright (c) 2016, 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 <Uefi.h>
 | |
| 
 | |
| 
 | |
| #include <Library/UefiBootServicesTableLib.h>
 | |
| #include <Library/DebugLib.h>
 | |
| 
 | |
| #include <Guid/MemoryProfile.h>
 | |
| 
 | |
| EDKII_MEMORY_PROFILE_PROTOCOL *mLibProfileProtocol;
 | |
| 
 | |
| /**
 | |
|   The constructor function initializes memory profile for DXE 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
 | |
| MemoryProfileLibConstructor (
 | |
|   IN EFI_HANDLE        ImageHandle,
 | |
|   IN EFI_SYSTEM_TABLE  *SystemTable
 | |
|   )
 | |
| {
 | |
|   EFI_STATUS           Status;
 | |
| 
 | |
|   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 (mLibProfileProtocol == NULL) {
 | |
|     return EFI_UNSUPPORTED;
 | |
|   }
 | |
|   return mLibProfileProtocol->Record (
 | |
|                                 mLibProfileProtocol,
 | |
|                                 CallerAddress,
 | |
|                                 Action,
 | |
|                                 MemoryType,
 | |
|                                 Buffer,
 | |
|                                 Size,
 | |
|                                 ActionString
 | |
|                                 );
 | |
| }
 | |
| 
 |