[PATCH u-boot-marvell v2 08/20] tools: kwbimage: Check the return value of image_headersz_v1()

[Pali Rohár]

Function image_headersz_v1() may return zero on fatal errors.
In this case the function already printed an error message.

Check the return value of image_headersz_v1() in kwbimage_generate(),
and exit on zero value with EXIT_FAILURE.

Signed-off-by: Pali Rohár <p...@kernel.org>
Reviewed-by: Marek Behún <marek.be...@nic.cz>
---
 tools/kwbimage.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tools/kwbimage.c b/tools/kwbimage.c
index c0f1bdac0210..a5b518f60bc8 100644
--- a/tools/kwbimage.c
+++ b/tools/kwbimage.c
@@ -2029,6 +2029,10 @@ static int kwbimage_generate(struct image_tool_params 
*params,
 
        case 1:
                alloc_len = image_headersz_v1(NULL);
+               if (!alloc_len) {
+                       free(image_cfg);
+                       exit(EXIT_FAILURE);
+               }
                break;
 
        default:
-- 
2.20.1