]> git.proxmox.com Git - mirror_edk2.git/blobdiff - PerformancePkg/Dp_App/Dp.c
PerformancePkg/Dp_App: Add check to avoid NULL pointer deference
[mirror_edk2.git] / PerformancePkg / Dp_App / Dp.c
index f5d0bda9f890f3371d471fef88410d3368fcf5ea..4cdc39edb42e75ccf0221839550df5fa9fab0f2b 100644 (file)
@@ -1,35 +1,37 @@
 /** @file\r
-  * Shell application for Displaying Performance Metrics.\r
-  *\r
-  * The Dp application reads performance data and presents it in several\r
-  * different formats depending upon the needs of the user.  Both\r
-  * Trace and Measured Profiling information is processed and presented.\r
-  *\r
-  * Dp uses the "PerformanceLib" to read the measurement records.\r
-  * The "TimerLib" provides information about the timer, such as frequency,\r
-  * beginning, and ending counter values.\r
-  * Measurement records contain identifying information (Handle, Token, Module)\r
-  * and start and end time values.\r
-  * Dp uses this information to group records in different ways.  It also uses\r
-  * timer information to calculate elapsed time for each measurement.\r
-  *\r
-  * Copyright (c) 2009-2010, Intel Corporation. All rights reserved.<BR>\r
-  * This program and the accompanying materials\r
-  * are licensed and made available under the terms and conditions of the BSD License\r
-  * which accompanies this distribution.  The full text of the license may be found at\r
-  * http://opensource.org/licenses/bsd-license.php\r
-  *\r
-  * THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-  * WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
+  Shell application for Displaying Performance Metrics.\r
+\r
+  The Dp application reads performance data and presents it in several\r
+  different formats depending upon the needs of the user.  Both\r
+  Trace and Measured Profiling information is processed and presented.\r
+\r
+  Dp uses the "PerformanceLib" to read the measurement records.\r
+  The "TimerLib" provides information about the timer, such as frequency,\r
+  beginning, and ending counter values.\r
+  Measurement records contain identifying information (Handle, Token, Module)\r
+  and start and end time values.\r
+  Dp uses this information to group records in different ways.  It also uses\r
+  timer information to calculate elapsed time for each measurement.\r
\r
+  Copyright (c) 2009 - 2017, Intel Corporation. All rights reserved.<BR>\r
+  (C) Copyright 2015-2016 Hewlett Packard Enterprise Development LP<BR>\r
+  This program and the accompanying materials\r
+  are licensed and made available under the terms and conditions of the BSD License\r
+  which accompanies this distribution.  The full text of the license may be found at\r
+  http://opensource.org/licenses/bsd-license.php\r
\r
+  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
+  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
 **/\r
 \r
 #include <Library/UefiApplicationEntryPoint.h>\r
+#include <Library/UefiBootServicesTableLib.h>\r
 #include <Library/ShellLib.h>\r
 #include <Library/BaseLib.h>\r
 #include <Library/MemoryAllocationLib.h>\r
 #include <Library/DebugLib.h>\r
-#include <Library/TimerLib.h>\r
 #include <Library/UefiLib.h>\r
+#include <Library/UefiHiiServicesLib.h>\r
 #include <Library/HiiLib.h>\r
 #include <Library/PcdLib.h>\r
 \r
 #include "Literals.h"\r
 #include "DpInternal.h"\r
 \r
+//\r
+// String token ID of help message text.\r
+// Shell supports to find help message in the resource section of an application image if\r
+// .MAN file is not found. This global variable is added to make build tool recognizes\r
+// that the help string is consumed by user and then build tool will add the string into\r
+// the resource section. Thus the application can use '-?' option to show help message in\r
+// Shell.\r
+//\r
+GLOBAL_REMOVE_IF_UNREFERENCED EFI_STRING_ID mDpStrEngHelpTokenId = STRING_TOKEN (STR_DP_HELP_INFORMATION);\r
+\r
 //\r
 /// Module-Global Variables\r
-/// @{\r
+///@{\r
 EFI_HII_HANDLE   gHiiHandle;\r
+SHELL_PARAM_ITEM *DpParamList       = NULL;\r
 CHAR16           *mPrintTokenBuffer = NULL;\r
