[adm5120] fix flash driver, it should work on RB150 as well

git-svn-id: svn://svn.openwrt.org/openwrt/trunk@9418 3c298f89-4303-0410-b956-a3cf2f4a3e73

1 files changed, 12 insertions, 6 deletions

diff --git a/target/linux/adm5120/files/drivers/mtd/maps/adm5120-flash.c b/target/linux/adm5120/files/drivers/mtd/maps/adm5120-flash.c
index e07fb96d2..20f67d625 100644
--- a/target/linux/adm5120/files/drivers/mtd/maps/adm5120-flash.c
+++ b/target/linux/adm5120/files/drivers/mtd/maps/adm5120-flash.c
@@ -244,7 +244,7 @@ static int adm5120_flash_initinfo(struct adm5120_flash_info *info,
 	struct map_info *map = &info->amap.map;
 	struct adm5120_flash_platform_data *pdata = dev->dev.platform_data;
 	struct flash_desc *fdesc;
-	u32 t;
+	u32 t = 0;
 
 	map->name = dev->dev.bus_id;
 
@@ -255,12 +255,18 @@ static int adm5120_flash_initinfo(struct adm5120_flash_info *info,
 
 	fdesc = &flash_descs[dev->id];
 
-	/* get memory window size */
-	t = SW_READ_REG(MEMCTRL) >> fdesc->srs_shift;
-	t &= MEMCTRL_SRS_MASK;
-	info->amap.window_size = flash_sizes[t];
+	if (pdata)
+		info->amap.window_size = pdata->window_size;
+
+	if (info->amap.window_size == 0) {
+		/* get memory window size */
+		t = SW_READ_REG(MEMCTRL) >> fdesc->srs_shift;
+		t &= MEMCTRL_SRS_MASK;
+		info->amap.window_size = flash_sizes[t];
+	}
+
 	if (info->amap.window_size == 0) {
-		MAP_ERR(map, "invalid flash size detected\n");
+		MAP_ERR(map, "unable to determine window size\n");
 		goto err_out;
 	}