123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- #include <linux/module.h>
- #include <linux/init.h>
- #include <linux/kernel.h>
- #include <linux/slab.h>
- #include <linux/mtd/mtd.h>
- #include <linux/mtd/partitions.h>
- #include <linux/byteorder/generic.h>
- #include "mtdsplit.h"
- #define TPLINK_NR_PARTS 2
- #define TPLINK_MIN_ROOTFS_OFFS 0x80000
- #define MD5SUM_LEN 16
- struct fw_v1 {
- char vendor_name[24];
- char fw_version[36];
- uint32_t hw_id;
- uint32_t hw_rev;
- uint32_t unk1;
- uint8_t md5sum1[MD5SUM_LEN];
- uint32_t unk2;
- uint8_t md5sum2[MD5SUM_LEN];
- uint32_t unk3;
- uint32_t kernel_la;
- uint32_t kernel_ep;
- uint32_t fw_length;
- uint32_t kernel_ofs;
- uint32_t kernel_len;
- uint32_t rootfs_ofs;
- uint32_t rootfs_len;
- uint32_t boot_ofs;
- uint32_t boot_len;
- uint8_t pad[360];
- } __attribute__ ((packed));
- struct fw_v2 {
- char fw_version[48];
- uint32_t hw_id;
- uint32_t hw_rev;
- uint32_t unk1;
- uint8_t md5sum1[MD5SUM_LEN];
- uint32_t unk2;
- uint8_t md5sum2[MD5SUM_LEN];
- uint32_t unk3;
- uint32_t kernel_la;
- uint32_t kernel_ep;
- uint32_t fw_length;
- uint32_t kernel_ofs;
- uint32_t kernel_len;
- uint32_t rootfs_ofs;
- uint32_t rootfs_len;
- uint32_t boot_ofs;
- uint32_t boot_len;
- uint16_t unk4;
- uint8_t sver_hi;
- uint8_t sver_lo;
- uint8_t unk5;
- uint8_t ver_hi;
- uint8_t ver_mid;
- uint8_t ver_lo;
- uint8_t pad[364];
- } __attribute__ ((packed));
- struct tplink_fw_header {
- uint32_t version;
- union {
- struct fw_v1 v1;
- struct fw_v2 v2;
- };
- };
- static int mtdsplit_parse_tplink(struct mtd_info *master,
- struct mtd_partition **pparts,
- struct mtd_part_parser_data *data)
- {
- struct tplink_fw_header hdr;
- size_t hdr_len, retlen, kernel_size;
- size_t rootfs_offset;
- struct mtd_partition *parts;
- int err;
- hdr_len = sizeof(hdr);
- err = mtd_read(master, 0, hdr_len, &retlen, (void *) &hdr);
- if (err)
- return err;
- if (retlen != hdr_len)
- return -EIO;
- switch (le32_to_cpu(hdr.version)) {
- case 1:
- if (be32_to_cpu(hdr.v1.kernel_ofs) != sizeof(hdr))
- return -EINVAL;
- kernel_size = sizeof(hdr) + be32_to_cpu(hdr.v1.kernel_len);
- break;
- case 2:
- case 3:
- if (be32_to_cpu(hdr.v2.kernel_ofs) != sizeof(hdr))
- return -EINVAL;
- kernel_size = sizeof(hdr) + be32_to_cpu(hdr.v2.kernel_len);
- break;
- default:
- return -EINVAL;
- }
- if (kernel_size > master->size)
- return -EINVAL;
-
- err = mtd_check_rootfs_magic(master, kernel_size, NULL);
- if (!err) {
- rootfs_offset = kernel_size;
- } else {
-
- err = mtd_find_rootfs_from(master, TPLINK_MIN_ROOTFS_OFFS,
- master->size, &rootfs_offset, NULL);
- if (err)
- return err;
- }
- parts = kzalloc(TPLINK_NR_PARTS * sizeof(*parts), GFP_KERNEL);
- if (!parts)
- return -ENOMEM;
- parts[0].name = KERNEL_PART_NAME;
- parts[0].offset = 0;
- parts[0].size = rootfs_offset;
- parts[1].name = ROOTFS_PART_NAME;
- parts[1].offset = rootfs_offset;
- parts[1].size = master->size - rootfs_offset;
- *pparts = parts;
- return TPLINK_NR_PARTS;
- }
- static struct mtd_part_parser mtdsplit_tplink_parser = {
- .owner = THIS_MODULE,
- .name = "tplink-fw",
- .parse_fn = mtdsplit_parse_tplink,
- .type = MTD_PARSER_TYPE_FIRMWARE,
- };
- static int __init mtdsplit_tplink_init(void)
- {
- register_mtd_parser(&mtdsplit_tplink_parser);
- return 0;
- }
- subsys_initcall(mtdsplit_tplink_init);
|