[PATCH u-boot-marvell 03/16] tools: kwbimage: Fix generating image with multiple DATA_DELAY commands

[Pali Rohár]

Register set header consists of sequence of DATA commands followed by
exactly one DATA_DELAY command. Thus if we are generating image with
multiple DATA_DELAY commands, we need to create more register set headers.

Fix calculation of image size with multiple DATA_DELAY commands and
correctly set pointer to struct register_set_hdr_v1 when initializing new
register set header.

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

diff --git a/tools/kwbimage.c b/tools/kwbimage.c
index 9ac74cfbe514..6432551065ab 100644
--- a/tools/kwbimage.c
+++ b/tools/kwbimage.c
@@ -991,7 +991,7 @@ static void *image_create_v0(size_t *imagesz, struct 
image_tool_params *params,
 
 static size_t image_headersz_v1(int *hasext)
 {
-       struct image_cfg_element *binarye;
+       struct image_cfg_element *binarye, *e;
        unsigned int count;
        size_t headersz;
        int cfgi;
@@ -1008,7 +1008,18 @@ static size_t image_headersz_v1(int *hasext)
                        *hasext = 1;
        }
 
-       count = image_count_options(IMAGE_CFG_DATA);
+       count = 0;
+       for (cfgi = 0; cfgi < cfgn; cfgi++) {
+               e = &image_cfg[cfgi];
+
+               if (e->type == IMAGE_CFG_DATA)
+                       count++;
+
+               if (e->type == IMAGE_CFG_DATA_DELAY) {
+                       headersz += sizeof(struct register_set_hdr_v1) + 8 * 
count + 4;
+                       count = 0;
+               }
+       }
        if (count > 0)
                headersz += sizeof(struct register_set_hdr_v1) + 8 * count + 4;
 
@@ -1366,12 +1377,13 @@ static void *image_create_v1(size_t *imagesz, struct 
image_tool_params *params,
        }
 
        datai = 0;
-       register_set_hdr = (struct register_set_hdr_v1 *)cur;
        for (cfgi = 0; cfgi < cfgn; cfgi++) {
                e = &image_cfg[cfgi];
                if (e->type != IMAGE_CFG_DATA &&
                    e->type != IMAGE_CFG_DATA_DELAY)
                        continue;
+               if (datai == 0)
+                       register_set_hdr = (struct register_set_hdr_v1 *)cur;
                if (e->type == IMAGE_CFG_DATA_DELAY) {
                        finish_register_set_header_v1(&cur, &next_ext, 
register_set_hdr,
                                                      &datai, e->regdata_delay);
-- 
2.20.1