Add checking for memory allocation in UI modules.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2219 6f19259b-4bc3-4df7-8a09-765794883524
2007-01-11 08:03:27 +00:00
parent b12dbde369
commit 41a907e423
4 changed files with 15 additions and 0 deletions
--- a/EdkModulePkg/Universal/UserInterface/SetupBrowser/Dxe/InputHandler.c
+++ b/EdkModulePkg/Universal/UserInterface/SetupBrowser/Dxe/InputHandler.c
@ -1233,6 +1233,7 @@ GetSelectionInputPopUp (
        //
        if (StrLen (StringPtr) > (PopUpWidth - 1)) {
          TempStringPtr = AllocateZeroPool (sizeof (CHAR16) * (PopUpWidth - 1));
+          ASSERT (TempStringPtr != NULL);
          CopyMem (TempStringPtr, StringPtr, (sizeof (CHAR16) * (PopUpWidth - 5)));
          gBS->FreePool (StringPtr);
          StringPtr = TempStringPtr;