tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ShellPkg/Dp: make sure memory is freed before exit 

Run dp command now:
Firstly it will get performance records from FPDT and then
parse the DP command. And if encounter invalid parameters,
it will exit directly. Thus the performance records got before
are invalid. And what's worse is that the memory allocated in
getting performance records phase is not freed.

This patch update the code to parse the command firstly and
then get the performance records. And make sure that all the
clean work has been done before exiting.

Cc: Liming Gao <liming.gao@intel.com>
Cc: Ruiyu Ni <ruiyu.ni@intel.com>
Cc: Jaben Carsey <jaben.carsey@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.1
Signed-off-by: Dandan Bi <dandan.bi@intel.com>
Reviewed-by: Ruiyu Ni <ruiyu.ni@intel.com>
This commit is contained in:
Dandan Bi 2018-05-11 14:02:08 +08:00 committed by Eric Dong
parent c25d390552
commit 37d533da76
1 changed files with 37 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
ShellPkg/DynamicCommand/DpDynamicCommand/Dp.c
View File

@ -392,7 +392,7 @@ BuildCachedGuidHandleTable (
FreePool (HandleBuffer);
HandleBuffer = NULL;
}
return Status;
return EFI_SUCCESS;
}
/**
@ -731,35 +731,6 @@ RunDp (
Status = ShellInitialize();
ASSERT_EFI_ERROR(Status);
//
// DP dump performance data by parsing FPDT table in ACPI table.
// Folloing 3 steps are to get the measurement form the FPDT table.
//
//
//1. Get FPDT from ACPI table.
//
Status = GetBootPerformanceTable ();
if (EFI_ERROR(Status)) {
return Status;
}
//
//2. Cache the ModuleGuid and hanlde mapping table.
//
Status = BuildCachedGuidHandleTable();
if (EFI_ERROR (Status)) {
return Status;
}
//
//3. Build the measurement array form the FPDT records.
//
Status = BuildMeasurementList ();
if (EFI_ERROR(Status)) {
return Status;
}
//
// Process Command Line arguments
//
@ -811,6 +782,38 @@ RunDp (
#endif // PROFILING_IMPLEMENTED
}
//
// DP dump performance data by parsing FPDT table in ACPI table.
// Folloing 3 steps are to get the measurement form the FPDT table.
//
//
//1. Get FPDT from ACPI table.
//
Status = GetBootPerformanceTable ();
if (EFI_ERROR (Status)) {
ShellStatus = Status;
goto Done;
}
//
//2. Cache the ModuleGuid and hanlde mapping table.
//
Status = BuildCachedGuidHandleTable();
if (EFI_ERROR (Status)) {
ShellStatus = Status;
goto Done;
}
//
//3. Build the measurement array form the FPDT records.
//
Status = BuildMeasurementList ();
if (EFI_ERROR (Status)) {
ShellStatus = SHELL_OUT_OF_RESOURCES;
goto Done;
}
//
// Initialize the pre-defined cumulative data.
//
@ -823,7 +826,8 @@ RunDp (
if (CustomCumulativeToken != NULL) {
CustomCumulativeData = AllocateZeroPool (sizeof (PERF_CUM_DATA));
if (CustomCumulativeData == NULL) {
return SHELL_OUT_OF_RESOURCES;
ShellStatus = SHELL_OUT_OF_RESOURCES;
goto Done;
}
CustomCumulativeData->MinDur = PERF_MAXDUR;
CustomCumulativeData->MaxDur = 0;
@ -832,8 +836,8 @@ RunDp (
NameSize = StrLen (CustomCumulativeToken) + 1;
CustomCumulativeData->Name = AllocateZeroPool (NameSize);
if (CustomCumulativeData->Name == NULL) {
FreePool (CustomCumulativeData);
return SHELL_OUT_OF_RESOURCES;
ShellStatus = SHELL_OUT_OF_RESOURCES;
goto Done;
}
UnicodeStrToAsciiStrS (CustomCumulativeToken, CustomCumulativeData->Name, NameSize);
}