X-Git-Url: http://git.rohieb.name/openwrt.git/blobdiff_plain/17c2fb2fc826af433e21fde62306cb776ae5c287..a808a8e5276ff8e3012d1b54b9c3f29744889620:/target/linux/ar71xx/files/drivers/mtd/tplinkpart.c

diff --git a/target/linux/ar71xx/files/drivers/mtd/tplinkpart.c b/target/linux/ar71xx/files/drivers/mtd/tplinkpart.c
index 7b2ac7e40..6f47f8435 100644
--- a/target/linux/ar71xx/files/drivers/mtd/tplinkpart.c
+++ b/target/linux/ar71xx/files/drivers/mtd/tplinkpart.c
@@ -10,6 +10,7 @@
 #include <linux/kernel.h>
 #include <linux/slab.h>
 #include <linux/vmalloc.h>
+#include <linux/magic.h>
 
 #include <linux/mtd/mtd.h>
 #include <linux/mtd/partitions.h>
@@ -83,6 +84,27 @@ err:
 	return NULL;
 }
 
+static int tplink_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 tplink_parse_partitions(struct mtd_info *master,
 				   struct mtd_partition **pparts,
 				   unsigned long origin)
@@ -93,6 +115,7 @@ static int tplink_parse_partitions(struct mtd_info *master,
 	size_t offset;
 	size_t art_offset;
 	size_t rootfs_offset;
+	size_t squashfs_offset;
 	int ret;
 
 	nr_parts = TPLINK_NUM_PARTS;
@@ -111,7 +134,15 @@ static int tplink_parse_partitions(struct mtd_info *master,
 		goto err_free_parts;
 	}
 
-	rootfs_offset = offset + be32_to_cpu(header->rootfs_ofs);
+	squashfs_offset = offset + sizeof(struct tplink_fw_header) +
+			  be32_to_cpu(header->kernel_len);
+
+	ret = tplink_check_rootfs_magic(master, squashfs_offset);
+	if (ret == 0)
+		rootfs_offset = squashfs_offset;
+	else
+		rootfs_offset = offset + be32_to_cpu(header->rootfs_ofs);
+
 	art_offset = master->size - TPLINK_ART_LEN;
 
 	parts[0].name = "u-boot";