[edk2] [patch] MdeModulePkg: Add ASSERT to make sure pointer 'OptionalData' not be NULL

[Dandan Bi]

Cc: Liming Gao <liming....@intel.com>
Cc: Eric Dong <eric.d...@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Dandan Bi <dandan...@intel.com>
---
 MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c 
b/MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c
index 929f383..2ecf5b6 100644
--- a/MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c
+++ b/MdeModulePkg/Library/BootMaintenanceManagerUiLib/Variable.c
@@ -612,10 +612,11 @@ Var_UpdateDriverOption (
   NewMenuEntry->DisplayStringToken = HiiSetString (HiiHandle, 0, 
NewMenuEntry->DisplayString, NULL);
   NewMenuEntry->HelpStringToken = HiiSetString (HiiHandle, 0, 
NewMenuEntry->HelpString, NULL);
 
   if (OptionalDataExist) {
     NewLoadContext->OptionalData = AllocateZeroPool 
(LoadOption.OptionalDataSize);
+    ASSERT (NewLoadContext->OptionalData != NULL);
     CopyMem (
       NewLoadContext->OptionalData,
       LoadOption.OptionalData,
       LoadOption.OptionalDataSize
       );
@@ -723,10 +724,11 @@ Var_UpdateBootOption (
   NewMenuEntry->DisplayStringToken = HiiSetString (CallbackData->BmmHiiHandle, 
0, NewMenuEntry->DisplayString, NULL);
   NewMenuEntry->HelpStringToken = HiiSetString (CallbackData->BmmHiiHandle, 0, 
NewMenuEntry->HelpString, NULL);
 
   if (OptionalDataExist) {
     NewLoadContext->OptionalData = AllocateZeroPool 
(LoadOption.OptionalDataSize);
+    ASSERT (NewLoadContext->OptionalData != NULL);
     CopyMem (
       NewLoadContext->OptionalData,
       LoadOption.OptionalData,
       LoadOption.OptionalDataSize
       );
-- 
1.9.5.msysgit.1

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