tools: kwboot: Fix inserting UART data checksum without -B option

Commit 83a1a335bc84 ("tools: kwboot: Fix parsing UART image without data
checksum") added fixup code to insert place for data checksum if UART image
does not have it. Together with option -B (change baudrate), kwboot
calculates this checksum. Without option -B, it inserts only place for
checksum but does not calculate it.

This commit fix above logic and calculate data checksum also when kwboot is
used without -B option.

Fixes: 83a1a335bc84 ("tools: kwboot: Fix parsing UART image without data checksum")
Signed-off-by: Pali Rohár <pali@kernel.org>
Reviewed-by: Stefan Roese <sr@denx.de>

diff --git a/tools/kwboot.c b/tools/kwboot.c
index ef108c8395b857c5fe8b47699f83ec8a7bd8a7fb..61a9c3065aac6bf14c22f7e4efa0370804eadb8c 100644 (file)
--- a/tools/kwboot.c
+++ b/tools/kwboot.c
@@ -2079,6 +2079,8 @@ kwboot_img_patch(void *img, size_t *size, int baudrate)
                        goto err;
                }
                kwboot_img_grow_data_right(img, size, sizeof(uint32_t));
+               /* Update the 32-bit data checksum */
+               *kwboot_img_csum32_ptr(img) = kwboot_img_csum32(img);
        }
 
        if (!kwboot_img_has_ddr_init(img) &&