[linux.git] / drivers / mtd / maps / rbtx4939-flash.c

/*
 * rbtx4939-flash (based on physmap.c)
 *
 * This is a simplified physmap driver with map_init callback function.
 *
 * 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.
 *
 * Copyright (C) 2009 Atsushi Nemoto <[email protected]>
 */

#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 <asm/txx9/rbtx4939.h>

struct rbtx4939_flash_info {
	struct mtd_info *mtd;
	struct map_info map;
#ifdef CONFIG_MTD_PARTITIONS
	int nr_parts;
	struct mtd_partition *parts;
#endif
};

static int rbtx4939_flash_remove(struct platform_device *dev)
{
	struct rbtx4939_flash_info *info;

	info = platform_get_drvdata(dev);
	if (!info)
		return 0;
	platform_set_drvdata(dev, NULL);

	if (info->mtd) {
#ifdef CONFIG_MTD_PARTITIONS
		struct rbtx4939_flash_data *pdata = dev->dev.platform_data;

		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", NULL };
#endif

static int rbtx4939_flash_probe(struct platform_device *dev)
{
	struct rbtx4939_flash_data *pdata;
	struct rbtx4939_flash_info *info;
	struct resource *res;
	const char **probe_type;
	int err = 0;
	unsigned long size;

	pdata = dev->dev.platform_data;
	if (!pdata)
		return -ENODEV;

	res = platform_get_resource(dev, IORESOURCE_MEM, 0);
	if (!res)
		return -ENODEV;
	info = devm_kzalloc(&dev->dev, sizeof(struct rbtx4939_flash_info),
			    GFP_KERNEL);
	if (!info)
		return -ENOMEM;

	platform_set_drvdata(dev, info);

	size = resource_size(res);
	pr_notice("rbtx4939 platform flash device: %pR\n", res);

	if (!devm_request_mem_region(&dev->dev, res->start, size,
				     dev_name(&dev->dev)))
		return -EBUSY;

	info->map.name = dev_name(&dev->dev);
	info->map.phys = res->start;
	info->map.size = size;
	info->map.bankwidth = pdata->width;

	info->map.virt = devm_ioremap(&dev->dev, info->map.phys, size);
	if (!info->map.virt)
		return -EBUSY;

	if (pdata->map_init)
		(*pdata->map_init)(&info->map);
	else
		simple_map_init(&info->map);

	probe_type = rom_probe_types;
	for (; !info->mtd && *probe_type; probe_type++)
		info->mtd = do_map_probe(*probe_type, &info->map);
	if (!info->mtd) {
		dev_err(&dev->dev, "map_probe failed\n");
		err = -ENXIO;
		goto err_out;
	}
	info->mtd->owner = THIS_MODULE;
	if (err)
		goto err_out;

#ifdef CONFIG_MTD_PARTITIONS
	err = parse_mtd_partitions(info->mtd, part_probe_types,
				&info->parts, 0);
	if (err > 0) {
		add_mtd_partitions(info->mtd, info->parts, err);
		info->nr_parts = err;
		return 0;
	}

	if (pdata->nr_parts) {
		pr_notice("Using rbtx4939 partition information\n");
		add_mtd_partitions(info->mtd, pdata->parts, pdata->nr_parts);
		return 0;
	}
#endif

	add_mtd_device(info->mtd);
	return 0;

err_out:
	rbtx4939_flash_remove(dev);
	return err;
}

#ifdef CONFIG_PM
static void rbtx4939_flash_shutdown(struct platform_device *dev)
{
	struct rbtx4939_flash_info *info = platform_get_drvdata(dev);

	if (info->mtd->suspend && info->mtd->resume)
		if (info->mtd->suspend(info->mtd) == 0)
			info->mtd->resume(info->mtd);
}
#else
#define rbtx4939_flash_shutdown NULL
#endif

static struct platform_driver rbtx4939_flash_driver = {
	.probe		= rbtx4939_flash_probe,
	.remove		= rbtx4939_flash_remove,
	.shutdown	= rbtx4939_flash_shutdown,
	.driver		= {
		.name	= "rbtx4939-flash",
		.owner	= THIS_MODULE,
	},
};

static int __init rbtx4939_flash_init(void)
{
	return platform_driver_register(&rbtx4939_flash_driver);
}

static void __exit rbtx4939_flash_exit(void)
{
	platform_driver_unregister(&rbtx4939_flash_driver);
}

module_init(rbtx4939_flash_init);
module_exit(rbtx4939_flash_exit);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("RBTX4939 MTD map driver");
MODULE_ALIAS("platform:rbtx4939-flash");
Commit	Line	Data
610f75e7 AN	1	/*
	2	* rbtx4939-flash (based on physmap.c)
	3	*
	4	* This is a simplified physmap driver with map_init callback function.
	5	*
	6	* This program is free software; you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License version 2 as
	8	* published by the Free Software Foundation.
	9	*
	10	* Copyright (C) 2009 Atsushi Nemoto <[email protected]>
	11	*/
	12
	13	#include <linux/module.h>
	14	#include <linux/types.h>
	15	#include <linux/kernel.h>
	16	#include <linux/init.h>
	17	#include <linux/slab.h>
	18	#include <linux/device.h>
	19	#include <linux/platform_device.h>
	20	#include <linux/mtd/mtd.h>
	21	#include <linux/mtd/map.h>
	22	#include <linux/mtd/partitions.h>
	23	#include <asm/txx9/rbtx4939.h>
	24
	25	struct rbtx4939_flash_info {
	26	struct mtd_info *mtd;
	27	struct map_info map;
	28	#ifdef CONFIG_MTD_PARTITIONS
	29	int nr_parts;
	30	struct mtd_partition *parts;
	31	#endif
	32	};
	33
	34	static int rbtx4939_flash_remove(struct platform_device *dev)
	35	{
	36	struct rbtx4939_flash_info *info;
	37
	38	info = platform_get_drvdata(dev);
	39	if (!info)
	40	return 0;
	41	platform_set_drvdata(dev, NULL);
	42
	43	if (info->mtd) {
	44	#ifdef CONFIG_MTD_PARTITIONS
	45	struct rbtx4939_flash_data *pdata = dev->dev.platform_data;
	46
	47	if (info->nr_parts) {
	48	del_mtd_partitions(info->mtd);
	49	kfree(info->parts);
	50	} else if (pdata->nr_parts)
	51	del_mtd_partitions(info->mtd);
	52	else
	53	del_mtd_device(info->mtd);
	54	#else
	55	del_mtd_device(info->mtd);
	56	#endif
	57	map_destroy(info->mtd);
	58	}
	59	return 0;
	60	}
	61
	62	static const char *rom_probe_types[] = { "cfi_probe", "jedec_probe", NULL };
	63	#ifdef CONFIG_MTD_PARTITIONS
	64	static const char *part_probe_types[] = { "cmdlinepart", NULL };
65	#endif
66
67	static int rbtx4939_flash_probe(struct platform_device *dev)
68	{
69	struct rbtx4939_flash_data *pdata;
70	struct rbtx4939_flash_info *info;
71	struct resource *res;
72	const char **probe_type;
73	int err = 0;
74	unsigned long size;
75
76	pdata = dev->dev.platform_data;
77	if (!pdata)
78	return -ENODEV;
79
80	res = platform_get_resource(dev, IORESOURCE_MEM, 0);
81	if (!res)
82	return -ENODEV;
83	info = devm_kzalloc(&dev->dev, sizeof(struct rbtx4939_flash_info),
84	GFP_KERNEL);
85	if (!info)
86	return -ENOMEM;
87
88	platform_set_drvdata(dev, info);
89
90	size = resource_size(res);
91	pr_notice("rbtx4939 platform flash device: %pR\n", res);
92
93	if (!devm_request_mem_region(&dev->dev, res->start, size,
94	dev_name(&dev->dev)))
95	return -EBUSY;
96
97	info->map.name = dev_name(&dev->dev);
98	info->map.phys = res->start;
99	info->map.size = size;
100	info->map.bankwidth = pdata->width;
101
102	info->map.virt = devm_ioremap(&dev->dev, info->map.phys, size);
103	if (!info->map.virt)
104	return -EBUSY;
105
106	if (pdata->map_init)
107	(*pdata->map_init)(&info->map);
108	else
109	simple_map_init(&info->map);
110
111	probe_type = rom_probe_types;
112	for (; !info->mtd && *probe_type; probe_type++)
113	info->mtd = do_map_probe(*probe_type, &info->map);
114	if (!info->mtd) {
115	dev_err(&dev->dev, "map_probe failed\n");
116	err = -ENXIO;
117	goto err_out;
118	}
119	info->mtd->owner = THIS_MODULE;
120	if (err)
121	goto err_out;
122
123	#ifdef CONFIG_MTD_PARTITIONS
124	err = parse_mtd_partitions(info->mtd, part_probe_types,
125	&info->parts, 0);
126	if (err > 0) {
127	add_mtd_partitions(info->mtd, info->parts, err);
128	info->nr_parts = err;
129	return 0;
130	}
131
132	if (pdata->nr_parts) {
133	pr_notice("Using rbtx4939 partition information\n");
134	add_mtd_partitions(info->mtd, pdata->parts, pdata->nr_parts);
135	return 0;
136	}
137	#endif
138
139	add_mtd_device(info->mtd);
140	return 0;
141
142	err_out:
143	rbtx4939_flash_remove(dev);
144	return err;
145	}
146
147	#ifdef CONFIG_PM
610f75e7 AN	148	static void rbtx4939_flash_shutdown(struct platform_device *dev)
	149	{
	150	struct rbtx4939_flash_info *info = platform_get_drvdata(dev);
	151
	152	if (info->mtd->suspend && info->mtd->resume)
	153	if (info->mtd->suspend(info->mtd) == 0)
	154	info->mtd->resume(info->mtd);
	155	}
	156	#else
610f75e7 AN	157	#define rbtx4939_flash_shutdown NULL
	158	#endif
	159
	160	static struct platform_driver rbtx4939_flash_driver = {
	161	.probe = rbtx4939_flash_probe,
	162	.remove = rbtx4939_flash_remove,
610f75e7 AN	163	.shutdown = rbtx4939_flash_shutdown,
	164	.driver = {
	165	.name = "rbtx4939-flash",
	166	.owner = THIS_MODULE,
	167	},
	168	};
	169
	170	static int __init rbtx4939_flash_init(void)
	171	{
	172	return platform_driver_register(&rbtx4939_flash_driver);
	173	}
	174
	175	static void __exit rbtx4939_flash_exit(void)
	176	{
	177	platform_driver_unregister(&rbtx4939_flash_driver);
	178	}
	179
	180	module_init(rbtx4939_flash_init);
	181	module_exit(rbtx4939_flash_exit);
	182
	183	MODULE_LICENSE("GPL");
	184	MODULE_DESCRIPTION("RBTX4939 MTD map driver");
	185	MODULE_ALIAS("platform:rbtx4939-flash");