Re: [edk2] [Patch] MdeModulePkg LoadFileOnFv2: Fix the potential NULL pointer access

2016-08-02 Thread Tian, Feng 
Reviewed-by: Feng Tian 

Thanks
Feng

-Original Message-
From: edk2-devel [mailto:edk2-devel-boun...@lists.01.org] On Behalf Of Liming 
Gao
Sent: Tuesday, August 2, 2016 1:39 PM
To: edk2-devel@lists.01.org
Subject: [edk2] [Patch] MdeModulePkg LoadFileOnFv2: Fix the potential NULL 
pointer access

Check NULL pointer before access it.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Liming Gao 
---
 MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c | 6 ++
 1 file changed, 6 insertions(+)

diff --git a/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c 
b/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
index 9eea50d..18a07d8 100644
--- a/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
+++ b/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
@@ -345,6 +345,9 @@ FvNotificationEvent (
   Index  = 0;
   BufferSize = sizeof (EFI_HANDLE);
   Handle = AllocateZeroPool (BufferSize);
+  if (Handle == NULL) {
+return;
+  }
   Status = gBS->LocateHandle (
 ByProtocol,
 &gEfiFirmwareVolume2ProtocolGuid, @@ -355,6 +358,9 @@ 
FvNotificationEvent (
   if (EFI_BUFFER_TOO_SMALL == Status) {
 FreePool (Handle);
 Handle = AllocateZeroPool (BufferSize);
+if (Handle == NULL) {
+  return;
+}
 Status = gBS->LocateHandle (
 ByProtocol,
 &gEfiFirmwareVolume2ProtocolGuid,
--
2.8.0.windows.1

___
edk2-devel mailing list
edk2-devel@lists.01.org
https://lists.01.org/mailman/listinfo/edk2-devel
___
edk2-devel mailing list
edk2-devel@lists.01.org
https://lists.01.org/mailman/listinfo/edk2-devel

[edk2] [Patch] MdeModulePkg LoadFileOnFv2: Fix the potential NULL pointer access

2016-08-01 Thread Liming Gao 
Check NULL pointer before access it.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Liming Gao 
---
 MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c | 6 ++
 1 file changed, 6 insertions(+)

diff --git a/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c 
b/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
index 9eea50d..18a07d8 100644
--- a/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
+++ b/MdeModulePkg/Universal/LoadFileOnFv2/LoadFileOnFv2.c
@@ -345,6 +345,9 @@ FvNotificationEvent (
   Index  = 0;
   BufferSize = sizeof (EFI_HANDLE);
   Handle = AllocateZeroPool (BufferSize);
+  if (Handle == NULL) {
+return;
+  }
   Status = gBS->LocateHandle (
 ByProtocol,
 &gEfiFirmwareVolume2ProtocolGuid,
@@ -355,6 +358,9 @@ FvNotificationEvent (
   if (EFI_BUFFER_TOO_SMALL == Status) {
 FreePool (Handle);
 Handle = AllocateZeroPool (BufferSize);
+if (Handle == NULL) {
+  return;
+}
 Status = gBS->LocateHandle (
 ByProtocol,
 &gEfiFirmwareVolume2ProtocolGuid,
-- 
2.8.0.windows.1

___
edk2-devel mailing list
edk2-devel@lists.01.org
https://lists.01.org/mailman/listinfo/edk2-devel