/* * $Id$ * * Parse MyLoader-style flash partition tables and produce a Linux partition * array to match. * * Copyright (C) 2007 OpenWrt.org * Copyright (C) 2007 Gabor Juhos <juhosg at openwrt.org> * * This file was based on drivers/mtd/redboot.c * Author: Red Hat, Inc. - David Woodhouse <dwmw2@cambridge.redhat.com> * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 as published * by the Free Software Foundation. * */ #include <linux/kernel.h> #include <linux/slab.h> #include <linux/init.h> #include <linux/vmalloc.h> #include <linux/mtd/mtd.h> #include <linux/mtd/partitions.h> #include <linux/byteorder/generic.h> #include <prom/myloader.h> #define NAME_LEN_MAX 20 #define NAME_MYLOADER "MyLoader" #define NAME_PARTITION_TABLE "Partition Table" #define BLOCK_LEN_MIN 0x10000 int parse_myloader_partitions(struct mtd_info *master, struct mtd_partition **pparts, unsigned long origin) { struct mylo_partition_table *tab; struct mylo_partition *part; struct mtd_partition *mtd_parts; struct mtd_partition *mtd_part; int num_parts; int ret, i; size_t retlen; char *names; unsigned long offset; unsigned long blocklen; tab = vmalloc(sizeof(*tab)); if (!tab) { return -ENOMEM; goto out; } blocklen = master->erasesize; if (blocklen < BLOCK_LEN_MIN) blocklen = BLOCK_LEN_MIN; /* Partition Table is always located on the second erase block */ offset = blocklen; printk(KERN_NOTICE "%s: searching for MyLoader partition table at " "offset 0x%lx\n", master->name, offset); ret = master->read(master, offset, sizeof(*tab), &retlen, (void *)tab); if (ret) goto out; if (retlen != sizeof(*tab)) { ret = -EIO; goto out_free_buf; } /* Check for Partition Table magic number */ if (tab->magic != le32_to_cpu(MYLO_MAGIC_PARTITIONS)) { printk(KERN_NOTICE "%s: no MyLoader partition table found\n", master->name); ret = 0; goto out_free_buf; } /* The MyLoader and the Partition Table is always present */ num_parts = 2; /* Detect number of used partitions */ for (i = 0; i < MYLO_MAX_PARTITIONS; i++) { part = &tab->partitions[i]; if (le16_to_cpu(part->type) == PARTITION_TYPE_FREE) continue; num_parts++; } mtd_parts = kzalloc((num_parts * sizeof(*mtd_part) + num_parts * NAME_LEN_MAX), GFP_KERNEL); if (!mtd_parts) { ret = -ENOMEM; goto out_free_buf; } mtd_part = mtd_parts; names = (char *)&mtd_parts[num_parts]; strncpy(names, NAME_MYLOADER, NAME_LEN_MAX-1); mtd_part->name = names; mtd_part->offset = 0; mtd_part->size = blocklen; mtd_part->mask_flags = MTD_WRITEABLE; mtd_part++; names += NAME_LEN_MAX; strncpy(names, NAME_PARTITION_TABLE, NAME_LEN_MAX-1); mtd_part->name = names; mtd_part->offset = blocklen; mtd_part->size = blocklen; mtd_part->mask_flags = MTD_WRITEABLE; mtd_part++; names += NAME_LEN_MAX; for (i = 0; i < MYLO_MAX_PARTITIONS; i++) { part = &tab->partitions[i]; if (le16_to_cpu(part->type) == PARTITION_TYPE_FREE) continue; sprintf(names, "partition%d", i); mtd_part->offset = le32_to_cpu(part->addr); mtd_part->size = le32_to_cpu(part->size); mtd_part->name = names; mtd_part++; names += NAME_LEN_MAX; } *pparts = mtd_parts; ret = num_parts; out_free_buf: vfree(tab); out: return ret; } static struct mtd_part_parser mylo_mtd_parser = { .owner = THIS_MODULE, .parse_fn = parse_myloader_partitions, .name = NAME_MYLOADER, }; static int __init mylo_mtd_parser_init(void) { return register_mtd_parser(&mylo_mtd_parser); } static void __exit mylo_mtd_parser_exit(void) { deregister_mtd_parser(&mylo_mtd_parser); } module_init(mylo_mtd_parser_init); module_exit(mylo_mtd_parser_exit); MODULE_AUTHOR("Gabor Juhos <juhosg at openwrt.org>"); MODULE_DESCRIPTION("Parsing code for MyLoader partition tables"); MODULE_LICENSE("GPL v2");