]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Core/PiSmmCore/SmramProfileRecord.c
MdeModulePkg: Apply uncrustify changes
[mirror_edk2.git] / MdeModulePkg / Core / PiSmmCore / SmramProfileRecord.c
index 002a8e43606557c6af6702a16f4b18b52f2018ce..b437e4c433095a99560d2710871c2212204cd626 100644 (file)
@@ -8,16 +8,16 @@
 \r
 #include "PiSmmCore.h"\r
 \r
-#define IS_SMRAM_PROFILE_ENABLED ((PcdGet8 (PcdMemoryProfilePropertyMask) & BIT1) != 0)\r
-#define IS_UEFI_MEMORY_PROFILE_ENABLED ((PcdGet8 (PcdMemoryProfilePropertyMask) & BIT0) != 0)\r
+#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
@@ -29,10 +29,10 @@ typedef struct {
 } MEMORY_PROFILE_DRIVER_INFO_DATA;\r
 \r
 typedef struct {\r
-  UINT32                        Signature;\r
-  MEMORY_PROFILE_ALLOC_INFO     AllocInfo;\r
-  CHAR8                         *ActionString;\r
-  LIST_ENTRY                    Link;\r
+  UINT32                       Signature;\r
+  MEMORY_PROFILE_ALLOC_INFO    AllocInfo;\r
+  CHAR8                        *ActionString;\r
+  LIST_ENTRY                   Link;\r
 } MEMORY_PROFILE_ALLOC_INFO_DATA;\r
 \r
 //\r