-CHAR16           mGaugeString[DXE_PERFORMANCE_STRING_SIZE];\r
-CHAR16           mUnicodeToken[PERF_TOKEN_LENGTH + 1];\r
+CHAR16           mGaugeString[DP_GAUGE_STRING_LENGTH + 1];\r
+CHAR16           mUnicodeToken[DXE_PERFORMANCE_STRING_SIZE];\r
 UINT64           mInterestThreshold;\r
+BOOLEAN          mShowId = FALSE;\r
 \r
 PERF_SUMMARY_DATA SummaryData = { 0 };    ///< Create the SummaryData structure and init. to ZERO.\r
 \r
@@ -65,27 +79,56 @@ PERF_CUM_DATA CumData[] = {
 /// Number of items for which we are gathering cumulative statistics.\r
 UINT32 const      NumCum = sizeof(CumData) / sizeof(PERF_CUM_DATA);\r
 \r
-SHELL_PARAM_ITEM  DpParamList[] = {\r
-  {STR_DP_OPTION_QH, TypeFlag},   // -?   Help\r
-  {STR_DP_OPTION_LH, TypeFlag},   // -h   Help\r
-  {STR_DP_OPTION_UH, TypeFlag},   // -H   Help\r
-  {STR_DP_OPTION_LV, TypeFlag},   // -v   Verbose Mode\r
-  {STR_DP_OPTION_UA, TypeFlag},   // -A   All, Cooked\r
-  {STR_DP_OPTION_UR, TypeFlag},   // -R   RAW All\r
-  {STR_DP_OPTION_LS, TypeFlag},   // -s   Summary\r
+PARAM_ITEM_LIST  ParamList[] = {\r
+  {STRING_TOKEN (STR_DP_OPTION_QH), TypeFlag},   // -?   Help\r
+  {STRING_TOKEN (STR_DP_OPTION_LH), TypeFlag},   // -h   Help\r
+  {STRING_TOKEN (STR_DP_OPTION_UH), TypeFlag},   // -H   Help\r
+  {STRING_TOKEN (STR_DP_OPTION_LV), TypeFlag},   // -v   Verbose Mode\r
+  {STRING_TOKEN (STR_DP_OPTION_UA), TypeFlag},   // -A   All, Cooked\r
+  {STRING_TOKEN (STR_DP_OPTION_UR), TypeFlag},   // -R   RAW All\r
+  {STRING_TOKEN (STR_DP_OPTION_LS), TypeFlag},   // -s   Summary\r
 #if PROFILING_IMPLEMENTED\r
-  {STR_DP_OPTION_UP, TypeFlag},   // -P   Dump Profile Data\r
-  {STR_DP_OPTION_UT, TypeFlag},   // -T   Dump Trace Data\r
+  {STRING_TOKEN (STR_DP_OPTION_UP), TypeFlag},   // -P   Dump Profile Data\r
+  {STRING_TOKEN (STR_DP_OPTION_UT), TypeFlag},   // -T   Dump Trace Data\r
 #endif\r
-  {STR_DP_OPTION_LX, TypeFlag},   // -x   eXclude Cumulative Items\r
-  {STR_DP_OPTION_LN, TypeValue},  // -n # Number of records to display for A and R\r
-  {STR_DP_OPTION_LT, TypeValue},  // -t # Threshold of interest\r
-  {NULL, TypeMax}\r
+  {STRING_TOKEN (STR_DP_OPTION_LX), TypeFlag},   // -x   eXclude Cumulative Items\r
+  {STRING_TOKEN (STR_DP_OPTION_LI), TypeFlag},   // -i   Display Identifier\r
+  {STRING_TOKEN (STR_DP_OPTION_LC), TypeValue},  // -c   Display cumulative data.\r
+  {STRING_TOKEN (STR_DP_OPTION_LN), TypeValue},  // -n # Number of records to display for A and R\r
+  {STRING_TOKEN (STR_DP_OPTION_LT), TypeValue}   // -t # Threshold of interest\r
   };\r
 \r
-/// @}\r
+///@}\r
+\r
+/**\r
+  Transfer the param list value and get the command line parse.\r
+\r
+**/\r
+VOID\r
+InitialShellParamList( void )\r
+{\r
+  UINT32            ListIndex;\r
+  UINT32            ListLength;  \r
+\r
+  //\r
+  // Allocate one more for the end tag.\r
+  //\r
+  ListLength = ARRAY_SIZE (ParamList) + 1;  \r
+  DpParamList = AllocatePool (sizeof (SHELL_PARAM_ITEM) * ListLength);\r
+  ASSERT (DpParamList != NULL);\r
+  \r
+  for (ListIndex = 0; ListIndex < ListLength - 1; ListIndex ++)\r
+  { \r
+    DpParamList[ListIndex].Name = HiiGetString (gHiiHandle, ParamList[ListIndex].Token, NULL);      \r
+    DpParamList[ListIndex].Type = ParamList[ListIndex].Type;\r
+  }\r
+  DpParamList[ListIndex].Name = NULL;\r
+  DpParamList[ListIndex].Type = TypeMax;\r
+}\r
 \r
-/// Display Usage and Help information.\r
+/**\r
+   Display Usage and Help information.\r
+**/\r
 VOID\r
 ShowHelp( void )\r
 {\r
@@ -107,19 +150,24 @@ ShowHelp( void )
 #endif // PROFILING_IMPLEMENTED\r
   PrintToken (STRING_TOKEN (STR_DP_HELP_THRESHOLD));\r
   PrintToken (STRING_TOKEN (STR_DP_HELP_COUNT));\r
+  PrintToken (STRING_TOKEN (STR_DP_HELP_ID));\r
+  PrintToken (STRING_TOKEN (STR_DP_HELP_CUM_DATA));\r
   PrintToken (STRING_TOKEN (STR_DP_HELP_HELP));\r
   Print(L"\n");\r
 }\r
 \r
-/// Display the trailing Verbose information.\r
+/**\r
+   Display the trailing Verbose information.\r
+**/\r
 VOID\r
 DumpStatistics( void )\r
 {\r
   EFI_STRING                StringPtr;\r
-\r
-  StringPtr = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_SECTION_STATISTICS), NULL);\r
+  EFI_STRING                StringPtrUnknown;\r
+  StringPtr        = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_SECTION_STATISTICS), NULL);\r
+  StringPtrUnknown = HiiGetString (gHiiHandle, STRING_TOKEN (STR_ALIT_UNKNOWN), NULL);  \r
   PrintToken( STRING_TOKEN (STR_DP_SECTION_HEADER),\r
-              (StringPtr == NULL) ? ALit_UNKNOWN: StringPtr);\r
+              (StringPtr == NULL) ? StringPtrUnknown : StringPtr);\r
 \r
   PrintToken( STRING_TOKEN (STR_DP_STATS_NUMTRACE),       SummaryData.NumTrace);\r
   PrintToken( STRING_TOKEN (STR_DP_STATS_NUMINCOMPLETE),  SummaryData.NumIncomplete);\r
