Verify memory allocations were successful.

[mirror_edk2.git] / ShellPkg / Library / UefiShellLevel3CommandsLib / Alias.c

diff --git a/ShellPkg/Library/UefiShellLevel3CommandsLib/Alias.c b/ShellPkg/Library/UefiShellLevel3CommandsLib/Alias.c
index da53af4c5cab7276a8d37e6c216e5d81c8c1dacf..1428d37a64065dfc8a3108828c3b0845a736d46d 100644 (file)
--- a/ShellPkg/Library/UefiShellLevel3CommandsLib/Alias.c
+++ b/ShellPkg/Library/UefiShellLevel3CommandsLib/Alias.c
@@ -41,6 +41,9 @@ PrintAllShellAlias(
     return (SHELL_SUCCESS);\r
   }\r
   Alias = AllocateZeroPool(StrSize(ConstAllAliasList));\r
+  if (Alias == NULL) {\r
+    return (SHELL_OUT_OF_RESOURCES);\r
+  }\r
   Walker = (CHAR16*)ConstAllAliasList;\r
 \r
   do {\r