git.oblomov.eu Git - linux-2.6/blob - drivers/mtd/nand/nand_bbt.c

   1 /*
   2  *  drivers/mtd/nand_bbt.c
   3  *
   4  *  Overview:
   5  *   Bad block table support for the NAND driver
   6  *
   7  *  Copyright (C) 2004 Thomas Gleixner (tglx@linutronix.de)
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License version 2 as
  11  * published by the Free Software Foundation.
  12  *
  13  * Description:
  14  *
  15  * When nand_scan_bbt is called, then it tries to find the bad block table
  16  * depending on the options in the bbt descriptor(s). If a bbt is found
  17  * then the contents are read and the memory based bbt is created. If a
  18  * mirrored bbt is selected then the mirror is searched too and the
  19  * versions are compared. If the mirror has a greater version number
  20  * than the mirror bbt is used to build the memory based bbt.
  21  * If the tables are not versioned, then we "or" the bad block information.
  22  * If one of the bbt's is out of date or does not exist it is (re)created.
  23  * If no bbt exists at all then the device is scanned for factory marked
  24  * good / bad blocks and the bad block tables are created.
  25  *
  26  * For manufacturer created bbts like the one found on M-SYS DOC devices
  27  * the bbt is searched and read but never created
  28  *
  29  * The autogenerated bad block table is located in the last good blocks
  30  * of the device. The table is mirrored, so it can be updated eventually.
  31  * The table is marked in the oob area with an ident pattern and a version
  32  * number which indicates which of both tables is more up to date.
  33  *
  34  * The table uses 2 bits per block
  35  * 11b:         block is good
  36  * 00b:         block is factory marked bad
  37  * 01b, 10b:    block is marked bad due to wear
  38  *
  39  * The memory bad block table uses the following scheme:
  40  * 00b:         block is good
  41  * 01b:         block is marked bad due to wear
  42  * 10b:         block is reserved (to protect the bbt area)
  43  * 11b:         block is factory marked bad
  44  *
  45  * Multichip devices like DOC store the bad block info per floor.
  46  *
  47  * Following assumptions are made:
  48  * - bbts start at a page boundary, if autolocated on a block boundary
  49  * - the space necessary for a bbt in FLASH does not exceed a block boundary
  50  *
  51  */
  52
  53 #include <linux/slab.h>
  54 #include <linux/types.h>
  55 #include <linux/mtd/mtd.h>
  56 #include <linux/mtd/nand.h>
  57 #include <linux/mtd/nand_ecc.h>
  58 #include <linux/mtd/compatmac.h>
  59 #include <linux/bitops.h>
  60 #include <linux/delay.h>
  61 #include <linux/vmalloc.h>
  62
  63 /**
  64  * check_pattern - [GENERIC] check if a pattern is in the buffer
  65  * @buf:        the buffer to search
  66  * @len:        the length of buffer to search
  67  * @paglen:     the pagelength
  68  * @td:         search pattern descriptor
  69  *
  70  * Check for a pattern at the given place. Used to search bad block
  71  * tables and good / bad block identifiers.
  72  * If the SCAN_EMPTY option is set then check, if all bytes except the
  73  * pattern area contain 0xff
  74  *
  75 */
  76 static int check_pattern(uint8_t *buf, int len, int paglen, struct nand_bbt_descr *td)
  77 {
  78         int i, end = 0;
  79         uint8_t *p = buf;
  80
  81         end = paglen + td->offs;
  82         if (td->options & NAND_BBT_SCANEMPTY) {
  83                 for (i = 0; i < end; i++) {
  84                         if (p[i] != 0xff)
  85                                 return -1;
  86                 }
  87         }
  88         p += end;
  89
  90         /* Compare the pattern */
  91         for (i = 0; i < td->len; i++) {
  92                 if (p[i] != td->pattern[i])
  93                         return -1;
  94         }
  95
  96         if (td->options & NAND_BBT_SCANEMPTY) {
  97                 p += td->len;
  98                 end += td->len;
  99                 for (i = end; i < len; i++) {
 100                         if (*p++ != 0xff)
 101                                 return -1;
 102                 }
 103         }
 104         return 0;
 105 }
 106
 107 /**
 108  * check_short_pattern - [GENERIC] check if a pattern is in the buffer
 109  * @buf:        the buffer to search
 110  * @td:         search pattern descriptor
 111  *
 112  * Check for a pattern at the given place. Used to search bad block
 113  * tables and good / bad block identifiers. Same as check_pattern, but
 114  * no optional empty check
 115  *
 116 */
 117 static int check_short_pattern(uint8_t *buf, struct nand_bbt_descr *td)
 118 {
 119         int i;
 120         uint8_t *p = buf;
 121
 122         /* Compare the pattern */
 123         for (i = 0; i < td->len; i++) {
 124                 if (p[td->offs + i] != td->pattern[i])
 125                         return -1;
 126         }
 127         return 0;
 128 }
 129
 130 /**
 131  * read_bbt - [GENERIC] Read the bad block table starting from page
 132  * @mtd:        MTD device structure
 133  * @buf:        temporary buffer
 134  * @page:       the starting page
 135  * @num:        the number of bbt descriptors to read
 136  * @bits:       number of bits per block
 137  * @offs:       offset in the memory table
 138  * @reserved_block_code:        Pattern to identify reserved blocks
 139  *
 140  * Read the bad block table starting from page.
 141  *
 142  */
 143 static int read_bbt(struct mtd_info *mtd, uint8_t *buf, int page, int num,
 144                     int bits, int offs, int reserved_block_code)
 145 {
 146         int res, i, j, act = 0;
 147         struct nand_chip *this = mtd->priv;
 148         size_t retlen, len, totlen;
 149         loff_t from;
 150         uint8_t msk = (uint8_t) ((1 << bits) - 1);
 151
 152         totlen = (num * bits) >> 3;
 153         from = ((loff_t) page) << this->page_shift;
 154
 155         while (totlen) {
 156                 len = min(totlen, (size_t) (1 << this->bbt_erase_shift));
 157                 res = mtd->read(mtd, from, len, &retlen, buf);
 158                 if (res < 0) {
 159                         if (retlen != len) {
 160                                 printk(KERN_INFO "nand_bbt: Error reading bad block table\n");
 161                                 return res;
 162                         }
 163                         printk(KERN_WARNING "nand_bbt: ECC error while reading bad block table\n");
 164                 }
 165
 166                 /* Analyse data */
 167                 for (i = 0; i < len; i++) {
 168                         uint8_t dat = buf[i];
 169                         for (j = 0; j < 8; j += bits, act += 2) {
 170                                 uint8_t tmp = (dat >> j) & msk;
 171                                 if (tmp == msk)
 172                                         continue;
 173                                 if (reserved_block_code && (tmp == reserved_block_code)) {
 174                                         printk(KERN_DEBUG "nand_read_bbt: Reserved block at 0x%012llx\n",
 175                                                (loff_t)((offs << 2) + (act >> 1)) << this->bbt_erase_shift);
 176                                         this->bbt[offs + (act >> 3)] |= 0x2 << (act & 0x06);
 177                                         mtd->ecc_stats.bbtblocks++;
 178                                         continue;
 179                                 }
 180                                 /* Leave it for now, if its matured we can move this
 181                                  * message to MTD_DEBUG_LEVEL0 */
 182                                 printk(KERN_DEBUG "nand_read_bbt: Bad block at 0x%012llx\n",
 183                                        (loff_t)((offs << 2) + (act >> 1)) << this->bbt_erase_shift);
 184                                 /* Factory marked bad or worn out ? */
 185                                 if (tmp == 0)
 186                                         this->bbt[offs + (act >> 3)] |= 0x3 << (act & 0x06);
 187                                 else
 188                                         this->bbt[offs + (act >> 3)] |= 0x1 << (act & 0x06);
 189                                 mtd->ecc_stats.badblocks++;
 190                         }
 191                 }
 192                 totlen -= len;
 193                 from += len;
 194         }
 195         return 0;
 196 }
 197
 198 /**
 199  * read_abs_bbt - [GENERIC] Read the bad block table starting at a given page
 200  * @mtd:        MTD device structure
 201  * @buf:        temporary buffer
 202  * @td:         descriptor for the bad block table
 203  * @chip:       read the table for a specific chip, -1 read all chips.
 204  *              Applies only if NAND_BBT_PERCHIP option is set
 205  *
 206  * Read the bad block table for all chips starting at a given page
 207  * We assume that the bbt bits are in consecutive order.
 208 */
 209 static int read_abs_bbt(struct mtd_info *mtd, uint8_t *buf, struct nand_bbt_descr *td, int chip)
 210 {
 211         struct nand_chip *this = mtd->priv;
 212         int res = 0, i;
 213         int bits;
 214
 215         bits = td->options & NAND_BBT_NRBITS_MSK;
 216         if (td->options & NAND_BBT_PERCHIP) {
 217                 int offs = 0;
 218                 for (i = 0; i < this->numchips; i++) {
 219                         if (chip == -1 || chip == i)
 220                                 res = read_bbt (mtd, buf, td->pages[i], this->chipsize >> this->bbt_erase_shift, bits, offs, td->reserved_block_code);
 221                         if (res)
 222                                 return res;
 223                         offs += this->chipsize >> (this->bbt_erase_shift + 2);
 224                 }
 225         } else {
 226                 res = read_bbt (mtd, buf, td->pages[0], mtd->size >> this->bbt_erase_shift, bits, 0, td->reserved_block_code);
 227                 if (res)
 228                         return res;
 229         }
 230         return 0;
 231 }
 232
 233 /*
 234  * Scan read raw data from flash
 235  */
 236 static int scan_read_raw(struct mtd_info *mtd, uint8_t *buf, loff_t offs,
 237                          size_t len)
 238 {
 239         struct mtd_oob_ops ops;
 240
 241         ops.mode = MTD_OOB_RAW;
 242         ops.ooboffs = 0;
 243         ops.ooblen = mtd->oobsize;
 244         ops.oobbuf = buf;
 245         ops.datbuf = buf;
 246         ops.len = len;
 247
 248         return mtd->read_oob(mtd, offs, &ops);
 249 }
 250
 251 /*
 252  * Scan write data with oob to flash
 253  */
 254 static int scan_write_bbt(struct mtd_info *mtd, loff_t offs, size_t len,
 255                           uint8_t *buf, uint8_t *oob)
 256 {
 257         struct mtd_oob_ops ops;
 258
 259         ops.mode = MTD_OOB_PLACE;
 260         ops.ooboffs = 0;
 261         ops.ooblen = mtd->oobsize;
 262         ops.datbuf = buf;
 263         ops.oobbuf = oob;
 264         ops.len = len;
 265
 266         return mtd->write_oob(mtd, offs, &ops);
 267 }
 268
 269 /**
 270  * read_abs_bbts - [GENERIC] Read the bad block table(s) for all chips starting at a given page
 271  * @mtd:        MTD device structure
 272  * @buf:        temporary buffer
 273  * @td:         descriptor for the bad block table
 274  * @md:         descriptor for the bad block table mirror
 275  *
 276  * Read the bad block table(s) for all chips starting at a given page
 277  * We assume that the bbt bits are in consecutive order.
 278  *
 279 */
 280 static int read_abs_bbts(struct mtd_info *mtd, uint8_t *buf,
 281                          struct nand_bbt_descr *td, struct nand_bbt_descr *md)
 282 {
 283         struct nand_chip *this = mtd->priv;
 284
 285         /* Read the primary version, if available */
 286         if (td->options & NAND_BBT_VERSION) {
 287                 scan_read_raw(mtd, buf, (loff_t)td->pages[0] << this->page_shift,
 288                               mtd->writesize);
 289                 td->version[0] = buf[mtd->writesize + td->veroffs];
 290                 printk(KERN_DEBUG "Bad block table at page %d, version 0x%02X\n",
 291                        td->pages[0], td->version[0]);
 292         }
 293
 294         /* Read the mirror version, if available */
 295         if (md && (md->options & NAND_BBT_VERSION)) {
 296                 scan_read_raw(mtd, buf, (loff_t)md->pages[0] << this->page_shift,
 297                               mtd->writesize);
 298                 md->version[0] = buf[mtd->writesize + md->veroffs];
 299                 printk(KERN_DEBUG "Bad block table at page %d, version 0x%02X\n",
 300                        md->pages[0], md->version[0]);
 301         }
 302         return 1;
 303 }
 304
 305 /*
 306  * Scan a given block full
 307  */
 308 static int scan_block_full(struct mtd_info *mtd, struct nand_bbt_descr *bd,
 309                            loff_t offs, uint8_t *buf, size_t readlen,
 310                            int scanlen, int len)
 311 {
 312         int ret, j;
 313
 314         ret = scan_read_raw(mtd, buf, offs, readlen);
 315         if (ret)
 316                 return ret;
 317
 318         for (j = 0; j < len; j++, buf += scanlen) {
 319                 if (check_pattern(buf, scanlen, mtd->writesize, bd))
 320                         return 1;
 321         }
 322         return 0;
 323 }
 324
 325 /*
 326  * Scan a given block partially
 327  */
 328 static int scan_block_fast(struct mtd_info *mtd, struct nand_bbt_descr *bd,
 329                            loff_t offs, uint8_t *buf, int len)
 330 {
 331         struct mtd_oob_ops ops;
 332         int j, ret;
 333
 334         ops.ooblen = mtd->oobsize;
 335         ops.oobbuf = buf;
 336         ops.ooboffs = 0;
 337         ops.datbuf = NULL;
 338         ops.mode = MTD_OOB_PLACE;
 339
 340         for (j = 0; j < len; j++) {
 341                 /*
 342                  * Read the full oob until read_oob is fixed to
 343                  * handle single byte reads for 16 bit
 344                  * buswidth
 345                  */
 346                 ret = mtd->read_oob(mtd, offs, &ops);
 347                 if (ret)
 348                         return ret;
 349
 350                 if (check_short_pattern(buf, bd))
 351                         return 1;
 352
 353                 offs += mtd->writesize;
 354         }
 355         return 0;
 356 }
 357
 358 /**
 359  * create_bbt - [GENERIC] Create a bad block table by scanning the device
 360  * @mtd:        MTD device structure
 361  * @buf:        temporary buffer
 362  * @bd:         descriptor for the good/bad block search pattern
 363  * @chip:       create the table for a specific chip, -1 read all chips.
 364  *              Applies only if NAND_BBT_PERCHIP option is set
 365  *
 366  * Create a bad block table by scanning the device
 367  * for the given good/bad block identify pattern
 368  */
 369 static int create_bbt(struct mtd_info *mtd, uint8_t *buf,
 370         struct nand_bbt_descr *bd, int chip)
 371 {
 372         struct nand_chip *this = mtd->priv;
 373         int i, numblocks, len, scanlen;
 374         int startblock;
 375         loff_t from;
 376         size_t readlen;
 377
 378         printk(KERN_INFO "Scanning device for bad blocks\n");
 379
 380         if (bd->options & NAND_BBT_SCANALLPAGES)
 381                 len = 1 << (this->bbt_erase_shift - this->page_shift);
 382         else {
 383                 if (bd->options & NAND_BBT_SCAN2NDPAGE)
 384                         len = 2;
 385                 else
 386                         len = 1;
 387         }
 388
 389         if (!(bd->options & NAND_BBT_SCANEMPTY)) {
 390                 /* We need only read few bytes from the OOB area */
 391                 scanlen = 0;
 392                 readlen = bd->len;
 393         } else {
 394                 /* Full page content should be read */
 395                 scanlen = mtd->writesize + mtd->oobsize;
 396                 readlen = len * mtd->writesize;
 397         }
 398
 399         if (chip == -1) {
 400                 /* Note that numblocks is 2 * (real numblocks) here, see i+=2
 401                  * below as it makes shifting and masking less painful */
 402                 numblocks = mtd->size >> (this->bbt_erase_shift - 1);
 403                 startblock = 0;
 404                 from = 0;
 405         } else {
 406                 if (chip >= this->numchips) {
 407                         printk(KERN_WARNING "create_bbt(): chipnr (%d) > available chips (%d)\n",
 408                                chip + 1, this->numchips);
 409                         return -EINVAL;
 410                 }
 411                 numblocks = this->chipsize >> (this->bbt_erase_shift - 1);
 412                 startblock = chip * numblocks;
 413                 numblocks += startblock;
 414                 from = (loff_t)startblock << (this->bbt_erase_shift - 1);
 415         }
 416
 417         for (i = startblock; i < numblocks;) {
 418                 int ret;
 419
 420                 if (bd->options & NAND_BBT_SCANALLPAGES)
 421                         ret = scan_block_full(mtd, bd, from, buf, readlen,
 422                                               scanlen, len);
 423                 else
 424                         ret = scan_block_fast(mtd, bd, from, buf, len);
 425
 426                 if (ret < 0)
 427                         return ret;
 428
 429                 if (ret) {
 430                         this->bbt[i >> 3] |= 0x03 << (i & 0x6);
 431                         printk(KERN_WARNING "Bad eraseblock %d at 0x%012llx\n",
 432                                i >> 1, (unsigned long long)from);
 433                         mtd->ecc_stats.badblocks++;
 434                 }
 435
 436                 i += 2;
 437                 from += (1 << this->bbt_erase_shift);
 438         }
 439         return 0;
 440 }
 441
 442 /**
 443  * search_bbt - [GENERIC] scan the device for a specific bad block table
 444  * @mtd:        MTD device structure
 445  * @buf:        temporary buffer
 446  * @td:         descriptor for the bad block table
 447  *
 448  * Read the bad block table by searching for a given ident pattern.
 449  * Search is preformed either from the beginning up or from the end of
 450  * the device downwards. The search starts always at the start of a
 451  * block.
 452  * If the option NAND_BBT_PERCHIP is given, each chip is searched
 453  * for a bbt, which contains the bad block information of this chip.
 454  * This is necessary to provide support for certain DOC devices.
 455  *
 456  * The bbt ident pattern resides in the oob area of the first page
 457  * in a block.
 458  */
 459 static int search_bbt(struct mtd_info *mtd, uint8_t *buf, struct nand_bbt_descr *td)
 460 {
 461         struct nand_chip *this = mtd->priv;
 462         int i, chips;
 463         int bits, startblock, block, dir;
 464         int scanlen = mtd->writesize + mtd->oobsize;
 465         int bbtblocks;
 466         int blocktopage = this->bbt_erase_shift - this->page_shift;
 467
 468         /* Search direction top -> down ? */
 469         if (td->options & NAND_BBT_LASTBLOCK) {
 470                 startblock = (mtd->size >> this->bbt_erase_shift) - 1;
 471                 dir = -1;
 472         } else {
 473                 startblock = 0;
 474                 dir = 1;
 475         }
 476
 477         /* Do we have a bbt per chip ? */
 478         if (td->options & NAND_BBT_PERCHIP) {
 479                 chips = this->numchips;
 480                 bbtblocks = this->chipsize >> this->bbt_erase_shift;
 481                 startblock &= bbtblocks - 1;
 482         } else {
 483                 chips = 1;
 484                 bbtblocks = mtd->size >> this->bbt_erase_shift;
 485         }
 486
 487         /* Number of bits for each erase block in the bbt */
 488         bits = td->options & NAND_BBT_NRBITS_MSK;
 489
 490         for (i = 0; i < chips; i++) {
 491                 /* Reset version information */
 492                 td->version[i] = 0;
 493                 td->pages[i] = -1;
 494                 /* Scan the maximum number of blocks */
 495                 for (block = 0; block < td->maxblocks; block++) {
 496
 497                         int actblock = startblock + dir * block;
 498                         loff_t offs = (loff_t)actblock << this->bbt_erase_shift;
 499
 500                         /* Read first page */
 501                         scan_read_raw(mtd, buf, offs, mtd->writesize);
 502                         if (!check_pattern(buf, scanlen, mtd->writesize, td)) {
 503                                 td->pages[i] = actblock << blocktopage;
 504                                 if (td->options & NAND_BBT_VERSION) {
 505                                         td->version[i] = buf[mtd->writesize + td->veroffs];
 506                                 }
 507                                 break;
 508                         }
 509                 }
 510                 startblock += this->chipsize >> this->bbt_erase_shift;
 511         }
 512         /* Check, if we found a bbt for each requested chip */
 513         for (i = 0; i < chips; i++) {
 514                 if (td->pages[i] == -1)
 515                         printk(KERN_WARNING "Bad block table not found for chip %d\n", i);
 516                 else
 517                         printk(KERN_DEBUG "Bad block table found at page %d, version 0x%02X\n", td->pages[i],
 518                                td->version[i]);
 519         }
 520         return 0;
 521 }
 522
 523 /**
 524  * search_read_bbts - [GENERIC] scan the device for bad block table(s)
 525  * @mtd:        MTD device structure
 526  * @buf:        temporary buffer
 527  * @td:         descriptor for the bad block table
 528  * @md:         descriptor for the bad block table mirror
 529  *
 530  * Search and read the bad block table(s)
 531 */
 532 static int search_read_bbts(struct mtd_info *mtd, uint8_t * buf, struct nand_bbt_descr *td, struct nand_bbt_descr *md)
 533 {
 534         /* Search the primary table */
 535         search_bbt(mtd, buf, td);
 536
 537         /* Search the mirror table */
 538         if (md)
 539                 search_bbt(mtd, buf, md);
 540
 541         /* Force result check */
 542         return 1;
 543 }
 544
 545 /**
 546  * write_bbt - [GENERIC] (Re)write the bad block table
 547  *
 548  * @mtd:        MTD device structure
 549  * @buf:        temporary buffer
 550  * @td:         descriptor for the bad block table
 551  * @md:         descriptor for the bad block table mirror
 552  * @chipsel:    selector for a specific chip, -1 for all
 553  *
 554  * (Re)write the bad block table
 555  *
 556 */
 557 static int write_bbt(struct mtd_info *mtd, uint8_t *buf,
 558                      struct nand_bbt_descr *td, struct nand_bbt_descr *md,
 559                      int chipsel)
 560 {
 561         struct nand_chip *this = mtd->priv;
 562         struct erase_info einfo;
 563         int i, j, res, chip = 0;
 564         int bits, startblock, dir, page, offs, numblocks, sft, sftmsk;
 565         int nrchips, bbtoffs, pageoffs, ooboffs;
 566         uint8_t msk[4];
 567         uint8_t rcode = td->reserved_block_code;
 568         size_t retlen, len = 0;
 569         loff_t to;
 570         struct mtd_oob_ops ops;
 571
 572         ops.ooblen = mtd->oobsize;
 573         ops.ooboffs = 0;
 574         ops.datbuf = NULL;
 575         ops.mode = MTD_OOB_PLACE;
 576
 577         if (!rcode)
 578                 rcode = 0xff;
 579         /* Write bad block table per chip rather than per device ? */
 580         if (td->options & NAND_BBT_PERCHIP) {
 581                 numblocks = (int)(this->chipsize >> this->bbt_erase_shift);
 582                 /* Full device write or specific chip ? */
 583                 if (chipsel == -1) {
 584                         nrchips = this->numchips;
 585                 } else {
 586                         nrchips = chipsel + 1;
 587                         chip = chipsel;
 588                 }
 589         } else {
 590                 numblocks = (int)(mtd->size >> this->bbt_erase_shift);
 591                 nrchips = 1;
 592         }
 593
 594         /* Loop through the chips */
 595         for (; chip < nrchips; chip++) {
 596
 597                 /* There was already a version of the table, reuse the page
 598                  * This applies for absolute placement too, as we have the
 599                  * page nr. in td->pages.
 600                  */
 601                 if (td->pages[chip] != -1) {
 602                         page = td->pages[chip];
 603                         goto write;
 604                 }
 605
 606                 /* Automatic placement of the bad block table */
 607                 /* Search direction top -> down ? */
 608                 if (td->options & NAND_BBT_LASTBLOCK) {
 609                         startblock = numblocks * (chip + 1) - 1;
 610                         dir = -1;
 611                 } else {
 612                         startblock = chip * numblocks;
 613                         dir = 1;
 614                 }
 615
 616                 for (i = 0; i < td->maxblocks; i++) {
 617                         int block = startblock + dir * i;
 618                         /* Check, if the block is bad */
 619                         switch ((this->bbt[block >> 2] >>
 620                                  (2 * (block & 0x03))) & 0x03) {
 621                         case 0x01:
 622                         case 0x03:
 623                                 continue;
 624                         }
 625                         page = block <<
 626                                 (this->bbt_erase_shift - this->page_shift);
 627                         /* Check, if the block is used by the mirror table */
 628                         if (!md || md->pages[chip] != page)
 629                                 goto write;
 630                 }
 631                 printk(KERN_ERR "No space left to write bad block table\n");
 632                 return -ENOSPC;
 633         write:
 634
 635                 /* Set up shift count and masks for the flash table */
 636                 bits = td->options & NAND_BBT_NRBITS_MSK;
 637                 msk[2] = ~rcode;
 638                 switch (bits) {
 639                 case 1: sft = 3; sftmsk = 0x07; msk[0] = 0x00; msk[1] = 0x01;
 640                         msk[3] = 0x01;
 641                         break;
 642                 case 2: sft = 2; sftmsk = 0x06; msk[0] = 0x00; msk[1] = 0x01;
 643                         msk[3] = 0x03;
 644                         break;
 645                 case 4: sft = 1; sftmsk = 0x04; msk[0] = 0x00; msk[1] = 0x0C;
 646                         msk[3] = 0x0f;
 647                         break;
 648                 case 8: sft = 0; sftmsk = 0x00; msk[0] = 0x00; msk[1] = 0x0F;
 649                         msk[3] = 0xff;
 650                         break;
 651                 default: return -EINVAL;
 652                 }
 653
 654                 bbtoffs = chip * (numblocks >> 2);
 655
 656                 to = ((loff_t) page) << this->page_shift;
 657
 658                 /* Must we save the block contents ? */
 659                 if (td->options & NAND_BBT_SAVECONTENT) {
 660                         /* Make it block aligned */
 661                         to &= ~((loff_t) ((1 << this->bbt_erase_shift) - 1));
 662                         len = 1 << this->bbt_erase_shift;
 663                         res = mtd->read(mtd, to, len, &retlen, buf);
 664                         if (res < 0) {
 665                                 if (retlen != len) {
 666                                         printk(KERN_INFO "nand_bbt: Error "
 667                                                "reading block for writing "
 668                                                "the bad block table\n");
 669                                         return res;
 670                                 }
 671                                 printk(KERN_WARNING "nand_bbt: ECC error "
 672                                        "while reading block for writing "
 673                                        "bad block table\n");
 674                         }
 675                         /* Read oob data */
 676                         ops.ooblen = (len >> this->page_shift) * mtd->oobsize;
 677                         ops.oobbuf = &buf[len];
 678                         res = mtd->read_oob(mtd, to + mtd->writesize, &ops);
 679                         if (res < 0 || ops.oobretlen != ops.ooblen)
 680                                 goto outerr;
 681
 682                         /* Calc the byte offset in the buffer */
 683                         pageoffs = page - (int)(to >> this->page_shift);
 684                         offs = pageoffs << this->page_shift;
 685                         /* Preset the bbt area with 0xff */
 686                         memset(&buf[offs], 0xff, (size_t) (numblocks >> sft));
 687                         ooboffs = len + (pageoffs * mtd->oobsize);
 688
 689                 } else {
 690                         /* Calc length */
 691                         len = (size_t) (numblocks >> sft);
 692                         /* Make it page aligned ! */
 693                         len = (len + (mtd->writesize - 1)) &
 694                                 ~(mtd->writesize - 1);
 695                         /* Preset the buffer with 0xff */
 696                         memset(buf, 0xff, len +
 697                                (len >> this->page_shift)* mtd->oobsize);
 698                         offs = 0;
 699                         ooboffs = len;
 700                         /* Pattern is located in oob area of first page */
 701                         memcpy(&buf[ooboffs + td->offs], td->pattern, td->len);
 702                 }
 703
 704                 if (td->options & NAND_BBT_VERSION)
 705                         buf[ooboffs + td->veroffs] = td->version[chip];
 706
 707                 /* walk through the memory table */
 708                 for (i = 0; i < numblocks;) {
 709                         uint8_t dat;
 710                         dat = this->bbt[bbtoffs + (i >> 2)];
 711                         for (j = 0; j < 4; j++, i++) {
 712                                 int sftcnt = (i << (3 - sft)) & sftmsk;
 713                                 /* Do not store the reserved bbt blocks ! */
 714                                 buf[offs + (i >> sft)] &=
 715                                         ~(msk[dat & 0x03] << sftcnt);
 716                                 dat >>= 2;
 717                         }
 718                 }
 719
 720                 memset(&einfo, 0, sizeof(einfo));
 721                 einfo.mtd = mtd;
 722                 einfo.addr = to;
 723                 einfo.len = 1 << this->bbt_erase_shift;
 724                 res = nand_erase_nand(mtd, &einfo, 1);
 725                 if (res < 0)
 726                         goto outerr;
 727
 728                 res = scan_write_bbt(mtd, to, len, buf, &buf[len]);
 729                 if (res < 0)
 730                         goto outerr;
 731
 732                 printk(KERN_DEBUG "Bad block table written to 0x%012llx, version "
 733                        "0x%02X\n", (unsigned long long)to, td->version[chip]);
 734
 735                 /* Mark it as used */
 736                 td->pages[chip] = page;
 737         }
 738         return 0;
 739
 740  outerr:
 741         printk(KERN_WARNING
 742                "nand_bbt: Error while writing bad block table %d\n", res);
 743         return res;
 744 }
 745
 746 /**
 747  * nand_memory_bbt - [GENERIC] create a memory based bad block table
 748  * @mtd:        MTD device structure
 749  * @bd:         descriptor for the good/bad block search pattern
 750  *
 751  * The function creates a memory based bbt by scanning the device
 752  * for manufacturer / software marked good / bad blocks
 753 */
 754 static inline int nand_memory_bbt(struct mtd_info *mtd, struct nand_bbt_descr *bd)
 755 {
 756         struct nand_chip *this = mtd->priv;
 757
 758         bd->options &= ~NAND_BBT_SCANEMPTY;
 759         return create_bbt(mtd, this->buffers->databuf, bd, -1);
 760 }
 761
 762 /**
 763  * check_create - [GENERIC] create and write bbt(s) if necessary
 764  * @mtd:        MTD device structure
 765  * @buf:        temporary buffer
 766  * @bd:         descriptor for the good/bad block search pattern
 767  *
 768  * The function checks the results of the previous call to read_bbt
 769  * and creates / updates the bbt(s) if necessary
 770  * Creation is necessary if no bbt was found for the chip/device
 771  * Update is necessary if one of the tables is missing or the
 772  * version nr. of one table is less than the other
 773 */
 774 static int check_create(struct mtd_info *mtd, uint8_t *buf, struct nand_bbt_descr *bd)
 775 {
 776         int i, chips, writeops, chipsel, res;
 777         struct nand_chip *this = mtd->priv;
 778         struct nand_bbt_descr *td = this->bbt_td;
 779         struct nand_bbt_descr *md = this->bbt_md;
 780         struct nand_bbt_descr *rd, *rd2;
 781
 782         /* Do we have a bbt per chip ? */
 783         if (td->options & NAND_BBT_PERCHIP)
 784                 chips = this->numchips;
 785         else
 786                 chips = 1;
 787
 788         for (i = 0; i < chips; i++) {
 789                 writeops = 0;
 790                 rd = NULL;
 791                 rd2 = NULL;
 792                 /* Per chip or per device ? */
 793                 chipsel = (td->options & NAND_BBT_PERCHIP) ? i : -1;
 794                 /* Mirrored table avilable ? */
 795                 if (md) {
 796                         if (td->pages[i] == -1 && md->pages[i] == -1) {
 797                                 writeops = 0x03;
 798                                 goto create;
 799                         }
 800
 801                         if (td->pages[i] == -1) {
 802                                 rd = md;
 803                                 td->version[i] = md->version[i];
 804                                 writeops = 1;
 805                                 goto writecheck;
 806                         }
 807
 808                         if (md->pages[i] == -1) {
 809                                 rd = td;
 810                                 md->version[i] = td->version[i];
 811                                 writeops = 2;
 812                                 goto writecheck;
 813                         }
 814
 815                         if (td->version[i] == md->version[i]) {
 816                                 rd = td;
 817                                 if (!(td->options & NAND_BBT_VERSION))
 818                                         rd2 = md;
 819                                 goto writecheck;
 820                         }
 821
 822                         if (((int8_t) (td->version[i] - md->version[i])) > 0) {
 823                                 rd = td;
 824                                 md->version[i] = td->version[i];
 825                                 writeops = 2;
 826                         } else {
 827                                 rd = md;
 828                                 td->version[i] = md->version[i];
 829                                 writeops = 1;
 830                         }
 831
 832                         goto writecheck;
 833
 834                 } else {
 835                         if (td->pages[i] == -1) {
 836                                 writeops = 0x01;
 837                                 goto create;
 838                         }
 839                         rd = td;
 840                         goto writecheck;
 841                 }
 842         create:
 843                 /* Create the bad block table by scanning the device ? */
 844                 if (!(td->options & NAND_BBT_CREATE))
 845                         continue;
 846
 847                 /* Create the table in memory by scanning the chip(s) */
 848                 create_bbt(mtd, buf, bd, chipsel);
 849
 850                 td->version[i] = 1;
 851                 if (md)
 852                         md->version[i] = 1;
 853         writecheck:
 854                 /* read back first ? */
 855                 if (rd)
 856                         read_abs_bbt(mtd, buf, rd, chipsel);
 857                 /* If they weren't versioned, read both. */
 858                 if (rd2)
 859                         read_abs_bbt(mtd, buf, rd2, chipsel);
 860
 861                 /* Write the bad block table to the device ? */
 862                 if ((writeops & 0x01) && (td->options & NAND_BBT_WRITE)) {
 863                         res = write_bbt(mtd, buf, td, md, chipsel);
 864                         if (res < 0)
 865                                 return res;
 866                 }
 867
 868                 /* Write the mirror bad block table to the device ? */
 869                 if ((writeops & 0x02) && md && (md->options & NAND_BBT_WRITE)) {
 870                         res = write_bbt(mtd, buf, md, td, chipsel);
 871                         if (res < 0)
 872                                 return res;
 873                 }
 874         }
 875         return 0;
 876 }
 877
 878 /**
 879  * mark_bbt_regions - [GENERIC] mark the bad block table regions
 880  * @mtd:        MTD device structure
 881  * @td:         bad block table descriptor
 882  *
 883  * The bad block table regions are marked as "bad" to prevent
 884  * accidental erasures / writes. The regions are identified by
 885  * the mark 0x02.
 886 */
 887 static void mark_bbt_region(struct mtd_info *mtd, struct nand_bbt_descr *td)
 888 {
 889         struct nand_chip *this = mtd->priv;
 890         int i, j, chips, block, nrblocks, update;
 891         uint8_t oldval, newval;
 892
 893         /* Do we have a bbt per chip ? */
 894         if (td->options & NAND_BBT_PERCHIP) {
 895                 chips = this->numchips;
 896                 nrblocks = (int)(this->chipsize >> this->bbt_erase_shift);
 897         } else {
 898                 chips = 1;
 899                 nrblocks = (int)(mtd->size >> this->bbt_erase_shift);
 900         }
 901
 902         for (i = 0; i < chips; i++) {
 903                 if ((td->options & NAND_BBT_ABSPAGE) ||
 904                     !(td->options & NAND_BBT_WRITE)) {
 905                         if (td->pages[i] == -1)
 906                                 continue;
 907                         block = td->pages[i] >> (this->bbt_erase_shift - this->page_shift);
 908                         block <<= 1;
 909                         oldval = this->bbt[(block >> 3)];
 910                         newval = oldval | (0x2 << (block & 0x06));
 911                         this->bbt[(block >> 3)] = newval;
 912                         if ((oldval != newval) && td->reserved_block_code)
 913                                 nand_update_bbt(mtd, (loff_t)block << (this->bbt_erase_shift - 1));
 914                         continue;
 915                 }
 916                 update = 0;
 917                 if (td->options & NAND_BBT_LASTBLOCK)
 918                         block = ((i + 1) * nrblocks) - td->maxblocks;
 919                 else
 920                         block = i * nrblocks;
 921                 block <<= 1;
 922                 for (j = 0; j < td->maxblocks; j++) {
 923                         oldval = this->bbt[(block >> 3)];
 924                         newval = oldval | (0x2 << (block & 0x06));
 925                         this->bbt[(block >> 3)] = newval;
 926                         if (oldval != newval)
 927                                 update = 1;
 928                         block += 2;
 929                 }
 930                 /* If we want reserved blocks to be recorded to flash, and some
 931                    new ones have been marked, then we need to update the stored
 932                    bbts.  This should only happen once. */
 933                 if (update && td->reserved_block_code)
 934                         nand_update_bbt(mtd, (loff_t)(block - 2) << (this->bbt_erase_shift - 1));
 935         }
 936 }
 937
 938 /**
 939  * nand_scan_bbt - [NAND Interface] scan, find, read and maybe create bad block table(s)
 940  * @mtd:        MTD device structure
 941  * @bd:         descriptor for the good/bad block search pattern
 942  *
 943  * The function checks, if a bad block table(s) is/are already
 944  * available. If not it scans the device for manufacturer
 945  * marked good / bad blocks and writes the bad block table(s) to
 946  * the selected place.
 947  *
 948  * The bad block table memory is allocated here. It must be freed
 949  * by calling the nand_free_bbt function.
 950  *
 951 */
 952 int nand_scan_bbt(struct mtd_info *mtd, struct nand_bbt_descr *bd)
 953 {
 954         struct nand_chip *this = mtd->priv;
 955         int len, res = 0;
 956         uint8_t *buf;
 957         struct nand_bbt_descr *td = this->bbt_td;
 958         struct nand_bbt_descr *md = this->bbt_md;
 959
 960         len = mtd->size >> (this->bbt_erase_shift + 2);
 961         /* Allocate memory (2bit per block) and clear the memory bad block table */
 962         this->bbt = kzalloc(len, GFP_KERNEL);
 963         if (!this->bbt) {
 964                 printk(KERN_ERR "nand_scan_bbt: Out of memory\n");
 965                 return -ENOMEM;
 966         }
 967
 968         /* If no primary table decriptor is given, scan the device
 969          * to build a memory based bad block table
 970          */
 971         if (!td) {
 972                 if ((res = nand_memory_bbt(mtd, bd))) {
 973                         printk(KERN_ERR "nand_bbt: Can't scan flash and build the RAM-based BBT\n");
 974                         kfree(this->bbt);
 975                         this->bbt = NULL;
 976                 }
 977                 return res;
 978         }
 979
 980         /* Allocate a temporary buffer for one eraseblock incl. oob */
 981         len = (1 << this->bbt_erase_shift);
 982         len += (len >> this->page_shift) * mtd->oobsize;
 983         buf = vmalloc(len);
 984         if (!buf) {
 985                 printk(KERN_ERR "nand_bbt: Out of memory\n");
 986                 kfree(this->bbt);
 987                 this->bbt = NULL;
 988                 return -ENOMEM;
 989         }
 990
 991         /* Is the bbt at a given page ? */
 992         if (td->options & NAND_BBT_ABSPAGE) {
 993                 res = read_abs_bbts(mtd, buf, td, md);
 994         } else {
 995                 /* Search the bad block table using a pattern in oob */
 996                 res = search_read_bbts(mtd, buf, td, md);
 997         }
 998
 999         if (res)
1000                 res = check_create(mtd, buf, bd);
1001
1002         /* Prevent the bbt regions from erasing / writing */
1003         mark_bbt_region(mtd, td);
1004         if (md)
1005                 mark_bbt_region(mtd, md);
1006
1007         vfree(buf);
1008         return res;
1009 }
1010
1011 /**
1012  * nand_update_bbt - [NAND Interface] update bad block table(s)
1013  * @mtd:        MTD device structure
1014  * @offs:       the offset of the newly marked block
1015  *
1016  * The function updates the bad block table(s)
1017 */
1018 int nand_update_bbt(struct mtd_info *mtd, loff_t offs)
1019 {
1020         struct nand_chip *this = mtd->priv;
1021         int len, res = 0, writeops = 0;
1022         int chip, chipsel;
1023         uint8_t *buf;
1024         struct nand_bbt_descr *td = this->bbt_td;
1025         struct nand_bbt_descr *md = this->bbt_md;
1026
1027         if (!this->bbt || !td)
1028                 return -EINVAL;
1029
1030         /* Allocate a temporary buffer for one eraseblock incl. oob */
1031         len = (1 << this->bbt_erase_shift);
1032         len += (len >> this->page_shift) * mtd->oobsize;
1033         buf = kmalloc(len, GFP_KERNEL);
1034         if (!buf) {
1035                 printk(KERN_ERR "nand_update_bbt: Out of memory\n");
1036                 return -ENOMEM;
1037         }
1038
1039         writeops = md != NULL ? 0x03 : 0x01;
1040
1041         /* Do we have a bbt per chip ? */
1042         if (td->options & NAND_BBT_PERCHIP) {
1043                 chip = (int)(offs >> this->chip_shift);
1044                 chipsel = chip;
1045         } else {
1046                 chip = 0;
1047                 chipsel = -1;
1048         }
1049
1050         td->version[chip]++;
1051         if (md)
1052                 md->version[chip]++;
1053
1054         /* Write the bad block table to the device ? */
1055         if ((writeops & 0x01) && (td->options & NAND_BBT_WRITE)) {
1056                 res = write_bbt(mtd, buf, td, md, chipsel);
1057                 if (res < 0)
1058                         goto out;
1059         }
1060         /* Write the mirror bad block table to the device ? */
1061         if ((writeops & 0x02) && md && (md->options & NAND_BBT_WRITE)) {
1062                 res = write_bbt(mtd, buf, md, td, chipsel);
1063         }
1064
1065  out:
1066         kfree(buf);
1067         return res;
1068 }
1069
1070 /* Define some generic bad / good block scan pattern which are used
1071  * while scanning a device for factory marked good / bad blocks. */
1072 static uint8_t scan_ff_pattern[] = { 0xff, 0xff };
1073
1074 static struct nand_bbt_descr smallpage_memorybased = {
1075         .options = NAND_BBT_SCAN2NDPAGE,
1076         .offs = 5,
1077         .len = 1,
1078         .pattern = scan_ff_pattern
1079 };
1080
1081 static struct nand_bbt_descr largepage_memorybased = {
1082         .options = 0,
1083         .offs = 0,
1084         .len = 2,
1085         .pattern = scan_ff_pattern
1086 };
1087
1088 static struct nand_bbt_descr smallpage_flashbased = {
1089         .options = NAND_BBT_SCAN2NDPAGE,
1090         .offs = 5,
1091         .len = 1,
1092         .pattern = scan_ff_pattern
1093 };
1094
1095 static struct nand_bbt_descr largepage_flashbased = {
1096         .options = NAND_BBT_SCAN2NDPAGE,
1097         .offs = 0,
1098         .len = 2,
1099         .pattern = scan_ff_pattern
1100 };
1101
1102 static uint8_t scan_agand_pattern[] = { 0x1C, 0x71, 0xC7, 0x1C, 0x71, 0xC7 };
1103
1104 static struct nand_bbt_descr agand_flashbased = {
1105         .options = NAND_BBT_SCANEMPTY | NAND_BBT_SCANALLPAGES,
1106         .offs = 0x20,
1107         .len = 6,
1108         .pattern = scan_agand_pattern
1109 };
1110
1111 /* Generic flash bbt decriptors
1112 */
1113 static uint8_t bbt_pattern[] = {'B', 'b', 't', '0' };
1114 static uint8_t mirror_pattern[] = {'1', 't', 'b', 'B' };
1115
1116 static struct nand_bbt_descr bbt_main_descr = {
1117         .options = NAND_BBT_LASTBLOCK | NAND_BBT_CREATE | NAND_BBT_WRITE
1118                 | NAND_BBT_2BIT | NAND_BBT_VERSION | NAND_BBT_PERCHIP,
1119         .offs = 8,
1120         .len = 4,
1121         .veroffs = 12,
1122         .maxblocks = 4,
1123         .pattern = bbt_pattern
1124 };
1125
1126 static struct nand_bbt_descr bbt_mirror_descr = {
1127         .options = NAND_BBT_LASTBLOCK | NAND_BBT_CREATE | NAND_BBT_WRITE
1128                 | NAND_BBT_2BIT | NAND_BBT_VERSION | NAND_BBT_PERCHIP,
1129         .offs = 8,
1130         .len = 4,
1131         .veroffs = 12,
1132         .maxblocks = 4,
1133         .pattern = mirror_pattern
1134 };
1135
1136 /**
1137  * nand_default_bbt - [NAND Interface] Select a default bad block table for the device
1138  * @mtd:        MTD device structure
1139  *
1140  * This function selects the default bad block table
1141  * support for the device and calls the nand_scan_bbt function
1142  *
1143 */
1144 int nand_default_bbt(struct mtd_info *mtd)
1145 {
1146         struct nand_chip *this = mtd->priv;
1147
1148         /* Default for AG-AND. We must use a flash based
1149          * bad block table as the devices have factory marked
1150          * _good_ blocks. Erasing those blocks leads to loss
1151          * of the good / bad information, so we _must_ store
1152          * this information in a good / bad table during
1153          * startup
1154          */
1155         if (this->options & NAND_IS_AND) {
1156                 /* Use the default pattern descriptors */
1157                 if (!this->bbt_td) {
1158                         this->bbt_td = &bbt_main_descr;
1159                         this->bbt_md = &bbt_mirror_descr;
1160                 }
1161                 this->options |= NAND_USE_FLASH_BBT;
1162                 return nand_scan_bbt(mtd, &agand_flashbased);
1163         }
1164
1165         /* Is a flash based bad block table requested ? */
1166         if (this->options & NAND_USE_FLASH_BBT) {
1167                 /* Use the default pattern descriptors */
1168                 if (!this->bbt_td) {
1169                         this->bbt_td = &bbt_main_descr;
1170                         this->bbt_md = &bbt_mirror_descr;
1171                 }
1172                 if (!this->badblock_pattern) {
1173                         this->badblock_pattern = (mtd->writesize > 512) ? &largepage_flashbased : &smallpage_flashbased;
1174                 }
1175         } else {
1176                 this->bbt_td = NULL;
1177                 this->bbt_md = NULL;
1178                 if (!this->badblock_pattern) {
1179                         this->badblock_pattern = (mtd->writesize > 512) ?
1180                             &largepage_memorybased : &smallpage_memorybased;
1181                 }
1182         }
1183         return nand_scan_bbt(mtd, this->badblock_pattern);
1184 }
1185
1186 /**
1187  * nand_isbad_bbt - [NAND Interface] Check if a block is bad
1188  * @mtd:        MTD device structure
1189  * @offs:       offset in the device
1190  * @allowbbt:   allow access to bad block table region
1191  *
1192 */
1193 int nand_isbad_bbt(struct mtd_info *mtd, loff_t offs, int allowbbt)
1194 {
1195         struct nand_chip *this = mtd->priv;
1196         int block;
1197         uint8_t res;
1198
1199         /* Get block number * 2 */
1200         block = (int)(offs >> (this->bbt_erase_shift - 1));
1201         res = (this->bbt[block >> 3] >> (block & 0x06)) & 0x03;
1202
1203         DEBUG(MTD_DEBUG_LEVEL2, "nand_isbad_bbt(): bbt info for offs 0x%08x: (block %d) 0x%02x\n",
1204               (unsigned int)offs, block >> 1, res);
1205
1206         switch ((int)res) {
1207         case 0x00:
1208                 return 0;
1209         case 0x01:
1210                 return 1;
1211         case 0x02:
1212                 return allowbbt ? 0 : 1;
1213         }
1214         return 1;
1215 }
1216
1217 EXPORT_SYMBOL(nand_scan_bbt);
1218 EXPORT_SYMBOL(nand_default_bbt);