@@ -130,17 +178,20 @@ DumpStatistics( void )
 #if PROFILING_IMPLEMENTED\r
   PrintToken( STRING_TOKEN (STR_DP_STATS_NUMPROFILE),     SummaryData.NumProfile);\r
 #endif // PROFILING_IMPLEMENTED\r
+  FreePool (StringPtr);\r
+  FreePool (StringPtrUnknown);\r
 }\r
 \r
-/** Dump performance data.\r
-  *\r
-  * @param[in]  ImageHandle     The image handle.\r
-  * @param[in]  SystemTable     The system table.\r
-  *\r
-  * @retval EFI_SUCCESS            Command completed successfully.\r
-  * @retval EFI_INVALID_PARAMETER  Command usage error.\r
-  * @retval value                  Unknown error.\r
-  *\r
+/**\r
+  Dump performance data.\r
+  \r
+  @param[in]  ImageHandle     The image handle.\r
+  @param[in]  SystemTable     The system table.\r
+\r
+  @retval EFI_SUCCESS            Command completed successfully.\r
+  @retval EFI_INVALID_PARAMETER  Command usage error.\r
+  @retval EFI_ABORTED            The user aborts the operation.\r
+  @retval value                  Unknown error.\r
 **/\r
 EFI_STATUS\r
 EFIAPI\r
