Fix write length so it includes the updated header (tag) CRC32

Signed-off-by: TJ <hacker@iam.tj>

diff --git a/firmware_header_editor.c b/firmware_header_editor.c
index b41f45d..ca0e24b 100644 (file)
--- a/firmware_header_editor.c
+++ b/firmware_header_editor.c
@@ -324,8 +324,9 @@ main(int argc, char **argv)
        memcpy(buffer + header_crc_offset, &tmp, sizeof(tmp));
 
        if (!opt_simulate) {
+         ssize_t write_len = header_len > header_crc_offset + sizeof(tmp) ? header_len : header_crc_offset + sizeof(tmp);
          lseek(fd, 0, SEEK_SET);
-         write(fd, buffer, header_len);
+         write(fd, buffer, write_len);
        }
 
        printf( format_spec,