1f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce/* 2f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * Copyright © 2007 Eugene Konev <ejka@openwrt.org> 3f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * 4f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * This program is free software; you can redistribute it and/or modify 5f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * it under the terms of the GNU General Public License as published by 6f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * the Free Software Foundation; either version 2 of the License, or 7f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * (at your option) any later version. 8f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * 9f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * This program is distributed in the hope that it will be useful, 10f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * but WITHOUT ANY WARRANTY; without even the implied warranty of 11f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * GNU General Public License for more details. 13f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * 14f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * You should have received a copy of the GNU General Public License 15f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * along with this program; if not, write to the Free Software 16f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * 18f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * TI AR7 flash partition table. 19f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * Based on ar7 map by Felix Fietkau <nbd@openwrt.org> 20f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce * 21f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce */ 22f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 23f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#include <linux/kernel.h> 24f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#include <linux/slab.h> 25f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 26f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#include <linux/mtd/mtd.h> 27f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#include <linux/mtd/partitions.h> 28f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#include <linux/bootmem.h> 29a0e5cc581b3fc0e0a909e3cab48d9ec286c2a276Paul Gortmaker#include <linux/module.h> 30f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 319ff407385441744ea4618f109535014c4b651f8aLuka Perkov#include <uapi/linux/magic.h> 329ff407385441744ea4618f109535014c4b651f8aLuka Perkov 33f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#define AR7_PARTS 4 34f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#define ROOT_OFFSET 0xe0000 35f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 36f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#define LOADER_MAGIC1 le32_to_cpu(0xfeedfa42) 37f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#define LOADER_MAGIC2 le32_to_cpu(0xfeed1281) 38f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 39f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocestruct ar7_bin_rec { 40f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int checksum; 41f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int length; 42f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int address; 43f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce}; 44f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 45f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocestatic int create_mtd_partitions(struct mtd_info *master, 46f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce struct mtd_partition **pparts, 47c7975330154af17aecc167b33ca866b6b3d98918Dmitry Eremin-Solenikov struct mtd_part_parser_data *data) 48f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce{ 49f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce struct ar7_bin_rec header; 50986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse unsigned int offset; 51986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse size_t len; 52f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int pre_size = master->erasesize, post_size = 0; 53f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int root_offset = ROOT_OFFSET; 54f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 55f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce int retries = 10; 5617b536cc43bcf2afcc20b4812f93a895881b5f4fAtsushi Nemoto struct mtd_partition *ar7_parts; 57f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 5817b536cc43bcf2afcc20b4812f93a895881b5f4fAtsushi Nemoto ar7_parts = kzalloc(sizeof(*ar7_parts) * AR7_PARTS, GFP_KERNEL); 5917b536cc43bcf2afcc20b4812f93a895881b5f4fAtsushi Nemoto if (!ar7_parts) 6017b536cc43bcf2afcc20b4812f93a895881b5f4fAtsushi Nemoto return -ENOMEM; 61f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[0].name = "loader"; 62f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[0].offset = 0; 63f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[0].size = master->erasesize; 64f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[0].mask_flags = MTD_WRITEABLE; 65f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 66f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].name = "config"; 67f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].offset = 0; 68f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].size = master->erasesize; 69f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].mask_flags = 0; 70f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 71f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce do { /* Try 10 blocks starting from master->erasesize */ 72f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce offset = pre_size; 73329ad399a9b3adf52c90637b21ca029fcf7f8795Artem Bityutskiy mtd_read(master, offset, sizeof(header), &len, 74329ad399a9b3adf52c90637b21ca029fcf7f8795Artem Bityutskiy (uint8_t *)&header); 75f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (!strncmp((char *)&header, "TIENV0.8", 8)) 76f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].offset = pre_size; 77f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (header.checksum == LOADER_MAGIC1) 78f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 79f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (header.checksum == LOADER_MAGIC2) 80f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 81f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce pre_size += master->erasesize; 82f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } while (retries--); 83f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 84f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce pre_size = offset; 85f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 86f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (!ar7_parts[1].offset) { 87f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].offset = master->size - master->erasesize; 88f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce post_size = master->erasesize; 89f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 90f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 91f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce switch (header.checksum) { 92f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce case LOADER_MAGIC1: 93f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce while (header.length) { 94f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce offset += sizeof(header) + header.length; 95329ad399a9b3adf52c90637b21ca029fcf7f8795Artem Bityutskiy mtd_read(master, offset, sizeof(header), &len, 96329ad399a9b3adf52c90637b21ca029fcf7f8795Artem Bityutskiy (uint8_t *)&header); 97f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 98f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce root_offset = offset + sizeof(header) + 4; 99f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 100f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce case LOADER_MAGIC2: 101f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce while (header.length) { 102f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce offset += sizeof(header) + header.length; 103329ad399a9b3adf52c90637b21ca029fcf7f8795Artem Bityutskiy mtd_read(master, offset, sizeof(header), &len, 104329ad399a9b3adf52c90637b21ca029fcf7f8795Artem Bityutskiy (uint8_t *)&header); 105f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 106f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce root_offset = offset + sizeof(header) + 4 + 0xff; 107986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse root_offset &= ~(uint32_t)0xff; 108f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 109f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce default: 110f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce printk(KERN_WARNING "Unknown magic: %08x\n", header.checksum); 111f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 112f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 113f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 114329ad399a9b3adf52c90637b21ca029fcf7f8795Artem Bityutskiy mtd_read(master, root_offset, sizeof(header), &len, (u8 *)&header); 115f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (header.checksum != SQUASHFS_MAGIC) { 116f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce root_offset += master->erasesize - 1; 117f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce root_offset &= ~(master->erasesize - 1); 118f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 119f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 120f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[2].name = "linux"; 121f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[2].offset = pre_size; 122f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[2].size = master->size - pre_size - post_size; 123f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[2].mask_flags = 0; 124f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 125f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[3].name = "rootfs"; 126f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[3].offset = root_offset; 127f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[3].size = master->size - root_offset - post_size; 128f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[3].mask_flags = 0; 129f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 130f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce *pparts = ar7_parts; 131f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce return AR7_PARTS; 132f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce} 133f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 134f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocestatic struct mtd_part_parser ar7_parser = { 135f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce .owner = THIS_MODULE, 136f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce .parse_fn = create_mtd_partitions, 137f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce .name = "ar7part", 138f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce}; 139f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 140f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocestatic int __init ar7_parser_init(void) 141f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce{ 1426e14a61d412eb87ef7bdcec8b08a95bead771a78Axel Lin register_mtd_parser(&ar7_parser); 1436e14a61d412eb87ef7bdcec8b08a95bead771a78Axel Lin return 0; 144f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce} 145f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 146422f3890a68333cfd39f2133cf7255ed067e92e6Lubomir Rintelstatic void __exit ar7_parser_exit(void) 147422f3890a68333cfd39f2133cf7255ed067e92e6Lubomir Rintel{ 148422f3890a68333cfd39f2133cf7255ed067e92e6Lubomir Rintel deregister_mtd_parser(&ar7_parser); 149422f3890a68333cfd39f2133cf7255ed067e92e6Lubomir Rintel} 150422f3890a68333cfd39f2133cf7255ed067e92e6Lubomir Rintel 151f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocemodule_init(ar7_parser_init); 152422f3890a68333cfd39f2133cf7255ed067e92e6Lubomir Rintelmodule_exit(ar7_parser_exit); 153f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 154f0797881d59ab93d7d92c55411e0573977d909d4Matteo CroceMODULE_LICENSE("GPL"); 155f0797881d59ab93d7d92c55411e0573977d909d4Matteo CroceMODULE_AUTHOR( "Felix Fietkau <nbd@openwrt.org>, " 156f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce "Eugene Konev <ejka@openwrt.org>"); 157f0797881d59ab93d7d92c55411e0573977d909d4Matteo CroceMODULE_DESCRIPTION("MTD partitioning for TI AR7"); 158