BaseTools/GenFw: Avoid possible NULL pointer dereference
Cc: Liming Gao <liming.gao@intel.com> Cc: Yonghong Zhu <yonghong.zhu@intel.com> Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Hao Wu <hao.a.wu@intel.com> Reviewed-by: Liming Gao <liming.gao@intel.com>
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
/** @file
|
||||
Elf64 convert solution
|
||||
|
||||
Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>
|
||||
Copyright (c) 2010 - 2016, Intel Corporation. All rights reserved.<BR>
|
||||
Portions copyright (c) 2013-2014, ARM Ltd. All rights reserved.<BR>
|
||||
|
||||
This program and the accompanying materials are licensed and made available
|
||||
@@ -172,6 +172,10 @@ InitializeElf64 (
|
||||
//
|
||||
VerboseMsg ("Create COFF Section Offset Buffer");
|
||||
mCoffSectionsOffset = (UINT32 *)malloc(mEhdr->e_shnum * sizeof (UINT32));
|
||||
if (mCoffSectionsOffset == NULL) {
|
||||
Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");
|
||||
return FALSE;
|
||||
}
|
||||
memset(mCoffSectionsOffset, 0, mEhdr->e_shnum * sizeof(UINT32));
|
||||
|
||||
//
|
||||
@@ -518,6 +522,10 @@ ScanSections64 (
|
||||
// Allocate base Coff file. Will be expanded later for relocations.
|
||||
//
|
||||
mCoffFile = (UINT8 *)malloc(mCoffOffset);
|
||||
if (mCoffFile == NULL) {
|
||||
Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");
|
||||
}
|
||||
assert (mCoffFile != NULL);
|
||||
memset(mCoffFile, 0, mCoffOffset);
|
||||
|
||||
//
|
||||
|
Reference in New Issue
Block a user