1 files changed, 77 insertions, 6 deletions

diff --git a/arch/mips/cavium-octeon/flash_setup.c b/arch/mips/cavium-octeon/flash_setup.c
index 237e5b1a72d8..a5e8f4a784af 100644
--- a/arch/mips/cavium-octeon/flash_setup.c
+++ b/arch/mips/cavium-octeon/flash_setup.c
@@ -8,9 +8,11 @@
  * Copyright (C) 2007, 2008 Cavium Networks
  */
 #include <linux/kernel.h>
-#include <linux/export.h>
+#include <linux/module.h>
+#include <linux/semaphore.h>
 #include <linux/mtd/mtd.h>
 #include <linux/mtd/map.h>
+#include <linux/of_platform.h>
 #include <linux/mtd/partitions.h>
 
 #include <asm/octeon/octeon.h>
@@ -25,19 +27,62 @@ static const char *part_probe_types[] = {
         NULL
 };
 
+static map_word octeon_flash_map_read(struct map_info *map, unsigned long ofs)
+{
+        map_word r;
+
+        down(&octeon_bootbus_sem);
+        r = inline_map_read(map, ofs);
+        up(&octeon_bootbus_sem);
+
+        return r;
+}
+
+static void octeon_flash_map_write(struct map_info *map, const map_word datum,
+                                   unsigned long ofs)
+{
+        down(&octeon_bootbus_sem);
+        inline_map_write(map, datum, ofs);
+        up(&octeon_bootbus_sem);
+}
+
+static void octeon_flash_map_copy_from(struct map_info *map, void *to,
+                                       unsigned long from, ssize_t len)
+{
+        down(&octeon_bootbus_sem);
+        inline_map_copy_from(map, to, from, len);
+        up(&octeon_bootbus_sem);
+}
+
+static void octeon_flash_map_copy_to(struct map_info *map, unsigned long to,
+                                     const void *from, ssize_t len)
+{
+        down(&octeon_bootbus_sem);
+        inline_map_copy_to(map, to, from, len);
+        up(&octeon_bootbus_sem);
+}
+
 /**
  * Module/ driver initialization.
  *
  * Returns Zero on success
  */
-static int __init flash_init(void)
+static int octeon_flash_probe(struct platform_device *pdev)
 {
+        union cvmx_mio_boot_reg_cfgx region_cfg;
+        u32 cs;
+        int r;
+        struct device_node *np = pdev->dev.of_node;
+
+        r = of_property_read_u32(np, "reg", &cs);
+        if (r)
+                return r;
+
         /*
          * Read the bootbus region 0 setup to determine the base
          * address of the flash.
          */
-        union cvmx_mio_boot_reg_cfgx region_cfg;
-        region_cfg.u64 = cvmx_read_csr(CVMX_MIO_BOOT_REG_CFGX(0));
+        region_cfg.u64 = cvmx_read_csr(CVMX_MIO_BOOT_REG_CFGX(cs));
         if (region_cfg.s.en) {
                 /*
                  * The bootloader always takes the flash and sets its
@@ -56,7 +101,11 @@ static int __init flash_init(void)
                 flash_map.virt = ioremap(flash_map.phys, flash_map.size);
                 pr_notice("Bootbus flash: Setting flash for %luMB flash at "
                           "0x%08llx\n", flash_map.size >> 20, flash_map.phys);
-                simple_map_init(&flash_map);
+                WARN_ON(!map_bankwidth_supported(flash_map.bankwidth));
+                flash_map.read = octeon_flash_map_read;
+                flash_map.write = octeon_flash_map_write;
+                flash_map.copy_from = octeon_flash_map_copy_from;
+                flash_map.copy_to = octeon_flash_map_copy_to;
                 mymtd = do_map_probe("cfi_probe", &flash_map);
                 if (mymtd) {
                         mymtd->owner = THIS_MODULE;
@@ -69,4 +118,26 @@ static int __init flash_init(void)
         return 0;
 }
 
-late_initcall(flash_init);
+static const struct of_device_id of_flash_match[] = {
+        {
+                .compatible     = "cfi-flash",
+        },
+        { },
+};
+MODULE_DEVICE_TABLE(of, of_flash_match);
+
+static struct platform_driver of_flash_driver = {
+        .driver = {
+                .name = "octeon-of-flash",
+                .of_match_table = of_flash_match,
+        },
+        .probe          = octeon_flash_probe,
+};
+
+static int octeon_flash_init(void)
+{
+        return platform_driver_register(&of_flash_driver);
+}
+late_initcall(octeon_flash_init);
+
+MODULE_LICENSE("GPL");