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>
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.
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>
22 #define WRGG_NR_PARTS 2
23 #define WRGG_MIN_ROOTFS_OFFS 0x80000 /* 512KiB */
24 #define WRGG03_MAGIC 0x20080321
26 struct wrgg03_header
{
39 } __attribute__ ((packed
));
42 static int mtdsplit_parse_wrgg(struct mtd_info
*master
,
43 const struct mtd_partition
**pparts
,
44 struct mtd_part_parser_data
*data
)
46 struct wrgg03_header hdr
;
47 size_t hdr_len
, retlen
, kernel_ent_size
;
49 struct mtd_partition
*parts
;
50 enum mtdsplit_part_type type
;
53 hdr_len
= sizeof(hdr
);
54 err
= mtd_read(master
, 0, hdr_len
, &retlen
, (void *) &hdr
);
58 if (retlen
!= hdr_len
)
62 if (le32_to_cpu(hdr
.magic1
) != WRGG03_MAGIC
)
65 kernel_ent_size
= hdr_len
+ be32_to_cpu(hdr
.size
);
67 if (kernel_ent_size
> master
->size
)
71 * The size in the header covers the rootfs as well.
72 * Start the search from an arbitrary offset.
74 err
= mtd_find_rootfs_from(master
, WRGG_MIN_ROOTFS_OFFS
,
75 master
->size
, &rootfs_offset
, &type
);
79 parts
= kzalloc(WRGG_NR_PARTS
* sizeof(*parts
), GFP_KERNEL
);
83 parts
[0].name
= KERNEL_PART_NAME
;
85 parts
[0].size
= rootfs_offset
;
87 parts
[1].name
= ROOTFS_PART_NAME
;
88 parts
[1].offset
= rootfs_offset
;
89 parts
[1].size
= master
->size
- rootfs_offset
;
95 static struct mtd_part_parser mtdsplit_wrgg_parser
= {
98 .parse_fn
= mtdsplit_parse_wrgg
,
99 .type
= MTD_PARSER_TYPE_FIRMWARE
,
102 static int __init
mtdsplit_wrgg_init(void)
104 register_mtd_parser(&mtdsplit_wrgg_parser
);
109 subsys_initcall(mtdsplit_wrgg_init
);