moko nand blocksize is 128k

[openwrt.git] / target / linux / ar7 / files / drivers / mtd / ar7part.c

diff --git a/target/linux/ar7/files/drivers/mtd/ar7part.c b/target/linux/ar7/files/drivers/mtd/ar7part.c
index 587bb31..3d160d4 100644 (file)
--- a/target/linux/ar7/files/drivers/mtd/ar7part.c
+++ b/target/linux/ar7/files/drivers/mtd/ar7part.c
@@ -1,24 +1,22 @@
 /*
- * $Id$
- * 
- * Copyright (C) 2007 OpenWrt.org
- * 
+ * Copyright (C) 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.
+ * Based on ar7 map by Felix Fietkau <nbd@openwrt.org>
  *
  */
 
@@ -28,7 +26,13 @@
 #include <linux/mtd/mtd.h>
 #include <linux/mtd/partitions.h>
 #include <linux/bootmem.h>
-#include <linux/squashfs_fs.h>
+#include <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;
@@ -36,19 +40,18 @@ struct ar7_bin_rec {
        unsigned int address;
 };
 
-static struct mtd_partition ar7_parts[5];
+static struct mtd_partition ar7_parts[AR7_PARTS];
 
-static int create_mtd_partitions(struct mtd_info *master, 
-                                struct mtd_partition **pparts, 
+static int create_mtd_partitions(struct mtd_info *master,
+                                struct mtd_partition **pparts,
                                 unsigned long origin)
 {
        struct ar7_bin_rec header;
        unsigned int offset, len;
-       unsigned int pre_size = master->erasesize, post_size = 0,
-               root_offset = 0xe0000;
-       int retries = 10;
+       unsigned int pre_size = master->erasesize, post_size = 0;
+       unsigned int root_offset = ROOT_OFFSET;
 
-       printk("Parsing AR7 partition map...\n");
+       int retries = 10;
 
        ar7_parts[0].name = "loader";
        ar7_parts[0].offset = 0;
@@ -60,14 +63,15 @@ static int create_mtd_partitions(struct mtd_info *master,
        ar7_parts[1].size = master->erasesize;
        ar7_parts[1].mask_flags = 0;
 
-       do {
+       do { /* Try 10 blocks starting from master->erasesize */
                offset = pre_size;
-               master->read(master, offset, sizeof(header), &len, (u_char *)&header);
+               master->read(master, offset,
+                       sizeof(header), &len, (u8 *)&header);
                if (!strncmp((char *)&header, "TIENV0.8", 8))
                        ar7_parts[1].offset = pre_size;
-               if (header.checksum == 0xfeedfa42)
+               if (header.checksum == LOADER_MAGIC1)
                        break;
-               if (header.checksum == 0xfeed1281)
+               if (header.checksum == LOADER_MAGIC2)
                        break;
                pre_size += master->erasesize;
        } while (retries--);
@@ -80,29 +84,30 @@ static int create_mtd_partitions(struct mtd_info *master,
        }
 
        switch (header.checksum) {
-       case 0xfeedfa42:
+       case LOADER_MAGIC1:
                while (header.length) {
                        offset += sizeof(header) + header.length;
                        master->read(master, offset, sizeof(header),
-                                    &len, (u_char *)&header); 
+                                    &len, (u8 *)&header);
                }
                root_offset = offset + sizeof(header) + 4;
                break;
-       case 0xfeed1281:
+       case LOADER_MAGIC2:
                while (header.length) {
                        offset += sizeof(header) + header.length;
                        master->read(master, offset, sizeof(header),
-                                    &len, (u_char *)&header); 
+                                    &len, (u8 *)&header);
                }
                root_offset = offset + sizeof(header) + 4 + 0xff;
                root_offset &= ~(u32)0xff;
                break;
        default:
-               printk("Unknown magic: %08x\n", header.checksum);
+               printk(KERN_WARNING "Unknown magic: %08x\n", header.checksum);
                break;
        }
 
-       master->read(master, root_offset, sizeof(header), &len, (u_char *)&header);
+       master->read(master, root_offset,
+               sizeof(header), &len, (u8 *)&header);
        if (header.checksum != SQUASHFS_MAGIC) {
                root_offset += master->erasesize - 1;
                root_offset &= ~(master->erasesize - 1);
@@ -119,7 +124,7 @@ static int create_mtd_partitions(struct mtd_info *master,
        ar7_parts[3].mask_flags = 0;
 
        *pparts = ar7_parts;
-       return 4;
+       return AR7_PARTS;
 }
 
 static struct mtd_part_parser ar7_parser = {
@@ -136,5 +141,6 @@ static int __init ar7_parser_init(void)
 module_init(ar7_parser_init);
 
 MODULE_LICENSE("GPL");
-MODULE_AUTHOR("Felix Fietkau, Eugene Konev");
+MODULE_AUTHOR( "Felix Fietkau <nbd@openwrt.org>, "
+               "Eugene Konev <ejka@openwrt.org>");
 MODULE_DESCRIPTION("MTD partitioning for TI AR7");