archive-gh-me/OpenCorePkg
1
0
Fork 0
mirror of https://github.com/acidanthera/OpenCorePkg.git synced 2025-12-08 19:25:01 +00:00
Code Issues Projects Releases Wiki Activity

Utilities: Corrects missing memory allocation checks in TestNtfsDxe

Browse Source
This commit is contained in:
Savva Mitrofanov 2023-03-01 16:23:36 +06:00
parent 344a2080b2
commit 4c19125820
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
Utilities/TestNtfsDxe/TestNtfsDxe.c
View File

@ -202,6 +202,11 @@ LLVMFuzzerTestOneInput (
Status = FileGetInfo (NewHandle, &gEfiFileInfoGuid, &Len, Info);
if (Status == EFI_BUFFER_TOO_SMALL) {
Info = AllocateZeroPool (Len);
if (Info == NULL) {
FreeAll (FileName, Instance);
return 0;
}
FileGetInfo (NewHandle, &gEfiFileInfoGuid, &Len, Info);
FreePool (Info);
}
@ -210,6 +215,11 @@ LLVMFuzzerTestOneInput (
Status = FileGetInfo (NewHandle, &gEfiFileSystemInfoGuid, &Len, Info);
if (Status == EFI_BUFFER_TOO_SMALL) {
Info = AllocateZeroPool (Len);
if (Info == NULL) {
FreeAll (FileName, Instance);
return 0;
}
FileGetInfo (NewHandle, &gEfiFileSystemInfoGuid, &Len, Info);
FreePool (Info);
}
@ -218,6 +228,11 @@ LLVMFuzzerTestOneInput (
Status = FileGetInfo (NewHandle, &gEfiFileSystemVolumeLabelInfoIdGuid, &Len, Info);
if (Status == EFI_BUFFER_TOO_SMALL) {
Info = AllocateZeroPool (Len);
if (Info == NULL) {
FreeAll (FileName, Instance);
return 0;
}
FileGetInfo (NewHandle, &gEfiFileSystemVolumeLabelInfoIdGuid, &Len, Info);
FreePool (Info);
}