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

BaseTools/VfrCompile: Add checks for array access

Browse Source 
Cc: Liming Gao <liming.gao@intel.com>
Cc: Yonghong Zhu <yonghong.zhu@intel.com>
Cc: Eric Dong <eric.dong@intel.com>
Cc: Dandan Bi <dandan.bi@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:
Hao Wu
2016-10-11 11:21:31 +08:00
parent 5acc8d3cdd
commit bab5ad2fd1
2 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
BaseTools/Source/C/VfrCompile/VfrUtilityLib.cpp
View File

@ -1474,6 +1474,10 @@ CVfrDataStorage::GetFreeVarStoreId (
}
}
if (Index == EFI_FREE_VARSTORE_ID_BITMAP_SIZE) {
return EFI_VARSTORE_ID_INVALID;
}
for (Offset = 0, Mask = 0x80000000; Mask != 0; Mask >>= 1, Offset++) {
if ((mFreeVarStoreIdBitMap[Index] & Mask) == 0) {
mFreeVarStoreIdBitMap[Index] |= Mask;
@ -2437,6 +2441,10 @@ CVfrQuestionDB::GetFreeQuestionId (
}
}
if (Index == EFI_FREE_QUESTION_ID_BITMAP_SIZE) {
return EFI_QUESTION_ID_INVALID;
}
for (Offset = 0, Mask = 0x80000000; Mask != 0; Mask >>= 1, Offset++) {
if ((mFreeQIdBitMap[Index] & Mask) == 0) {
mFreeQIdBitMap[Index] |= Mask;