ar7part.c revision 17b536cc43bcf2afcc20b4812f93a895881b5f4f
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> 29f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#include <linux/magic.h> 30f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 31f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#define AR7_PARTS 4 32f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#define ROOT_OFFSET 0xe0000 33f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 34f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#define LOADER_MAGIC1 le32_to_cpu(0xfeedfa42) 35f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce#define LOADER_MAGIC2 le32_to_cpu(0xfeed1281) 36f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 37986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse#ifndef SQUASHFS_MAGIC 38986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse#define SQUASHFS_MAGIC 0x73717368 39986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse#endif 40986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse 41f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocestruct ar7_bin_rec { 42f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int checksum; 43f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int length; 44f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int address; 45f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce}; 46f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 47f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocestatic int create_mtd_partitions(struct mtd_info *master, 48f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce struct mtd_partition **pparts, 49f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned long origin) 50f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce{ 51f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce struct ar7_bin_rec header; 52986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse unsigned int offset; 53986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse size_t len; 54f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int pre_size = master->erasesize, post_size = 0; 55f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce unsigned int root_offset = ROOT_OFFSET; 56f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 57f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce int retries = 10; 5817b536cc43bcf2afcc20b4812f93a895881b5f4fAtsushi Nemoto struct mtd_partition *ar7_parts; 59f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 6017b536cc43bcf2afcc20b4812f93a895881b5f4fAtsushi Nemoto ar7_parts = kzalloc(sizeof(*ar7_parts) * AR7_PARTS, GFP_KERNEL); 6117b536cc43bcf2afcc20b4812f93a895881b5f4fAtsushi Nemoto if (!ar7_parts) 6217b536cc43bcf2afcc20b4812f93a895881b5f4fAtsushi Nemoto return -ENOMEM; 63f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[0].name = "loader"; 64f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[0].offset = 0; 65f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[0].size = master->erasesize; 66f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[0].mask_flags = MTD_WRITEABLE; 67f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 68f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].name = "config"; 69f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].offset = 0; 70f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].size = master->erasesize; 71f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].mask_flags = 0; 72f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 73f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce do { /* Try 10 blocks starting from master->erasesize */ 74f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce offset = pre_size; 75f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce master->read(master, offset, 76986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse sizeof(header), &len, (uint8_t *)&header); 77f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (!strncmp((char *)&header, "TIENV0.8", 8)) 78f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].offset = pre_size; 79f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (header.checksum == LOADER_MAGIC1) 80f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 81f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (header.checksum == LOADER_MAGIC2) 82f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 83f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce pre_size += master->erasesize; 84f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } while (retries--); 85f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 86f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce pre_size = offset; 87f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 88f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (!ar7_parts[1].offset) { 89f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[1].offset = master->size - master->erasesize; 90f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce post_size = master->erasesize; 91f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 92f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 93f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce switch (header.checksum) { 94f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce case LOADER_MAGIC1: 95f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce while (header.length) { 96f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce offset += sizeof(header) + header.length; 97f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce master->read(master, offset, sizeof(header), 98986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse &len, (uint8_t *)&header); 99f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 100f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce root_offset = offset + sizeof(header) + 4; 101f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 102f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce case LOADER_MAGIC2: 103f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce while (header.length) { 104f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce offset += sizeof(header) + header.length; 105f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce master->read(master, offset, sizeof(header), 106986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse &len, (uint8_t *)&header); 107f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 108f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce root_offset = offset + sizeof(header) + 4 + 0xff; 109986ee0139a91ab8b6b07d29d7a112c8033b5f8e0David Woodhouse root_offset &= ~(uint32_t)0xff; 110f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 111f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce default: 112f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce printk(KERN_WARNING "Unknown magic: %08x\n", header.checksum); 113f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce break; 114f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 115f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 116f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce master->read(master, root_offset, 117f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce sizeof(header), &len, (u8 *)&header); 118f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce if (header.checksum != SQUASHFS_MAGIC) { 119f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce root_offset += master->erasesize - 1; 120f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce root_offset &= ~(master->erasesize - 1); 121f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce } 122f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 123f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[2].name = "linux"; 124f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[2].offset = pre_size; 125f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[2].size = master->size - pre_size - post_size; 126f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[2].mask_flags = 0; 127f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 128f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[3].name = "rootfs"; 129f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[3].offset = root_offset; 130f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[3].size = master->size - root_offset - post_size; 131f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce ar7_parts[3].mask_flags = 0; 132f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 133f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce *pparts = ar7_parts; 134f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce return AR7_PARTS; 135f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce} 136f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 137f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocestatic struct mtd_part_parser ar7_parser = { 138f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce .owner = THIS_MODULE, 139f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce .parse_fn = create_mtd_partitions, 140f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce .name = "ar7part", 141f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce}; 142f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 143f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocestatic int __init ar7_parser_init(void) 144f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce{ 145f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce return register_mtd_parser(&ar7_parser); 146f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce} 147f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 148f0797881d59ab93d7d92c55411e0573977d909d4Matteo Crocemodule_init(ar7_parser_init); 149f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce 150f0797881d59ab93d7d92c55411e0573977d909d4Matteo CroceMODULE_LICENSE("GPL"); 151f0797881d59ab93d7d92c55411e0573977d909d4Matteo CroceMODULE_AUTHOR( "Felix Fietkau <nbd@openwrt.org>, " 152f0797881d59ab93d7d92c55411e0573977d909d4Matteo Croce "Eugene Konev <ejka@openwrt.org>"); 153f0797881d59ab93d7d92c55411e0573977d909d4Matteo CroceMODULE_DESCRIPTION("MTD partitioning for TI AR7"); 154