@@ -40,7 +40,7 @@ typedef struct {
 //\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
@@ -50,8 +50,8 @@ GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_FREE_MEMORY mSmramFreeMemory = {
   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
@@ -61,21 +61,21 @@ GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_CONTEXT_DATA mSmramProfileContext =
     },\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 = NULL;\r
+GLOBAL_REMOVE_IF_UNREFERENCED MEMORY_PROFILE_CONTEXT_DATA  *mSmramProfileContextPtr = NULL;\r
 \r
-GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN mSmramReadyToLock;\r
-GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN mSmramProfileGettingStatus = FALSE;\r
-GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN mSmramProfileRecordingEnable = MEMORY_PROFILE_RECORDING_DISABLE;\r
-GLOBAL_REMOVE_IF_UNREFERENCED EFI_DEVICE_PATH_PROTOCOL *mSmramProfileDriverPath;\r
-GLOBAL_REMOVE_IF_UNREFERENCED UINTN                    mSmramProfileDriverPathSize;\r
+GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN                   mSmramReadyToLock;\r
+GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN                   mSmramProfileGettingStatus   = FALSE;\r
+GLOBAL_REMOVE_IF_UNREFERENCED BOOLEAN                   mSmramProfileRecordingEnable = MEMORY_PROFILE_RECORDING_DISABLE;\r
+GLOBAL_REMOVE_IF_UNREFERENCED EFI_DEVICE_PATH_PROTOCOL  *mSmramProfileDriverPath;\r
+GLOBAL_REMOVE_IF_UNREFERENCED UINTN                     mSmramProfileDriverPathSize;\r
 \r
 /**\r
   Dump SMRAM information.\r
@@ -105,7 +105,7 @@ EFIAPI
 SmramProfileProtocolGetData (\r
   IN     EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
   IN OUT UINT64                             *ProfileSize,\r
-     OUT VOID                               *ProfileBuffer\r
+  OUT VOID                                  *ProfileBuffer\r
   );\r
 \r
 /**\r
@@ -226,7 +226,7 @@ SmramProfileProtocolRecord (
   IN CHAR8                              *ActionString OPTIONAL\r
   );\r
 \r
-GLOBAL_REMOVE_IF_UNREFERENCED EDKII_SMM_MEMORY_PROFILE_PROTOCOL mSmmProfileProtocol = {\r
+GLOBAL_REMOVE_IF_UNREFERENCED EDKII_SMM_MEMORY_PROFILE_PROTOCOL  mSmmProfileProtocol = {\r
   SmramProfileProtocolGetData,\r
   SmramProfileProtocolRegisterImage,\r
   SmramProfileProtocolUnregisterImage,\r
@@ -269,22 +269,22 @@ InternalPeCoffGetSubsystem (
 \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
+  } 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
@@ -319,23 +319,23 @@ InternalPeCoffGetEntryPoint (
   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
@@ -343,10 +343,10 @@ InternalPeCoffGetEntryPoint (
   // 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
@@ -369,30 +369,30 @@ InternalPeCoffGetEntryPoint (
 **/\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
-  CHAR8                             *PdbString;\r
-  UINTN                             PdbSize;\r
-  UINTN                             PdbOccupiedSize;\r
-\r
-  PdbSize = 0;\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
+  PdbString       = NULL;\r
   if (ImageBase != 0) {\r
-    PdbString = PeCoffLoaderGetPdbPointer ((VOID*) (UINTN) ImageBase);\r
+    PdbString = PeCoffLoaderGetPdbPointer ((VOID *)(UINTN)ImageBase);\r
     if (PdbString != NULL) {\r
-      PdbSize = AsciiStrSize (PdbString);\r
+      PdbSize         = AsciiStrSize (PdbString);\r
       PdbOccupiedSize = GET_OCCUPIED_SIZE (PdbSize, sizeof (UINT64));\r
     }\r
   }\r
@@ -403,53 +403,56 @@ BuildDriverInfo (
   Status = SmmInternalAllocatePool (\r
              EfiRuntimeServicesData,\r
              sizeof (*DriverInfoData) + sizeof (LIST_ENTRY) + PdbSize,\r
-             (VOID **) &DriverInfoData\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 = (UINT16) (sizeof (MEMORY_PROFILE_DRIVER_INFO) + PdbOccupiedSize);\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
+    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
+    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
@@ -466,28 +469,27 @@ BuildDriverInfo (
 **/\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
+  EFI_STATUS                         Status;\r
+  EDKII_MEMORY_PROFILE_PROTOCOL      *ProfileProtocol;\r
+  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH  *FilePath;\r
+  UINT8                              TempBuffer[sizeof (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof (EFI_DEVICE_PATH_PROTOCOL)];\r
 \r
   if (IS_UEFI_MEMORY_PROFILE_ENABLED) {\r
-\r
     FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)TempBuffer;\r
-    Status = gBS->LocateProtocol (&gEdkiiMemoryProfileGuid, NULL, (VOID **) &ProfileProtocol);\r
+    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
@@ -506,27 +508,26 @@ RegisterImageToDxe (
 **/\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
+  EFI_STATUS                         Status;\r
+  EDKII_MEMORY_PROFILE_PROTOCOL      *ProfileProtocol;\r
+  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH  *FilePath;\r
+  UINT8                              TempBuffer[sizeof (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof (EFI_DEVICE_PATH_PROTOCOL)];\r
 \r
   if (IS_UEFI_MEMORY_PROFILE_ENABLED) {\r
-\r
     FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)TempBuffer;\r
-    Status = gBS->LocateProtocol (&gEdkiiMemoryProfileGuid, NULL, (VOID *) &ProfileProtocol);\r
+    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
@@ -545,13 +546,13 @@ UnregisterImageFromDxe (
 **/\r
 BOOLEAN\r
 NeedRecordThisDriver (\r
-  IN EFI_DEVICE_PATH_PROTOCOL       *DriverFilePath\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
+  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
@@ -563,7 +564,7 @@ NeedRecordThisDriver (
   //\r
   // Record FilePath without end node.\r
   //\r
-  FilePathSize = GetDevicePathSize (DriverFilePath) - sizeof(EFI_DEVICE_PATH_PROTOCOL);\r
+  FilePathSize = GetDevicePathSize (DriverFilePath) - sizeof (EFI_DEVICE_PATH_PROTOCOL);\r
 \r
   DevicePathInstance = mSmramProfileDriverPath;\r
   do {\r
@@ -580,14 +581,15 @@ NeedRecordThisDriver (
     //\r
     DevicePathSize = (UINTN)TmpDevicePath - (UINTN)DevicePathInstance;\r
     if ((FilePathSize == DevicePathSize) &&\r
-        (CompareMem (DriverFilePath, DevicePathInstance, DevicePathSize) == 0)) {\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
+    DevicePathInstance = (EFI_DEVICE_PATH_PROTOCOL *)((UINTN)DevicePathInstance + DevicePathSize + DevicePathNodeLength (TmpDevicePath));\r
   } while (DevicePathSubType (TmpDevicePath) != END_ENTIRE_DEVICE_PATH_SUBTYPE);\r
 \r
   return FALSE;\r
@@ -604,30 +606,30 @@ NeedRecordThisDriver (
 **/\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
-  UINT8                             TempBuffer[sizeof(MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof(EFI_DEVICE_PATH_PROTOCOL)];\r
-  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FilePath;\r
+  MEMORY_PROFILE_DRIVER_INFO_DATA    *DriverInfoData;\r
+  PHYSICAL_ADDRESS                   ImageBase;\r
+  UINT8                              TempBuffer[sizeof (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof (EFI_DEVICE_PATH_PROTOCOL)];\r
+  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH  *FilePath;\r
 \r
-  FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) TempBuffer;\r
+  FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)TempBuffer;\r
   EfiInitializeFwVolDevicepathNode (FilePath, &gEfiCallerIdGuid);\r
   SetDevicePathEndNode (FilePath + 1);\r
 \r
-  if (!NeedRecordThisDriver ((EFI_DEVICE_PATH_PROTOCOL *) FilePath)) {\r
+  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
@@ -646,7 +648,7 @@ SmramProfileInit (
   VOID\r
   )\r
 {\r
-  MEMORY_PROFILE_CONTEXT_DATA *SmramProfileContext;\r
+  MEMORY_PROFILE_CONTEXT_DATA  *SmramProfileContext;\r
 \r
   RegisterImageToDxe (\r
     &gEfiCallerIdGuid,\r
@@ -670,9 +672,10 @@ SmramProfileInit (
   } else {\r
     mSmramProfileRecordingEnable = MEMORY_PROFILE_RECORDING_ENABLE;\r
   }\r
+\r
   mSmramProfileDriverPathSize = PcdGetSize (PcdMemoryProfileDriverPath);\r
-  mSmramProfileDriverPath = AllocateCopyPool (mSmramProfileDriverPathSize, PcdGetPtr (PcdMemoryProfileDriverPath));\r
-  mSmramProfileContextPtr = &mSmramProfileContext;\r
+  mSmramProfileDriverPath     = AllocateCopyPool (mSmramProfileDriverPathSize, PcdGetPtr (PcdMemoryProfileDriverPath));\r
+  mSmramProfileContextPtr     = &mSmramProfileContext;\r
 \r
   RegisterSmmCore (&mSmramProfileContext);\r
 \r
@@ -688,8 +691,8 @@ SmramProfileInstallProtocol (
   VOID\r
   )\r
 {\r
-  EFI_HANDLE    Handle;\r
-  EFI_STATUS    Status;\r
+  EFI_HANDLE  Handle;\r
+  EFI_STATUS  Status;\r
 \r
   if (!IS_SMRAM_PROFILE_ENABLED) {\r
     return;\r
@@ -715,21 +718,22 @@ SmramProfileInstallProtocol (
 **/\r
 EFI_GUID *\r
 GetFileNameFromFilePath (\r
-  IN EFI_DEVICE_PATH_PROTOCOL   *FilePath\r
+  IN EFI_DEVICE_PATH_PROTOCOL  *FilePath\r
   )\r
 {\r
-  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH     *ThisFilePath;\r
-  EFI_GUID                              *FileName;\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
+    ThisFilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)FilePath;\r
     while (!IsDevicePathEnd (ThisFilePath)) {\r
       FileName = EfiGetNameGuidFromFwVolDevicePathNode (ThisFilePath);\r
       if (FileName != NULL) {\r
         break;\r
       }\r
-      ThisFilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) NextDevicePathNode (ThisFilePath);\r
+\r
+      ThisFilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *)NextDevicePathNode (ThisFilePath);\r
     }\r
   }\r
 \r
@@ -750,14 +754,14 @@ GetFileNameFromFilePath (
 **/\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
-  UINT8                             TempBuffer[sizeof(MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof(EFI_DEVICE_PATH_PROTOCOL)];\r
-  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FilePath;\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
@@ -772,11 +776,11 @@ RegisterSmramProfileImage (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
-  FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) TempBuffer;\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
+  if (!NeedRecordThisDriver ((EFI_DEVICE_PATH_PROTOCOL *)FilePath)) {\r
     return EFI_UNSUPPORTED;\r
   }\r
 \r
@@ -791,7 +795,7 @@ RegisterSmramProfileImage (
                      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
@@ -813,21 +817,22 @@ RegisterSmramProfileImage (
 **/\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
@@ -837,7 +842,8 @@ GetMemoryProfileDriverInfoByFileNameAndAddress (
     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
@@ -857,20 +863,21 @@ GetMemoryProfileDriverInfoByFileNameAndAddress (
 **/\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
@@ -879,7 +886,8 @@ GetMemoryProfileDriverInfoFromAddress (
                        );\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
@@ -905,14 +913,14 @@ UnregisterSmramProfileImage (
   IN BOOLEAN               UnregisterFromDxe\r
   )\r
 {\r
-  EFI_STATUS                        Status;\r
-  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
-  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
-  EFI_GUID                          *FileName;\r
-  PHYSICAL_ADDRESS                  ImageAddress;\r
-  VOID                              *EntryPointInImage;\r
-  UINT8                             TempBuffer[sizeof(MEDIA_FW_VOL_FILEPATH_DEVICE_PATH) + sizeof(EFI_DEVICE_PATH_PROTOCOL)];\r
-  MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *FilePath;\r
+  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
@@ -926,11 +934,11 @@ UnregisterSmramProfileImage (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
-  FilePath = (MEDIA_FW_VOL_FILEPATH_DEVICE_PATH *) TempBuffer;\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
+  if (!NeedRecordThisDriver ((EFI_DEVICE_PATH_PROTOCOL *)FilePath)) {\r
     return EFI_UNSUPPORTED;\r
   }\r
 \r
@@ -940,23 +948,26 @@ UnregisterSmramProfileImage (
   }\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 EFI_NOT_FOUND;\r
   }\r
@@ -964,11 +975,11 @@ UnregisterSmramProfileImage (
   ContextData->Context.TotalImageSize -= DriverInfoData->DriverInfo.ImageSize;\r
 \r
   // Keep the ImageBase for RVA calculation in Application.\r
-  //DriverInfoData->DriverInfo.ImageBase = 0;\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
@@ -991,13 +1002,14 @@ UnregisterSmramProfileImage (
 **/\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 (MemoryType != EfiRuntimeServicesCode &&\r
-      MemoryType != EfiRuntimeServicesData) {\r
+  if ((MemoryType != EfiRuntimeServicesCode) &&\r
+      (MemoryType != EfiRuntimeServicesData))\r
+  {\r
     return FALSE;\r
   }\r
 \r
@@ -1023,7 +1035,7 @@ SmmCoreNeedRecordProfile (
 **/\r
 EFI_MEMORY_TYPE\r
 GetProfileMemoryIndex (\r
-  IN EFI_MEMORY_TYPE    MemoryType\r
+  IN EFI_MEMORY_TYPE  MemoryType\r
   )\r
 {\r
   return MemoryType;\r
@@ -1040,17 +1052,18 @@ SmramProfileUpdateFreePages (
   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
@@ -1087,17 +1100,17 @@ SmmCoreUpdateProfileAllocate (
   IN CHAR8                  *ActionString OPTIONAL\r
   )\r
 {\r
-  EFI_STATUS                        Status;\r
-  MEMORY_PROFILE_CONTEXT            *Context;\r
-  MEMORY_PROFILE_DRIVER_INFO        *DriverInfo;\r
-  MEMORY_PROFILE_ALLOC_INFO         *AllocInfo;\r
-  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
-  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
-  MEMORY_PROFILE_ALLOC_INFO_DATA    *AllocInfoData;\r
-  EFI_MEMORY_TYPE                   ProfileMemoryIndex;\r
-  MEMORY_PROFILE_ACTION             BasicAction;\r
-  UINTN                             ActionStringSize;\r
-  UINTN                             ActionStringOccupiedSize;\r
+  EFI_STATUS                       Status;\r
+  MEMORY_PROFILE_CONTEXT           *Context;\r
+  MEMORY_PROFILE_DRIVER_INFO       *DriverInfo;\r
+  MEMORY_PROFILE_ALLOC_INFO        *AllocInfo;\r
+  MEMORY_PROFILE_CONTEXT_DATA      *ContextData;\r
+  MEMORY_PROFILE_DRIVER_INFO_DATA  *DriverInfoData;\r
+  MEMORY_PROFILE_ALLOC_INFO_DATA   *AllocInfoData;\r
+  EFI_MEMORY_TYPE                  ProfileMemoryIndex;\r
+  MEMORY_PROFILE_ACTION            BasicAction;\r
+  UINTN                            ActionStringSize;\r
+  UINTN                            ActionStringOccupiedSize;\r
 \r
   BasicAction = Action & MEMORY_PROFILE_ACTION_BASIC_MASK;\r
 \r
@@ -1111,10 +1124,10 @@ SmmCoreUpdateProfileAllocate (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
-  ActionStringSize = 0;\r
+  ActionStringSize         = 0;\r
   ActionStringOccupiedSize = 0;\r
   if (ActionString != NULL) {\r
-    ActionStringSize = AsciiStrSize (ActionString);\r
+    ActionStringSize         = AsciiStrSize (ActionString);\r
     ActionStringOccupiedSize = GET_OCCUPIED_SIZE (ActionStringSize, sizeof (UINT64));\r
   }\r
 \r
@@ -1122,48 +1135,49 @@ SmmCoreUpdateProfileAllocate (
   // Use SmmInternalAllocatePool() that will not update profile for this AllocatePool action.\r
   //\r
   AllocInfoData = NULL;\r
-  Status = SmmInternalAllocatePool (\r
-             EfiRuntimeServicesData,\r
-             sizeof (*AllocInfoData) + ActionStringSize,\r
-             (VOID **) &AllocInfoData\r
-             );\r
+  Status        = SmmInternalAllocatePool (\r
+                    EfiRuntimeServicesData,\r
+                    sizeof (*AllocInfoData) + ActionStringSize,\r
+                    (VOID **)&AllocInfoData\r
+                    );\r
   if (EFI_ERROR (Status)) {\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
+\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
+    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
+    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
+    AllocInfoData->ActionString   = NULL;\r
   }\r
 \r
   InsertTailList (DriverInfoData->AllocInfoList, &AllocInfoData->Link);\r
 \r
-  Context = &ContextData->Context;\r
+  Context    = &ContextData->Context;\r
   DriverInfo = &DriverInfoData->DriverInfo;\r
-  DriverInfo->AllocRecordCount ++;\r
+  DriverInfo->AllocRecordCount++;\r
 \r
   //\r
   // Update summary if and only if it is basic action.\r
@@ -1175,6 +1189,7 @@ SmmCoreUpdateProfileAllocate (
     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
@@ -1184,6 +1199,7 @@ SmmCoreUpdateProfileAllocate (
     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
@@ -1207,22 +1223,23 @@ SmmCoreUpdateProfileAllocate (
 **/\r
 MEMORY_PROFILE_ALLOC_INFO_DATA *\r
 GetMemoryProfileAllocInfoFromAddress (\r
-  IN MEMORY_PROFILE_DRIVER_INFO_DATA    *DriverInfoData,\r
-  IN MEMORY_PROFILE_ACTION              BasicAction,\r
-  IN UINTN                              Size,\r
-  IN VOID                               *Buffer\r
+  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
@@ -1233,17 +1250,21 @@ GetMemoryProfileAllocInfoFromAddress (
     if ((AllocInfo->Action & MEMORY_PROFILE_ACTION_BASIC_MASK) != BasicAction) {\r
       continue;\r
     }\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
@@ -1306,7 +1327,7 @@ SmmCoreUpdateProfileFree (
   // Need use do-while loop to find all possible record,\r
   // because one address might be recorded multiple times.\r
   //\r
-  Found = FALSE;\r
+  Found         = FALSE;\r
   AllocInfoData = NULL;\r
   do {\r
     if (DriverInfoData != NULL) {\r
@@ -1323,6 +1344,7 @@ SmmCoreUpdateProfileFree (
           break;\r
       }\r
     }\r
+\r
     if (AllocInfoData == NULL) {\r
       //\r
       // Legal case, because driver A might free memory allocated by driver B, by some protocol.\r
@@ -1331,7 +1353,8 @@ SmmCoreUpdateProfileFree (
 \r
       for (DriverLink = DriverInfoList->ForwardLink;\r
            DriverLink != DriverInfoList;\r
-           DriverLink = DriverLink->ForwardLink) {\r
+           DriverLink = DriverLink->ForwardLink)\r
+      {\r
         ThisDriverInfoData = CR (\r
                                DriverLink,\r
                                MEMORY_PROFILE_DRIVER_INFO_DATA,\r
@@ -1350,6 +1373,7 @@ SmmCoreUpdateProfileFree (
             AllocInfoData = NULL;\r
             break;\r
         }\r
+\r
         if (AllocInfoData != NULL) {\r
           DriverInfoData = ThisDriverInfoData;\r
           break;\r
@@ -1374,44 +1398,45 @@ SmmCoreUpdateProfileFree (
 \r
     Found = TRUE;\r
 \r
-    Context = &ContextData->Context;\r
+    Context    = &ContextData->Context;\r
     DriverInfo = &DriverInfoData->DriverInfo;\r
-    AllocInfo = &AllocInfoData->AllocInfo;\r
+    AllocInfo  = &AllocInfoData->AllocInfo;\r
 \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
-      Context->CurrentTotalUsage -= AllocInfo->Size;\r
+      Context->CurrentTotalUsage                           -= AllocInfo->Size;\r
       Context->CurrentTotalUsageByType[ProfileMemoryIndex] -= AllocInfo->Size;\r
 \r
-      DriverInfo->CurrentUsage -= AllocInfo->Size;\r
+      DriverInfo->CurrentUsage                           -= AllocInfo->Size;\r
       DriverInfo->CurrentUsageByType[ProfileMemoryIndex] -= AllocInfo->Size;\r
     }\r
 \r
     RemoveEntryList (&AllocInfoData->Link);\r
 \r
     if (BasicAction == MemoryProfileActionFreePages) {\r
-      if (AllocInfo->Buffer != (PHYSICAL_ADDRESS) (UINTN) Buffer) {\r
+      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
+          (UINTN)((PHYSICAL_ADDRESS)(UINTN)Buffer - AllocInfo->Buffer),\r
+          (VOID *)(UINTN)AllocInfo->Buffer,\r
           AllocInfoData->ActionString\r
           );\r
       }\r
-      if (AllocInfo->Buffer + AllocInfo->Size != ((PHYSICAL_ADDRESS) (UINTN) Buffer + Size)) {\r
+\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
+          (UINTN)((AllocInfo->Buffer + AllocInfo->Size) - ((PHYSICAL_ADDRESS)(UINTN)Buffer + Size)),\r
+          (VOID *)((UINTN)Buffer + Size),\r
           AllocInfoData->ActionString\r
           );\r
       }\r
@@ -1457,9 +1482,9 @@ SmmCoreUpdateProfile (
   IN CHAR8                  *ActionString OPTIONAL\r
   )\r
 {\r
-  EFI_STATUS                    Status;\r
-  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
-  MEMORY_PROFILE_ACTION         BasicAction;\r
+  EFI_STATUS                   Status;\r
+  MEMORY_PROFILE_CONTEXT_DATA  *ContextData;\r
+  MEMORY_PROFILE_ACTION        BasicAction;\r
 \r
   if (!IS_SMRAM_PROFILE_ENABLED) {\r
     return EFI_UNSUPPORTED;\r
@@ -1547,21 +1572,21 @@ SmramProfileGetDataSize (
   VOID\r
   )\r
 {\r
-  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
-  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
-  MEMORY_PROFILE_ALLOC_INFO_DATA    *AllocInfoData;\r
-  LIST_ENTRY                        *DriverInfoList;\r
-  LIST_ENTRY                        *DriverLink;\r
-  LIST_ENTRY                        *AllocInfoList;\r
-  LIST_ENTRY                        *AllocLink;\r
-  UINTN                             TotalSize;\r
-  LIST_ENTRY                        *Node;\r
-  LIST_ENTRY                        *FreePageList;\r
-  LIST_ENTRY                        *FreePoolList;\r
-  FREE_POOL_HEADER                  *Pool;\r
-  UINTN                             PoolListIndex;\r
-  UINTN                             Index;\r
-  UINTN                             SmmPoolTypeIndex;\r
+  MEMORY_PROFILE_CONTEXT_DATA      *ContextData;\r
+  MEMORY_PROFILE_DRIVER_INFO_DATA  *DriverInfoData;\r
+  MEMORY_PROFILE_ALLOC_INFO_DATA   *AllocInfoData;\r
+  LIST_ENTRY                       *DriverInfoList;\r
+  LIST_ENTRY                       *DriverLink;\r
+  LIST_ENTRY                       *AllocInfoList;\r
+  LIST_ENTRY                       *AllocLink;\r
+  UINTN                            TotalSize;\r
+  LIST_ENTRY                       *Node;\r
+  LIST_ENTRY                       *FreePageList;\r
+  LIST_ENTRY                       *FreePoolList;\r
+  FREE_POOL_HEADER                 *Pool;\r
+  UINTN                            PoolListIndex;\r
+  UINTN                            Index;\r
+  UINTN                            SmmPoolTypeIndex;\r
 \r
   ContextData = GetSmramProfileContext ();\r
   if (ContextData == NULL) {\r
@@ -1573,19 +1598,21 @@ SmramProfileGetDataSize (
   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
+                       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
+         AllocLink = AllocLink->ForwardLink)\r
+    {\r
       AllocInfoData = CR (\r
                         AllocLink,\r
                         MEMORY_PROFILE_ALLOC_INFO_DATA,\r
@@ -1596,20 +1623,22 @@ SmramProfileGetDataSize (
     }\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
+\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
+           Node = Node->BackLink)\r
+      {\r
         Pool = BASE_CR (Node, FREE_POOL_HEADER, Link);\r
         if (Pool->Header.Available) {\r
           Index++;\r
@@ -1636,9 +1665,9 @@ SmramProfileGetDataSize (
 **/\r
 VOID\r
 SmramProfileCopyData (\r
-  OUT VOID      *ProfileBuffer,\r
-  IN OUT UINT64 *ProfileSize,\r
-  IN OUT UINT64 *ProfileOffset\r
+  OUT VOID       *ProfileBuffer,\r
+  IN OUT UINT64  *ProfileSize,\r
+  IN OUT UINT64  *ProfileOffset\r
   )\r
 {\r
   MEMORY_PROFILE_CONTEXT           *Context;\r
@@ -1647,50 +1676,52 @@ SmramProfileCopyData (
   MEMORY_PROFILE_CONTEXT_DATA      *ContextData;\r
   MEMORY_PROFILE_DRIVER_INFO_DATA  *DriverInfoData;\r
   MEMORY_PROFILE_ALLOC_INFO_DATA   *AllocInfoData;\r
-  LIST_ENTRY                      *DriverInfoList;\r
-  LIST_ENTRY                      *DriverLink;\r
-  LIST_ENTRY                      *AllocInfoList;\r
-  LIST_ENTRY                      *AllocLink;\r
-  LIST_ENTRY                      *Node;\r
-  FREE_PAGE_LIST                  *Pages;\r
-  LIST_ENTRY                      *FreePageList;\r
-  LIST_ENTRY                      *FreePoolList;\r
-  FREE_POOL_HEADER                *Pool;\r
-  UINTN                           PoolListIndex;\r
-  UINT32                          Index;\r
-  MEMORY_PROFILE_FREE_MEMORY      *FreeMemory;\r
-  MEMORY_PROFILE_MEMORY_RANGE     *MemoryRange;\r
-  MEMORY_PROFILE_DESCRIPTOR       *MemoryProfileDescriptor;\r
-  UINT64                          Offset;\r
-  UINT64                          RemainingSize;\r
-  UINTN                           PdbSize;\r
-  UINTN                           ActionStringSize;\r
-  UINTN                           SmmPoolTypeIndex;\r
+  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
   RemainingSize = *ProfileSize;\r
-  Offset = 0;\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
+      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
@@ -1703,20 +1734,23 @@ SmramProfileCopyData (
         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
+          CopyMem ((VOID *)((UINTN)DriverInfo + DriverInfo->PdbStringOffset), DriverInfoData->PdbString, PdbSize);\r
         }\r
+\r
         RemainingSize -= DriverInfo->Header.Length;\r
-        ProfileBuffer = (UINT8 *) ProfileBuffer + 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
@@ -1729,36 +1763,40 @@ SmramProfileCopyData (
           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
+            CopyMem ((VOID *)((UINTN)AllocInfo + AllocInfo->ActionStringOffset), AllocInfoData->ActionString, ActionStringSize);\r
           }\r
+\r
           RemainingSize -= AllocInfo->Header.Length;\r
-          ProfileBuffer = (UINT8 *) ProfileBuffer + AllocInfo->Header.Length;\r
+          ProfileBuffer  = (UINT8 *)ProfileBuffer + AllocInfo->Header.Length;\r
         } else {\r
           goto Done;\r
         }\r
       }\r
+\r
       Offset += AllocInfoData->AllocInfo.Header.Length;\r
     }\r
   }\r
 \r
-\r
   if (*ProfileOffset < (Offset + sizeof (MEMORY_PROFILE_FREE_MEMORY))) {\r
     if (RemainingSize >= sizeof (MEMORY_PROFILE_FREE_MEMORY)) {\r
       FreeMemory = ProfileBuffer;\r
       CopyMem (FreeMemory, &mSmramFreeMemory, sizeof (MEMORY_PROFILE_FREE_MEMORY));\r
-      Index = 0;\r
+      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
+\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
+               Node = Node->BackLink)\r
+          {\r
             Pool = BASE_CR (Node, FREE_POOL_HEADER, Link);\r
             if (Pool->Header.Available) {\r
               Index++;\r
@@ -1766,60 +1804,67 @@ SmramProfileCopyData (
           }\r
         }\r
       }\r
+\r
       FreeMemory->FreeMemoryEntryCount = Index;\r
 \r
       RemainingSize -= sizeof (MEMORY_PROFILE_FREE_MEMORY);\r
-      ProfileBuffer = (UINT8 *) ProfileBuffer + sizeof (MEMORY_PROFILE_FREE_MEMORY);\r
+      ProfileBuffer  = (UINT8 *)ProfileBuffer + sizeof (MEMORY_PROFILE_FREE_MEMORY);\r
     } else {\r
       goto Done;\r
     }\r
   }\r
-  Offset += sizeof (MEMORY_PROFILE_FREE_MEMORY);\r
+\r
+  Offset      += sizeof (MEMORY_PROFILE_FREE_MEMORY);\r
   FreePageList = &mSmmMemoryMap;\r
   for (Node = FreePageList->BackLink;\r
        Node != FreePageList;\r
-       Node = Node->BackLink) {\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
+        Pages                                     = BASE_CR (Node, FREE_PAGE_LIST, Link);\r
+        MemoryProfileDescriptor                   = ProfileBuffer;\r
         MemoryProfileDescriptor->Header.Signature = MEMORY_PROFILE_DESCRIPTOR_SIGNATURE;\r
-        MemoryProfileDescriptor->Header.Length = sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
-        MemoryProfileDescriptor->Header.Revision = MEMORY_PROFILE_DESCRIPTOR_REVISION;\r
-        MemoryProfileDescriptor->Address = (PHYSICAL_ADDRESS) (UINTN) Pages;\r
-        MemoryProfileDescriptor->Size = EFI_PAGES_TO_SIZE (Pages->NumberOfPages);\r
+        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
+        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
+           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                   = 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->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
+              ProfileBuffer  = (UINT8 *)ProfileBuffer + sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
             } else {\r
               goto Done;\r
             }\r
           }\r
+\r
           Offset += sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
         }\r
       }\r
@@ -1828,35 +1873,37 @@ SmramProfileCopyData (
 \r
   if (*ProfileOffset < (Offset + sizeof (MEMORY_PROFILE_MEMORY_RANGE))) {\r
     if (RemainingSize >= sizeof (MEMORY_PROFILE_MEMORY_RANGE)) {\r
-      MemoryRange = ProfileBuffer;\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
+      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
+      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
     if (*ProfileOffset < (Offset + sizeof (MEMORY_PROFILE_DESCRIPTOR))) {\r
       if (RemainingSize >= sizeof (MEMORY_PROFILE_DESCRIPTOR)) {\r
-        MemoryProfileDescriptor = ProfileBuffer;\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
+        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
+        ProfileBuffer  = (UINT8 *)ProfileBuffer + sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
       } else {\r
         goto Done;\r
       }\r
     }\r
+\r
     Offset += sizeof (MEMORY_PROFILE_DESCRIPTOR);\r
   }\r
 \r
@@ -1890,26 +1937,26 @@ EFIAPI
 SmramProfileProtocolGetData (\r
   IN     EDKII_SMM_MEMORY_PROFILE_PROTOCOL  *This,\r
   IN OUT UINT64                             *ProfileSize,\r
-     OUT VOID                               *ProfileBuffer\r
+  OUT VOID                                  *ProfileBuffer\r
   )\r
 {\r
-  UINT64                                Size;\r
-  UINT64                                Offset;\r
-  MEMORY_PROFILE_CONTEXT_DATA           *ContextData;\r
-  BOOLEAN                               SmramProfileGettingStatus;\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
+  SmramProfileGettingStatus  = mSmramProfileGettingStatus;\r
   mSmramProfileGettingStatus = TRUE;\r
 \r
   Size = SmramProfileGetDataSize ();\r
 \r
   if (*ProfileSize < Size) {\r
-    *ProfileSize = Size;\r
+    *ProfileSize               = Size;\r
     mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
     return EFI_BUFFER_TOO_SMALL;\r
   }\r
@@ -1947,21 +1994,22 @@ SmramProfileProtocolRegisterImage (
   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
+  EFI_STATUS            Status;\r
+  EFI_SMM_DRIVER_ENTRY  DriverEntry;\r
+  VOID                  *EntryPointInImage;\r
+  EFI_GUID              *Name;\r
 \r
   ZeroMem (&DriverEntry, sizeof (DriverEntry));\r
   Name = GetFileNameFromFilePath (FilePath);\r
   if (Name != NULL) {\r
     CopyMem (&DriverEntry.FileName, Name, sizeof (EFI_GUID));\r
   }\r
-  DriverEntry.ImageBuffer = ImageBase;\r
-  DriverEntry.NumberOfPage = EFI_SIZE_TO_PAGES ((UINTN) ImageSize);\r
-  Status = InternalPeCoffGetEntryPoint ((VOID *) (UINTN) DriverEntry.ImageBuffer, &EntryPointInImage);\r
+\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
+  DriverEntry.ImageEntryPoint = (PHYSICAL_ADDRESS)(UINTN)EntryPointInImage;\r
 \r
   return RegisterSmramProfileImage (&DriverEntry, FALSE);\r
 }\r
@@ -1989,21 +2037,22 @@ SmramProfileProtocolUnregisterImage (
   IN UINT64                             ImageSize\r
   )\r
 {\r
-  EFI_STATUS                        Status;\r
-  EFI_SMM_DRIVER_ENTRY              DriverEntry;\r
-  VOID                              *EntryPointInImage;\r
-  EFI_GUID                          *Name;\r
+  EFI_STATUS            Status;\r
+  EFI_SMM_DRIVER_ENTRY  DriverEntry;\r
+  VOID                  *EntryPointInImage;\r
+  EFI_GUID              *Name;\r
 \r
   ZeroMem (&DriverEntry, sizeof (DriverEntry));\r
   Name = GetFileNameFromFilePath (FilePath);\r
   if (Name != NULL) {\r
     CopyMem (&DriverEntry.FileName, Name, sizeof (EFI_GUID));\r
   }\r
-  DriverEntry.ImageBuffer = ImageBase;\r
-  DriverEntry.NumberOfPage = EFI_SIZE_TO_PAGES ((UINTN) ImageSize);\r
-  Status = InternalPeCoffGetEntryPoint ((VOID *) (UINTN) DriverEntry.ImageBuffer, &EntryPointInImage);\r
+\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
+  DriverEntry.ImageEntryPoint = (PHYSICAL_ADDRESS)(UINTN)EntryPointInImage;\r
 \r
   return UnregisterSmramProfileImage (&DriverEntry, FALSE);\r
 }\r
@@ -2026,7 +2075,7 @@ SmramProfileProtocolGetRecordingState (
   OUT BOOLEAN                           *RecordingState\r
   )\r
 {\r
-  MEMORY_PROFILE_CONTEXT_DATA           *ContextData;\r
+  MEMORY_PROFILE_CONTEXT_DATA  *ContextData;\r
 \r
   ContextData = GetSmramProfileContext ();\r
   if (ContextData == NULL) {\r
@@ -2036,6 +2085,7 @@ SmramProfileProtocolGetRecordingState (
   if (RecordingState == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
+\r
   *RecordingState = mSmramProfileRecordingEnable;\r
   return EFI_SUCCESS;\r
 }\r
@@ -2057,7 +2107,7 @@ SmramProfileProtocolSetRecordingState (
   IN BOOLEAN                            RecordingState\r
   )\r
 {\r
-  MEMORY_PROFILE_CONTEXT_DATA           *ContextData;\r
+  MEMORY_PROFILE_CONTEXT_DATA  *ContextData;\r
 \r
   ContextData = GetSmramProfileContext ();\r
   if (ContextData == NULL) {\r
@@ -2114,21 +2164,21 @@ SmramProfileProtocolRecord (
 **/\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                       SmramProfileGettingStatus;\r
+  MEMORY_PROFILE_CONTEXT_DATA  *ContextData;\r
+  BOOLEAN                      SmramProfileGettingStatus;\r
 \r
   ContextData = GetSmramProfileContext ();\r
   if (ContextData == NULL) {\r
-    return ;\r
+    return;\r
   }\r
 \r
-  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+  SmramProfileGettingStatus  = mSmramProfileGettingStatus;\r
   mSmramProfileGettingStatus = TRUE;\r
 \r
-  SmramProfileParameterGetInfo->ProfileSize = SmramProfileGetDataSize();\r
+  SmramProfileParameterGetInfo->ProfileSize         = SmramProfileGetDataSize ();\r
   SmramProfileParameterGetInfo->Header.ReturnStatus = 0;\r
 \r
   mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
@@ -2142,7 +2192,7 @@ SmramProfileHandlerGetInfo (
 **/\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
@@ -2153,36 +2203,35 @@ SmramProfileHandlerGetData (
 \r
   ContextData = GetSmramProfileContext ();\r
   if (ContextData == NULL) {\r
-    return ;\r
+    return;\r
   }\r
 \r
-  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+  SmramProfileGettingStatus  = mSmramProfileGettingStatus;\r
   mSmramProfileGettingStatus = TRUE;\r
 \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
+  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
+    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
   ProfileOffset = 0;\r
-  SmramProfileCopyData ((VOID *) (UINTN) SmramProfileGetData.ProfileBuffer, &ProfileSize, &ProfileOffset);\r
-  SmramProfileParameterGetData->ProfileSize = ProfileSize;\r
+  SmramProfileCopyData ((VOID *)(UINTN)SmramProfileGetData.ProfileBuffer, &ProfileSize, &ProfileOffset);\r
+  SmramProfileParameterGetData->ProfileSize         = ProfileSize;\r
   SmramProfileParameterGetData->Header.ReturnStatus = 0;\r
 \r
 Done:\r
@@ -2197,34 +2246,33 @@ Done:
 **/\r
 VOID\r
 SmramProfileHandlerGetDataByOffset (\r
-  IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET     *SmramProfileParameterGetDataByOffset\r
+  IN SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET  *SmramProfileParameterGetDataByOffset\r
   )\r
 {\r
-  SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET    SmramProfileGetDataByOffset;\r
-  MEMORY_PROFILE_CONTEXT_DATA                           *ContextData;\r
-  BOOLEAN                                               SmramProfileGettingStatus;\r
+  SMRAM_PROFILE_PARAMETER_GET_PROFILE_DATA_BY_OFFSET  SmramProfileGetDataByOffset;\r
+  MEMORY_PROFILE_CONTEXT_DATA                         *ContextData;\r
+  BOOLEAN                                             SmramProfileGettingStatus;\r
 \r
   ContextData = GetSmramProfileContext ();\r
   if (ContextData == NULL) {\r
-    return ;\r
+    return;\r
   }\r
 \r
-  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+  SmramProfileGettingStatus  = mSmramProfileGettingStatus;\r
   mSmramProfileGettingStatus = TRUE;\r
 \r
-\r
   CopyMem (&SmramProfileGetDataByOffset, SmramProfileParameterGetDataByOffset, sizeof (SmramProfileGetDataByOffset));\r
 \r
   //\r
   // Sanity check\r
   //\r
-  if (!SmmIsBufferOutsideSmmValid ((UINTN) SmramProfileGetDataByOffset.ProfileBuffer, (UINTN) SmramProfileGetDataByOffset.ProfileSize)) {\r
+  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
+    SmramProfileParameterGetDataByOffset->Header.ReturnStatus = (UINT64)(INT64)(INTN)EFI_ACCESS_DENIED;\r
     goto Done;\r
   }\r
 \r
-  SmramProfileCopyData ((VOID *) (UINTN) SmramProfileGetDataByOffset.ProfileBuffer, &SmramProfileGetDataByOffset.ProfileSize, &SmramProfileGetDataByOffset.ProfileOffset);\r
+  SmramProfileCopyData ((VOID *)(UINTN)SmramProfileGetDataByOffset.ProfileBuffer, &SmramProfileGetDataByOffset.ProfileSize, &SmramProfileGetDataByOffset.ProfileOffset);\r
   CopyMem (SmramProfileParameterGetDataByOffset, &SmramProfileGetDataByOffset, sizeof (SmramProfileGetDataByOffset));\r
   SmramProfileParameterGetDataByOffset->Header.ReturnStatus = 0;\r
 \r
@@ -2266,7 +2314,7 @@ SmramProfileHandler (
   //\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
@@ -2282,72 +2330,77 @@ SmramProfileHandler (
     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 ((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
-    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
-    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
-    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
-    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
-    ParameterRecordingState = (SMRAM_PROFILE_PARAMETER_RECORDING_STATE *) (UINTN) CommBuffer;\r
-    mSmramProfileRecordingEnable = ParameterRecordingState->RecordingState;\r
-    ParameterRecordingState->Header.ReturnStatus = 0;\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
-  //\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
+      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
+    default:\r
+      break;\r
   }\r
 \r
   DEBUG ((DEBUG_ERROR, "SmramProfileHandler Exit\n"));\r
@@ -2364,8 +2417,8 @@ RegisterSmramProfileHandler (
   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
@@ -2390,16 +2443,16 @@ DumpSmramRange (
   VOID\r
   )\r
 {\r
-  UINTN                         Index;\r
-  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
-  BOOLEAN                       SmramProfileGettingStatus;\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
-  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+  SmramProfileGettingStatus  = mSmramProfileGettingStatus;\r
   mSmramProfileGettingStatus = TRUE;\r
 \r
   DEBUG ((DEBUG_INFO, "FullSmramRange address - 0x%08x\n", mFullSmramRanges));\r
@@ -2429,19 +2482,19 @@ DumpFreePagesList (
   VOID\r
   )\r
 {\r
-  LIST_ENTRY                    *FreePageList;\r
-  LIST_ENTRY                    *Node;\r
-  FREE_PAGE_LIST                *Pages;\r
-  UINTN                         Index;\r
-  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
-  BOOLEAN                       SmramProfileGettingStatus;\r
+  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
-  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+  SmramProfileGettingStatus  = mSmramProfileGettingStatus;\r
   mSmramProfileGettingStatus = TRUE;\r
 \r
   DEBUG ((DEBUG_INFO, "======= SmramProfile begin =======\n"));\r
@@ -2450,10 +2503,11 @@ DumpFreePagesList (
   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 ((DEBUG_INFO, "  Index - 0x%x\n", Index));\r
-    DEBUG ((DEBUG_INFO, "    PhysicalStart - 0x%016lx\n", (PHYSICAL_ADDRESS) (UINTN) Pages));\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
@@ -2471,21 +2525,21 @@ DumpFreePoolList (
   VOID\r
   )\r
 {\r
-  LIST_ENTRY                    *FreePoolList;\r
-  LIST_ENTRY                    *Node;\r
-  FREE_POOL_HEADER              *Pool;\r
-  UINTN                         Index;\r
-  UINTN                         PoolListIndex;\r
-  MEMORY_PROFILE_CONTEXT_DATA   *ContextData;\r
-  BOOLEAN                       SmramProfileGettingStatus;\r
-  UINTN                         SmmPoolTypeIndex;\r
+  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
-  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+  SmramProfileGettingStatus  = mSmramProfileGettingStatus;\r
   mSmramProfileGettingStatus = TRUE;\r
 \r
   DEBUG ((DEBUG_INFO, "======= SmramProfile begin =======\n"));\r
@@ -2496,10 +2550,11 @@ DumpFreePoolList (
       FreePoolList = &mSmmPoolLists[SmmPoolTypeIndex][PoolListIndex];\r
       for (Node = FreePoolList->BackLink, Index = 0;\r
            Node != FreePoolList;\r
-           Node = Node->BackLink, Index++) {\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, "    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
@@ -2511,7 +2566,7 @@ DumpFreePoolList (
   mSmramProfileGettingStatus = SmramProfileGettingStatus;\r
 }\r
 \r
-GLOBAL_REMOVE_IF_UNREFERENCED CHAR8 *mSmmActionString[] = {\r
+GLOBAL_REMOVE_IF_UNREFERENCED CHAR8  *mSmmActionString[] = {\r
   "SmmUnknown",\r
   "gSmst->SmmAllocatePages",\r
   "gSmst->SmmFreePages",\r
@@ -2520,42 +2575,42 @@ GLOBAL_REMOVE_IF_UNREFERENCED CHAR8 *mSmmActionString[] = {
 };\r
 \r
 typedef struct {\r
-  MEMORY_PROFILE_ACTION  Action;\r
-  CHAR8                 *String;\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
+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
+  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
+GLOBAL_REMOVE_IF_UNREFERENCED PROFILE_MEMORY_TYPE_STRING  mMemoryTypeString[] = {\r
+  { EfiRuntimeServicesCode, "EfiRuntimeServicesCode" },\r
+  { EfiRuntimeServicesData, "EfiRuntimeServicesData" }\r
 };\r
 \r
 /**\r
@@ -2568,10 +2623,11 @@ GLOBAL_REMOVE_IF_UNREFERENCED PROFILE_MEMORY_TYPE_STRING mMemoryTypeString[] = {
 **/\r
 CHAR8 *\r
 ProfileMemoryTypeToStr (\r
-  IN EFI_MEMORY_TYPE    MemoryType\r
+  IN EFI_MEMORY_TYPE  MemoryType\r
   )\r
 {\r
-  UINTN     Index;\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
@@ -2594,16 +2650,17 @@ ProfileActionToStr (
   IN MEMORY_PROFILE_ACTION  Action\r
   )\r
 {\r
-  UINTN     Index;\r
-  UINTN     ActionStringCount;\r
-  CHAR8     **ActionString;\r
+  UINTN  Index;\r
+  UINTN  ActionStringCount;\r
+  CHAR8  **ActionString;\r
 \r
-  ActionString = mSmmActionString;\r
+  ActionString      = mSmmActionString;\r
   ActionStringCount = ARRAY_SIZE (mSmmActionString);\r
 \r
-  if ((UINTN) (UINT32) Action < ActionStringCount) {\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
@@ -2622,27 +2679,27 @@ DumpSmramProfile (
   VOID\r
   )\r
 {\r
-  MEMORY_PROFILE_CONTEXT            *Context;\r
-  MEMORY_PROFILE_DRIVER_INFO        *DriverInfo;\r
-  MEMORY_PROFILE_ALLOC_INFO         *AllocInfo;\r
-  MEMORY_PROFILE_CONTEXT_DATA       *ContextData;\r
-  MEMORY_PROFILE_DRIVER_INFO_DATA   *DriverInfoData;\r
-  MEMORY_PROFILE_ALLOC_INFO_DATA    *AllocInfoData;\r
-  LIST_ENTRY                        *SmramDriverInfoList;\r
-  UINTN                             DriverIndex;\r
-  LIST_ENTRY                        *DriverLink;\r
-  LIST_ENTRY                        *AllocInfoList;\r
-  UINTN                             AllocIndex;\r
-  LIST_ENTRY                        *AllocLink;\r
-  BOOLEAN                           SmramProfileGettingStatus;\r
-  UINTN                             TypeIndex;\r
+  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
-  SmramProfileGettingStatus = mSmramProfileGettingStatus;\r
+  SmramProfileGettingStatus  = mSmramProfileGettingStatus;\r
   mSmramProfileGettingStatus = TRUE;\r
 \r
   Context = &ContextData->Context;\r
@@ -2653,11 +2710,13 @@ DumpSmramProfile (
   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
+        (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
+\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
@@ -2665,13 +2724,14 @@ DumpSmramProfile (
   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 ((DEBUG_INFO, "  MEMORY_PROFILE_DRIVER_INFO (0x%x)\n", DriverIndex));\r
     DEBUG ((DEBUG_INFO, "    FileName            - %g\n", &DriverInfo->FileName));\r
@@ -2684,23 +2744,26 @@ DumpSmramProfile (
     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
+          (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
+\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 ((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
@@ -2714,6 +2777,7 @@ DumpSmramProfile (
       } 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
@@ -2734,12 +2798,13 @@ DumpSmramInfo (
   VOID\r
   )\r
 {\r
-    DEBUG_CODE_BEGIN ();\r
-    if (IS_SMRAM_PROFILE_ENABLED) {\r
-      DumpSmramProfile ();\r
-      DumpFreePagesList ();\r
-      DumpFreePoolList ();\r
-      DumpSmramRange ();\r
-    }\r
-    DEBUG_CODE_END ();\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