MdeModulePkg DxeCore: Call UnregisterMemoryProfileImage correctly.

Use if (Image->Started) condition judgement before call to
UnregisterMemoryProfileImage() in CoreUnloadAndCloseImage().

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Star Zeng <star.zeng@intel.com>
Reviewed-by: Jiewen Yao <jiewen.yao@intel.com>

git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@17597 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
Star Zeng 2015-06-09 09:41:02 +00:00 committed by lzeng14
parent eadbec0109
commit 57400f34db
1 changed files with 5 additions and 2 deletions

View File

@ -1,7 +1,7 @@
/** @file /** @file
Core image handling services to load and unload PeImage. Core image handling services to load and unload PeImage.
Copyright (c) 2006 - 2014, Intel Corporation. All rights reserved.<BR> Copyright (c) 2006 - 2015, Intel Corporation. All rights reserved.<BR>
This program and the accompanying materials This program and the accompanying materials
are licensed and made available under the terms and conditions of the BSD License are licensed and made available under the terms and conditions of the BSD License
which accompanies this distribution. The full text of the license may be found at which accompanies this distribution. The full text of the license may be found at
@ -858,6 +858,10 @@ CoreUnloadAndCloseImage (
HandleBuffer = NULL; HandleBuffer = NULL;
ProtocolGuidArray = NULL; ProtocolGuidArray = NULL;
if (Image->Started) {
UnregisterMemoryProfileImage (Image);
}
if (Image->Ebc != NULL) { if (Image->Ebc != NULL) {
// //
// If EBC protocol exists we must perform cleanups for this image. // If EBC protocol exists we must perform cleanups for this image.
@ -1854,7 +1858,6 @@ CoreUnloadImage (
Status = EFI_INVALID_PARAMETER; Status = EFI_INVALID_PARAMETER;
goto Done; goto Done;
} }
UnregisterMemoryProfileImage (Image);
if (Image->Started) { if (Image->Started) {
// //