target/linux/generic/files/drivers/mtd/mtdsplit/mtdsplit_wrgg.c

   1 /*
   2  *  Copyright (C) 2013 Gabor Juhos <juhosg@openwrt.org>
   3  *  Copyright (C) 2014 Felix Fietkau <nbd@nbd.name>
   4  *  Copyright (C) 2016 Stijn Tintel <stijn@linux-ipv6.be>
   5  *
   6  *  This program is free software; you can redistribute it and/or modify it
   7  *  under the terms of the GNU General Public License version 2 as published
   8  *  by the Free Software Foundation.
   9  *
  10  */
  11
  12 #include <linux/module.h>
  13 #include <linux/init.h>
  14 #include <linux/kernel.h>
  15 #include <linux/slab.h>
  16 #include <linux/mtd/mtd.h>
  17 #include <linux/mtd/partitions.h>
  18 #include <linux/byteorder/generic.h>
  19
  20 #include "mtdsplit.h"
  21
  22 #define WRGG_NR_PARTS           2
  23 #define WRGG_MIN_ROOTFS_OFFS    0x80000 /* 512KiB */
  24 #define WRGG03_MAGIC            0x20080321
  25 #define WRG_MAGIC               0x20040220
  26
  27 struct wrgg03_header {
  28         char            signature[32];
  29         uint32_t        magic1;
  30         uint32_t        magic2;
  31         char            version[16];
  32         char            model[16];
  33         uint32_t        flag[2];
  34         uint32_t        reserve[2];
  35         char            buildno[16];
  36         uint32_t        size;
  37         uint32_t        offset;
  38         char            devname[32];
  39         char            digest[16];
  40 } __attribute__ ((packed));
  41
  42 struct wrg_header {
  43         char            signature[32];
  44         uint32_t        magic1;
  45         uint32_t        magic2;
  46         uint32_t        size;
  47         uint32_t        offset;
  48         char            devname[32];
  49         char            digest[16];
  50 } __attribute__ ((packed));
  51
  52
  53 static int mtdsplit_parse_wrgg(struct mtd_info *master,
  54                                const struct mtd_partition **pparts,
  55                                struct mtd_part_parser_data *data)
  56 {
  57         struct wrgg03_header hdr;
  58         size_t hdr_len, retlen, kernel_ent_size;
  59         size_t rootfs_offset;
  60         struct mtd_partition *parts;
  61         enum mtdsplit_part_type type;
  62         int err;
  63
  64         hdr_len = sizeof(hdr);
  65         err = mtd_read(master, 0, hdr_len, &retlen, (void *) &hdr);
  66         if (err)
  67                 return err;
  68
  69         if (retlen != hdr_len)
  70                 return -EIO;
  71
  72         /* sanity checks */
  73         if (le32_to_cpu(hdr.magic1) == WRGG03_MAGIC) {
  74                 kernel_ent_size = hdr_len + be32_to_cpu(hdr.size);
  75         } else if (le32_to_cpu(hdr.magic1) == WRG_MAGIC) {
  76                 kernel_ent_size = sizeof(struct wrg_header) + le32_to_cpu(
  77                                   ((struct wrg_header*)&hdr)->size);
  78         } else {
  79                 return -EINVAL;
  80         }
  81
  82         if (kernel_ent_size > master->size)
  83                 return -EINVAL;
  84
  85         /*
  86          * The size in the header covers the rootfs as well.
  87          * Start the search from an arbitrary offset.
  88          */
  89         err = mtd_find_rootfs_from(master, WRGG_MIN_ROOTFS_OFFS,
  90                                    master->size, &rootfs_offset, &type);
  91         if (err)
  92                 return err;
  93
  94         parts = kzalloc(WRGG_NR_PARTS * sizeof(*parts), GFP_KERNEL);
  95         if (!parts)
  96                 return -ENOMEM;
  97
  98         parts[0].name = KERNEL_PART_NAME;
  99         parts[0].offset = 0;
 100         parts[0].size = rootfs_offset;
 101
 102         parts[1].name = ROOTFS_PART_NAME;
 103         parts[1].offset = rootfs_offset;
 104         parts[1].size = master->size - rootfs_offset;
 105
 106         *pparts = parts;
 107         return WRGG_NR_PARTS;
 108 }
 109
 110 static struct mtd_part_parser mtdsplit_wrgg_parser = {
 111         .owner = THIS_MODULE,
 112         .name = "wrgg-fw",
 113         .parse_fn = mtdsplit_parse_wrgg,
 114         .type = MTD_PARSER_TYPE_FIRMWARE,
 115 };
 116
 117 static int __init mtdsplit_wrgg_init(void)
 118 {
 119         register_mtd_parser(&mtdsplit_wrgg_parser);
 120
 121         return 0;
 122 }
 123
 124 subsys_initcall(mtdsplit_wrgg_init);