ar71xx: use dynamically allocated partitions in wrt160nl_part

SVN-Revision: 29970

1 files changed, 14 insertions, 4 deletions

diff --git a/target/linux/ar71xx/files/drivers/mtd/wrt160nl_part.c b/target/linux/ar71xx/files/drivers/mtd/wrt160nl_part.c
index d979365..1f1879a 100644
--- a/target/linux/ar71xx/files/drivers/mtd/wrt160nl_part.c
+++ b/target/linux/ar71xx/files/drivers/mtd/wrt160nl_part.c
@@ -77,8 +77,6 @@ struct wrt160nl_header {
 	struct uimage_header	uimage;
 } __attribute__ ((packed));
 
-static struct mtd_partition trx_parts[TRX_PARTS];
-
 #define WRT160NL_UBOOT_LEN	0x40000
 #define WRT160NL_ART_LEN	0x10000
 #define WRT160NL_NVRAM_LEN	0x10000
@@ -90,6 +88,7 @@ static int wrt160nl_parse_partitions(struct mtd_info *master,
 	struct wrt160nl_header *header;
 	struct trx_header *theader;
 	struct uimage_header *uheader;
+	struct mtd_partition *trx_parts;
 	size_t retlen;
 	unsigned int kernel_len;
 	unsigned int uboot_len;
@@ -101,10 +100,17 @@ static int wrt160nl_parse_partitions(struct mtd_info *master,
 	nvram_len = max_t(unsigned int, master->erasesize, WRT160NL_NVRAM_LEN);
 	art_len = max_t(unsigned int, master->erasesize, WRT160NL_ART_LEN);
 
+	trx_parts = kzalloc(TRX_PARTS * sizeof(struct mtd_partition),
+			    GFP_KERNEL);
+	if (!trx_parts) {
+		ret = -ENOMEM;
+		goto out;
+	}
+
 	header = vmalloc(sizeof(*header));
 	if (!header) {
 		return -ENOMEM;
-		goto out;
+		goto free_parts;
 	}
 
 	ret = master->read(master, uboot_len, sizeof(*header),
@@ -169,11 +175,15 @@ static int wrt160nl_parse_partitions(struct mtd_info *master,
 	trx_parts[5].size = master->size - uboot_len - nvram_len - art_len;
 	trx_parts[5].mask_flags = 0;
 
+	vfree(header);
+
 	*pparts = trx_parts;
-	ret = TRX_PARTS;
+	return TRX_PARTS;
 
 free_hdr:
 	vfree(header);
+free_parts:
+	kfree(trx_parts);
 out:
 	return ret;
 }