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

Make sure gBS FreePool() is used to free the buffer always allocated by gBS AllocatePool() service. 

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10695 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
lgao4 2010-07-26 16:20:34 +00:00
parent 082be7a58a
commit 0433d8f018
2 changed files with 40 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
MdePkg/Library/DxeServicesLib/DxeServicesLib.c
View File

@ -122,6 +122,7 @@ InternalGetSectionFromFv (
EFI_STATUS Status;
EFI_FIRMWARE_VOLUME2_PROTOCOL *Fv;
UINT32 AuthenticationStatus;
VOID* TempBuffer;
ASSERT (NameGuid != NULL);
ASSERT (Buffer != NULL);
@ -170,6 +171,17 @@ InternalGetSectionFromFv (
);
}
if (!EFI_ERROR (Status)) {
//
// The found buffer by FV protocol is allocated by gBS AllocatePool() service.
// Copy the found buffer to the allocated buffer by AllocatePool().
// So, the returned buffer can be freed by FreePool().
//
TempBuffer = AllocateCopyPool (*Size, *Buffer);
gBS->FreePool (*Buffer);
*Buffer = TempBuffer;
}
return Status;
}
@ -312,7 +324,11 @@ GetSectionFromAnyFvByFileType (
Done:
if (HandleBuffer != NULL) {
FreePool(HandleBuffer);
//
// HandleBuffer is allocated by gBS AllocatePool() service.
// So, gBS FreePool() service is used to free HandleBuffer.
//
gBS->FreePool (HandleBuffer);
}
return Status;
@ -435,8 +451,12 @@ GetSectionFromAnyFv (
Done:
if (HandleBuffer != NULL) {
FreePool(HandleBuffer);
if (HandleBuffer != NULL) {
//
// HandleBuffer is allocated by gBS AllocatePool() service.
// So, gBS FreePool() service is used to free HandleBuffer.
//
gBS->FreePool (HandleBuffer);
}
return Status;
@ -615,6 +635,7 @@ GetFileBufferByFilePath (
EFI_FIRMWARE_VOLUME2_PROTOCOL *FwVol;
EFI_SECTION_TYPE SectionType;
UINT8 *ImageBuffer;
UINT8 *TempBuffer;
UINTN ImageBufferSize;
EFI_FV_FILETYPE Type;
EFI_FV_FILE_ATTRIBUTES Attrib;
@ -642,6 +663,7 @@ GetFileBufferByFilePath (
FileInfo = NULL;
FileHandle = NULL;
ImageBuffer = NULL;
TempBuffer = NULL;
ImageBufferSize = 0;
*AuthenticationStatus = 0;
@ -702,6 +724,16 @@ GetFileBufferByFilePath (
AuthenticationStatus
);
}
if (!EFI_ERROR (Status)) {
//
// The found buffer by FV protocol is allocated by gBS AllocatePool() service.
// Copy the found buffer to the allocated buffer by AllocatePool().
// Then, this returned buffer can be freed by FreePool().
//
TempBuffer = AllocateCopyPool (ImageBufferSize, ImageBuffer);
gBS->FreePool (ImageBuffer);
ImageBuffer = TempBuffer;
}
}
}
goto Finish;

6
MdePkg/Library/UefiPciSegmentLibPciRootBridgeIo/PciSegmentLib.c
View File

@ -96,7 +96,11 @@ PciSegmentLibConstructor (
ASSERT (Descriptors->Desc != ACPI_END_TAG_DESCRIPTOR);
}
FreePool(HandleBuffer);
//
// HandleBuffer is allocated by gBS AllocatePool() service.
// So, gBS FreePool() service is used to free HandleBuffer.
//
gBS->FreePool (HandleBuffer);
return EFI_SUCCESS;
}