drivers/mtd/spi-nor/spi-nor.c

   1 /*
   2  * Based on m25p80.c, by Mike Lavender (mike@steroidmicros.com), with
   3  * influence from lart.c (Abraham Van Der Merwe) and mtd_dataflash.c
   4  *
   5  * Copyright (C) 2005, Intec Automation Inc.
   6  * Copyright (C) 2014, Freescale Semiconductor, Inc.
   7  *
   8  * This code is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 as
  10  * published by the Free Software Foundation.
  11  */
  12
  13 #include <linux/err.h>
  14 #include <linux/errno.h>
  15 #include <linux/module.h>
  16 #include <linux/device.h>
  17 #include <linux/mutex.h>
  18 #include <linux/math64.h>
  19 #include <linux/sizes.h>
  20
  21 #include <linux/mtd/mtd.h>
  22 #include <linux/of_platform.h>
  23 #include <linux/spi/flash.h>
  24 #include <linux/mtd/spi-nor.h>
  25 #include <linux/spi/spi.h>
  26
  27 /* Define max times to check status register before we give up. */
  28
  29 /*
  30  * For everything but full-chip erase; probably could be much smaller, but kept
  31  * around for safety for now
  32  */
  33 #define DEFAULT_READY_WAIT_JIFFIES              (40UL * HZ)
  34
  35 /*
  36  * For full-chip erase, calibrated to a 2MB flash (M25P16); should be scaled up
  37  * for larger flash
  38  */
  39 #define CHIP_ERASE_2MB_READY_WAIT_JIFFIES       (40UL * HZ)
  40
  41 #define SPI_NOR_MAX_ID_LEN      6
  42 #define SPI_NOR_MAX_ADDR_WIDTH  4
  43
  44 struct flash_info {
  45         char            *name;
  46
  47         /*
  48          * This array stores the ID bytes.
  49          * The first three bytes are the JEDIC ID.
  50          * JEDEC ID zero means "no ID" (mostly older chips).
  51          */
  52         u8              id[SPI_NOR_MAX_ID_LEN];
  53         u8              id_len;
  54
  55         /* The size listed here is what works with SPINOR_OP_SE, which isn't
  56          * necessarily called a "sector" by the vendor.
  57          */
  58         unsigned        sector_size;
  59         u16             n_sectors;
  60
  61         u16             page_size;
  62         u16             addr_width;
  63
  64         u16             flags;
  65 #define SECT_4K                 BIT(0)  /* SPINOR_OP_BE_4K works uniformly */
  66 #define SPI_NOR_NO_ERASE        BIT(1)  /* No erase command needed */
  67 #define SST_WRITE               BIT(2)  /* use SST byte programming */
  68 #define SPI_NOR_NO_FR           BIT(3)  /* Can't do fastread */
  69 #define SECT_4K_PMC             BIT(4)  /* SPINOR_OP_BE_4K_PMC works uniformly */
  70 #define SPI_NOR_DUAL_READ       BIT(5)  /* Flash supports Dual Read */
  71 #define SPI_NOR_QUAD_READ       BIT(6)  /* Flash supports Quad Read */
  72 #define USE_FSR                 BIT(7)  /* use flag status register */
  73 #define SPI_NOR_HAS_LOCK        BIT(8)  /* Flash supports lock/unlock via SR */
  74 #define SPI_NOR_HAS_TB          BIT(9)  /*
  75                                          * Flash SR has Top/Bottom (TB) protect
  76                                          * bit. Must be used with
  77                                          * SPI_NOR_HAS_LOCK.
  78                                                 */
  79 #define SST_GLOBAL_PROT_UNLK    BIT(10) /* Unlock the Global protection for
  80                                          * sst flashes
  81                                          */
  82
  83 };
  84
  85 #define JEDEC_MFR(info) ((info)->id[0])
  86
  87 static const struct flash_info *spi_nor_match_id(const char *name);
  88
  89 /*
  90  * Read the status register, returning its value in the location
  91  * Return the status register value.
  92  * Returns negative if error occurred.
  93  */
  94 static int read_sr(struct spi_nor *nor)
  95 {
  96         int ret;
  97         u8 val[2];
  98
  99         if (nor->isparallel) {
 100                 ret = nor->read_reg(nor, SPINOR_OP_RDSR, &val[0], 2);
 101                 if (ret < 0) {
 102                         pr_err("error %d reading SR\n", (int) ret);
 103                         return ret;
 104                 }
 105                 val[0] |= val[1];
 106         } else {
 107                 ret = nor->read_reg(nor, SPINOR_OP_RDSR, &val[0], 1);
 108                 if (ret < 0) {
 109                         pr_err("error %d reading SR\n", (int) ret);
 110                         return ret;
 111                 }
 112         }
 113
 114         return val[0];
 115 }
 116
 117 /*
 118  * Read the flag status register, returning its value in the location
 119  * Return the status register value.
 120  * Returns negative if error occurred.
 121  */
 122 static int read_fsr(struct spi_nor *nor)
 123 {
 124         int ret;
 125         u8 val[2];
 126
 127         if (nor->isparallel) {
 128                 ret = nor->read_reg(nor, SPINOR_OP_RDFSR, &val[0], 2);
 129                 if (ret < 0) {
 130                         pr_err("error %d reading FSR\n", ret);
 131                         return ret;
 132                 }
 133                 val[0] &= val[1];
 134         } else {
 135                 ret = nor->read_reg(nor, SPINOR_OP_RDFSR, &val[0], 1);
 136                 if (ret < 0) {
 137                         pr_err("error %d reading FSR\n", ret);
 138                         return ret;
 139                 }
 140         }
 141
 142         return val[0];
 143 }
 144
 145 /*
 146  * Read configuration register, returning its value in the
 147  * location. Return the configuration register value.
 148  * Returns negative if error occured.
 149  */
 150 static int read_cr(struct spi_nor *nor)
 151 {
 152         int ret;
 153         u8 val;
 154
 155         ret = nor->read_reg(nor, SPINOR_OP_RDCR, &val, 1);
 156         if (ret < 0) {
 157                 dev_err(nor->dev, "error %d reading CR\n", ret);
 158                 return ret;
 159         }
 160
 161         return val;
 162 }
 163
 164 /*
 165  * Dummy Cycle calculation for different type of read.
 166  * It can be used to support more commands with
 167  * different dummy cycle requirements.
 168  */
 169 static inline int spi_nor_read_dummy_cycles(struct spi_nor *nor)
 170 {
 171         switch (nor->flash_read) {
 172         case SPI_NOR_FAST:
 173         case SPI_NOR_DUAL:
 174         case SPI_NOR_QUAD:
 175                 return 8;
 176         case SPI_NOR_NORMAL:
 177                 return 0;
 178         }
 179         return 0;
 180 }
 181
 182 /*
 183  * Write status register 1 byte
 184  * Returns negative if error occurred.
 185  */
 186 static inline int write_sr(struct spi_nor *nor, u8 val)
 187 {
 188         nor->cmd_buf[0] = val;
 189         return nor->write_reg(nor, SPINOR_OP_WRSR, nor->cmd_buf, 1);
 190 }
 191
 192 /*
 193  * Write status Register and configuration register with 2 bytes
 194  * The first byte will be written to the status register, while the
 195  * second byte will be written to the configuration register.
 196  * Return negative if error occured.
 197  */
 198 static int write_sr_cr(struct spi_nor *nor, u16 val)
 199 {
 200         nor->cmd_buf[0] = val & 0xff;
 201         nor->cmd_buf[1] = (val >> 8);
 202
 203         return nor->write_reg(nor, SPINOR_OP_WRSR, nor->cmd_buf, 2);
 204 }
 205
 206 /*
 207  * Set write enable latch with Write Enable command.
 208  * Returns negative if error occurred.
 209  */
 210 static inline int write_enable(struct spi_nor *nor)
 211 {
 212         return nor->write_reg(nor, SPINOR_OP_WREN, NULL, 0);
 213 }
 214
 215 /*
 216  * Send write disble instruction to the chip.
 217  */
 218 static inline int write_disable(struct spi_nor *nor)
 219 {
 220         return nor->write_reg(nor, SPINOR_OP_WRDI, NULL, 0);
 221 }
 222
 223 static inline struct spi_nor *mtd_to_spi_nor(struct mtd_info *mtd)
 224 {
 225         return mtd->priv;
 226 }
 227
 228 /* Enable/disable 4-byte addressing mode. */
 229 static inline int set_4byte(struct spi_nor *nor, const struct flash_info *info,
 230                             int enable)
 231 {
 232         int status;
 233         bool need_wren = false;
 234         u8 cmd;
 235
 236         switch (JEDEC_MFR(info)) {
 237         case SNOR_MFR_MICRON:
 238                 /* Some Micron need WREN command; all will accept it */
 239                 need_wren = true;
 240         case SNOR_MFR_MACRONIX:
 241         case SNOR_MFR_WINBOND:
 242                 if (need_wren)
 243                         write_enable(nor);
 244
 245                 cmd = enable ? SPINOR_OP_EN4B : SPINOR_OP_EX4B;
 246                 status = nor->write_reg(nor, cmd, NULL, 0);
 247                 if (need_wren)
 248                         write_disable(nor);
 249
 250                 return status;
 251         default:
 252                 /* Spansion style */
 253                 nor->cmd_buf[0] = enable << 7;
 254                 return nor->write_reg(nor, SPINOR_OP_BRWR, nor->cmd_buf, 1);
 255         }
 256 }
 257
 258 /**
 259  * read_ear - Get the extended/bank address register value
 260  * @nor:        Pointer to the flash control structure
 261  *
 262  * This routine reads the Extended/bank address register value
 263  *
 264  * Return:      Negative if error occured.
 265  */
 266 static int read_ear(struct spi_nor *nor, struct flash_info *info)
 267 {
 268         int ret;
 269         u8 val;
 270         u8 code;
 271
 272         /* This is actually Spansion */
 273         if (JEDEC_MFR(info) == CFI_MFR_AMD)
 274                 code = SPINOR_OP_BRRD;
 275         /* This is actually Micron */
 276         else if (JEDEC_MFR(info) == CFI_MFR_ST)
 277                 code = SPINOR_OP_RDEAR;
 278         else
 279                 return -EINVAL;
 280
 281         ret = nor->read_reg(nor, code, &val, 1);
 282         if (ret < 0)
 283                 return ret;
 284
 285         return val;
 286 }
 287
 288
 289 static inline int spi_nor_sr_ready(struct spi_nor *nor)
 290 {
 291         int sr = read_sr(nor);
 292         if (sr < 0)
 293                 return sr;
 294         else
 295                 return !(sr & SR_WIP);
 296 }
 297
 298 static inline int spi_nor_fsr_ready(struct spi_nor *nor)
 299 {
 300         int fsr = read_fsr(nor);
 301         if (fsr < 0)
 302                 return fsr;
 303         else
 304                 return fsr & FSR_READY;
 305 }
 306
 307 static int spi_nor_ready(struct spi_nor *nor)
 308 {
 309         int sr, fsr;
 310         sr = spi_nor_sr_ready(nor);
 311         if (sr < 0)
 312                 return sr;
 313         fsr = nor->flags & SNOR_F_USE_FSR ? spi_nor_fsr_ready(nor) : 1;
 314         if (fsr < 0)
 315                 return fsr;
 316         return sr && fsr;
 317 }
 318
 319 /*
 320  * Service routine to read status register until ready, or timeout occurs.
 321  * Returns non-zero if error.
 322  */
 323 static int spi_nor_wait_till_ready_with_timeout(struct spi_nor *nor,
 324                                                 unsigned long timeout_jiffies)
 325 {
 326         unsigned long deadline;
 327         int timeout = 0, ret;
 328
 329         deadline = jiffies + timeout_jiffies;
 330
 331         while (!timeout) {
 332                 if (time_after_eq(jiffies, deadline))
 333                         timeout = 1;
 334
 335                 ret = spi_nor_ready(nor);
 336                 if (ret < 0)
 337                         return ret;
 338                 if (ret)
 339                         return 0;
 340
 341                 cond_resched();
 342         }
 343
 344         dev_err(nor->dev, "flash operation timed out\n");
 345
 346         return -ETIMEDOUT;
 347 }
 348
 349 static int spi_nor_wait_till_ready(struct spi_nor *nor)
 350 {
 351         return spi_nor_wait_till_ready_with_timeout(nor,
 352                                                     DEFAULT_READY_WAIT_JIFFIES);
 353 }
 354
 355 /*
 356  * Update Extended Address/bank selection Register.
 357  * Call with flash->lock locked.
 358  */
 359 static int write_ear(struct spi_nor *nor, u32 addr)
 360 {
 361         u8 code;
 362         u8 ear;
 363         int ret;
 364         struct mtd_info *mtd = &nor->mtd;
 365
 366         /* Wait until finished previous write command. */
 367         if (spi_nor_wait_till_ready(nor))
 368                 return 1;
 369
 370         if (mtd->size <= (0x1000000) << nor->shift)
 371                 return 0;
 372
 373         addr = addr % (u32) mtd->size;
 374         ear = addr >> 24;
 375
 376         if ((!nor->isstacked) && (ear == nor->curbank))
 377                 return 0;
 378
 379         if (nor->isstacked && (mtd->size <= 0x2000000))
 380                 return 0;
 381
 382         if (nor->jedec_id == CFI_MFR_AMD)
 383                 code = SPINOR_OP_BRWR;
 384         if (nor->jedec_id == CFI_MFR_ST) {
 385                 write_enable(nor);
 386                 code = SPINOR_OP_WREAR;
 387         }
 388         nor->cmd_buf[0] = ear;
 389
 390         ret = nor->write_reg(nor, code, nor->cmd_buf, 1);
 391         if (ret < 0)
 392                 return ret;
 393
 394         nor->curbank = ear;
 395
 396         return 0;
 397 }
 398
 399 /*
 400  * Erase the whole flash memory
 401  *
 402  * Returns 0 if successful, non-zero otherwise.
 403  */
 404 static int erase_chip(struct spi_nor *nor)
 405 {
 406         u32 ret;
 407
 408         dev_dbg(nor->dev, " %lldKiB\n", (long long)(nor->mtd.size >> 10));
 409
 410         if (nor->isstacked)
 411                 nor->spi->master->flags &= ~SPI_MASTER_U_PAGE;
 412
 413         ret = nor->write_reg(nor, SPINOR_OP_CHIP_ERASE, NULL, 0);
 414         if (ret)
 415                 return ret;
 416
 417         if (nor->isstacked) {
 418                 /* Wait until previous write command finished */
 419                 ret = spi_nor_wait_till_ready(nor);
 420                 if (ret)
 421                         return ret;
 422
 423                 nor->spi->master->flags |= SPI_MASTER_U_PAGE;
 424
 425                 ret = nor->write_reg(nor, SPINOR_OP_CHIP_ERASE, NULL, 0);
 426         }
 427         return ret;
 428
 429 }
 430
 431 static int spi_nor_lock_and_prep(struct spi_nor *nor, enum spi_nor_ops ops)
 432 {
 433         int ret = 0;
 434
 435         mutex_lock(&nor->lock);
 436
 437         if (nor->prepare) {
 438                 ret = nor->prepare(nor, ops);
 439                 if (ret) {
 440                         dev_err(nor->dev, "failed in the preparation.\n");
 441                         mutex_unlock(&nor->lock);
 442                         return ret;
 443                 }
 444         }
 445         return ret;
 446 }
 447
 448 static void spi_nor_unlock_and_unprep(struct spi_nor *nor, enum spi_nor_ops ops)
 449 {
 450         if (nor->unprepare)
 451                 nor->unprepare(nor, ops);
 452         mutex_unlock(&nor->lock);
 453 }
 454
 455 /*
 456  * Initiate the erasure of a single sector
 457  */
 458 static int spi_nor_erase_sector(struct spi_nor *nor, u32 addr)
 459 {
 460         u8 buf[SPI_NOR_MAX_ADDR_WIDTH];
 461         int i;
 462
 463         if (nor->erase)
 464                 return nor->erase(nor, addr);
 465
 466         /*
 467          * Default implementation, if driver doesn't have a specialized HW
 468          * control
 469          */
 470         for (i = nor->addr_width - 1; i >= 0; i--) {
 471                 buf[i] = addr & 0xff;
 472                 addr >>= 8;
 473         }
 474
 475         return nor->write_reg(nor, nor->erase_opcode, buf, nor->addr_width);
 476 }
 477
 478 /*
 479  * Erase an address range on the nor chip.  The address range may extend
 480  * one or more erase sectors.  Return an error is there is a problem erasing.
 481  */
 482 static int spi_nor_erase(struct mtd_info *mtd, struct erase_info *instr)
 483 {
 484         struct spi_nor *nor = mtd_to_spi_nor(mtd);
 485         u32 addr, len, offset;
 486         uint32_t rem;
 487         int ret;
 488
 489         dev_dbg(nor->dev, "at 0x%llx, len %lld\n", (long long)instr->addr,
 490                         (long long)instr->len);
 491
 492         div_u64_rem(instr->len, mtd->erasesize, &rem);
 493         if (rem)
 494                 return -EINVAL;
 495
 496         addr = instr->addr;
 497         len = instr->len;
 498
 499         ret = spi_nor_lock_and_prep(nor, SPI_NOR_OPS_ERASE);
 500         if (ret)
 501                 return ret;
 502
 503         /* whole-chip erase? */
 504         if (len == mtd->size) {
 505                 unsigned long timeout;
 506
 507                 write_enable(nor);
 508
 509                 if (erase_chip(nor)) {
 510                         ret = -EIO;
 511                         goto erase_err;
 512                 }
 513
 514                 /*
 515                  * Scale the timeout linearly with the size of the flash, with
 516                  * a minimum calibrated to an old 2MB flash. We could try to
 517                  * pull these from CFI/SFDP, but these values should be good
 518                  * enough for now.
 519                  */
 520                 timeout = max(CHIP_ERASE_2MB_READY_WAIT_JIFFIES,
 521                               CHIP_ERASE_2MB_READY_WAIT_JIFFIES *
 522                               (unsigned long)(mtd->size / SZ_2M));
 523                 ret = spi_nor_wait_till_ready_with_timeout(nor, timeout);
 524                 if (ret)
 525                         goto erase_err;
 526
 527         /* REVISIT in some cases we could speed up erasing large regions
 528          * by using SPINOR_OP_SE instead of SPINOR_OP_BE_4K.  We may have set up
 529          * to use "small sector erase", but that's not always optimal.
 530          */
 531
 532         /* "sector"-at-a-time erase */
 533         } else {
 534                 while (len) {
 535
 536                         write_enable(nor);
 537                         offset = addr;
 538                         if (nor->isparallel == 1)
 539                                 offset /= 2;
 540
 541                         if (nor->isstacked == 1) {
 542                                 if (offset >= (mtd->size / 2)) {
 543                                         offset = offset - (mtd->size / 2);
 544                                         nor->spi->master->flags |=
 545                                                 SPI_MASTER_U_PAGE;
 546                                 } else {
 547                                         nor->spi->master->flags &=
 548                                                 ~SPI_MASTER_U_PAGE;
 549                                 }
 550                         }
 551                         if (nor->addr_width == 3) {
 552                                 /* Update Extended Address Register */
 553                                 ret = write_ear(nor, offset);
 554                                 if (ret)
 555                                         goto erase_err;
 556                         }
 557                         ret = spi_nor_wait_till_ready(nor);
 558                         if (ret)
 559                                 goto erase_err;
 560
 561                         write_enable(nor);
 562
 563                         ret = spi_nor_erase_sector(nor, offset);
 564                         if (ret)
 565                                 goto erase_err;
 566
 567                         addr += mtd->erasesize;
 568                         len -= mtd->erasesize;
 569
 570                         ret = spi_nor_wait_till_ready(nor);
 571                         if (ret)
 572                                 goto erase_err;
 573                 }
 574         }
 575
 576         write_disable(nor);
 577
 578 erase_err:
 579         spi_nor_unlock_and_unprep(nor, SPI_NOR_OPS_ERASE);
 580
 581         instr->state = ret ? MTD_ERASE_FAILED : MTD_ERASE_DONE;
 582         mtd_erase_callback(instr);
 583
 584         return ret;
 585 }
 586
 587 static inline uint16_t min_lockable_sectors(struct spi_nor *nor,
 588                                             uint16_t n_sectors)
 589 {
 590         uint16_t lock_granularity;
 591
 592         /*
 593          * Revisit - SST (not used by us) has the same JEDEC ID as micron but
 594          * protected area table is similar to that of spansion.
 595          */
 596         lock_granularity = max(1, n_sectors/M25P_MAX_LOCKABLE_SECTORS);
 597         if (nor->jedec_id == CFI_MFR_ST)        /* Micron */
 598                 lock_granularity = 1;
 599
 600         return lock_granularity;
 601 }
 602
 603 static inline uint32_t get_protected_area_start(struct spi_nor *nor,
 604                                                 uint8_t lock_bits)
 605 {
 606         u16 n_sectors;
 607         u32 sector_size;
 608         uint64_t mtd_size;
 609         struct mtd_info *mtd = &nor->mtd;
 610
 611         n_sectors = nor->n_sectors;
 612         sector_size = nor->sector_size;
 613         mtd_size = mtd->size;
 614
 615         if (nor->isparallel) {
 616                 sector_size = (nor->sector_size >> 1);
 617                 mtd_size = (mtd->size >> 1);
 618         }
 619         if (nor->isstacked) {
 620                 n_sectors = (nor->n_sectors >> 1);
 621                 mtd_size = (mtd->size >> 1);
 622         }
 623
 624         return mtd_size - (1<<(lock_bits-1)) *
 625                 min_lockable_sectors(nor, n_sectors) * sector_size;
 626 }
 627
 628 static uint8_t min_protected_area_including_offset(struct spi_nor *nor,
 629                                                    uint32_t offset)
 630 {
 631         uint8_t lock_bits, lockbits_limit;
 632
 633         /*
 634          * Revisit - SST (not used by us) has the same JEDEC ID as micron but
 635          * protected area table is similar to that of spansion.
 636          * Mircon has 4 block protect bits.
 637          */
 638         lockbits_limit = 7;
 639         if (nor->jedec_id == CFI_MFR_ST)        /* Micron */
 640                 lockbits_limit = 15;
 641
 642         for (lock_bits = 1; lock_bits < lockbits_limit; lock_bits++) {
 643                 if (offset >= get_protected_area_start(nor, lock_bits))
 644                         break;
 645         }
 646         return lock_bits;
 647 }
 648
 649 static int write_sr_modify_protection(struct spi_nor *nor, uint8_t status,
 650                                       uint8_t lock_bits)
 651 {
 652         uint8_t status_new, bp_mask;
 653         u16 val;
 654
 655         status_new = status & ~SR_BP_BIT_MASK;
 656         bp_mask = (lock_bits << SR_BP_BIT_OFFSET) & SR_BP_BIT_MASK;
 657
 658         /* Micron */
 659         if (nor->jedec_id == CFI_MFR_ST) {
 660                 /* To support chips with more than 896 sectors (56MB) */
 661                 status_new &= ~SR_BP3;
 662
 663                 /* Protected area starts from top */
 664                 status_new &= ~SR_BP_TB;
 665
 666                 if (lock_bits > 7)
 667                         bp_mask |= SR_BP3;
 668         }
 669
 670         status_new |= bp_mask;
 671
 672         write_enable(nor);
 673
 674         /* For spansion flashes */
 675         if (nor->jedec_id == CFI_MFR_AMD) {
 676                 val = read_cr(nor) << 8;
 677                 val |= status_new;
 678                 if (write_sr_cr(nor, val) < 0)
 679                         return 1;
 680         } else {
 681                 if (write_sr(nor, status_new) < 0)
 682                         return 1;
 683         }
 684         return 0;
 685 }
 686
 687 static uint8_t bp_bits_from_sr(struct spi_nor *nor, uint8_t status)
 688 {
 689         uint8_t ret;
 690
 691         ret = (((status) & SR_BP_BIT_MASK) >> SR_BP_BIT_OFFSET);
 692         if (nor->jedec_id == 0x20)
 693                 ret |= ((status & SR_BP3) >> (SR_BP_BIT_OFFSET + 1));
 694
 695         return ret;
 696 }
 697
 698
 699 static void stm_get_locked_range(struct spi_nor *nor, u8 sr, loff_t *ofs,
 700                                  uint64_t *len)
 701 {
 702         struct mtd_info *mtd = &nor->mtd;
 703         u8 mask = SR_BP2 | SR_BP1 | SR_BP0;
 704         int shift = ffs(mask) - 1;
 705         int pow;
 706
 707         if (!(sr & mask)) {
 708                 /* No protection */
 709                 *ofs = 0;
 710                 *len = 0;
 711         } else {
 712                 pow = ((sr & mask) ^ mask) >> shift;
 713                 *len = mtd->size >> pow;
 714                 if (nor->flags & SNOR_F_HAS_SR_TB && sr & SR_TB)
 715                         *ofs = 0;
 716                 else
 717                         *ofs = mtd->size - *len;
 718         }
 719 }
 720
 721 /*
 722  * Return 1 if the entire region is locked (if @locked is true) or unlocked (if
 723  * @locked is false); 0 otherwise
 724  */
 725 static int stm_check_lock_status_sr(struct spi_nor *nor, loff_t ofs, uint64_t len,
 726                                     u8 sr, bool locked)
 727 {
 728         loff_t lock_offs;
 729         uint64_t lock_len;
 730
 731         if (!len)
 732                 return 1;
 733
 734         stm_get_locked_range(nor, sr, &lock_offs, &lock_len);
 735
 736         if (locked)
 737                 /* Requested range is a sub-range of locked range */
 738                 return (ofs + len <= lock_offs + lock_len) && (ofs >= lock_offs);
 739         else
 740                 /* Requested range does not overlap with locked range */
 741                 return (ofs >= lock_offs + lock_len) || (ofs + len <= lock_offs);
 742 }
 743
 744 static int stm_is_locked_sr(struct spi_nor *nor, loff_t ofs, uint64_t len,
 745                             u8 sr)
 746 {
 747         return stm_check_lock_status_sr(nor, ofs, len, sr, true);
 748 }
 749
 750 static int stm_is_unlocked_sr(struct spi_nor *nor, loff_t ofs, uint64_t len,
 751                               u8 sr)
 752 {
 753         return stm_check_lock_status_sr(nor, ofs, len, sr, false);
 754 }
 755
 756 /*
 757  * Lock a region of the flash. Compatible with ST Micro and similar flash.
 758  * Supports the block protection bits BP{0,1,2} in the status register
 759  * (SR). Does not support these features found in newer SR bitfields:
 760  *   - SEC: sector/block protect - only handle SEC=0 (block protect)
 761  *   - CMP: complement protect - only support CMP=0 (range is not complemented)
 762  *
 763  * Support for the following is provided conditionally for some flash:
 764  *   - TB: top/bottom protect
 765  *
 766  * Sample table portion for 8MB flash (Winbond w25q64fw):
 767  *
 768  *   SEC  |  TB   |  BP2  |  BP1  |  BP0  |  Prot Length  | Protected Portion
 769  *  --------------------------------------------------------------------------
 770  *    X   |   X   |   0   |   0   |   0   |  NONE         | NONE
 771  *    0   |   0   |   0   |   0   |   1   |  128 KB       | Upper 1/64
 772  *    0   |   0   |   0   |   1   |   0   |  256 KB       | Upper 1/32
 773  *    0   |   0   |   0   |   1   |   1   |  512 KB       | Upper 1/16
 774  *    0   |   0   |   1   |   0   |   0   |  1 MB         | Upper 1/8
 775  *    0   |   0   |   1   |   0   |   1   |  2 MB         | Upper 1/4
 776  *    0   |   0   |   1   |   1   |   0   |  4 MB         | Upper 1/2
 777  *    X   |   X   |   1   |   1   |   1   |  8 MB         | ALL
 778  *  ------|-------|-------|-------|-------|---------------|-------------------
 779  *    0   |   1   |   0   |   0   |   1   |  128 KB       | Lower 1/64
 780  *    0   |   1   |   0   |   1   |   0   |  256 KB       | Lower 1/32
 781  *    0   |   1   |   0   |   1   |   1   |  512 KB       | Lower 1/16
 782  *    0   |   1   |   1   |   0   |   0   |  1 MB         | Lower 1/8
 783  *    0   |   1   |   1   |   0   |   1   |  2 MB         | Lower 1/4
 784  *    0   |   1   |   1   |   1   |   0   |  4 MB         | Lower 1/2
 785  *
 786  * Returns negative on errors, 0 on success.
 787  */
 788 static int stm_lock(struct spi_nor *nor, loff_t ofs, uint64_t len)
 789 {
 790         struct mtd_info *mtd = &nor->mtd;
 791         int status_old, status_new;
 792         u8 mask = SR_BP2 | SR_BP1 | SR_BP0;
 793         u8 shift = ffs(mask) - 1, pow, val;
 794         loff_t lock_len;
 795         bool can_be_top = true, can_be_bottom = nor->flags & SNOR_F_HAS_SR_TB;
 796         bool use_top;
 797         int ret;
 798
 799         ofs = ofs >> nor->shift;
 800
 801         status_old = read_sr(nor);
 802         if (status_old < 0)
 803                 return status_old;
 804
 805         /* If nothing in our range is unlocked, we don't need to do anything */
 806         if (stm_is_locked_sr(nor, ofs, len, status_old))
 807                 return 0;
 808
 809         /* If anything below us is unlocked, we can't use 'bottom' protection */
 810         if (!stm_is_locked_sr(nor, 0, ofs, status_old))
 811                 can_be_bottom = false;
 812
 813         /* If anything above us is unlocked, we can't use 'top' protection */
 814         if (!stm_is_locked_sr(nor, ofs + len, mtd->size - (ofs + len),
 815                                 status_old))
 816                 can_be_top = false;
 817
 818         if (!can_be_bottom && !can_be_top)
 819                 return -EINVAL;
 820
 821         /* Prefer top, if both are valid */
 822         use_top = can_be_top;
 823
 824         /* lock_len: length of region that should end up locked */
 825         if (use_top)
 826                 lock_len = mtd->size - ofs;
 827         else
 828                 lock_len = ofs + len;
 829
 830         /*
 831          * Need smallest pow such that:
 832          *
 833          *   1 / (2^pow) <= (len / size)
 834          *
 835          * so (assuming power-of-2 size) we do:
 836          *
 837          *   pow = ceil(log2(size / len)) = log2(size) - floor(log2(len))
 838          */
 839         pow = ilog2(mtd->size) - ilog2(lock_len);
 840         val = mask - (pow << shift);
 841         if (val & ~mask)
 842                 return -EINVAL;
 843         /* Don't "lock" with no region! */
 844         if (!(val & mask))
 845                 return -EINVAL;
 846
 847         status_new = (status_old & ~mask & ~SR_TB) | val;
 848
 849         /* Disallow further writes if WP pin is asserted */
 850         status_new |= SR_SRWD;
 851
 852         if (!use_top)
 853                 status_new |= SR_TB;
 854
 855         /* Don't bother if they're the same */
 856         if (status_new == status_old)
 857                 return 0;
 858
 859         /* Only modify protection if it will not unlock other areas */
 860         if ((status_new & mask) < (status_old & mask))
 861                 return -EINVAL;
 862
 863         write_enable(nor);
 864         ret = write_sr(nor, status_new);
 865         if (ret)
 866                 return ret;
 867         return spi_nor_wait_till_ready(nor);
 868 }
 869
 870 /*
 871  * Unlock a region of the flash. See stm_lock() for more info
 872  *
 873  * Returns negative on errors, 0 on success.
 874  */
 875 static int stm_unlock(struct spi_nor *nor, loff_t ofs, uint64_t len)
 876 {
 877         struct mtd_info *mtd = &nor->mtd;
 878         int status_old, status_new;
 879         u8 mask = SR_BP2 | SR_BP1 | SR_BP0;
 880         u8 shift = ffs(mask) - 1, pow, val;
 881         loff_t lock_len;
 882         bool can_be_top = true, can_be_bottom = nor->flags & SNOR_F_HAS_SR_TB;
 883         bool use_top;
 884         int ret;
 885
 886         ofs = ofs >> nor->shift;
 887
 888         status_old = read_sr(nor);
 889         if (status_old < 0)
 890                 return status_old;
 891
 892         /* If nothing in our range is locked, we don't need to do anything */
 893         if (stm_is_unlocked_sr(nor, ofs, len, status_old))
 894                 return 0;
 895
 896         /* If anything below us is locked, we can't use 'top' protection */
 897         if (!stm_is_unlocked_sr(nor, 0, ofs, status_old))
 898                 can_be_top = false;
 899
 900         /* If anything above us is locked, we can't use 'bottom' protection */
 901         if (!stm_is_unlocked_sr(nor, ofs + len, mtd->size - (ofs + len),
 902                                 status_old))
 903                 can_be_bottom = false;
 904
 905         if (!can_be_bottom && !can_be_top)
 906                 return -EINVAL;
 907
 908         /* Prefer top, if both are valid */
 909         use_top = can_be_top;
 910
 911         /* lock_len: length of region that should remain locked */
 912         if (use_top)
 913                 lock_len = mtd->size - (ofs + len);
 914         else
 915                 lock_len = ofs;
 916
 917         /*
 918          * Need largest pow such that:
 919          *
 920          *   1 / (2^pow) >= (len / size)
 921          *
 922          * so (assuming power-of-2 size) we do:
 923          *
 924          *   pow = floor(log2(size / len)) = log2(size) - ceil(log2(len))
 925          */
 926         pow = ilog2(mtd->size) - order_base_2(lock_len);
 927         if (lock_len == 0) {
 928                 val = 0; /* fully unlocked */
 929         } else {
 930                 val = mask - (pow << shift);
 931                 /* Some power-of-two sizes are not supported */
 932                 if (val & ~mask)
 933                         return -EINVAL;
 934         }
 935
 936         status_new = (status_old & ~mask & ~SR_TB) | val;
 937
 938         /* Don't protect status register if we're fully unlocked */
 939         if (lock_len == 0)
 940                 status_new &= ~SR_SRWD;
 941
 942         if (!use_top)
 943                 status_new |= SR_TB;
 944
 945         /* Don't bother if they're the same */
 946         if (status_new == status_old)
 947                 return 0;
 948
 949         /* Only modify protection if it will not lock other areas */
 950         if ((status_new & mask) > (status_old & mask))
 951                 return -EINVAL;
 952
 953         write_enable(nor);
 954         ret = write_sr(nor, status_new);
 955         if (ret)
 956                 return ret;
 957         return spi_nor_wait_till_ready(nor);
 958 }
 959
 960 /*
 961  * Check if a region of the flash is (completely) locked. See stm_lock() for
 962  * more info.
 963  *
 964  * Returns 1 if entire region is locked, 0 if any portion is unlocked, and
 965  * negative on errors.
 966  */
 967 static int stm_is_locked(struct spi_nor *nor, loff_t ofs, uint64_t len)
 968 {
 969         int status;
 970
 971         status = read_sr(nor);
 972         if (status < 0)
 973                 return status;
 974
 975         return stm_is_locked_sr(nor, ofs, len, status);
 976 }
 977
 978 static int spi_nor_lock(struct mtd_info *mtd, loff_t ofs, uint64_t len)
 979 {
 980         struct spi_nor *nor = mtd_to_spi_nor(mtd);
 981         int ret;
 982         uint8_t status;
 983         uint8_t lock_bits;
 984
 985         ret = spi_nor_lock_and_prep(nor, SPI_NOR_OPS_LOCK);
 986         if (ret)
 987                 return ret;
 988
 989         if (nor->isparallel == 1)
 990                 ofs = ofs >> nor->shift;
 991
 992         if (nor->isstacked == 1) {
 993                 if (ofs >= (mtd->size / 2)) {
 994                         ofs = ofs - (mtd->size / 2);
 995                         nor->spi->master->flags |= SPI_MASTER_U_PAGE;
 996                 } else
 997                         nor->spi->master->flags &= ~SPI_MASTER_U_PAGE;
 998         }
 999
1000         ret = nor->flash_lock(nor, ofs, len);
1001         /* Wait until finished previous command */
1002         ret = spi_nor_wait_till_ready(nor);
1003         if (ret)
1004                 goto err;
1005
1006         status = read_sr(nor);
1007
1008         lock_bits = min_protected_area_including_offset(nor, ofs);
1009
1010         /* Only modify protection if it will not unlock other areas */
1011         if (lock_bits > bp_bits_from_sr(nor, status))
1012                 ret = write_sr_modify_protection(nor, status, lock_bits);
1013         else
1014                 dev_err(nor->dev, "trying to unlock already locked area\n");
1015 err:
1016         spi_nor_unlock_and_unprep(nor, SPI_NOR_OPS_UNLOCK);
1017         return ret;
1018 }
1019
1020 static int spi_nor_unlock(struct mtd_info *mtd, loff_t ofs, uint64_t len)
1021 {
1022         struct spi_nor *nor = mtd_to_spi_nor(mtd);
1023         int ret;
1024         uint8_t status;
1025         uint8_t lock_bits;
1026
1027         ret = spi_nor_lock_and_prep(nor, SPI_NOR_OPS_UNLOCK);
1028         if (ret)
1029                 return ret;
1030
1031         if (nor->isparallel == 1)
1032                 ofs = ofs >> nor->shift;
1033
1034         if (nor->isstacked == 1) {
1035                 if (ofs >= (mtd->size / 2)) {
1036                         ofs = ofs - (mtd->size / 2);
1037                         nor->spi->master->flags |= SPI_MASTER_U_PAGE;
1038                 } else
1039                         nor->spi->master->flags &= ~SPI_MASTER_U_PAGE;
1040         }
1041
1042         ret = nor->flash_unlock(nor, ofs, len);
1043         /* Wait until finished previous command */
1044         ret = spi_nor_wait_till_ready(nor);
1045         if (ret)
1046                 goto err;
1047
1048         status = read_sr(nor);
1049
1050         lock_bits = min_protected_area_including_offset(nor, ofs+len) - 1;
1051
1052         /* Only modify protection if it will not lock other areas */
1053         if (lock_bits < bp_bits_from_sr(nor, status))
1054                 ret = write_sr_modify_protection(nor, status, lock_bits);
1055         else
1056                 dev_err(nor->dev, "trying to lock already unlocked area\n");
1057 err:
1058         spi_nor_unlock_and_unprep(nor, SPI_NOR_OPS_LOCK);
1059         return ret;
1060 }
1061
1062 static int spi_nor_is_locked(struct mtd_info *mtd, loff_t ofs, uint64_t len)
1063 {
1064         struct spi_nor *nor = mtd_to_spi_nor(mtd);
1065         int ret;
1066
1067         ret = spi_nor_lock_and_prep(nor, SPI_NOR_OPS_UNLOCK);
1068         if (ret)
1069                 return ret;
1070
1071         ret = nor->flash_is_locked(nor, ofs, len);
1072
1073         spi_nor_unlock_and_unprep(nor, SPI_NOR_OPS_LOCK);
1074         return ret;
1075 }
1076
1077 /* Used when the "_ext_id" is two bytes at most */
1078 #define INFO(_jedec_id, _ext_id, _sector_size, _n_sectors, _flags)      \
1079                 .id = {                                                 \
1080                         ((_jedec_id) >> 16) & 0xff,                     \
1081                         ((_jedec_id) >> 8) & 0xff,                      \
1082                         (_jedec_id) & 0xff,                             \
1083                         ((_ext_id) >> 8) & 0xff,                        \
1084                         (_ext_id) & 0xff,                               \
1085                         },                                              \
1086                 .id_len = (!(_jedec_id) ? 0 : (3 + ((_ext_id) ? 2 : 0))),       \
1087                 .sector_size = (_sector_size),                          \
1088                 .n_sectors = (_n_sectors),                              \
1089                 .page_size = 256,                                       \
1090                 .flags = (_flags),
1091
1092 #define INFO6(_jedec_id, _ext_id, _sector_size, _n_sectors, _flags)     \
1093                 .id = {                                                 \
1094                         ((_jedec_id) >> 16) & 0xff,                     \
1095                         ((_jedec_id) >> 8) & 0xff,                      \
1096                         (_jedec_id) & 0xff,                             \
1097                         ((_ext_id) >> 16) & 0xff,                       \
1098                         ((_ext_id) >> 8) & 0xff,                        \
1099                         (_ext_id) & 0xff,                               \
1100                         },                                              \
1101                 .id_len = 6,                                            \
1102                 .sector_size = (_sector_size),                          \
1103                 .n_sectors = (_n_sectors),                              \
1104                 .page_size = 256,                                       \
1105                 .flags = (_flags),
1106
1107 #define CAT25_INFO(_sector_size, _n_sectors, _page_size, _addr_width, _flags)   \
1108                 .sector_size = (_sector_size),                          \
1109                 .n_sectors = (_n_sectors),                              \
1110                 .page_size = (_page_size),                              \
1111                 .addr_width = (_addr_width),                            \
1112                 .flags = (_flags),
1113
1114 /* NOTE: double check command sets and memory organization when you add
1115  * more nor chips.  This current list focusses on newer chips, which
1116  * have been converging on command sets which including JEDEC ID.
1117  *
1118  * All newly added entries should describe *hardware* and should use SECT_4K
1119  * (or SECT_4K_PMC) if hardware supports erasing 4 KiB sectors. For usage
1120  * scenarios excluding small sectors there is config option that can be
1121  * disabled: CONFIG_MTD_SPI_NOR_USE_4K_SECTORS.
1122  * For historical (and compatibility) reasons (before we got above config) some
1123  * old entries may be missing 4K flag.
1124  */
1125 static const struct flash_info spi_nor_ids[] = {
1126         /* Atmel -- some are (confusingly) marketed as "DataFlash" */
1127         { "at25fs010",  INFO(0x1f6601, 0, 32 * 1024,   4, SECT_4K) },
1128         { "at25fs040",  INFO(0x1f6604, 0, 64 * 1024,   8, SECT_4K) },
1129
1130         { "at25df041a", INFO(0x1f4401, 0, 64 * 1024,   8, SECT_4K) },
1131         { "at25df321a", INFO(0x1f4701, 0, 64 * 1024,  64, SECT_4K) },
1132         { "at25df641",  INFO(0x1f4800, 0, 64 * 1024, 128, SECT_4K) },
1133
1134         { "at26f004",   INFO(0x1f0400, 0, 64 * 1024,  8, SECT_4K) },
1135         { "at26df081a", INFO(0x1f4501, 0, 64 * 1024, 16, SECT_4K) },
1136         { "at26df161a", INFO(0x1f4601, 0, 64 * 1024, 32, SECT_4K) },
1137         { "at26df321",  INFO(0x1f4700, 0, 64 * 1024, 64, SECT_4K) },
1138
1139         { "at45db081d", INFO(0x1f2500, 0, 64 * 1024, 16, SECT_4K) },
1140
1141         /* EON -- en25xxx */
1142         { "en25f32",    INFO(0x1c3116, 0, 64 * 1024,   64, SECT_4K) },
1143         { "en25p32",    INFO(0x1c2016, 0, 64 * 1024,   64, 0) },
1144         { "en25q32b",   INFO(0x1c3016, 0, 64 * 1024,   64, 0) },
1145         { "en25p64",    INFO(0x1c2017, 0, 64 * 1024,  128, 0) },
1146         { "en25q64",    INFO(0x1c3017, 0, 64 * 1024,  128, SECT_4K) },
1147         { "en25qh128",  INFO(0x1c7018, 0, 64 * 1024,  256, 0) },
1148         { "en25qh256",  INFO(0x1c7019, 0, 64 * 1024,  512, 0) },
1149         { "en25s64",    INFO(0x1c3817, 0, 64 * 1024,  128, SECT_4K) },
1150
1151         /* ESMT */
1152         { "f25l32pa", INFO(0x8c2016, 0, 64 * 1024, 64, SECT_4K) },
1153
1154         /* Everspin */
1155         { "mr25h256", CAT25_INFO( 32 * 1024, 1, 256, 2, SPI_NOR_NO_ERASE | SPI_NOR_NO_FR) },
1156         { "mr25h10",  CAT25_INFO(128 * 1024, 1, 256, 3, SPI_NOR_NO_ERASE | SPI_NOR_NO_FR) },
1157
1158         /* Fujitsu */
1159         { "mb85rs1mt", INFO(0x047f27, 0, 128 * 1024, 1, SPI_NOR_NO_ERASE) },
1160
1161         /* GigaDevice */
1162         {
1163                 "gd25q32", INFO(0xc84016, 0, 64 * 1024,  64,
1164                         SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ |
1165                         SPI_NOR_HAS_LOCK | SPI_NOR_HAS_TB)
1166         },
1167         {
1168                 "gd25q64", INFO(0xc84017, 0, 64 * 1024, 128,
1169                         SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ |
1170                         SPI_NOR_HAS_LOCK | SPI_NOR_HAS_TB)
1171         },
1172         {
1173                 "gd25lq64c", INFO(0xc86017, 0, 64 * 1024, 128,
1174                         SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ |
1175                         SPI_NOR_HAS_LOCK | SPI_NOR_HAS_TB)
1176         },
1177         {
1178                 "gd25q128", INFO(0xc84018, 0, 64 * 1024, 256,
1179                         SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ |
1180                         SPI_NOR_HAS_LOCK | SPI_NOR_HAS_TB)
1181         },
1182
1183         /* Intel/Numonyx -- xxxs33b */
1184         { "160s33b",  INFO(0x898911, 0, 64 * 1024,  32, 0) },
1185         { "320s33b",  INFO(0x898912, 0, 64 * 1024,  64, 0) },
1186         { "640s33b",  INFO(0x898913, 0, 64 * 1024, 128, 0) },
1187
1188         /* ISSI */
1189         { "is25lp256d", INFO(0x9d6019, 0, 64 * 1024, 512, SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ | USE_FSR | SPI_NOR_HAS_LOCK) },
1190         { "is25cd512", INFO(0x7f9d20, 0, 32 * 1024,   2, SECT_4K) },
1191
1192         /* Macronix */
1193         { "mx25l512e",   INFO(0xc22010, 0, 64 * 1024,   1, SECT_4K) },
1194         { "mx25l2005a",  INFO(0xc22012, 0, 64 * 1024,   4, SECT_4K) },
1195         { "mx25l4005a",  INFO(0xc22013, 0, 64 * 1024,   8, SECT_4K) },
1196         { "mx25l8005",   INFO(0xc22014, 0, 64 * 1024,  16, 0) },
1197         { "mx25l1606e",  INFO(0xc22015, 0, 64 * 1024,  32, SECT_4K) },
1198         { "mx25l3205d",  INFO(0xc22016, 0, 64 * 1024,  64, SECT_4K) },
1199         { "mx25l3255e",  INFO(0xc29e16, 0, 64 * 1024,  64, SECT_4K) },
1200         { "mx25l6405d",  INFO(0xc22017, 0, 64 * 1024, 128, SECT_4K) },
1201         { "mx25u6435f",  INFO(0xc22537, 0, 64 * 1024, 128, SECT_4K) },
1202         { "mx25l12805d", INFO(0xc22018, 0, 64 * 1024, 256, 0) },
1203         { "mx25l12855e", INFO(0xc22618, 0, 64 * 1024, 256, 0) },
1204         { "mx25l25635e", INFO(0xc22019, 0, 64 * 1024, 512, 0) },
1205         { "mx25l25655e", INFO(0xc22619, 0, 64 * 1024, 512, 0) },
1206         { "mx66l51235l", INFO(0xc2201a, 0, 64 * 1024, 1024, SPI_NOR_QUAD_READ) },
1207         { "mx66l1g55g",  INFO(0xc2261b, 0, 64 * 1024, 2048, SPI_NOR_QUAD_READ) },
1208
1209         /* Micron */
1210         { "n25q032",     INFO(0x20ba16, 0, 64 * 1024,   64, SPI_NOR_QUAD_READ) },
1211         { "n25q032a",    INFO(0x20bb16, 0, 64 * 1024,   64, SPI_NOR_QUAD_READ) },
1212         { "n25q064",     INFO(0x20ba17, 0, 64 * 1024,  128, SECT_4K | SPI_NOR_QUAD_READ) },
1213         { "n25q064a",    INFO(0x20bb17, 0, 64 * 1024,  128, SECT_4K | SPI_NOR_QUAD_READ) },
1214         { "n25q128a11",  INFO(0x20bb18, 0, 64 * 1024,  256, SECT_4K | SPI_NOR_QUAD_READ | USE_FSR | SPI_NOR_HAS_LOCK) },
1215         { "n25q128a13",  INFO(0x20ba18, 0, 64 * 1024,  256, SECT_4K | SPI_NOR_QUAD_READ | USE_FSR | SPI_NOR_HAS_LOCK) },
1216         { "n25q256a",    INFO(0x20bb19, 0, 64 * 1024,  512, SECT_4K | SPI_NOR_QUAD_READ | USE_FSR| SPI_NOR_HAS_LOCK) },
1217         { "n25q256a13",  INFO(0x20ba19, 0, 64 * 1024,  512, SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ | USE_FSR | SPI_NOR_HAS_LOCK) },
1218         { "n25q512a",    INFO(0x20bb20, 0, 64 * 1024, 1024, SECT_4K | USE_FSR | SPI_NOR_QUAD_READ | SPI_NOR_HAS_LOCK) },
1219         { "n25q512a13",  INFO(0x20ba20, 0, 64 * 1024, 1024, SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ | USE_FSR | SPI_NOR_HAS_LOCK) },
1220         { "n25q512ax3",  INFO(0x20ba20, 0, 64 * 1024, 1024, SECT_4K | USE_FSR | SPI_NOR_QUAD_READ | SPI_NOR_HAS_LOCK) },
1221         { "n25q00",      INFO(0x20ba21, 0, 64 * 1024, 2048, SECT_4K | USE_FSR | SPI_NOR_QUAD_READ | SPI_NOR_HAS_LOCK) },
1222         { "n25q00a",     INFO(0x20bb21, 0, 64 * 1024, 2048, SECT_4K | USE_FSR | SPI_NOR_QUAD_READ | SPI_NOR_HAS_LOCK) },
1223
1224         /* PMC */
1225         { "pm25lv512",   INFO(0,        0, 32 * 1024,    2, SECT_4K_PMC) },
1226         { "pm25lv010",   INFO(0,        0, 32 * 1024,    4, SECT_4K_PMC) },
1227         { "pm25lq032",   INFO(0x7f9d46, 0, 64 * 1024,   64, SECT_4K) },
1228
1229         /* Spansion -- single (large) sector size only, at least
1230          * for the chips listed here (without boot sectors).
1231          */
1232         { "s25sl032p",  INFO(0x010215, 0x4d00,  64 * 1024,  64, SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1233         { "s25sl064p",  INFO(0x010216, 0x4d00,  64 * 1024, 128, SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1234         { "s25fl256s0", INFO(0x010219, 0x4d00, 256 * 1024, 128, SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ | SPI_NOR_HAS_LOCK) },
1235         { "s25fl256s1", INFO(0x010219, 0x4d01,  64 * 1024, 512, SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1236         { "s25fl512s",  INFO(0x010220, 0x4d00, 256 * 1024, 256, SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1237         { "s70fl01gs",  INFO(0x010221, 0x4d00, 256 * 1024, 256, SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1238         { "s25sl12800", INFO(0x012018, 0x0300, 256 * 1024,  64, SPI_NOR_HAS_LOCK) },
1239         { "s25sl12801", INFO(0x012018, 0x0301,  64 * 1024, 256, SPI_NOR_HAS_LOCK) },
1240         { "s25fl128s",  INFO6(0x012018, 0x4d0180, 64 * 1024, 256, SPI_NOR_QUAD_READ) },
1241         { "s25fl129p0", INFO(0x012018, 0x4d00, 256 * 1024,  64, SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ | SPI_NOR_HAS_LOCK) },
1242         { "s25fl129p1", INFO(0x012018, 0x4d01,  64 * 1024, 256, SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ | SPI_NOR_HAS_LOCK) },
1243         { "s25sl004a",  INFO(0x010212,      0,  64 * 1024,   8, 0) },
1244         { "s25sl008a",  INFO(0x010213,      0,  64 * 1024,  16, 0) },
1245         { "s25sl016a",  INFO(0x010214,      0,  64 * 1024,  32, 0) },
1246         { "s25sl032a",  INFO(0x010215,      0,  64 * 1024,  64, 0) },
1247         { "s25sl064a",  INFO(0x010216,      0,  64 * 1024, 128, 0) },
1248         { "s25fl004k",  INFO(0xef4013,      0,  64 * 1024,   8, SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1249         { "s25fl008k",  INFO(0xef4014,      0,  64 * 1024,  16, SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1250         { "s25fl016k",  INFO(0xef4015,      0,  64 * 1024,  32, SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1251         { "s25fl064k",  INFO(0xef4017,      0,  64 * 1024, 128, SECT_4K) },
1252         { "s25fl116k",  INFO(0x014015,      0,  64 * 1024,  32, SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1253         { "s25fl132k",  INFO(0x014016,      0,  64 * 1024,  64, SECT_4K) },
1254         { "s25fl164k",  INFO(0x014017,      0,  64 * 1024, 128, SECT_4K) },
1255         { "s25fl204k",  INFO(0x014013,      0,  64 * 1024,   8, SECT_4K | SPI_NOR_DUAL_READ) },
1256         { "sst26wf016B", INFO(0xbf2651, 0, 64 * 1024, 32, SECT_4K | SST_GLOBAL_PROT_UNLK) },
1257
1258         /* SST -- large erase sizes are "overlays", "sectors" are 4K */
1259         { "sst25vf040b", INFO(0xbf258d, 0, 64 * 1024,  8, SECT_4K | SST_WRITE) },
1260         { "sst25vf080b", INFO(0xbf258e, 0, 64 * 1024, 16, SECT_4K | SST_WRITE) },
1261         { "sst25vf016b", INFO(0xbf2541, 0, 64 * 1024, 32, SECT_4K | SST_WRITE) },
1262         { "sst25vf032b", INFO(0xbf254a, 0, 64 * 1024, 64, SECT_4K | SST_WRITE) },
1263         { "sst25vf064c", INFO(0xbf254b, 0, 64 * 1024, 128, SECT_4K) },
1264         { "sst25wf512",  INFO(0xbf2501, 0, 64 * 1024,  1, SECT_4K | SST_WRITE) },
1265         { "sst25wf010",  INFO(0xbf2502, 0, 64 * 1024,  2, SECT_4K | SST_WRITE) },
1266         { "sst25wf020",  INFO(0xbf2503, 0, 64 * 1024,  4, SECT_4K | SST_WRITE) },
1267         { "sst25wf020a", INFO(0x621612, 0, 64 * 1024,  4, SECT_4K) },
1268         { "sst25wf040b", INFO(0x621613, 0, 64 * 1024,  8, SECT_4K) },
1269         { "sst25wf040",  INFO(0xbf2504, 0, 64 * 1024,  8, SECT_4K | SST_WRITE) },
1270         { "sst25wf080",  INFO(0xbf2505, 0, 64 * 1024, 16, SECT_4K | SST_WRITE) },
1271
1272         /* ST Microelectronics -- newer production may have feature updates */
1273         { "m25p05",  INFO(0x202010,  0,  32 * 1024,   2, 0) },
1274         { "m25p10",  INFO(0x202011,  0,  32 * 1024,   4, 0) },
1275         { "m25p20",  INFO(0x202012,  0,  64 * 1024,   4, 0) },
1276         { "m25p40",  INFO(0x202013,  0,  64 * 1024,   8, 0) },
1277         { "m25p80",  INFO(0x202014,  0,  64 * 1024,  16, 0) },
1278         { "m25p16",  INFO(0x202015,  0,  64 * 1024,  32, 0) },
1279         { "m25p32",  INFO(0x202016,  0,  64 * 1024,  64, 0) },
1280         { "m25p64",  INFO(0x202017,  0,  64 * 1024, 128, 0) },
1281         { "m25p128", INFO(0x202018,  0, 256 * 1024,  64, 0) },
1282
1283         { "m25p05-nonjedec",  INFO(0, 0,  32 * 1024,   2, 0) },
1284         { "m25p10-nonjedec",  INFO(0, 0,  32 * 1024,   4, 0) },
1285         { "m25p20-nonjedec",  INFO(0, 0,  64 * 1024,   4, 0) },
1286         { "m25p40-nonjedec",  INFO(0, 0,  64 * 1024,   8, 0) },
1287         { "m25p80-nonjedec",  INFO(0, 0,  64 * 1024,  16, 0) },
1288         { "m25p16-nonjedec",  INFO(0, 0,  64 * 1024,  32, 0) },
1289         { "m25p32-nonjedec",  INFO(0, 0,  64 * 1024,  64, 0) },
1290         { "m25p64-nonjedec",  INFO(0, 0,  64 * 1024, 128, 0) },
1291         { "m25p128-nonjedec", INFO(0, 0, 256 * 1024,  64, 0) },
1292
1293         { "m45pe10", INFO(0x204011,  0, 64 * 1024,    2, 0) },
1294         { "m45pe80", INFO(0x204014,  0, 64 * 1024,   16, 0) },
1295         { "m45pe16", INFO(0x204015,  0, 64 * 1024,   32, 0) },
1296
1297         { "m25pe20", INFO(0x208012,  0, 64 * 1024,  4,       0) },
1298         { "m25pe80", INFO(0x208014,  0, 64 * 1024, 16,       0) },
1299         { "m25pe16", INFO(0x208015,  0, 64 * 1024, 32, SECT_4K) },
1300
1301         { "m25px16",    INFO(0x207115,  0, 64 * 1024, 32, SECT_4K) },
1302         { "m25px32",    INFO(0x207116,  0, 64 * 1024, 64, SECT_4K) },
1303         { "m25px32-s0", INFO(0x207316,  0, 64 * 1024, 64, SECT_4K) },
1304         { "m25px32-s1", INFO(0x206316,  0, 64 * 1024, 64, SECT_4K) },
1305         { "m25px64",    INFO(0x207117,  0, 64 * 1024, 128, 0) },
1306         { "m25px80",    INFO(0x207114,  0, 64 * 1024, 16, 0) },
1307
1308         /* Winbond -- w25x "blocks" are 64K, "sectors" are 4KiB */
1309         { "w25x05", INFO(0xef3010, 0, 64 * 1024,  1,  SECT_4K) },
1310         { "w25x10", INFO(0xef3011, 0, 64 * 1024,  2,  SECT_4K) },
1311         { "w25x20", INFO(0xef3012, 0, 64 * 1024,  4,  SECT_4K) },
1312         { "w25x40", INFO(0xef3013, 0, 64 * 1024,  8,  SECT_4K) },
1313         { "w25x80", INFO(0xef3014, 0, 64 * 1024,  16, SECT_4K) },
1314         { "w25x16", INFO(0xef3015, 0, 64 * 1024,  32, SECT_4K) },
1315         { "w25x32", INFO(0xef3016, 0, 64 * 1024,  64, SECT_4K) },
1316         { "w25q32", INFO(0xef4016, 0, 64 * 1024,  64, SECT_4K) },
1317         {
1318                 "w25q32dw", INFO(0xef6016, 0, 64 * 1024,  64,
1319                         SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ |
1320                         SPI_NOR_HAS_LOCK | SPI_NOR_HAS_TB)
1321         },
1322         { "w25x64", INFO(0xef3017, 0, 64 * 1024, 128, SECT_4K) },
1323         { "w25q64", INFO(0xef4017, 0, 64 * 1024, 128, SECT_4K) },
1324         {
1325                 "w25q64dw", INFO(0xef6017, 0, 64 * 1024, 128,
1326                         SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ |
1327                         SPI_NOR_HAS_LOCK | SPI_NOR_HAS_TB)
1328         },
1329         {
1330                 "w25q128fw", INFO(0xef6018, 0, 64 * 1024, 256,
1331                         SECT_4K | SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ |
1332                         SPI_NOR_HAS_LOCK | SPI_NOR_HAS_TB)
1333         },
1334         { "w25q80", INFO(0xef5014, 0, 64 * 1024,  16, SECT_4K) },
1335         { "w25q80bl", INFO(0xef4014, 0, 64 * 1024,  16, SECT_4K) },
1336         { "w25q128", INFO(0xef4018, 0, 64 * 1024, 256, SECT_4K| SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1337         { "w25q256", INFO(0xef4019, 0, 64 * 1024, 512, SECT_4K| SPI_NOR_DUAL_READ | SPI_NOR_QUAD_READ) },
1338
1339         /* Catalyst / On Semiconductor -- non-JEDEC */
1340         { "cat25c11", CAT25_INFO(  16, 8, 16, 1, SPI_NOR_NO_ERASE | SPI_NOR_NO_FR) },
1341         { "cat25c03", CAT25_INFO(  32, 8, 16, 2, SPI_NOR_NO_ERASE | SPI_NOR_NO_FR) },
1342         { "cat25c09", CAT25_INFO( 128, 8, 32, 2, SPI_NOR_NO_ERASE | SPI_NOR_NO_FR) },
1343         { "cat25c17", CAT25_INFO( 256, 8, 32, 2, SPI_NOR_NO_ERASE | SPI_NOR_NO_FR) },
1344         { "cat25128", CAT25_INFO(2048, 8, 64, 2, SPI_NOR_NO_ERASE | SPI_NOR_NO_FR) },
1345         { },
1346 };
1347
1348 static const struct flash_info *spi_nor_read_id(struct spi_nor *nor)
1349 {
1350         int                     tmp;
1351         u8                      id[SPI_NOR_MAX_ID_LEN];
1352         const struct flash_info *info;
1353
1354         tmp = nor->read_reg(nor, SPINOR_OP_RDID, id, SPI_NOR_MAX_ID_LEN);
1355         if (tmp < 0) {
1356                 dev_dbg(nor->dev, "error %d reading JEDEC ID\n", tmp);
1357                 return ERR_PTR(tmp);
1358         }
1359
1360         for (tmp = 0; tmp < ARRAY_SIZE(spi_nor_ids) - 1; tmp++) {
1361                 info = &spi_nor_ids[tmp];
1362                 if (info->id_len) {
1363                         if (!memcmp(info->id, id, info->id_len))
1364                                 return &spi_nor_ids[tmp];
1365                 }
1366         }
1367         dev_err(nor->dev, "unrecognized JEDEC id bytes: %02x, %02x, %02x\n",
1368                 id[0], id[1], id[2]);
1369         return ERR_PTR(-ENODEV);
1370 }
1371
1372 static int spi_nor_read(struct mtd_info *mtd, loff_t from, size_t len,
1373                         size_t *retlen, u_char *buf)
1374 {
1375         struct spi_nor *nor = mtd_to_spi_nor(mtd);
1376         int ret;
1377         u32 offset = from;
1378         u32 stack_shift = 0;
1379         u32 read_len = 0;
1380         u32 rem_bank_len = 0;
1381         u8 bank;
1382         u8 is_ofst_odd = 0;
1383
1384 #define OFFSET_16_MB 0x1000000
1385
1386         dev_dbg(nor->dev, "from 0x%08x, len %zd\n", (u32)from, len);
1387
1388         if ((nor->isparallel) && (offset & 1)) {
1389                 /* We can hit this case when we use file system like ubifs */
1390                 from = (loff_t)(from - 1);
1391                 len = (size_t)(len + 1);
1392                 is_ofst_odd = 1;
1393         }
1394
1395         ret = spi_nor_lock_and_prep(nor, SPI_NOR_OPS_READ);
1396         if (ret)
1397                 return ret;
1398
1399         while (len) {
1400
1401                 if (nor->addr_width == 3) {
1402                         bank = (u32)from / (OFFSET_16_MB << nor->shift);
1403                         rem_bank_len = ((OFFSET_16_MB << nor->shift) *
1404                                                         (bank + 1)) - from;
1405                 }
1406                 offset = from;
1407
1408                 if (nor->isparallel == 1)
1409                         offset /= 2;
1410
1411                 if (nor->isstacked == 1) {
1412                         stack_shift = 1;
1413                         if (offset >= (mtd->size / 2)) {
1414                                 offset = offset - (mtd->size / 2);
1415                                 nor->spi->master->flags |= SPI_MASTER_U_PAGE;
1416                         } else {
1417                                 nor->spi->master->flags &= ~SPI_MASTER_U_PAGE;
1418                         }
1419                 }
1420
1421                 /* Die cross over issue is not handled */
1422                 if (nor->addr_width == 4) {
1423                         rem_bank_len = (mtd->size >> stack_shift) -
1424                                         (offset << nor->shift);
1425                 }
1426                 if (nor->addr_width == 3)
1427                         write_ear(nor, offset);
1428                 if (len < rem_bank_len)
1429                         read_len = len;
1430                 else
1431                         read_len = rem_bank_len;
1432
1433                 /* Wait till previous write/erase is done. */
1434                 ret = spi_nor_wait_till_ready(nor);
1435                 if (ret)
1436                         goto read_err;
1437
1438                 ret = nor->read(nor, offset, read_len, buf);
1439                 if (ret == 0) {
1440                         /* We shouldn't see 0-length reads */
1441                         ret = -EIO;
1442                         goto read_err;
1443                 }
1444                 if (ret < 0)
1445                         goto read_err;
1446
1447                 WARN_ON(ret > len);
1448                 if (is_ofst_odd == 1) {
1449                         memcpy(buf, (buf + 1), (len - 1));
1450                         *retlen += (ret - 1);
1451                 } else {
1452                         *retlen += ret;
1453                 }
1454                 buf += ret;
1455                 from += ret;
1456                 len -= ret;
1457         }
1458         ret = 0;
1459
1460 read_err:
1461         spi_nor_unlock_and_unprep(nor, SPI_NOR_OPS_READ);
1462         return ret;
1463 }
1464
1465 static int sst_write(struct mtd_info *mtd, loff_t to, size_t len,
1466                 size_t *retlen, const u_char *buf)
1467 {
1468         struct spi_nor *nor = mtd_to_spi_nor(mtd);
1469         size_t actual;
1470         int ret;
1471
1472         dev_dbg(nor->dev, "to 0x%08x, len %zd\n", (u32)to, len);
1473
1474         ret = spi_nor_lock_and_prep(nor, SPI_NOR_OPS_WRITE);
1475         if (ret)
1476                 return ret;
1477
1478         write_enable(nor);
1479
1480         nor->sst_write_second = false;
1481
1482         actual = to % 2;
1483         /* Start write from odd address. */
1484         if (actual) {
1485                 nor->program_opcode = SPINOR_OP_BP;
1486
1487                 /* write one byte. */
1488                 ret = nor->write(nor, to, 1, buf);
1489                 if (ret < 0)
1490                         goto sst_write_err;
1491                 WARN(ret != 1, "While writing 1 byte written %i bytes\n",
1492                      (int)ret);
1493                 ret = spi_nor_wait_till_ready(nor);
1494                 if (ret)
1495                         goto sst_write_err;
1496         }
1497         to += actual;
1498
1499         /* Write out most of the data here. */
1500         for (; actual < len - 1; actual += 2) {
1501                 nor->program_opcode = SPINOR_OP_AAI_WP;
1502
1503                 /* write two bytes. */
1504                 ret = nor->write(nor, to, 2, buf + actual);
1505                 if (ret < 0)
1506                         goto sst_write_err;
1507                 WARN(ret != 2, "While writing 2 bytes written %i bytes\n",
1508                      (int)ret);
1509                 ret = spi_nor_wait_till_ready(nor);
1510                 if (ret)
1511                         goto sst_write_err;
1512                 to += 2;
1513                 nor->sst_write_second = true;
1514         }
1515         nor->sst_write_second = false;
1516
1517         write_disable(nor);
1518         ret = spi_nor_wait_till_ready(nor);
1519         if (ret)
1520                 goto sst_write_err;
1521
1522         /* Write out trailing byte if it exists. */
1523         if (actual != len) {
1524                 write_enable(nor);
1525
1526                 nor->program_opcode = SPINOR_OP_BP;
1527                 ret = nor->write(nor, to, 1, buf + actual);
1528                 if (ret < 0)
1529                         goto sst_write_err;
1530                 WARN(ret != 1, "While writing 1 byte written %i bytes\n",
1531                      (int)ret);
1532                 ret = spi_nor_wait_till_ready(nor);
1533                 if (ret)
1534                         goto sst_write_err;
1535                 write_disable(nor);
1536                 actual += 1;
1537         }
1538 sst_write_err:
1539         *retlen += actual;
1540         spi_nor_unlock_and_unprep(nor, SPI_NOR_OPS_WRITE);
1541         return ret;
1542 }
1543
1544 /*
1545  * Write an address range to the nor chip.  Data must be written in
1546  * FLASH_PAGESIZE chunks.  The address range may be any size provided
1547  * it is within the physical boundaries.
1548  */
1549 static int spi_nor_write(struct mtd_info *mtd, loff_t to, size_t len,
1550         size_t *retlen, const u_char *buf)
1551 {
1552         struct spi_nor *nor = mtd_to_spi_nor(mtd);
1553         size_t page_offset, page_remain, i;
1554         ssize_t ret;
1555         u32 offset, stack_shift=0;
1556         u8 bank = 0;
1557         u32 rem_bank_len = 0;
1558
1559 #define OFFSET_16_MB 0x1000000
1560
1561         dev_dbg(nor->dev, "to 0x%08x, len %zd\n", (u32)to, len);
1562         ret = spi_nor_lock_and_prep(nor, SPI_NOR_OPS_WRITE);
1563         if (ret)
1564                 return ret;
1565         for (i = 0; i < len; ) {
1566                 ssize_t written;
1567
1568                 if (nor->addr_width == 3) {
1569                         bank = (u32)to / (OFFSET_16_MB << nor->shift);
1570                         rem_bank_len = ((OFFSET_16_MB << nor->shift) *
1571                                                         (bank + 1)) - to;
1572                 }
1573
1574                 page_offset = ((to + i)) & (nor->page_size - 1);
1575
1576                 offset = (to + i);
1577
1578                 if (nor->isparallel == 1)
1579                         offset /= 2;
1580
1581                 if (nor->isstacked == 1) {
1582                         stack_shift = 1;
1583                         if (offset >= (mtd->size / 2)) {
1584                                 offset = offset - (mtd->size / 2);
1585                                 nor->spi->master->flags |= SPI_MASTER_U_PAGE;
1586                         } else {
1587                                 nor->spi->master->flags &= ~SPI_MASTER_U_PAGE;
1588                         }
1589                 }
1590
1591                 /* Die cross over issue is not handled */
1592                 if (nor->addr_width == 4)
1593                         rem_bank_len = (mtd->size >> stack_shift) - offset;
1594                 if (nor->addr_width == 3)
1595                         write_ear(nor, (offset >> nor->shift));
1596                 if (len < rem_bank_len) {
1597                         page_remain = min_t(size_t,
1598                                     nor->page_size - page_offset, len - i);
1599
1600                 }
1601                 else {
1602                 /* the size of data remaining on the first page */
1603                         page_remain = rem_bank_len;
1604                 }
1605                 ret = spi_nor_wait_till_ready(nor);
1606                 if (ret)
1607                         goto write_err;
1608
1609                 write_enable(nor);
1610
1611                 ret = nor->write(nor, (offset), page_remain, buf + i);
1612                 if (ret < 0)
1613                         goto write_err;
1614                 written = ret;
1615
1616                 ret = spi_nor_wait_till_ready(nor);
1617                 if (ret)
1618                         goto write_err;
1619                 *retlen += written;
1620                 i += written;
1621                 if (written != page_remain) {
1622                         dev_err(nor->dev,
1623                                 "While writing %zu bytes written %zd bytes\n",
1624                                 page_remain, written);
1625                         ret = -EIO;
1626                         goto write_err;
1627                 }
1628         }
1629
1630 write_err:
1631         spi_nor_unlock_and_unprep(nor, SPI_NOR_OPS_WRITE);
1632         return ret;
1633 }
1634
1635 static int macronix_quad_enable(struct spi_nor *nor)
1636 {
1637         int ret, val;
1638
1639         val = read_sr(nor);
1640         if (val < 0)
1641                 return val;
1642         write_enable(nor);
1643
1644         write_sr(nor, val | SR_QUAD_EN_MX);
1645
1646         if (spi_nor_wait_till_ready(nor))
1647                 return 1;
1648
1649         ret = read_sr(nor);
1650         if (!(ret > 0 && (ret & SR_QUAD_EN_MX))) {
1651                 dev_err(nor->dev, "Macronix Quad bit not set\n");
1652                 return -EINVAL;
1653         }
1654
1655         return 0;
1656 }
1657
1658 static int spansion_quad_enable(struct spi_nor *nor)
1659 {
1660         int ret;
1661         int quad_en = CR_QUAD_EN_SPAN << 8;
1662
1663         quad_en |= read_sr(nor);
1664         quad_en |= (read_cr(nor) << 8);
1665
1666         write_enable(nor);
1667
1668         ret = write_sr_cr(nor, quad_en);
1669         if (ret < 0) {
1670                 dev_err(nor->dev,
1671                         "error while writing configuration register\n");
1672                 return -EINVAL;
1673         }
1674
1675         /* read back and check it */
1676         ret = read_cr(nor);
1677         if (!(ret > 0 && (ret & CR_QUAD_EN_SPAN))) {
1678                 dev_err(nor->dev, "Spansion Quad bit not set\n");
1679                 return -EINVAL;
1680         }
1681
1682         return 0;
1683 }
1684
1685 static int set_quad_mode(struct spi_nor *nor, const struct flash_info *info)
1686 {
1687         int status;
1688
1689         switch (JEDEC_MFR(info)) {
1690         case SNOR_MFR_MACRONIX:
1691         case SNOR_MFR_ISSI:
1692                 status = macronix_quad_enable(nor);
1693                 if (status) {
1694                         dev_err(nor->dev, "Macronix quad-read not enabled\n");
1695                         return -EINVAL;
1696                 }
1697                 return status;
1698         case SNOR_MFR_MICRON:
1699                 return 0;
1700         default:
1701                 status = spansion_quad_enable(nor);
1702                 if (status) {
1703                         dev_err(nor->dev, "Spansion quad-read not enabled\n");
1704                         return -EINVAL;
1705                 }
1706                 return status;
1707         }
1708 }
1709
1710 static int spi_nor_check(struct spi_nor *nor)
1711 {
1712         if (!nor->dev || !nor->read || !nor->write ||
1713                 !nor->read_reg || !nor->write_reg) {
1714                 pr_err("spi-nor: please fill all the necessary fields!\n");
1715                 return -EINVAL;
1716         }
1717
1718         return 0;
1719 }
1720
1721 int spi_nor_scan(struct spi_nor *nor, const char *name, enum read_mode mode)
1722 {
1723         struct flash_info *info = NULL;
1724         struct device *dev = nor->dev;
1725         struct mtd_info *mtd = &nor->mtd;
1726         struct device_node *np = spi_nor_get_flash_node(nor);
1727         struct device_node *np_spi ;
1728         int ret;
1729         int i;
1730         u32 is_dual;
1731
1732         ret = spi_nor_check(nor);
1733         if (ret)
1734                 return ret;
1735
1736         if (name)
1737                 info = (struct flash_info *)spi_nor_match_id(name);
1738         /* Try to auto-detect if chip name wasn't specified or not found */
1739         if (!info)
1740                 info = (struct flash_info *)spi_nor_read_id(nor);
1741         if (IS_ERR_OR_NULL(info))
1742                 return -ENOENT;
1743
1744         /*
1745          * If caller has specified name of flash model that can normally be
1746          * detected using JEDEC, let's verify it.
1747          */
1748         if (name && info->id_len) {
1749                 const struct flash_info *jinfo;
1750
1751                 jinfo = spi_nor_read_id(nor);
1752                 if (IS_ERR(jinfo)) {
1753                         return PTR_ERR(jinfo);
1754                 } else if (jinfo != info) {
1755                         /*
1756                          * JEDEC knows better, so overwrite platform ID. We
1757                          * can't trust partitions any longer, but we'll let
1758                          * mtd apply them anyway, since some partitions may be
1759                          * marked read-only, and we don't want to lose that
1760                          * information, even if it's not 100% accurate.
1761                          */
1762                         dev_warn(dev, "found %s, expected %s\n",
1763                                  jinfo->name, info->name);
1764                         info = (struct flash_info *)jinfo;
1765                 }
1766         }
1767
1768         mutex_init(&nor->lock);
1769
1770         /*
1771          * Atmel, SST, Intel/Numonyx, and others serial NOR tend to power up
1772          * with the software protection bits set
1773          */
1774
1775         if (JEDEC_MFR(info) == SNOR_MFR_ATMEL ||
1776             JEDEC_MFR(info) == SNOR_MFR_INTEL ||
1777             JEDEC_MFR(info) == SNOR_MFR_SST ||
1778             info->flags & SPI_NOR_HAS_LOCK) {
1779                 write_enable(nor);
1780                 write_sr(nor, 0);
1781                 if (info->flags & SST_GLOBAL_PROT_UNLK) {
1782                         write_enable(nor);
1783                         /* Unlock global write protection bits */
1784                         nor->write_reg(nor, GLOBAL_BLKPROT_UNLK, NULL, 0);
1785                 }
1786                 spi_nor_wait_till_ready(nor);
1787         }
1788
1789         if (!mtd->name)
1790                 mtd->name = dev_name(dev);
1791         mtd->priv = nor;
1792         mtd->type = MTD_NORFLASH;
1793         mtd->writesize = 1;
1794         mtd->flags = MTD_CAP_NORFLASH;
1795         mtd->size = info->sector_size * info->n_sectors;
1796         mtd->_erase = spi_nor_erase;
1797         mtd->_read = spi_nor_read;
1798 #ifdef CONFIG_OF
1799         np_spi = of_get_next_parent(np);
1800         if ((of_property_match_string(np_spi, "compatible",
1801                     "xlnx,zynq-qspi-1.0") >= 0) ||
1802                         (of_property_match_string(np_spi, "compatible",
1803                                         "xlnx,zynqmp-qspi-1.0") >= 0)) {
1804                         if (of_property_read_u32(np_spi, "is-dual",
1805                                                  &is_dual) < 0) {
1806                                 /* Default to single if prop not defined */
1807                                 nor->shift = 0;
1808                                 nor->isstacked = 0;
1809                                 nor->isparallel = 0;
1810                         } else {
1811                                 if (is_dual == 1) {
1812                                         /* dual parallel */
1813                                         nor->shift = 1;
1814                                         info->sector_size <<= nor->shift;
1815                                         info->page_size <<= nor->shift;
1816                                         mtd->size <<= nor->shift;
1817                                         nor->isparallel = 1;
1818                                         nor->isstacked = 0;
1819                                         nor->spi->master->flags |=
1820                                                         (SPI_MASTER_DATA_STRIPE
1821                                                         | SPI_MASTER_BOTH_CS);
1822                                 } else {
1823 #ifdef CONFIG_SPI_ZYNQ_QSPI_DUAL_STACKED
1824                                         /* dual stacked */
1825                                         nor->shift = 0;
1826                                         mtd->size <<= 1;
1827                                         info->n_sectors <<= 1;
1828                                         nor->isstacked = 1;
1829                                         nor->isparallel = 0;
1830 #else
1831                                         u32 is_stacked;
1832                                         if (of_property_read_u32(np_spi,
1833                                                         "is-stacked",
1834                                                         &is_stacked) < 0) {
1835                                                 is_stacked = 0;
1836                                         }
1837                                         if (is_stacked) {
1838                                                 /* dual stacked */
1839                                                 nor->shift = 0;
1840                                                 mtd->size <<= 1;
1841                                                 info->n_sectors <<= 1;
1842                                                 nor->isstacked = 1;
1843                                                 nor->isparallel = 0;
1844                                         } else {
1845                                                 /* single */
1846                                                 nor->shift = 0;
1847                                                 nor->isstacked = 0;
1848                                                 nor->isparallel = 0;
1849                                         }
1850 #endif
1851                                 }
1852                         }
1853         }
1854 #if 0
1855         pr_info("parallel %d stacked %d shift %d mtsize %d\n",
1856                 nor->isparallel, nor->isstacked, nor->shift, mtd->size);
1857 #endif
1858 #else
1859         /* Default to single */
1860         nor->shift = 0;
1861         nor->isstacked = 0;
1862         nor->isparallel = 0;
1863 #endif
1864
1865         /* NOR protection support for STmicro/Micron chips and similar */
1866         if (JEDEC_MFR(info) == SNOR_MFR_MICRON ||
1867                         info->flags & SPI_NOR_HAS_LOCK) {
1868                 nor->flash_lock = stm_lock;
1869                 nor->flash_unlock = stm_unlock;
1870                 nor->flash_is_locked = stm_is_locked;
1871         }
1872
1873         if (nor->flash_lock && nor->flash_unlock && nor->flash_is_locked) {
1874                 mtd->_lock = spi_nor_lock;
1875                 mtd->_unlock = spi_nor_unlock;
1876                 mtd->_is_locked = spi_nor_is_locked;
1877         }
1878
1879         /* sst nor chips use AAI word program */
1880         if (info->flags & SST_WRITE)
1881                 mtd->_write = sst_write;
1882         else
1883                 mtd->_write = spi_nor_write;
1884
1885         if (info->flags & USE_FSR)
1886                 nor->flags |= SNOR_F_USE_FSR;
1887         if (info->flags & SPI_NOR_HAS_TB)
1888                 nor->flags |= SNOR_F_HAS_SR_TB;
1889
1890 #ifdef CONFIG_MTD_SPI_NOR_USE_4K_SECTORS
1891         /* prefer "small sector" erase if possible */
1892         if (info->flags & SECT_4K) {
1893                 nor->erase_opcode = SPINOR_OP_BE_4K;
1894                 mtd->erasesize = 4096 << nor->shift;
1895         } else if (info->flags & SECT_4K_PMC) {
1896                 nor->erase_opcode = SPINOR_OP_BE_4K_PMC;
1897                 mtd->erasesize = 4096 << nor->shift;
1898         } else
1899 #endif
1900         {
1901                 nor->erase_opcode = SPINOR_OP_SE;
1902                 mtd->erasesize = info->sector_size;
1903         }
1904
1905         if (info->flags & SPI_NOR_NO_ERASE)
1906                 mtd->flags |= MTD_NO_ERASE;
1907
1908         nor->jedec_id = info->id[0];
1909         mtd->dev.parent = dev;
1910         nor->page_size = info->page_size;
1911         mtd->writebufsize = nor->page_size;
1912
1913         if (np) {
1914                 /* If we were instantiated by DT, use it */
1915                 if (of_property_read_bool(np, "m25p,fast-read"))
1916                         nor->flash_read = SPI_NOR_FAST;
1917                 else
1918                         nor->flash_read = SPI_NOR_NORMAL;
1919         } else {
1920                 /* If we weren't instantiated by DT, default to fast-read */
1921                 nor->flash_read = SPI_NOR_FAST;
1922         }
1923
1924         /* Some devices cannot do fast-read, no matter what DT tells us */
1925         if (info->flags & SPI_NOR_NO_FR)
1926                 nor->flash_read = SPI_NOR_NORMAL;
1927
1928         /* Quad/Dual-read mode takes precedence over fast/normal */
1929         if (mode == SPI_NOR_QUAD && info->flags & SPI_NOR_QUAD_READ) {
1930                 ret = set_quad_mode(nor, info);
1931                 if (ret) {
1932                         dev_err(dev, "quad mode not supported\n");
1933                         return ret;
1934                 }
1935                 nor->flash_read = SPI_NOR_QUAD;
1936         } else if (mode == SPI_NOR_DUAL && info->flags & SPI_NOR_DUAL_READ) {
1937                 nor->flash_read = SPI_NOR_DUAL;
1938         }
1939
1940         /* Default commands */
1941         switch (nor->flash_read) {
1942         case SPI_NOR_QUAD:
1943                 nor->read_opcode = SPINOR_OP_READ_1_1_4;
1944                 break;
1945         case SPI_NOR_DUAL:
1946                 nor->read_opcode = SPINOR_OP_READ_1_1_2;
1947                 break;
1948         case SPI_NOR_FAST:
1949                 nor->read_opcode = SPINOR_OP_READ_FAST;
1950                 break;
1951         case SPI_NOR_NORMAL:
1952                 nor->read_opcode = SPINOR_OP_READ;
1953                 break;
1954         default:
1955                 dev_err(dev, "No Read opcode defined\n");
1956                 return -EINVAL;
1957         }
1958
1959         nor->program_opcode = SPINOR_OP_PP;
1960
1961         if (info->addr_width)
1962                 nor->addr_width = info->addr_width;
1963         else if (mtd->size > 0x1000000) {
1964 #ifdef CONFIG_OF
1965                 np_spi = of_get_next_parent(np);
1966                 if (of_property_match_string(np_spi, "compatible",
1967                                              "xlnx,zynq-qspi-1.0") >= 0) {
1968                         int status;
1969
1970                         nor->addr_width = 3;
1971                         set_4byte(nor, info, 0);
1972                         status = read_ear(nor, info);
1973                         if (status < 0)
1974                                 dev_warn(dev, "failed to read ear reg\n");
1975                         else
1976                                 nor->curbank = status & EAR_SEGMENT_MASK;
1977                 } else {
1978 #endif
1979                 /* enable 4-byte addressing if the device exceeds 16MiB */
1980                 nor->addr_width = 4;
1981                 if (JEDEC_MFR(info) == SNOR_MFR_SPANSION) {
1982                         /* Dedicated 4-byte command set */
1983                         switch (nor->flash_read) {
1984                         case SPI_NOR_QUAD:
1985                                 nor->read_opcode = SPINOR_OP_READ4_1_1_4;
1986                                 break;
1987                         case SPI_NOR_DUAL:
1988                                 nor->read_opcode = SPINOR_OP_READ4_1_1_2;
1989                                 break;
1990                         case SPI_NOR_FAST:
1991                                 nor->read_opcode = SPINOR_OP_READ4_FAST;
1992                                 break;
1993                         case SPI_NOR_NORMAL:
1994                                 nor->read_opcode = SPINOR_OP_READ4;
1995                                 break;
1996                         }
1997                         nor->program_opcode = SPINOR_OP_PP_4B;
1998                         /* No small sector erase for 4-byte command set */
1999                         nor->erase_opcode = SPINOR_OP_SE_4B;
2000                         mtd->erasesize = info->sector_size;
2001                 } else {
2002                         np_spi = of_get_next_parent(np);
2003                         if (of_property_match_string(np_spi, "compatible",
2004                                                 "xlnx,xps-spi-2.00.a") >= 0) {
2005                                 nor->addr_width = 3;
2006                                 set_4byte(nor, info, 0);
2007                         } else {
2008                                 set_4byte(nor, info, 1);
2009                                 if (nor->isstacked) {
2010                                         nor->spi->master->flags |=
2011                                                         SPI_MASTER_U_PAGE;
2012                                         set_4byte(nor, info, 1);
2013                                         nor->spi->master->flags &=
2014                                                         ~SPI_MASTER_U_PAGE;
2015                                 }
2016                         }
2017                 }
2018 #ifdef CONFIG_OF
2019                 }
2020 #endif
2021         } else {
2022                 nor->addr_width = 3;
2023         }
2024
2025         if (nor->addr_width > SPI_NOR_MAX_ADDR_WIDTH) {
2026                 dev_err(dev, "address width is too large: %u\n",
2027                         nor->addr_width);
2028                 return -EINVAL;
2029         }
2030
2031         nor->read_dummy = spi_nor_read_dummy_cycles(nor);
2032
2033         dev_info(dev, "%s (%lld Kbytes)\n", info->name,
2034                         (long long)mtd->size >> 10);
2035
2036         dev_dbg(dev,
2037                 "mtd .name = %s, .size = 0x%llx (%lldMiB), "
2038                 ".erasesize = 0x%.8x (%uKiB) .numeraseregions = %d\n",
2039                 mtd->name, (long long)mtd->size, (long long)(mtd->size >> 20),
2040                 mtd->erasesize, mtd->erasesize / 1024, mtd->numeraseregions);
2041
2042         if (mtd->numeraseregions)
2043                 for (i = 0; i < mtd->numeraseregions; i++)
2044                         dev_dbg(dev,
2045                                 "mtd.eraseregions[%d] = { .offset = 0x%llx, "
2046                                 ".erasesize = 0x%.8x (%uKiB), "
2047                                 ".numblocks = %d }\n",
2048                                 i, (long long)mtd->eraseregions[i].offset,
2049                                 mtd->eraseregions[i].erasesize,
2050                                 mtd->eraseregions[i].erasesize / 1024,
2051                                 mtd->eraseregions[i].numblocks);
2052         return 0;
2053 }
2054 EXPORT_SYMBOL_GPL(spi_nor_scan);
2055
2056 static const struct flash_info *spi_nor_match_id(const char *name)
2057 {
2058         const struct flash_info *id = spi_nor_ids;
2059
2060         while (id->name) {
2061                 if (!strcmp(name, id->name))
2062                         return id;
2063                 id++;
2064         }
2065         return NULL;
2066 }
2067
2068 MODULE_LICENSE("GPL");
2069 MODULE_AUTHOR("Huang Shijie <shijie8@gmail.com>");
2070 MODULE_AUTHOR("Mike Lavender");
2071 MODULE_DESCRIPTION("framework for SPI NOR");