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