tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

BaseTools/VolInfo: Fix potential memory leak 

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:27:55 +08:00
parent bdf5f73120
commit 8500665494
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
BaseTools/Source/C/VolInfo/VolInfo.c
View File

@ -258,6 +258,14 @@ Returns:
continue; continue;
} }
if ((stricmp (argv[0], "--hash") == 0)) { if ((stricmp (argv[0], "--hash") == 0)) {
if (EnableHash == TRUE) {
//
// --hash already given in the option, ignore this one
//
argc --;
argv ++;
continue;
}
EnableHash = TRUE; EnableHash = TRUE;
OpenSslCommand = "openssl"; OpenSslCommand = "openssl";
OpenSslEnv = getenv("OPENSSL_PATH"); OpenSslEnv = getenv("OPENSSL_PATH");
@ -1784,8 +1792,14 @@ Returns:
} }
ScratchBuffer = malloc (ScratchSize); ScratchBuffer = malloc (ScratchSize);
if (ScratchBuffer == NULL) {
Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");
return EFI_OUT_OF_RESOURCES;
}
UncompressedBuffer = malloc (UncompressedLength); UncompressedBuffer = malloc (UncompressedLength);
if ((ScratchBuffer == NULL) || (UncompressedBuffer == NULL)) { if (UncompressedBuffer == NULL) {
free (ScratchBuffer);
Error (NULL, 0, 4001, "Resource", "memory cannot be allocated!");
return EFI_OUT_OF_RESOURCES; return EFI_OUT_OF_RESOURCES;
} }
Status = DecompressFunction ( Status = DecompressFunction (