mtd: support -c (datasize) option for fixseama command

[openwrt/staging/chunkeey.git] / package / system / mtd / src / seama.c

diff --git a/package/system/mtd/src/seama.c b/package/system/mtd/src/seama.c
index f8b677b1897c1369bc1c923e4e329bbe7d8a6c14..bcda321918d03a772ee6a420501aa899af0ca300 100644 (file)
--- a/package/system/mtd/src/seama.c
+++ b/package/system/mtd/src/seama.c
@@ -104,14 +104,13 @@ err_out:
 }
 
 int
-mtd_fixseama(const char *mtd, size_t offset)
+mtd_fixseama(const char *mtd, size_t offset, size_t data_size)
 {
        int fd;
        char *first_block;
        ssize_t res;
        size_t block_offset;
        size_t data_offset;
-       size_t data_size;
        struct seama_entity_header *shdr;
 
        if (quiet < 2)
@@ -155,7 +154,8 @@ mtd_fixseama(const char *mtd, size_t offset)
        }
 
        data_offset = offset + sizeof(struct seama_entity_header) + ntohs(shdr->metasize);
-       data_size = mtdsize - data_offset;
+       if (!data_size)
+               data_size = mtdsize - data_offset;
        if (data_size > ntohl(shdr->size))
                data_size = ntohl(shdr->size);
        if (seama_fix_md5(shdr, fd, data_offset, data_size))