diff options
author | jcarsey <jcarsey@6f19259b-4bc3-4df7-8a09-765794883524> | 2011-04-19 17:54:42 +0000 |
---|---|---|
committer | jcarsey <jcarsey@6f19259b-4bc3-4df7-8a09-765794883524> | 2011-04-19 17:54:42 +0000 |
commit | f8d3e6898a4c7660291917de31c9ba77d18c0b35 (patch) | |
tree | 296be61c3cf6d3c2dbec87029682817a54258db9 /ShellPkg/Library/UefiShellLib | |
parent | f20076daaf8509e815a37fd0cb5453b9c3b9a79d (diff) | |
download | edk2-f8d3e6898a4c7660291917de31c9ba77d18c0b35.zip edk2-f8d3e6898a4c7660291917de31c9ba77d18c0b35.tar.gz edk2-f8d3e6898a4c7660291917de31c9ba77d18c0b35.tar.bz2 |
check that memory allocation was successful.
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11568 6f19259b-4bc3-4df7-8a09-765794883524
Diffstat (limited to 'ShellPkg/Library/UefiShellLib')
-rw-r--r-- | ShellPkg/Library/UefiShellLib/UefiShellLib.c | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/ShellPkg/Library/UefiShellLib/UefiShellLib.c b/ShellPkg/Library/UefiShellLib/UefiShellLib.c index d4a80e4..d65f427 100644 --- a/ShellPkg/Library/UefiShellLib/UefiShellLib.c +++ b/ShellPkg/Library/UefiShellLib/UefiShellLib.c @@ -2606,6 +2606,12 @@ InternalShellPrintWorker( mPostReplaceFormat = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
mPostReplaceFormat2 = AllocateZeroPool (PcdGet16 (PcdShellPrintBufferSize));
+ if (mPostReplaceFormat == NULL || mPostReplaceFormat2 == NULL) {
+ SHELL_FREE_NON_NULL(mPostReplaceFormat);
+ SHELL_FREE_NON_NULL(mPostReplaceFormat2);
+ return (EFI_OUT_OF_RESOURCES);
+ }
+
Status = EFI_SUCCESS;
OriginalAttribute = gST->ConOut->Mode->Attribute;
|