mtd: return error code from mtd_unpoint
[pandora-kernel.git] / drivers / mtd / mtdpart.c
1 /*
2  * Simple MTD partitioning layer
3  *
4  * Copyright © 2000 Nicolas Pitre <nico@fluxnic.net>
5  * Copyright © 2002 Thomas Gleixner <gleixner@linutronix.de>
6  * Copyright © 2000-2010 David Woodhouse <dwmw2@infradead.org>
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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
21  *
22  */
23
24 #include <linux/module.h>
25 #include <linux/types.h>
26 #include <linux/kernel.h>
27 #include <linux/slab.h>
28 #include <linux/list.h>
29 #include <linux/kmod.h>
30 #include <linux/mtd/mtd.h>
31 #include <linux/mtd/partitions.h>
32 #include <linux/err.h>
33
34 #include "mtdcore.h"
35
36 /* Our partition linked list */
37 static LIST_HEAD(mtd_partitions);
38 static DEFINE_MUTEX(mtd_partitions_mutex);
39
40 /* Our partition node structure */
41 struct mtd_part {
42         struct mtd_info mtd;
43         struct mtd_info *master;
44         uint64_t offset;
45         struct list_head list;
46 };
47
48 /*
49  * Given a pointer to the MTD object in the mtd_part structure, we can retrieve
50  * the pointer to that structure with this macro.
51  */
52 #define PART(x)  ((struct mtd_part *)(x))
53
54
55 /*
56  * MTD methods which simply translate the effective address and pass through
57  * to the _real_ device.
58  */
59
60 static int part_read(struct mtd_info *mtd, loff_t from, size_t len,
61                 size_t *retlen, u_char *buf)
62 {
63         struct mtd_part *part = PART(mtd);
64         struct mtd_ecc_stats stats;
65         int res;
66
67         stats = part->master->ecc_stats;
68
69         if (from >= mtd->size)
70                 len = 0;
71         else if (from + len > mtd->size)
72                 len = mtd->size - from;
73         res = mtd_read(part->master, from + part->offset, len, retlen, buf);
74         if (unlikely(res)) {
75                 if (mtd_is_bitflip(res))
76                         mtd->ecc_stats.corrected += part->master->ecc_stats.corrected - stats.corrected;
77                 if (mtd_is_eccerr(res))
78                         mtd->ecc_stats.failed += part->master->ecc_stats.failed - stats.failed;
79         }
80         return res;
81 }
82
83 static int part_point(struct mtd_info *mtd, loff_t from, size_t len,
84                 size_t *retlen, void **virt, resource_size_t *phys)
85 {
86         struct mtd_part *part = PART(mtd);
87         if (from >= mtd->size)
88                 len = 0;
89         else if (from + len > mtd->size)
90                 len = mtd->size - from;
91         return mtd_point(part->master, from + part->offset, len, retlen,
92                          virt, phys);
93 }
94
95 static int part_unpoint(struct mtd_info *mtd, loff_t from, size_t len)
96 {
97         struct mtd_part *part = PART(mtd);
98
99         return mtd_unpoint(part->master, from + part->offset, len);
100 }
101
102 static unsigned long part_get_unmapped_area(struct mtd_info *mtd,
103                                             unsigned long len,
104                                             unsigned long offset,
105                                             unsigned long flags)
106 {
107         struct mtd_part *part = PART(mtd);
108
109         offset += part->offset;
110         return mtd_get_unmapped_area(part->master, len, offset, flags);
111 }
112
113 static int part_read_oob(struct mtd_info *mtd, loff_t from,
114                 struct mtd_oob_ops *ops)
115 {
116         struct mtd_part *part = PART(mtd);
117         int res;
118
119         if (from >= mtd->size)
120                 return -EINVAL;
121         if (ops->datbuf && from + ops->len > mtd->size)
122                 return -EINVAL;
123
124         /*
125          * If OOB is also requested, make sure that we do not read past the end
126          * of this partition.
127          */
128         if (ops->oobbuf) {
129                 size_t len, pages;
130
131                 if (ops->mode == MTD_OPS_AUTO_OOB)
132                         len = mtd->oobavail;
133                 else
134                         len = mtd->oobsize;
135                 pages = mtd_div_by_ws(mtd->size, mtd);
136                 pages -= mtd_div_by_ws(from, mtd);
137                 if (ops->ooboffs + ops->ooblen > pages * len)
138                         return -EINVAL;
139         }
140
141         res = mtd_read_oob(part->master, from + part->offset, ops);
142         if (unlikely(res)) {
143                 if (mtd_is_bitflip(res))
144                         mtd->ecc_stats.corrected++;
145                 if (mtd_is_eccerr(res))
146                         mtd->ecc_stats.failed++;
147         }
148         return res;
149 }
150
151 static int part_read_user_prot_reg(struct mtd_info *mtd, loff_t from,
152                 size_t len, size_t *retlen, u_char *buf)
153 {
154         struct mtd_part *part = PART(mtd);
155         return mtd_read_user_prot_reg(part->master, from, len, retlen, buf);
156 }
157
158 static int part_get_user_prot_info(struct mtd_info *mtd,
159                 struct otp_info *buf, size_t len)
160 {
161         struct mtd_part *part = PART(mtd);
162         return mtd_get_user_prot_info(part->master, buf, len);
163 }
164
165 static int part_read_fact_prot_reg(struct mtd_info *mtd, loff_t from,
166                 size_t len, size_t *retlen, u_char *buf)
167 {
168         struct mtd_part *part = PART(mtd);
169         return mtd_read_fact_prot_reg(part->master, from, len, retlen, buf);
170 }
171
172 static int part_get_fact_prot_info(struct mtd_info *mtd, struct otp_info *buf,
173                 size_t len)
174 {
175         struct mtd_part *part = PART(mtd);
176         return mtd_get_fact_prot_info(part->master, buf, len);
177 }
178
179 static int part_write(struct mtd_info *mtd, loff_t to, size_t len,
180                 size_t *retlen, const u_char *buf)
181 {
182         struct mtd_part *part = PART(mtd);
183         if (!(mtd->flags & MTD_WRITEABLE))
184                 return -EROFS;
185         if (to >= mtd->size)
186                 len = 0;
187         else if (to + len > mtd->size)
188                 len = mtd->size - to;
189         return mtd_write(part->master, to + part->offset, len, retlen, buf);
190 }
191
192 static int part_panic_write(struct mtd_info *mtd, loff_t to, size_t len,
193                 size_t *retlen, const u_char *buf)
194 {
195         struct mtd_part *part = PART(mtd);
196         if (!(mtd->flags & MTD_WRITEABLE))
197                 return -EROFS;
198         if (to >= mtd->size)
199                 len = 0;
200         else if (to + len > mtd->size)
201                 len = mtd->size - to;
202         return mtd_panic_write(part->master, to + part->offset, len, retlen,
203                                buf);
204 }
205
206 static int part_write_oob(struct mtd_info *mtd, loff_t to,
207                 struct mtd_oob_ops *ops)
208 {
209         struct mtd_part *part = PART(mtd);
210
211         if (!(mtd->flags & MTD_WRITEABLE))
212                 return -EROFS;
213
214         if (to >= mtd->size)
215                 return -EINVAL;
216         if (ops->datbuf && to + ops->len > mtd->size)
217                 return -EINVAL;
218         return mtd_write_oob(part->master, to + part->offset, ops);
219 }
220
221 static int part_write_user_prot_reg(struct mtd_info *mtd, loff_t from,
222                 size_t len, size_t *retlen, u_char *buf)
223 {
224         struct mtd_part *part = PART(mtd);
225         return mtd_write_user_prot_reg(part->master, from, len, retlen, buf);
226 }
227
228 static int part_lock_user_prot_reg(struct mtd_info *mtd, loff_t from,
229                 size_t len)
230 {
231         struct mtd_part *part = PART(mtd);
232         return mtd_lock_user_prot_reg(part->master, from, len);
233 }
234
235 static int part_writev(struct mtd_info *mtd, const struct kvec *vecs,
236                 unsigned long count, loff_t to, size_t *retlen)
237 {
238         struct mtd_part *part = PART(mtd);
239         if (!(mtd->flags & MTD_WRITEABLE))
240                 return -EROFS;
241         return mtd_writev(part->master, vecs, count, to + part->offset,
242                           retlen);
243 }
244
245 static int part_erase(struct mtd_info *mtd, struct erase_info *instr)
246 {
247         struct mtd_part *part = PART(mtd);
248         int ret;
249         if (!(mtd->flags & MTD_WRITEABLE))
250                 return -EROFS;
251         if (instr->addr >= mtd->size)
252                 return -EINVAL;
253         instr->addr += part->offset;
254         ret = mtd_erase(part->master, instr);
255         if (ret) {
256                 if (instr->fail_addr != MTD_FAIL_ADDR_UNKNOWN)
257                         instr->fail_addr -= part->offset;
258                 instr->addr -= part->offset;
259         }
260         return ret;
261 }
262
263 void mtd_erase_callback(struct erase_info *instr)
264 {
265         if (instr->mtd->_erase == part_erase) {
266                 struct mtd_part *part = PART(instr->mtd);
267
268                 if (instr->fail_addr != MTD_FAIL_ADDR_UNKNOWN)
269                         instr->fail_addr -= part->offset;
270                 instr->addr -= part->offset;
271         }
272         if (instr->callback)
273                 instr->callback(instr);
274 }
275 EXPORT_SYMBOL_GPL(mtd_erase_callback);
276
277 static int part_lock(struct mtd_info *mtd, loff_t ofs, uint64_t len)
278 {
279         struct mtd_part *part = PART(mtd);
280         if ((len + ofs) > mtd->size)
281                 return -EINVAL;
282         return mtd_lock(part->master, ofs + part->offset, len);
283 }
284
285 static int part_unlock(struct mtd_info *mtd, loff_t ofs, uint64_t len)
286 {
287         struct mtd_part *part = PART(mtd);
288         if ((len + ofs) > mtd->size)
289                 return -EINVAL;
290         return mtd_unlock(part->master, ofs + part->offset, len);
291 }
292
293 static int part_is_locked(struct mtd_info *mtd, loff_t ofs, uint64_t len)
294 {
295         struct mtd_part *part = PART(mtd);
296         if ((len + ofs) > mtd->size)
297                 return -EINVAL;
298         return mtd_is_locked(part->master, ofs + part->offset, len);
299 }
300
301 static void part_sync(struct mtd_info *mtd)
302 {
303         struct mtd_part *part = PART(mtd);
304         mtd_sync(part->master);
305 }
306
307 static int part_suspend(struct mtd_info *mtd)
308 {
309         struct mtd_part *part = PART(mtd);
310         return mtd_suspend(part->master);
311 }
312
313 static void part_resume(struct mtd_info *mtd)
314 {
315         struct mtd_part *part = PART(mtd);
316         mtd_resume(part->master);
317 }
318
319 static int part_block_isbad(struct mtd_info *mtd, loff_t ofs)
320 {
321         struct mtd_part *part = PART(mtd);
322         if (ofs >= mtd->size)
323                 return -EINVAL;
324         ofs += part->offset;
325         return mtd_block_isbad(part->master, ofs);
326 }
327
328 static int part_block_markbad(struct mtd_info *mtd, loff_t ofs)
329 {
330         struct mtd_part *part = PART(mtd);
331         int res;
332
333         if (!(mtd->flags & MTD_WRITEABLE))
334                 return -EROFS;
335         if (ofs >= mtd->size)
336                 return -EINVAL;
337         ofs += part->offset;
338         res = mtd_block_markbad(part->master, ofs);
339         if (!res)
340                 mtd->ecc_stats.badblocks++;
341         return res;
342 }
343
344 static inline void free_partition(struct mtd_part *p)
345 {
346         kfree(p->mtd.name);
347         kfree(p);
348 }
349
350 /*
351  * This function unregisters and destroy all slave MTD objects which are
352  * attached to the given master MTD object.
353  */
354
355 int del_mtd_partitions(struct mtd_info *master)
356 {
357         struct mtd_part *slave, *next;
358         int ret, err = 0;
359
360         mutex_lock(&mtd_partitions_mutex);
361         list_for_each_entry_safe(slave, next, &mtd_partitions, list)
362                 if (slave->master == master) {
363                         ret = del_mtd_device(&slave->mtd);
364                         if (ret < 0) {
365                                 err = ret;
366                                 continue;
367                         }
368                         list_del(&slave->list);
369                         free_partition(slave);
370                 }
371         mutex_unlock(&mtd_partitions_mutex);
372
373         return err;
374 }
375
376 static struct mtd_part *allocate_partition(struct mtd_info *master,
377                         const struct mtd_partition *part, int partno,
378                         uint64_t cur_offset)
379 {
380         struct mtd_part *slave;
381         char *name;
382
383         /* allocate the partition structure */
384         slave = kzalloc(sizeof(*slave), GFP_KERNEL);
385         name = kstrdup(part->name, GFP_KERNEL);
386         if (!name || !slave) {
387                 printk(KERN_ERR"memory allocation error while creating partitions for \"%s\"\n",
388                        master->name);
389                 kfree(name);
390                 kfree(slave);
391                 return ERR_PTR(-ENOMEM);
392         }
393
394         /* set up the MTD object for this partition */
395         slave->mtd.type = master->type;
396         slave->mtd.flags = master->flags & ~part->mask_flags;
397         slave->mtd.size = part->size;
398         slave->mtd.writesize = master->writesize;
399         slave->mtd.writebufsize = master->writebufsize;
400         slave->mtd.oobsize = master->oobsize;
401         slave->mtd.oobavail = master->oobavail;
402         slave->mtd.subpage_sft = master->subpage_sft;
403
404         slave->mtd.name = name;
405         slave->mtd.owner = master->owner;
406         slave->mtd.backing_dev_info = master->backing_dev_info;
407
408         /* NOTE:  we don't arrange MTDs as a tree; it'd be error-prone
409          * to have the same data be in two different partitions.
410          */
411         slave->mtd.dev.parent = master->dev.parent;
412
413         slave->mtd._read = part_read;
414         slave->mtd._write = part_write;
415
416         if (master->_panic_write)
417                 slave->mtd._panic_write = part_panic_write;
418
419         if (master->_point && master->_unpoint) {
420                 slave->mtd._point = part_point;
421                 slave->mtd._unpoint = part_unpoint;
422         }
423
424         if (master->_get_unmapped_area)
425                 slave->mtd._get_unmapped_area = part_get_unmapped_area;
426         if (master->_read_oob)
427                 slave->mtd._read_oob = part_read_oob;
428         if (master->_write_oob)
429                 slave->mtd._write_oob = part_write_oob;
430         if (master->_read_user_prot_reg)
431                 slave->mtd._read_user_prot_reg = part_read_user_prot_reg;
432         if (master->_read_fact_prot_reg)
433                 slave->mtd._read_fact_prot_reg = part_read_fact_prot_reg;
434         if (master->_write_user_prot_reg)
435                 slave->mtd._write_user_prot_reg = part_write_user_prot_reg;
436         if (master->_lock_user_prot_reg)
437                 slave->mtd._lock_user_prot_reg = part_lock_user_prot_reg;
438         if (master->_get_user_prot_info)
439                 slave->mtd._get_user_prot_info = part_get_user_prot_info;
440         if (master->_get_fact_prot_info)
441                 slave->mtd._get_fact_prot_info = part_get_fact_prot_info;
442         if (master->_sync)
443                 slave->mtd._sync = part_sync;
444         if (!partno && !master->dev.class && master->_suspend &&
445             master->_resume) {
446                         slave->mtd._suspend = part_suspend;
447                         slave->mtd._resume = part_resume;
448         }
449         if (master->_writev)
450                 slave->mtd._writev = part_writev;
451         if (master->_lock)
452                 slave->mtd._lock = part_lock;
453         if (master->_unlock)
454                 slave->mtd._unlock = part_unlock;
455         if (master->_is_locked)
456                 slave->mtd._is_locked = part_is_locked;
457         if (master->_block_isbad)
458                 slave->mtd._block_isbad = part_block_isbad;
459         if (master->_block_markbad)
460                 slave->mtd._block_markbad = part_block_markbad;
461         slave->mtd._erase = part_erase;
462         slave->master = master;
463         slave->offset = part->offset;
464
465         if (slave->offset == MTDPART_OFS_APPEND)
466                 slave->offset = cur_offset;
467         if (slave->offset == MTDPART_OFS_NXTBLK) {
468                 slave->offset = cur_offset;
469                 if (mtd_mod_by_eb(cur_offset, master) != 0) {
470                         /* Round up to next erasesize */
471                         slave->offset = (mtd_div_by_eb(cur_offset, master) + 1) * master->erasesize;
472                         printk(KERN_NOTICE "Moving partition %d: "
473                                "0x%012llx -> 0x%012llx\n", partno,
474                                (unsigned long long)cur_offset, (unsigned long long)slave->offset);
475                 }
476         }
477         if (slave->offset == MTDPART_OFS_RETAIN) {
478                 slave->offset = cur_offset;
479                 if (master->size - slave->offset >= slave->mtd.size) {
480                         slave->mtd.size = master->size - slave->offset
481                                                         - slave->mtd.size;
482                 } else {
483                         printk(KERN_ERR "mtd partition \"%s\" doesn't have enough space: %#llx < %#llx, disabled\n",
484                                 part->name, master->size - slave->offset,
485                                 slave->mtd.size);
486                         /* register to preserve ordering */
487                         goto out_register;
488                 }
489         }
490         if (slave->mtd.size == MTDPART_SIZ_FULL)
491                 slave->mtd.size = master->size - slave->offset;
492
493         printk(KERN_NOTICE "0x%012llx-0x%012llx : \"%s\"\n", (unsigned long long)slave->offset,
494                 (unsigned long long)(slave->offset + slave->mtd.size), slave->mtd.name);
495
496         /* let's do some sanity checks */
497         if (slave->offset >= master->size) {
498                 /* let's register it anyway to preserve ordering */
499                 slave->offset = 0;
500                 slave->mtd.size = 0;
501                 printk(KERN_ERR"mtd: partition \"%s\" is out of reach -- disabled\n",
502                         part->name);
503                 goto out_register;
504         }
505         if (slave->offset + slave->mtd.size > master->size) {
506                 slave->mtd.size = master->size - slave->offset;
507                 printk(KERN_WARNING"mtd: partition \"%s\" extends beyond the end of device \"%s\" -- size truncated to %#llx\n",
508                         part->name, master->name, (unsigned long long)slave->mtd.size);
509         }
510         if (master->numeraseregions > 1) {
511                 /* Deal with variable erase size stuff */
512                 int i, max = master->numeraseregions;
513                 u64 end = slave->offset + slave->mtd.size;
514                 struct mtd_erase_region_info *regions = master->eraseregions;
515
516                 /* Find the first erase regions which is part of this
517                  * partition. */
518                 for (i = 0; i < max && regions[i].offset <= slave->offset; i++)
519                         ;
520                 /* The loop searched for the region _behind_ the first one */
521                 if (i > 0)
522                         i--;
523
524                 /* Pick biggest erasesize */
525                 for (; i < max && regions[i].offset < end; i++) {
526                         if (slave->mtd.erasesize < regions[i].erasesize) {
527                                 slave->mtd.erasesize = regions[i].erasesize;
528                         }
529                 }
530                 BUG_ON(slave->mtd.erasesize == 0);
531         } else {
532                 /* Single erase size */
533                 slave->mtd.erasesize = master->erasesize;
534         }
535
536         if ((slave->mtd.flags & MTD_WRITEABLE) &&
537             mtd_mod_by_eb(slave->offset, &slave->mtd)) {
538                 /* Doesn't start on a boundary of major erase size */
539                 /* FIXME: Let it be writable if it is on a boundary of
540                  * _minor_ erase size though */
541                 slave->mtd.flags &= ~MTD_WRITEABLE;
542                 printk(KERN_WARNING"mtd: partition \"%s\" doesn't start on an erase block boundary -- force read-only\n",
543                         part->name);
544         }
545         if ((slave->mtd.flags & MTD_WRITEABLE) &&
546             mtd_mod_by_eb(slave->mtd.size, &slave->mtd)) {
547                 slave->mtd.flags &= ~MTD_WRITEABLE;
548                 printk(KERN_WARNING"mtd: partition \"%s\" doesn't end on an erase block -- force read-only\n",
549                         part->name);
550         }
551
552         slave->mtd.ecclayout = master->ecclayout;
553         if (master->_block_isbad) {
554                 uint64_t offs = 0;
555
556                 while (offs < slave->mtd.size) {
557                         if (mtd_block_isbad(master, offs + slave->offset))
558                                 slave->mtd.ecc_stats.badblocks++;
559                         offs += slave->mtd.erasesize;
560                 }
561         }
562
563 out_register:
564         return slave;
565 }
566
567 int mtd_add_partition(struct mtd_info *master, char *name,
568                       long long offset, long long length)
569 {
570         struct mtd_partition part;
571         struct mtd_part *p, *new;
572         uint64_t start, end;
573         int ret = 0;
574
575         /* the direct offset is expected */
576         if (offset == MTDPART_OFS_APPEND ||
577             offset == MTDPART_OFS_NXTBLK)
578                 return -EINVAL;
579
580         if (length == MTDPART_SIZ_FULL)
581                 length = master->size - offset;
582
583         if (length <= 0)
584                 return -EINVAL;
585
586         part.name = name;
587         part.size = length;
588         part.offset = offset;
589         part.mask_flags = 0;
590         part.ecclayout = NULL;
591
592         new = allocate_partition(master, &part, -1, offset);
593         if (IS_ERR(new))
594                 return PTR_ERR(new);
595
596         start = offset;
597         end = offset + length;
598
599         mutex_lock(&mtd_partitions_mutex);
600         list_for_each_entry(p, &mtd_partitions, list)
601                 if (p->master == master) {
602                         if ((start >= p->offset) &&
603                             (start < (p->offset + p->mtd.size)))
604                                 goto err_inv;
605
606                         if ((end >= p->offset) &&
607                             (end < (p->offset + p->mtd.size)))
608                                 goto err_inv;
609                 }
610
611         list_add(&new->list, &mtd_partitions);
612         mutex_unlock(&mtd_partitions_mutex);
613
614         add_mtd_device(&new->mtd);
615
616         return ret;
617 err_inv:
618         mutex_unlock(&mtd_partitions_mutex);
619         free_partition(new);
620         return -EINVAL;
621 }
622 EXPORT_SYMBOL_GPL(mtd_add_partition);
623
624 int mtd_del_partition(struct mtd_info *master, int partno)
625 {
626         struct mtd_part *slave, *next;
627         int ret = -EINVAL;
628
629         mutex_lock(&mtd_partitions_mutex);
630         list_for_each_entry_safe(slave, next, &mtd_partitions, list)
631                 if ((slave->master == master) &&
632                     (slave->mtd.index == partno)) {
633                         ret = del_mtd_device(&slave->mtd);
634                         if (ret < 0)
635                                 break;
636
637                         list_del(&slave->list);
638                         free_partition(slave);
639                         break;
640                 }
641         mutex_unlock(&mtd_partitions_mutex);
642
643         return ret;
644 }
645 EXPORT_SYMBOL_GPL(mtd_del_partition);
646
647 /*
648  * This function, given a master MTD object and a partition table, creates
649  * and registers slave MTD objects which are bound to the master according to
650  * the partition definitions.
651  *
652  * We don't register the master, or expect the caller to have done so,
653  * for reasons of data integrity.
654  */
655
656 int add_mtd_partitions(struct mtd_info *master,
657                        const struct mtd_partition *parts,
658                        int nbparts)
659 {
660         struct mtd_part *slave;
661         uint64_t cur_offset = 0;
662         int i;
663
664         printk(KERN_NOTICE "Creating %d MTD partitions on \"%s\":\n", nbparts, master->name);
665
666         for (i = 0; i < nbparts; i++) {
667                 slave = allocate_partition(master, parts + i, i, cur_offset);
668                 if (IS_ERR(slave))
669                         return PTR_ERR(slave);
670
671                 mutex_lock(&mtd_partitions_mutex);
672                 list_add(&slave->list, &mtd_partitions);
673                 mutex_unlock(&mtd_partitions_mutex);
674
675                 add_mtd_device(&slave->mtd);
676
677                 cur_offset = slave->offset + slave->mtd.size;
678         }
679
680         return 0;
681 }
682
683 static DEFINE_SPINLOCK(part_parser_lock);
684 static LIST_HEAD(part_parsers);
685
686 static struct mtd_part_parser *get_partition_parser(const char *name)
687 {
688         struct mtd_part_parser *p, *ret = NULL;
689
690         spin_lock(&part_parser_lock);
691
692         list_for_each_entry(p, &part_parsers, list)
693                 if (!strcmp(p->name, name) && try_module_get(p->owner)) {
694                         ret = p;
695                         break;
696                 }
697
698         spin_unlock(&part_parser_lock);
699
700         return ret;
701 }
702
703 #define put_partition_parser(p) do { module_put((p)->owner); } while (0)
704
705 int register_mtd_parser(struct mtd_part_parser *p)
706 {
707         spin_lock(&part_parser_lock);
708         list_add(&p->list, &part_parsers);
709         spin_unlock(&part_parser_lock);
710
711         return 0;
712 }
713 EXPORT_SYMBOL_GPL(register_mtd_parser);
714
715 int deregister_mtd_parser(struct mtd_part_parser *p)
716 {
717         spin_lock(&part_parser_lock);
718         list_del(&p->list);
719         spin_unlock(&part_parser_lock);
720         return 0;
721 }
722 EXPORT_SYMBOL_GPL(deregister_mtd_parser);
723
724 /*
725  * Do not forget to update 'parse_mtd_partitions()' kerneldoc comment if you
726  * are changing this array!
727  */
728 static const char *default_mtd_part_types[] = {
729         "cmdlinepart",
730         "ofpart",
731         NULL
732 };
733
734 /**
735  * parse_mtd_partitions - parse MTD partitions
736  * @master: the master partition (describes whole MTD device)
737  * @types: names of partition parsers to try or %NULL
738  * @pparts: array of partitions found is returned here
739  * @data: MTD partition parser-specific data
740  *
741  * This function tries to find partition on MTD device @master. It uses MTD
742  * partition parsers, specified in @types. However, if @types is %NULL, then
743  * the default list of parsers is used. The default list contains only the
744  * "cmdlinepart" and "ofpart" parsers ATM.
745  *
746  * This function may return:
747  * o a negative error code in case of failure
748  * o zero if no partitions were found
749  * o a positive number of found partitions, in which case on exit @pparts will
750  *   point to an array containing this number of &struct mtd_info objects.
751  */
752 int parse_mtd_partitions(struct mtd_info *master, const char **types,
753                          struct mtd_partition **pparts,
754                          struct mtd_part_parser_data *data)
755 {
756         struct mtd_part_parser *parser;
757         int ret = 0;
758
759         if (!types)
760                 types = default_mtd_part_types;
761
762         for ( ; ret <= 0 && *types; types++) {
763                 parser = get_partition_parser(*types);
764                 if (!parser && !request_module("%s", *types))
765                         parser = get_partition_parser(*types);
766                 if (!parser)
767                         continue;
768                 ret = (*parser->parse_fn)(master, pparts, data);
769                 if (ret > 0) {
770                         printk(KERN_NOTICE "%d %s partitions found on MTD device %s\n",
771                                ret, parser->name, master->name);
772                 }
773                 put_partition_parser(parser);
774         }
775         return ret;
776 }
777
778 int mtd_is_partition(struct mtd_info *mtd)
779 {
780         struct mtd_part *part;
781         int ispart = 0;
782
783         mutex_lock(&mtd_partitions_mutex);
784         list_for_each_entry(part, &mtd_partitions, list)
785                 if (&part->mtd == mtd) {
786                         ispart = 1;
787                         break;
788                 }
789         mutex_unlock(&mtd_partitions_mutex);
790
791         return ispart;
792 }
793 EXPORT_SYMBOL_GPL(mtd_is_partition);