ShellPkg/Ping6: Handle memory allocation failure

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Ruiyu Ni <ruiyu.ni@intel.com>
Reviewed-by: Jaben Carsey <jaben.carsey@intel.com>

diff --git a/ShellPkg/Library/UefiShellNetwork2CommandsLib/Ping6.c b/ShellPkg/Library/UefiShellNetwork2CommandsLib/Ping6.c
index 4496802eac6dbce77657fbaaf31b552fadf9a353..90a2604388552727230ad91499f88d2d3a0110bd 100644 (file)
--- a/ShellPkg/Library/UefiShellNetwork2CommandsLib/Ping6.c
+++ b/ShellPkg/Library/UefiShellNetwork2CommandsLib/Ping6.c
@@ -961,7 +961,11 @@ ShellPing6 (
   ShellStatus = SHELL_SUCCESS;\r
   Private = AllocateZeroPool (sizeof (PING6_PRIVATE_DATA));\r
 \r
-  ASSERT (Private != NULL);\r
+  if (Private == NULL) {\r
+    ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellNetwork2HiiHandle, L"Ping6");\r
+    ShellStatus = SHELL_OUT_OF_RESOURCES;\r
+    goto ON_EXIT;\r
+  }\r
 \r
   Private->ImageHandle = ImageHandle;\r
   Private->SendNum     = SendNumber;\r