4 * Copyright (C) 2007 OpenWrt.org
5 * Copyright (C) Gabor Juhos <juhosg at openwrt.org>
7 * This program is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version 2
10 * of the License, or (at your option) any later version.
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, write to the
19 * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
20 * Boston, MA 02110-1301, USA.
23 #include <linux/module.h>
24 #include <linux/types.h>
25 #include <linux/kernel.h>
26 #include <linux/slab.h>
27 #include <linux/list.h>
28 #include <linux/kmod.h>
29 #include <linux/root_dev.h>
31 #include <linux/mtd/mtd.h>
32 #include <linux/mtd/partitions.h>
34 #include <linux/byteorder/generic.h>
36 #define PFX "trxsplit: "
38 #define TRX_MAGIC 0x30524448 /* "HDR0" */
40 #define TRX_MAX_LEN 0x3A0000
41 #define TRX_NO_HEADER 0x1 /* do not write TRX header */
42 #define TRX_GZ_FILES 0x2 /* contains individual gzip files */
43 #define TRX_MAX_OFFSET 3
44 #define TRX_MIN_KERNEL_SIZE 256*1024
47 u32 magic; /* "HDR0" */
48 u32 len; /* Length of file including header */
49 u32 crc32; /* 32-bit CRC from flag_version to end of file */
50 u32 flag_version; /* 0:15 flags, 16:31 version */
51 u32 offsets[TRX_MAX_OFFSET]; /* Offsets of partitions */
54 #define TRX_ALIGN 0x1000
56 static int trx_nr_parts;
57 static unsigned long trx_offset;
58 static struct mtd_info *trx_mtd;
59 static struct mtd_partition trx_parts[TRX_MAX_OFFSET];
60 static struct trx_header trx_hdr;
62 static int trxsplit_refresh_partitions(struct mtd_info *mtd);
64 static int trxsplit_checktrx(struct mtd_info *mtd, unsigned long offset)
69 err = mtd->read(mtd, offset, sizeof(trx_hdr), &retlen,
72 printk(KERN_ALERT PFX "unable to read from '%s'\n", mtd->name);
76 if (retlen != sizeof(trx_hdr)) {
77 printk(KERN_ALERT PFX "reading failed on '%s'\n", mtd->name);
81 trx_hdr.magic = le32_to_cpu(trx_hdr.magic);
82 trx_hdr.len = le32_to_cpu(trx_hdr.len);
83 trx_hdr.crc32 = le32_to_cpu(trx_hdr.crc32);
84 trx_hdr.flag_version = le32_to_cpu(trx_hdr.flag_version);
85 trx_hdr.offsets[0] = le32_to_cpu(trx_hdr.offsets[0]);
86 trx_hdr.offsets[1] = le32_to_cpu(trx_hdr.offsets[1]);
87 trx_hdr.offsets[2] = le32_to_cpu(trx_hdr.offsets[2]);
90 if (trx_hdr.magic != TRX_MAGIC)
93 if (trx_hdr.len > mtd->size - offset)
96 /* TODO: add crc32 checking too? */
104 static void trxsplit_findtrx(struct mtd_info *mtd)
106 unsigned long offset;
109 printk(KERN_INFO PFX "searching TRX header in '%s'\n", mtd->name);
112 for (offset = 0; offset < mtd->size; offset += TRX_ALIGN) {
113 err = trxsplit_checktrx(mtd, offset);
121 printk(KERN_INFO PFX "TRX header found at 0x%lX\n", offset);
127 static void trxsplit_create_partitions(struct mtd_info *mtd)
129 struct mtd_partition *part = trx_parts;
133 for (i = 0; i < TRX_MAX_OFFSET; i++) {
134 part = &trx_parts[i];
135 if (trx_hdr.offsets[i] == 0)
137 part->offset = trx_offset + trx_hdr.offsets[i];
141 for (i = 0; i < trx_nr_parts-1; i++)
142 trx_parts[i].size = trx_parts[i+1].offset - trx_parts[i].offset;
144 trx_parts[i].size = mtd->size - trx_parts[i].offset;
147 part = &trx_parts[i];
148 if (part->size < TRX_MIN_KERNEL_SIZE) {
149 part->name = "loader";
153 part = &trx_parts[i];
154 part->name = "kernel";
157 part = &trx_parts[i];
158 part->name = "rootfs";
160 err = add_mtd_partitions(mtd, trx_parts, trx_nr_parts);
162 printk(KERN_ALERT PFX "adding TRX partitions failed\n");
166 mtd->refresh_device = trxsplit_refresh_partitions;
169 static int trxsplit_refresh_partitions(struct mtd_info *mtd)
171 printk(KERN_INFO PFX "refreshing TRX partitions in '%s' (%d,%d)\n",
172 mtd->name, MTD_BLOCK_MAJOR, mtd->index);
174 /* remove old partitions */
175 del_mtd_partitions(mtd);
177 trxsplit_findtrx(mtd);
181 trxsplit_create_partitions(trx_mtd);
188 static void __init trxsplit_add_mtd(struct mtd_info *mtd)
190 if (mtd->type != MTD_NORFLASH) {
191 printk(KERN_INFO PFX "'%s' is not a NOR flash, skipped\n",
197 trxsplit_findtrx(mtd);
200 static void __init trxsplit_remove_mtd(struct mtd_info *mtd)
205 static struct mtd_notifier trxsplit_notifier __initdata = {
206 .add = trxsplit_add_mtd,
207 .remove = trxsplit_remove_mtd,
210 static void __init trxsplit_scan(void)
212 register_mtd_user(&trxsplit_notifier);
213 unregister_mtd_user(&trxsplit_notifier);
216 static int __init trxsplit_init(void)
221 printk(KERN_INFO PFX "creating TRX partitions in '%s' "
222 "(%d,%d)\n", trx_mtd->name, MTD_BLOCK_MAJOR,
224 trxsplit_create_partitions(trx_mtd);
230 late_initcall(trxsplit_init);