[mirror_edk2.git] / MdeModulePkg / Core / PiSmmCore / SmramProfileRecord.c

/** @file\r
  Support routines for SMRAM profile.\r
\r
  Copyright (c) 2014 - 2018, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "PiSmmCore.h"\r
\r
#define IS_SMRAM_PROFILE_ENABLED ((PcdGet8 (PcdMemoryProfilePropertyMask) & BIT1) != 0)\r
#define IS_UEFI_MEMORY_PROFILE_ENABLED ((PcdGet8 (PcdMemoryProfilePropertyMask) & BIT0) != 0)\r
\r
#define GET_OCCUPIED_SIZE(ActualSize, Alignment) \\r
  ((ActualSize) + (((Alignment) - ((ActualSize) & ((Alignment) - 1))) & ((Alignment) - 1)))\r
\r
typedef struct {\r
  UINT32                        Signature;\r
  MEMORY_PROFILE_CONTEXT        Context;\r
  LIST_ENTRY                    *DriverInfoList;\r
} MEMORY_PROFILE_CONTEXT_DATA;\r
\r
typedef struct {\r
  UINT32                        Signature;\r
  MEMORY_PROFILE_DRIVER_INFO    DriverInfo;\r
  LIST_ENTRY                    *AllocInfoList;\r
  CHAR8                         *PdbString;\r
  LIST_ENTRY                    Link;\r
} MEMORY_PROFILE_DRIVER_INFO_DATA;\r
\r
typedef struct {\r
  UINT32                        Signature;\r
  MEMORY_PROFILE_ALLOC_INFO     AllocInfo;\r
  CHAR8                         *ActionString;\r
  LIST_ENTRY                    Link;\r
} MEMORY_PROFILE_ALLOC_INFO_DATA;\r
\r
//\r
// When free memory less than 4 pages, dump it.\r
//\r
#define SMRAM_INFO_DUMP_PAGE_THRESHOLD  4\r
\r
GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_FREE_MEMORY mSmramFreeMemory = {\r
  {\r
    MEMORY_PROFILE_FREE_MEMORY_SIGNATURE,\r
    sizeof (MEMORY_PROFILE_FREE_MEMORY),\r
    MEMORY_PROFILE_FREE_MEMORY_REVISION\r
  },\r
  0,\r
  0\r
};\r
\r
GLOBAL_REMOVE_IF_UNREFERENCED LIST_ENTRY  mImageQueue = INITIALIZE_LIST_HEAD_VARIABLE (mImageQueue);\r
GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_CONTEXT_DATA mSmramProfileContext = {\r
  MEMORY_PROFILE_CONTEXT_SIGNATURE,\r
  {\r
    {\r
      MEMORY_PROFILE_CONTEXT_SIGNATURE,\r
      sizeof (MEMORY_PROFILE_CONTEXT),\r
      MEMORY_PROFILE_CONTEXT_REVISION\r
    },\r
    0,\r
    0,\r
    {0},\r
    {0},\r
    0,\r
    0,\r
    0\r
  },\r
  &mImageQueue,\r
};\r
GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_CONTEXT_DATA *mSmramProfileContextPtr = NULL;\r
\r
GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN mSmramReadyToLock;\r
GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN mSmramProfileGettingStatus = FALSE;\r
GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN mSmramProfileRecordingEnable = MEMORY_PROFILE_RECORDING_DISABLE;\r
GLOBAL_REMOVE_IF_UNREFERENCED EFI_DEVICE_PATH_PROTOCOL *mSmramProfileDriverPath;\r
GLOBAL_REMOVE_IF_UNREFERENCED UINTN                    mSmramProfileDriverPathSize;\r
\r
/**\r
  Dump SMRAM information.\r
\r
**/\r
VOID\r
DumpSmramInfo (\r
  VOID\r
  );\r
\r
/**\r
  Get memory profile data.\r
\r
  @param[in]      This              The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in, out] ProfileSize       On entry, points to the size in bytes of the ProfileBuffer.\r
                                    On return, points to the size of the data returned in ProfileBuffer.\r
  @param[out]     ProfileBuffer     Profile buffer.\r
\r
  @return EFI_SUCCESS               Get the memory profile data successfully.\r
  @return EFI_UNSUPPORTED           Memory profile is unsupported.\r
  @return EFI_BUFFER_TO_SMALL       The ProfileSize is too small for the resulting data.\r
                                    ProfileSize is updated with the size required.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolGetData (\r
  IN     EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN OUT UINT64                             *ProfileSize,\r
     OUT VOID                               *ProfileBuffer\r
  );\r
\r
/**\r
  Register image to memory profile.\r
\r
  @param[in] This               The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in] FilePath           File path of the image.\r
  @param[in] ImageBase          Image base address.\r
  @param[in] ImageSize          Image size.\r
  @param[in] FileType           File type of the image.\r
\r
  @return EFI_SUCCESS           Register successfully.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required.\r
  @return EFI_OUT_OF_RESOURCE   No enough resource for this register.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolRegisterImage (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN EFI_DEVICE_PATH_PROTOCOL           *FilePath,\r
  IN PHYSICAL_ADDRESS                   ImageBase,\r
  IN UINT64                             ImageSize,\r
  IN EFI_FV_FILETYPE                    FileType\r
  );\r
\r
/**\r
  Unregister image from memory profile.\r
\r
  @param[in] This               The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in] FilePath           File path of the image.\r
  @param[in] ImageBase          Image base address.\r
  @param[in] ImageSize          Image size.\r
\r
  @return EFI_SUCCESS           Unregister successfully.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required.\r
  @return EFI_NOT_FOUND         The image is not found.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolUnregisterImage (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN EFI_DEVICE_PATH_PROTOCOL           *FilePath,\r
  IN PHYSICAL_ADDRESS                   ImageBase,\r
  IN UINT64                             ImageSize\r
  );\r
\r
/**\r
  Get memory profile recording state.\r
\r
  @param[in]  This              The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[out] RecordingState    Recording state.\r
\r
  @return EFI_SUCCESS           Memory profile recording state is returned.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported.\r
  @return EFI_INVALID_PARAMETER RecordingState is NULL.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolGetRecordingState (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  OUT BOOLEAN                           *RecordingState\r
  );\r
\r
/**\r
  Set memory profile recording state.\r
\r
  @param[in] This               The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in] RecordingState     Recording state.\r
\r
  @return EFI_SUCCESS           Set memory profile recording state successfully.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolSetRecordingState (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN BOOLEAN                            RecordingState\r
  );\r
\r
/**\r
  Record memory profile of multilevel caller.\r
\r
  @param[in] This               The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in] CallerAddress      Address of caller.\r
  @param[in] Action             Memory profile action.\r
  @param[in] MemoryType         Memory type.\r
                                EfiMaxMemoryType means the MemoryType is unknown.\r
  @param[in] Buffer             Buffer address.\r
  @param[in] Size               Buffer size.\r
  @param[in] ActionString       String for memory profile action.\r
                                Only needed for user defined allocate action.\r
\r
  @return EFI_SUCCESS           Memory profile is updated.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required,\r
                                or memory profile for the memory type is not required.\r
  @return EFI_ACCESS_DENIED     It is during memory profile data getting.\r
  @return EFI_ABORTED           Memory profile recording is not enabled.\r
  @return EFI_OUT_OF_RESOURCES  No enough resource to update memory profile for allocate action.\r
  @return EFI_NOT_FOUND         No matched allocate info found for free action.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolRecord (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN PHYSICAL_ADDRESS                   CallerAddress,\r
  IN MEMORY_PROFILE_ACTION              Action,\r
  IN EFI_MEMORY_TYPE                    MemoryType,\r
  IN VOID                               *Buffer,\r
  IN UINTN                              Size,\r
  IN CHAR8                              *ActionString OPTIONAL\r
  );\r
\r
GLOBAL_REMOVE_IF_UNREFERENCED EDKII_SMM_MEMORY_PROFILE_PROTOCOL mSmmProfileProtocol = {\r
  SmramProfileProtocolGetData,\r
  SmramProfileProtocolRegisterImage,\r
  SmramProfileProtocolUnregisterImage,\r
  SmramProfileProtocolGetRecordingState,\r
  SmramProfileProtocolSetRecordingState,\r
  SmramProfileProtocolRecord,\r
};\r
\r
/**\r
  Return SMRAM profile context.\r
\r
  @return SMRAM profile context.\r
\r
**/\r
MEMORY_PROFILE_CONTEXT_DATA *\r
GetSmramProfileContext (\r
  VOID\r
  )\r
{\r
  return mSmramProfileContextPtr;\r
}\r
\r
/**\r
  Retrieves and returns the Subsystem of a PE/COFF image that has been loaded into system memory.\r
  If Pe32Data is NULL, then ASSERT().\r
\r
  @param Pe32Data   The pointer to the PE/COFF image that is loaded in system memory.\r
\r
  @return The Subsystem of the PE/COFF image.\r
\r
**/\r
UINT16\r
InternalPeCoffGetSubsystem (\r
  IN VOID  *Pe32Data\r
  )\r
{\r
  EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION  Hdr;\r
  EFI_IMAGE_DOS_HEADER                 *DosHdr;\r
  UINT16                               Magic;\r
\r
  ASSERT (Pe32Data != NULL);\r
\r
  DosHdr = (EFI_IMAGE_DOS_HEADER *) Pe32Data;\r
  if (DosHdr->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r
    //\r
    // DOS image header is present, so read the PE header after the DOS image header.\r
    //\r
    Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *) ((UINTN) Pe32Data + (UINTN) ((DosHdr->e_lfanew) & 0x0ffff));\r
  } else {\r
    //\r
    // DOS image header is not present, so PE header is at the image base.\r
    //\r
    Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *) Pe32Data;\r
  }\r
\r
  if (Hdr.Te->Signature == EFI_TE_IMAGE_HEADER_SIGNATURE) {\r
    return Hdr.Te->Subsystem;\r
  } else if (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE)  {\r
    Magic = Hdr.Pe32->OptionalHeader.Magic;\r
    if (Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {\r
      return Hdr.Pe32->OptionalHeader.Subsystem;\r
    } else if (Magic == EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC) {\r
      return Hdr.Pe32Plus->OptionalHeader.Subsystem;\r
    }\r
  }\r
\r
  return 0x0000;\r
}\r
\r
/**\r
  Retrieves and returns a pointer to the entry point to a PE/COFF image that has been loaded\r
  into system memory with the PE/COFF Loader Library functions.\r
\r
  Retrieves the entry point to the PE/COFF image specified by Pe32Data and returns this entry\r
  point in EntryPoint.  If the entry point could not be retrieved from the PE/COFF image, then\r
  return RETURN_INVALID_PARAMETER.  Otherwise return RETURN_SUCCESS.\r
  If Pe32Data is NULL, then ASSERT().\r
  If EntryPoint is NULL, then ASSERT().\r
\r
  @param  Pe32Data                  The pointer to the PE/COFF image that is loaded in system memory.\r
  @param  EntryPoint                The pointer to entry point to the PE/COFF image to return.\r
\r
  @retval RETURN_SUCCESS            EntryPoint was returned.\r
  @retval RETURN_INVALID_PARAMETER  The entry point could not be found in the PE/COFF image.\r
\r
**/\r
RETURN_STATUS\r
InternalPeCoffGetEntryPoint (\r
  IN  VOID  *Pe32Data,\r
  OUT VOID  **EntryPoint\r
  )\r
{\r
  EFI_IMAGE_DOS_HEADER                  *DosHdr;\r
  EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION   Hdr;\r
\r
  ASSERT (Pe32Data   != NULL);\r
  ASSERT (EntryPoint != NULL);\r
\r
  DosHdr = (EFI_IMAGE_DOS_HEADER *) Pe32Data;\r
  if (DosHdr->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r
    //\r
    // DOS image header is present, so read the PE header after the DOS image header.\r
    //\r
    Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *) ((UINTN) Pe32Data + (UINTN) ((DosHdr->e_lfanew) & 0x0ffff));\r
  } else {\r
    //\r
    // DOS image header is not present, so PE header is at the image base.\r
    //\r
    Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *) Pe32Data;\r
  }\r
\r
  //\r
  // Calculate the entry point relative to the start of the image.\r
  // AddressOfEntryPoint is common for PE32 & PE32+\r
  //\r
  if (Hdr.Te->Signature == EFI_TE_IMAGE_HEADER_SIGNATURE) {\r
    *EntryPoint = (VOID *) ((UINTN) Pe32Data + (UINTN) (Hdr.Te->AddressOfEntryPoint & 0x0ffffffff) + sizeof (EFI_TE_IMAGE_HEADER) - Hdr.Te->StrippedSize);\r
    return RETURN_SUCCESS;\r
  } else if (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE) {\r
    *EntryPoint = (VOID *) ((UINTN) Pe32Data + (UINTN) (Hdr.Pe32->OptionalHeader.AddressOfEntryPoint & 0x0ffffffff));\r
    return RETURN_SUCCESS;\r
  }\r
