From 9ea69f8a05b808b4bab81b608436a02e2f2fba09 Mon Sep 17 00:00:00 2001
From: jcarsey <jcarsey@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Mon, 4 Oct 2010 16:30:40 +0000
Subject: Verify memory allocations were successful.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10909 6f19259b-4bc3-4df7-8a09-765794883524
---
 ShellPkg/Library/UefiShellLevel1CommandsLib/For.c | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'ShellPkg/Library/UefiShellLevel1CommandsLib')

diff --git a/ShellPkg/Library/UefiShellLevel1CommandsLib/For.c b/ShellPkg/Library/UefiShellLevel1CommandsLib/For.c
index 0e68119b8d..297a137664 100644
--- a/ShellPkg/Library/UefiShellLevel1CommandsLib/For.c
+++ b/ShellPkg/Library/UefiShellLevel1CommandsLib/For.c
@@ -103,8 +103,11 @@ typedef struct {
   @param[in] Alias              The alias to test for.
   @param[in] CommandString      The updated command string.
   @param[in,out] List           The list to search.
+
+  @retval EFI_SUCCESS           The operation was completed successfully.
+  @retval EFI_OUT_OF_RESOURCES  There was not enough free memory.
 **/
-VOID
+EFI_STATUS
 EFIAPI
 InternalUpdateAliasOnList(
   IN CONST CHAR16       *Alias,
@@ -139,12 +142,16 @@ InternalUpdateAliasOnList(
   }
   if (!Found) {
     Node = AllocateZeroPool(sizeof(ALIAS_LIST));
+    if (Node == NULL) {
+      return (EFI_OUT_OF_RESOURCES);
+    }
     ASSERT(Node->Alias == NULL);
     Node->Alias         = StrnCatGrow(&Node->Alias, NULL, Alias, 0);
     ASSERT(Node->CommandString == NULL);
     Node->CommandString = StrnCatGrow(&Node->CommandString, NULL, CommandString, 0);
     InsertTailList(List, &Node->Link);
   }
+  return (EFI_SUCCESS);
 }
 
 /**
-- 
cgit v1.2.3