[mirror_ubuntu-zesty-kernel.git] / drivers / mtd / ar7part.c

/*
 * Copyright © 2007 Eugene Konev <ejka@openwrt.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * TI AR7 flash partition table.
 * Based on ar7 map by Felix Fietkau <nbd@openwrt.org>
 *
 */

#include <linux/kernel.h>
#include <linux/slab.h>

#include <linux/mtd/mtd.h>
#include <linux/mtd/partitions.h>
#include <linux/bootmem.h>
#include <linux/module.h>

#include <uapi/linux/magic.h>

#define AR7_PARTS	4
#define ROOT_OFFSET	0xe0000

#define LOADER_MAGIC1	le32_to_cpu(0xfeedfa42)
#define LOADER_MAGIC2	le32_to_cpu(0xfeed1281)

struct ar7_bin_rec {
	unsigned int checksum;
	unsigned int length;
	unsigned int address;
};

static int create_mtd_partitions(struct mtd_info *master,
				 const struct mtd_partition **pparts,
				 struct mtd_part_parser_data *data)
{
	struct ar7_bin_rec header;
	unsigned int offset;
	size_t len;
	unsigned int pre_size = master->erasesize, post_size = 0;
	unsigned int root_offset = ROOT_OFFSET;

	int retries = 10;
	struct mtd_partition *ar7_parts;

	ar7_parts = kzalloc(sizeof(*ar7_parts) * AR7_PARTS, GFP_KERNEL);
	if (!ar7_parts)
		return -ENOMEM;
	ar7_parts[0].name = "loader";
	ar7_parts[0].offset = 0;
	ar7_parts[0].size = master->erasesize;
	ar7_parts[0].mask_flags = MTD_WRITEABLE;

	ar7_parts[1].name = "config";
	ar7_parts[1].offset = 0;
	ar7_parts[1].size = master->erasesize;
	ar7_parts[1].mask_flags = 0;

	do { /* Try 10 blocks starting from master->erasesize */
		offset = pre_size;
		mtd_read(master, offset, sizeof(header), &len,
			 (uint8_t *)&header);
		if (!strncmp((char *)&header, "TIENV0.8", 8))
			ar7_parts[1].offset = pre_size;
		if (header.checksum == LOADER_MAGIC1)
			break;
		if (header.checksum == LOADER_MAGIC2)
			break;
		pre_size += master->erasesize;
	} while (retries--);

	pre_size = offset;

	if (!ar7_parts[1].offset) {
		ar7_parts[1].offset = master->size - master->erasesize;
		post_size = master->erasesize;
	}

	switch (header.checksum) {
	case LOADER_MAGIC1:
		while (header.length) {
			offset += sizeof(header) + header.length;
			mtd_read(master, offset, sizeof(header), &len,
				 (uint8_t *)&header);
		}
		root_offset = offset + sizeof(header) + 4;
		break;
	case LOADER_MAGIC2:
		while (header.length) {
			offset += sizeof(header) + header.length;
			mtd_read(master, offset, sizeof(header), &len,
				 (uint8_t *)&header);
		}
		root_offset = offset + sizeof(header) + 4 + 0xff;
		root_offset &= ~(uint32_t)0xff;
		break;
	default:
		printk(KERN_WARNING "Unknown magic: %08x\n", header.checksum);
		break;
	}

	mtd_read(master, root_offset, sizeof(header), &len, (u8 *)&header);
	if (header.checksum != SQUASHFS_MAGIC) {
		root_offset += master->erasesize - 1;
		root_offset &= ~(master->erasesize - 1);
	}

	ar7_parts[2].name = "linux";
	ar7_parts[2].offset = pre_size;
	ar7_parts[2].size = master->size - pre_size - post_size;
	ar7_parts[2].mask_flags = 0;

	ar7_parts[3].name = "rootfs";
	ar7_parts[3].offset = root_offset;
	ar7_parts[3].size = master->size - root_offset - post_size;
	ar7_parts[3].mask_flags = 0;

	*pparts = ar7_parts;
	return AR7_PARTS;
}

static struct mtd_part_parser ar7_parser = {
	.parse_fn = create_mtd_partitions,
	.name = "ar7part",
};
module_mtd_part_parser(ar7_parser);

MODULE_LICENSE("GPL");
MODULE_AUTHOR(	"Felix Fietkau <nbd@openwrt.org>, "
		"Eugene Konev <ejka@openwrt.org>");
MODULE_DESCRIPTION("MTD partitioning for TI AR7");
Commit	Line	Data
f0797881 MC	1	/*
	2	* Copyright © 2007 Eugene Konev <ejka@openwrt.org>
	3	*
	4	* This program is free software; you can redistribute it and/or modify
	5	* it under the terms of the GNU General Public License as published by
	6	* the Free Software Foundation; either version 2 of the License, or
	7	* (at your option) any later version.
	8	*
	9	* This program is distributed in the hope that it will be useful,
	10	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	11	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	12	* GNU General Public License for more details.
	13	*
	14	* You should have received a copy of the GNU General Public License
	15	* along with this program; if not, write to the Free Software
	16	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
	17	*
	18	* TI AR7 flash partition table.
	19	* Based on ar7 map by Felix Fietkau <nbd@openwrt.org>
	20	*
	21	*/
	22
	23	#include <linux/kernel.h>
	24	#include <linux/slab.h>
	25
	26	#include <linux/mtd/mtd.h>
	27	#include <linux/mtd/partitions.h>
	28	#include <linux/bootmem.h>
