ar71xx: add mtd partition parser for the TP-Link boards

SVN-Revision: 29415

diff --git a/target/linux/ar71xx/config-2.6.39 b/target/linux/ar71xx/config-2.6.39
index 1837deb6765a8db30d31658d36016e218ff7d229..8cca8e2000c71a897b52e783b3f50fcbaf25e37f 100644 (file)
--- a/target/linux/ar71xx/config-2.6.39
+++ b/target/linux/ar71xx/config-2.6.39
@@ -155,6 +155,7 @@ CONFIG_MTD_M25P80=y
 CONFIG_MTD_MYLOADER_PARTS=y
 CONFIG_MTD_REDBOOT_DIRECTORY_BLOCK=-2
 CONFIG_MTD_REDBOOT_PARTS=y
+CONFIG_MTD_TPLINK_PARTS=y
 CONFIG_MTD_WRT160NL_PARTS=y
 CONFIG_MYLOADER=y
 CONFIG_NEED_DMA_MAP_STATE=y

diff --git a/target/linux/ar71xx/files/drivers/mtd/tplinkpart.c b/target/linux/ar71xx/files/drivers/mtd/tplinkpart.c
new file mode 100644 (file)
index 0000000..7b2ac7e
--- /dev/null
+++ b/target/linux/ar71xx/files/drivers/mtd/tplinkpart.c
@@ -0,0 +1,165 @@
+/*
+ * Copyright (C) 2011 Gabor Juhos <juhosg@openwrt.org>
+ *
+ * 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/vmalloc.h>
+
+#include <linux/mtd/mtd.h>
+#include <linux/mtd/partitions.h>
+
+#define TPLINK_NUM_PARTS       5
+#define TPLINK_HEADER_V1       0x01000000
+#define MD5SUM_LEN             16
+
+#define TPLINK_ART_LEN         0x10000
+#define TPLINK_KERNEL_OFFS     0x20000
+
+struct tplink_fw_header {
+       uint32_t        version;        /* header version */
+       char            vendor_name[24];
+       char            fw_version[36];
+       uint32_t        hw_id;          /* hardware id */
+       uint32_t        hw_rev;         /* hardware revision */
+       uint32_t        unk1;
+       uint8_t         md5sum1[MD5SUM_LEN];
+       uint32_t        unk2;
+       uint8_t         md5sum2[MD5SUM_LEN];
+       uint32_t        unk3;
+       uint32_t        kernel_la;      /* kernel load address */
+       uint32_t        kernel_ep;      /* kernel entry point */
+       uint32_t        fw_length;      /* total length of the firmware */
+       uint32_t        kernel_ofs;     /* kernel data offset */
+       uint32_t        kernel_len;     /* kernel data length */
+       uint32_t        rootfs_ofs;     /* rootfs data offset */
+       uint32_t        rootfs_len;     /* rootfs data length */
+       uint32_t        boot_ofs;       /* bootloader data offset */
+       uint32_t        boot_len;       /* bootloader data length */
+       uint8_t         pad[360];
+} __attribute__ ((packed));
+
+static struct tplink_fw_header *
+tplink_read_header(struct mtd_info *mtd, size_t offset)
+{
+       struct tplink_fw_header *header;
+       size_t header_len;
+       size_t retlen;
+       int ret;
+       u32 t;
+
+       header = vmalloc(sizeof(*header));
+       if (!header)
+               goto err;
+
+       header_len = sizeof(struct tplink_fw_header);
+       ret = mtd->read(mtd, offset, header_len, &retlen,
+                       (unsigned char *) header);
+       if (ret)
+               goto err_free_header;
+
+       if (retlen != header_len)
+               goto err_free_header;
+
+       /* sanity checks */
+       t = be32_to_cpu(header->version);
+       if (t != TPLINK_HEADER_V1)
+               goto err_free_header;
+
+       t = be32_to_cpu(header->kernel_ofs);
+       if (t != header_len)
+               goto err_free_header;
+
+       return header;
+
+err_free_header:
+       vfree(header);
+err:
+       return NULL;
+}
+
+static int tplink_parse_partitions(struct mtd_info *master,
+                                  struct mtd_partition **pparts,
+                                  unsigned long origin)
+{
+       struct mtd_partition *parts;
+       struct tplink_fw_header *header;
+       int nr_parts;
+       size_t offset;
+       size_t art_offset;
+       size_t rootfs_offset;
+       int ret;
+
+       nr_parts = TPLINK_NUM_PARTS;
+       parts = kzalloc(nr_parts * sizeof(struct mtd_partition), GFP_KERNEL);
+       if (!parts) {
+               ret = -ENOMEM;
+               goto err;
+       }
+
+       offset = TPLINK_KERNEL_OFFS;
+
+       header = tplink_read_header(master, offset);
+       if (!header) {
+               pr_notice("%s: no TP-Link header found\n", master->name);
+               ret = -ENODEV;
+               goto err_free_parts;
+       }
+
+       rootfs_offset = offset + be32_to_cpu(header->rootfs_ofs);
+       art_offset = master->size - TPLINK_ART_LEN;
+
+       parts[0].name = "u-boot";
+       parts[0].offset = 0;
+       parts[0].size = offset;
+       parts[0].mask_flags = MTD_WRITEABLE;
+
+       parts[1].name = "kernel";
+       parts[1].offset = offset;
+       parts[1].size = rootfs_offset - offset;
+
+       parts[2].name = "rootfs";
+       parts[2].offset = rootfs_offset;
+       parts[2].size = art_offset - rootfs_offset;
+
+       parts[3].name = "art";
+       parts[3].offset = art_offset;
+       parts[3].size = TPLINK_ART_LEN;
+       parts[3].mask_flags = MTD_WRITEABLE;
+
+       parts[4].name = "firmware";
+       parts[4].offset = offset;
+       parts[4].size = art_offset - offset;
+
+       vfree(header);
+
+       *pparts = parts;
+       return nr_parts;
+
+err_free_parts:
+       kfree(parts);
+err:
+       *pparts = NULL;
+       return ret;
+}
+
+static struct mtd_part_parser tplink_parser = {
+       .owner          = THIS_MODULE,
+       .parse_fn       = tplink_parse_partitions,
+       .name           = "tp-link",
+};
+
+static int __init tplink_parser_init(void)
+{
+       return register_mtd_parser(&tplink_parser);
+}
+
+module_init(tplink_parser_init);
+
+MODULE_LICENSE("GPL v2");
+MODULE_AUTHOR("Gabor Juhos <juhosg@openwrt.org>");