@@ -149,34 +200,105 @@ InitializeDp (
   IN EFI_SYSTEM_TABLE         *SystemTable\r
   )\r
 {\r
-  UINT64                    Freq;\r
-  UINT64                    Ticker;\r
+  PERFORMANCE_PROPERTY          *PerformanceProperty;\r
+  UINT32                        ListIndex;\r
 \r
-  LIST_ENTRY                *ParamPackage;\r
-  CONST CHAR16              *CmdLineArg;\r
-  EFI_STRING                StringPtr;\r
-  UINTN                     Number2Display;\r
+  LIST_ENTRY                    *ParamPackage;\r
+  CONST CHAR16                  *CmdLineArg;\r
+  EFI_STRING                    StringPtr;\r
+  UINTN                         Number2Display;\r
+\r
+  EFI_STATUS                    Status;\r
+  BOOLEAN                       SummaryMode;\r
+  BOOLEAN                       VerboseMode;\r
+  BOOLEAN                       AllMode;\r
+  BOOLEAN                       RawMode;\r
+  BOOLEAN                       TraceMode;\r
+  BOOLEAN                       ProfileMode;\r
+  BOOLEAN                       ExcludeMode;\r
+  BOOLEAN                       CumulativeMode;\r
+  CONST CHAR16                  *CustomCumulativeToken;\r
+  PERF_CUM_DATA                 *CustomCumulativeData;\r
+  UINTN                         NameSize;\r
+  EFI_HII_PACKAGE_LIST_HEADER   *PackageList;\r
 \r
-  EFI_STATUS                Status;\r
-  BOOLEAN                   SummaryMode     = FALSE;\r
-  BOOLEAN                   VerboseMode     = FALSE;\r
-  BOOLEAN                   AllMode         = FALSE;\r
-  BOOLEAN                   RawMode         = FALSE;\r
-  BOOLEAN                   TraceMode       = FALSE;\r
-  BOOLEAN                   ProfileMode     = FALSE;\r
-  BOOLEAN                   ExcludeMode     = FALSE;\r
+  EFI_STRING                StringDpOptionQh;\r
+  EFI_STRING                StringDpOptionLh;\r
+  EFI_STRING                StringDpOptionUh;\r
+  EFI_STRING                StringDpOptionLv;\r
+  EFI_STRING                StringDpOptionUs;\r
+  EFI_STRING                StringDpOptionLs;\r
+  EFI_STRING                StringDpOptionUa;\r
+  EFI_STRING                StringDpOptionUr;\r
+  EFI_STRING                StringDpOptionUt;\r
+  EFI_STRING                StringDpOptionUp;\r
+  EFI_STRING                StringDpOptionLx;\r
+  EFI_STRING                StringDpOptionLn;\r
+  EFI_STRING                StringDpOptionLt;\r
+  EFI_STRING                StringDpOptionLi;\r
+  EFI_STRING                StringDpOptionLc;\r
+  \r
+  SummaryMode     = FALSE;\r
+  VerboseMode     = FALSE;\r
+  AllMode         = FALSE;\r
+  RawMode         = FALSE;\r
+  TraceMode       = FALSE;\r
+  ProfileMode     = FALSE;\r
+  ExcludeMode     = FALSE;\r
+  CumulativeMode = FALSE;\r
+  CustomCumulativeData = NULL;\r
 \r
+  StringDpOptionQh = NULL;\r
+  StringDpOptionLh = NULL;\r
+  StringDpOptionUh = NULL;\r
+  StringDpOptionLv = NULL;\r
+  StringDpOptionUs = NULL;\r
+  StringDpOptionLs = NULL;\r
+  StringDpOptionUa = NULL;\r
+  StringDpOptionUr = NULL;\r
+  StringDpOptionUt = NULL;\r
+  StringDpOptionUp = NULL;\r
+  StringDpOptionLx = NULL;\r
+  StringDpOptionLn = NULL;\r
+  StringDpOptionLt = NULL;\r
+  StringDpOptionLi = NULL;\r
+  StringDpOptionLc = NULL;\r
+  StringPtr        = NULL;\r
 \r
-  // Get DP's entry time as soon as possible.\r
-  // This is used as the Shell-Phase end time.\r
   //\r
-  Ticker  = GetPerformanceCounter ();\r
+  // Retrieve HII package list from ImageHandle\r
+  //\r
+  Status = gBS->OpenProtocol (\r
+                  ImageHandle,\r
+                  &gEfiHiiPackageListProtocolGuid,\r
+                  (VOID **) &PackageList,\r
+                  ImageHandle,\r
+                  NULL,\r
+                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
 \r
-  // Register our string package with HII and return the handle to it.\r
   //\r
-  gHiiHandle = HiiAddPackages (&gEfiCallerIdGuid, ImageHandle, DPStrings, NULL);\r
+  // Publish HII package list to HII Database.\r
+  //\r
+  Status = gHiiDatabase->NewPackageList (\r
+                          gHiiDatabase,\r
+                          PackageList,\r
+                          NULL,\r
+                          &gHiiHandle\r
+                          );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
   ASSERT (gHiiHandle != NULL);\r
+  \r
 \r
+  // Initial the command list\r
+  //\r
+  InitialShellParamList ();\r
+  \r
 /****************************************************************************\r
 ****            Process Command Line arguments                           ****\r
 ****************************************************************************/\r
@@ -187,27 +309,47 @@ InitializeDp (
     ShowHelp();\r
   }\r
   else {\r
-    if (ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_QH)  ||\r
-        ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_LH)  ||\r
-        ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_UH))\r
+    StringDpOptionQh = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_QH), NULL);\r
+    StringDpOptionLh = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_LH), NULL);\r
+    StringDpOptionUh = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_UH), NULL);\r
+    \r
+    if (ShellCommandLineGetFlag (ParamPackage, StringDpOptionQh)  ||\r
+        ShellCommandLineGetFlag (ParamPackage, StringDpOptionLh)  ||\r
+        ShellCommandLineGetFlag (ParamPackage, StringDpOptionUh))\r
     {\r
       ShowHelp();\r
     }\r
     else {\r
+      StringDpOptionLv = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_LV), NULL);\r
+      StringDpOptionUs = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_US), NULL);\r
+      StringDpOptionLs = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_LS), NULL);\r
+      StringDpOptionUa = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_UA), NULL);\r
+      StringDpOptionUr = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_UR), NULL);\r
+      StringDpOptionUt = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_UT), NULL);\r
+      StringDpOptionUp = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_UP), NULL);\r
+      StringDpOptionLx = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_LX), NULL);\r
+      StringDpOptionLn = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_LN), NULL);\r
+      StringDpOptionLt = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_LT), NULL);\r
+      StringDpOptionLi = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_LI), NULL);\r
+      StringDpOptionLc = HiiGetString (gHiiHandle, STRING_TOKEN (STR_DP_OPTION_LC), NULL);\r
+      \r
       // Boolean Options\r
