1 files changed, 14 insertions, 10 deletions

diff --git a/drivers/mtd/onenand/generic.c b/drivers/mtd/onenand/generic.c
index 3a496c33fb52..e78914938c5c 100644
--- a/drivers/mtd/onenand/generic.c
+++ b/drivers/mtd/onenand/generic.c
@@ -19,12 +19,16 @@
 #include <linux/mtd/mtd.h>
 #include <linux/mtd/onenand.h>
 #include <linux/mtd/partitions.h>
-
 #include <asm/io.h>
-#include <asm/mach/flash.h>
-
-#define DRIVER_NAME     "onenand"
 
+/*
+ * Note: Driver name and platform data format have been updated!
+ *
+ * This version of the driver is named "onenand-flash" and takes struct
+ * onenand_platform_data as platform data. The old ARM-specific version
+ * with the name "onenand" used to take struct flash_platform_data.
+ */
+#define DRIVER_NAME     "onenand-flash"
 
 #ifdef CONFIG_MTD_PARTITIONS
 static const char *part_probes[] = { "cmdlinepart", NULL,  };
@@ -39,16 +43,16 @@ struct onenand_info {
 static int __devinit generic_onenand_probe(struct platform_device *pdev)
 {
         struct onenand_info *info;
-        struct flash_platform_data *pdata = pdev->dev.platform_data;
+        struct onenand_platform_data *pdata = pdev->dev.platform_data;
         struct resource *res = pdev->resource;
-        unsigned long size = res->end - res->start + 1;
+        unsigned long size = resource_size(res);
         int err;
 
         info = kzalloc(sizeof(struct onenand_info), GFP_KERNEL);
         if (!info)
                 return -ENOMEM;
 
-        if (!request_mem_region(res->start, size, pdev->dev.driver->name)) {
+        if (!request_mem_region(res->start, size, dev_name(&pdev->dev))) {
                 err = -EBUSY;
                 goto out_free_info;
         }
@@ -59,7 +63,7 @@ static int __devinit generic_onenand_probe(struct platform_device *pdev)
                 goto out_release_mem_region;
         }
 
-        info->onenand.mmcontrol = pdata->mmcontrol;
+        info->onenand.mmcontrol = pdata ? pdata->mmcontrol : 0;
         info->onenand.irq = platform_get_irq(pdev, 0);
 
         info->mtd.name = dev_name(&pdev->dev);
@@ -75,7 +79,7 @@ static int __devinit generic_onenand_probe(struct platform_device *pdev)
         err = parse_mtd_partitions(&info->mtd, part_probes, &info->parts, 0);
         if (err > 0)
                 add_mtd_partitions(&info->mtd, info->parts, err);
-        else if (err <= 0 && pdata->parts)
+        else if (err <= 0 && pdata && pdata->parts)
                 add_mtd_partitions(&info->mtd, pdata->parts, pdata->nr_parts);
         else
 #endif
@@ -99,7 +103,7 @@ static int __devexit generic_onenand_remove(struct platform_device *pdev)
 {
         struct onenand_info *info = platform_get_drvdata(pdev);
         struct resource *res = pdev->resource;
-        unsigned long size = res->end - res->start + 1;
+        unsigned long size = resource_size(res);
 
         platform_set_drvdata(pdev, NULL);