From e7a5a238c32f62441ed5a156525b9db96f0892da Mon Sep 17 00:00:00 2001
From: Ruiyu Ni <ruiyu.ni@intel.com>
Date: Wed, 13 Jul 2016 17:43:47 +0800
Subject: [PATCH] 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>
---
 ShellPkg/Library/UefiShellNetwork2CommandsLib/Ping6.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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