a0e5cc58	29	#include <linux/module.h>
f0797881	30
9ff40738 LP	31	#include <uapi/linux/magic.h>
9ff40738 LP	32
f0797881 MC	33	#define AR7_PARTS 4
	34	#define ROOT_OFFSET 0xe0000
	35
	36	#define LOADER_MAGIC1 le32_to_cpu(0xfeedfa42)
	37	#define LOADER_MAGIC2 le32_to_cpu(0xfeed1281)
	38
	39	struct ar7_bin_rec {
	40	unsigned int checksum;
	41	unsigned int length;
	42	unsigned int address;
	43	};
	44
f0797881	45	static int create_mtd_partitions(struct mtd_info *master,
b9adf469	46	const struct mtd_partition **pparts,
c7975330	47	struct mtd_part_parser_data *data)
f0797881 MC	48	{
f0797881 MC	49	struct ar7_bin_rec header;
986ee013 DW	50	unsigned int offset;
986ee013 DW	51	size_t len;
f0797881 MC	52	unsigned int pre_size = master->erasesize, post_size = 0;
	53	unsigned int root_offset = ROOT_OFFSET;
	54
	55	int retries = 10;
17b536cc	56	struct mtd_partition *ar7_parts;
f0797881	57
17b536cc AN	58	ar7_parts = kzalloc(sizeof(ar7_parts) AR7_PARTS, GFP_KERNEL);
	59	if (!ar7_parts)
	60	return -ENOMEM;
f0797881 MC	61	ar7_parts[0].name = "loader";
	62	ar7_parts[0].offset = 0;
	63	ar7_parts[0].size = master->erasesize;
	64	ar7_parts[0].mask_flags = MTD_WRITEABLE;
	65
	66	ar7_parts[1].name = "config";
	67	ar7_parts[1].offset = 0;
	68	ar7_parts[1].size = master->erasesize;
	69	ar7_parts[1].mask_flags = 0;
	70
	71	do { /* Try 10 blocks starting from master->erasesize */
	72	offset = pre_size;
329ad399 AB	73	mtd_read(master, offset, sizeof(header), &len,
329ad399 AB	74	(uint8_t *)&header);
f0797881 MC	75	if (!strncmp((char *)&header, "TIENV0.8", 8))
	76	ar7_parts[1].offset = pre_size;
	77	if (header.checksum == LOADER_MAGIC1)
	78	break;
	79	if (header.checksum == LOADER_MAGIC2)
	80	break;
	81	pre_size += master->erasesize;
	82	} while (retries--);
	83
	84	pre_size = offset;
	85
	86	if (!ar7_parts[1].offset) {
	87	ar7_parts[1].offset = master->size - master->erasesize;
	88	post_size = master->erasesize;
	89	}
	90
	91	switch (header.checksum) {
	92	case LOADER_MAGIC1:
	93	while (header.length) {
	94	offset += sizeof(header) + header.length;
329ad399 AB	95	mtd_read(master, offset, sizeof(header), &len,
329ad399 AB	96	(uint8_t *)&header);
f0797881 MC	97	}
	98	root_offset = offset + sizeof(header) + 4;
	99	break;
	100	case LOADER_MAGIC2:
	101	while (header.length) {
	102	offset += sizeof(header) + header.length;
329ad399 AB	103	mtd_read(master, offset, sizeof(header), &len,
329ad399 AB	104	(uint8_t *)&header);
f0797881 MC	105	}
f0797881 MC	106	root_offset = offset + sizeof(header) + 4 + 0xff;
986ee013	107	root_offset &= ~(uint32_t)0xff;
f0797881 MC	108	break;
	109	default:
	110	printk(KERN_WARNING "Unknown magic: %08x\n", header.checksum);
	111	break;
	112	}
	113
329ad399	114	mtd_read(master, root_offset, sizeof(header), &len, (u8 *)&header);
f0797881 MC	115	if (header.checksum != SQUASHFS_MAGIC) {
	116	root_offset += master->erasesize - 1;
	117	root_offset &= ~(master->erasesize - 1);
	118	}
	119
	120	ar7_parts[2].name = "linux";
	121	ar7_parts[2].offset = pre_size;
	122	ar7_parts[2].size = master->size - pre_size - post_size;
	123	ar7_parts[2].mask_flags = 0;
	124
	125	ar7_parts[3].name = "rootfs";
	126	ar7_parts[3].offset = root_offset;
	127	ar7_parts[3].size = master->size - root_offset - post_size;
	128	ar7_parts[3].mask_flags = 0;
	129
	130	*pparts = ar7_parts;
	131	return AR7_PARTS;
	132	}
	133
	134	static struct mtd_part_parser ar7_parser = {
f0797881 MC	135	.parse_fn = create_mtd_partitions,
	136	.name = "ar7part",
	137	};
b8f70bad	138	module_mtd_part_parser(ar7_parser);
f0797881 MC	139
	140	MODULE_LICENSE("GPL");
	141	MODULE_AUTHOR( "Felix Fietkau <nbd@openwrt.org>, "
	142	"Eugene Konev <ejka@openwrt.org>");
	143	MODULE_DESCRIPTION("MTD partitioning for TI AR7");