--- /dev/null
+/*
+ * Parallel flash driver for the Atheros AR91xx SoC
+ *
+ * Copyright (C) 2009 Gabor Juhos <juhosg@openwrt.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ *
+ */
+
+#include <linux/module.h>
+#include <linux/types.h>
+#include <linux/kernel.h>
+#include <linux/init.h>
+#include <linux/slab.h>
+#include <linux/device.h>
+#include <linux/platform_device.h>
+#include <linux/mtd/mtd.h>
+#include <linux/mtd/map.h>
+#include <linux/mtd/partitions.h>
+#include <linux/io.h>
+
+#include <asm/mach-ar71xx/ar71xx.h>
+#include <asm/mach-ar71xx/platform.h>
+
+#define DRV_NAME "ar91xx-flash"
+
+struct ar91xx_flash_info {
+ struct mtd_info *mtd;
+ struct map_info map;
+#ifdef CONFIG_MTD_PARTITIONS
+ int nr_parts;
+ struct mtd_partition *parts;
+#endif
+};
+
+static map_word ar91xx_flash_read(struct map_info *map, unsigned long ofs)
+{
+ map_word val;
+
+ if (map_bankwidth_is_1(map))
+ val.x[0] = __raw_readb(map->virt + (ofs ^ 3));
+ else if (map_bankwidth_is_2(map))
+ val.x[0] = __raw_readw(map->virt + (ofs ^ 2));
+ else
+ val = map_word_ff(map);
+
+ return val;
+}
+
+static void ar91xx_flash_write(struct map_info *map, map_word d,
+ unsigned long ofs)
+{
+ if (map_bankwidth_is_1(map))
+ __raw_writeb(d.x[0], map->virt + (ofs ^ 3));
+ else if (map_bankwidth_is_2(map))
+ __raw_writew(d.x[0], map->virt + (ofs ^ 2));
+
+ mb();
+}
+
+static int ar91xx_flash_remove(struct platform_device *pdev)
+{
+ struct ar91xx_flash_platform_data *pdata;
+ struct ar91xx_flash_info *info;
+
+ info = platform_get_drvdata(pdev);
+ if (info == NULL)
+ return 0;
+
+ platform_set_drvdata(pdev, NULL);
+
+ if (info->mtd == NULL)
+ return 0;
+
+ pdata = pdev->dev.platform_data;
+#ifdef CONFIG_MTD_PARTITIONS
+ if (info->nr_parts) {
+ del_mtd_partitions(info->mtd);
+ kfree(info->parts);
+ } else if (pdata->nr_parts) {
+ del_mtd_partitions(info->mtd);
+ } else {
+ del_mtd_device(info->mtd);
+ }
+#else
+ del_mtd_device(info->mtd);
+#endif
+ map_destroy(info->mtd);
+
+ return 0;
+}
+
+static const char *rom_probe_types[] = { "cfi_probe", "jedec_probe", NULL };
+#ifdef CONFIG_MTD_PARTITIONS
+static const char *part_probe_types[] = { "cmdlinepart", "RedBoot", NULL };
+#endif
+
+static int ar91xx_flash_probe(struct platform_device *pdev)
+{
+ struct ar91xx_flash_platform_data *pdata;
+ struct ar91xx_flash_info *info;
+ struct resource *res;
+ struct resource *region;
+ const char **probe_type;
+ int err = 0;
+
+ pdata = pdev->dev.platform_data;
+ if (pdata == NULL)
+ return -EINVAL;
+
+ info = devm_kzalloc(&pdev->dev, sizeof(struct ar91xx_flash_info),
+ GFP_KERNEL);
+ if (info == NULL) {
+ err = -ENOMEM;
+ goto err_out;
+ }
+
+ platform_set_drvdata(pdev, info);
+
+ res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
+ if (res == NULL) {
+ err = -ENOENT;
+ goto err_out;
+ }
+
+ dev_info(&pdev->dev, "%.8llx at %.8llx\n",
+ (unsigned long long)(res->end - res->start + 1),
+ (unsigned long long)res->start);
+
+ region = devm_request_mem_region(&pdev->dev,
+ res->start, res->end - res->start + 1,
+ dev_name(&pdev->dev));
+ if (region == NULL) {
+ dev_err(&pdev->dev, "could not reserve memory region\n");
+ err = -ENOMEM;
+ goto err_out;
+ }
+
+ info->map.name = dev_name(&pdev->dev);
+ info->map.phys = res->start;
+ info->map.size = res->end - res->start + 1;
+ info->map.bankwidth = pdata->width;
+
+ info->map.virt = devm_ioremap(&pdev->dev, info->map.phys,
+ info->map.size);
+ if (info->map.virt == NULL) {
+ dev_err(&pdev->dev, "failed to ioremap flash region\n");
+ err = -EIO;
+ goto err_out;
+ }
+
+ simple_map_init(&info->map);
+ info->map.read = ar91xx_flash_read;
+ info->map.write = ar91xx_flash_write;
+
+ probe_type = rom_probe_types;
+ for (; info->mtd == NULL && *probe_type != NULL; probe_type++)
+ info->mtd = do_map_probe(*probe_type, &info->map);
+
+ if (info->mtd == NULL) {
+ dev_err(&pdev->dev, "map_probe failed\n");
+ err = -ENXIO;
+ goto err_out;
+ }
+
+ info->mtd->owner = THIS_MODULE;
+
+#ifdef CONFIG_MTD_PARTITIONS
+ if (pdata->nr_parts) {
+ dev_info(&pdev->dev, "using static partition mapping\n");
+ add_mtd_partitions(info->mtd, pdata->parts, pdata->nr_parts);
+ return 0;
+ }
+
+ err = parse_mtd_partitions(info->mtd, part_probe_types,
+ &info->parts, 0);
+ if (err > 0) {
+ add_mtd_partitions(info->mtd, info->parts, err);
+ return 0;
+ }
+#endif
+
+ add_mtd_device(info->mtd);
+ return 0;
+
+ err_out:
+ ar91xx_flash_remove(pdev);
+ return err;
+}
+
+#ifdef CONFIG_PM
+static int ar91xx_flash_suspend(struct platform_device *dev, pm_message_t state)
+{
+ struct ar91xx_flash_info *info = platform_get_drvdata(dev);
+ int ret = 0;
+
+ if (info->mtd->suspend)
+ ret = info->mtd->suspend(info->mtd);
+
+ if (ret)
+ goto fail;
+
+ return 0;
+
+ fail:
+ if (info->mtd->suspend) {
+ BUG_ON(!info->mtd->resume);
+ info->mtd->resume(info->mtd);
+ }
+
+ return ret;
+}
+
+static int ar91xx_flash_resume(struct platform_device *pdev)
+{
+ struct ar91xx_flash_info *info = platform_get_drvdata(pdev);
+
+ if (info->mtd->resume)
+ info->mtd->resume(info->mtd);
+
+ return 0;
+}
+
+static void ar91xx_flash_shutdown(struct platform_device *pdev)
+{
+ struct ar91xx_flash_info *info = platform_get_drvdata(pdev);
+
+ if (info->mtd->suspend && info->mtd->resume)
+ if (info->mtd->suspend(info->mtd) == 0)
+ info->mtd->resume(info->mtd);
+}
+#else
+#define ar91xx_flash_suspend NULL
+#define ar91xx_flash_resume NULL
+#define ar91xx_flash_shutdown NULL
+#endif
+
+static struct platform_driver ar91xx_flash_driver = {
+ .probe = ar91xx_flash_probe,
+ .remove = ar91xx_flash_remove,
+ .suspend = ar91xx_flash_suspend,
+ .resume = ar91xx_flash_resume,
+ .shutdown = ar91xx_flash_shutdown,
+ .driver = {
+ .name = DRV_NAME,
+ .owner = THIS_MODULE,
+ },
+};
+
+static int __init ar91xx_flash_init(void)
+{
+ return platform_driver_register(&ar91xx_flash_driver);
+}
+
+static void __exit ar91xx_flash_exit(void)
+{
+ platform_driver_unregister(&ar91xx_flash_driver);
+}
+
+module_init(ar91xx_flash_init);
+module_exit(ar91xx_flash_exit);
+
+MODULE_LICENSE("GPL v2");
+MODULE_AUTHOR("Gabor Juhos <juhosg@openwrt.org>");
+MODULE_DESCRIPTION("Parallel flash driver for the Atheros AR91xx SoC");
+MODULE_ALIAS("platform:" DRV_NAME);