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/module.h>
12 #include <linux/slab.h>
13 #include <linux/vmalloc.h>
14 #include <linux/magic.h>
16 #include <linux/mtd/mtd.h>
17 #include <linux/mtd/partitions.h>
19 #define TPLINK_NUM_PARTS 5
20 #define TPLINK_HEADER_V1 0x01000000
21 #define TPLINK_HEADER_V2 0x02000000
24 #define TPLINK_ART_LEN 0x10000
25 #define TPLINK_KERNEL_OFFS 0x20000
26 #define TPLINK_64K_KERNEL_OFFS 0x10000
28 struct tplink_fw_header
{
29 uint32_t version
; /* header version */
32 uint32_t hw_id
; /* hardware id */
33 uint32_t hw_rev
; /* hardware revision */
35 uint8_t md5sum1
[MD5SUM_LEN
];
37 uint8_t md5sum2
[MD5SUM_LEN
];
39 uint32_t kernel_la
; /* kernel load address */
40 uint32_t kernel_ep
; /* kernel entry point */
41 uint32_t fw_length
; /* total length of the firmware */
42 uint32_t kernel_ofs
; /* kernel data offset */
43 uint32_t kernel_len
; /* kernel data length */
44 uint32_t rootfs_ofs
; /* rootfs data offset */
45 uint32_t rootfs_len
; /* rootfs data length */
46 uint32_t boot_ofs
; /* bootloader data offset */
47 uint32_t boot_len
; /* bootloader data length */
49 } __attribute__ ((packed
));
51 static struct tplink_fw_header
*
52 tplink_read_header(struct mtd_info
*mtd
, size_t offset
)
54 struct tplink_fw_header
*header
;
60 header
= vmalloc(sizeof(*header
));
64 header_len
= sizeof(struct tplink_fw_header
);
65 ret
= mtd_read(mtd
, offset
, header_len
, &retlen
,
66 (unsigned char *) header
);
70 if (retlen
!= header_len
)
74 t
= be32_to_cpu(header
->version
);
75 if ((t
!= TPLINK_HEADER_V1
) && (t
!= TPLINK_HEADER_V2
))
78 t
= be32_to_cpu(header
->kernel_ofs
);
90 static int tplink_check_rootfs_magic(struct mtd_info
*mtd
, size_t offset
)
96 ret
= mtd_read(mtd
, offset
, sizeof(magic
), &retlen
,
97 (unsigned char *) &magic
);
101 if (retlen
!= sizeof(magic
))
104 if (le32_to_cpu(magic
) != SQUASHFS_MAGIC
&&
111 static int tplink_parse_partitions_offset(struct mtd_info
*master
,
112 struct mtd_partition
**pparts
,
113 struct mtd_part_parser_data
*data
,
116 struct mtd_partition
*parts
;
117 struct tplink_fw_header
*header
;
120 size_t rootfs_offset
;
121 size_t squashfs_offset
;
124 nr_parts
= TPLINK_NUM_PARTS
;
125 parts
= kzalloc(nr_parts
* sizeof(struct mtd_partition
), GFP_KERNEL
);
131 header
= tplink_read_header(master
, offset
);
133 pr_notice("%s: no TP-Link header found\n", master
->name
);
138 squashfs_offset
= offset
+ sizeof(struct tplink_fw_header
) +
139 be32_to_cpu(header
->kernel_len
);
141 ret
= tplink_check_rootfs_magic(master
, squashfs_offset
);
143 rootfs_offset
= squashfs_offset
;
145 rootfs_offset
= offset
+ be32_to_cpu(header
->rootfs_ofs
);
147 art_offset
= master
->size
- TPLINK_ART_LEN
;
149 parts
[0].name
= "u-boot";
151 parts
[0].size
= offset
;
152 parts
[0].mask_flags
= MTD_WRITEABLE
;
154 parts
[1].name
= "kernel";
155 parts
[1].offset
= offset
;
156 parts
[1].size
= rootfs_offset
- offset
;
158 parts
[2].name
= "rootfs";
159 parts
[2].offset
= rootfs_offset
;
160 parts
[2].size
= art_offset
- rootfs_offset
;
162 parts
[3].name
= "art";
163 parts
[3].offset
= art_offset
;
164 parts
[3].size
= TPLINK_ART_LEN
;
165 parts
[3].mask_flags
= MTD_WRITEABLE
;
167 parts
[4].name
= "firmware";
168 parts
[4].offset
= offset
;
169 parts
[4].size
= art_offset
- offset
;
183 static int tplink_parse_partitions(struct mtd_info
*master
,
184 struct mtd_partition
**pparts
,
185 struct mtd_part_parser_data
*data
)
187 return tplink_parse_partitions_offset(master
, pparts
, data
,
191 static int tplink_parse_64k_partitions(struct mtd_info
*master
,
192 struct mtd_partition
**pparts
,
193 struct mtd_part_parser_data
*data
)
195 return tplink_parse_partitions_offset(master
, pparts
, data
,
196 TPLINK_64K_KERNEL_OFFS
);
199 static struct mtd_part_parser tplink_parser
= {
200 .owner
= THIS_MODULE
,
201 .parse_fn
= tplink_parse_partitions
,
205 static struct mtd_part_parser tplink_64k_parser
= {
206 .owner
= THIS_MODULE
,
207 .parse_fn
= tplink_parse_64k_partitions
,
208 .name
= "tp-link-64k",
211 static int __init
tplink_parser_init(void)
213 register_mtd_parser(&tplink_parser
);
214 register_mtd_parser(&tplink_64k_parser
);
219 module_init(tplink_parser_init
);
221 MODULE_LICENSE("GPL v2");
222 MODULE_AUTHOR("Gabor Juhos <juhosg@openwrt.org>");