Add checking for memory allocation in UI modules.
authorxli24 <xli24@de2fecce-e211-0410-80a6-f3fac2684e05>
Thu, 11 Jan 2007 08:03:27 +0000 (08:03 +0000)
committerxli24 <xli24@de2fecce-e211-0410-80a6-f3fac2684e05>
Thu, 11 Jan 2007 08:03:27 +0000 (08:03 +0000)
git-svn-id: https://edk2.tianocore.org/svn/edk2/trunk@2219 de2fecce-e211-0410-80a6-f3fac2684e05

edk2/EdkModulePkg/Universal/UserInterface/DriverSample/DriverSample.c
edk2/EdkModulePkg/Universal/UserInterface/HiiDataBase/Dxe/Forms.c
edk2/EdkModulePkg/Universal/UserInterface/SetupBrowser/Dxe/InputHandler.c
edk2/EdkModulePkg/Universal/UserInterface/SetupBrowser/Dxe/Ui.c

index 5f3823c..c383d16 100644 (file)
@@ -94,11 +94,13 @@ Returns:
     // Need some memory for OptionList. Allow for up to 8 options.\r
     //\r
     OptionList = AllocateZeroPool (sizeof (IFR_OPTION) * 8);\r
+    ASSERT (OptionList != NULL);\r
 \r
     //\r
     // Allocate space for creation of Buffer\r
     //\r
     UpdateData = AllocateZeroPool (0x1000);\r
+    ASSERT (UpdateData != NULL);\r
 \r
     //\r
     // Remove all the op-codes starting with Label 0x2222 to next Label (second label is for convenience\r
@@ -170,11 +172,13 @@ Returns:
     // Need some memory for OptionList. Allow for up to 8 options.\r
     //\r
     OptionList = AllocateZeroPool (sizeof (IFR_OPTION) * 8);\r
+    ASSERT (OptionList != NULL);\r
 \r
     //\r
     // Allocate space for creation of Buffer\r
     //\r
     UpdateData = AllocateZeroPool (0x1000);\r
+    ASSERT (UpdateData != NULL);\r
 \r
     //\r
     // Remove all the op-codes starting with Label 0x2222 to next Label (second label is for convenience\r
@@ -244,6 +248,7 @@ Returns:
                     0x1000,\r
                     (VOID **) &UpdateData\r
                     );\r
+    ASSERT_EFI_ERROR (Status);\r
 \r
     ZeroMem (UpdateData, 0x1000);\r
 \r
@@ -297,6 +302,7 @@ Returns:
                     0x1000,\r
                     (VOID **)&UpdateData\r
                     );\r
+    ASSERT_EFI_ERROR (Status);\r
 \r
     ZeroMem (UpdateData, 0x1000);\r
 \r
@@ -366,6 +372,7 @@ Returns:
                     sizeof (EFI_HII_CALLBACK_PACKET) + sizeof (SAMPLE_STRING) + 2,\r
                     (VOID **) Packet\r
                     );\r
+    ASSERT_EFI_ERROR (Status);\r
 \r
     ZeroMem (*Packet, sizeof (EFI_HII_CALLBACK_PACKET) + sizeof (SAMPLE_STRING) + 2);\r
 \r
@@ -384,6 +391,7 @@ Returns:
                     sizeof (EFI_HII_CALLBACK_PACKET) + 2,\r
                     (VOID **) Packet\r
                     );\r
+    ASSERT_EFI_ERROR (Status);\r
 \r
     ZeroMem (*Packet, sizeof (EFI_HII_CALLBACK_PACKET) + 2);\r
 \r
@@ -585,6 +593,7 @@ DriverSampleInit (
                   0x1000,\r
                   (VOID **) &UpdateData\r
                   );\r
+  ASSERT_EFI_ERROR (Status);\r
 \r
   ZeroMem (UpdateData, 0x1000);\r
 \r
index 2c625fe..4178284 100644 (file)
@@ -1409,6 +1409,8 @@ Returns:
   TempBufferSize += PackageInstance->IfrSize + PackageInstance->StringSize;\r
 \r
   TempBuffer      = AllocateZeroPool (TempBufferSize);\r
+  ASSERT (TempBuffer != NULL);\r
+\r
   OrigTempBuffer  = TempBuffer;\r
 \r
   //\r
index d84fcf0..466039f 100644 (file)
@@ -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
index 7a2ec04..369d2b4 100644 (file)
@@ -3003,7 +3003,10 @@ Returns:
         FileFormTags->VariableDefinitions->VariableFakeSize = (UINT16) (FileFormTags->VariableDefinitions->VariableFakeSize + FileFormTags->VariableDefinitions->VariableSize);\r
         \r
         FileFormTags->VariableDefinitions->NvRamMap = AllocateZeroPool (FileFormTags->VariableDefinitions->VariableSize);\r
+        ASSERT (FileFormTags->VariableDefinitions->NvRamMap != NULL);\r
+\r
         FileFormTags->VariableDefinitions->FakeNvRamMap = AllocateZeroPool (NvMapSize + FileFormTags->VariableDefinitions->VariableFakeSize);\r
+        ASSERT (FileFormTags->VariableDefinitions->FakeNvRamMap != NULL);\r
 \r
         CopyMem (FileFormTags->VariableDefinitions->NvRamMap, NvMap, NvMapSize);\r
         gBS->FreePool (NvMapListHead);\r