2 * Copyright (C) 2011 Gabor Juhos <juhosg@openwrt.org>
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License version 2 as published
6 * by the Free Software Foundation.
10 #include <linux/kernel.h>
11 #include <linux/slab.h>
12 #include <linux/vmalloc.h>
14 #include <linux/mtd/mtd.h>
15 #include <linux/mtd/partitions.h>
17 #define TPLINK_NUM_PARTS 5
18 #define TPLINK_HEADER_V1 0x01000000
21 #define TPLINK_ART_LEN 0x10000
22 #define TPLINK_KERNEL_OFFS 0x20000
24 struct tplink_fw_header {
25 uint32_t version; /* header version */
28 uint32_t hw_id; /* hardware id */
29 uint32_t hw_rev; /* hardware revision */
31 uint8_t md5sum1[MD5SUM_LEN];
33 uint8_t md5sum2[MD5SUM_LEN];
35 uint32_t kernel_la; /* kernel load address */
36 uint32_t kernel_ep; /* kernel entry point */
37 uint32_t fw_length; /* total length of the firmware */
38 uint32_t kernel_ofs; /* kernel data offset */
39 uint32_t kernel_len; /* kernel data length */
40 uint32_t rootfs_ofs; /* rootfs data offset */
41 uint32_t rootfs_len; /* rootfs data length */
42 uint32_t boot_ofs; /* bootloader data offset */
43 uint32_t boot_len; /* bootloader data length */
45 } __attribute__ ((packed));
47 static struct tplink_fw_header *
48 tplink_read_header(struct mtd_info *mtd, size_t offset)
50 struct tplink_fw_header *header;
56 header = vmalloc(sizeof(*header));
60 header_len = sizeof(struct tplink_fw_header);
61 ret = mtd->read(mtd, offset, header_len, &retlen,
62 (unsigned char *) header);
66 if (retlen != header_len)
70 t = be32_to_cpu(header->version);
71 if (t != TPLINK_HEADER_V1)
74 t = be32_to_cpu(header->kernel_ofs);
86 static int tplink_parse_partitions(struct mtd_info *master,
87 struct mtd_partition **pparts,
90 struct mtd_partition *parts;
91 struct tplink_fw_header *header;
98 nr_parts = TPLINK_NUM_PARTS;
99 parts = kzalloc(nr_parts * sizeof(struct mtd_partition), GFP_KERNEL);
105 offset = TPLINK_KERNEL_OFFS;
107 header = tplink_read_header(master, offset);
109 pr_notice("%s: no TP-Link header found\n", master->name);
114 rootfs_offset = offset + be32_to_cpu(header->rootfs_ofs);
115 art_offset = master->size - TPLINK_ART_LEN;
117 parts[0].name = "u-boot";
119 parts[0].size = offset;
120 parts[0].mask_flags = MTD_WRITEABLE;
122 parts[1].name = "kernel";
123 parts[1].offset = offset;
124 parts[1].size = rootfs_offset - offset;
126 parts[2].name = "rootfs";
127 parts[2].offset = rootfs_offset;
128 parts[2].size = art_offset - rootfs_offset;
130 parts[3].name = "art";
131 parts[3].offset = art_offset;
132 parts[3].size = TPLINK_ART_LEN;
133 parts[3].mask_flags = MTD_WRITEABLE;
135 parts[4].name = "firmware";
136 parts[4].offset = offset;
137 parts[4].size = art_offset - offset;
151 static struct mtd_part_parser tplink_parser = {
152 .owner = THIS_MODULE,
153 .parse_fn = tplink_parse_partitions,
157 static int __init tplink_parser_init(void)
159 return register_mtd_parser(&tplink_parser);
162 module_init(tplink_parser_init);
164 MODULE_LICENSE("GPL v2");
165 MODULE_AUTHOR("Gabor Juhos <juhosg@openwrt.org>");