aboutsummaryrefslogtreecommitdiffstats
path: root/target/linux/ar71xx/files/drivers/mtd
diff options
context:
space:
mode:
Diffstat (limited to 'target/linux/ar71xx/files/drivers/mtd')
-rw-r--r--target/linux/ar71xx/files/drivers/mtd/tltpart.c206
1 files changed, 206 insertions, 0 deletions
diff --git a/target/linux/ar71xx/files/drivers/mtd/tltpart.c b/target/linux/ar71xx/files/drivers/mtd/tltpart.c
new file mode 100644
index 0000000000..92f558d778
--- /dev/null
+++ b/target/linux/ar71xx/files/drivers/mtd/tltpart.c
@@ -0,0 +1,206 @@
+/*
+ * 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/module.h>
+#include <linux/slab.h>
+#include <linux/vmalloc.h>
+#include <linux/magic.h>
+
+#include <linux/mtd/mtd.h>
+#include <linux/mtd/partitions.h>
+
+#define TLT_NUM_PARTS 7
+#define TLT_HEADER_V1 0x01000000
+#define MD5SUM_LEN 16
+
+#define TLT_UBOOT_LEN 0x20000
+#define TLT_CONFIG_LEN 0x10000
+#define TLT_ART_LEN 0x10000
+#define EVENTLOG_LEN 0x90000
+
+struct tlt_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 tlt_fw_header *
+tlt_read_header(struct mtd_info *mtd, size_t offset)
+{
+ struct tlt_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 tlt_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 != TLT_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 tlt_check_rootfs_magic(struct mtd_info *mtd, size_t offset)
+{
+ u32 magic;
+ size_t retlen;
+ int ret;
+
+ ret = mtd_read(mtd, offset, sizeof(magic), &retlen,
+ (unsigned char *) &magic);
+ if (ret)
+ return ret;
+
+ if (retlen != sizeof(magic))
+ return -EIO;
+
+ if (le32_to_cpu(magic) != SQUASHFS_MAGIC &&
+ magic != 0x19852003)
+ return -EINVAL;
+
+ return 0;
+}
+
+static int tlt_parse_partitions(struct mtd_info *master,
+ struct mtd_partition **pparts,
+ struct mtd_part_parser_data *data)
+{
+ struct mtd_partition *parts;
+ struct tlt_fw_header *header;
+ int nr_parts;
+ size_t kernel_offset;
+ size_t rootfs_offset;
+ size_t squashfs_offset;
+ int ret;
+
+ nr_parts = TLT_NUM_PARTS;
+ parts = kzalloc(nr_parts * sizeof(struct mtd_partition), GFP_KERNEL);
+ if (!parts) {
+ ret = -ENOMEM;
+ goto err;
+ }
+
+ kernel_offset = TLT_UBOOT_LEN + TLT_CONFIG_LEN + TLT_ART_LEN;
+
+ header = tlt_read_header(master, kernel_offset);
+ if (!header) {
+ pr_notice("%s: no TP-Link header found\n", master->name);
+ ret = -ENODEV;
+ goto err_free_parts;
+ }
+
+ squashfs_offset = kernel_offset + sizeof(struct tlt_fw_header) +
+ be32_to_cpu(header->kernel_len);
+
+ ret = tlt_check_rootfs_magic(master, squashfs_offset);
+ if (ret == 0)
+ rootfs_offset = squashfs_offset;
+ else
+ rootfs_offset = kernel_offset + be32_to_cpu(header->rootfs_ofs);
+
+ vfree(header);
+
+ parts[0].name = "u-boot";
+ parts[0].offset = 0;
+ parts[0].size = TLT_UBOOT_LEN;
+ parts[0].mask_flags = MTD_WRITEABLE;
+
+ parts[1].name = "config";
+ parts[1].offset = TLT_UBOOT_LEN;
+ parts[1].size = TLT_CONFIG_LEN;
+
+ parts[2].name = "art";
+ parts[2].offset = TLT_UBOOT_LEN + TLT_CONFIG_LEN;
+ parts[2].size = TLT_ART_LEN;
+ parts[2].mask_flags = MTD_WRITEABLE;
+
+ parts[3].name = "kernel";
+ parts[3].offset = kernel_offset;
+ parts[3].size = rootfs_offset - kernel_offset;
+
+ parts[4].name = "rootfs";
+ parts[4].offset = rootfs_offset;
+ parts[4].size = master->size - rootfs_offset - EVENTLOG_LEN;
+
+ parts[5].name = "firmware";
+ parts[5].offset = kernel_offset;
+ parts[5].size = master->size - kernel_offset - EVENTLOG_LEN;
+
+ parts[6].name = "event-log";
+ parts[6].offset = master->size - EVENTLOG_LEN;
+ parts[6].size = EVENTLOG_LEN;
+
+ *pparts = parts;
+ return nr_parts;
+
+err_free_parts:
+ kfree(parts);
+err:
+ *pparts = NULL;
+ return ret;
+}
+
+static struct mtd_part_parser tlt_parser = {
+ .owner = THIS_MODULE,
+ .parse_fn = tlt_parse_partitions,
+ .name = "tlt",
+};
+
+static int __init tlt_parser_init(void)
+{
+ register_mtd_parser(&tlt_parser);
+
+ return 0;
+}
+
+module_init(tlt_parser_init);
+
+MODULE_LICENSE("GPL v2");
+MODULE_AUTHOR("Gabor Juhos <juhosg@openwrt.org>");