fix: memory leak: buf

Signed-off-by: yuxiaojun <[email protected]>

1 files changed, 7 insertions, 0 deletions

diff --git a/hw/mcu/sony/cxd56/mkspk/mkspk.c b/hw/mcu/sony/cxd56/mkspk/mkspk.c
index 3feb235f6..78d98a08e 100644
--- a/hw/mcu/sony/cxd56/mkspk/mkspk.c
+++ b/hw/mcu/sony/cxd56/mkspk/mkspk.c
@@ -183,6 +183,7 @@ static struct elf_file *load_elf(const char *filename)
   buf = (char *)malloc(fsize);
   if (!buf)
     {
+      free(ef);
       fclose(fp);
       return NULL;
     }
@@ -191,6 +192,8 @@ static struct elf_file *load_elf(const char *filename)
   fclose(fp);
   if (ret != 1)
     {
+      free(ef);
+      free(buf);
       return NULL;
     }
 
@@ -351,7 +354,11 @@ int main(int argc, char **argv)
     }
 
   spkimage = create_image(elf, args->core, args->savename, &size);
+  if(elf->data) {
+      free(elf->data);
+  }
   free(elf);
+  
 
   c = cipher_init(vmk, NULL);
   cipher_calc_cmac(c, spkimage, size, (uint8_t *) spkimage + size);