BaseTools/GenPage: 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>

diff --git a/BaseTools/Source/C/GenPage/GenPage.c b/BaseTools/Source/C/GenPage/GenPage.c
index e22485c4ae26f352c48ee771a0a9f24251140cec..3bb05efa36cd06b86c31f4b6eb08ee8144d83058 100644 (file)
--- a/BaseTools/Source/C/GenPage/GenPage.c
+++ b/BaseTools/Source/C/GenPage/GenPage.c
@@ -431,9 +431,11 @@ main (
   //\r
   result = GenBinPage (BaseMemory, InputFile, OutputFile);\r
   if (result < 0) {\r
   //\r
   result = GenBinPage (BaseMemory, InputFile, OutputFile);\r
   if (result < 0) {\r

+    free (BaseMemory);\r

     return STATUS_ERROR;\r
   }\r
 \r
     return STATUS_ERROR;\r
   }\r
 \r

+  free (BaseMemory);\r

   return 0;\r
 }\r
 \r
   return 0;\r
 }\r
 \r