mtd: Add armflash support for multiple blocks of flash
[pandora-kernel.git] / drivers / mtd / maps / integrator-flash.c
1 /*======================================================================
2
3     drivers/mtd/maps/integrator-flash.c: ARM Integrator flash map driver
4
5     Copyright (C) 2000 ARM Limited
6     Copyright (C) 2003 Deep Blue Solutions Ltd.
7
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21
22    This is access code for flashes using ARM's flash partitioning
23    standards.
24
25 ======================================================================*/
26
27 #include <linux/module.h>
28 #include <linux/types.h>
29 #include <linux/kernel.h>
30 #include <linux/slab.h>
31 #include <linux/ioport.h>
32 #include <linux/platform_device.h>
33 #include <linux/init.h>
34 #include <linux/io.h>
35
36 #include <linux/mtd/mtd.h>
37 #include <linux/mtd/map.h>
38 #include <linux/mtd/partitions.h>
39 #include <linux/mtd/concat.h>
40
41 #include <asm/mach/flash.h>
42 #include <mach/hardware.h>
43 #include <asm/system.h>
44
45 #define SUBDEV_NAME_SIZE        (BUS_ID_SIZE + 2)
46
47 struct armflash_subdev_info {
48         char                    name[SUBDEV_NAME_SIZE];
49         struct mtd_info         *mtd;
50         struct map_info         map;
51         struct flash_platform_data *plat;
52 };
53
54 struct armflash_info {
55         struct resource         *res;
56         struct mtd_partition    *parts;
57         struct mtd_info         *mtd;
58         int                     nr_subdev;
59         struct armflash_subdev_info subdev[0];
60 };
61
62 static void armflash_set_vpp(struct map_info *map, int on)
63 {
64         struct armflash_subdev_info *info =
65                 container_of(map, struct armflash_subdev_info, map);
66
67         if (info->plat && info->plat->set_vpp)
68                 info->plat->set_vpp(on);
69 }
70
71 static const char *probes[] = { "cmdlinepart", "RedBoot", "afs", NULL };
72
73 static int armflash_subdev_probe(struct armflash_subdev_info *subdev,
74                                  struct resource *res)
75 {
76         struct flash_platform_data *plat = subdev->plat;
77         resource_size_t size = res->end - res->start + 1;
78         void __iomem *base;
79         int err = 0;
80
81         if (!request_mem_region(res->start, size, subdev->name)) {
82                 err = -EBUSY;
83                 goto out;
84         }
85
86         base = ioremap(res->start, size);
87         if (!base) {
88                 err = -ENOMEM;
89                 goto no_mem;
90         }
91
92         /*
93          * look for CFI based flash parts fitted to this board
94          */
95         subdev->map.size        = size;
96         subdev->map.bankwidth   = plat->width;
97         subdev->map.phys        = res->start;
98         subdev->map.virt        = base;
99         subdev->map.name        = subdev->name;
100         subdev->map.set_vpp     = armflash_set_vpp;
101
102         simple_map_init(&subdev->map);
103
104         /*
105          * Also, the CFI layer automatically works out what size
106          * of chips we have, and does the necessary identification
107          * for us automatically.
108          */
109         subdev->mtd = do_map_probe(plat->map_name, &subdev->map);
110         if (!subdev->mtd) {
111                 err = -ENXIO;
112                 goto no_device;
113         }
114
115         subdev->mtd->owner = THIS_MODULE;
116
117         /* Successful? */
118         if (err == 0)
119                 return err;
120
121         if (subdev->mtd)
122                 map_destroy(subdev->mtd);
123  no_device:
124         iounmap(base);
125  no_mem:
126         release_mem_region(res->start, size);
127  out:
128         return err;
129 }
130
131 static void armflash_subdev_remove(struct armflash_subdev_info *subdev)
132 {
133         if (subdev->mtd)
134                 map_destroy(subdev->mtd);
135         if (subdev->map.virt)
136                 iounmap(subdev->map.virt);
137         release_mem_region(subdev->map.phys, subdev->map.size);
138 }
139
140 static int armflash_probe(struct platform_device *dev)
141 {
142         struct flash_platform_data *plat = dev->dev.platform_data;
143         unsigned int size;
144         struct armflash_info *info;
145         int i, nr, err;
146
147         /* Count the number of devices */
148         for (nr = 0; ; nr++)
149                 if (!platform_get_resource(dev, IORESOURCE_MEM, nr))
150                         break;
151         if (nr == 0) {
152                 err = -ENODEV;
153                 goto out;
154         }
155
156         size = sizeof(struct armflash_info) +
157                 sizeof(struct armflash_subdev_info) * nr;
158         info = kzalloc(size, GFP_KERNEL);
159         if (!info) {
160                 err = -ENOMEM;
161                 goto out;
162         }
163
164         if (plat && plat->init) {
165                 err = plat->init();
166                 if (err)
167                         goto no_resource;
168         }
169
170         for (i = 0; i < nr; i++) {
171                 struct armflash_subdev_info *subdev = &info->subdev[i];
172                 struct resource *res;
173
174                 res = platform_get_resource(dev, IORESOURCE_MEM, i);
175                 if (!res)
176                         break;
177
178                 if (nr == 1)
179                         /* No MTD concatenation, just use the default name */
180                         snprintf(subdev->name, SUBDEV_NAME_SIZE, "%s",
181                                  dev_name(&dev->dev));
182                 else
183                         snprintf(subdev->name, SUBDEV_NAME_SIZE, "%s-%d",
184                                  dev_name(&dev->dev), i);
185                 subdev->plat = plat;
186
187                 err = armflash_subdev_probe(subdev, res);
188                 if (err)
189                         break;
190         }
191         info->nr_subdev = i;
192
193         if (err)
194                 goto subdev_err;
195
196         if (info->nr_subdev == 1)
197                 info->mtd = info->subdev[0].mtd;
198         else if (info->nr_subdev > 1) {
199 #ifdef CONFIG_MTD_CONCAT
200                 struct mtd_info *cdev[info->nr_subdev];
201
202                 /*
203                  * We detected multiple devices.  Concatenate them together.
204                  */
205                 for (i = 0; i < info->nr_subdev; i++)
206                         cdev[i] = info->subdev[i].mtd;
207
208                 info->mtd = mtd_concat_create(cdev, info->nr_subdev,
209                                               dev_name(&dev->dev));
210                 if (info->mtd == NULL)
211                         err = -ENXIO;
212 #else
213                 printk(KERN_ERR "armflash: multiple devices found but "
214                        "MTD concat support disabled.\n");
215                 err = -ENXIO;
216 #endif
217         }
218
219         if (err < 0)
220                 goto cleanup;
221
222         err = parse_mtd_partitions(info->mtd, probes, &info->parts, 0);
223         if (err > 0) {
224                 err = add_mtd_partitions(info->mtd, info->parts, err);
225                 if (err)
226                         printk(KERN_ERR
227                                "mtd partition registration failed: %d\n", err);
228         }
229
230         if (err == 0) {
231                 platform_set_drvdata(dev, info);
232                 return err;
233         }
234
235         /*
236          * We got an error, free all resources.
237          */
238  cleanup:
239         if (info->mtd) {
240                 del_mtd_partitions(info->mtd);
241 #ifdef CONFIG_MTD_CONCAT
242                 if (info->mtd != info->subdev[0].mtd)
243                         mtd_concat_destroy(info->mtd);
244 #endif
245         }
246         kfree(info->parts);
247  subdev_err:
248         for (i = info->nr_subdev - 1; i >= 0; i--)
249                 armflash_subdev_remove(&info->subdev[i]);
250  no_resource:
251         if (plat && plat->exit)
252                 plat->exit();
253         kfree(info);
254  out:
255         return err;
256 }
257
258 static int armflash_remove(struct platform_device *dev)
259 {
260         struct armflash_info *info = platform_get_drvdata(dev);
261         struct flash_platform_data *plat = dev->dev.platform_data;
262         int i;
263
264         platform_set_drvdata(dev, NULL);
265
266         if (info) {
267                 if (info->mtd) {
268                         del_mtd_partitions(info->mtd);
269 #ifdef CONFIG_MTD_CONCAT
270                         if (info->mtd != info->subdev[0].mtd)
271                                 mtd_concat_destroy(info->mtd);
272 #endif
273                 }
274                 kfree(info->parts);
275
276                 for (i = info->nr_subdev - 1; i >= 0; i--)
277                         armflash_subdev_remove(&info->subdev[i]);
278
279                 if (plat && plat->exit)
280                         plat->exit();
281
282                 kfree(info);
283         }
284
285         return 0;
286 }
287
288 static struct platform_driver armflash_driver = {
289         .probe          = armflash_probe,
290         .remove         = armflash_remove,
291         .driver         = {
292                 .name   = "armflash",
293                 .owner  = THIS_MODULE,
294         },
295 };
296
297 static int __init armflash_init(void)
298 {
299         return platform_driver_register(&armflash_driver);
300 }
301
302 static void __exit armflash_exit(void)
303 {
304         platform_driver_unregister(&armflash_driver);
305 }
306
307 module_init(armflash_init);
308 module_exit(armflash_exit);
309
310 MODULE_AUTHOR("ARM Ltd");
311 MODULE_DESCRIPTION("ARM Integrator CFI map driver");
312 MODULE_LICENSE("GPL");
313 MODULE_ALIAS("platform:armflash");