fix: memory leak: buf

Signed-off-by: yuxiaojun <yuxiaojun@uniontech.com>
This commit is contained in:
yuxiaojun
2023-11-28 17:14:05 +08:00
committed by hathach
parent 178c4144ad
commit 86710c9bb2

View File

@@ -183,6 +183,7 @@ static struct elf_file *load_elf(const char *filename)
buf = (char *)malloc(fsize); buf = (char *)malloc(fsize);
if (!buf) if (!buf)
{ {
free(ef);
fclose(fp); fclose(fp);
return NULL; return NULL;
} }
@@ -191,6 +192,8 @@ static struct elf_file *load_elf(const char *filename)
fclose(fp); fclose(fp);
if (ret != 1) if (ret != 1)
{ {
free(ef);
free(buf);
return NULL; return NULL;
} }
@@ -351,7 +354,11 @@ int main(int argc, char **argv)
} }
spkimage = create_image(elf, args->core, args->savename, &size); spkimage = create_image(elf, args->core, args->savename, &size);
if(elf->data) {
free(elf->data);
}
free(elf); free(elf);
c = cipher_init(vmk, NULL); c = cipher_init(vmk, NULL);
cipher_calc_cmac(c, spkimage, size, (uint8_t *) spkimage + size); cipher_calc_cmac(c, spkimage, size, (uint8_t *) spkimage + size);