/** @file\r
Support routines for SMRAM profile.\r
\r
- Copyright (c) 2014 - 2015, Intel Corporation. All rights reserved.<BR>\r
- This program and the accompanying materials\r
- are licensed and made available under the terms and conditions of the BSD License\r
- which accompanies this distribution. The full text of the license may be found at\r
- http://opensource.org/licenses/bsd-license.php.\r
-\r
- THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
- WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\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_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
+ 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
- LIST_ENTRY Link;\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
//\r
#define SMRAM_INFO_DUMP_PAGE_THRESHOLD 4\r
\r
-GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_FREE_MEMORY mSmramFreeMemory = {\r
+GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_FREE_MEMORY mSmramFreeMemory = {\r
{\r
MEMORY_PROFILE_FREE_MEMORY_SIGNATURE,\r
sizeof (MEMORY_PROFILE_FREE_MEMORY),\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
+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
},\r
0,\r
0,\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;\r
+GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_CONTEXT_DATA *mSmramProfileContextPtr = NULL;\r
\r
-BOOLEAN mSmramReadyToLock;\r
-BOOLEAN mSmramProfileRecordingStatus = FALSE;\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
- Return SMRAM profile context.\r
-\r
- @return SMRAM profile context.\r
+ Dump SMRAM information.\r
\r
**/\r
-MEMORY_PROFILE_CONTEXT_DATA *\r
-GetSmramProfileContext (\r
+VOID\r
+DumpSmramInfo (\r
VOID\r
- )\r
-{\r
- return mSmramProfileContextPtr;\r
-}\r
+ );\r
\r
/**\r
- Retrieves the magic value from the PE/COFF header.\r
+ Get memory profile data.\r
\r
- @param Hdr The buffer in which to return the PE32, PE32+, or TE header.\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_IMAGE_NT_OPTIONAL_HDR32_MAGIC - Image is PE32\r
- @return EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC - Image is PE32+\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
-UINT16\r
-InternalPeCoffGetPeHeaderMagicValue (\r
- IN EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Hdr\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
- //\r
- // NOTE: Some versions of Linux ELILO for Itanium have an incorrect magic value\r
- // in the PE/COFF Header. If the MachineType is Itanium(IA64) and the\r
- // Magic value in the OptionalHeader is EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC\r
- // then override the returned value to EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC\r
- //\r
- if (Hdr.Pe32->FileHeader.Machine == IMAGE_FILE_MACHINE_IA64 && Hdr.Pe32->OptionalHeader.Magic == EFI_IMAGE_NT_OPTIONAL_HDR32_MAGIC) {\r
- return EFI_IMAGE_NT_OPTIONAL_HDR64_MAGIC;\r
- }\r
- //\r
- // Return the magic value from the PC/COFF Optional Header\r
- //\r
- return Hdr.Pe32->OptionalHeader.Magic;\r
+ return mSmramProfileContextPtr;\r
}\r
\r
/**\r
\r
ASSERT (Pe32Data != NULL);\r
\r
- DosHdr = (EFI_IMAGE_DOS_HEADER *) Pe32Data;\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
+ 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
+ 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 = InternalPeCoffGetPeHeaderMagicValue (Hdr);\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
OUT VOID **EntryPoint\r
)\r
{\r
- EFI_IMAGE_DOS_HEADER *DosHdr;\r
- EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION Hdr;\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
+ 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
+ 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
+ Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)Pe32Data;\r
}\r
\r
//\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
+ *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
+ *EntryPoint = (VOID *)((UINTN)Pe32Data + (UINTN)(Hdr.Pe32->OptionalHeader.AddressOfEntryPoint & 0x0ffffffff));\r
return RETURN_SUCCESS;\r
}\r
\r
@param ImageSize Image size.\r
@param EntryPoint Entry point of the image.\r
@param ImageSubsystem Image subsystem of the image.\r
-\r
@param FileType File type of the image.\r
\r
@return Pointer to memory profile driver info.\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
+ 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
+ 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),\r
- (VOID **) &DriverInfoData\r
+ sizeof (*DriverInfoData) + sizeof (LIST_ENTRY) + PdbSize,\r
+ (VOID **)&DriverInfoData\r
);\r
if (EFI_ERROR (Status)) {\r
return NULL;\r
}\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 = &DriverInfoData->DriverInfo;\r
+ DriverInfoData->Signature = MEMORY_PROFILE_DRIVER_INFO_SIGNATURE;\r
DriverInfo->Header.Signature = MEMORY_PROFILE_DRIVER_INFO_SIGNATURE;\r
- DriverInfo->Header.Length = sizeof (MEMORY_PROFILE_DRIVER_INFO);\r
- DriverInfo->Header.Revision = MEMORY_PROFILE_DRIVER_INFO_REVISION;\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
+\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
+ Status = InternalPeCoffGetEntryPoint ((VOID *)(UINTN)ImageBase, &EntryPointInImage);\r
ASSERT_EFI_ERROR (Status);\r
- DriverInfo->ImageBase = ImageBase + EntryPoint - (PHYSICAL_ADDRESS) (UINTN) EntryPointInImage;\r
+ DriverInfo->ImageBase = ImageBase + EntryPoint - (PHYSICAL_ADDRESS)(UINTN)EntryPointInImage;\r
}\r
- DriverInfo->FileType = FileType;\r
- DriverInfoData->AllocInfoList = (LIST_ENTRY *) (DriverInfoData + 1);\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->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.ImageCount++;\r
ContextData->Context.TotalImageSize += DriverInfo->ImageSize;\r
\r
return DriverInfoData;\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
+ 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_SMRAM_PROFILE_ENABLED) {\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
FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)TempBuffer;\r
- Status = gBS->LocateProtocol (&gEdkiiMemoryProfileGuid, NULL, (VOID **) &ProfileProtocol);\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
+ (EFI_DEVICE_PATH_PROTOCOL *)FilePath,\r
ImageBase,\r
ImageSize,\r
FileType\r
**/\r
VOID\r
UnregisterImageFromDxe (\r
- IN EFI_GUID *FileName,\r
- IN PHYSICAL_ADDRESS ImageBase,\r
- IN UINT64 ImageSize\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_SMRAM_PROFILE_ENABLED) {\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
FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)TempBuffer;\r
- Status = gBS->LocateProtocol (&gEdkiiMemoryProfileGuid, NULL, (VOID *) &ProfileProtocol);\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
+ (EFI_DEVICE_PATH_PROTOCOL *)FilePath,\r
ImageBase,\r
ImageSize\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
+ {\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
**/\r
BOOLEAN\r
RegisterSmmCore (\r
- IN MEMORY_PROFILE_CONTEXT_DATA *ContextData\r
+ IN MEMORY_PROFILE_CONTEXT_DATA *ContextData\r
)\r
{\r
- MEMORY_PROFILE_DRIVER_INFO_DATA *DriverInfoData;\r
- PHYSICAL_ADDRESS ImageBase;\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
- ASSERT (ContextData != NULL);\r
+ FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)TempBuffer;\r
+ EfiInitializeFwVolDevicepathNode (FilePath, &gEfiCallerIdGuid);\r
+ SetDevicePathEndNode (FilePath + 1);\r
\r
- RegisterImageToDxe (\r
- &gEfiCallerIdGuid,\r
- gSmmCorePrivate->PiSmmCoreImageBase,\r
- gSmmCorePrivate->PiSmmCoreImageSize,\r
- EFI_FV_FILETYPE_SMM_CORE\r
- );\r
+ if (!NeedRecordThisDriver ((EFI_DEVICE_PATH_PROTOCOL *)FilePath)) {\r
+ return FALSE;\r
+ }\r
\r
- ImageBase = gSmmCorePrivate->PiSmmCoreImageBase;\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
+ InternalPeCoffGetSubsystem ((VOID *)(UINTN)ImageBase),\r
EFI_FV_FILETYPE_SMM_CORE\r
);\r
if (DriverInfoData == NULL) {\r
VOID\r
)\r
{\r
- MEMORY_PROFILE_CONTEXT_DATA *SmramProfileContext;\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
return;\r
}\r
\r
- mSmramProfileRecordingStatus = TRUE;\r
- mSmramProfileContextPtr = &mSmramProfileContext;\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
+\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
+ DEBUG ((DEBUG_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
+\r
+ ThisFilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)NextDevicePathNode (ThisFilePath);\r
+ }\r
+ }\r
+\r
+ return FileName;\r
}\r
\r
/**\r
@param DriverEntry SMM image info.\r
@param RegisterToDxe Register image to DXE.\r
\r
- @retval TRUE Register success.\r
- @retval FALSE Register fail.\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
-BOOLEAN\r
+EFI_STATUS\r
RegisterSmramProfileImage (\r
- IN EFI_SMM_DRIVER_ENTRY *DriverEntry,\r
- IN BOOLEAN RegisterToDxe\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
-\r
- if (!IS_SMRAM_PROFILE_ENABLED) {\r
- return FALSE;\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
);\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 FALSE;\r
+ return EFI_UNSUPPORTED;\r
}\r
\r
DriverInfoData = BuildDriverInfo (\r
DriverEntry->ImageBuffer,\r
EFI_PAGES_TO_SIZE (DriverEntry->NumberOfPage),\r
DriverEntry->ImageEntryPoint,\r
- InternalPeCoffGetSubsystem ((VOID *) (UINTN) DriverEntry->ImageBuffer),\r
+ InternalPeCoffGetSubsystem ((VOID *)(UINTN)DriverEntry->ImageBuffer),\r
EFI_FV_FILETYPE_SMM\r
);\r
if (DriverInfoData == NULL) {\r
- return FALSE;\r
+ return EFI_OUT_OF_RESOURCES;\r
}\r
\r
- return TRUE;\r
+ return EFI_SUCCESS;\r
}\r
\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
+ 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
+ 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
+ DriverLink = DriverLink->ForwardLink)\r
+ {\r
DriverInfoData = CR (\r
DriverLink,\r
MEMORY_PROFILE_DRIVER_INFO_DATA,\r
DriverInfo = &DriverInfoData->DriverInfo;\r
if ((CompareGuid (&DriverInfo->FileName, FileName)) &&\r
(Address >= DriverInfo->ImageBase) &&\r
- (Address < (DriverInfo->ImageBase + DriverInfo->ImageSize))) {\r
+ (Address < (DriverInfo->ImageBase + DriverInfo->ImageSize)))\r
+ {\r
return DriverInfoData;\r
}\r
}\r
return NULL;\r
}\r
\r
-/**\r
- Search dummy image from SMRAM profile.\r
-\r
- @param ContextData Memory profile context.\r
-\r
- @return Pointer to memory profile driver info.\r
-\r
-**/\r
-MEMORY_PROFILE_DRIVER_INFO_DATA *\r
-FindDummyImage (\r
- IN MEMORY_PROFILE_CONTEXT_DATA *ContextData\r
- )\r
-{\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
- if (CompareGuid (&gZeroGuid, &DriverInfoData->DriverInfo.FileName)) {\r
- return DriverInfoData;\r
- }\r
- }\r
-\r
- return BuildDriverInfo (ContextData, &gZeroGuid, 0, 0, 0, 0, 0);\r
-}\r
-\r
/**\r
Search image from memory profile.\r
It will return image, if (Address >= ImageBuffer) AND (Address < ImageBuffer + ImageSize)\r
**/\r
MEMORY_PROFILE_DRIVER_INFO_DATA *\r
GetMemoryProfileDriverInfoFromAddress (\r
- IN MEMORY_PROFILE_CONTEXT_DATA *ContextData,\r
- IN PHYSICAL_ADDRESS Address\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
+ 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
+ DriverLink = DriverLink->ForwardLink)\r
+ {\r
DriverInfoData = CR (\r
DriverLink,\r
MEMORY_PROFILE_DRIVER_INFO_DATA,\r
);\r
DriverInfo = &DriverInfoData->DriverInfo;\r
if ((Address >= DriverInfo->ImageBase) &&\r
- (Address < (DriverInfo->ImageBase + DriverInfo->ImageSize))) {\r
+ (Address < (DriverInfo->ImageBase + DriverInfo->ImageSize)))\r
+ {\r
return DriverInfoData;\r
}\r
}\r
\r
- //\r
- // Should never come here.\r
- //\r
- return FindDummyImage (ContextData);\r
+ return NULL;\r
}\r
\r
/**\r
@param DriverEntry SMM image info.\r
@param UnregisterFromDxe Unregister image from DXE.\r
\r
- @retval TRUE Unregister success.\r
- @retval FALSE Unregister fail.\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
-BOOLEAN\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
-\r
- if (!IS_SMRAM_PROFILE_ENABLED) {\r
- return FALSE;\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
);\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 FALSE;\r
+ return EFI_UNSUPPORTED;\r
}\r
\r
DriverInfoData = NULL;\r
- FileName = &DriverEntry->FileName;\r
- ImageAddress = DriverEntry->ImageBuffer;\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
+ Status = InternalPeCoffGetEntryPoint ((VOID *)(UINTN)ImageAddress, &EntryPointInImage);\r
ASSERT_EFI_ERROR (Status);\r
- ImageAddress = ImageAddress + (UINTN) DriverEntry->ImageEntryPoint - (UINTN) EntryPointInImage;\r
+ ImageAddress = ImageAddress + (UINTN)DriverEntry->ImageEntryPoint - (UINTN)EntryPointInImage;\r
}\r
+\r
if (FileName != NULL) {\r
DriverInfoData = GetMemoryProfileDriverInfoByFileNameAndAddress (ContextData, FileName, ImageAddress);\r
}\r
+\r
if (DriverInfoData == NULL) {\r
DriverInfoData = GetMemoryProfileDriverInfoFromAddress (ContextData, ImageAddress);\r
}\r
+\r
if (DriverInfoData == NULL) {\r
- return FALSE;\r
+ return EFI_NOT_FOUND;\r
}\r
\r
ContextData->Context.TotalImageSize -= DriverInfoData->DriverInfo.ImageSize;\r
\r
- DriverInfoData->DriverInfo.ImageBase = 0;\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
+ ContextData->Context.ImageCount--;\r
RemoveEntryList (&DriverInfoData->Link);\r
//\r
// Use SmmInternalFreePool() that will not update profile for this FreePool action.\r
SmmInternalFreePool (DriverInfoData);\r
}\r
\r
- return TRUE;\r
+ return EFI_SUCCESS;\r
}\r
\r
/**\r
Return if this memory type needs to be recorded into memory profile.\r
- If BIOS memory type (0 ~ EfiMaxMemoryType), it checks bit (1 << MemoryType).\r
- If OS memory type (0x80000000 ~ 0xFFFFFFFF), it checks bit63 - 0x8000000000000000.\r
+ Only need to record EfiRuntimeServicesCode and EfiRuntimeServicesData for SMRAM profile.\r
\r
@param MemoryType Memory type.\r
\r
**/\r
BOOLEAN\r
SmmCoreNeedRecordProfile (\r
- IN EFI_MEMORY_TYPE MemoryType\r
+ IN EFI_MEMORY_TYPE MemoryType\r
)\r
{\r
- UINT64 TestBit;\r
+ UINT64 TestBit;\r
\r
- if ((UINT32) MemoryType >= 0x80000000) {\r
- TestBit = BIT63;\r
- } else {\r
- TestBit = LShiftU64 (1, MemoryType);\r
+ if ((MemoryType != EfiRuntimeServicesCode) &&\r
+ (MemoryType != EfiRuntimeServicesData))\r
+ {\r
+ return FALSE;\r
}\r
\r
+ TestBit = LShiftU64 (1, MemoryType);\r
+\r
if ((PcdGet64 (PcdMemoryProfileMemoryType) & TestBit) != 0) {\r
return TRUE;\r
} else {\r
\r
/**\r
Convert EFI memory type to profile memory index. The rule is:\r
- If BIOS memory type (0 ~ EfiMaxMemoryType), ProfileMemoryIndex = MemoryType.\r
- If OS memory type (0x80000000 ~ 0xFFFFFFFF), ProfileMemoryIndex = EfiMaxMemoryType.\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
**/\r
EFI_MEMORY_TYPE\r
GetProfileMemoryIndex (\r
- IN EFI_MEMORY_TYPE MemoryType\r
+ IN EFI_MEMORY_TYPE MemoryType\r
)\r
{\r
- if ((UINT32) MemoryType >= 0x80000000) {\r
- return EfiMaxMemoryType;\r
- } else {\r
- return MemoryType;\r
- }\r
+ return MemoryType;\r
}\r
\r
/**\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
+ LIST_ENTRY *Node;\r
+ FREE_PAGE_LIST *Pages;\r
+ LIST_ENTRY *FreePageList;\r
+ UINTN NumberOfPages;\r
\r
NumberOfPages = 0;\r
- FreePageList = &mSmmMemoryMap;\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
+ Node = Node->BackLink)\r
+ {\r
+ Pages = BASE_CR (Node, FREE_PAGE_LIST, Link);\r
NumberOfPages += Pages->NumberOfPages;\r
}\r
\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
- @retval TRUE Profile udpate success.\r
- @retval FALSE Profile update fail.\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
-BOOLEAN\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 VOID *Buffer,\r
+ IN CHAR8 *ActionString OPTIONAL\r
)\r
{\r
- EFI_STATUS Status;\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_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
- AllocInfoData = NULL;\r
+ BasicAction = Action & MEMORY_PROFILE_ACTION_BASIC_MASK;\r
\r
ContextData = GetSmramProfileContext ();\r
if (ContextData == NULL) {\r
- return FALSE;\r
+ return EFI_UNSUPPORTED;\r
}\r
\r
DriverInfoData = GetMemoryProfileDriverInfoFromAddress (ContextData, CallerAddress);\r
- ASSERT (DriverInfoData != NULL);\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
- Status = SmmInternalAllocatePool (\r
- EfiRuntimeServicesData,\r
- sizeof (*AllocInfoData),\r
- (VOID **) &AllocInfoData\r
- );\r
+ AllocInfoData = NULL;\r
+ Status = SmmInternalAllocatePool (\r
+ EfiRuntimeServicesData,\r
+ sizeof (*AllocInfoData) + ActionStringSize,\r
+ (VOID **)&AllocInfoData\r
+ );\r
if (EFI_ERROR (Status)) {\r
- return FALSE;\r
+ return EFI_OUT_OF_RESOURCES;\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 = sizeof (MEMORY_PROFILE_ALLOC_INFO);\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
\r
- InsertTailList (DriverInfoData->AllocInfoList, &AllocInfoData->Link);\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
- ProfileMemoryIndex = GetProfileMemoryIndex (MemoryType);\r
+ InsertTailList (DriverInfoData->AllocInfoList, &AllocInfoData->Link);\r
\r
+ Context = &ContextData->Context;\r
DriverInfo = &DriverInfoData->DriverInfo;\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
- DriverInfo->AllocRecordCount ++;\r
+ DriverInfo->AllocRecordCount++;\r
\r
- Context = &ContextData->Context;\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
+ // 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
+\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
+\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
- Context->SequenceCount ++;\r
\r
- SmramProfileUpdateFreePages (ContextData);\r
- return TRUE;\r
+ return EFI_SUCCESS;\r
}\r
\r
/**\r
Get memory profile alloc info from memory profile\r
\r
@param DriverInfoData Driver info\r
- @param Action This Free action\r
+ @param BasicAction This Free basic action\r
@param Size Buffer size\r
@param Buffer Buffer address\r
\r
**/\r
MEMORY_PROFILE_ALLOC_INFO_DATA *\r
GetMemoryProfileAllocInfoFromAddress (\r
- IN MEMORY_PROFILE_DRIVER_INFO_DATA *DriverInfoData,\r
- IN MEMORY_PROFILE_ACTION Action,\r
- IN UINTN Size,\r
- IN VOID *Buffer\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
+ 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
+ AllocLink = AllocLink->ForwardLink)\r
+ {\r
AllocInfoData = CR (\r
AllocLink,\r
MEMORY_PROFILE_ALLOC_INFO_DATA,\r
MEMORY_PROFILE_ALLOC_INFO_SIGNATURE\r
);\r
AllocInfo = &AllocInfoData->AllocInfo;\r
- if (AllocInfo->Action != Action) {\r
+ if ((AllocInfo->Action & MEMORY_PROFILE_ACTION_BASIC_MASK) != BasicAction) {\r
continue;\r
}\r
- switch (Action) {\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
+ if ((AllocInfo->Buffer <= (PHYSICAL_ADDRESS)(UINTN)Buffer) &&\r
+ ((AllocInfo->Buffer + AllocInfo->Size) >= ((PHYSICAL_ADDRESS)(UINTN)Buffer + Size)))\r
+ {\r
return AllocInfoData;\r
}\r
+\r
break;\r
case MemoryProfileActionAllocatePool:\r
- if (AllocInfo->Buffer == (PHYSICAL_ADDRESS) (UINTN) Buffer) {\r
+ if (AllocInfo->Buffer == (PHYSICAL_ADDRESS)(UINTN)Buffer) {\r
return AllocInfoData;\r
}\r
+\r
break;\r
default:\r
ASSERT (FALSE);\r
@param Size Buffer size.\r
@param Buffer Buffer address.\r
\r
- @retval TRUE Profile udpate success.\r
- @retval FALSE Profile update fail.\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
-BOOLEAN\r
+EFI_STATUS\r
SmmCoreUpdateProfileFree (\r
IN PHYSICAL_ADDRESS CallerAddress,\r
IN MEMORY_PROFILE_ACTION Action,\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 FALSE;\r
+ return EFI_UNSUPPORTED;\r
}\r
\r
DriverInfoData = GetMemoryProfileDriverInfoFromAddress (ContextData, CallerAddress);\r
- ASSERT (DriverInfoData != NULL);\r
\r
- switch (Action) {\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
- 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 (Action) {\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 (ThisDriverInfoData, MemoryProfileActionAllocatePages, Size, Buffer);\r
+ AllocInfoData = GetMemoryProfileAllocInfoFromAddress (DriverInfoData, MemoryProfileActionAllocatePages, Size, Buffer);\r
break;\r
case MemoryProfileActionFreePool:\r
- AllocInfoData = GetMemoryProfileAllocInfoFromAddress (ThisDriverInfoData, MemoryProfileActionAllocatePool, 0, Buffer);\r
+ AllocInfoData = GetMemoryProfileAllocInfoFromAddress (DriverInfoData, 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
- // No matched allocate operation is found for this free operation.\r
- // It is because the specified memory type allocate operation has been\r
- // filtered by CoreNeedRecordProfile(), but free operations have no\r
- // memory type information, they can not be filtered by CoreNeedRecordProfile().\r
- // Then, they will be filtered here.\r
+ // Legal case, because driver A might free memory allocated by driver B, by some protocol.\r
//\r
- return FALSE;\r
+ DriverInfoList = ContextData->DriverInfoList;\r
+\r
+ for (DriverLink = DriverInfoList->ForwardLink;\r
+ DriverLink != DriverInfoList;\r
+ DriverLink = DriverLink->ForwardLink)\r
+ {\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
+\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
\r
- Context = &ContextData->Context;\r
- DriverInfo = &DriverInfoData->DriverInfo;\r
- AllocInfo = &AllocInfoData->AllocInfo;\r
+ ASSERT (DriverInfoData != NULL);\r
+ ASSERT (AllocInfoData != NULL);\r
\r
- ProfileMemoryIndex = GetProfileMemoryIndex (AllocInfo->MemoryType);\r
+ Found = TRUE;\r
\r
- Context->CurrentTotalUsage -= AllocInfo->Size;\r
- Context->CurrentTotalUsageByType[ProfileMemoryIndex] -= AllocInfo->Size;\r
+ Context = &ContextData->Context;\r
+ DriverInfo = &DriverInfoData->DriverInfo;\r
+ AllocInfo = &AllocInfoData->AllocInfo;\r
\r
- DriverInfo->CurrentUsage -= AllocInfo->Size;\r
- DriverInfo->CurrentUsageByType[ProfileMemoryIndex] -= AllocInfo->Size;\r
- DriverInfo->AllocRecordCount --;\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
- RemoveEntryList (&AllocInfoData->Link);\r
+ Context->CurrentTotalUsage -= AllocInfo->Size;\r
+ Context->CurrentTotalUsageByType[ProfileMemoryIndex] -= AllocInfo->Size;\r
\r
- if (Action == MemoryProfileActionFreePages) {\r
- if (AllocInfo->Buffer != (PHYSICAL_ADDRESS) (UINTN) Buffer) {\r
- SmmCoreUpdateProfileAllocate (\r
- AllocInfo->CallerAddress,\r
- MemoryProfileActionAllocatePages,\r
- AllocInfo->MemoryType,\r
- (UINTN) ((PHYSICAL_ADDRESS) (UINTN) Buffer - AllocInfo->Buffer),\r
- (VOID *) (UINTN) AllocInfo->Buffer\r
- );\r
- }\r
- if (AllocInfo->Buffer + AllocInfo->Size != ((PHYSICAL_ADDRESS) (UINTN) Buffer + Size)) {\r
- SmmCoreUpdateProfileAllocate (\r
- AllocInfo->CallerAddress,\r
- MemoryProfileActionAllocatePages,\r
- AllocInfo->MemoryType,\r
- (UINTN) ((AllocInfo->Buffer + AllocInfo->Size) - ((PHYSICAL_ADDRESS) (UINTN) Buffer + Size)),\r
- (VOID *) ((UINTN) Buffer + Size)\r
- );\r
+ DriverInfo->CurrentUsage -= AllocInfo->Size;\r
+ DriverInfo->CurrentUsageByType[ProfileMemoryIndex] -= AllocInfo->Size;\r
}\r
- }\r
\r
- //\r
- // Use SmmInternalFreePool() that will not update profile for this FreePool action.\r
- //\r
- SmmInternalFreePool (AllocInfoData);\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
\r
- return TRUE;\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
@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
-\r
- @retval TRUE Profile udpate success.\r
- @retval FALSE Profile update fail.\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
-BOOLEAN\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 VOID *Buffer,\r
+ IN CHAR8 *ActionString OPTIONAL\r
)\r
{\r
- MEMORY_PROFILE_CONTEXT_DATA *ContextData;\r
+ EFI_STATUS Status;\r
+ MEMORY_PROFILE_CONTEXT_DATA *ContextData;\r
+ MEMORY_PROFILE_ACTION BasicAction;\r
\r
if (!IS_SMRAM_PROFILE_ENABLED) {\r
- return FALSE;\r
+ return EFI_UNSUPPORTED;\r
}\r
\r
- if (!mSmramProfileRecordingStatus) {\r
- return FALSE;\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 ((Action == MemoryProfileActionAllocatePages) || (Action == MemoryProfileActionAllocatePool)) {\r
+ if ((BasicAction == MemoryProfileActionAllocatePages) || (BasicAction == MemoryProfileActionAllocatePool)) {\r
//\r
// Only record limited MemoryType.\r
//\r
if (!SmmCoreNeedRecordProfile (MemoryType)) {\r
- return FALSE;\r
+ return EFI_UNSUPPORTED;\r
}\r
}\r
\r
ContextData = GetSmramProfileContext ();\r
if (ContextData == NULL) {\r
- return FALSE;\r
+ return EFI_UNSUPPORTED;\r
}\r
\r
- switch (Action) {\r
+ switch (BasicAction) {\r
case MemoryProfileActionAllocatePages:\r
- SmmCoreUpdateProfileAllocate (CallerAddress, Action, MemoryType, Size, Buffer);\r
+ Status = SmmCoreUpdateProfileAllocate (CallerAddress, Action, MemoryType, Size, Buffer, ActionString);\r
break;\r
case MemoryProfileActionFreePages:\r
- SmmCoreUpdateProfileFree (CallerAddress, Action, Size, Buffer);\r
+ Status = SmmCoreUpdateProfileFree (CallerAddress, Action, Size, Buffer);\r
break;\r
case MemoryProfileActionAllocatePool:\r
- SmmCoreUpdateProfileAllocate (CallerAddress, Action, MemoryType, Size, Buffer);\r
+ Status = SmmCoreUpdateProfileAllocate (CallerAddress, Action, MemoryType, Size, Buffer, ActionString);\r
break;\r
case MemoryProfileActionFreePool:\r
- SmmCoreUpdateProfileFree (CallerAddress, Action, 0, Buffer);\r
+ Status = SmmCoreUpdateProfileFree (CallerAddress, Action, 0, Buffer);\r
break;\r
default:\r
ASSERT (FALSE);\r
+ Status = EFI_UNSUPPORTED;\r
break;\r
}\r
\r
- return TRUE;\r
+ return Status;\r
}\r
\r
/**\r
return;\r
}\r
\r
- DEBUG ((EFI_D_INFO, "SmramProfileReadyToLock\n"));\r
+ DEBUG ((DEBUG_INFO, "SmramProfileReadyToLock\n"));\r
mSmramReadyToLock = TRUE;\r
}\r
\r
{\r
MEMORY_PROFILE_CONTEXT_DATA *ContextData;\r
MEMORY_PROFILE_DRIVER_INFO_DATA *DriverInfoData;\r
- LIST_ENTRY *DriverInfoList;\r
- LIST_ENTRY *DriverLink;\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
+ 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
}\r
\r
TotalSize = sizeof (MEMORY_PROFILE_CONTEXT);\r
- TotalSize += sizeof (MEMORY_PROFILE_DRIVER_INFO) * (UINTN) ContextData->Context.ImageCount;\r
\r
DriverInfoList = ContextData->DriverInfoList;\r
for (DriverLink = DriverInfoList->ForwardLink;\r
DriverLink != DriverInfoList;\r
- DriverLink = DriverLink->ForwardLink) {\r
+ DriverLink = DriverLink->ForwardLink)\r
+ {\r
DriverInfoData = CR (\r
- DriverLink,\r
- MEMORY_PROFILE_DRIVER_INFO_DATA,\r
- Link,\r
- MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
- );\r
- TotalSize += sizeof (MEMORY_PROFILE_ALLOC_INFO) * (UINTN) DriverInfoData->DriverInfo.AllocRecordCount;\r
- }\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
+ {\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
- Index = 0;\r
+ Index = 0;\r
FreePageList = &mSmmMemoryMap;\r
for (Node = FreePageList->BackLink;\r
Node != FreePageList;\r
- Node = Node->BackLink) {\r
+ Node = Node->BackLink)\r
+ {\r
Index++;\r
}\r
- for (PoolListIndex = 0; PoolListIndex < MAX_POOL_INDEX; PoolListIndex++) {\r
- FreePoolList = &mSmmPoolLists[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
+ 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
+ {\r
+ Pool = BASE_CR (Node, FREE_POOL_HEADER, Link);\r
+ if (Pool->Header.Available) {\r
+ Index++;\r
+ }\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
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
- IN VOID *ProfileBuffer\r
+ OUT VOID *ProfileBuffer,\r
+ IN OUT UINT64 *ProfileSize,\r
+ IN OUT UINT64 *ProfileOffset\r
)\r
{\r
MEMORY_PROFILE_CONTEXT *Context;\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
+ 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
+ return;\r
}\r
\r
- Context = ProfileBuffer;\r
- CopyMem (Context, &ContextData->Context, sizeof (MEMORY_PROFILE_CONTEXT));\r
- DriverInfo = (MEMORY_PROFILE_DRIVER_INFO *) (Context + 1);\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
+\r
+ Offset += sizeof (MEMORY_PROFILE_CONTEXT);\r
\r
DriverInfoList = ContextData->DriverInfoList;\r
for (DriverLink = DriverInfoList->ForwardLink;\r
DriverLink != DriverInfoList;\r
- DriverLink = DriverLink->ForwardLink) {\r
+ DriverLink = DriverLink->ForwardLink)\r
+ {\r
DriverInfoData = CR (\r
DriverLink,\r
MEMORY_PROFILE_DRIVER_INFO_DATA,\r
Link,\r
MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
);\r
- CopyMem (DriverInfo, &DriverInfoData->DriverInfo, sizeof (MEMORY_PROFILE_DRIVER_INFO));\r
- AllocInfo = (MEMORY_PROFILE_ALLOC_INFO *) (DriverInfo + 1);\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
+\r
+ RemainingSize -= DriverInfo->Header.Length;\r
+ ProfileBuffer = (UINT8 *)ProfileBuffer + DriverInfo->Header.Length;\r
+ } else {\r
+ goto Done;\r
+ }\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
+ AllocLink = AllocLink->ForwardLink)\r
+ {\r
AllocInfoData = CR (\r
AllocLink,\r
MEMORY_PROFILE_ALLOC_INFO_DATA,\r
Link,\r
MEMORY_PROFILE_ALLOC_INFO_SIGNATURE\r
);\r
- CopyMem (AllocInfo, &AllocInfoData->AllocInfo, sizeof (MEMORY_PROFILE_ALLOC_INFO));\r
- AllocInfo += 1;\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
+\r
+ RemainingSize -= AllocInfo->Header.Length;\r
+ ProfileBuffer = (UINT8 *)ProfileBuffer + AllocInfo->Header.Length;\r
+ } else {\r
+ goto Done;\r
+ }\r
+ }\r
\r
- DriverInfo = (MEMORY_PROFILE_DRIVER_INFO *) ((UINTN) (DriverInfo + 1) + sizeof (MEMORY_PROFILE_ALLOC_INFO) * (UINTN) DriverInfo->AllocRecordCount);\r
+ Offset += AllocInfoData->AllocInfo.Header.Length;\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
+ {\r
+ Index++;\r
+ }\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
+ {\r
+ Pool = BASE_CR (Node, FREE_POOL_HEADER, Link);\r
+ if (Pool->Header.Available) {\r
+ Index++;\r
+ }\r
+ }\r
+ }\r
+ }\r
\r
- FreeMemory = (MEMORY_PROFILE_FREE_MEMORY *) DriverInfo;\r
- CopyMem (FreeMemory, &mSmramFreeMemory, sizeof (MEMORY_PROFILE_FREE_MEMORY));\r
- MemoryProfileDescriptor = (MEMORY_PROFILE_DESCRIPTOR *) (FreeMemory + 1);\r
- Index = 0;\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
+\r
+ Offset += sizeof (MEMORY_PROFILE_FREE_MEMORY);\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
- 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
- MemoryProfileDescriptor++;\r
- Index++;\r
- }\r
- for (PoolListIndex = 0; PoolListIndex < MAX_POOL_INDEX; PoolListIndex++) {\r
- FreePoolList = &mSmmPoolLists[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
+ Node = Node->BackLink)\r
+ {\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) Pool;\r
- MemoryProfileDescriptor->Size = Pool->Header.Size;\r
- MemoryProfileDescriptor++;\r
- Index++;\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
+\r
+ Offset += sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
+ }\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
+ {\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
+\r
+ Offset += sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
+ }\r
}\r
}\r
}\r
- FreeMemory->FreeMemoryEntryCount = Index;\r
\r
- MemoryRange = (MEMORY_PROFILE_MEMORY_RANGE *) MemoryProfileDescriptor;\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
- MemoryProfileDescriptor = (MEMORY_PROFILE_DESCRIPTOR *) (MemoryRange + 1);\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
+\r
+ Offset += sizeof (MEMORY_PROFILE_MEMORY_RANGE);\r
for (Index = 0; Index < mFullSmramRangeCount; Index++) {\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
- MemoryProfileDescriptor++; \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
+\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
+\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
+\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
+\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
**/\r
VOID\r
SmramProfileHandlerGetInfo (\r
- IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_INFO *SmramProfileParameterGetInfo\r
+ IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_INFO *SmramProfileParameterGetInfo\r
)\r
{\r
- MEMORY_PROFILE_CONTEXT_DATA *ContextData;\r
- BOOLEAN SmramProfileRecordingStatus;\r
+ MEMORY_PROFILE_CONTEXT_DATA *ContextData;\r
+ BOOLEAN SmramProfileGettingStatus;\r
\r
ContextData = GetSmramProfileContext ();\r
if (ContextData == NULL) {\r
- return ;\r
+ return;\r
}\r
\r
- SmramProfileRecordingStatus = mSmramProfileRecordingStatus;\r
- mSmramProfileRecordingStatus = FALSE;\r
+ SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+ mSmramProfileGettingStatus = TRUE;\r
\r
- SmramProfileParameterGetInfo->ProfileSize = SmramProfileGetDataSize();\r
+ SmramProfileParameterGetInfo->ProfileSize = SmramProfileGetDataSize ();\r
SmramProfileParameterGetInfo->Header.ReturnStatus = 0;\r
\r
- mSmramProfileRecordingStatus = SmramProfileRecordingStatus;\r
+ mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
**/\r
VOID\r
SmramProfileHandlerGetData (\r
- IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA *SmramProfileParameterGetData\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 SmramProfileRecordingStatus;\r
+ BOOLEAN SmramProfileGettingStatus;\r
\r
ContextData = GetSmramProfileContext ();\r
if (ContextData == NULL) {\r
- return ;\r
+ return;\r
}\r
\r
- SmramProfileRecordingStatus = mSmramProfileRecordingStatus;\r
- mSmramProfileRecordingStatus = FALSE;\r
-\r
+ SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+ mSmramProfileGettingStatus = TRUE;\r
\r
CopyMem (&SmramProfileGetData, SmramProfileParameterGetData, sizeof (SmramProfileGetData));\r
\r
- ProfileSize = SmramProfileGetDataSize();\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
+ if (!SmmIsBufferOutsideSmmValid ((UINTN)SmramProfileGetData.ProfileBuffer, (UINTN)ProfileSize)) {\r
+ DEBUG ((DEBUG_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
+ SmramProfileParameterGetData->ProfileSize = ProfileSize;\r
+ SmramProfileParameterGetData->Header.ReturnStatus = (UINT64)(INT64)(INTN)EFI_BUFFER_TOO_SMALL;\r
goto Done;\r
}\r
\r
- SmramProfileParameterGetData->ProfileSize = ProfileSize;\r
- SmramProfileCopyData ((VOID *) (UINTN) SmramProfileGetData.ProfileBuffer);\r
+ ProfileOffset = 0;\r
+ SmramProfileCopyData ((VOID *)(UINTN)SmramProfileGetData.ProfileBuffer, &ProfileSize, &ProfileOffset);\r
+ SmramProfileParameterGetData->ProfileSize = ProfileSize;\r
SmramProfileParameterGetData->Header.ReturnStatus = 0;\r
\r
Done:\r
- mSmramProfileRecordingStatus = SmramProfileRecordingStatus;\r
+ mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
- SMRAM profile handler to register SMM image.\r
+ SMRAM profile handler to get profile data by offset.\r
\r
- @param SmramProfileParameterRegisterImage The parameter of SMM profile register image.\r
+ @param SmramProfileParameterGetDataByOffset The parameter of SMM profile get data by offset.\r
\r
**/\r
VOID\r
-SmramProfileHandlerRegisterImage (\r
- IN SMRAM_PROFILE_PARAMETER_REGISTER_IMAGE *SmramProfileParameterRegisterImage\r
+SmramProfileHandlerGetDataByOffset (\r
+ IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET *SmramProfileParameterGetDataByOffset\r
)\r
{\r
- EFI_STATUS Status;\r
- EFI_SMM_DRIVER_ENTRY DriverEntry;\r
- VOID *EntryPointInImage;\r
- BOOLEAN Ret;\r
+ SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET SmramProfileGetDataByOffset;\r
+ MEMORY_PROFILE_CONTEXT_DATA *ContextData;\r
+ BOOLEAN SmramProfileGettingStatus;\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
- Ret = RegisterSmramProfileImage (&DriverEntry, FALSE);\r
- if (Ret) {\r
- SmramProfileParameterRegisterImage->Header.ReturnStatus = 0;\r
+ ContextData = GetSmramProfileContext ();\r
+ if (ContextData == NULL) {\r
+ return;\r
}\r
-}\r
\r
-/**\r
- SMRAM profile handler to unregister SMM image.\r
+ SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+ mSmramProfileGettingStatus = TRUE;\r
\r
- @param SmramProfileParameterUnregisterImage The parameter of SMM profile unregister image.\r
+ CopyMem (&SmramProfileGetDataByOffset, SmramProfileParameterGetDataByOffset, sizeof (SmramProfileGetDataByOffset));\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
- BOOLEAN Ret;\r
+ //\r
+ // Sanity check\r
+ //\r
+ if (!SmmIsBufferOutsideSmmValid ((UINTN)SmramProfileGetDataByOffset.ProfileBuffer, (UINTN)SmramProfileGetDataByOffset.ProfileSize)) {\r
+ DEBUG ((DEBUG_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
- 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
+ SmramProfileCopyData ((VOID *)(UINTN)SmramProfileGetDataByOffset.ProfileBuffer, &SmramProfileGetDataByOffset.ProfileSize, &SmramProfileGetDataByOffset.ProfileOffset);\r
+ CopyMem (SmramProfileParameterGetDataByOffset, &SmramProfileGetDataByOffset, sizeof (SmramProfileGetDataByOffset));\r
+ SmramProfileParameterGetDataByOffset->Header.ReturnStatus = 0;\r
\r
- Ret = UnregisterSmramProfileImage (&DriverEntry, FALSE);\r
- if (Ret) {\r
- SmramProfileParameterUnregisterImage->Header.ReturnStatus = 0;\r
- }\r
+Done:\r
+ mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\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
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandler Enter\n"));\r
\r
//\r
// If input is invalid, stop processing this SMI\r
//\r
- if (CommBuffer == NULL || CommBufferSize == NULL) {\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
+ DEBUG ((DEBUG_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
+ DEBUG ((DEBUG_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
+ 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
+ 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_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
- default:\r
- break;\r
+ case SMRAM_PROFILE_COMMAND_GET_PROFILE_INFO:\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandlerGetInfo\n"));\r
+ if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_GET_PROFILE_INFO)) {\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
+ return EFI_SUCCESS;\r
+ }\r
+\r
+ SmramProfileHandlerGetInfo ((SMRAM_PROFILE_PARAMETER_GET_PROFILE_INFO *)(UINTN)CommBuffer);\r
+ break;\r
+ case SMRAM_PROFILE_COMMAND_GET_PROFILE_DATA:\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandlerGetData\n"));\r
+ if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA)) {\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
+ return EFI_SUCCESS;\r
+ }\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 ((DEBUG_ERROR, "SmramProfileHandlerGetDataByOffset\n"));\r
+ if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET)) {\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
+ return EFI_SUCCESS;\r
+ }\r
+\r
+ SmramProfileHandlerGetDataByOffset ((SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET *)(UINTN)CommBuffer);\r
+ break;\r
+ case SMRAM_PROFILE_COMMAND_GET_RECORDING_STATE:\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandlerGetRecordingState\n"));\r
+ if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_RECORDING_STATE)) {\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
+ return EFI_SUCCESS;\r
+ }\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 ((DEBUG_ERROR, "SmramProfileHandlerSetRecordingState\n"));\r
+ if (TempCommBufferSize != sizeof (SMRAM_PROFILE_PARAMETER_RECORDING_STATE)) {\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandler: SMM communication buffer size invalid!\n"));\r
+ return EFI_SUCCESS;\r
+ }\r
+\r
+ ParameterRecordingState = (SMRAM_PROFILE_PARAMETER_RECORDING_STATE *)(UINTN)CommBuffer;\r
+ mSmramProfileRecordingEnable = ParameterRecordingState->RecordingState;\r
+ ParameterRecordingState->Header.ReturnStatus = 0;\r
+ break;\r
+\r
+ //\r
+ // Below 2 commands have been deprecated. They may not be (re-)used.\r
+ //\r
+ case SMRAM_PROFILE_COMMAND_DEPRECATED1:\r
+ case SMRAM_PROFILE_COMMAND_DEPRECATED2:\r
+ ASSERT (FALSE);\r
+ //\r
+ // Fall-through to the default (unrecognized command) case.\r
+ //\r
+ default:\r
+ break;\r
}\r
\r
- DEBUG ((EFI_D_ERROR, "SmramProfileHandler Exit\n"));\r
+ DEBUG ((DEBUG_ERROR, "SmramProfileHandler Exit\n"));\r
\r
return EFI_SUCCESS;\r
}\r
VOID\r
)\r
{\r
- EFI_STATUS Status;\r
- EFI_HANDLE DispatchHandle;\r
+ EFI_STATUS Status;\r
+ EFI_HANDLE DispatchHandle;\r
\r
if (!IS_SMRAM_PROFILE_ENABLED) {\r
return;\r
VOID\r
)\r
{\r
- UINTN Index;\r
- MEMORY_PROFILE_CONTEXT_DATA *ContextData;\r
- BOOLEAN SmramProfileRecordingStatus;\r
+ UINTN Index;\r
+ MEMORY_PROFILE_CONTEXT_DATA *ContextData;\r
+ BOOLEAN SmramProfileGettingStatus;\r
\r
ContextData = GetSmramProfileContext ();\r
if (ContextData == NULL) {\r
- return ;\r
+ return;\r
}\r
\r
- SmramProfileRecordingStatus = mSmramProfileRecordingStatus;\r
- mSmramProfileRecordingStatus = FALSE;\r
+ SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+ mSmramProfileGettingStatus = TRUE;\r
\r
- DEBUG ((EFI_D_INFO, "FullSmramRange address - 0x%08x\n", mFullSmramRanges));\r
+ DEBUG ((DEBUG_INFO, "FullSmramRange address - 0x%08x\n", mFullSmramRanges));\r
\r
- DEBUG ((EFI_D_INFO, "======= SmramProfile begin =======\n"));\r
+ DEBUG ((DEBUG_INFO, "======= SmramProfile begin =======\n"));\r
\r
- DEBUG ((EFI_D_INFO, "FullSmramRange:\n"));\r
+ DEBUG ((DEBUG_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
+ DEBUG ((DEBUG_INFO, " FullSmramRange (0x%x)\n", Index));\r
+ DEBUG ((DEBUG_INFO, " PhysicalStart - 0x%016lx\n", mFullSmramRanges[Index].PhysicalStart));\r
+ DEBUG ((DEBUG_INFO, " CpuStart - 0x%016lx\n", mFullSmramRanges[Index].CpuStart));\r
+ DEBUG ((DEBUG_INFO, " PhysicalSize - 0x%016lx\n", mFullSmramRanges[Index].PhysicalSize));\r
+ DEBUG ((DEBUG_INFO, " RegionState - 0x%016lx\n", mFullSmramRanges[Index].RegionState));\r
}\r
\r
- DEBUG ((EFI_D_INFO, "======= SmramProfile end =======\n"));\r
+ DEBUG ((DEBUG_INFO, "======= SmramProfile end =======\n"));\r
\r
- mSmramProfileRecordingStatus = SmramProfileRecordingStatus;\r
+ mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\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 SmramProfileRecordingStatus;\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
+ return;\r
}\r
\r
- SmramProfileRecordingStatus = mSmramProfileRecordingStatus;\r
- mSmramProfileRecordingStatus = FALSE;\r
+ SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+ mSmramProfileGettingStatus = TRUE;\r
\r
- DEBUG ((EFI_D_INFO, "======= SmramProfile begin =======\n"));\r
+ DEBUG ((DEBUG_INFO, "======= SmramProfile begin =======\n"));\r
\r
- DEBUG ((EFI_D_INFO, "FreePagesList:\n"));\r
+ DEBUG ((DEBUG_INFO, "FreePagesList:\n"));\r
FreePageList = &mSmmMemoryMap;\r
for (Node = FreePageList->BackLink, Index = 0;\r
Node != FreePageList;\r
- Node = Node->BackLink, Index++) {\r
+ Node = Node->BackLink, Index++)\r
+ {\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
+ DEBUG ((DEBUG_INFO, " Index - 0x%x\n", Index));\r
+ DEBUG ((DEBUG_INFO, " PhysicalStart - 0x%016lx\n", (PHYSICAL_ADDRESS)(UINTN)Pages));\r
+ DEBUG ((DEBUG_INFO, " NumberOfPages - 0x%08x\n", Pages->NumberOfPages));\r
}\r
\r
- DEBUG ((EFI_D_INFO, "======= SmramProfile end =======\n"));\r
+ DEBUG ((DEBUG_INFO, "======= SmramProfile end =======\n"));\r
\r
- mSmramProfileRecordingStatus = SmramProfileRecordingStatus;\r
+ mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\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 SmramProfileRecordingStatus;\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
+ return;\r
}\r
\r
- SmramProfileRecordingStatus = mSmramProfileRecordingStatus;\r
- mSmramProfileRecordingStatus = FALSE;\r
-\r
- DEBUG ((EFI_D_INFO, "======= SmramProfile begin =======\n"));\r
-\r
- for (PoolListIndex = 0; PoolListIndex < MAX_POOL_INDEX; PoolListIndex++) {\r
- DEBUG ((EFI_D_INFO, "FreePoolList (%d):\n", PoolListIndex));\r
- FreePoolList = &mSmmPoolLists[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 ((EFI_D_INFO, " Index - 0x%x\n", Index));\r
- DEBUG ((EFI_D_INFO, " PhysicalStart - 0x%016lx\n", (PHYSICAL_ADDRESS) (UINTN) Pool));\r
- DEBUG ((EFI_D_INFO, " Size - 0x%08x\n", Pool->Header.Size));\r
- DEBUG ((EFI_D_INFO, " Available - 0x%02x\n", Pool->Header.Available));\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
+ {\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 ((EFI_D_INFO, "======= SmramProfile end =======\n"));\r
+ DEBUG ((DEBUG_INFO, "======= SmramProfile end =======\n"));\r
\r
- mSmramProfileRecordingStatus = SmramProfileRecordingStatus;\r
+ mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
-GLOBAL_REMOVE_IF_UNREFERENCED CHAR16 *mActionString[] = {\r
- L"Unknown",\r
- L"AllocatePages",\r
- L"FreePages",\r
- L"AllocatePool",\r
- L"FreePool",\r
+GLOBAL_REMOVE_IF_UNREFERENCED CHAR8 *mSmmActionString[] = {\r
+ "SmmUnknown",\r
+ "gSmst->SmmAllocatePages",\r
+ "gSmst->SmmFreePages",\r
+ "gSmst->SmmAllocatePool",\r
+ "gSmst->SmmFreePool",\r
};\r
\r
-GLOBAL_REMOVE_IF_UNREFERENCED CHAR16 *mMemoryTypeString[] = {\r
- L"EfiReservedMemoryType",\r
- L"EfiLoaderCode",\r
- L"EfiLoaderData",\r
- L"EfiBootServicesCode",\r
- L"EfiBootServicesData",\r
- L"EfiRuntimeServicesCode",\r
- L"EfiRuntimeServicesData",\r
- L"EfiConventionalMemory",\r
- L"EfiUnusableMemory",\r
- L"EfiACPIReclaimMemory",\r
- L"EfiACPIMemoryNVS",\r
- L"EfiMemoryMappedIO",\r
- L"EfiMemoryMappedIOPortSpace",\r
- L"EfiPalCode",\r
- L"EfiOSReserved",\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
+\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
+\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
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 SmramProfileRecordingStatus;\r
- UINTN TypeIndex;\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
+ return;\r
}\r
\r
- SmramProfileRecordingStatus = mSmramProfileRecordingStatus;\r
- mSmramProfileRecordingStatus = FALSE;\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
+ DEBUG ((DEBUG_INFO, "======= SmramProfile begin =======\n"));\r
+ DEBUG ((DEBUG_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 <= EfiMaxMemoryType; TypeIndex++) {\r
+ DEBUG ((DEBUG_INFO, " CurrentTotalUsage - 0x%016lx\n", Context->CurrentTotalUsage));\r
+ DEBUG ((DEBUG_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 (%s)\n", TypeIndex, Context->CurrentTotalUsageByType[TypeIndex], mMemoryTypeString[TypeIndex]));\r
- DEBUG ((EFI_D_INFO, " PeakTotalUsage[0x%02x] - 0x%016lx (%s)\n", TypeIndex, Context->PeakTotalUsageByType[TypeIndex], mMemoryTypeString[TypeIndex]));\r
+ (Context->PeakTotalUsageByType[TypeIndex] != 0))\r
+ {\r
+ DEBUG ((DEBUG_INFO, " CurrentTotalUsage[0x%02x] - 0x%016lx (%a)\n", TypeIndex, Context->CurrentTotalUsageByType[TypeIndex], ProfileMemoryTypeToStr (TypeIndex)));\r
+ DEBUG ((DEBUG_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
+ DEBUG ((DEBUG_INFO, " TotalImageSize - 0x%016lx\n", Context->TotalImageSize));\r
+ DEBUG ((DEBUG_INFO, " ImageCount - 0x%08x\n", Context->ImageCount));\r
+ DEBUG ((DEBUG_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
+ DriverLink = DriverLink->ForwardLink, DriverIndex++)\r
+ {\r
DriverInfoData = CR (\r
- DriverLink,\r
- MEMORY_PROFILE_DRIVER_INFO_DATA,\r
- Link,\r
- MEMORY_PROFILE_DRIVER_INFO_SIGNATURE\r
- );\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 <= EfiMaxMemoryType; TypeIndex++) {\r
+ DEBUG ((DEBUG_INFO, " MEMORY_PROFILE_DRIVER_INFO (0x%x)\n", DriverIndex));\r
+ DEBUG ((DEBUG_INFO, " FileName - %g\n", &DriverInfo->FileName));\r
+ DEBUG ((DEBUG_INFO, " ImageBase - 0x%016lx\n", DriverInfo->ImageBase));\r
+ DEBUG ((DEBUG_INFO, " ImageSize - 0x%016lx\n", DriverInfo->ImageSize));\r
+ DEBUG ((DEBUG_INFO, " EntryPoint - 0x%016lx\n", DriverInfo->EntryPoint));\r
+ DEBUG ((DEBUG_INFO, " ImageSubsystem - 0x%04x\n", DriverInfo->ImageSubsystem));\r
+ DEBUG ((DEBUG_INFO, " FileType - 0x%02x\n", DriverInfo->FileType));\r
+ DEBUG ((DEBUG_INFO, " CurrentUsage - 0x%016lx\n", DriverInfo->CurrentUsage));\r
+ DEBUG ((DEBUG_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 (%s)\n", TypeIndex, DriverInfo->CurrentUsageByType[TypeIndex], mMemoryTypeString[TypeIndex]));\r
- DEBUG ((EFI_D_INFO, " PeakUsage[0x%02x] - 0x%016lx (%s)\n", TypeIndex, DriverInfo->PeakUsageByType[TypeIndex], mMemoryTypeString[TypeIndex]));\r
+ (DriverInfo->PeakUsageByType[TypeIndex] != 0))\r
+ {\r
+ DEBUG ((DEBUG_INFO, " CurrentUsage[0x%02x] - 0x%016lx (%a)\n", TypeIndex, DriverInfo->CurrentUsageByType[TypeIndex], ProfileMemoryTypeToStr (TypeIndex)));\r
+ DEBUG ((DEBUG_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
+ DEBUG ((DEBUG_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
+ AllocLink = AllocLink->ForwardLink, AllocIndex++)\r
+ {\r
AllocInfoData = CR (\r
- AllocLink,\r
- MEMORY_PROFILE_ALLOC_INFO_DATA,\r
- Link,\r
- MEMORY_PROFILE_ALLOC_INFO_SIGNATURE\r
- );\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
- DEBUG ((EFI_D_INFO, " Action - 0x%08x (%s)\n", AllocInfo->Action, mActionString[(AllocInfo->Action < sizeof(mActionString)/sizeof(mActionString[0])) ? AllocInfo->Action : 0]));\r
- DEBUG ((EFI_D_INFO, " MemoryType - 0x%08x\n", 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
+ DEBUG ((DEBUG_INFO, " MEMORY_PROFILE_ALLOC_INFO (0x%x)\n", AllocIndex));\r
+ DEBUG ((DEBUG_INFO, " CallerAddress - 0x%016lx (Offset: 0x%08x)\n", AllocInfo->CallerAddress, AllocInfo->CallerAddress - DriverInfo->ImageBase));\r
+ DEBUG ((DEBUG_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 ((DEBUG_INFO, " Action - 0x%08x (%a)\n", AllocInfo->Action, AllocInfoData->ActionString));\r
+ } else {\r
+ DEBUG ((DEBUG_INFO, " Action - 0x%08x (UserDefined-0x%08x)\n", AllocInfo->Action, AllocInfo->Action));\r
+ }\r
+ } else {\r
+ DEBUG ((DEBUG_INFO, " Action - 0x%08x (%a)\n", AllocInfo->Action, ProfileActionToStr (AllocInfo->Action)));\r
+ }\r
+\r
+ DEBUG ((DEBUG_INFO, " MemoryType - 0x%08x (%a)\n", AllocInfo->MemoryType, ProfileMemoryTypeToStr (AllocInfo->MemoryType)));\r
+ DEBUG ((DEBUG_INFO, " Buffer - 0x%016lx\n", AllocInfo->Buffer));\r
+ DEBUG ((DEBUG_INFO, " Size - 0x%016lx\n", AllocInfo->Size));\r
}\r
}\r
\r
- DEBUG ((EFI_D_INFO, "======= SmramProfile end =======\n"));\r
+ DEBUG ((DEBUG_INFO, "======= SmramProfile end =======\n"));\r
\r
- mSmramProfileRecordingStatus = SmramProfileRecordingStatus;\r
+ mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
}\r
\r
/**\r
- Dump SMRAM infromation.\r
+ Dump SMRAM information.\r
\r
**/\r
VOID\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
+ DEBUG_CODE_BEGIN ();\r
+ if (IS_SMRAM_PROFILE_ENABLED) {\r
+ DumpSmramProfile ();\r
+ DumpFreePagesList ();\r
+ DumpFreePoolList ();\r
+ DumpSmramRange ();\r
+ }\r
\r
+ DEBUG_CODE_END ();\r
+}\r
projects / mirror_edk2.git / blobdiff