tyler.durden/audk
1
0
Fork 0
mirror of https://github.com/acidanthera/audk.git synced 2025-07-26 15:14:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

BaseTools/GenFv: Fix file handles not being closed

Browse Source 
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:
Hao Wu 2016-10-11 15:47:55 +08:00
parent 9639f7d378
commit 320ba37a56
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
BaseTools/Source/C/GenFv/GenFvInternalLib.c
View File

@ -1161,6 +1161,7 @@ Returns:
// //
FileBuffer = malloc (FileSize); FileBuffer = malloc (FileSize);
if (FileBuffer == NULL) { if (FileBuffer == NULL) {
fclose (NewFile);
Error (NULL, 0, 4001, "Resouce", "memory cannot be allocated!"); Error (NULL, 0, 4001, "Resouce", "memory cannot be allocated!");
return EFI_OUT_OF_RESOURCES; return EFI_OUT_OF_RESOURCES;
} }
@ -3506,6 +3507,7 @@ Returns:
PeFileSize = _filelength (fileno (PeFile)); PeFileSize = _filelength (fileno (PeFile));
PeFileBuffer = (UINT8 *) malloc (PeFileSize); PeFileBuffer = (UINT8 *) malloc (PeFileSize);
if (PeFileBuffer == NULL) { if (PeFileBuffer == NULL) {
fclose (PeFile);
Error (NULL, 0, 4001, "Resource", "memory cannot be allocated on rebase of %s", FileName); Error (NULL, 0, 4001, "Resource", "memory cannot be allocated on rebase of %s", FileName);
return EFI_OUT_OF_RESOURCES; return EFI_OUT_OF_RESOURCES;
} }
@ -3761,6 +3763,7 @@ Returns:
PeFileSize = _filelength (fileno (PeFile)); PeFileSize = _filelength (fileno (PeFile));
PeFileBuffer = (UINT8 *) malloc (PeFileSize); PeFileBuffer = (UINT8 *) malloc (PeFileSize);
if (PeFileBuffer == NULL) { if (PeFileBuffer == NULL) {
fclose (PeFile);
Error (NULL, 0, 4001, "Resource", "memory cannot be allocated on rebase of %s", FileName); Error (NULL, 0, 4001, "Resource", "memory cannot be allocated on rebase of %s", FileName);
return EFI_OUT_OF_RESOURCES; return EFI_OUT_OF_RESOURCES;
} }