2 * Parse MyLoader-style flash partition tables and produce a Linux partition
5 * Copyright (C) 2007-2009 Gabor Juhos <juhosg@openwrt.org>
7 * This file was based on drivers/mtd/redboot.c
8 * Author: Red Hat, Inc. - David Woodhouse <dwmw2@cambridge.redhat.com>
10 * This program is free software; you can redistribute it and/or modify it
11 * under the terms of the GNU General Public License version 2 as published
12 * by the Free Software Foundation.
16 #include <linux/kernel.h>
17 #include <linux/module.h>
18 #include <linux/version.h>
19 #include <linux/slab.h>
20 #include <linux/init.h>
21 #include <linux/vmalloc.h>
22 #include <linux/mtd/mtd.h>
23 #include <linux/mtd/partitions.h>
24 #include <linux/byteorder/generic.h>
25 #include <linux/myloader.h>
27 #define BLOCK_LEN_MIN 0x10000
28 #define PART_NAME_LEN 32
31 struct mylo_partition_table tab;
32 char names[MYLO_MAX_PARTITIONS][PART_NAME_LEN];
35 #if (LINUX_VERSION_CODE >= KERNEL_VERSION(3,2,0))
36 static int myloader_parse_partitions(struct mtd_info *master,
37 struct mtd_partition **pparts,
38 struct mtd_part_parser_data *data)
40 static int myloader_parse_partitions(struct mtd_info *master,
41 struct mtd_partition **pparts,
45 struct part_data *buf;
46 struct mylo_partition_table *tab;
47 struct mylo_partition *part;
48 struct mtd_partition *mtd_parts;
49 struct mtd_partition *mtd_part;
55 unsigned long blocklen;
57 buf = vmalloc(sizeof(*buf));
64 blocklen = master->erasesize;
65 if (blocklen < BLOCK_LEN_MIN)
66 blocklen = BLOCK_LEN_MIN;
70 /* Find the partition table */
71 for (i = 0; i < 4; i++, offset += blocklen) {
72 printk(KERN_DEBUG "%s: searching for MyLoader partition table"
73 " at offset 0x%lx\n", master->name, offset);
75 ret = mtd_read(master, offset, sizeof(*buf), &retlen,
80 if (retlen != sizeof(*buf)) {
85 /* Check for Partition Table magic number */
86 if (tab->magic == le32_to_cpu(MYLO_MAGIC_PARTITIONS))
91 if (tab->magic != le32_to_cpu(MYLO_MAGIC_PARTITIONS)) {
92 printk(KERN_DEBUG "%s: no MyLoader partition table found\n",
98 /* The MyLoader and the Partition Table is always present */
101 /* Detect number of used partitions */
102 for (i = 0; i < MYLO_MAX_PARTITIONS; i++) {
103 part = &tab->partitions[i];
105 if (le16_to_cpu(part->type) == PARTITION_TYPE_FREE)
111 mtd_parts = kzalloc((num_parts * sizeof(*mtd_part) +
112 num_parts * PART_NAME_LEN), GFP_KERNEL);
119 mtd_part = mtd_parts;
120 names = (char *)&mtd_parts[num_parts];
122 strncpy(names, "myloader", PART_NAME_LEN);
123 mtd_part->name = names;
124 mtd_part->offset = 0;
125 mtd_part->size = offset;
126 mtd_part->mask_flags = MTD_WRITEABLE;
128 names += PART_NAME_LEN;
130 strncpy(names, "partition_table", PART_NAME_LEN);
131 mtd_part->name = names;
132 mtd_part->offset = offset;
133 mtd_part->size = blocklen;
134 mtd_part->mask_flags = MTD_WRITEABLE;
136 names += PART_NAME_LEN;
138 for (i = 0; i < MYLO_MAX_PARTITIONS; i++) {
139 part = &tab->partitions[i];
141 if (le16_to_cpu(part->type) == PARTITION_TYPE_FREE)
144 if ((buf->names[i][0]) && (buf->names[i][0] != '\xff'))
145 strncpy(names, buf->names[i], PART_NAME_LEN);
147 snprintf(names, PART_NAME_LEN, "partition%d", i);
149 mtd_part->offset = le32_to_cpu(part->addr);
150 mtd_part->size = le32_to_cpu(part->size);
151 mtd_part->name = names;
153 names += PART_NAME_LEN;
165 static struct mtd_part_parser myloader_mtd_parser = {
166 .owner = THIS_MODULE,
167 .parse_fn = myloader_parse_partitions,
171 static int __init myloader_mtd_parser_init(void)
173 return register_mtd_parser(&myloader_mtd_parser);
176 static void __exit myloader_mtd_parser_exit(void)
178 deregister_mtd_parser(&myloader_mtd_parser);
181 module_init(myloader_mtd_parser_init);
182 module_exit(myloader_mtd_parser_exit);
184 MODULE_AUTHOR("Gabor Juhos <juhosg@openwrt.org>");
185 MODULE_DESCRIPTION("Parsing code for MyLoader partition tables");
186 MODULE_LICENSE("GPL v2");