-      VerboseMode = (ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_LV));\r
-      SummaryMode = (ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_US) ||\r
-                     ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_LS));\r
-      AllMode     = (ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_UA));\r
-      RawMode     = (ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_UR));\r
+      // \r
+      VerboseMode = ShellCommandLineGetFlag (ParamPackage, StringDpOptionLv);\r
+      SummaryMode = (BOOLEAN) (ShellCommandLineGetFlag (ParamPackage, StringDpOptionUs) ||\r
+                    ShellCommandLineGetFlag (ParamPackage, StringDpOptionLs));\r
+      AllMode     = ShellCommandLineGetFlag (ParamPackage, StringDpOptionUa);\r
+      RawMode     = ShellCommandLineGetFlag (ParamPackage, StringDpOptionUr);\r
 #if PROFILING_IMPLEMENTED\r
-      TraceMode   = (ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_UT));\r
-      ProfileMode = (ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_UP));\r
+      TraceMode   = ShellCommandLineGetFlag (ParamPackage, StringDpOptionUt);\r
+      ProfileMode = ShellCommandLineGetFlag (ParamPackage, StringDpOptionUp);\r
 #endif  // PROFILING_IMPLEMENTED\r
-      ExcludeMode = (ShellCommandLineGetFlag (ParamPackage, STR_DP_OPTION_LX));\r
+      ExcludeMode = ShellCommandLineGetFlag (ParamPackage, StringDpOptionLx);\r
+      mShowId     =  ShellCommandLineGetFlag (ParamPackage, StringDpOptionLi);\r
+      CumulativeMode = ShellCommandLineGetFlag (ParamPackage, StringDpOptionLc);\r
 \r
       // Options with Values\r