diff --git a/target/linux/ar71xx/patches-2.6.39/112-mtd-tp-link-partition-parser.patch b/target/linux/ar71xx/patches-2.6.39/112-mtd-tp-link-partition-parser.patch
new file mode 100644 (file)
index 0000000..90bdfc9
--- /dev/null
+++ b/target/linux/ar71xx/patches-2.6.39/112-mtd-tp-link-partition-parser.patch
@@ -0,0 +1,25 @@
+--- a/drivers/mtd/Kconfig
++++ b/drivers/mtd/Kconfig
+@@ -198,6 +198,12 @@ config MTD_MYLOADER_PARTS
+         You will still need the parsing functions to be called by the driver
+         for your particular device. It won't happen automatically.
+ 
++config MTD_TPLINK_PARTS
++      tristate "TP-Link AR7XXX/AR9XXX partitioning support"
++      depends on MTD_PARTITIONS && ATHEROS_AR71XX
++      ---help---
++        TBD.
++
+ comment "User Modules And Translation Layers"
+ 
+ config MTD_CHAR
+--- a/drivers/mtd/Makefile
++++ b/drivers/mtd/Makefile
+@@ -14,6 +14,7 @@ obj-$(CONFIG_MTD_AFS_PARTS)  += afs.o
+ obj-$(CONFIG_MTD_AR7_PARTS)   += ar7part.o
+ obj-$(CONFIG_MTD_WRT160NL_PARTS) += wrt160nl_part.o
+ obj-$(CONFIG_MTD_MYLOADER_PARTS) += myloader.o
++obj-$(CONFIG_MTD_TPLINK_PARTS)        += tplinkpart.o
+ 
+ # 'Users' - code which presents functionality to userspace.
+ obj-$(CONFIG_MTD_CHAR)                += mtdchar.o