Add checking for memory allocation in UI modules.

[mirror_edk2.git] / EdkModulePkg / Universal / UserInterface / SetupBrowser / Dxe / InputHandler.c

diff --git a/EdkModulePkg/Universal/UserInterface/SetupBrowser/Dxe/InputHandler.c b/EdkModulePkg/Universal/UserInterface/SetupBrowser/Dxe/InputHandler.c
index d84fcf02b11ea64e37a16877de49a42b21740fad..466039f8dcb56e75d8a5c2258a506993f01530a3 100644 (file)
--- a/EdkModulePkg/Universal/UserInterface/SetupBrowser/Dxe/InputHandler.c
+++ b/EdkModulePkg/Universal/UserInterface/SetupBrowser/Dxe/InputHandler.c
@@ -1233,6 +1233,7 @@ GetSelectionInputPopUp (
         //\r
         if (StrLen (StringPtr) > (PopUpWidth - 1)) {\r
           TempStringPtr = AllocateZeroPool (sizeof (CHAR16) * (PopUpWidth - 1));\r
+          ASSERT (TempStringPtr != NULL);\r
           CopyMem (TempStringPtr, StringPtr, (sizeof (CHAR16) * (PopUpWidth - 5)));\r
           gBS->FreePool (StringPtr);\r
           StringPtr = TempStringPtr;\r