-      CmdLineArg  = ( ShellCommandLineGetValue (ParamPackage, STR_DP_OPTION_LN));\r
+      CmdLineArg  = ShellCommandLineGetValue (ParamPackage, StringDpOptionLn);\r
       if (CmdLineArg == NULL) {\r
         Number2Display = DEFAULT_DISPLAYCOUNT;\r
       }\r
@@ -217,7 +359,7 @@ InitializeDp (
           Number2Display = MAXIMUM_DISPLAYCOUNT;\r
         }\r
       }\r
-      CmdLineArg  = (ShellCommandLineGetValue (ParamPackage, STR_DP_OPTION_LT));\r
+      CmdLineArg  = ShellCommandLineGetValue (ParamPackage, StringDpOptionLt);\r
       if (CmdLineArg == NULL) {\r
         mInterestThreshold = DEFAULT_THRESHOLD;  // 1ms := 1,000 us\r
       }\r
@@ -233,6 +375,22 @@ InitializeDp (
 #endif  // PROFILING_IMPLEMENTED\r
       }\r
 \r
+  //\r
+  // Init the custom cumulative data.\r
+  //\r
+  CustomCumulativeToken = ShellCommandLineGetValue (ParamPackage, StringDpOptionLc);\r
+  if (CustomCumulativeToken != NULL) {\r
+    CustomCumulativeData = AllocateZeroPool (sizeof (PERF_CUM_DATA));\r
+    ASSERT (CustomCumulativeData != NULL);\r
+    CustomCumulativeData->MinDur = 0;\r
+    CustomCumulativeData->MaxDur = 0;\r
+    CustomCumulativeData->Count  = 0;\r
+    CustomCumulativeData->Duration = 0;\r
+    NameSize = StrLen (CustomCumulativeToken) + 1;\r
+    CustomCumulativeData->Name   = AllocateZeroPool (NameSize);\r
+    UnicodeStrToAsciiStrS (CustomCumulativeToken, CustomCumulativeData->Name, NameSize);\r
+  }\r
+\r
 /****************************************************************************\r
 ****            Timer specific processing                                ****\r
 ****************************************************************************/\r
