Guarantee no memory corruption in an out of memory condition even in production builds.
Signed-off-by: niruiyu
Reviewed-by: mdkinney
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11777
6f19259b-4bc3-4df7-8a09-
765794883524
This library instance uses EFI_HOB_TYPE_CV defined in Intel framework HOB specification v0.9\r
to implement HobLib BuildCvHob() API. \r
\r
This library instance uses EFI_HOB_TYPE_CV defined in Intel framework HOB specification v0.9\r
to implement HobLib BuildCvHob() API. \r
\r
-Copyright (c) 2007 - 2010, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2007 - 2011, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution. The full text of the license may be found at\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution. The full text of the license may be found at\r
@param Type Type of the new HOB.\r
@param Length Length of the new HOB to allocate.\r
\r
@param Type Type of the new HOB.\r
@param Length Length of the new HOB to allocate.\r
\r
- @return The address of new HOB.\r
+ @retval NULL The HOB could not be allocated.\r
+ @retval others The address of new HOB.\r
VOID *Hob;\r
\r
Status = PeiServicesCreateHob (Type, Length, &Hob);\r
VOID *Hob;\r
\r
Status = PeiServicesCreateHob (Type, Length, &Hob);\r
+ if (EFI_ERROR (Status)) {\r
+ Hob = NULL;\r
+ }\r
//\r
// Assume the process of HOB building is always successful.\r
//\r
//\r
// Assume the process of HOB building is always successful.\r
//\r
- ASSERT_EFI_ERROR (Status);\r
+ ASSERT (Hob != NULL);\r
((ModuleLength & (EFI_PAGE_SIZE - 1)) == 0));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, (UINT16) sizeof (EFI_HOB_MEMORY_ALLOCATION_MODULE));\r
((ModuleLength & (EFI_PAGE_SIZE - 1)) == 0));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, (UINT16) sizeof (EFI_HOB_MEMORY_ALLOCATION_MODULE));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
CopyGuid (&(Hob->MemoryAllocationHeader.Name), &gEfiHobMemoryAllocModuleGuid);\r
Hob->MemoryAllocationHeader.MemoryBaseAddress = MemoryAllocationModule;\r
\r
CopyGuid (&(Hob->MemoryAllocationHeader.Name), &gEfiHobMemoryAllocModuleGuid);\r
Hob->MemoryAllocationHeader.MemoryBaseAddress = MemoryAllocationModule;\r
EFI_HOB_RESOURCE_DESCRIPTOR *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_RESOURCE_DESCRIPTOR, (UINT16) sizeof (EFI_HOB_RESOURCE_DESCRIPTOR));\r
EFI_HOB_RESOURCE_DESCRIPTOR *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_RESOURCE_DESCRIPTOR, (UINT16) sizeof (EFI_HOB_RESOURCE_DESCRIPTOR));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
Hob->ResourceType = ResourceType;\r
Hob->ResourceAttribute = ResourceAttribute;\r
\r
Hob->ResourceType = ResourceType;\r
Hob->ResourceAttribute = ResourceAttribute;\r
@param Guid The GUID to tag the customized HOB.\r
@param DataLength The size of the data payload for the GUID HOB.\r
\r
@param Guid The GUID to tag the customized HOB.\r
@param DataLength The size of the data payload for the GUID HOB.\r
\r
- @return The start address of GUID HOB data.\r
+ @retval NULL The GUID HOB could not be allocated.\r
+ @retval others The start address of GUID HOB data.\r
{\r
EFI_HOB_GUID_TYPE *Hob;\r
\r
{\r
EFI_HOB_GUID_TYPE *Hob;\r
\r
+ //\r
+ // Make sure Guid is valid\r
+ //\r
+ ASSERT (Guid != NULL);\r
+ \r
//\r
// Make sure that data length is not too long.\r
//\r
ASSERT (DataLength <= (0xffff - sizeof (EFI_HOB_GUID_TYPE)));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_GUID_EXTENSION, (UINT16) (sizeof (EFI_HOB_GUID_TYPE) + DataLength));\r
//\r
// Make sure that data length is not too long.\r
//\r
ASSERT (DataLength <= (0xffff - sizeof (EFI_HOB_GUID_TYPE)));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_GUID_EXTENSION, (UINT16) (sizeof (EFI_HOB_GUID_TYPE) + DataLength));\r
+ if (Hob == NULL) {\r
+ return Hob;\r
+ }\r
CopyGuid (&Hob->Name, Guid);\r
return Hob + 1;\r
}\r
CopyGuid (&Hob->Name, Guid);\r
return Hob + 1;\r
}\r
@param Data The data to be copied into the data field of the GUID HOB.\r
@param DataLength The size of the data payload for the GUID HOB.\r
\r
@param Data The data to be copied into the data field of the GUID HOB.\r
@param DataLength The size of the data payload for the GUID HOB.\r
\r
- @return The start address of GUID HOB data.\r
+ @retval NULL The GUID HOB could not be allocated.\r
+ @retval others The start address of GUID HOB data.\r
ASSERT (Data != NULL || DataLength == 0);\r
\r
HobData = BuildGuidHob (Guid, DataLength);\r
ASSERT (Data != NULL || DataLength == 0);\r
\r
HobData = BuildGuidHob (Guid, DataLength);\r
+ if (HobData == NULL) {\r
+ return HobData;\r
+ }\r
\r
return CopyMem (HobData, Data, DataLength);\r
}\r
\r
return CopyMem (HobData, Data, DataLength);\r
}\r
EFI_HOB_FIRMWARE_VOLUME *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_FV, (UINT16) sizeof (EFI_HOB_FIRMWARE_VOLUME));\r
EFI_HOB_FIRMWARE_VOLUME *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_FV, (UINT16) sizeof (EFI_HOB_FIRMWARE_VOLUME));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
Hob->BaseAddress = BaseAddress;\r
Hob->Length = Length;\r
\r
Hob->BaseAddress = BaseAddress;\r
Hob->Length = Length;\r
EFI_HOB_FIRMWARE_VOLUME2 *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_FV2, (UINT16) sizeof (EFI_HOB_FIRMWARE_VOLUME2));\r
EFI_HOB_FIRMWARE_VOLUME2 *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_FV2, (UINT16) sizeof (EFI_HOB_FIRMWARE_VOLUME2));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
Hob->BaseAddress = BaseAddress;\r
Hob->Length = Length;\r
\r
Hob->BaseAddress = BaseAddress;\r
Hob->Length = Length;\r
EFI_HOB_CAPSULE_VOLUME *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_CV, (UINT16) sizeof (EFI_HOB_CAPSULE_VOLUME));\r
EFI_HOB_CAPSULE_VOLUME *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_CV, (UINT16) sizeof (EFI_HOB_CAPSULE_VOLUME));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
Hob->BaseAddress = BaseAddress;\r
Hob->Length = Length;\r
\r
Hob->BaseAddress = BaseAddress;\r
Hob->Length = Length;\r
EFI_HOB_CPU *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_CPU, (UINT16) sizeof (EFI_HOB_CPU));\r
EFI_HOB_CPU *Hob;\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_CPU, (UINT16) sizeof (EFI_HOB_CPU));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
Hob->SizeOfMemorySpace = SizeOfMemorySpace;\r
Hob->SizeOfIoSpace = SizeOfIoSpace;\r
\r
Hob->SizeOfMemorySpace = SizeOfMemorySpace;\r
Hob->SizeOfIoSpace = SizeOfIoSpace;\r
((Length & (EFI_PAGE_SIZE - 1)) == 0));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, (UINT16) sizeof (EFI_HOB_MEMORY_ALLOCATION_STACK));\r
((Length & (EFI_PAGE_SIZE - 1)) == 0));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, (UINT16) sizeof (EFI_HOB_MEMORY_ALLOCATION_STACK));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
CopyGuid (&(Hob->AllocDescriptor.Name), &gEfiHobMemoryAllocStackGuid);\r
Hob->AllocDescriptor.MemoryBaseAddress = BaseAddress;\r
\r
CopyGuid (&(Hob->AllocDescriptor.Name), &gEfiHobMemoryAllocStackGuid);\r
Hob->AllocDescriptor.MemoryBaseAddress = BaseAddress;\r
((Length & (EFI_PAGE_SIZE - 1)) == 0));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, (UINT16) sizeof (EFI_HOB_MEMORY_ALLOCATION_BSP_STORE));\r
((Length & (EFI_PAGE_SIZE - 1)) == 0));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, (UINT16) sizeof (EFI_HOB_MEMORY_ALLOCATION_BSP_STORE));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
CopyGuid (&(Hob->AllocDescriptor.Name), &gEfiHobMemoryAllocBspStoreGuid);\r
Hob->AllocDescriptor.MemoryBaseAddress = BaseAddress;\r
\r
CopyGuid (&(Hob->AllocDescriptor.Name), &gEfiHobMemoryAllocBspStoreGuid);\r
Hob->AllocDescriptor.MemoryBaseAddress = BaseAddress;\r
((Length & (EFI_PAGE_SIZE - 1)) == 0));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, (UINT16) sizeof (EFI_HOB_MEMORY_ALLOCATION));\r
((Length & (EFI_PAGE_SIZE - 1)) == 0));\r
\r
Hob = InternalPeiCreateHob (EFI_HOB_TYPE_MEMORY_ALLOCATION, (UINT16) sizeof (EFI_HOB_MEMORY_ALLOCATION));\r
+ if (Hob == NULL) {\r
+ return;\r
+ }\r
\r
ZeroMem (&(Hob->AllocDescriptor.Name), sizeof (EFI_GUID));\r
Hob->AllocDescriptor.MemoryBaseAddress = BaseAddress;\r
\r
ZeroMem (&(Hob->AllocDescriptor.Name), sizeof (EFI_GUID));\r
Hob->AllocDescriptor.MemoryBaseAddress = BaseAddress;\r