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
25 #define WRG_MAGIC 0x20040220
27 struct wrgg03_header
{
40 } __attribute__ ((packed
));
50 } __attribute__ ((packed
));
53 static int mtdsplit_parse_wrgg(struct mtd_info
*master
,
54 const struct mtd_partition
**pparts
,
55 struct mtd_part_parser_data
*data
)
57 struct wrgg03_header hdr
;
58 size_t hdr_len
, retlen
, kernel_ent_size
;
60 struct mtd_partition
*parts
;
61 enum mtdsplit_part_type type
;
64 hdr_len
= sizeof(hdr
);
65 err
= mtd_read(master
, 0, hdr_len
, &retlen
, (void *) &hdr
);
69 if (retlen
!= hdr_len
)
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
);
82 if (kernel_ent_size
> master
->size
)
86 * The size in the header covers the rootfs as well.
87 * Start the search from an arbitrary offset.
89 err
= mtd_find_rootfs_from(master
, WRGG_MIN_ROOTFS_OFFS
,
90 master
->size
, &rootfs_offset
, &type
);
94 parts
= kzalloc(WRGG_NR_PARTS
* sizeof(*parts
), GFP_KERNEL
);
98 parts
[0].name
= KERNEL_PART_NAME
;
100 parts
[0].size
= rootfs_offset
;
102 parts
[1].name
= ROOTFS_PART_NAME
;
103 parts
[1].offset
= rootfs_offset
;
104 parts
[1].size
= master
->size
- rootfs_offset
;
107 return WRGG_NR_PARTS
;
110 static const struct of_device_id mtdsplit_wrgg_of_match_table
[] = {
111 { .compatible
= "wrg" },
114 MODULE_DEVICE_TABLE(of
, mtdsplit_wrgg_of_match_table
);
116 static struct mtd_part_parser mtdsplit_wrgg_parser
= {
117 .owner
= THIS_MODULE
,
119 .of_match_table
= mtdsplit_wrgg_of_match_table
,
120 .parse_fn
= mtdsplit_parse_wrgg
,
121 .type
= MTD_PARSER_TYPE_FIRMWARE
,
124 static int __init
mtdsplit_wrgg_init(void)
126 register_mtd_parser(&mtdsplit_wrgg_parser
);
131 subsys_initcall(mtdsplit_wrgg_init
);