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