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

View File

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