sravan/system76-edk2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add checking for memory allocation in UI modules.

Browse Source 
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2219 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
xli24
2007-01-11 08:03:27 +00:00
parent b12dbde369
commit 41a907e423
4 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
EdkModulePkg/Universal/UserInterface/DriverSample/DriverSample.c
View File

@ -94,11 +94,13 @@ Returns:
// Need some memory for OptionList. Allow for up to 8 options.
//
OptionList = AllocateZeroPool (sizeof (IFR_OPTION) * 8);
ASSERT (OptionList != NULL);
//
// Allocate space for creation of Buffer
//
UpdateData = AllocateZeroPool (0x1000);
ASSERT (UpdateData != NULL);
//
// Remove all the op-codes starting with Label 0x2222 to next Label (second label is for convenience
@ -170,11 +172,13 @@ Returns:
// Need some memory for OptionList. Allow for up to 8 options.
//
OptionList = AllocateZeroPool (sizeof (IFR_OPTION) * 8);
ASSERT (OptionList != NULL);
//
// Allocate space for creation of Buffer
//
UpdateData = AllocateZeroPool (0x1000);
ASSERT (UpdateData != NULL);
//
// Remove all the op-codes starting with Label 0x2222 to next Label (second label is for convenience
@ -244,6 +248,7 @@ Returns:
0x1000,
(VOID **) &UpdateData
);
ASSERT_EFI_ERROR (Status);
ZeroMem (UpdateData, 0x1000);
@ -297,6 +302,7 @@ Returns:
0x1000,
(VOID **)&UpdateData
);
ASSERT_EFI_ERROR (Status);
ZeroMem (UpdateData, 0x1000);
@ -366,6 +372,7 @@ Returns:
sizeof (EFI_HII_CALLBACK_PACKET) + sizeof (SAMPLE_STRING) + 2,
(VOID **) Packet
);
ASSERT_EFI_ERROR (Status);
ZeroMem (*Packet, sizeof (EFI_HII_CALLBACK_PACKET) + sizeof (SAMPLE_STRING) + 2);
@ -384,6 +391,7 @@ Returns:
sizeof (EFI_HII_CALLBACK_PACKET) + 2,
(VOID **) Packet
);
ASSERT_EFI_ERROR (Status);
ZeroMem (*Packet, sizeof (EFI_HII_CALLBACK_PACKET) + 2);
@ -585,6 +593,7 @@ DriverSampleInit (
0x1000,
(VOID **) &UpdateData
);
ASSERT_EFI_ERROR (Status);
ZeroMem (UpdateData, 0x1000);