Author: rmilecki
Date: 2016-05-09 14:53:29 +0200 (Mon, 09 May 2016)
New Revision: 49306

Modified:
   trunk/package/system/mtd/src/seama.c
   trunk/package/system/mtd/src/seama.h
Log:
mtd: seama: add md5 to header struct

This allows us to drop some extra offset calculations and simplifies
code a bit.

Signed-off-by: Rafa{U+0142} Mi{U+0142}ecki <[email protected]>

Modified: trunk/package/system/mtd/src/seama.c
===================================================================
--- trunk/package/system/mtd/src/seama.c        2016-05-09 09:55:35 UTC (rev 
49305)
+++ trunk/package/system/mtd/src/seama.c        2016-05-09 12:53:29 UTC (rev 
49306)
@@ -53,7 +53,7 @@
 int
 seama_fix_md5(char *buf, size_t len)
 {
-       struct seama_hdr *shdr = (struct seama_hdr *) buf;
+       struct seama_entity_header *shdr = (struct seama_entity_header *) buf;
        char *data;
        size_t msize;
        size_t isize;
@@ -61,7 +61,7 @@
        unsigned char digest[16];
        int i;
 
-       if (len < sizeof(struct seama_hdr))
+       if (len < sizeof(struct seama_entity_header))
                return -1;
 
        isize = ntohl(shdr->size);
@@ -71,17 +71,17 @@
                return -1;
        }
 
-       len -= sizeof(struct seama_hdr) + sizeof(digest) + msize;
+       len -= sizeof(struct seama_entity_header) + msize;
        if (isize > len)
                isize = len;
 
-       data = buf + sizeof(struct seama_hdr) + sizeof(digest) + msize;
+       data = buf + sizeof(struct seama_entity_header) + msize;
 
        MD5_Init(&ctx);
        MD5_Update(&ctx, data, isize);
        MD5_Final(digest, &ctx);
 
-       if (!memcmp(digest, &buf[sizeof(struct seama_hdr)], sizeof(digest))) {
+       if (!memcmp(digest, shdr->md5, sizeof(digest))) {
                if (quiet < 2)
                        fprintf(stderr, "the header is fixed already\n");
                return -1;
@@ -99,8 +99,7 @@
        shdr->size = htonl(isize);
 
        /* update the checksum in the image */
-       for (i = 0; i < sizeof(digest); i++)
-               buf[sizeof(struct seama_hdr) + i] = digest[i];
+       memcpy(shdr->md5, digest, sizeof(digest));
 
        return 0;
 }
@@ -113,7 +112,7 @@
        char *buf;
        ssize_t res;
        size_t block_offset;
-       struct seama_hdr *shdr;
+       struct seama_entity_header *shdr;
 
        if (quiet < 2)
                fprintf(stderr, "Trying to fix SEAMA header in %s at 0x%x...\n",
@@ -146,7 +145,7 @@
                exit(1);
        }
 
-       shdr = (struct seama_hdr *)first_block;
+       shdr = (struct seama_entity_header *)first_block;
        if (shdr->magic != htonl(SEAMA_MAGIC)) {
                fprintf(stderr, "No SEAMA header found\n");
                return -1;

Modified: trunk/package/system/mtd/src/seama.h
===================================================================
--- trunk/package/system/mtd/src/seama.h        2016-05-09 09:55:35 UTC (rev 
49305)
+++ trunk/package/system/mtd/src/seama.h        2016-05-09 12:53:29 UTC (rev 
49306)
@@ -94,14 +94,14 @@
  *     | image size                |
  *  +-------------+-------------+
  */
+
 /* seama header */
-typedef struct seama_hdr       seamahdr_t;
-struct seama_hdr
-{
+struct seama_entity_header {
        uint32_t        magic;                  /* should always be 
SEAMA_MAGIC. */
        uint16_t        reserved;               /* reserved for  */
        uint16_t        metasize;               /* size of the META data */
        uint32_t        size;                   /* size of the image */
+       uint8_t         md5[16];
 } __attribute__ ((packed));
 
 
_______________________________________________
openwrt-commits mailing list
[email protected]
https://lists.openwrt.org/cgi-bin/mailman/listinfo/openwrt-commits

Reply via email to