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