sravan/system76-edk2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

MdeModulePkg LoadFileOnFv2: Fix the potential NULL pointer access

Browse Source 
Check NULL pointer before access it.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Liming Gao <liming.gao@intel.com>
Reviewed-by: Feng Tian <feng.tian@intel.com>
This commit is contained in:
Liming Gao
2016-08-02 13:37:55 +08:00
parent b40ad7b54d
commit deaacda3b2
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
View File

@@ -345,6 +345,9 @@ FvNotificationEvent (
Index = 0; Index = 0;
BufferSize = sizeof (EFI_HANDLE); BufferSize = sizeof (EFI_HANDLE);
Handle = AllocateZeroPool (BufferSize); Handle = AllocateZeroPool (BufferSize);
if (Handle == NULL) {
return;
}
Status = gBS->LocateHandle ( Status = gBS->LocateHandle (
ByProtocol, ByProtocol,
&gEfiFirmwareVolume2ProtocolGuid, &gEfiFirmwareVolume2ProtocolGuid,
@@ -355,6 +358,9 @@ FvNotificationEvent (
if (EFI_BUFFER_TOO_SMALL == Status) { if (EFI_BUFFER_TOO_SMALL == Status) {
FreePool (Handle); FreePool (Handle);
Handle = AllocateZeroPool (BufferSize); Handle = AllocateZeroPool (BufferSize);
if (Handle == NULL) {
return;
}
Status = gBS->LocateHandle ( Status = gBS->LocateHandle (
ByProtocol, ByProtocol,
&gEfiFirmwareVolume2ProtocolGuid, &gEfiFirmwareVolume2ProtocolGuid,