From c2766270efcba795f1d14f9a51f07898eb8a62c1 Mon Sep 17 00:00:00 2001 From: Hyeonsik Song Date: Mon, 25 Jun 2018 11:02:43 +0900 Subject: [PATCH] firmware-utils: tplink-safeloader: add calloc error handling Signed-off-by: Hyeonsik Song --- tools/firmware-utils/src/tplink-safeloader.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/firmware-utils/src/tplink-safeloader.c b/tools/firmware-utils/src/tplink-safeloader.c index b567b2950cd..92d44856e5b 100644 --- a/tools/firmware-utils/src/tplink-safeloader.c +++ b/tools/firmware-utils/src/tplink-safeloader.c @@ -1638,6 +1638,10 @@ static int add_flash_partition( } part_list->name = calloc(1, strlen(name) + 1); + if (!part_list->name) { + error(1, 0, "Unable to allocate memory"); + } + memcpy((char *)part_list->name, name, strlen(name)); part_list->base = base; part_list->size = size;