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/HiiDataBase/Dxe/Forms.c
+++ b/EdkModulePkg/Universal/UserInterface/HiiDataBase/Dxe/Forms.c
@@ -1409,6 +1409,8 @@ Returns:
  TempBufferSize += PackageInstance->IfrSize + PackageInstance->StringSize;

  TempBuffer      = AllocateZeroPool (TempBufferSize);
+  ASSERT (TempBuffer != NULL);
+
  OrigTempBuffer  = TempBuffer;

  //