\r
  return RETURN_UNSUPPORTED;\r
}\r
\r
/**\r
  Build driver info.\r
\r
  @param ContextData    Memory profile context.\r
  @param FileName       File name of the image.\r
  @param ImageBase      Image base address.\r
  @param ImageSize      Image size.\r
  @param EntryPoint     Entry point of the image.\r
  @param ImageSubsystem Image subsystem of the image.\r
  @param FileType       File type of the image.\r
\r
  @return Pointer to memory profile driver info.\r
\r
**/\r
MEMORY_PROFILE_DRIVER_INFO_DATA *\r
BuildDriverInfo (\r
  IN MEMORY_PROFILE_CONTEXT_DATA    *ContextData,\r
  IN EFI_GUID                       *FileName,\r
  IN PHYSICAL_ADDRESS               ImageBase,\r
  IN UINT64                         ImageSize,\r
  IN PHYSICAL_ADDRESS               EntryPoint,\r
  IN UINT16                         ImageSubsystem,\r
  IN EFI_FV_FILETYPE                FileType\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  MEMORY_PROFILE_DRIVER_INFO        *DriverInfo;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  VOID                              *EntryPointInImage;\r
  CHAR8                             *PdbString;\r
  UINTN                             PdbSize;\r
  UINTN                             PdbOccupiedSize;\r
\r
  PdbSize = 0;\r
  PdbOccupiedSize = 0;\r
  PdbString = NULL;\r
  if (ImageBase != 0) {\r
    PdbString = PeCoffLoaderGetPdbPointer ((VOID*) (UINTN) ImageBase);\r
    if (PdbString != NULL) {\r
      PdbSize = AsciiStrSize (PdbString);\r
      PdbOccupiedSize = GET_OCCUPIED_SIZE (PdbSize, sizeof (UINT64));\r
    }\r
  }\r
\r
  //\r
  // Use SmmInternalAllocatePool() that will not update profile for this AllocatePool action.\r
  //\r
  Status = SmmInternalAllocatePool (\r
             EfiRuntimeServicesData,\r
             sizeof (*DriverInfoData) + sizeof (LIST_ENTRY) + PdbSize,\r
             (VOID **) &DriverInfoData\r
             );\r
  if (EFI_ERROR (Status)) {\r
    return NULL;\r
  }\r
  ASSERT (DriverInfoData != NULL);\r
\r
  ZeroMem (DriverInfoData, sizeof (*DriverInfoData));\r
\r
  DriverInfo = &DriverInfoData->DriverInfo;\r
  DriverInfoData->Signature = MEMORY_PROFILE_DRIVER_INFO_SIGNATURE;\r
  DriverInfo->Header.Signature = MEMORY_PROFILE_DRIVER_INFO_SIGNATURE;\r
  DriverInfo->Header.Length = (UINT16) (sizeof (MEMORY_PROFILE_DRIVER_INFO) + PdbOccupiedSize);\r
  DriverInfo->Header.Revision = MEMORY_PROFILE_DRIVER_INFO_REVISION;\r
  if (FileName != NULL) {\r
    CopyMem (&DriverInfo->FileName, FileName, sizeof (EFI_GUID));\r
  }\r
  DriverInfo->ImageBase = ImageBase;\r
  DriverInfo->ImageSize = ImageSize;\r
  DriverInfo->EntryPoint = EntryPoint;\r
  DriverInfo->ImageSubsystem = ImageSubsystem;\r
  if ((EntryPoint != 0) && ((EntryPoint < ImageBase) || (EntryPoint >= (ImageBase + ImageSize)))) {\r
    //\r
    // If the EntryPoint is not in the range of image buffer, it should come from emulation environment.\r
    // So patch ImageBuffer here to align the EntryPoint.\r
    //\r
    Status = InternalPeCoffGetEntryPoint ((VOID *) (UINTN) ImageBase, &EntryPointInImage);\r
    ASSERT_EFI_ERROR (Status);\r
    DriverInfo->ImageBase = ImageBase + EntryPoint - (PHYSICAL_ADDRESS) (UINTN) EntryPointInImage;\r
  }\r
  DriverInfo->FileType = FileType;\r
  DriverInfoData->AllocInfoList = (LIST_ENTRY *) (DriverInfoData + 1);\r
  InitializeListHead (DriverInfoData->AllocInfoList);\r
  DriverInfo->CurrentUsage = 0;\r
  DriverInfo->PeakUsage = 0;\r
  DriverInfo->AllocRecordCount = 0;\r
  if (PdbSize != 0) {\r
    DriverInfo->PdbStringOffset = (UINT16) sizeof (MEMORY_PROFILE_DRIVER_INFO);\r
    DriverInfoData->PdbString = (CHAR8 *) (DriverInfoData->AllocInfoList + 1);\r
    CopyMem (DriverInfoData->PdbString, PdbString, PdbSize);\r
  } else {\r
    DriverInfo->PdbStringOffset = 0;\r
    DriverInfoData->PdbString = NULL;\r
  }\r
\r
  InsertTailList (ContextData->DriverInfoList, &DriverInfoData->Link);\r
  ContextData->Context.ImageCount ++;\r
  ContextData->Context.TotalImageSize += DriverInfo->ImageSize;\r
\r
  return DriverInfoData;\r
}\r
\r
/**\r
  Register image to DXE.\r
\r
  @param FileName       File name of the image.\r
  @param ImageBase      Image base address.\r
  @param ImageSize      Image size.\r
  @param FileType       File type of the image.\r
\r
**/\r
VOID\r
RegisterImageToDxe (\r
  IN EFI_GUID                       *FileName,\r
  IN PHYSICAL_ADDRESS               ImageBase,\r
  IN UINT64                         ImageSize,\r
  IN EFI_FV_FILETYPE                FileType\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EDKII_MEMORY_PROFILE_PROTOCOL     *ProfileProtocol;\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FilePath;\r
  UINT8                             TempBuffer[sizeof (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof (EFI_DEVICE_PATH_PROTOCOL)];\r
\r
  if (IS_UEFI_MEMORY_PROFILE_ENABLED) {\r
\r
    FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)TempBuffer;\r
    Status = gBS->LocateProtocol (&gEdkiiMemoryProfileGuid, NULL, (VOID **) &ProfileProtocol);\r
    if (!EFI_ERROR (Status)) {\r
      EfiInitializeFwVolDevicepathNode (FilePath, FileName);\r
      SetDevicePathEndNode (FilePath + 1);\r
\r
      Status = ProfileProtocol->RegisterImage (\r
                                  ProfileProtocol,\r
                                  (EFI_DEVICE_PATH_PROTOCOL *) FilePath,\r
                                  ImageBase,\r
                                  ImageSize,\r
                                  FileType\r
                                  );\r
    }\r
  }\r
}\r
\r
/**\r
  Unregister image from DXE.\r
\r
  @param FileName       File name of the image.\r
  @param ImageBase      Image base address.\r
  @param ImageSize      Image size.\r
\r
**/\r
VOID\r
UnregisterImageFromDxe (\r
  IN EFI_GUID                       *FileName,\r
  IN PHYSICAL_ADDRESS               ImageBase,\r
  IN UINT64                         ImageSize\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EDKII_MEMORY_PROFILE_PROTOCOL     *ProfileProtocol;\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FilePath;\r
  UINT8                             TempBuffer[sizeof (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof (EFI_DEVICE_PATH_PROTOCOL)];\r
\r
  if (IS_UEFI_MEMORY_PROFILE_ENABLED) {\r
\r
    FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)TempBuffer;\r
    Status = gBS->LocateProtocol (&gEdkiiMemoryProfileGuid, NULL, (VOID *) &ProfileProtocol);\r
    if (!EFI_ERROR (Status)) {\r
      EfiInitializeFwVolDevicepathNode (FilePath, FileName);\r
      SetDevicePathEndNode (FilePath + 1);\r
\r
      Status = ProfileProtocol->UnregisterImage (\r
                                  ProfileProtocol,\r
                                  (EFI_DEVICE_PATH_PROTOCOL *) FilePath,\r
                                  ImageBase,\r
                                  ImageSize\r
                                  );\r
    }\r
  }\r
}\r
\r
/**\r
  Return if record for this driver is needed..\r
\r
  @param DriverFilePath     Driver file path.\r
\r
  @retval TRUE              Record for this driver is needed.\r
  @retval FALSE             Record for this driver is not needed.\r
\r
**/\r
BOOLEAN\r
NeedRecordThisDriver (\r
  IN EFI_DEVICE_PATH_PROTOCOL       *DriverFilePath\r
  )\r
{\r
  EFI_DEVICE_PATH_PROTOCOL    *TmpDevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL    *DevicePathInstance;\r
  UINTN                       DevicePathSize;\r
  UINTN                       FilePathSize;\r
\r
  if (!IsDevicePathValid (mSmramProfileDriverPath, mSmramProfileDriverPathSize)) {\r
    //\r
    // Invalid Device Path means record all.\r
    //\r
    return TRUE;\r
  }\r
\r
  //\r
  // Record FilePath without end node.\r
  //\r
  FilePathSize = GetDevicePathSize (DriverFilePath) - sizeof(EFI_DEVICE_PATH_PROTOCOL);\r
\r
  DevicePathInstance = mSmramProfileDriverPath;\r
  do {\r
    //\r
    // Find End node (it might be END_ENTIRE or END_INSTANCE)\r
    //\r
    TmpDevicePath = DevicePathInstance;\r
    while (!IsDevicePathEndType (TmpDevicePath)) {\r
      TmpDevicePath = NextDevicePathNode (TmpDevicePath);\r
    }\r
\r
    //\r
    // Do not compare END node\r
    //\r
    DevicePathSize = (UINTN)TmpDevicePath - (UINTN)DevicePathInstance;\r
    if ((FilePathSize == DevicePathSize) &&\r
        (CompareMem (DriverFilePath, DevicePathInstance, DevicePathSize) == 0)) {\r
      return TRUE;\r
    }\r
\r
    //\r
    // Get next instance\r
    //\r
    DevicePathInstance = (EFI_DEVICE_PATH_PROTOCOL *)((UINTN)DevicePathInstance + DevicePathSize + DevicePathNodeLength(TmpDevicePath));\r
  } while (DevicePathSubType (TmpDevicePath) != END_ENTIRE_DEVICE_PATH_SUBTYPE);\r
\r
  return FALSE;\r
}\r
\r
/**\r
  Register SMM Core to SMRAM profile.\r
\r
  @param ContextData    SMRAM profile context.\r
\r
  @retval TRUE          Register success.\r
  @retval FALSE         Register fail.\r
\r
**/\r
BOOLEAN\r
RegisterSmmCore (\r
  IN MEMORY_PROFILE_CONTEXT_DATA    *ContextData\r
  )\r
{\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  PHYSICAL_ADDRESS                  ImageBase;\r
  UINT8                             TempBuffer[sizeof(MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof(EFI_DEVICE_PATH_PROTOCOL)];\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FilePath;\r
\r
  FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) TempBuffer;\r
  EfiInitializeFwVolDevicepathNode (FilePath, &gEfiCallerIdGuid);\r
  SetDevicePathEndNode (FilePath + 1);\r
\r
  if (!NeedRecordThisDriver ((EFI_DEVICE_PATH_PROTOCOL *) FilePath)) {\r
    return FALSE;\r
  }\r
\r
  ImageBase = gSmmCorePrivate->PiSmmCoreImageBase;\r
  DriverInfoData = BuildDriverInfo (\r
                     ContextData,\r
                     &gEfiCallerIdGuid,\r
                     ImageBase,\r
                     gSmmCorePrivate->PiSmmCoreImageSize,\r
                     gSmmCorePrivate->PiSmmCoreEntryPoint,\r
                     InternalPeCoffGetSubsystem ((VOID *) (UINTN) ImageBase),\r
                     EFI_FV_FILETYPE_SMM_CORE\r
                     );\r
  if (DriverInfoData == NULL) {\r
    return FALSE;\r
  }\r
\r
  return TRUE;\r
}\r
\r
/**\r
  Initialize SMRAM profile.\r
\r
**/\r
VOID\r
SmramProfileInit (\r
  VOID\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT_DATA *SmramProfileContext;\r
\r
  RegisterImageToDxe (\r
    &gEfiCallerIdGuid,\r
    gSmmCorePrivate->PiSmmCoreImageBase,\r
    gSmmCorePrivate->PiSmmCoreImageSize,\r
    EFI_FV_FILETYPE_SMM_CORE\r
    );\r
\r
  if (!IS_SMRAM_PROFILE_ENABLED) {\r
    return;\r
  }\r
\r
  SmramProfileContext = GetSmramProfileContext ();\r
  if (SmramProfileContext != NULL) {\r
    return;\r
  }\r
\r
  mSmramProfileGettingStatus = FALSE;\r
  if ((PcdGet8 (PcdMemoryProfilePropertyMask) & BIT7) != 0) {\r
    mSmramProfileRecordingEnable = MEMORY_PROFILE_RECORDING_DISABLE;\r
  } else {\r
    mSmramProfileRecordingEnable = MEMORY_PROFILE_RECORDING_ENABLE;\r
  }\r
  mSmramProfileDriverPathSize = PcdGetSize (PcdMemoryProfileDriverPath);\r
  mSmramProfileDriverPath = AllocateCopyPool (mSmramProfileDriverPathSize, PcdGetPtr (PcdMemoryProfileDriverPath));\r
  mSmramProfileContextPtr = &mSmramProfileContext;\r
\r
  RegisterSmmCore (&mSmramProfileContext);\r
\r
  DEBUG ((EFI_D_INFO, "SmramProfileInit SmramProfileContext - 0x%x\n", &mSmramProfileContext));\r
}\r
\r
/**\r
  Install SMRAM profile protocol.\r
\r
**/\r
VOID\r
SmramProfileInstallProtocol (\r
  VOID\r
  )\r
{\r
  EFI_HANDLE    Handle;\r
  EFI_STATUS    Status;\r
\r
  if (!IS_SMRAM_PROFILE_ENABLED) {\r
    return;\r
  }\r
\r
  Handle = NULL;\r
  Status = SmmInstallProtocolInterface (\r
             &Handle,\r
             &gEdkiiSmmMemoryProfileGuid,\r
             EFI_NATIVE_INTERFACE,\r
             &mSmmProfileProtocol\r
             );\r
  ASSERT_EFI_ERROR (Status);\r
}\r
\r
/**\r
  Get the GUID file name from the file path.\r
\r
  @param FilePath  File path.\r
\r
  @return The GUID file name from the file path.\r
\r
**/\r
EFI_GUID *\r
GetFileNameFromFilePath (\r
  IN EFI_DEVICE_PATH_PROTOCOL   *FilePath\r
  )\r
{\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH     *ThisFilePath;\r
  EFI_GUID                              *FileName;\r
\r
  FileName = NULL;\r
  if (FilePath != NULL) {\r
    ThisFilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) FilePath;\r
    while (!IsDevicePathEnd (ThisFilePath)) {\r
      FileName = EfiGetNameGuidFromFwVolDevicePathNode (ThisFilePath);\r
      if (FileName != NULL) {\r
        break;\r
      }\r
      ThisFilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) NextDevicePathNode (ThisFilePath);\r
    }\r
  }\r