@@ -241,10 +399,16 @@ InitializeDp (
       //    StartCount = Value loaded into the counter when it starts counting\r
       //      EndCount = Value counter counts to before it needs to be reset\r
       //\r
-      Freq = GetPerformanceCounterProperties (&TimerInfo.StartCount, &TimerInfo.EndCount);\r
+      Status = EfiGetSystemConfigurationTable (&gPerformanceProtocolGuid, (VOID **) &PerformanceProperty);\r
+      if (EFI_ERROR (Status) || (PerformanceProperty == NULL)) {\r
+        PrintToken (STRING_TOKEN (STR_PERF_PROPERTY_NOT_FOUND));\r
+        goto Done;\r
+      }\r
 \r
       // Convert the Frequency from Hz to KHz\r
-      TimerInfo.Frequency = (UINT32)DivU64x32 (Freq, 1000);\r
+      TimerInfo.Frequency  = (UINT32)DivU64x32 (PerformanceProperty->Frequency, 1000);\r
+      TimerInfo.StartCount = PerformanceProperty->TimerStartValue;\r
+      TimerInfo.EndCount   = PerformanceProperty->TimerEndValue;\r
 \r
       // Determine in which direction the performance counter counts.\r
       TimerInfo.CountUp = (BOOLEAN) (TimerInfo.EndCount >= TimerInfo.StartCount);\r
@@ -262,7 +426,7 @@ InitializeDp (
           (! RawMode)\r
          ) {\r
         StringPtr = HiiGetString (gHiiHandle,\r
-                      TimerInfo.CountUp ? STRING_TOKEN (STR_DP_UP) : STRING_TOKEN (STR_DP_DOWN),\r
+                      (EFI_STRING_ID) (TimerInfo.CountUp ? STRING_TOKEN (STR_DP_UP) : STRING_TOKEN (STR_DP_DOWN)),\r
                       NULL);\r
         ASSERT (StringPtr != NULL);\r
         PrintToken (STRING_TOKEN (STR_DP_TIMER_PROPERTIES),   // Print Timer count range and direction\r
@@ -294,10 +458,15 @@ InitializeDp (
 ****    !T &&  P  := (2) Only Profile records are displayed\r
 ****     T &&  P  := (3) Same as Default, both are displayed\r
 ****************************************************************************/\r
-      GatherStatistics();\r
-      if (AllMode) {\r
+      GatherStatistics (CustomCumulativeData);\r
+      if (CumulativeMode) {                       \r
+        ProcessCumulative (CustomCumulativeData);\r
+      } else if (AllMode) {\r
         if (TraceMode) {\r
-          DumpAllTrace( Number2Display, ExcludeMode);\r
+          Status = DumpAllTrace( Number2Display, ExcludeMode);\r
+          if (Status == EFI_ABORTED) {\r
+            goto Done;\r
+          }\r
         }\r
         if (ProfileMode) {\r
           DumpAllProfile( Number2Display, ExcludeMode);\r
@@ -305,7 +474,10 @@ InitializeDp (
       }\r
       else if (RawMode) {\r
         if (TraceMode) {\r
-          DumpRawTrace( Number2Display, ExcludeMode);\r
+          Status = DumpRawTrace( Number2Display, ExcludeMode);\r
+          if (Status == EFI_ABORTED) {\r
+            goto Done;\r
+          }\r
         }\r
         if (ProfileMode) {\r
           DumpRawProfile( Number2Display, ExcludeMode);\r
@@ -314,14 +486,24 @@ InitializeDp (
       else {\r
         //------------- Begin Cooked Mode Processing\r
         if (TraceMode) {\r
-          ProcessPhases ( Ticker );\r
+          ProcessPhases ();\r
           if ( ! SummaryMode) {\r
             Status = ProcessHandles ( ExcludeMode);\r
-            if ( ! EFI_ERROR( Status)) {\r
-              ProcessPeims (     );\r
-              ProcessGlobal (    );\r
-              ProcessCumulative ();\r
+            if (Status == EFI_ABORTED) {\r
+              goto Done;\r
             }\r
+\r
+            Status = ProcessPeims ();\r
+            if (Status == EFI_ABORTED) {\r
+              goto Done;\r
+            }\r
+\r
+            Status = ProcessGlobal ();\r
+            if (Status == EFI_ABORTED) {\r
+              goto Done;\r
+            }\r
+\r
+            ProcessCumulative (NULL);\r
           }\r
         }\r
         if (ProfileMode) {\r
@@ -333,7 +515,45 @@ InitializeDp (
       }\r
     }\r
   }\r
-  (void) FreePool (mPrintTokenBuffer);\r
+\r
+Done:\r
+\r
+  //\r
+  // Free the memory allocate from HiiGetString\r
+  //\r
+  ListIndex = 0;\r
+  while (DpParamList[ListIndex].Name != NULL) {\r
+    FreePool (DpParamList[ListIndex].Name);\r
+    ListIndex ++;\r
+  }  \r
+  FreePool (DpParamList);\r
+\r
+  SafeFreePool (StringDpOptionQh);\r
+  SafeFreePool (StringDpOptionLh);\r
+  SafeFreePool (StringDpOptionUh);\r
+  SafeFreePool (StringDpOptionLv);\r
+  SafeFreePool (StringDpOptionUs);\r
+  SafeFreePool (StringDpOptionLs);\r
+  SafeFreePool (StringDpOptionUa);\r
+  SafeFreePool (StringDpOptionUr);\r
+  SafeFreePool (StringDpOptionUt);\r
+  SafeFreePool (StringDpOptionUp);\r
+  SafeFreePool (StringDpOptionLx);\r
+  SafeFreePool (StringDpOptionLn);\r
+  SafeFreePool (StringDpOptionLt);\r
+  SafeFreePool (StringDpOptionLi);\r
+  SafeFreePool (StringDpOptionLc);\r
+  SafeFreePool (StringPtr);\r
+  SafeFreePool (mPrintTokenBuffer);\r
+\r
+  if (ParamPackage != NULL) {\r
+    ShellCommandLineFreeVarList (ParamPackage);\r
+  }\r
+  if (CustomCumulativeData != NULL) {\r
+    SafeFreePool (CustomCumulativeData->Name);\r
+  }\r
+  SafeFreePool (CustomCumulativeData);\r
+\r
   HiiRemovePackages (gHiiHandle);\r
   return Status;\r
 }\r