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 <anemo@mba.ocn.ne.jp>
11  */
12 
13 #include <linux/module.h>
14 #include <linux/types.h>
15 #include <linux/kernel.h>
16 #include <linux/slab.h>
17 #include <linux/device.h>
18 #include <linux/platform_device.h>
19 #include <linux/mtd/mtd.h>
20 #include <linux/mtd/map.h>
21 #include <linux/mtd/partitions.h>
22 #include <asm/txx9/rbtx4939.h>
23 
24 struct rbtx4939_flash_info {
25 	struct mtd_info *mtd;
26 	struct map_info map;
27 };
28 
rbtx4939_flash_remove(struct platform_device * dev)29 static int rbtx4939_flash_remove(struct platform_device *dev)
30 {
31 	struct rbtx4939_flash_info *info;
32 
33 	info = platform_get_drvdata(dev);
34 	if (!info)
35 		return 0;
36 
37 	if (info->mtd) {
38 		mtd_device_unregister(info->mtd);
39 		map_destroy(info->mtd);
40 	}
41 	return 0;
42 }
43 
44 static const char * const rom_probe_types[] = {
45 	"cfi_probe", "jedec_probe", NULL };
46 
rbtx4939_flash_probe(struct platform_device * dev)47 static int rbtx4939_flash_probe(struct platform_device *dev)
48 {
49 	struct rbtx4939_flash_data *pdata;
50 	struct rbtx4939_flash_info *info;
51 	struct resource *res;
52 	const char * const *probe_type;
53 	int err = 0;
54 	unsigned long size;
55 
56 	pdata = dev_get_platdata(&dev->dev);
57 	if (!pdata)
58 		return -ENODEV;
59 
60 	res = platform_get_resource(dev, IORESOURCE_MEM, 0);
61 	if (!res)
62 		return -ENODEV;
63 	info = devm_kzalloc(&dev->dev, sizeof(struct rbtx4939_flash_info),
64 			    GFP_KERNEL);
65 	if (!info)
66 		return -ENOMEM;
67 
68 	platform_set_drvdata(dev, info);
69 
70 	size = resource_size(res);
71 	pr_notice("rbtx4939 platform flash device: %pR\n", res);
72 
73 	if (!devm_request_mem_region(&dev->dev, res->start, size,
74 				     dev_name(&dev->dev)))
75 		return -EBUSY;
76 
77 	info->map.name = dev_name(&dev->dev);
78 	info->map.phys = res->start;
79 	info->map.size = size;
80 	info->map.bankwidth = pdata->width;
81 
82 	info->map.virt = devm_ioremap(&dev->dev, info->map.phys, size);
83 	if (!info->map.virt)
84 		return -EBUSY;
85 
86 	if (pdata->map_init)
87 		(*pdata->map_init)(&info->map);
88 	else
89 		simple_map_init(&info->map);
90 
91 	probe_type = rom_probe_types;
92 	for (; !info->mtd && *probe_type; probe_type++)
93 		info->mtd = do_map_probe(*probe_type, &info->map);
94 	if (!info->mtd) {
95 		dev_err(&dev->dev, "map_probe failed\n");
96 		err = -ENXIO;
97 		goto err_out;
98 	}
99 	info->mtd->dev.parent = &dev->dev;
100 	err = mtd_device_register(info->mtd, pdata->parts, pdata->nr_parts);
101 
102 	if (err)
103 		goto err_out;
104 	return 0;
105 
106 err_out:
107 	rbtx4939_flash_remove(dev);
108 	return err;
109 }
110 
111 #ifdef CONFIG_PM
rbtx4939_flash_shutdown(struct platform_device * dev)112 static void rbtx4939_flash_shutdown(struct platform_device *dev)
113 {
114 	struct rbtx4939_flash_info *info = platform_get_drvdata(dev);
115 
116 	if (mtd_suspend(info->mtd) == 0)
117 		mtd_resume(info->mtd);
118 }
119 #else
120 #define rbtx4939_flash_shutdown NULL
121 #endif
122 
123 static struct platform_driver rbtx4939_flash_driver = {
124 	.probe		= rbtx4939_flash_probe,
125 	.remove		= rbtx4939_flash_remove,
126 	.shutdown	= rbtx4939_flash_shutdown,
127 	.driver		= {
128 		.name	= "rbtx4939-flash",
129 	},
130 };
131 
132 module_platform_driver(rbtx4939_flash_driver);
133 
134 MODULE_LICENSE("GPL");
135 MODULE_DESCRIPTION("RBTX4939 MTD map driver");
136 MODULE_ALIAS("platform:rbtx4939-flash");
137