\r
  return FileName;\r
}\r
\r
/**\r
  Register SMM image to SMRAM profile.\r
\r
  @param DriverEntry    SMM image info.\r
  @param RegisterToDxe  Register image to DXE.\r
\r
  @return EFI_SUCCESS           Register successfully.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required.\r
  @return EFI_OUT_OF_RESOURCES  No enough resource for this register.\r
\r
**/\r
EFI_STATUS\r
RegisterSmramProfileImage (\r
  IN EFI_SMM_DRIVER_ENTRY   *DriverEntry,\r
  IN BOOLEAN                RegisterToDxe\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  UINT8                             TempBuffer[sizeof(MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof(EFI_DEVICE_PATH_PROTOCOL)];\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FilePath;\r
\r
  if (RegisterToDxe) {\r
    RegisterImageToDxe (\r
      &DriverEntry->FileName,\r
      DriverEntry->ImageBuffer,\r
      EFI_PAGES_TO_SIZE (DriverEntry->NumberOfPage),\r
      EFI_FV_FILETYPE_SMM\r
      );\r
  }\r
\r
  if (!IS_SMRAM_PROFILE_ENABLED) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) TempBuffer;\r
  EfiInitializeFwVolDevicepathNode (FilePath, &DriverEntry->FileName);\r
  SetDevicePathEndNode (FilePath + 1);\r
\r
  if (!NeedRecordThisDriver ((EFI_DEVICE_PATH_PROTOCOL *) FilePath)) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  DriverInfoData = BuildDriverInfo (\r
                     ContextData,\r
                     &DriverEntry->FileName,\r
                     DriverEntry->ImageBuffer,\r
                     EFI_PAGES_TO_SIZE (DriverEntry->NumberOfPage),\r
                     DriverEntry->ImageEntryPoint,\r
                     InternalPeCoffGetSubsystem ((VOID *) (UINTN) DriverEntry->ImageBuffer),\r
                     EFI_FV_FILETYPE_SMM\r
                     );\r
  if (DriverInfoData == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Search image from memory profile.\r
\r
  @param ContextData    Memory profile context.\r
  @param FileName       Image file name.\r
  @param Address        Image Address.\r
\r
  @return Pointer to memory profile driver info.\r
\r
**/\r
MEMORY_PROFILE_DRIVER_INFO_DATA *\r
GetMemoryProfileDriverInfoByFileNameAndAddress (\r
  IN MEMORY_PROFILE_CONTEXT_DATA    *ContextData,\r
  IN EFI_GUID                       *FileName,\r
  IN PHYSICAL_ADDRESS               Address\r
  )\r
{\r
  MEMORY_PROFILE_DRIVER_INFO        *DriverInfo;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  LIST_ENTRY                        *DriverLink;\r
  LIST_ENTRY                        *DriverInfoList;\r
\r
  DriverInfoList = ContextData->DriverInfoList;\r
\r
  for (DriverLink = DriverInfoList->ForwardLink;\r
       DriverLink != DriverInfoList;\r
       DriverLink = DriverLink->ForwardLink) {\r
    DriverInfoData = CR (\r
                       DriverLink,\r
                       MEMORY_PROFILE_DRIVER_INFO_DATA,\r
                       Link,\r
                       MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
                       );\r
    DriverInfo = &DriverInfoData->DriverInfo;\r
    if ((CompareGuid (&DriverInfo->FileName, FileName)) &&\r
        (Address >= DriverInfo->ImageBase) &&\r
        (Address < (DriverInfo->ImageBase + DriverInfo->ImageSize))) {\r
      return DriverInfoData;\r
    }\r
  }\r
\r
  return NULL;\r
}\r
\r
/**\r
  Search image from memory profile.\r
  It will return image, if (Address >= ImageBuffer) AND (Address < ImageBuffer + ImageSize)\r
\r
  @param ContextData    Memory profile context.\r
  @param Address        Image or Function address.\r
\r
  @return Pointer to memory profile driver info.\r
\r
**/\r
MEMORY_PROFILE_DRIVER_INFO_DATA *\r
GetMemoryProfileDriverInfoFromAddress (\r
  IN MEMORY_PROFILE_CONTEXT_DATA    *ContextData,\r
  IN PHYSICAL_ADDRESS               Address\r
  )\r
{\r
  MEMORY_PROFILE_DRIVER_INFO        *DriverInfo;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  LIST_ENTRY                        *DriverLink;\r
  LIST_ENTRY                        *DriverInfoList;\r
\r
  DriverInfoList = ContextData->DriverInfoList;\r
\r
  for (DriverLink = DriverInfoList->ForwardLink;\r
       DriverLink != DriverInfoList;\r
       DriverLink = DriverLink->ForwardLink) {\r
    DriverInfoData = CR (\r
                       DriverLink,\r
                       MEMORY_PROFILE_DRIVER_INFO_DATA,\r
                       Link,\r
                       MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
                       );\r
    DriverInfo = &DriverInfoData->DriverInfo;\r
    if ((Address >= DriverInfo->ImageBase) &&\r
        (Address < (DriverInfo->ImageBase + DriverInfo->ImageSize))) {\r
      return DriverInfoData;\r
    }\r
  }\r
\r
  return NULL;\r
}\r
\r
/**\r
  Unregister image from SMRAM profile.\r
\r
  @param DriverEntry        SMM image info.\r
  @param UnregisterFromDxe  Unregister image from DXE.\r
\r
  @return EFI_SUCCESS           Unregister successfully.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required.\r
  @return EFI_NOT_FOUND         The image is not found.\r
\r
**/\r
EFI_STATUS\r
UnregisterSmramProfileImage (\r
  IN EFI_SMM_DRIVER_ENTRY  *DriverEntry,\r
  IN BOOLEAN               UnregisterFromDxe\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  EFI_GUID                          *FileName;\r
  PHYSICAL_ADDRESS                  ImageAddress;\r
  VOID                              *EntryPointInImage;\r
  UINT8                             TempBuffer[sizeof(MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof(EFI_DEVICE_PATH_PROTOCOL)];\r
  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FilePath;\r
\r
  if (UnregisterFromDxe) {\r
    UnregisterImageFromDxe (\r
      &DriverEntry->FileName,\r
      DriverEntry->ImageBuffer,\r
      EFI_PAGES_TO_SIZE (DriverEntry->NumberOfPage)\r
      );\r
  }\r
\r
  if (!IS_SMRAM_PROFILE_ENABLED) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) TempBuffer;\r
  EfiInitializeFwVolDevicepathNode (FilePath, &DriverEntry->FileName);\r
  SetDevicePathEndNode (FilePath + 1);\r
\r
  if (!NeedRecordThisDriver ((EFI_DEVICE_PATH_PROTOCOL *) FilePath)) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  DriverInfoData = NULL;\r
  FileName = &DriverEntry->FileName;\r
  ImageAddress = DriverEntry->ImageBuffer;\r
  if ((DriverEntry->ImageEntryPoint < ImageAddress) || (DriverEntry->ImageEntryPoint >= (ImageAddress + EFI_PAGES_TO_SIZE (DriverEntry->NumberOfPage)))) {\r
    //\r
    // If the EntryPoint is not in the range of image buffer, it should come from emulation environment.\r
    // So patch ImageAddress here to align the EntryPoint.\r
    //\r
    Status = InternalPeCoffGetEntryPoint ((VOID *) (UINTN) ImageAddress, &EntryPointInImage);\r
    ASSERT_EFI_ERROR (Status);\r
    ImageAddress = ImageAddress + (UINTN) DriverEntry->ImageEntryPoint - (UINTN) EntryPointInImage;\r
  }\r
  if (FileName != NULL) {\r
    DriverInfoData = GetMemoryProfileDriverInfoByFileNameAndAddress (ContextData, FileName, ImageAddress);\r
  }\r
  if (DriverInfoData == NULL) {\r
    DriverInfoData = GetMemoryProfileDriverInfoFromAddress (ContextData, ImageAddress);\r
  }\r
  if (DriverInfoData == NULL) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  ContextData->Context.TotalImageSize -= DriverInfoData->DriverInfo.ImageSize;\r
\r
  // Keep the ImageBase for RVA calculation in Application.\r
  //DriverInfoData->DriverInfo.ImageBase = 0;\r
  DriverInfoData->DriverInfo.ImageSize = 0;\r
\r
  if (DriverInfoData->DriverInfo.PeakUsage == 0) {\r
    ContextData->Context.ImageCount --;\r
    RemoveEntryList (&DriverInfoData->Link);\r
    //\r
    // Use SmmInternalFreePool() that will not update profile for this FreePool action.\r
    //\r
    SmmInternalFreePool (DriverInfoData);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Return if this memory type needs to be recorded into memory profile.\r
  Only need to record EfiRuntimeServicesCode and EfiRuntimeServicesData for SMRAM profile.\r
\r
  @param MemoryType     Memory type.\r
\r
  @retval TRUE          This memory type need to be recorded.\r
  @retval FALSE         This memory type need not to be recorded.\r
\r
**/\r
BOOLEAN\r
SmmCoreNeedRecordProfile (\r
  IN EFI_MEMORY_TYPE    MemoryType\r
  )\r
{\r
  UINT64 TestBit;\r
\r
  if (MemoryType != EfiRuntimeServicesCode &&\r
      MemoryType != EfiRuntimeServicesData) {\r
    return FALSE;\r
  }\r
\r
  TestBit = LShiftU64 (1, MemoryType);\r
\r
  if ((PcdGet64 (PcdMemoryProfileMemoryType) & TestBit) != 0) {\r
    return TRUE;\r
  } else {\r
    return FALSE;\r
  }\r
}\r
\r
/**\r
  Convert EFI memory type to profile memory index. The rule is:\r
  If BIOS memory type (0 ~ EfiMaxMemoryType - 1), ProfileMemoryIndex = MemoryType.\r
  As SMRAM profile is only to record EfiRuntimeServicesCode and EfiRuntimeServicesData,\r
  so return input memory type directly.\r
\r
  @param MemoryType     Memory type.\r
\r
  @return EFI memory type as profile memory index.\r
\r
**/\r
EFI_MEMORY_TYPE\r
GetProfileMemoryIndex (\r
  IN EFI_MEMORY_TYPE    MemoryType\r
  )\r
{\r
  return MemoryType;\r
}\r
\r
/**\r
  Update SMRAM profile FreeMemoryPages information\r
\r
  @param ContextData    Memory profile context.\r
\r
**/\r
VOID\r
SmramProfileUpdateFreePages (\r
  IN MEMORY_PROFILE_CONTEXT_DATA  *ContextData\r
  )\r
{\r
  LIST_ENTRY                      *Node;\r
  FREE_PAGE_LIST                  *Pages;\r
  LIST_ENTRY                      *FreePageList;\r
  UINTN                           NumberOfPages;\r
\r
  NumberOfPages = 0;\r
  FreePageList = &mSmmMemoryMap;\r
  for (Node = FreePageList->BackLink;\r
       Node != FreePageList;\r
       Node = Node->BackLink) {\r
    Pages = BASE_CR (Node, FREE_PAGE_LIST, Link);\r
    NumberOfPages += Pages->NumberOfPages;\r
  }\r
\r
  mSmramFreeMemory.TotalFreeMemoryPages = NumberOfPages;\r
\r
  if (NumberOfPages <= SMRAM_INFO_DUMP_PAGE_THRESHOLD) {\r
    DumpSmramInfo ();\r
  }\r
}\r
\r
/**\r
  Update SMRAM profile Allocate information.\r
\r
  @param CallerAddress  Address of caller who call Allocate.\r
  @param Action         This Allocate action.\r
  @param MemoryType     Memory type.\r
  @param Size           Buffer size.\r
  @param Buffer         Buffer address.\r
  @param ActionString   String for memory profile action.\r
\r
  @return EFI_SUCCESS           Memory profile is updated.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required.\r
  @return EFI_OUT_OF_RESOURCES  No enough resource to update memory profile for allocate action.\r
\r
**/\r
EFI_STATUS\r
SmmCoreUpdateProfileAllocate (\r
  IN PHYSICAL_ADDRESS       CallerAddress,\r
  IN MEMORY_PROFILE_ACTION  Action,\r
  IN EFI_MEMORY_TYPE        MemoryType,\r
  IN UINTN                  Size,\r
  IN VOID                   *Buffer,\r
  IN CHAR8                  *ActionString OPTIONAL\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  MEMORY_PROFILE_CONTEXT            *Context;\r
  MEMORY_PROFILE_DRIVER_INFO        *DriverInfo;\r
  MEMORY_PROFILE_ALLOC_INFO         *AllocInfo;\r
  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  MEMORY_PROFILE_ALLOC_INFO_DATA    *AllocInfoData;\r
  EFI_MEMORY_TYPE                   ProfileMemoryIndex;\r
  MEMORY_PROFILE_ACTION             BasicAction;\r
  UINTN                             ActionStringSize;\r
  UINTN                             ActionStringOccupiedSize;\r
\r
  BasicAction = Action & MEMORY_PROFILE_ACTION_BASIC_MASK;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  DriverInfoData = GetMemoryProfileDriverInfoFromAddress (ContextData, CallerAddress);\r
  if (DriverInfoData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  ActionStringSize = 0;\r
  ActionStringOccupiedSize = 0;\r
  if (ActionString != NULL) {\r
    ActionStringSize = AsciiStrSize (ActionString);\r
    ActionStringOccupiedSize = GET_OCCUPIED_SIZE (ActionStringSize, sizeof (UINT64));\r
  }\r
\r
  //\r
  // Use SmmInternalAllocatePool() that will not update profile for this AllocatePool action.\r
  //\r
  AllocInfoData = NULL;\r
  Status = SmmInternalAllocatePool (\r
             EfiRuntimeServicesData,\r
             sizeof (*AllocInfoData) + ActionStringSize,\r
             (VOID **) &AllocInfoData\r
             );\r
  if (EFI_ERROR (Status)) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  ASSERT (AllocInfoData != NULL);\r
\r
  //\r
  // Only update SequenceCount if and only if it is basic action.\r
  //\r
  if (Action == BasicAction) {\r
    ContextData->Context.SequenceCount ++;\r
  }\r
\r
  AllocInfo = &AllocInfoData->AllocInfo;\r
  AllocInfoData->Signature      = MEMORY_PROFILE_ALLOC_INFO_SIGNATURE;\r
  AllocInfo->Header.Signature   = MEMORY_PROFILE_ALLOC_INFO_SIGNATURE;\r
  AllocInfo->Header.Length      = (UINT16) (sizeof (MEMORY_PROFILE_ALLOC_INFO) + ActionStringOccupiedSize);\r
  AllocInfo->Header.Revision    = MEMORY_PROFILE_ALLOC_INFO_REVISION;\r
  AllocInfo->CallerAddress      = CallerAddress;\r
  AllocInfo->SequenceId         = ContextData->Context.SequenceCount;\r
  AllocInfo->Action             = Action;\r
  AllocInfo->MemoryType         = MemoryType;\r
  AllocInfo->Buffer             = (PHYSICAL_ADDRESS) (UINTN) Buffer;\r
  AllocInfo->Size               = Size;\r
  if (ActionString != NULL) {\r
    AllocInfo->ActionStringOffset = (UINT16) sizeof (MEMORY_PROFILE_ALLOC_INFO);\r
    AllocInfoData->ActionString = (CHAR8 *) (AllocInfoData + 1);\r
    CopyMem (AllocInfoData->ActionString, ActionString, ActionStringSize);\r
  } else {\r
    AllocInfo->ActionStringOffset = 0;\r
    AllocInfoData->ActionString = NULL;\r
  }\r
\r
  InsertTailList (DriverInfoData->AllocInfoList, &AllocInfoData->Link);\r
\r
  Context = &ContextData->Context;\r
  DriverInfo = &DriverInfoData->DriverInfo;\r
  DriverInfo->AllocRecordCount ++;\r
\r
  //\r
  // Update summary if and only if it is basic action.\r
  //\r
  if (Action == BasicAction) {\r
    ProfileMemoryIndex = GetProfileMemoryIndex (MemoryType);\r
\r
    DriverInfo->CurrentUsage += Size;\r
    if (DriverInfo->PeakUsage < DriverInfo->CurrentUsage) {\r
      DriverInfo->PeakUsage = DriverInfo->CurrentUsage;\r
    }\r
    DriverInfo->CurrentUsageByType[ProfileMemoryIndex] += Size;\r
    if (DriverInfo->PeakUsageByType[ProfileMemoryIndex] < DriverInfo->CurrentUsageByType[ProfileMemoryIndex]) {\r
      DriverInfo->PeakUsageByType[ProfileMemoryIndex] = DriverInfo->CurrentUsageByType[ProfileMemoryIndex];\r
    }\r
\r
    Context->CurrentTotalUsage += Size;\r
    if (Context->PeakTotalUsage < Context->CurrentTotalUsage) {\r
      Context->PeakTotalUsage = Context->CurrentTotalUsage;\r
    }\r
    Context->CurrentTotalUsageByType[ProfileMemoryIndex] += Size;\r
    if (Context->PeakTotalUsageByType[ProfileMemoryIndex] < Context->CurrentTotalUsageByType[ProfileMemoryIndex]) {\r
      Context->PeakTotalUsageByType[ProfileMemoryIndex] = Context->CurrentTotalUsageByType[ProfileMemoryIndex];\r
    }\r
\r
    SmramProfileUpdateFreePages (ContextData);\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Get memory profile alloc info from memory profile\r
\r
  @param DriverInfoData     Driver info\r
  @param BasicAction        This Free basic action\r
  @param Size               Buffer size\r
  @param Buffer             Buffer address\r
\r
  @return Pointer to memory profile alloc info.\r
**/\r
MEMORY_PROFILE_ALLOC_INFO_DATA *\r
GetMemoryProfileAllocInfoFromAddress (\r
  IN MEMORY_PROFILE_DRIVER_INFO_DATA    *DriverInfoData,\r
  IN MEMORY_PROFILE_ACTION              BasicAction,\r
  IN UINTN                              Size,\r
  IN VOID                               *Buffer\r
  )\r
{\r
  LIST_ENTRY                        *AllocInfoList;\r
  LIST_ENTRY                        *AllocLink;\r
  MEMORY_PROFILE_ALLOC_INFO         *AllocInfo;\r
  MEMORY_PROFILE_ALLOC_INFO_DATA    *AllocInfoData;\r
\r
  AllocInfoList = DriverInfoData->AllocInfoList;\r
\r
  for (AllocLink = AllocInfoList->ForwardLink;\r
       AllocLink != AllocInfoList;\r
       AllocLink = AllocLink->ForwardLink) {\r
    AllocInfoData = CR (\r
                      AllocLink,\r
                      MEMORY_PROFILE_ALLOC_INFO_DATA,\r
                      Link,\r
                      MEMORY_PROFILE_ALLOC_INFO_SIGNATURE\r
                      );\r
    AllocInfo = &AllocInfoData->AllocInfo;\r
    if ((AllocInfo->Action & MEMORY_PROFILE_ACTION_BASIC_MASK) != BasicAction) {\r
      continue;\r
    }\r
    switch (BasicAction) {\r
      case MemoryProfileActionAllocatePages:\r
        if ((AllocInfo->Buffer <= (PHYSICAL_ADDRESS) (UINTN) Buffer) &&\r
            ((AllocInfo->Buffer + AllocInfo->Size) >= ((PHYSICAL_ADDRESS) (UINTN) Buffer + Size))) {\r
          return AllocInfoData;\r
        }\r
        break;\r
      case MemoryProfileActionAllocatePool:\r
        if (AllocInfo->Buffer == (PHYSICAL_ADDRESS) (UINTN) Buffer) {\r
          return AllocInfoData;\r
        }\r
        break;\r
      default:\r
        ASSERT (FALSE);\r
        break;\r
    }\r
  }\r
\r
  return NULL;\r
}\r
\r
/**\r
  Update SMRAM profile Free information.\r
\r
  @param CallerAddress  Address of caller who call Free.\r
  @param Action         This Free action.\r
  @param Size           Buffer size.\r
  @param Buffer         Buffer address.\r
\r
  @return EFI_SUCCESS           Memory profile is updated.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported.\r
  @return EFI_NOT_FOUND         No matched allocate info found for free action.\r
\r
**/\r
EFI_STATUS\r
SmmCoreUpdateProfileFree (\r
  IN PHYSICAL_ADDRESS       CallerAddress,\r
  IN MEMORY_PROFILE_ACTION  Action,\r
  IN UINTN                  Size,\r
  IN VOID                   *Buffer\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT           *Context;\r
  MEMORY_PROFILE_DRIVER_INFO       *DriverInfo;\r
  MEMORY_PROFILE_ALLOC_INFO        *AllocInfo;\r
  MEMORY_PROFILE_CONTEXT_DATA      *ContextData;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA  *DriverInfoData;\r
  LIST_ENTRY                       *DriverLink;\r
  LIST_ENTRY                       *DriverInfoList;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA  *ThisDriverInfoData;\r
  MEMORY_PROFILE_ALLOC_INFO_DATA   *AllocInfoData;\r
  EFI_MEMORY_TYPE                  ProfileMemoryIndex;\r
  MEMORY_PROFILE_ACTION            BasicAction;\r
  BOOLEAN                          Found;\r
\r
  BasicAction = Action & MEMORY_PROFILE_ACTION_BASIC_MASK;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  DriverInfoData = GetMemoryProfileDriverInfoFromAddress (ContextData, CallerAddress);\r
\r
  //\r
  // Do not return if DriverInfoData == NULL here,\r
  // because driver A might free memory allocated by driver B.\r
  //\r
\r
  //\r
  // Need use do-while loop to find all possible record,\r
  // because one address might be recorded multiple times.\r
  //\r
  Found = FALSE;\r
  AllocInfoData = NULL;\r
  do {\r
    if (DriverInfoData != NULL) {\r
      switch (BasicAction) {\r
        case MemoryProfileActionFreePages:\r
          AllocInfoData = GetMemoryProfileAllocInfoFromAddress (DriverInfoData, MemoryProfileActionAllocatePages, Size, Buffer);\r
          break;\r
        case MemoryProfileActionFreePool:\r
          AllocInfoData = GetMemoryProfileAllocInfoFromAddress (DriverInfoData, MemoryProfileActionAllocatePool, 0, Buffer);\r
          break;\r
        default:\r
          ASSERT (FALSE);\r
          AllocInfoData = NULL;\r
          break;\r
      }\r
    }\r
    if (AllocInfoData == NULL) {\r
      //\r
      // Legal case, because driver A might free memory allocated by driver B, by some protocol.\r
      //\r
      DriverInfoList = ContextData->DriverInfoList;\r
\r
      for (DriverLink = DriverInfoList->ForwardLink;\r
           DriverLink != DriverInfoList;\r
           DriverLink = DriverLink->ForwardLink) {\r
        ThisDriverInfoData = CR (\r
                               DriverLink,\r
                               MEMORY_PROFILE_DRIVER_INFO_DATA,\r
                               Link,\r
                               MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
                               );\r
        switch (BasicAction) {\r
          case MemoryProfileActionFreePages:\r
            AllocInfoData = GetMemoryProfileAllocInfoFromAddress (ThisDriverInfoData, MemoryProfileActionAllocatePages, Size, Buffer);\r
            break;\r
          case MemoryProfileActionFreePool:\r
            AllocInfoData = GetMemoryProfileAllocInfoFromAddress (ThisDriverInfoData, MemoryProfileActionAllocatePool, 0, Buffer);\r
            break;\r
          default:\r
            ASSERT (FALSE);\r
            AllocInfoData = NULL;\r
            break;\r
        }\r
        if (AllocInfoData != NULL) {\r
          DriverInfoData = ThisDriverInfoData;\r
          break;\r
        }\r
      }\r
\r
      if (AllocInfoData == NULL) {\r
        //\r
        // If (!Found), no matched allocate info is found for this free action.\r
        // It is because the specified memory type allocate actions have been filtered by\r
        // CoreNeedRecordProfile(), but free actions have no memory type information,\r
        // they can not be filtered by CoreNeedRecordProfile(). Then, they will be\r
        // filtered here.\r
        //\r
        // If (Found), it is normal exit path.\r
        return (Found ? EFI_SUCCESS : EFI_NOT_FOUND);\r
      }\r
    }\r
\r
    ASSERT (DriverInfoData != NULL);\r
    ASSERT (AllocInfoData != NULL);\r
\r
    Found = TRUE;\r
\r
    Context = &ContextData->Context;\r
    DriverInfo = &DriverInfoData->DriverInfo;\r
    AllocInfo = &AllocInfoData->AllocInfo;\r
\r
    DriverInfo->AllocRecordCount --;\r
    //\r
    // Update summary if and only if it is basic action.\r
    //\r
    if (AllocInfo->Action == (AllocInfo->Action & MEMORY_PROFILE_ACTION_BASIC_MASK)) {\r
      ProfileMemoryIndex = GetProfileMemoryIndex (AllocInfo->MemoryType);\r
\r
      Context->CurrentTotalUsage -= AllocInfo->Size;\r
      Context->CurrentTotalUsageByType[ProfileMemoryIndex] -= AllocInfo->Size;\r
\r
      DriverInfo->CurrentUsage -= AllocInfo->Size;\r
      DriverInfo->CurrentUsageByType[ProfileMemoryIndex] -= AllocInfo->Size;\r
    }\r
\r
    RemoveEntryList (&AllocInfoData->Link);\r
\r
    if (BasicAction == MemoryProfileActionFreePages) {\r
      if (AllocInfo->Buffer != (PHYSICAL_ADDRESS) (UINTN) Buffer) {\r
        SmmCoreUpdateProfileAllocate (\r
          AllocInfo->CallerAddress,\r
          AllocInfo->Action,\r
          AllocInfo->MemoryType,\r
          (UINTN) ((PHYSICAL_ADDRESS) (UINTN) Buffer - AllocInfo->Buffer),\r
          (VOID *) (UINTN) AllocInfo->Buffer,\r
          AllocInfoData->ActionString\r
          );\r
      }\r
      if (AllocInfo->Buffer + AllocInfo->Size != ((PHYSICAL_ADDRESS) (UINTN) Buffer + Size)) {\r
        SmmCoreUpdateProfileAllocate (\r
          AllocInfo->CallerAddress,\r
          AllocInfo->Action,\r
          AllocInfo->MemoryType,\r
          (UINTN) ((AllocInfo->Buffer + AllocInfo->Size) - ((PHYSICAL_ADDRESS) (UINTN) Buffer + Size)),\r
          (VOID *) ((UINTN) Buffer + Size),\r
          AllocInfoData->ActionString\r
          );\r
      }\r
    }\r
\r
    //\r
    // Use SmmInternalFreePool() that will not update profile for this FreePool action.\r
    //\r
    SmmInternalFreePool (AllocInfoData);\r
  } while (TRUE);\r
}\r
\r
/**\r
  Update SMRAM profile information.\r
\r
  @param CallerAddress  Address of caller who call Allocate or Free.\r
  @param Action         This Allocate or Free action.\r
  @param MemoryType     Memory type.\r
                        EfiMaxMemoryType means the MemoryType is unknown.\r
  @param Size           Buffer size.\r
  @param Buffer         Buffer address.\r
  @param ActionString   String for memory profile action.\r
                        Only needed for user defined allocate action.\r
\r
  @return EFI_SUCCESS           Memory profile is updated.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required,\r
                                or memory profile for the memory type is not required.\r
  @return EFI_ACCESS_DENIED     It is during memory profile data getting.\r
  @return EFI_ABORTED           Memory profile recording is not enabled.\r
  @return EFI_OUT_OF_RESOURCES  No enough resource to update memory profile for allocate action.\r
  @return EFI_NOT_FOUND         No matched allocate info found for free action.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmmCoreUpdateProfile (\r
  IN PHYSICAL_ADDRESS       CallerAddress,\r
  IN MEMORY_PROFILE_ACTION  Action,\r
  IN EFI_MEMORY_TYPE        MemoryType, // Valid for AllocatePages/AllocatePool\r
  IN UINTN                  Size,       // Valid for AllocatePages/FreePages/AllocatePool\r
  IN VOID                   *Buffer,\r
  IN CHAR8                  *ActionString OPTIONAL\r
  )\r
{\r
  EFI_STATUS                    Status;\r
  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
  MEMORY_PROFILE_ACTION         BasicAction;\r
\r
  if (!IS_SMRAM_PROFILE_ENABLED) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  if (mSmramProfileGettingStatus) {\r
    return EFI_ACCESS_DENIED;\r
  }\r
\r
  if (!mSmramProfileRecordingEnable) {\r
    return EFI_ABORTED;\r
  }\r
\r
  //\r
  // Get the basic action to know how to process the record\r
  //\r
  BasicAction = Action & MEMORY_PROFILE_ACTION_BASIC_MASK;\r
\r
  //\r
  // Free operations have no memory type information, so skip the check.\r
  //\r
  if ((BasicAction == MemoryProfileActionAllocatePages) || (BasicAction == MemoryProfileActionAllocatePool)) {\r
    //\r
    // Only record limited MemoryType.\r
    //\r
    if (!SmmCoreNeedRecordProfile (MemoryType)) {\r
      return EFI_UNSUPPORTED;\r
    }\r
  }\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  switch (BasicAction) {\r
    case MemoryProfileActionAllocatePages:\r
      Status = SmmCoreUpdateProfileAllocate (CallerAddress, Action, MemoryType, Size, Buffer, ActionString);\r
      break;\r
    case MemoryProfileActionFreePages:\r
      Status = SmmCoreUpdateProfileFree (CallerAddress, Action, Size, Buffer);\r
      break;\r
    case MemoryProfileActionAllocatePool:\r
      Status = SmmCoreUpdateProfileAllocate (CallerAddress, Action, MemoryType, Size, Buffer, ActionString);\r
      break;\r
    case MemoryProfileActionFreePool:\r
      Status = SmmCoreUpdateProfileFree (CallerAddress, Action, 0, Buffer);\r
      break;\r
    default:\r
      ASSERT (FALSE);\r
      Status = EFI_UNSUPPORTED;\r
      break;\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  SMRAM profile ready to lock callback function.\r
\r
**/\r
VOID\r
SmramProfileReadyToLock (\r
  VOID\r
  )\r
{\r
  if (!IS_SMRAM_PROFILE_ENABLED) {\r
    return;\r
  }\r
\r
  DEBUG ((EFI_D_INFO, "SmramProfileReadyToLock\n"));\r
  mSmramReadyToLock = TRUE;\r
}\r
\r
////////////////////\r
\r
/**\r
  Get SMRAM profile data size.\r
\r
  @return SMRAM profile data size.\r
\r
**/\r
UINTN\r
SmramProfileGetDataSize (\r
  VOID\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  MEMORY_PROFILE_ALLOC_INFO_DATA    *AllocInfoData;\r
  LIST_ENTRY                        *DriverInfoList;\r
  LIST_ENTRY                        *DriverLink;\r
  LIST_ENTRY                        *AllocInfoList;\r
  LIST_ENTRY                        *AllocLink;\r
  UINTN                             TotalSize;\r
  LIST_ENTRY                        *Node;\r
  LIST_ENTRY                        *FreePageList;\r
  LIST_ENTRY                        *FreePoolList;\r
  FREE_POOL_HEADER                  *Pool;\r
  UINTN                             PoolListIndex;\r
  UINTN                             Index;\r
  UINTN                             SmmPoolTypeIndex;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return 0;\r
  }\r
\r
  TotalSize = sizeof (MEMORY_PROFILE_CONTEXT);\r
\r
  DriverInfoList = ContextData->DriverInfoList;\r
  for (DriverLink = DriverInfoList->ForwardLink;\r
       DriverLink != DriverInfoList;\r
       DriverLink = DriverLink->ForwardLink) {\r
    DriverInfoData = CR (\r
                   DriverLink,\r
                   MEMORY_PROFILE_DRIVER_INFO_DATA,\r
                   Link,\r
                   MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
                   );\r
    TotalSize += DriverInfoData->DriverInfo.Header.Length;\r
\r
    AllocInfoList = DriverInfoData->AllocInfoList;\r
    for (AllocLink = AllocInfoList->ForwardLink;\r
         AllocLink != AllocInfoList;\r
         AllocLink = AllocLink->ForwardLink) {\r
      AllocInfoData = CR (\r
                        AllocLink,\r
                        MEMORY_PROFILE_ALLOC_INFO_DATA,\r
                        Link,\r
                        MEMORY_PROFILE_ALLOC_INFO_SIGNATURE\r
                        );\r
      TotalSize += AllocInfoData->AllocInfo.Header.Length;\r
    }\r
  }\r
\r
\r
  Index = 0;\r
  FreePageList = &mSmmMemoryMap;\r
  for (Node = FreePageList->BackLink;\r
       Node != FreePageList;\r
       Node = Node->BackLink) {\r
    Index++;\r
  }\r
  for (SmmPoolTypeIndex = 0; SmmPoolTypeIndex < SmmPoolTypeMax; SmmPoolTypeIndex++) {\r
    for (PoolListIndex = 0; PoolListIndex < MAX_POOL_INDEX; PoolListIndex++) {\r
      FreePoolList = &mSmmPoolLists[SmmPoolTypeIndex][PoolListIndex];\r
      for (Node = FreePoolList->BackLink;\r
           Node != FreePoolList;\r
           Node = Node->BackLink) {\r
        Pool = BASE_CR (Node, FREE_POOL_HEADER, Link);\r
        if (Pool->Header.Available) {\r
          Index++;\r
        }\r
      }\r
    }\r
  }\r
\r
  TotalSize += (sizeof (MEMORY_PROFILE_FREE_MEMORY) + Index * sizeof (MEMORY_PROFILE_DESCRIPTOR));\r
  TotalSize += (sizeof (MEMORY_PROFILE_MEMORY_RANGE) + mFullSmramRangeCount * sizeof (MEMORY_PROFILE_DESCRIPTOR));\r
\r
  return TotalSize;\r
}\r
\r
/**\r
  Copy SMRAM profile data.\r
\r
  @param ProfileBuffer  The buffer to hold SMRAM profile data.\r
  @param ProfileSize    On input, profile buffer size.\r
                        On output, actual profile data size copied.\r
  @param ProfileOffset  On input, profile buffer offset to copy.\r
                        On output, next time profile buffer offset to copy.\r
\r
**/\r
VOID\r
SmramProfileCopyData (\r
  OUT VOID      *ProfileBuffer,\r
  IN OUT UINT64 *ProfileSize,\r
  IN OUT UINT64 *ProfileOffset\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT           *Context;\r
  MEMORY_PROFILE_DRIVER_INFO       *DriverInfo;\r
  MEMORY_PROFILE_ALLOC_INFO        *AllocInfo;\r
  MEMORY_PROFILE_CONTEXT_DATA      *ContextData;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA  *DriverInfoData;\r
  MEMORY_PROFILE_ALLOC_INFO_DATA   *AllocInfoData;\r
  LIST_ENTRY                      *DriverInfoList;\r
  LIST_ENTRY                      *DriverLink;\r
  LIST_ENTRY                      *AllocInfoList;\r
  LIST_ENTRY                      *AllocLink;\r
  LIST_ENTRY                      *Node;\r
  FREE_PAGE_LIST                  *Pages;\r
  LIST_ENTRY                      *FreePageList;\r
  LIST_ENTRY                      *FreePoolList;\r
  FREE_POOL_HEADER                *Pool;\r
  UINTN                           PoolListIndex;\r
  UINT32                          Index;\r
  MEMORY_PROFILE_FREE_MEMORY      *FreeMemory;\r
  MEMORY_PROFILE_MEMORY_RANGE     *MemoryRange;\r
  MEMORY_PROFILE_DESCRIPTOR       *MemoryProfileDescriptor;\r
  UINT64                          Offset;\r
  UINT64                          RemainingSize;\r
  UINTN                           PdbSize;\r
  UINTN                           ActionStringSize;\r
  UINTN                           SmmPoolTypeIndex;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return ;\r
  }\r
\r
  RemainingSize = *ProfileSize;\r
  Offset = 0;\r
\r
  if (*ProfileOffset < sizeof (MEMORY_PROFILE_CONTEXT)) {\r
    if (RemainingSize >= sizeof (MEMORY_PROFILE_CONTEXT)) {\r
      Context = ProfileBuffer;\r
      CopyMem (Context, &ContextData->Context, sizeof (MEMORY_PROFILE_CONTEXT));\r
      RemainingSize -= sizeof (MEMORY_PROFILE_CONTEXT);\r
      ProfileBuffer = (UINT8 *) ProfileBuffer + sizeof (MEMORY_PROFILE_CONTEXT);\r
    } else {\r
      goto Done;\r
    }\r
  }\r
  Offset += sizeof (MEMORY_PROFILE_CONTEXT);\r
\r
  DriverInfoList = ContextData->DriverInfoList;\r
  for (DriverLink = DriverInfoList->ForwardLink;\r
       DriverLink != DriverInfoList;\r
       DriverLink = DriverLink->ForwardLink) {\r
    DriverInfoData = CR (\r
                       DriverLink,\r
                       MEMORY_PROFILE_DRIVER_INFO_DATA,\r
                       Link,\r
                       MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
                       );\r
    if (*ProfileOffset < (Offset + DriverInfoData->DriverInfo.Header.Length)) {\r
      if (RemainingSize >= DriverInfoData->DriverInfo.Header.Length) {\r
        DriverInfo = ProfileBuffer;\r
        CopyMem (DriverInfo, &DriverInfoData->DriverInfo, sizeof (MEMORY_PROFILE_DRIVER_INFO));\r
        if (DriverInfo->PdbStringOffset != 0) {\r
          PdbSize = AsciiStrSize (DriverInfoData->PdbString);\r
          CopyMem ((VOID *) ((UINTN) DriverInfo + DriverInfo->PdbStringOffset), DriverInfoData->PdbString, PdbSize);\r
        }\r
        RemainingSize -= DriverInfo->Header.Length;\r
        ProfileBuffer = (UINT8 *) ProfileBuffer + DriverInfo->Header.Length;\r
      } else {\r
        goto Done;\r
      }\r
    }\r
    Offset += DriverInfoData->DriverInfo.Header.Length;\r
\r
    AllocInfoList = DriverInfoData->AllocInfoList;\r
    for (AllocLink = AllocInfoList->ForwardLink;\r
         AllocLink != AllocInfoList;\r
         AllocLink = AllocLink->ForwardLink) {\r
      AllocInfoData = CR (\r
                        AllocLink,\r
                        MEMORY_PROFILE_ALLOC_INFO_DATA,\r
                        Link,\r
                        MEMORY_PROFILE_ALLOC_INFO_SIGNATURE\r
                        );\r
      if (*ProfileOffset < (Offset + AllocInfoData->AllocInfo.Header.Length)) {\r
        if (RemainingSize >= AllocInfoData->AllocInfo.Header.Length) {\r
          AllocInfo = ProfileBuffer;\r
          CopyMem (AllocInfo, &AllocInfoData->AllocInfo, sizeof (MEMORY_PROFILE_ALLOC_INFO));\r
          if (AllocInfo->ActionStringOffset) {\r
            ActionStringSize = AsciiStrSize (AllocInfoData->ActionString);\r
            CopyMem ((VOID *) ((UINTN) AllocInfo + AllocInfo->ActionStringOffset), AllocInfoData->ActionString, ActionStringSize);\r
          }\r
          RemainingSize -= AllocInfo->Header.Length;\r
          ProfileBuffer = (UINT8 *) ProfileBuffer + AllocInfo->Header.Length;\r
        } else {\r
          goto Done;\r
        }\r
      }\r
      Offset += AllocInfoData->AllocInfo.Header.Length;\r
    }\r
  }\r
\r
\r
  if (*ProfileOffset < (Offset + sizeof (MEMORY_PROFILE_FREE_MEMORY))) {\r
    if (RemainingSize >= sizeof (MEMORY_PROFILE_FREE_MEMORY)) {\r
      FreeMemory = ProfileBuffer;\r
      CopyMem (FreeMemory, &mSmramFreeMemory, sizeof (MEMORY_PROFILE_FREE_MEMORY));\r
      Index = 0;\r
      FreePageList = &mSmmMemoryMap;\r
      for (Node = FreePageList->BackLink;\r
           Node != FreePageList;\r
           Node = Node->BackLink) {\r
        Index++;\r
      }\r
      for (SmmPoolTypeIndex = 0; SmmPoolTypeIndex < SmmPoolTypeMax; SmmPoolTypeIndex++) {\r
        for (PoolListIndex = 0; PoolListIndex < MAX_POOL_INDEX; PoolListIndex++) {\r
          FreePoolList = &mSmmPoolLists[SmmPoolTypeIndex][MAX_POOL_INDEX - PoolListIndex - 1];\r
          for (Node = FreePoolList->BackLink;\r
               Node != FreePoolList;\r
               Node = Node->BackLink) {\r
            Pool = BASE_CR (Node, FREE_POOL_HEADER, Link);\r
            if (Pool->Header.Available) {\r
              Index++;\r
            }\r
          }\r
        }\r
      }\r
      FreeMemory->FreeMemoryEntryCount = Index;\r
\r
      RemainingSize -= sizeof (MEMORY_PROFILE_FREE_MEMORY);\r
      ProfileBuffer = (UINT8 *) ProfileBuffer + sizeof (MEMORY_PROFILE_FREE_MEMORY);\r
    } else {\r
      goto Done;\r
    }\r
  }\r
  Offset += sizeof (MEMORY_PROFILE_FREE_MEMORY);\r
  FreePageList = &mSmmMemoryMap;\r
  for (Node = FreePageList->BackLink;\r
       Node != FreePageList;\r
       Node = Node->BackLink) {\r
    if (*ProfileOffset < (Offset + sizeof (MEMORY_PROFILE_DESCRIPTOR))) {\r
      if (RemainingSize >= sizeof (MEMORY_PROFILE_DESCRIPTOR)) {\r
        Pages = BASE_CR (Node, FREE_PAGE_LIST, Link);\r
        MemoryProfileDescriptor = ProfileBuffer;\r
        MemoryProfileDescriptor->Header.Signature = MEMORY_PROFILE_DESCRIPTOR_SIGNATURE;\r
        MemoryProfileDescriptor->Header.Length = sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
        MemoryProfileDescriptor->Header.Revision = MEMORY_PROFILE_DESCRIPTOR_REVISION;\r
        MemoryProfileDescriptor->Address = (PHYSICAL_ADDRESS) (UINTN) Pages;\r
        MemoryProfileDescriptor->Size = EFI_PAGES_TO_SIZE (Pages->NumberOfPages);\r
\r
        RemainingSize -= sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
        ProfileBuffer = (UINT8 *) ProfileBuffer + sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
      } else {\r
        goto Done;\r
      }\r
    }\r
    Offset += sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
  }\r
  for (SmmPoolTypeIndex = 0; SmmPoolTypeIndex < SmmPoolTypeMax; SmmPoolTypeIndex++) {\r
    for (PoolListIndex = 0; PoolListIndex < MAX_POOL_INDEX; PoolListIndex++) {\r
      FreePoolList = &mSmmPoolLists[SmmPoolTypeIndex][MAX_POOL_INDEX - PoolListIndex - 1];\r
      for (Node = FreePoolList->BackLink;\r
           Node != FreePoolList;\r
           Node = Node->BackLink) {\r
        Pool = BASE_CR (Node, FREE_POOL_HEADER, Link);\r
        if (Pool->Header.Available) {\r
          if (*ProfileOffset < (Offset + sizeof (MEMORY_PROFILE_DESCRIPTOR))) {\r
            if (RemainingSize >= sizeof (MEMORY_PROFILE_DESCRIPTOR)) {\r
              MemoryProfileDescriptor = ProfileBuffer;\r
              MemoryProfileDescriptor->Header.Signature = MEMORY_PROFILE_DESCRIPTOR_SIGNATURE;\r
              MemoryProfileDescriptor->Header.Length = sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
              MemoryProfileDescriptor->Header.Revision = MEMORY_PROFILE_DESCRIPTOR_REVISION;\r
              MemoryProfileDescriptor->Address = (PHYSICAL_ADDRESS) (UINTN) Pool;\r
              MemoryProfileDescriptor->Size = Pool->Header.Size;\r
\r
              RemainingSize -= sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
              ProfileBuffer = (UINT8 *) ProfileBuffer + sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
            } else {\r
              goto Done;\r
            }\r
          }\r
          Offset += sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
        }\r
      }\r
    }\r
  }\r
\r
  if (*ProfileOffset < (Offset + sizeof (MEMORY_PROFILE_MEMORY_RANGE))) {\r
    if (RemainingSize >= sizeof (MEMORY_PROFILE_MEMORY_RANGE)) {\r
      MemoryRange = ProfileBuffer;\r
      MemoryRange->Header.Signature = MEMORY_PROFILE_MEMORY_RANGE_SIGNATURE;\r
      MemoryRange->Header.Length = sizeof (MEMORY_PROFILE_MEMORY_RANGE);\r
      MemoryRange->Header.Revision = MEMORY_PROFILE_MEMORY_RANGE_REVISION;\r
      MemoryRange->MemoryRangeCount = (UINT32) mFullSmramRangeCount;\r
\r
      RemainingSize -= sizeof (MEMORY_PROFILE_MEMORY_RANGE);\r
      ProfileBuffer = (UINT8 *) ProfileBuffer + sizeof (MEMORY_PROFILE_MEMORY_RANGE);\r
    } else {\r
      goto Done;\r
    }\r
  }\r
  Offset += sizeof (MEMORY_PROFILE_MEMORY_RANGE);\r
  for (Index = 0; Index < mFullSmramRangeCount; Index++) {\r
    if (*ProfileOffset < (Offset + sizeof (MEMORY_PROFILE_DESCRIPTOR))) {\r
      if (RemainingSize >= sizeof (MEMORY_PROFILE_DESCRIPTOR)) {\r
        MemoryProfileDescriptor = ProfileBuffer;\r
        MemoryProfileDescriptor->Header.Signature = MEMORY_PROFILE_DESCRIPTOR_SIGNATURE;\r
        MemoryProfileDescriptor->Header.Length = sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
        MemoryProfileDescriptor->Header.Revision = MEMORY_PROFILE_DESCRIPTOR_REVISION;\r
        MemoryProfileDescriptor->Address = mFullSmramRanges[Index].PhysicalStart;\r
        MemoryProfileDescriptor->Size = mFullSmramRanges[Index].PhysicalSize;\r
\r
        RemainingSize -= sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
        ProfileBuffer = (UINT8 *) ProfileBuffer + sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
      } else {\r
        goto Done;\r
      }\r
    }\r
    Offset += sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
  }\r
\r
Done:\r
  //\r
  // On output, actual profile data size copied.\r
  //\r
  *ProfileSize -= RemainingSize;\r
  //\r
  // On output, next time profile buffer offset to copy.\r
  //\r
  *ProfileOffset = Offset;\r
}\r
\r
/**\r
  Get memory profile data.\r
\r
  @param[in]      This              The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in, out] ProfileSize       On entry, points to the size in bytes of the ProfileBuffer.\r
                                    On return, points to the size of the data returned in ProfileBuffer.\r
  @param[out]     ProfileBuffer     Profile buffer.\r
\r
  @return EFI_SUCCESS               Get the memory profile data successfully.\r
  @return EFI_UNSUPPORTED           Memory profile is unsupported.\r
  @return EFI_BUFFER_TO_SMALL       The ProfileSize is too small for the resulting data.\r
                                    ProfileSize is updated with the size required.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolGetData (\r
  IN     EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN OUT UINT64                             *ProfileSize,\r
     OUT VOID                               *ProfileBuffer\r
  )\r
{\r
  UINT64                                Size;\r
  UINT64                                Offset;\r
  MEMORY_PROFILE_CONTEXT_DATA           *ContextData;\r
  BOOLEAN                               SmramProfileGettingStatus;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
  mSmramProfileGettingStatus = TRUE;\r
\r
  Size = SmramProfileGetDataSize ();\r
\r
  if (*ProfileSize < Size) {\r
    *ProfileSize = Size;\r
    mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
    return EFI_BUFFER_TOO_SMALL;\r
  }\r
\r
  Offset = 0;\r
  SmramProfileCopyData (ProfileBuffer, &Size, &Offset);\r
  *ProfileSize = Size;\r
\r
  mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Register image to memory profile.\r
\r
  @param[in] This               The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in] FilePath           File path of the image.\r
  @param[in] ImageBase          Image base address.\r
  @param[in] ImageSize          Image size.\r
  @param[in] FileType           File type of the image.\r
\r
  @return EFI_SUCCESS           Register successfully.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required.\r
  @return EFI_OUT_OF_RESOURCES  No enough resource for this register.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolRegisterImage (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN EFI_DEVICE_PATH_PROTOCOL           *FilePath,\r
  IN PHYSICAL_ADDRESS                   ImageBase,\r
  IN UINT64                             ImageSize,\r
  IN EFI_FV_FILETYPE                    FileType\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_SMM_DRIVER_ENTRY              DriverEntry;\r
  VOID                              *EntryPointInImage;\r
  EFI_GUID                          *Name;\r
\r
  ZeroMem (&DriverEntry, sizeof (DriverEntry));\r
  Name = GetFileNameFromFilePath (FilePath);\r
  if (Name != NULL) {\r
    CopyMem (&DriverEntry.FileName, Name, sizeof (EFI_GUID));\r
  }\r
  DriverEntry.ImageBuffer = ImageBase;\r
  DriverEntry.NumberOfPage = EFI_SIZE_TO_PAGES ((UINTN) ImageSize);\r
  Status = InternalPeCoffGetEntryPoint ((VOID *) (UINTN) DriverEntry.ImageBuffer, &EntryPointInImage);\r
  ASSERT_EFI_ERROR (Status);\r
  DriverEntry.ImageEntryPoint = (PHYSICAL_ADDRESS) (UINTN) EntryPointInImage;\r
\r
  return RegisterSmramProfileImage (&DriverEntry, FALSE);\r
}\r
\r
/**\r
  Unregister image from memory profile.\r
\r
  @param[in] This               The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in] FilePath           File path of the image.\r
  @param[in] ImageBase          Image base address.\r
  @param[in] ImageSize          Image size.\r
\r
  @return EFI_SUCCESS           Unregister successfully.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required.\r
  @return EFI_NOT_FOUND         The image is not found.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolUnregisterImage (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN EFI_DEVICE_PATH_PROTOCOL           *FilePath,\r
  IN PHYSICAL_ADDRESS                   ImageBase,\r
  IN UINT64                             ImageSize\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_SMM_DRIVER_ENTRY              DriverEntry;\r
  VOID                              *EntryPointInImage;\r
  EFI_GUID                          *Name;\r
\r
  ZeroMem (&DriverEntry, sizeof (DriverEntry));\r
  Name = GetFileNameFromFilePath (FilePath);\r
  if (Name != NULL) {\r
    CopyMem (&DriverEntry.FileName, Name, sizeof (EFI_GUID));\r
  }\r
  DriverEntry.ImageBuffer = ImageBase;\r
  DriverEntry.NumberOfPage = EFI_SIZE_TO_PAGES ((UINTN) ImageSize);\r
  Status = InternalPeCoffGetEntryPoint ((VOID *) (UINTN) DriverEntry.ImageBuffer, &EntryPointInImage);\r
  ASSERT_EFI_ERROR (Status);\r
  DriverEntry.ImageEntryPoint = (PHYSICAL_ADDRESS) (UINTN) EntryPointInImage;\r
\r
  return UnregisterSmramProfileImage (&DriverEntry, FALSE);\r
}\r
\r
/**\r
  Get memory profile recording state.\r
\r
  @param[in]  This              The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[out] RecordingState    Recording state.\r
\r
  @return EFI_SUCCESS           Memory profile recording state is returned.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported.\r
  @return EFI_INVALID_PARAMETER RecordingState is NULL.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolGetRecordingState (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  OUT BOOLEAN                           *RecordingState\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT_DATA           *ContextData;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  if (RecordingState == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  *RecordingState = mSmramProfileRecordingEnable;\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Set memory profile recording state.\r
\r
  @param[in] This               The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in] RecordingState     Recording state.\r
\r
  @return EFI_SUCCESS           Set memory profile recording state successfully.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolSetRecordingState (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN BOOLEAN                            RecordingState\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT_DATA           *ContextData;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  mSmramProfileRecordingEnable = RecordingState;\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Record memory profile of multilevel caller.\r
\r
  @param[in] This               The EDKII_SMM_MEMORY_PROFILE_PROTOCOL instance.\r
  @param[in] CallerAddress      Address of caller.\r
  @param[in] Action             Memory profile action.\r
  @param[in] MemoryType         Memory type.\r
                                EfiMaxMemoryType means the MemoryType is unknown.\r
  @param[in] Buffer             Buffer address.\r
  @param[in] Size               Buffer size.\r
  @param[in] ActionString       String for memory profile action.\r
                                Only needed for user defined allocate action.\r
\r
  @return EFI_SUCCESS           Memory profile is updated.\r
  @return EFI_UNSUPPORTED       Memory profile is unsupported,\r
                                or memory profile for the image is not required,\r
                                or memory profile for the memory type is not required.\r
  @return EFI_ACCESS_DENIED     It is during memory profile data getting.\r
  @return EFI_ABORTED           Memory profile recording is not enabled.\r
  @return EFI_OUT_OF_RESOURCES  No enough resource to update memory profile for allocate action.\r
  @return EFI_NOT_FOUND         No matched allocate info found for free action.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileProtocolRecord (\r
  IN EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
  IN PHYSICAL_ADDRESS                   CallerAddress,\r
  IN MEMORY_PROFILE_ACTION              Action,\r
  IN EFI_MEMORY_TYPE                    MemoryType,\r
  IN VOID                               *Buffer,\r
  IN UINTN                              Size,\r
  IN CHAR8                              *ActionString OPTIONAL\r
  )\r
{\r
  return SmmCoreUpdateProfile (CallerAddress, Action, MemoryType, Size, Buffer, ActionString);\r
}\r
\r
/**\r
  SMRAM profile handler to get profile info.\r
\r
  @param SmramProfileParameterGetInfo The parameter of SMM profile get size.\r
\r
**/\r
VOID\r
SmramProfileHandlerGetInfo (\r
  IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_INFO   *SmramProfileParameterGetInfo\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
  BOOLEAN                       SmramProfileGettingStatus;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return ;\r
  }\r
\r
  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
  mSmramProfileGettingStatus = TRUE;\r
\r
  SmramProfileParameterGetInfo->ProfileSize = SmramProfileGetDataSize();\r
  SmramProfileParameterGetInfo->Header.ReturnStatus = 0;\r
\r
  mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
  SMRAM profile handler to get profile data.\r
\r
  @param SmramProfileParameterGetData The parameter of SMM profile get data.\r
\r
**/\r
VOID\r
SmramProfileHandlerGetData (\r
  IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA   *SmramProfileParameterGetData\r
  )\r
{\r
  UINT64                                    ProfileSize;\r
  UINT64                                    ProfileOffset;\r
  SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA  SmramProfileGetData;\r
  MEMORY_PROFILE_CONTEXT_DATA               *ContextData;\r
  BOOLEAN                                   SmramProfileGettingStatus;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return ;\r
  }\r
\r
  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
  mSmramProfileGettingStatus = TRUE;\r
\r
\r
  CopyMem (&SmramProfileGetData, SmramProfileParameterGetData, sizeof (SmramProfileGetData));\r
\r
  ProfileSize = SmramProfileGetDataSize();\r
\r
  //\r
  // Sanity check\r
  //\r
  if (!SmmIsBufferOutsideSmmValid ((UINTN) SmramProfileGetData.ProfileBuffer, (UINTN) ProfileSize)) {\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerGetData: SMM ProfileBuffer in SMRAM or overflow!\n"));\r
    SmramProfileParameterGetData->ProfileSize = ProfileSize;\r
    SmramProfileParameterGetData->Header.ReturnStatus = (UINT64) (INT64) (INTN) EFI_ACCESS_DENIED;\r
    goto Done;\r
  }\r
\r
  if (SmramProfileGetData.ProfileSize < ProfileSize) {\r
    SmramProfileParameterGetData->ProfileSize = ProfileSize;\r
    SmramProfileParameterGetData->Header.ReturnStatus = (UINT64) (INT64) (INTN) EFI_BUFFER_TOO_SMALL;\r
    goto Done;\r
  }\r
\r
  ProfileOffset = 0;\r
  SmramProfileCopyData ((VOID *) (UINTN) SmramProfileGetData.ProfileBuffer, &ProfileSize, &ProfileOffset);\r
  SmramProfileParameterGetData->ProfileSize = ProfileSize;\r
  SmramProfileParameterGetData->Header.ReturnStatus = 0;\r
\r
Done:\r
  mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
  SMRAM profile handler to get profile data by offset.\r
\r
  @param SmramProfileParameterGetDataByOffset   The parameter of SMM profile get data by offset.\r
\r
**/\r
VOID\r
SmramProfileHandlerGetDataByOffset (\r
  IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET     *SmramProfileParameterGetDataByOffset\r
  )\r
{\r
  SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET    SmramProfileGetDataByOffset;\r
  MEMORY_PROFILE_CONTEXT_DATA                           *ContextData;\r
  BOOLEAN                                               SmramProfileGettingStatus;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return ;\r
  }\r
\r
  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
  mSmramProfileGettingStatus = TRUE;\r
\r
\r
  CopyMem (&SmramProfileGetDataByOffset, SmramProfileParameterGetDataByOffset, sizeof (SmramProfileGetDataByOffset));\r
\r
  //\r
  // Sanity check\r
  //\r
  if (!SmmIsBufferOutsideSmmValid ((UINTN) SmramProfileGetDataByOffset.ProfileBuffer, (UINTN) SmramProfileGetDataByOffset.ProfileSize)) {\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerGetDataByOffset: SMM ProfileBuffer in SMRAM or overflow!\n"));\r
    SmramProfileParameterGetDataByOffset->Header.ReturnStatus = (UINT64) (INT64) (INTN) EFI_ACCESS_DENIED;\r
    goto Done;\r
  }\r
\r
  SmramProfileCopyData ((VOID *) (UINTN) SmramProfileGetDataByOffset.ProfileBuffer, &SmramProfileGetDataByOffset.ProfileSize, &SmramProfileGetDataByOffset.ProfileOffset);\r
  CopyMem (SmramProfileParameterGetDataByOffset, &SmramProfileGetDataByOffset, sizeof (SmramProfileGetDataByOffset));\r
  SmramProfileParameterGetDataByOffset->Header.ReturnStatus = 0;\r
\r
Done:\r
  mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
  SMRAM profile handler to register SMM image.\r
\r
  @param SmramProfileParameterRegisterImage The parameter of SMM profile register image.\r
\r
**/\r
VOID\r
SmramProfileHandlerRegisterImage (\r
  IN SMRAM_PROFILE_PARAMETER_REGISTER_IMAGE *SmramProfileParameterRegisterImage\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_SMM_DRIVER_ENTRY              DriverEntry;\r
  VOID                              *EntryPointInImage;\r
\r
  ZeroMem (&DriverEntry, sizeof (DriverEntry));\r
  CopyMem (&DriverEntry.FileName, &SmramProfileParameterRegisterImage->FileName, sizeof(EFI_GUID));\r
  DriverEntry.ImageBuffer = SmramProfileParameterRegisterImage->ImageBuffer;\r
  DriverEntry.NumberOfPage = (UINTN) SmramProfileParameterRegisterImage->NumberOfPage;\r
  Status = InternalPeCoffGetEntryPoint ((VOID *) (UINTN) DriverEntry.ImageBuffer, &EntryPointInImage);\r
  ASSERT_EFI_ERROR (Status);\r
  DriverEntry.ImageEntryPoint = (PHYSICAL_ADDRESS) (UINTN) EntryPointInImage;\r
\r
  Status = RegisterSmramProfileImage (&DriverEntry, FALSE);\r
  if (!EFI_ERROR (Status)) {\r
    SmramProfileParameterRegisterImage->Header.ReturnStatus = 0;\r
  }\r
}\r
\r
/**\r
  SMRAM profile handler to unregister SMM image.\r
\r
  @param SmramProfileParameterUnregisterImage The parameter of SMM profile unregister image.\r
\r
**/\r
VOID\r
SmramProfileHandlerUnregisterImage (\r
  IN SMRAM_PROFILE_PARAMETER_UNREGISTER_IMAGE *SmramProfileParameterUnregisterImage\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_SMM_DRIVER_ENTRY              DriverEntry;\r
  VOID                              *EntryPointInImage;\r
\r
  ZeroMem (&DriverEntry, sizeof (DriverEntry));\r
  CopyMem (&DriverEntry.FileName, &SmramProfileParameterUnregisterImage->FileName, sizeof (EFI_GUID));\r
  DriverEntry.ImageBuffer = SmramProfileParameterUnregisterImage->ImageBuffer;\r
  DriverEntry.NumberOfPage = (UINTN) SmramProfileParameterUnregisterImage->NumberOfPage;\r
  Status = InternalPeCoffGetEntryPoint ((VOID *) (UINTN) DriverEntry.ImageBuffer, &EntryPointInImage);\r
  ASSERT_EFI_ERROR (Status);\r
  DriverEntry.ImageEntryPoint = (PHYSICAL_ADDRESS) (UINTN) EntryPointInImage;\r
\r
  Status = UnregisterSmramProfileImage (&DriverEntry, FALSE);\r
  if (!EFI_ERROR (Status)) {\r
    SmramProfileParameterUnregisterImage->Header.ReturnStatus = 0;\r
  }\r
}\r
\r
/**\r
  Dispatch function for a Software SMI handler.\r
\r
  Caution: This function may receive untrusted input.\r
  Communicate buffer and buffer size are external input, so this function will do basic validation.\r
\r
  @param DispatchHandle  The unique handle assigned to this handler by SmiHandlerRegister().\r
  @param Context         Points to an optional handler context which was specified when the\r
                         handler was registered.\r
  @param CommBuffer      A pointer to a collection of data in memory that will\r
                         be conveyed from a non-SMM environment into an SMM environment.\r
  @param CommBufferSize  The size of the CommBuffer.\r
\r
  @retval EFI_SUCCESS Command is handled successfully.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
SmramProfileHandler (\r
  IN EFI_HANDLE  DispatchHandle,\r
  IN CONST VOID  *Context         OPTIONAL,\r
  IN OUT VOID    *CommBuffer      OPTIONAL,\r
  IN OUT UINTN   *CommBufferSize  OPTIONAL\r
  )\r
{\r
  SMRAM_PROFILE_PARAMETER_HEADER           *SmramProfileParameterHeader;\r
  UINTN                                    TempCommBufferSize;\r
  SMRAM_PROFILE_PARAMETER_RECORDING_STATE  *ParameterRecordingState;\r
\r
  DEBUG ((EFI_D_ERROR, "SmramProfileHandler Enter\n"));\r
\r
  //\r
  // If input is invalid, stop processing this SMI\r
  //\r
  if (CommBuffer == NULL || CommBufferSize == NULL) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  TempCommBufferSize = *CommBufferSize;\r
\r
  if (TempCommBufferSize < sizeof (SMRAM_PROFILE_PARAMETER_HEADER)) {\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
    return EFI_SUCCESS;\r
  }\r
\r
  if (mSmramReadyToLock && !SmmIsBufferOutsideSmmValid ((UINTN)CommBuffer, TempCommBufferSize)) {\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer in SMRAM or overflow!\n"));\r
    return EFI_SUCCESS;\r
  }\r
\r
  SmramProfileParameterHeader = (SMRAM_PROFILE_PARAMETER_HEADER *) ((UINTN) CommBuffer);\r
\r
  SmramProfileParameterHeader->ReturnStatus = (UINT64)-1;\r
\r
  if (GetSmramProfileContext () == NULL) {\r
    SmramProfileParameterHeader->ReturnStatus = (UINT64) (INT64) (INTN) EFI_UNSUPPORTED;\r
    return EFI_SUCCESS;\r
  }\r
\r
  switch (SmramProfileParameterHeader->Command) {\r
  case SMRAM_PROFILE_COMMAND_GET_PROFILE_INFO:\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerGetInfo\n"));\r
    if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_GET_PROFILE_INFO)) {\r
      DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
      return EFI_SUCCESS;\r
    }\r
    SmramProfileHandlerGetInfo ((SMRAM_PROFILE_PARAMETER_GET_PROFILE_INFO *) (UINTN) CommBuffer);\r
    break;\r
  case SMRAM_PROFILE_COMMAND_GET_PROFILE_DATA:\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerGetData\n"));\r
    if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA)) {\r
      DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
      return EFI_SUCCESS;\r
    }\r
    SmramProfileHandlerGetData ((SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA *) (UINTN) CommBuffer);\r
    break;\r
  case SMRAM_PROFILE_COMMAND_GET_PROFILE_DATA_BY_OFFSET:\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerGetDataByOffset\n"));\r
    if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET)) {\r
      DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
      return EFI_SUCCESS;\r
    }\r
    SmramProfileHandlerGetDataByOffset ((SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET *) (UINTN) CommBuffer);\r
    break;\r
  case SMRAM_PROFILE_COMMAND_REGISTER_IMAGE:\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerRegisterImage\n"));\r
    if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_REGISTER_IMAGE)) {\r
      DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
      return EFI_SUCCESS;\r
    }\r
    if (mSmramReadyToLock) {\r
      return EFI_SUCCESS;\r
    }\r
    SmramProfileHandlerRegisterImage ((SMRAM_PROFILE_PARAMETER_REGISTER_IMAGE *) (UINTN) CommBuffer);\r
    break;\r
  case SMRAM_PROFILE_COMMAND_UNREGISTER_IMAGE:\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerUnregisterImage\n"));\r
    if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_UNREGISTER_IMAGE)) {\r
      DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
      return EFI_SUCCESS;\r
    }\r
    if (mSmramReadyToLock) {\r
      return EFI_SUCCESS;\r
    }\r
    SmramProfileHandlerUnregisterImage ((SMRAM_PROFILE_PARAMETER_UNREGISTER_IMAGE *) (UINTN) CommBuffer);\r
    break;\r
  case SMRAM_PROFILE_COMMAND_GET_RECORDING_STATE:\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerGetRecordingState\n"));\r
    if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_RECORDING_STATE)) {\r
      DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
      return EFI_SUCCESS;\r
    }\r
    ParameterRecordingState = (SMRAM_PROFILE_PARAMETER_RECORDING_STATE *) (UINTN) CommBuffer;\r
    ParameterRecordingState->RecordingState = mSmramProfileRecordingEnable;\r
    ParameterRecordingState->Header.ReturnStatus = 0;\r
    break;\r
  case SMRAM_PROFILE_COMMAND_SET_RECORDING_STATE:\r
    DEBUG ((EFI_D_ERROR, "SmramProfileHandlerSetRecordingState\n"));\r
    if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_RECORDING_STATE)) {\r
      DEBUG ((EFI_D_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
      return EFI_SUCCESS;\r
    }\r
    ParameterRecordingState = (SMRAM_PROFILE_PARAMETER_RECORDING_STATE *) (UINTN) CommBuffer;\r
    mSmramProfileRecordingEnable = ParameterRecordingState->RecordingState;\r
    ParameterRecordingState->Header.ReturnStatus = 0;\r
    break;\r
\r
  default:\r
    break;\r
  }\r
\r
  DEBUG ((EFI_D_ERROR, "SmramProfileHandler Exit\n"));\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Register SMRAM profile handler.\r
\r
**/\r
VOID\r
RegisterSmramProfileHandler (\r
  VOID\r
  )\r
{\r
  EFI_STATUS    Status;\r
  EFI_HANDLE    DispatchHandle;\r
\r
  if (!IS_SMRAM_PROFILE_ENABLED) {\r
    return;\r
  }\r
\r
  Status = SmiHandlerRegister (\r
             SmramProfileHandler,\r
             &gEdkiiMemoryProfileGuid,\r
             &DispatchHandle\r
             );\r
  ASSERT_EFI_ERROR (Status);\r
}\r
\r
////////////////////\r
\r
/**\r
  Dump SMRAM range.\r
\r
**/\r
VOID\r
DumpSmramRange (\r
  VOID\r
  )\r
{\r
  UINTN                         Index;\r
  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
  BOOLEAN                       SmramProfileGettingStatus;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return ;\r
  }\r
\r
  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
  mSmramProfileGettingStatus = TRUE;\r
\r
  DEBUG ((EFI_D_INFO, "FullSmramRange address - 0x%08x\n", mFullSmramRanges));\r
\r
  DEBUG ((EFI_D_INFO, "======= SmramProfile begin =======\n"));\r
\r
  DEBUG ((EFI_D_INFO, "FullSmramRange:\n"));\r
  for (Index = 0; Index < mFullSmramRangeCount; Index++) {\r
    DEBUG ((EFI_D_INFO, "  FullSmramRange (0x%x)\n", Index));\r
    DEBUG ((EFI_D_INFO, "    PhysicalStart      - 0x%016lx\n", mFullSmramRanges[Index].PhysicalStart));\r
    DEBUG ((EFI_D_INFO, "    CpuStart           - 0x%016lx\n", mFullSmramRanges[Index].CpuStart));\r
    DEBUG ((EFI_D_INFO, "    PhysicalSize       - 0x%016lx\n", mFullSmramRanges[Index].PhysicalSize));\r
    DEBUG ((EFI_D_INFO, "    RegionState        - 0x%016lx\n", mFullSmramRanges[Index].RegionState));\r
  }\r
\r
  DEBUG ((EFI_D_INFO, "======= SmramProfile end =======\n"));\r
\r
  mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
  Dump SMRAM free page list.\r
\r
**/\r
VOID\r
DumpFreePagesList (\r
  VOID\r
  )\r
{\r
  LIST_ENTRY                    *FreePageList;\r
  LIST_ENTRY                    *Node;\r
  FREE_PAGE_LIST                *Pages;\r
  UINTN                         Index;\r
  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
  BOOLEAN                       SmramProfileGettingStatus;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return ;\r
  }\r
\r
  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
  mSmramProfileGettingStatus = TRUE;\r
\r
  DEBUG ((EFI_D_INFO, "======= SmramProfile begin =======\n"));\r
\r
  DEBUG ((EFI_D_INFO, "FreePagesList:\n"));\r
  FreePageList = &mSmmMemoryMap;\r
  for (Node = FreePageList->BackLink, Index = 0;\r
       Node != FreePageList;\r
       Node = Node->BackLink, Index++) {\r
    Pages = BASE_CR (Node, FREE_PAGE_LIST, Link);\r
    DEBUG ((EFI_D_INFO, "  Index - 0x%x\n", Index));\r
    DEBUG ((EFI_D_INFO, "    PhysicalStart - 0x%016lx\n", (PHYSICAL_ADDRESS) (UINTN) Pages));\r
    DEBUG ((EFI_D_INFO, "    NumberOfPages - 0x%08x\n", Pages->NumberOfPages));\r
  }\r
\r
  DEBUG ((EFI_D_INFO, "======= SmramProfile end =======\n"));\r
\r
  mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
  Dump SMRAM free pool list.\r
\r
**/\r
VOID\r
DumpFreePoolList (\r
  VOID\r
  )\r
{\r
  LIST_ENTRY                    *FreePoolList;\r
  LIST_ENTRY                    *Node;\r
  FREE_POOL_HEADER              *Pool;\r
  UINTN                         Index;\r
  UINTN                         PoolListIndex;\r
  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
  BOOLEAN                       SmramProfileGettingStatus;\r
  UINTN                         SmmPoolTypeIndex;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return ;\r
  }\r
\r
  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
  mSmramProfileGettingStatus = TRUE;\r
\r
  DEBUG ((DEBUG_INFO, "======= SmramProfile begin =======\n"));\r
\r
  for (SmmPoolTypeIndex = 0; SmmPoolTypeIndex < SmmPoolTypeMax; SmmPoolTypeIndex++) {\r
    for (PoolListIndex = 0; PoolListIndex < MAX_POOL_INDEX; PoolListIndex++) {\r
      DEBUG ((DEBUG_INFO, "FreePoolList(%d)(%d):\n", SmmPoolTypeIndex, PoolListIndex));\r
      FreePoolList = &mSmmPoolLists[SmmPoolTypeIndex][PoolListIndex];\r
      for (Node = FreePoolList->BackLink, Index = 0;\r
           Node != FreePoolList;\r
           Node = Node->BackLink, Index++) {\r
        Pool = BASE_CR (Node, FREE_POOL_HEADER, Link);\r
        DEBUG ((DEBUG_INFO, "  Index - 0x%x\n", Index));\r
        DEBUG ((DEBUG_INFO, "    PhysicalStart - 0x%016lx\n", (PHYSICAL_ADDRESS) (UINTN) Pool));\r
        DEBUG ((DEBUG_INFO, "    Size          - 0x%08x\n", Pool->Header.Size));\r
        DEBUG ((DEBUG_INFO, "    Available     - 0x%02x\n", Pool->Header.Available));\r
      }\r
    }\r
  }\r
\r
  DEBUG ((DEBUG_INFO, "======= SmramProfile end =======\n"));\r
\r
  mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
GLOBAL_REMOVE_IF_UNREFERENCED CHAR8 *mSmmActionString[] = {\r
  "SmmUnknown",\r
  "gSmst->SmmAllocatePages",\r
  "gSmst->SmmFreePages",\r
  "gSmst->SmmAllocatePool",\r
  "gSmst->SmmFreePool",\r
};\r
\r
typedef struct {\r
  MEMORY_PROFILE_ACTION  Action;\r
  CHAR8                 *String;\r
} ACTION_STRING;\r
\r
GLOBAL_REMOVE_IF_UNREFERENCED ACTION_STRING mExtActionString[] = {\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_PAGES,                    "Lib:AllocatePages"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_RUNTIME_PAGES,            "Lib:AllocateRuntimePages"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_RESERVED_PAGES,           "Lib:AllocateReservedPages"},\r
  {MEMORY_PROFILE_ACTION_LIB_FREE_PAGES,                        "Lib:FreePages"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_ALIGNED_PAGES,            "Lib:AllocateAlignedPages"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_ALIGNED_RUNTIME_PAGES,    "Lib:AllocateAlignedRuntimePages"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_ALIGNED_RESERVED_PAGES,   "Lib:AllocateAlignedReservedPages"},\r
  {MEMORY_PROFILE_ACTION_LIB_FREE_ALIGNED_PAGES,                "Lib:FreeAlignedPages"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_POOL,                     "Lib:AllocatePool"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_RUNTIME_POOL,             "Lib:AllocateRuntimePool"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_RESERVED_POOL,            "Lib:AllocateReservedPool"},\r
  {MEMORY_PROFILE_ACTION_LIB_FREE_POOL,                         "Lib:FreePool"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_ZERO_POOL,                "Lib:AllocateZeroPool"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_RUNTIME_ZERO_POOL,        "Lib:AllocateRuntimeZeroPool"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_RESERVED_ZERO_POOL,       "Lib:AllocateReservedZeroPool"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_COPY_POOL,                "Lib:AllocateCopyPool"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_RUNTIME_COPY_POOL,        "Lib:AllocateRuntimeCopyPool"},\r
  {MEMORY_PROFILE_ACTION_LIB_ALLOCATE_RESERVED_COPY_POOL,       "Lib:AllocateReservedCopyPool"},\r
  {MEMORY_PROFILE_ACTION_LIB_REALLOCATE_POOL,                   "Lib:ReallocatePool"},\r
  {MEMORY_PROFILE_ACTION_LIB_REALLOCATE_RUNTIME_POOL,           "Lib:ReallocateRuntimePool"},\r
  {MEMORY_PROFILE_ACTION_LIB_REALLOCATE_RESERVED_POOL,          "Lib:ReallocateReservedPool"},\r
};\r
\r
typedef struct {\r
  EFI_MEMORY_TYPE   MemoryType;\r
  CHAR8             *MemoryTypeStr;\r
} PROFILE_MEMORY_TYPE_STRING;\r
\r
GLOBAL_REMOVE_IF_UNREFERENCED PROFILE_MEMORY_TYPE_STRING mMemoryTypeString[] = {\r
  {EfiRuntimeServicesCode, "EfiRuntimeServicesCode"},\r
  {EfiRuntimeServicesData, "EfiRuntimeServicesData"}\r
};\r
\r
/**\r
  Memory type to string.\r
\r
  @param[in] MemoryType Memory type.\r
\r
  @return Pointer to string.\r
\r
**/\r
CHAR8 *\r
ProfileMemoryTypeToStr (\r
  IN EFI_MEMORY_TYPE    MemoryType\r
  )\r
{\r
  UINTN     Index;\r
  for (Index = 0; Index < ARRAY_SIZE (mMemoryTypeString); Index++) {\r
    if (mMemoryTypeString[Index].MemoryType == MemoryType) {\r
      return mMemoryTypeString[Index].MemoryTypeStr;\r
    }\r
  }\r
\r
  return "UnexpectedMemoryType";\r
}\r
\r
/**\r
  Action to string.\r
\r
  @param[in] Action                     Profile action.\r
\r
  @return Pointer to string.\r
\r
**/\r
CHAR8 *\r
ProfileActionToStr (\r
  IN MEMORY_PROFILE_ACTION  Action\r
  )\r
{\r
  UINTN     Index;\r
  UINTN     ActionStringCount;\r
  CHAR8     **ActionString;\r
\r
  ActionString = mSmmActionString;\r
  ActionStringCount = ARRAY_SIZE (mSmmActionString);\r
\r
  if ((UINTN) (UINT32) Action < ActionStringCount) {\r
    return ActionString[Action];\r
  }\r
  for (Index = 0; Index < ARRAY_SIZE (mExtActionString); Index++) {\r
    if (mExtActionString[Index].Action == Action) {\r
      return mExtActionString[Index].String;\r
    }\r
  }\r
\r
  return ActionString[0];\r
}\r
\r
/**\r
  Dump SMRAM profile.\r
\r
**/\r
VOID\r
DumpSmramProfile (\r
  VOID\r
  )\r
{\r
  MEMORY_PROFILE_CONTEXT            *Context;\r
  MEMORY_PROFILE_DRIVER_INFO        *DriverInfo;\r
  MEMORY_PROFILE_ALLOC_INFO         *AllocInfo;\r
  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
  MEMORY_PROFILE_ALLOC_INFO_DATA    *AllocInfoData;\r
  LIST_ENTRY                        *SmramDriverInfoList;\r
  UINTN                             DriverIndex;\r
  LIST_ENTRY                        *DriverLink;\r
  LIST_ENTRY                        *AllocInfoList;\r
  UINTN                             AllocIndex;\r
  LIST_ENTRY                        *AllocLink;\r
  BOOLEAN                           SmramProfileGettingStatus;\r
  UINTN                             TypeIndex;\r
\r
  ContextData = GetSmramProfileContext ();\r
  if (ContextData == NULL) {\r
    return ;\r
  }\r
\r
  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
  mSmramProfileGettingStatus = TRUE;\r
\r
  Context = &ContextData->Context;\r
  DEBUG ((EFI_D_INFO, "======= SmramProfile begin =======\n"));\r
  DEBUG ((EFI_D_INFO, "MEMORY_PROFILE_CONTEXT\n"));\r
\r
  DEBUG ((EFI_D_INFO, "  CurrentTotalUsage     - 0x%016lx\n", Context->CurrentTotalUsage));\r
  DEBUG ((EFI_D_INFO, "  PeakTotalUsage        - 0x%016lx\n", Context->PeakTotalUsage));\r
  for (TypeIndex = 0; TypeIndex < sizeof (Context->CurrentTotalUsageByType) / sizeof (Context->CurrentTotalUsageByType[0]); TypeIndex++) {\r
    if ((Context->CurrentTotalUsageByType[TypeIndex] != 0) ||\r
        (Context->PeakTotalUsageByType[TypeIndex] != 0)) {\r
      DEBUG ((EFI_D_INFO, "  CurrentTotalUsage[0x%02x]  - 0x%016lx (%a)\n", TypeIndex, Context->CurrentTotalUsageByType[TypeIndex], ProfileMemoryTypeToStr (TypeIndex)));\r
      DEBUG ((EFI_D_INFO, "  PeakTotalUsage[0x%02x]     - 0x%016lx (%a)\n", TypeIndex, Context->PeakTotalUsageByType[TypeIndex], ProfileMemoryTypeToStr (TypeIndex)));\r
    }\r
  }\r
  DEBUG ((EFI_D_INFO, "  TotalImageSize        - 0x%016lx\n", Context->TotalImageSize));\r
  DEBUG ((EFI_D_INFO, "  ImageCount            - 0x%08x\n", Context->ImageCount));\r
  DEBUG ((EFI_D_INFO, "  SequenceCount         - 0x%08x\n", Context->SequenceCount));\r
\r
  SmramDriverInfoList = ContextData->DriverInfoList;\r
  for (DriverLink = SmramDriverInfoList->ForwardLink, DriverIndex = 0;\r
       DriverLink != SmramDriverInfoList;\r
       DriverLink = DriverLink->ForwardLink, DriverIndex++) {\r
    DriverInfoData = CR (\r
                   DriverLink,\r
                   MEMORY_PROFILE_DRIVER_INFO_DATA,\r
                   Link,\r
                   MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
                   );\r
    DriverInfo = &DriverInfoData->DriverInfo;\r
    DEBUG ((EFI_D_INFO, "  MEMORY_PROFILE_DRIVER_INFO (0x%x)\n", DriverIndex));\r
    DEBUG ((EFI_D_INFO, "    FileName            - %g\n", &DriverInfo->FileName));\r
    DEBUG ((EFI_D_INFO, "    ImageBase           - 0x%016lx\n", DriverInfo->ImageBase));\r
    DEBUG ((EFI_D_INFO, "    ImageSize           - 0x%016lx\n", DriverInfo->ImageSize));\r
    DEBUG ((EFI_D_INFO, "    EntryPoint          - 0x%016lx\n", DriverInfo->EntryPoint));\r
    DEBUG ((EFI_D_INFO, "    ImageSubsystem      - 0x%04x\n", DriverInfo->ImageSubsystem));\r
    DEBUG ((EFI_D_INFO, "    FileType            - 0x%02x\n", DriverInfo->FileType));\r
    DEBUG ((EFI_D_INFO, "    CurrentUsage        - 0x%016lx\n", DriverInfo->CurrentUsage));\r
    DEBUG ((EFI_D_INFO, "    PeakUsage           - 0x%016lx\n", DriverInfo->PeakUsage));\r
    for (TypeIndex = 0; TypeIndex < sizeof (DriverInfo->CurrentUsageByType) / sizeof (DriverInfo->CurrentUsageByType[0]); TypeIndex++) {\r
      if ((DriverInfo->CurrentUsageByType[TypeIndex] != 0) ||\r
          (DriverInfo->PeakUsageByType[TypeIndex] != 0)) {\r
        DEBUG ((EFI_D_INFO, "    CurrentUsage[0x%02x]     - 0x%016lx (%a)\n", TypeIndex, DriverInfo->CurrentUsageByType[TypeIndex], ProfileMemoryTypeToStr (TypeIndex)));\r
        DEBUG ((EFI_D_INFO, "    PeakUsage[0x%02x]        - 0x%016lx (%a)\n", TypeIndex, DriverInfo->PeakUsageByType[TypeIndex], ProfileMemoryTypeToStr (TypeIndex)));\r
      }\r
    }\r
    DEBUG ((EFI_D_INFO, "    AllocRecordCount    - 0x%08x\n", DriverInfo->AllocRecordCount));\r
\r
    AllocInfoList = DriverInfoData->AllocInfoList;\r
    for (AllocLink = AllocInfoList->ForwardLink, AllocIndex = 0;\r
         AllocLink != AllocInfoList;\r
         AllocLink = AllocLink->ForwardLink, AllocIndex++) {\r
      AllocInfoData = CR (\r
                     AllocLink,\r
                     MEMORY_PROFILE_ALLOC_INFO_DATA,\r
                     Link,\r
                     MEMORY_PROFILE_ALLOC_INFO_SIGNATURE\r
                     );\r
      AllocInfo = &AllocInfoData->AllocInfo;\r
      DEBUG ((EFI_D_INFO, "    MEMORY_PROFILE_ALLOC_INFO (0x%x)\n", AllocIndex));\r
      DEBUG ((EFI_D_INFO, "      CallerAddress  - 0x%016lx (Offset: 0x%08x)\n", AllocInfo->CallerAddress, AllocInfo->CallerAddress - DriverInfo->ImageBase));\r
      DEBUG ((EFI_D_INFO, "      SequenceId     - 0x%08x\n", AllocInfo->SequenceId));\r
      if ((AllocInfo->Action & MEMORY_PROFILE_ACTION_USER_DEFINED_MASK) != 0) {\r
        if (AllocInfoData->ActionString != NULL) {\r
          DEBUG ((EFI_D_INFO, "      Action         - 0x%08x (%a)\n", AllocInfo->Action, AllocInfoData->ActionString));\r
        } else {\r
          DEBUG ((EFI_D_INFO, "      Action         - 0x%08x (UserDefined-0x%08x)\n", AllocInfo->Action, AllocInfo->Action));\r
        }\r
      } else {\r
        DEBUG ((EFI_D_INFO, "      Action         - 0x%08x (%a)\n", AllocInfo->Action, ProfileActionToStr (AllocInfo->Action)));\r
      }\r
      DEBUG ((EFI_D_INFO, "      MemoryType     - 0x%08x (%a)\n", AllocInfo->MemoryType, ProfileMemoryTypeToStr (AllocInfo->MemoryType)));\r
      DEBUG ((EFI_D_INFO, "      Buffer         - 0x%016lx\n", AllocInfo->Buffer));\r
      DEBUG ((EFI_D_INFO, "      Size           - 0x%016lx\n", AllocInfo->Size));\r
    }\r
  }\r
\r
  DEBUG ((EFI_D_INFO, "======= SmramProfile end =======\n"));\r
\r
  mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
  Dump SMRAM information.\r
\r
**/\r
VOID\r
DumpSmramInfo (\r
  VOID\r
  )\r
{\r
  DEBUG_CODE (\r
    if (IS_SMRAM_PROFILE_ENABLED) {\r
      DumpSmramProfile ();\r
      DumpFreePagesList ();\r
      DumpFreePoolList ();\r
      DumpSmramRange ();\r
    }\r
  );\r
}\r
\r