drivers/mtd/nand/atmel_nand.c

   1 /*
   2  *  Copyright © 2003 Rick Bronson
   3  *
   4  *  Derived from drivers/mtd/nand/autcpu12.c
   5  *       Copyright © 2001 Thomas Gleixner (gleixner@autronix.de)
   6  *
   7  *  Derived from drivers/mtd/spia.c
   8  *       Copyright © 2000 Steven J. Hill (sjhill@cotw.com)
   9  *
  10  *
  11  *  Add Hardware ECC support for AT91SAM9260 / AT91SAM9263
  12  *     Richard Genoud (richard.genoud@gmail.com), Adeneo Copyright © 2007
  13  *
  14  *     Derived from Das U-Boot source code
  15  *              (u-boot-1.1.5/board/atmel/at91sam9263ek/nand.c)
  16  *     © Copyright 2006 ATMEL Rousset, Lacressonniere Nicolas
  17  *
  18  *  Add Programmable Multibit ECC support for various AT91 SoC
  19  *     © Copyright 2012 ATMEL, Hong Xu
  20  *
  21  *  Add Nand Flash Controller support for SAMA5 SoC
  22  *     © Copyright 2013 ATMEL, Josh Wu (josh.wu@atmel.com)
  23  *
  24  * This program is free software; you can redistribute it and/or modify
  25  * it under the terms of the GNU General Public License version 2 as
  26  * published by the Free Software Foundation.
  27  *
  28  */
  29
  30 #include <linux/dma-mapping.h>
  31 #include <linux/slab.h>
  32 #include <linux/module.h>
  33 #include <linux/moduleparam.h>
  34 #include <linux/platform_device.h>
  35 #include <linux/of.h>
  36 #include <linux/of_device.h>
  37 #include <linux/of_gpio.h>
  38 #include <linux/of_mtd.h>
  39 #include <linux/mtd/mtd.h>
  40 #include <linux/mtd/nand.h>
  41 #include <linux/mtd/partitions.h>
  42
  43 #include <linux/delay.h>
  44 #include <linux/dmaengine.h>
  45 #include <linux/gpio.h>
  46 #include <linux/interrupt.h>
  47 #include <linux/io.h>
  48 #include <linux/platform_data/atmel.h>
  49
  50 static int use_dma = 1;
  51 module_param(use_dma, int, 0);
  52
  53 static int on_flash_bbt = 0;
  54 module_param(on_flash_bbt, int, 0);
  55
  56 /* Register access macros */
  57 #define ecc_readl(add, reg)                             \
  58         __raw_readl(add + ATMEL_ECC_##reg)
  59 #define ecc_writel(add, reg, value)                     \
  60         __raw_writel((value), add + ATMEL_ECC_##reg)
  61
  62 #include "atmel_nand_ecc.h"     /* Hardware ECC registers */
  63 #include "atmel_nand_nfc.h"     /* Nand Flash Controller definition */
  64
  65 /* oob layout for large page size
  66  * bad block info is on bytes 0 and 1
  67  * the bytes have to be consecutives to avoid
  68  * several NAND_CMD_RNDOUT during read
  69  */
  70 static struct nand_ecclayout atmel_oobinfo_large = {
  71         .eccbytes = 4,
  72         .eccpos = {60, 61, 62, 63},
  73         .oobfree = {
  74                 {2, 58}
  75         },
  76 };
  77
  78 /* oob layout for small page size
  79  * bad block info is on bytes 4 and 5
  80  * the bytes have to be consecutives to avoid
  81  * several NAND_CMD_RNDOUT during read
  82  */
  83 static struct nand_ecclayout atmel_oobinfo_small = {
  84         .eccbytes = 4,
  85         .eccpos = {0, 1, 2, 3},
  86         .oobfree = {
  87                 {6, 10}
  88         },
  89 };
  90
  91 struct atmel_nfc {
  92         void __iomem            *base_cmd_regs;
  93         void __iomem            *hsmc_regs;
  94         void __iomem            *sram_bank0;
  95         dma_addr_t              sram_bank0_phys;
  96         bool                    use_nfc_sram;
  97         bool                    write_by_sram;
  98
  99         bool                    is_initialized;
 100         struct completion       comp_ready;
 101         struct completion       comp_cmd_done;
 102         struct completion       comp_xfer_done;
 103
 104         /* Point to the sram bank which include readed data via NFC */
 105         void __iomem            *data_in_sram;
 106         bool                    will_write_sram;
 107 };
 108 static struct atmel_nfc nand_nfc;
 109
 110 struct atmel_nand_host {
 111         struct nand_chip        nand_chip;
 112         struct mtd_info         mtd;
 113         void __iomem            *io_base;
 114         dma_addr_t              io_phys;
 115         struct atmel_nand_data  board;
 116         struct device           *dev;
 117         void __iomem            *ecc;
 118
 119         struct completion       comp;
 120         struct dma_chan         *dma_chan;
 121
 122         struct atmel_nfc        *nfc;
 123
 124         bool                    has_pmecc;
 125         u8                      pmecc_corr_cap;
 126         u16                     pmecc_sector_size;
 127         u32                     pmecc_lookup_table_offset;
 128         u32                     pmecc_lookup_table_offset_512;
 129         u32                     pmecc_lookup_table_offset_1024;
 130
 131         int                     pmecc_bytes_per_sector;
 132         int                     pmecc_sector_number;
 133         int                     pmecc_degree;   /* Degree of remainders */
 134         int                     pmecc_cw_len;   /* Length of codeword */
 135
 136         void __iomem            *pmerrloc_base;
 137         void __iomem            *pmecc_rom_base;
 138
 139         /* lookup table for alpha_to and index_of */
 140         void __iomem            *pmecc_alpha_to;
 141         void __iomem            *pmecc_index_of;
 142
 143         /* data for pmecc computation */
 144         int16_t                 *pmecc_partial_syn;
 145         int16_t                 *pmecc_si;
 146         int16_t                 *pmecc_smu;     /* Sigma table */
 147         int16_t                 *pmecc_lmu;     /* polynomal order */
 148         int                     *pmecc_mu;
 149         int                     *pmecc_dmu;
 150         int                     *pmecc_delta;
 151 };
 152
 153 static struct nand_ecclayout atmel_pmecc_oobinfo;
 154
 155 /*
 156  * Enable NAND.
 157  */
 158 static void atmel_nand_enable(struct atmel_nand_host *host)
 159 {
 160         if (gpio_is_valid(host->board.enable_pin))
 161                 gpio_set_value(host->board.enable_pin, 0);
 162 }
 163
 164 /*
 165  * Disable NAND.
 166  */
 167 static void atmel_nand_disable(struct atmel_nand_host *host)
 168 {
 169         if (gpio_is_valid(host->board.enable_pin))
 170                 gpio_set_value(host->board.enable_pin, 1);
 171 }
 172
 173 /*
 174  * Hardware specific access to control-lines
 175  */
 176 static void atmel_nand_cmd_ctrl(struct mtd_info *mtd, int cmd, unsigned int ctrl)
 177 {
 178         struct nand_chip *nand_chip = mtd->priv;
 179         struct atmel_nand_host *host = nand_chip->priv;
 180
 181         if (ctrl & NAND_CTRL_CHANGE) {
 182                 if (ctrl & NAND_NCE)
 183                         atmel_nand_enable(host);
 184                 else
 185                         atmel_nand_disable(host);
 186         }
 187         if (cmd == NAND_CMD_NONE)
 188                 return;
 189
 190         if (ctrl & NAND_CLE)
 191                 writeb(cmd, host->io_base + (1 << host->board.cle));
 192         else
 193                 writeb(cmd, host->io_base + (1 << host->board.ale));
 194 }
 195
 196 /*
 197  * Read the Device Ready pin.
 198  */
 199 static int atmel_nand_device_ready(struct mtd_info *mtd)
 200 {
 201         struct nand_chip *nand_chip = mtd->priv;
 202         struct atmel_nand_host *host = nand_chip->priv;
 203
 204         return gpio_get_value(host->board.rdy_pin) ^
 205                 !!host->board.rdy_pin_active_low;
 206 }
 207
 208 /* Set up for hardware ready pin and enable pin. */
 209 static int atmel_nand_set_enable_ready_pins(struct mtd_info *mtd)
 210 {
 211         struct nand_chip *chip = mtd->priv;
 212         struct atmel_nand_host *host = chip->priv;
 213         int res = 0;
 214
 215         if (gpio_is_valid(host->board.rdy_pin)) {
 216                 res = devm_gpio_request(host->dev,
 217                                 host->board.rdy_pin, "nand_rdy");
 218                 if (res < 0) {
 219                         dev_err(host->dev,
 220                                 "can't request rdy gpio %d\n",
 221                                 host->board.rdy_pin);
 222                         return res;
 223                 }
 224
 225                 res = gpio_direction_input(host->board.rdy_pin);
 226                 if (res < 0) {
 227                         dev_err(host->dev,
 228                                 "can't request input direction rdy gpio %d\n",
 229                                 host->board.rdy_pin);
 230                         return res;
 231                 }
 232
 233                 chip->dev_ready = atmel_nand_device_ready;
 234         }
 235
 236         if (gpio_is_valid(host->board.enable_pin)) {
 237                 res = devm_gpio_request(host->dev,
 238                                 host->board.enable_pin, "nand_enable");
 239                 if (res < 0) {
 240                         dev_err(host->dev,
 241                                 "can't request enable gpio %d\n",
 242                                 host->board.enable_pin);
 243                         return res;
 244                 }
 245
 246                 res = gpio_direction_output(host->board.enable_pin, 1);
 247                 if (res < 0) {
 248                         dev_err(host->dev,
 249                                 "can't request output direction enable gpio %d\n",
 250                                 host->board.enable_pin);
 251                         return res;
 252                 }
 253         }
 254
 255         return res;
 256 }
 257
 258 static void memcpy32_fromio(void *trg, const void __iomem  *src, size_t size)
 259 {
 260         int i;
 261         u32 *t = trg;
 262         const __iomem u32 *s = src;
 263
 264         for (i = 0; i < (size >> 2); i++)
 265                 *t++ = readl_relaxed(s++);
 266 }
 267
 268 static void memcpy32_toio(void __iomem *trg, const void *src, int size)
 269 {
 270         int i;
 271         u32 __iomem *t = trg;
 272         const u32 *s = src;
 273
 274         for (i = 0; i < (size >> 2); i++)
 275                 writel_relaxed(*s++, t++);
 276 }
 277
 278 /*
 279  * Minimal-overhead PIO for data access.
 280  */
 281 static void atmel_read_buf8(struct mtd_info *mtd, u8 *buf, int len)
 282 {
 283         struct nand_chip        *nand_chip = mtd->priv;
 284         struct atmel_nand_host *host = nand_chip->priv;
 285
 286         if (host->nfc && host->nfc->use_nfc_sram && host->nfc->data_in_sram) {
 287                 memcpy32_fromio(buf, host->nfc->data_in_sram, len);
 288                 host->nfc->data_in_sram += len;
 289         } else {
 290                 __raw_readsb(nand_chip->IO_ADDR_R, buf, len);
 291         }
 292 }
 293
 294 static void atmel_read_buf16(struct mtd_info *mtd, u8 *buf, int len)
 295 {
 296         struct nand_chip        *nand_chip = mtd->priv;
 297         struct atmel_nand_host *host = nand_chip->priv;
 298
 299         if (host->nfc && host->nfc->use_nfc_sram && host->nfc->data_in_sram) {
 300                 memcpy32_fromio(buf, host->nfc->data_in_sram, len);
 301                 host->nfc->data_in_sram += len;
 302         } else {
 303                 __raw_readsw(nand_chip->IO_ADDR_R, buf, len / 2);
 304         }
 305 }
 306
 307 static void atmel_write_buf8(struct mtd_info *mtd, const u8 *buf, int len)
 308 {
 309         struct nand_chip        *nand_chip = mtd->priv;
 310
 311         __raw_writesb(nand_chip->IO_ADDR_W, buf, len);
 312 }
 313
 314 static void atmel_write_buf16(struct mtd_info *mtd, const u8 *buf, int len)
 315 {
 316         struct nand_chip        *nand_chip = mtd->priv;
 317
 318         __raw_writesw(nand_chip->IO_ADDR_W, buf, len / 2);
 319 }
 320
 321 static void dma_complete_func(void *completion)
 322 {
 323         complete(completion);
 324 }
 325
 326 static int nfc_set_sram_bank(struct atmel_nand_host *host, unsigned int bank)
 327 {
 328         /* NFC only has two banks. Must be 0 or 1 */
 329         if (bank > 1)
 330                 return -EINVAL;
 331
 332         if (bank) {
 333                 /* Only for a 2k-page or lower flash, NFC can handle 2 banks */
 334                 if (host->mtd.writesize > 2048)
 335                         return -EINVAL;
 336                 nfc_writel(host->nfc->hsmc_regs, BANK, ATMEL_HSMC_NFC_BANK1);
 337         } else {
 338                 nfc_writel(host->nfc->hsmc_regs, BANK, ATMEL_HSMC_NFC_BANK0);
 339         }
 340
 341         return 0;
 342 }
 343
 344 static uint nfc_get_sram_off(struct atmel_nand_host *host)
 345 {
 346         if (nfc_readl(host->nfc->hsmc_regs, BANK) & ATMEL_HSMC_NFC_BANK1)
 347                 return NFC_SRAM_BANK1_OFFSET;
 348         else
 349                 return 0;
 350 }
 351
 352 static dma_addr_t nfc_sram_phys(struct atmel_nand_host *host)
 353 {
 354         if (nfc_readl(host->nfc->hsmc_regs, BANK) & ATMEL_HSMC_NFC_BANK1)
 355                 return host->nfc->sram_bank0_phys + NFC_SRAM_BANK1_OFFSET;
 356         else
 357                 return host->nfc->sram_bank0_phys;
 358 }
 359
 360 static int atmel_nand_dma_op(struct mtd_info *mtd, void *buf, int len,
 361                                int is_read)
 362 {
 363         struct dma_device *dma_dev;
 364         enum dma_ctrl_flags flags;
 365         dma_addr_t dma_src_addr, dma_dst_addr, phys_addr;
 366         struct dma_async_tx_descriptor *tx = NULL;
 367         dma_cookie_t cookie;
 368         struct nand_chip *chip = mtd->priv;
 369         struct atmel_nand_host *host = chip->priv;
 370         void *p = buf;
 371         int err = -EIO;
 372         enum dma_data_direction dir = is_read ? DMA_FROM_DEVICE : DMA_TO_DEVICE;
 373         struct atmel_nfc *nfc = host->nfc;
 374
 375         if (buf >= high_memory)
 376                 goto err_buf;
 377
 378         dma_dev = host->dma_chan->device;
 379
 380         flags = DMA_CTRL_ACK | DMA_PREP_INTERRUPT;
 381
 382         phys_addr = dma_map_single(dma_dev->dev, p, len, dir);
 383         if (dma_mapping_error(dma_dev->dev, phys_addr)) {
 384                 dev_err(host->dev, "Failed to dma_map_single\n");
 385                 goto err_buf;
 386         }
 387
 388         if (is_read) {
 389                 if (nfc && nfc->data_in_sram)
 390                         dma_src_addr = nfc_sram_phys(host) + (nfc->data_in_sram
 391                                 - (nfc->sram_bank0 + nfc_get_sram_off(host)));
 392                 else
 393                         dma_src_addr = host->io_phys;
 394
 395                 dma_dst_addr = phys_addr;
 396         } else {
 397                 dma_src_addr = phys_addr;
 398
 399                 if (nfc && nfc->write_by_sram)
 400                         dma_dst_addr = nfc_sram_phys(host);
 401                 else
 402                         dma_dst_addr = host->io_phys;
 403         }
 404
 405         tx = dma_dev->device_prep_dma_memcpy(host->dma_chan, dma_dst_addr,
 406                                              dma_src_addr, len, flags);
 407         if (!tx) {
 408                 dev_err(host->dev, "Failed to prepare DMA memcpy\n");
 409                 goto err_dma;
 410         }
 411
 412         init_completion(&host->comp);
 413         tx->callback = dma_complete_func;
 414         tx->callback_param = &host->comp;
 415
 416         cookie = tx->tx_submit(tx);
 417         if (dma_submit_error(cookie)) {
 418                 dev_err(host->dev, "Failed to do DMA tx_submit\n");
 419                 goto err_dma;
 420         }
 421
 422         dma_async_issue_pending(host->dma_chan);
 423         wait_for_completion(&host->comp);
 424
 425         if (is_read && nfc && nfc->data_in_sram)
 426                 /* After read data from SRAM, need to increase the position */
 427                 nfc->data_in_sram += len;
 428
 429         err = 0;
 430
 431 err_dma:
 432         dma_unmap_single(dma_dev->dev, phys_addr, len, dir);
 433 err_buf:
 434         if (err != 0)
 435                 dev_dbg(host->dev, "Fall back to CPU I/O\n");
 436         return err;
 437 }
 438
 439 static void atmel_read_buf(struct mtd_info *mtd, u8 *buf, int len)
 440 {
 441         struct nand_chip *chip = mtd->priv;
 442         struct atmel_nand_host *host = chip->priv;
 443
 444         if (use_dma && len > mtd->oobsize)
 445                 /* only use DMA for bigger than oob size: better performances */
 446                 if (atmel_nand_dma_op(mtd, buf, len, 1) == 0)
 447                         return;
 448
 449         if (host->board.bus_width_16)
 450                 atmel_read_buf16(mtd, buf, len);
 451         else
 452                 atmel_read_buf8(mtd, buf, len);
 453 }
 454
 455 static void atmel_write_buf(struct mtd_info *mtd, const u8 *buf, int len)
 456 {
 457         struct nand_chip *chip = mtd->priv;
 458         struct atmel_nand_host *host = chip->priv;
 459
 460         if (use_dma && len > mtd->oobsize)
 461                 /* only use DMA for bigger than oob size: better performances */
 462                 if (atmel_nand_dma_op(mtd, (void *)buf, len, 0) == 0)
 463                         return;
 464
 465         if (host->board.bus_width_16)
 466                 atmel_write_buf16(mtd, buf, len);
 467         else
 468                 atmel_write_buf8(mtd, buf, len);
 469 }
 470
 471 /*
 472  * Return number of ecc bytes per sector according to sector size and
 473  * correction capability
 474  *
 475  * Following table shows what at91 PMECC supported:
 476  * Correction Capability        Sector_512_bytes        Sector_1024_bytes
 477  * =====================        ================        =================
 478  *                2-bits                 4-bytes                  4-bytes
 479  *                4-bits                 7-bytes                  7-bytes
 480  *                8-bits                13-bytes                 14-bytes
 481  *               12-bits                20-bytes                 21-bytes
 482  *               24-bits                39-bytes                 42-bytes
 483  */
 484 static int pmecc_get_ecc_bytes(int cap, int sector_size)
 485 {
 486         int m = 12 + sector_size / 512;
 487         return (m * cap + 7) / 8;
 488 }
 489
 490 static void pmecc_config_ecc_layout(struct nand_ecclayout *layout,
 491                                     int oobsize, int ecc_len)
 492 {
 493         int i;
 494
 495         layout->eccbytes = ecc_len;
 496
 497         /* ECC will occupy the last ecc_len bytes continuously */
 498         for (i = 0; i < ecc_len; i++)
 499                 layout->eccpos[i] = oobsize - ecc_len + i;
 500
 501         layout->oobfree[0].offset = 2;
 502         layout->oobfree[0].length =
 503                 oobsize - ecc_len - layout->oobfree[0].offset;
 504 }
 505
 506 static void __iomem *pmecc_get_alpha_to(struct atmel_nand_host *host)
 507 {
 508         int table_size;
 509
 510         table_size = host->pmecc_sector_size == 512 ?
 511                 PMECC_LOOKUP_TABLE_SIZE_512 : PMECC_LOOKUP_TABLE_SIZE_1024;
 512
 513         return host->pmecc_rom_base + host->pmecc_lookup_table_offset +
 514                         table_size * sizeof(int16_t);
 515 }
 516
 517 static int pmecc_data_alloc(struct atmel_nand_host *host)
 518 {
 519         const int cap = host->pmecc_corr_cap;
 520         int size;
 521
 522         size = (2 * cap + 1) * sizeof(int16_t);
 523         host->pmecc_partial_syn = devm_kzalloc(host->dev, size, GFP_KERNEL);
 524         host->pmecc_si = devm_kzalloc(host->dev, size, GFP_KERNEL);
 525         host->pmecc_lmu = devm_kzalloc(host->dev,
 526                         (cap + 1) * sizeof(int16_t), GFP_KERNEL);
 527         host->pmecc_smu = devm_kzalloc(host->dev,
 528                         (cap + 2) * size, GFP_KERNEL);
 529
 530         size = (cap + 1) * sizeof(int);
 531         host->pmecc_mu = devm_kzalloc(host->dev, size, GFP_KERNEL);
 532         host->pmecc_dmu = devm_kzalloc(host->dev, size, GFP_KERNEL);
 533         host->pmecc_delta = devm_kzalloc(host->dev, size, GFP_KERNEL);
 534
 535         if (!host->pmecc_partial_syn ||
 536                 !host->pmecc_si ||
 537                 !host->pmecc_lmu ||
 538                 !host->pmecc_smu ||
 539                 !host->pmecc_mu ||
 540                 !host->pmecc_dmu ||
 541                 !host->pmecc_delta)
 542                 return -ENOMEM;
 543
 544         return 0;
 545 }
 546
 547 static void pmecc_gen_syndrome(struct mtd_info *mtd, int sector)
 548 {
 549         struct nand_chip *nand_chip = mtd->priv;
 550         struct atmel_nand_host *host = nand_chip->priv;
 551         int i;
 552         uint32_t value;
 553
 554         /* Fill odd syndromes */
 555         for (i = 0; i < host->pmecc_corr_cap; i++) {
 556                 value = pmecc_readl_rem_relaxed(host->ecc, sector, i / 2);
 557                 if (i & 1)
 558                         value >>= 16;
 559                 value &= 0xffff;
 560                 host->pmecc_partial_syn[(2 * i) + 1] = (int16_t)value;
 561         }
 562 }
 563
 564 static void pmecc_substitute(struct mtd_info *mtd)
 565 {
 566         struct nand_chip *nand_chip = mtd->priv;
 567         struct atmel_nand_host *host = nand_chip->priv;
 568         int16_t __iomem *alpha_to = host->pmecc_alpha_to;
 569         int16_t __iomem *index_of = host->pmecc_index_of;
 570         int16_t *partial_syn = host->pmecc_partial_syn;
 571         const int cap = host->pmecc_corr_cap;
 572         int16_t *si;
 573         int i, j;
 574
 575         /* si[] is a table that holds the current syndrome value,
 576          * an element of that table belongs to the field
 577          */
 578         si = host->pmecc_si;
 579
 580         memset(&si[1], 0, sizeof(int16_t) * (2 * cap - 1));
 581
 582         /* Computation 2t syndromes based on S(x) */
 583         /* Odd syndromes */
 584         for (i = 1; i < 2 * cap; i += 2) {
 585                 for (j = 0; j < host->pmecc_degree; j++) {
 586                         if (partial_syn[i] & ((unsigned short)0x1 << j))
 587                                 si[i] = readw_relaxed(alpha_to + i * j) ^ si[i];
 588                 }
 589         }
 590         /* Even syndrome = (Odd syndrome) ** 2 */
 591         for (i = 2, j = 1; j <= cap; i = ++j << 1) {
 592                 if (si[j] == 0) {
 593                         si[i] = 0;
 594                 } else {
 595                         int16_t tmp;
 596
 597                         tmp = readw_relaxed(index_of + si[j]);
 598                         tmp = (tmp * 2) % host->pmecc_cw_len;
 599                         si[i] = readw_relaxed(alpha_to + tmp);
 600                 }
 601         }
 602
 603         return;
 604 }
 605
 606 static void pmecc_get_sigma(struct mtd_info *mtd)
 607 {
 608         struct nand_chip *nand_chip = mtd->priv;
 609         struct atmel_nand_host *host = nand_chip->priv;
 610
 611         int16_t *lmu = host->pmecc_lmu;
 612         int16_t *si = host->pmecc_si;
 613         int *mu = host->pmecc_mu;
 614         int *dmu = host->pmecc_dmu;     /* Discrepancy */
 615         int *delta = host->pmecc_delta; /* Delta order */
 616         int cw_len = host->pmecc_cw_len;
 617         const int16_t cap = host->pmecc_corr_cap;
 618         const int num = 2 * cap + 1;
 619         int16_t __iomem *index_of = host->pmecc_index_of;
 620         int16_t __iomem *alpha_to = host->pmecc_alpha_to;
 621         int i, j, k;
 622         uint32_t dmu_0_count, tmp;
 623         int16_t *smu = host->pmecc_smu;
 624
 625         /* index of largest delta */
 626         int ro;
 627         int largest;
 628         int diff;
 629
 630         dmu_0_count = 0;
 631
 632         /* First Row */
 633
 634         /* Mu */
 635         mu[0] = -1;
 636
 637         memset(smu, 0, sizeof(int16_t) * num);
 638         smu[0] = 1;
 639
 640         /* discrepancy set to 1 */
 641         dmu[0] = 1;
 642         /* polynom order set to 0 */
 643         lmu[0] = 0;
 644         delta[0] = (mu[0] * 2 - lmu[0]) >> 1;
 645
 646         /* Second Row */
 647
 648         /* Mu */
 649         mu[1] = 0;
 650         /* Sigma(x) set to 1 */
 651         memset(&smu[num], 0, sizeof(int16_t) * num);
 652         smu[num] = 1;
 653
 654         /* discrepancy set to S1 */
 655         dmu[1] = si[1];
 656
 657         /* polynom order set to 0 */
 658         lmu[1] = 0;
 659
 660         delta[1] = (mu[1] * 2 - lmu[1]) >> 1;
 661
 662         /* Init the Sigma(x) last row */
 663         memset(&smu[(cap + 1) * num], 0, sizeof(int16_t) * num);
 664
 665         for (i = 1; i <= cap; i++) {
 666                 mu[i + 1] = i << 1;
 667                 /* Begin Computing Sigma (Mu+1) and L(mu) */
 668                 /* check if discrepancy is set to 0 */
 669                 if (dmu[i] == 0) {
 670                         dmu_0_count++;
 671
 672                         tmp = ((cap - (lmu[i] >> 1) - 1) / 2);
 673                         if ((cap - (lmu[i] >> 1) - 1) & 0x1)
 674                                 tmp += 2;
 675                         else
 676                                 tmp += 1;
 677
 678                         if (dmu_0_count == tmp) {
 679                                 for (j = 0; j <= (lmu[i] >> 1) + 1; j++)
 680                                         smu[(cap + 1) * num + j] =
 681                                                         smu[i * num + j];
 682
 683                                 lmu[cap + 1] = lmu[i];
 684                                 return;
 685                         }
 686
 687                         /* copy polynom */
 688                         for (j = 0; j <= lmu[i] >> 1; j++)
 689                                 smu[(i + 1) * num + j] = smu[i * num + j];
 690
 691                         /* copy previous polynom order to the next */
 692                         lmu[i + 1] = lmu[i];
 693                 } else {
 694                         ro = 0;
 695                         largest = -1;
 696                         /* find largest delta with dmu != 0 */
 697                         for (j = 0; j < i; j++) {
 698                                 if ((dmu[j]) && (delta[j] > largest)) {
 699                                         largest = delta[j];
 700                                         ro = j;
 701                                 }
 702                         }
 703
 704                         /* compute difference */
 705                         diff = (mu[i] - mu[ro]);
 706
 707                         /* Compute degree of the new smu polynomial */
 708                         if ((lmu[i] >> 1) > ((lmu[ro] >> 1) + diff))
 709                                 lmu[i + 1] = lmu[i];
 710                         else
 711                                 lmu[i + 1] = ((lmu[ro] >> 1) + diff) * 2;
 712
 713                         /* Init smu[i+1] with 0 */
 714                         for (k = 0; k < num; k++)
 715                                 smu[(i + 1) * num + k] = 0;
 716
 717                         /* Compute smu[i+1] */
 718                         for (k = 0; k <= lmu[ro] >> 1; k++) {
 719                                 int16_t a, b, c;
 720
 721                                 if (!(smu[ro * num + k] && dmu[i]))
 722                                         continue;
 723                                 a = readw_relaxed(index_of + dmu[i]);
 724                                 b = readw_relaxed(index_of + dmu[ro]);
 725                                 c = readw_relaxed(index_of + smu[ro * num + k]);
 726                                 tmp = a + (cw_len - b) + c;
 727                                 a = readw_relaxed(alpha_to + tmp % cw_len);
 728                                 smu[(i + 1) * num + (k + diff)] = a;
 729                         }
 730
 731                         for (k = 0; k <= lmu[i] >> 1; k++)
 732                                 smu[(i + 1) * num + k] ^= smu[i * num + k];
 733                 }
 734
 735                 /* End Computing Sigma (Mu+1) and L(mu) */
 736                 /* In either case compute delta */
 737                 delta[i + 1] = (mu[i + 1] * 2 - lmu[i + 1]) >> 1;
 738
 739                 /* Do not compute discrepancy for the last iteration */
 740                 if (i >= cap)
 741                         continue;
 742
 743                 for (k = 0; k <= (lmu[i + 1] >> 1); k++) {
 744                         tmp = 2 * (i - 1);
 745                         if (k == 0) {
 746                                 dmu[i + 1] = si[tmp + 3];
 747                         } else if (smu[(i + 1) * num + k] && si[tmp + 3 - k]) {
 748                                 int16_t a, b, c;
 749                                 a = readw_relaxed(index_of +
 750                                                 smu[(i + 1) * num + k]);
 751                                 b = si[2 * (i - 1) + 3 - k];
 752                                 c = readw_relaxed(index_of + b);
 753                                 tmp = a + c;
 754                                 tmp %= cw_len;
 755                                 dmu[i + 1] = readw_relaxed(alpha_to + tmp) ^
 756                                         dmu[i + 1];
 757                         }
 758                 }
 759         }
 760
 761         return;
 762 }
 763
 764 static int pmecc_err_location(struct mtd_info *mtd)
 765 {
 766         struct nand_chip *nand_chip = mtd->priv;
 767         struct atmel_nand_host *host = nand_chip->priv;
 768         unsigned long end_time;
 769         const int cap = host->pmecc_corr_cap;
 770         const int num = 2 * cap + 1;
 771         int sector_size = host->pmecc_sector_size;
 772         int err_nbr = 0;        /* number of error */
 773         int roots_nbr;          /* number of roots */
 774         int i;
 775         uint32_t val;
 776         int16_t *smu = host->pmecc_smu;
 777
 778         pmerrloc_writel(host->pmerrloc_base, ELDIS, PMERRLOC_DISABLE);
 779
 780         for (i = 0; i <= host->pmecc_lmu[cap + 1] >> 1; i++) {
 781                 pmerrloc_writel_sigma_relaxed(host->pmerrloc_base, i,
 782                                       smu[(cap + 1) * num + i]);
 783                 err_nbr++;
 784         }
 785
 786         val = (err_nbr - 1) << 16;
 787         if (sector_size == 1024)
 788                 val |= 1;
 789
 790         pmerrloc_writel(host->pmerrloc_base, ELCFG, val);
 791         pmerrloc_writel(host->pmerrloc_base, ELEN,
 792                         sector_size * 8 + host->pmecc_degree * cap);
 793
 794         end_time = jiffies + msecs_to_jiffies(PMECC_MAX_TIMEOUT_MS);
 795         while (!(pmerrloc_readl_relaxed(host->pmerrloc_base, ELISR)
 796                  & PMERRLOC_CALC_DONE)) {
 797                 if (unlikely(time_after(jiffies, end_time))) {
 798                         dev_err(host->dev, "PMECC: Timeout to calculate error location.\n");
 799                         return -1;
 800                 }
 801                 cpu_relax();
 802         }
 803
 804         roots_nbr = (pmerrloc_readl_relaxed(host->pmerrloc_base, ELISR)
 805                 & PMERRLOC_ERR_NUM_MASK) >> 8;
 806         /* Number of roots == degree of smu hence <= cap */
 807         if (roots_nbr == host->pmecc_lmu[cap + 1] >> 1)
 808                 return err_nbr - 1;
 809
 810         /* Number of roots does not match the degree of smu
 811          * unable to correct error */
 812         return -1;
 813 }
 814
 815 static void pmecc_correct_data(struct mtd_info *mtd, uint8_t *buf, uint8_t *ecc,
 816                 int sector_num, int extra_bytes, int err_nbr)
 817 {
 818         struct nand_chip *nand_chip = mtd->priv;
 819         struct atmel_nand_host *host = nand_chip->priv;
 820         int i = 0;
 821         int byte_pos, bit_pos, sector_size, pos;
 822         uint32_t tmp;
 823         uint8_t err_byte;
 824
 825         sector_size = host->pmecc_sector_size;
 826
 827         while (err_nbr) {
 828                 tmp = pmerrloc_readl_el_relaxed(host->pmerrloc_base, i) - 1;
 829                 byte_pos = tmp / 8;
 830                 bit_pos  = tmp % 8;
 831
 832                 if (byte_pos >= (sector_size + extra_bytes))
 833                         BUG();  /* should never happen */
 834
 835                 if (byte_pos < sector_size) {
 836                         err_byte = *(buf + byte_pos);
 837                         *(buf + byte_pos) ^= (1 << bit_pos);
 838
 839                         pos = sector_num * host->pmecc_sector_size + byte_pos;
 840                         dev_info(host->dev, "Bit flip in data area, byte_pos: %d, bit_pos: %d, 0x%02x -> 0x%02x\n",
 841                                 pos, bit_pos, err_byte, *(buf + byte_pos));
 842                 } else {
 843                         /* Bit flip in OOB area */
 844                         tmp = sector_num * host->pmecc_bytes_per_sector
 845                                         + (byte_pos - sector_size);
 846                         err_byte = ecc[tmp];
 847                         ecc[tmp] ^= (1 << bit_pos);
 848
 849                         pos = tmp + nand_chip->ecc.layout->eccpos[0];
 850                         dev_info(host->dev, "Bit flip in OOB, oob_byte_pos: %d, bit_pos: %d, 0x%02x -> 0x%02x\n",
 851                                 pos, bit_pos, err_byte, ecc[tmp]);
 852                 }
 853
 854                 i++;
 855                 err_nbr--;
 856         }
 857
 858         return;
 859 }
 860
 861 static int pmecc_correction(struct mtd_info *mtd, u32 pmecc_stat, uint8_t *buf,
 862         u8 *ecc)
 863 {
 864         struct nand_chip *nand_chip = mtd->priv;
 865         struct atmel_nand_host *host = nand_chip->priv;
 866         int i, err_nbr;
 867         uint8_t *buf_pos;
 868         int total_err = 0;
 869
 870         for (i = 0; i < nand_chip->ecc.total; i++)
 871                 if (ecc[i] != 0xff)
 872                         goto normal_check;
 873         /* Erased page, return OK */
 874         return 0;
 875
 876 normal_check:
 877         for (i = 0; i < host->pmecc_sector_number; i++) {
 878                 err_nbr = 0;
 879                 if (pmecc_stat & 0x1) {
 880                         buf_pos = buf + i * host->pmecc_sector_size;
 881
 882                         pmecc_gen_syndrome(mtd, i);
 883                         pmecc_substitute(mtd);
 884                         pmecc_get_sigma(mtd);
 885
 886                         err_nbr = pmecc_err_location(mtd);
 887                         if (err_nbr == -1) {
 888                                 dev_err(host->dev, "PMECC: Too many errors\n");
 889                                 mtd->ecc_stats.failed++;
 890                                 return -EIO;
 891                         } else {
 892                                 pmecc_correct_data(mtd, buf_pos, ecc, i,
 893                                         host->pmecc_bytes_per_sector, err_nbr);
 894                                 mtd->ecc_stats.corrected += err_nbr;
 895                                 total_err += err_nbr;
 896                         }
 897                 }
 898                 pmecc_stat >>= 1;
 899         }
 900
 901         return total_err;
 902 }
 903
 904 static void pmecc_enable(struct atmel_nand_host *host, int ecc_op)
 905 {
 906         u32 val;
 907
 908         if (ecc_op != NAND_ECC_READ && ecc_op != NAND_ECC_WRITE) {
 909                 dev_err(host->dev, "atmel_nand: wrong pmecc operation type!");
 910                 return;
 911         }
 912
 913         pmecc_writel(host->ecc, CTRL, PMECC_CTRL_RST);
 914         pmecc_writel(host->ecc, CTRL, PMECC_CTRL_DISABLE);
 915         val = pmecc_readl_relaxed(host->ecc, CFG);
 916
 917         if (ecc_op == NAND_ECC_READ)
 918                 pmecc_writel(host->ecc, CFG, (val & ~PMECC_CFG_WRITE_OP)
 919                         | PMECC_CFG_AUTO_ENABLE);
 920         else
 921                 pmecc_writel(host->ecc, CFG, (val | PMECC_CFG_WRITE_OP)
 922                         & ~PMECC_CFG_AUTO_ENABLE);
 923
 924         pmecc_writel(host->ecc, CTRL, PMECC_CTRL_ENABLE);
 925         pmecc_writel(host->ecc, CTRL, PMECC_CTRL_DATA);
 926 }
 927
 928 static int atmel_nand_pmecc_read_page(struct mtd_info *mtd,
 929         struct nand_chip *chip, uint8_t *buf, int oob_required, int page)
 930 {
 931         struct atmel_nand_host *host = chip->priv;
 932         int eccsize = chip->ecc.size * chip->ecc.steps;
 933         uint8_t *oob = chip->oob_poi;
 934         uint32_t *eccpos = chip->ecc.layout->eccpos;
 935         uint32_t stat;
 936         unsigned long end_time;
 937         int bitflips = 0;
 938
 939         if (!host->nfc || !host->nfc->use_nfc_sram)
 940                 pmecc_enable(host, NAND_ECC_READ);
 941
 942         chip->read_buf(mtd, buf, eccsize);
 943         chip->read_buf(mtd, oob, mtd->oobsize);
 944
 945         end_time = jiffies + msecs_to_jiffies(PMECC_MAX_TIMEOUT_MS);
 946         while ((pmecc_readl_relaxed(host->ecc, SR) & PMECC_SR_BUSY)) {
 947                 if (unlikely(time_after(jiffies, end_time))) {
 948                         dev_err(host->dev, "PMECC: Timeout to get error status.\n");
 949                         return -EIO;
 950                 }
 951                 cpu_relax();
 952         }
 953
 954         stat = pmecc_readl_relaxed(host->ecc, ISR);
 955         if (stat != 0) {
 956                 bitflips = pmecc_correction(mtd, stat, buf, &oob[eccpos[0]]);
 957                 if (bitflips < 0)
 958                         /* uncorrectable errors */
 959                         return 0;
 960         }
 961
 962         return bitflips;
 963 }
 964
 965 static int atmel_nand_pmecc_write_page(struct mtd_info *mtd,
 966                 struct nand_chip *chip, const uint8_t *buf, int oob_required)
 967 {
 968         struct atmel_nand_host *host = chip->priv;
 969         uint32_t *eccpos = chip->ecc.layout->eccpos;
 970         int i, j;
 971         unsigned long end_time;
 972
 973         if (!host->nfc || !host->nfc->write_by_sram) {
 974                 pmecc_enable(host, NAND_ECC_WRITE);
 975                 chip->write_buf(mtd, (u8 *)buf, mtd->writesize);
 976         }
 977
 978         end_time = jiffies + msecs_to_jiffies(PMECC_MAX_TIMEOUT_MS);
 979         while ((pmecc_readl_relaxed(host->ecc, SR) & PMECC_SR_BUSY)) {
 980                 if (unlikely(time_after(jiffies, end_time))) {
 981                         dev_err(host->dev, "PMECC: Timeout to get ECC value.\n");
 982                         return -EIO;
 983                 }
 984                 cpu_relax();
 985         }
 986
 987         for (i = 0; i < host->pmecc_sector_number; i++) {
 988                 for (j = 0; j < host->pmecc_bytes_per_sector; j++) {
 989                         int pos;
 990
 991                         pos = i * host->pmecc_bytes_per_sector + j;
 992                         chip->oob_poi[eccpos[pos]] =
 993                                 pmecc_readb_ecc_relaxed(host->ecc, i, j);
 994                 }
 995         }
 996         chip->write_buf(mtd, chip->oob_poi, mtd->oobsize);
 997
 998         return 0;
 999 }
1000
1001 static void atmel_pmecc_core_init(struct mtd_info *mtd)
1002 {
1003         struct nand_chip *nand_chip = mtd->priv;
1004         struct atmel_nand_host *host = nand_chip->priv;
1005         uint32_t val = 0;
1006         struct nand_ecclayout *ecc_layout;
1007
1008         pmecc_writel(host->ecc, CTRL, PMECC_CTRL_RST);
1009         pmecc_writel(host->ecc, CTRL, PMECC_CTRL_DISABLE);
1010
1011         switch (host->pmecc_corr_cap) {
1012         case 2:
1013                 val = PMECC_CFG_BCH_ERR2;
1014                 break;
1015         case 4:
1016                 val = PMECC_CFG_BCH_ERR4;
1017                 break;
1018         case 8:
1019                 val = PMECC_CFG_BCH_ERR8;
1020                 break;
1021         case 12:
1022                 val = PMECC_CFG_BCH_ERR12;
1023                 break;
1024         case 24:
1025                 val = PMECC_CFG_BCH_ERR24;
1026                 break;
1027         }
1028
1029         if (host->pmecc_sector_size == 512)
1030                 val |= PMECC_CFG_SECTOR512;
1031         else if (host->pmecc_sector_size == 1024)
1032                 val |= PMECC_CFG_SECTOR1024;
1033
1034         switch (host->pmecc_sector_number) {
1035         case 1:
1036                 val |= PMECC_CFG_PAGE_1SECTOR;
1037                 break;
1038         case 2:
1039                 val |= PMECC_CFG_PAGE_2SECTORS;
1040                 break;
1041         case 4:
1042                 val |= PMECC_CFG_PAGE_4SECTORS;
1043                 break;
1044         case 8:
1045                 val |= PMECC_CFG_PAGE_8SECTORS;
1046                 break;
1047         }
1048
1049         val |= (PMECC_CFG_READ_OP | PMECC_CFG_SPARE_DISABLE
1050                 | PMECC_CFG_AUTO_DISABLE);
1051         pmecc_writel(host->ecc, CFG, val);
1052
1053         ecc_layout = nand_chip->ecc.layout;
1054         pmecc_writel(host->ecc, SAREA, mtd->oobsize - 1);
1055         pmecc_writel(host->ecc, SADDR, ecc_layout->eccpos[0]);
1056         pmecc_writel(host->ecc, EADDR,
1057                         ecc_layout->eccpos[ecc_layout->eccbytes - 1]);
1058         /* See datasheet about PMECC Clock Control Register */
1059         pmecc_writel(host->ecc, CLK, 2);
1060         pmecc_writel(host->ecc, IDR, 0xff);
1061         pmecc_writel(host->ecc, CTRL, PMECC_CTRL_ENABLE);
1062 }
1063
1064 /*
1065  * Get minimum ecc requirements from NAND.
1066  * If pmecc-cap, pmecc-sector-size in DTS are not specified, this function
1067  * will set them according to minimum ecc requirement. Otherwise, use the
1068  * value in DTS file.
1069  * return 0 if success. otherwise return error code.
1070  */
1071 static int pmecc_choose_ecc(struct atmel_nand_host *host,
1072                 int *cap, int *sector_size)
1073 {
1074         /* Get minimum ECC requirements */
1075         if (host->nand_chip.ecc_strength_ds) {
1076                 *cap = host->nand_chip.ecc_strength_ds;
1077                 *sector_size = host->nand_chip.ecc_step_ds;
1078                 dev_info(host->dev, "minimum ECC: %d bits in %d bytes\n",
1079                                 *cap, *sector_size);
1080         } else {
1081                 *cap = 2;
1082                 *sector_size = 512;
1083                 dev_info(host->dev, "can't detect min. ECC, assume 2 bits in 512 bytes\n");
1084         }
1085
1086         /* If device tree doesn't specify, use NAND's minimum ECC parameters */
1087         if (host->pmecc_corr_cap == 0) {
1088                 /* use the most fitable ecc bits (the near bigger one ) */
1089                 if (*cap <= 2)
1090                         host->pmecc_corr_cap = 2;
1091                 else if (*cap <= 4)
1092                         host->pmecc_corr_cap = 4;
1093                 else if (*cap <= 8)
1094                         host->pmecc_corr_cap = 8;
1095                 else if (*cap <= 12)
1096                         host->pmecc_corr_cap = 12;
1097                 else if (*cap <= 24)
1098                         host->pmecc_corr_cap = 24;
1099                 else
1100                         return -EINVAL;
1101         }
1102         if (host->pmecc_sector_size == 0) {
1103                 /* use the most fitable sector size (the near smaller one ) */
1104                 if (*sector_size >= 1024)
1105                         host->pmecc_sector_size = 1024;
1106                 else if (*sector_size >= 512)
1107                         host->pmecc_sector_size = 512;
1108                 else
1109                         return -EINVAL;
1110         }
1111         return 0;
1112 }
1113
1114 static int atmel_pmecc_nand_init_params(struct platform_device *pdev,
1115                                          struct atmel_nand_host *host)
1116 {
1117         struct mtd_info *mtd = &host->mtd;
1118         struct nand_chip *nand_chip = &host->nand_chip;
1119         struct resource *regs, *regs_pmerr, *regs_rom;
1120         int cap, sector_size, err_no;
1121
1122         err_no = pmecc_choose_ecc(host, &cap, &sector_size);
1123         if (err_no) {
1124                 dev_err(host->dev, "The NAND flash's ECC requirement are not support!");
1125                 return err_no;
1126         }
1127
1128         if (cap > host->pmecc_corr_cap ||
1129                         sector_size != host->pmecc_sector_size)
1130                 dev_info(host->dev, "WARNING: Be Caution! Using different PMECC parameters from Nand ONFI ECC reqirement.\n");
1131
1132         cap = host->pmecc_corr_cap;
1133         sector_size = host->pmecc_sector_size;
1134         host->pmecc_lookup_table_offset = (sector_size == 512) ?
1135                         host->pmecc_lookup_table_offset_512 :
1136                         host->pmecc_lookup_table_offset_1024;
1137
1138         dev_info(host->dev, "Initialize PMECC params, cap: %d, sector: %d\n",
1139                  cap, sector_size);
1140
1141         regs = platform_get_resource(pdev, IORESOURCE_MEM, 1);
1142         if (!regs) {
1143                 dev_warn(host->dev,
1144                         "Can't get I/O resource regs for PMECC controller, rolling back on software ECC\n");
1145                 nand_chip->ecc.mode = NAND_ECC_SOFT;
1146                 return 0;
1147         }
1148
1149         host->ecc = devm_ioremap_resource(&pdev->dev, regs);
1150         if (IS_ERR(host->ecc)) {
1151                 dev_err(host->dev, "ioremap failed\n");
1152                 err_no = PTR_ERR(host->ecc);
1153                 goto err;
1154         }
1155
1156         regs_pmerr = platform_get_resource(pdev, IORESOURCE_MEM, 2);
1157         host->pmerrloc_base = devm_ioremap_resource(&pdev->dev, regs_pmerr);
1158         if (IS_ERR(host->pmerrloc_base)) {
1159                 dev_err(host->dev,
1160                         "Can not get I/O resource for PMECC ERRLOC controller!\n");
1161                 err_no = PTR_ERR(host->pmerrloc_base);
1162                 goto err;
1163         }
1164
1165         regs_rom = platform_get_resource(pdev, IORESOURCE_MEM, 3);
1166         host->pmecc_rom_base = devm_ioremap_resource(&pdev->dev, regs_rom);
1167         if (IS_ERR(host->pmecc_rom_base)) {
1168                 dev_err(host->dev, "Can not get I/O resource for ROM!\n");
1169                 err_no = PTR_ERR(host->pmecc_rom_base);
1170                 goto err;
1171         }
1172
1173         nand_chip->ecc.size = sector_size;
1174
1175         /* set ECC page size and oob layout */
1176         switch (mtd->writesize) {
1177         case 2048:
1178                 host->pmecc_degree = (sector_size == 512) ?
1179                         PMECC_GF_DIMENSION_13 : PMECC_GF_DIMENSION_14;
1180                 host->pmecc_cw_len = (1 << host->pmecc_degree) - 1;
1181                 host->pmecc_sector_number = mtd->writesize / sector_size;
1182                 host->pmecc_bytes_per_sector = pmecc_get_ecc_bytes(
1183                         cap, sector_size);
1184                 host->pmecc_alpha_to = pmecc_get_alpha_to(host);
1185                 host->pmecc_index_of = host->pmecc_rom_base +
1186                         host->pmecc_lookup_table_offset;
1187
1188                 nand_chip->ecc.steps = host->pmecc_sector_number;
1189                 nand_chip->ecc.strength = cap;
1190                 nand_chip->ecc.bytes = host->pmecc_bytes_per_sector;
1191                 nand_chip->ecc.total = host->pmecc_bytes_per_sector *
1192                                        host->pmecc_sector_number;
1193                 if (nand_chip->ecc.total > mtd->oobsize - 2) {
1194                         dev_err(host->dev, "No room for ECC bytes\n");
1195                         err_no = -EINVAL;
1196                         goto err;
1197                 }
1198                 pmecc_config_ecc_layout(&atmel_pmecc_oobinfo,
1199                                         mtd->oobsize,
1200                                         nand_chip->ecc.total);
1201
1202                 nand_chip->ecc.layout = &atmel_pmecc_oobinfo;
1203                 break;
1204         case 512:
1205         case 1024:
1206         case 4096:
1207                 /* TODO */
1208                 dev_warn(host->dev,
1209                         "Unsupported page size for PMECC, use Software ECC\n");
1210         default:
1211                 /* page size not handled by HW ECC */
1212                 /* switching back to soft ECC */
1213                 nand_chip->ecc.mode = NAND_ECC_SOFT;
1214                 return 0;
1215         }
1216
1217         /* Allocate data for PMECC computation */
1218         err_no = pmecc_data_alloc(host);
1219         if (err_no) {
1220                 dev_err(host->dev,
1221                                 "Cannot allocate memory for PMECC computation!\n");
1222                 goto err;
1223         }
1224
1225         nand_chip->options |= NAND_NO_SUBPAGE_WRITE;
1226         nand_chip->ecc.read_page = atmel_nand_pmecc_read_page;
1227         nand_chip->ecc.write_page = atmel_nand_pmecc_write_page;
1228
1229         atmel_pmecc_core_init(mtd);
1230
1231         return 0;
1232
1233 err:
1234         return err_no;
1235 }
1236
1237 /*
1238  * Calculate HW ECC
1239  *
1240  * function called after a write
1241  *
1242  * mtd:        MTD block structure
1243  * dat:        raw data (unused)
1244  * ecc_code:   buffer for ECC
1245  */
1246 static int atmel_nand_calculate(struct mtd_info *mtd,
1247                 const u_char *dat, unsigned char *ecc_code)
1248 {
1249         struct nand_chip *nand_chip = mtd->priv;
1250         struct atmel_nand_host *host = nand_chip->priv;
1251         unsigned int ecc_value;
1252
1253         /* get the first 2 ECC bytes */
1254         ecc_value = ecc_readl(host->ecc, PR);
1255
1256         ecc_code[0] = ecc_value & 0xFF;
1257         ecc_code[1] = (ecc_value >> 8) & 0xFF;
1258
1259         /* get the last 2 ECC bytes */
1260         ecc_value = ecc_readl(host->ecc, NPR) & ATMEL_ECC_NPARITY;
1261
1262         ecc_code[2] = ecc_value & 0xFF;
1263         ecc_code[3] = (ecc_value >> 8) & 0xFF;
1264
1265         return 0;
1266 }
1267
1268 /*
1269  * HW ECC read page function
1270  *
1271  * mtd:        mtd info structure
1272  * chip:       nand chip info structure
1273  * buf:        buffer to store read data
1274  * oob_required:    caller expects OOB data read to chip->oob_poi
1275  */
1276 static int atmel_nand_read_page(struct mtd_info *mtd, struct nand_chip *chip,
1277                                 uint8_t *buf, int oob_required, int page)
1278 {
1279         int eccsize = chip->ecc.size;
1280         int eccbytes = chip->ecc.bytes;
1281         uint32_t *eccpos = chip->ecc.layout->eccpos;
1282         uint8_t *p = buf;
1283         uint8_t *oob = chip->oob_poi;
1284         uint8_t *ecc_pos;
1285         int stat;
1286         unsigned int max_bitflips = 0;
1287
1288         /*
1289          * Errata: ALE is incorrectly wired up to the ECC controller
1290          * on the AP7000, so it will include the address cycles in the
1291          * ECC calculation.
1292          *
1293          * Workaround: Reset the parity registers before reading the
1294          * actual data.
1295          */
1296         struct atmel_nand_host *host = chip->priv;
1297         if (host->board.need_reset_workaround)
1298                 ecc_writel(host->ecc, CR, ATMEL_ECC_RST);
1299
1300         /* read the page */
1301         chip->read_buf(mtd, p, eccsize);
1302
1303         /* move to ECC position if needed */
1304         if (eccpos[0] != 0) {
1305                 /* This only works on large pages
1306                  * because the ECC controller waits for
1307                  * NAND_CMD_RNDOUTSTART after the
1308                  * NAND_CMD_RNDOUT.
1309                  * anyway, for small pages, the eccpos[0] == 0
1310                  */
1311                 chip->cmdfunc(mtd, NAND_CMD_RNDOUT,
1312                                 mtd->writesize + eccpos[0], -1);
1313         }
1314
1315         /* the ECC controller needs to read the ECC just after the data */
1316         ecc_pos = oob + eccpos[0];
1317         chip->read_buf(mtd, ecc_pos, eccbytes);
1318
1319         /* check if there's an error */
1320         stat = chip->ecc.correct(mtd, p, oob, NULL);
1321
1322         if (stat < 0) {
1323                 mtd->ecc_stats.failed++;
1324         } else {
1325                 mtd->ecc_stats.corrected += stat;
1326                 max_bitflips = max_t(unsigned int, max_bitflips, stat);
1327         }
1328
1329         /* get back to oob start (end of page) */
1330         chip->cmdfunc(mtd, NAND_CMD_RNDOUT, mtd->writesize, -1);
1331
1332         /* read the oob */
1333         chip->read_buf(mtd, oob, mtd->oobsize);
1334
1335         return max_bitflips;
1336 }
1337
1338 /*
1339  * HW ECC Correction
1340  *
1341  * function called after a read
1342  *
1343  * mtd:        MTD block structure
1344  * dat:        raw data read from the chip
1345  * read_ecc:   ECC from the chip (unused)
1346  * isnull:     unused
1347  *
1348  * Detect and correct a 1 bit error for a page
1349  */
1350 static int atmel_nand_correct(struct mtd_info *mtd, u_char *dat,
1351                 u_char *read_ecc, u_char *isnull)
1352 {
1353         struct nand_chip *nand_chip = mtd->priv;
1354         struct atmel_nand_host *host = nand_chip->priv;
1355         unsigned int ecc_status;
1356         unsigned int ecc_word, ecc_bit;
1357
1358         /* get the status from the Status Register */
1359         ecc_status = ecc_readl(host->ecc, SR);
1360
1361         /* if there's no error */
1362         if (likely(!(ecc_status & ATMEL_ECC_RECERR)))
1363                 return 0;
1364
1365         /* get error bit offset (4 bits) */
1366         ecc_bit = ecc_readl(host->ecc, PR) & ATMEL_ECC_BITADDR;
1367         /* get word address (12 bits) */
1368         ecc_word = ecc_readl(host->ecc, PR) & ATMEL_ECC_WORDADDR;
1369         ecc_word >>= 4;
1370
1371         /* if there are multiple errors */
1372         if (ecc_status & ATMEL_ECC_MULERR) {
1373                 /* check if it is a freshly erased block
1374                  * (filled with 0xff) */
1375                 if ((ecc_bit == ATMEL_ECC_BITADDR)
1376                                 && (ecc_word == (ATMEL_ECC_WORDADDR >> 4))) {
1377                         /* the block has just been erased, return OK */
1378                         return 0;
1379                 }
1380                 /* it doesn't seems to be a freshly
1381                  * erased block.
1382                  * We can't correct so many errors */
1383                 dev_dbg(host->dev, "atmel_nand : multiple errors detected."
1384                                 " Unable to correct.\n");
1385                 return -EIO;
1386         }
1387
1388         /* if there's a single bit error : we can correct it */
1389         if (ecc_status & ATMEL_ECC_ECCERR) {
1390                 /* there's nothing much to do here.
1391                  * the bit error is on the ECC itself.
1392                  */
1393                 dev_dbg(host->dev, "atmel_nand : one bit error on ECC code."
1394                                 " Nothing to correct\n");
1395                 return 0;
1396         }
1397
1398         dev_dbg(host->dev, "atmel_nand : one bit error on data."
1399                         " (word offset in the page :"
1400                         " 0x%x bit offset : 0x%x)\n",
1401                         ecc_word, ecc_bit);
1402         /* correct the error */
1403         if (nand_chip->options & NAND_BUSWIDTH_16) {
1404                 /* 16 bits words */
1405                 ((unsigned short *) dat)[ecc_word] ^= (1 << ecc_bit);
1406         } else {
1407                 /* 8 bits words */
1408                 dat[ecc_word] ^= (1 << ecc_bit);
1409         }
1410         dev_dbg(host->dev, "atmel_nand : error corrected\n");
1411         return 1;
1412 }
1413
1414 /*
1415  * Enable HW ECC : unused on most chips
1416  */
1417 static void atmel_nand_hwctl(struct mtd_info *mtd, int mode)
1418 {
1419         struct nand_chip *nand_chip = mtd->priv;
1420         struct atmel_nand_host *host = nand_chip->priv;
1421
1422         if (host->board.need_reset_workaround)
1423                 ecc_writel(host->ecc, CR, ATMEL_ECC_RST);
1424 }
1425
1426 static int atmel_of_init_port(struct atmel_nand_host *host,
1427                               struct device_node *np)
1428 {
1429         u32 val;
1430         u32 offset[2];
1431         int ecc_mode;
1432         struct atmel_nand_data *board = &host->board;
1433         enum of_gpio_flags flags = 0;
1434
1435         if (of_property_read_u32(np, "atmel,nand-addr-offset", &val) == 0) {
1436                 if (val >= 32) {
1437                         dev_err(host->dev, "invalid addr-offset %u\n", val);
1438                         return -EINVAL;
1439                 }
1440                 board->ale = val;
1441         }
1442
1443         if (of_property_read_u32(np, "atmel,nand-cmd-offset", &val) == 0) {
1444                 if (val >= 32) {
1445                         dev_err(host->dev, "invalid cmd-offset %u\n", val);
1446                         return -EINVAL;
1447                 }
1448                 board->cle = val;
1449         }
1450
1451         ecc_mode = of_get_nand_ecc_mode(np);
1452
1453         board->ecc_mode = ecc_mode < 0 ? NAND_ECC_SOFT : ecc_mode;
1454
1455         board->on_flash_bbt = of_get_nand_on_flash_bbt(np);
1456
1457         board->has_dma = of_property_read_bool(np, "atmel,nand-has-dma");
1458
1459         if (of_get_nand_bus_width(np) == 16)
1460                 board->bus_width_16 = 1;
1461
1462         board->rdy_pin = of_get_gpio_flags(np, 0, &flags);
1463         board->rdy_pin_active_low = (flags == OF_GPIO_ACTIVE_LOW);
1464
1465         board->enable_pin = of_get_gpio(np, 1);
1466         board->det_pin = of_get_gpio(np, 2);
1467
1468         host->has_pmecc = of_property_read_bool(np, "atmel,has-pmecc");
1469
1470         /* load the nfc driver if there is */
1471         of_platform_populate(np, NULL, NULL, host->dev);
1472
1473         if (!(board->ecc_mode == NAND_ECC_HW) || !host->has_pmecc)
1474                 return 0;       /* Not using PMECC */
1475
1476         /* use PMECC, get correction capability, sector size and lookup
1477          * table offset.
1478          * If correction bits and sector size are not specified, then find
1479          * them from NAND ONFI parameters.
1480          */
1481         if (of_property_read_u32(np, "atmel,pmecc-cap", &val) == 0) {
1482                 if ((val != 2) && (val != 4) && (val != 8) && (val != 12) &&
1483                                 (val != 24)) {
1484                         dev_err(host->dev,
1485                                 "Unsupported PMECC correction capability: %d; should be 2, 4, 8, 12 or 24\n",
1486                                 val);
1487                         return -EINVAL;
1488                 }
1489                 host->pmecc_corr_cap = (u8)val;
1490         }
1491
1492         if (of_property_read_u32(np, "atmel,pmecc-sector-size", &val) == 0) {
1493                 if ((val != 512) && (val != 1024)) {
1494                         dev_err(host->dev,
1495                                 "Unsupported PMECC sector size: %d; should be 512 or 1024 bytes\n",
1496                                 val);
1497                         return -EINVAL;
1498                 }
1499                 host->pmecc_sector_size = (u16)val;
1500         }
1501
1502         if (of_property_read_u32_array(np, "atmel,pmecc-lookup-table-offset",
1503                         offset, 2) != 0) {
1504                 dev_err(host->dev, "Cannot get PMECC lookup table offset\n");
1505                 return -EINVAL;
1506         }
1507         if (!offset[0] && !offset[1]) {
1508                 dev_err(host->dev, "Invalid PMECC lookup table offset\n");
1509                 return -EINVAL;
1510         }
1511         host->pmecc_lookup_table_offset_512 = offset[0];
1512         host->pmecc_lookup_table_offset_1024 = offset[1];
1513
1514         return 0;
1515 }
1516
1517 static int atmel_hw_nand_init_params(struct platform_device *pdev,
1518                                          struct atmel_nand_host *host)
1519 {
1520         struct mtd_info *mtd = &host->mtd;
1521         struct nand_chip *nand_chip = &host->nand_chip;
1522         struct resource         *regs;
1523
1524         regs = platform_get_resource(pdev, IORESOURCE_MEM, 1);
1525         if (!regs) {
1526                 dev_err(host->dev,
1527                         "Can't get I/O resource regs, use software ECC\n");
1528                 nand_chip->ecc.mode = NAND_ECC_SOFT;
1529                 return 0;
1530         }
1531
1532         host->ecc = devm_ioremap_resource(&pdev->dev, regs);
1533         if (IS_ERR(host->ecc)) {
1534                 dev_err(host->dev, "ioremap failed\n");
1535                 return PTR_ERR(host->ecc);
1536         }
1537
1538         /* ECC is calculated for the whole page (1 step) */
1539         nand_chip->ecc.size = mtd->writesize;
1540
1541         /* set ECC page size and oob layout */
1542         switch (mtd->writesize) {
1543         case 512:
1544                 nand_chip->ecc.layout = &atmel_oobinfo_small;
1545                 ecc_writel(host->ecc, MR, ATMEL_ECC_PAGESIZE_528);
1546                 break;
1547         case 1024:
1548                 nand_chip->ecc.layout = &atmel_oobinfo_large;
1549                 ecc_writel(host->ecc, MR, ATMEL_ECC_PAGESIZE_1056);
1550                 break;
1551         case 2048:
1552                 nand_chip->ecc.layout = &atmel_oobinfo_large;
1553                 ecc_writel(host->ecc, MR, ATMEL_ECC_PAGESIZE_2112);
1554                 break;
1555         case 4096:
1556                 nand_chip->ecc.layout = &atmel_oobinfo_large;
1557                 ecc_writel(host->ecc, MR, ATMEL_ECC_PAGESIZE_4224);
1558                 break;
1559         default:
1560                 /* page size not handled by HW ECC */
1561                 /* switching back to soft ECC */
1562                 nand_chip->ecc.mode = NAND_ECC_SOFT;
1563                 return 0;
1564         }
1565
1566         /* set up for HW ECC */
1567         nand_chip->ecc.calculate = atmel_nand_calculate;
1568         nand_chip->ecc.correct = atmel_nand_correct;
1569         nand_chip->ecc.hwctl = atmel_nand_hwctl;
1570         nand_chip->ecc.read_page = atmel_nand_read_page;
1571         nand_chip->ecc.bytes = 4;
1572         nand_chip->ecc.strength = 1;
1573
1574         return 0;
1575 }
1576
1577 static inline u32 nfc_read_status(struct atmel_nand_host *host)
1578 {
1579         u32 err_flags = NFC_SR_DTOE | NFC_SR_UNDEF | NFC_SR_AWB | NFC_SR_ASE;
1580         u32 nfc_status = nfc_readl(host->nfc->hsmc_regs, SR);
1581
1582         if (unlikely(nfc_status & err_flags)) {
1583                 if (nfc_status & NFC_SR_DTOE)
1584                         dev_err(host->dev, "NFC: Waiting Nand R/B Timeout Error\n");
1585                 else if (nfc_status & NFC_SR_UNDEF)
1586                         dev_err(host->dev, "NFC: Access Undefined Area Error\n");
1587                 else if (nfc_status & NFC_SR_AWB)
1588                         dev_err(host->dev, "NFC: Access memory While NFC is busy\n");
1589                 else if (nfc_status & NFC_SR_ASE)
1590                         dev_err(host->dev, "NFC: Access memory Size Error\n");
1591         }
1592
1593         return nfc_status;
1594 }
1595
1596 /* SMC interrupt service routine */
1597 static irqreturn_t hsmc_interrupt(int irq, void *dev_id)
1598 {
1599         struct atmel_nand_host *host = dev_id;
1600         u32 status, mask, pending;
1601         irqreturn_t ret = IRQ_NONE;
1602
1603         status = nfc_read_status(host);
1604         mask = nfc_readl(host->nfc->hsmc_regs, IMR);
1605         pending = status & mask;
1606
1607         if (pending & NFC_SR_XFR_DONE) {
1608                 complete(&host->nfc->comp_xfer_done);
1609                 nfc_writel(host->nfc->hsmc_regs, IDR, NFC_SR_XFR_DONE);
1610                 ret = IRQ_HANDLED;
1611         }
1612         if (pending & NFC_SR_RB_EDGE) {
1613                 complete(&host->nfc->comp_ready);
1614                 nfc_writel(host->nfc->hsmc_regs, IDR, NFC_SR_RB_EDGE);
1615                 ret = IRQ_HANDLED;
1616         }
1617         if (pending & NFC_SR_CMD_DONE) {
1618                 complete(&host->nfc->comp_cmd_done);
1619                 nfc_writel(host->nfc->hsmc_regs, IDR, NFC_SR_CMD_DONE);
1620                 ret = IRQ_HANDLED;
1621         }
1622
1623         return ret;
1624 }
1625
1626 /* NFC(Nand Flash Controller) related functions */
1627 static void nfc_prepare_interrupt(struct atmel_nand_host *host, u32 flag)
1628 {
1629         if (flag & NFC_SR_XFR_DONE)
1630                 init_completion(&host->nfc->comp_xfer_done);
1631
1632         if (flag & NFC_SR_RB_EDGE)
1633                 init_completion(&host->nfc->comp_ready);
1634
1635         if (flag & NFC_SR_CMD_DONE)
1636                 init_completion(&host->nfc->comp_cmd_done);
1637
1638         /* Enable interrupt that need to wait for */
1639         nfc_writel(host->nfc->hsmc_regs, IER, flag);
1640 }
1641
1642 static int nfc_wait_interrupt(struct atmel_nand_host *host, u32 flag)
1643 {
1644         int i, index = 0;
1645         struct completion *comp[3];     /* Support 3 interrupt completion */
1646
1647         if (flag & NFC_SR_XFR_DONE)
1648                 comp[index++] = &host->nfc->comp_xfer_done;
1649
1650         if (flag & NFC_SR_RB_EDGE)
1651                 comp[index++] = &host->nfc->comp_ready;
1652
1653         if (flag & NFC_SR_CMD_DONE)
1654                 comp[index++] = &host->nfc->comp_cmd_done;
1655
1656         if (index == 0) {
1657                 dev_err(host->dev, "Unkown interrupt flag: 0x%08x\n", flag);
1658                 return -EINVAL;
1659         }
1660
1661         for (i = 0; i < index; i++) {
1662                 if (wait_for_completion_timeout(comp[i],
1663                                 msecs_to_jiffies(NFC_TIME_OUT_MS)))
1664                         continue;       /* wait for next completion */
1665                 else
1666                         goto err_timeout;
1667         }
1668
1669         return 0;
1670
1671 err_timeout:
1672         dev_err(host->dev, "Time out to wait for interrupt: 0x%08x\n", flag);
1673         /* Disable the interrupt as it is not handled by interrupt handler */
1674         nfc_writel(host->nfc->hsmc_regs, IDR, flag);
1675         return -ETIMEDOUT;
1676 }
1677
1678 static int nfc_send_command(struct atmel_nand_host *host,
1679         unsigned int cmd, unsigned int addr, unsigned char cycle0)
1680 {
1681         unsigned long timeout;
1682         u32 flag = NFC_SR_CMD_DONE;
1683         flag |= cmd & NFCADDR_CMD_DATAEN ? NFC_SR_XFR_DONE : 0;
1684
1685         dev_dbg(host->dev,
1686                 "nfc_cmd: 0x%08x, addr1234: 0x%08x, cycle0: 0x%02x\n",
1687                 cmd, addr, cycle0);
1688
1689         timeout = jiffies + msecs_to_jiffies(NFC_TIME_OUT_MS);
1690         while (nfc_cmd_readl(NFCADDR_CMD_NFCBUSY, host->nfc->base_cmd_regs)
1691                         & NFCADDR_CMD_NFCBUSY) {
1692                 if (time_after(jiffies, timeout)) {
1693                         dev_err(host->dev,
1694                                 "Time out to wait CMD_NFCBUSY ready!\n");
1695                         return -ETIMEDOUT;
1696                 }
1697         }
1698
1699         nfc_prepare_interrupt(host, flag);
1700         nfc_writel(host->nfc->hsmc_regs, CYCLE0, cycle0);
1701         nfc_cmd_addr1234_writel(cmd, addr, host->nfc->base_cmd_regs);
1702         return nfc_wait_interrupt(host, flag);
1703 }
1704
1705 static int nfc_device_ready(struct mtd_info *mtd)
1706 {
1707         u32 status, mask;
1708         struct nand_chip *nand_chip = mtd->priv;
1709         struct atmel_nand_host *host = nand_chip->priv;
1710
1711         status = nfc_read_status(host);
1712         mask = nfc_readl(host->nfc->hsmc_regs, IMR);
1713
1714         /* The mask should be 0. If not we may lost interrupts */
1715         if (unlikely(mask & status))
1716                 dev_err(host->dev, "Lost the interrupt flags: 0x%08x\n",
1717                                 mask & status);
1718
1719         return status & NFC_SR_RB_EDGE;
1720 }
1721
1722 static void nfc_select_chip(struct mtd_info *mtd, int chip)
1723 {
1724         struct nand_chip *nand_chip = mtd->priv;
1725         struct atmel_nand_host *host = nand_chip->priv;
1726
1727         if (chip == -1)
1728                 nfc_writel(host->nfc->hsmc_regs, CTRL, NFC_CTRL_DISABLE);
1729         else
1730                 nfc_writel(host->nfc->hsmc_regs, CTRL, NFC_CTRL_ENABLE);
1731 }
1732
1733 static int nfc_make_addr(struct mtd_info *mtd, int command, int column,
1734                 int page_addr, unsigned int *addr1234, unsigned int *cycle0)
1735 {
1736         struct nand_chip *chip = mtd->priv;
1737
1738         int acycle = 0;
1739         unsigned char addr_bytes[8];
1740         int index = 0, bit_shift;
1741
1742         BUG_ON(addr1234 == NULL || cycle0 == NULL);
1743
1744         *cycle0 = 0;
1745         *addr1234 = 0;
1746
1747         if (column != -1) {
1748                 if (chip->options & NAND_BUSWIDTH_16 &&
1749                                 !nand_opcode_8bits(command))
1750                         column >>= 1;
1751                 addr_bytes[acycle++] = column & 0xff;
1752                 if (mtd->writesize > 512)
1753                         addr_bytes[acycle++] = (column >> 8) & 0xff;
1754         }
1755
1756         if (page_addr != -1) {
1757                 addr_bytes[acycle++] = page_addr & 0xff;
1758                 addr_bytes[acycle++] = (page_addr >> 8) & 0xff;
1759                 if (chip->chipsize > (128 << 20))
1760                         addr_bytes[acycle++] = (page_addr >> 16) & 0xff;
1761         }
1762
1763         if (acycle > 4)
1764                 *cycle0 = addr_bytes[index++];
1765
1766         for (bit_shift = 0; index < acycle; bit_shift += 8)
1767                 *addr1234 += addr_bytes[index++] << bit_shift;
1768
1769         /* return acycle in cmd register */
1770         return acycle << NFCADDR_CMD_ACYCLE_BIT_POS;
1771 }
1772
1773 static void nfc_nand_command(struct mtd_info *mtd, unsigned int command,
1774                                 int column, int page_addr)
1775 {
1776         struct nand_chip *chip = mtd->priv;
1777         struct atmel_nand_host *host = chip->priv;
1778         unsigned long timeout;
1779         unsigned int nfc_addr_cmd = 0;
1780
1781         unsigned int cmd1 = command << NFCADDR_CMD_CMD1_BIT_POS;
1782
1783         /* Set default settings: no cmd2, no addr cycle. read from nand */
1784         unsigned int cmd2 = 0;
1785         unsigned int vcmd2 = 0;
1786         int acycle = NFCADDR_CMD_ACYCLE_NONE;
1787         int csid = NFCADDR_CMD_CSID_3;
1788         int dataen = NFCADDR_CMD_DATADIS;
1789         int nfcwr = NFCADDR_CMD_NFCRD;
1790         unsigned int addr1234 = 0;
1791         unsigned int cycle0 = 0;
1792         bool do_addr = true;
1793         host->nfc->data_in_sram = NULL;
1794
1795         dev_dbg(host->dev, "%s: cmd = 0x%02x, col = 0x%08x, page = 0x%08x\n",
1796              __func__, command, column, page_addr);
1797
1798         switch (command) {
1799         case NAND_CMD_RESET:
1800                 nfc_addr_cmd = cmd1 | acycle | csid | dataen | nfcwr;
1801                 nfc_send_command(host, nfc_addr_cmd, addr1234, cycle0);
1802                 udelay(chip->chip_delay);
1803
1804                 nfc_nand_command(mtd, NAND_CMD_STATUS, -1, -1);
1805                 timeout = jiffies + msecs_to_jiffies(NFC_TIME_OUT_MS);
1806                 while (!(chip->read_byte(mtd) & NAND_STATUS_READY)) {
1807                         if (time_after(jiffies, timeout)) {
1808                                 dev_err(host->dev,
1809                                         "Time out to wait status ready!\n");
1810                                 break;
1811                         }
1812                 }
1813                 return;
1814         case NAND_CMD_STATUS:
1815                 do_addr = false;
1816                 break;
1817         case NAND_CMD_PARAM:
1818         case NAND_CMD_READID:
1819                 do_addr = false;
1820                 acycle = NFCADDR_CMD_ACYCLE_1;
1821                 if (column != -1)
1822                         addr1234 = column;
1823                 break;
1824         case NAND_CMD_RNDOUT:
1825                 cmd2 = NAND_CMD_RNDOUTSTART << NFCADDR_CMD_CMD2_BIT_POS;
1826                 vcmd2 = NFCADDR_CMD_VCMD2;
1827                 break;
1828         case NAND_CMD_READ0:
1829         case NAND_CMD_READOOB:
1830                 if (command == NAND_CMD_READOOB) {
1831                         column += mtd->writesize;
1832                         command = NAND_CMD_READ0; /* only READ0 is valid */
1833                         cmd1 = command << NFCADDR_CMD_CMD1_BIT_POS;
1834                 }
1835                 if (host->nfc->use_nfc_sram) {
1836                         /* Enable Data transfer to sram */
1837                         dataen = NFCADDR_CMD_DATAEN;
1838
1839                         /* Need enable PMECC now, since NFC will transfer
1840                          * data in bus after sending nfc read command.
1841                          */
1842                         if (chip->ecc.mode == NAND_ECC_HW && host->has_pmecc)
1843                                 pmecc_enable(host, NAND_ECC_READ);
1844                 }
1845
1846                 cmd2 = NAND_CMD_READSTART << NFCADDR_CMD_CMD2_BIT_POS;
1847                 vcmd2 = NFCADDR_CMD_VCMD2;
1848                 break;
1849         /* For prgramming command, the cmd need set to write enable */
1850         case NAND_CMD_PAGEPROG:
1851         case NAND_CMD_SEQIN:
1852         case NAND_CMD_RNDIN:
1853                 nfcwr = NFCADDR_CMD_NFCWR;
1854                 if (host->nfc->will_write_sram && command == NAND_CMD_SEQIN)
1855                         dataen = NFCADDR_CMD_DATAEN;
1856                 break;
1857         default:
1858                 break;
1859         }
1860
1861         if (do_addr)
1862                 acycle = nfc_make_addr(mtd, command, column, page_addr,
1863                                 &addr1234, &cycle0);
1864
1865         nfc_addr_cmd = cmd1 | cmd2 | vcmd2 | acycle | csid | dataen | nfcwr;
1866         nfc_send_command(host, nfc_addr_cmd, addr1234, cycle0);
1867
1868         /*
1869          * Program and erase have their own busy handlers status, sequential
1870          * in, and deplete1 need no delay.
1871          */
1872         switch (command) {
1873         case NAND_CMD_CACHEDPROG:
1874         case NAND_CMD_PAGEPROG:
1875         case NAND_CMD_ERASE1:
1876         case NAND_CMD_ERASE2:
1877         case NAND_CMD_RNDIN:
1878         case NAND_CMD_STATUS:
1879         case NAND_CMD_RNDOUT:
1880         case NAND_CMD_SEQIN:
1881         case NAND_CMD_READID:
1882                 return;
1883
1884         case NAND_CMD_READ0:
1885                 if (dataen == NFCADDR_CMD_DATAEN) {
1886                         host->nfc->data_in_sram = host->nfc->sram_bank0 +
1887                                 nfc_get_sram_off(host);
1888                         return;
1889                 }
1890                 /* fall through */
1891         default:
1892                 nfc_prepare_interrupt(host, NFC_SR_RB_EDGE);
1893                 nfc_wait_interrupt(host, NFC_SR_RB_EDGE);
1894         }
1895 }
1896
1897 static int nfc_sram_write_page(struct mtd_info *mtd, struct nand_chip *chip,
1898                         uint32_t offset, int data_len, const uint8_t *buf,
1899                         int oob_required, int page, int cached, int raw)
1900 {
1901         int cfg, len;
1902         int status = 0;
1903         struct atmel_nand_host *host = chip->priv;
1904         void __iomem *sram = host->nfc->sram_bank0 + nfc_get_sram_off(host);
1905
1906         /* Subpage write is not supported */
1907         if (offset || (data_len < mtd->writesize))
1908                 return -EINVAL;
1909
1910         cfg = nfc_readl(host->nfc->hsmc_regs, CFG);
1911         len = mtd->writesize;
1912
1913         if (unlikely(raw)) {
1914                 len += mtd->oobsize;
1915                 nfc_writel(host->nfc->hsmc_regs, CFG, cfg | NFC_CFG_WSPARE);
1916         } else
1917                 nfc_writel(host->nfc->hsmc_regs, CFG, cfg & ~NFC_CFG_WSPARE);
1918
1919         /* Copy page data to sram that will write to nand via NFC */
1920         if (use_dma) {
1921                 if (atmel_nand_dma_op(mtd, (void *)buf, len, 0) != 0)
1922                         /* Fall back to use cpu copy */
1923                         memcpy32_toio(sram, buf, len);
1924         } else {
1925                 memcpy32_toio(sram, buf, len);
1926         }
1927
1928         if (chip->ecc.mode == NAND_ECC_HW && host->has_pmecc)
1929                 /*
1930                  * When use NFC sram, need set up PMECC before send
1931                  * NAND_CMD_SEQIN command. Since when the nand command
1932                  * is sent, nfc will do transfer from sram and nand.
1933                  */
1934                 pmecc_enable(host, NAND_ECC_WRITE);
1935
1936         host->nfc->will_write_sram = true;
1937         chip->cmdfunc(mtd, NAND_CMD_SEQIN, 0x00, page);
1938         host->nfc->will_write_sram = false;
1939
1940         if (likely(!raw))
1941                 /* Need to write ecc into oob */
1942                 status = chip->ecc.write_page(mtd, chip, buf, oob_required);
1943
1944         if (status < 0)
1945                 return status;
1946
1947         chip->cmdfunc(mtd, NAND_CMD_PAGEPROG, -1, -1);
1948         status = chip->waitfunc(mtd, chip);
1949
1950         if ((status & NAND_STATUS_FAIL) && (chip->errstat))
1951                 status = chip->errstat(mtd, chip, FL_WRITING, status, page);
1952
1953         if (status & NAND_STATUS_FAIL)
1954                 return -EIO;
1955
1956         return 0;
1957 }
1958
1959 static int nfc_sram_init(struct mtd_info *mtd)
1960 {
1961         struct nand_chip *chip = mtd->priv;
1962         struct atmel_nand_host *host = chip->priv;
1963         int res = 0;
1964
1965         /* Initialize the NFC CFG register */
1966         unsigned int cfg_nfc = 0;
1967
1968         /* set page size and oob layout */
1969         switch (mtd->writesize) {
1970         case 512:
1971                 cfg_nfc = NFC_CFG_PAGESIZE_512;
1972                 break;
1973         case 1024:
1974                 cfg_nfc = NFC_CFG_PAGESIZE_1024;
1975                 break;
1976         case 2048:
1977                 cfg_nfc = NFC_CFG_PAGESIZE_2048;
1978                 break;
1979         case 4096:
1980                 cfg_nfc = NFC_CFG_PAGESIZE_4096;
1981                 break;
1982         case 8192:
1983                 cfg_nfc = NFC_CFG_PAGESIZE_8192;
1984                 break;
1985         default:
1986                 dev_err(host->dev, "Unsupported page size for NFC.\n");
1987                 res = -ENXIO;
1988                 return res;
1989         }
1990
1991         /* oob bytes size = (NFCSPARESIZE + 1) * 4
1992          * Max support spare size is 512 bytes. */
1993         cfg_nfc |= (((mtd->oobsize / 4) - 1) << NFC_CFG_NFC_SPARESIZE_BIT_POS
1994                 & NFC_CFG_NFC_SPARESIZE);
1995         /* default set a max timeout */
1996         cfg_nfc |= NFC_CFG_RSPARE |
1997                         NFC_CFG_NFC_DTOCYC | NFC_CFG_NFC_DTOMUL;
1998
1999         nfc_writel(host->nfc->hsmc_regs, CFG, cfg_nfc);
2000
2001         host->nfc->will_write_sram = false;
2002         nfc_set_sram_bank(host, 0);
2003
2004         /* Use Write page with NFC SRAM only for PMECC or ECC NONE. */
2005         if (host->nfc->write_by_sram) {
2006                 if ((chip->ecc.mode == NAND_ECC_HW && host->has_pmecc) ||
2007                                 chip->ecc.mode == NAND_ECC_NONE)
2008                         chip->write_page = nfc_sram_write_page;
2009                 else
2010                         host->nfc->write_by_sram = false;
2011         }
2012
2013         dev_info(host->dev, "Using NFC Sram read %s\n",
2014                         host->nfc->write_by_sram ? "and write" : "");
2015         return 0;
2016 }
2017
2018 static struct platform_driver atmel_nand_nfc_driver;
2019 /*
2020  * Probe for the NAND device.
2021  */
2022 static int atmel_nand_probe(struct platform_device *pdev)
2023 {
2024         struct atmel_nand_host *host;
2025         struct mtd_info *mtd;
2026         struct nand_chip *nand_chip;
2027         struct resource *mem;
2028         struct mtd_part_parser_data ppdata = {};
2029         int res, irq;
2030
2031         /* Allocate memory for the device structure (and zero it) */
2032         host = devm_kzalloc(&pdev->dev, sizeof(*host), GFP_KERNEL);
2033         if (!host)
2034                 return -ENOMEM;
2035
2036         res = platform_driver_register(&atmel_nand_nfc_driver);
2037         if (res)
2038                 dev_err(&pdev->dev, "atmel_nand: can't register NFC driver\n");
2039
2040         mem = platform_get_resource(pdev, IORESOURCE_MEM, 0);
2041         host->io_base = devm_ioremap_resource(&pdev->dev, mem);
2042         if (IS_ERR(host->io_base)) {
2043                 dev_err(&pdev->dev, "atmel_nand: ioremap resource failed\n");
2044                 res = PTR_ERR(host->io_base);
2045                 goto err_nand_ioremap;
2046         }
2047         host->io_phys = (dma_addr_t)mem->start;
2048
2049         mtd = &host->mtd;
2050         nand_chip = &host->nand_chip;
2051         host->dev = &pdev->dev;
2052         if (IS_ENABLED(CONFIG_OF) && pdev->dev.of_node) {
2053                 /* Only when CONFIG_OF is enabled of_node can be parsed */
2054                 res = atmel_of_init_port(host, pdev->dev.of_node);
2055                 if (res)
2056                         goto err_nand_ioremap;
2057         } else {
2058                 memcpy(&host->board, dev_get_platdata(&pdev->dev),
2059                        sizeof(struct atmel_nand_data));
2060         }
2061
2062         nand_chip->priv = host;         /* link the private data structures */
2063         mtd->priv = nand_chip;
2064         mtd->owner = THIS_MODULE;
2065
2066         /* Set address of NAND IO lines */
2067         nand_chip->IO_ADDR_R = host->io_base;
2068         nand_chip->IO_ADDR_W = host->io_base;
2069
2070         if (nand_nfc.is_initialized) {
2071                 /* NFC driver is probed and initialized */
2072                 host->nfc = &nand_nfc;
2073
2074                 nand_chip->select_chip = nfc_select_chip;
2075                 nand_chip->dev_ready = nfc_device_ready;
2076                 nand_chip->cmdfunc = nfc_nand_command;
2077
2078                 /* Initialize the interrupt for NFC */
2079                 irq = platform_get_irq(pdev, 0);
2080                 if (irq < 0) {
2081                         dev_err(host->dev, "Cannot get HSMC irq!\n");
2082                         res = irq;
2083                         goto err_nand_ioremap;
2084                 }
2085
2086                 res = devm_request_irq(&pdev->dev, irq, hsmc_interrupt,
2087                                 0, "hsmc", host);
2088                 if (res) {
2089                         dev_err(&pdev->dev, "Unable to request HSMC irq %d\n",
2090                                 irq);
2091                         goto err_nand_ioremap;
2092                 }
2093         } else {
2094                 res = atmel_nand_set_enable_ready_pins(mtd);
2095                 if (res)
2096                         goto err_nand_ioremap;
2097
2098                 nand_chip->cmd_ctrl = atmel_nand_cmd_ctrl;
2099         }
2100
2101         nand_chip->ecc.mode = host->board.ecc_mode;
2102         nand_chip->chip_delay = 20;             /* 20us command delay time */
2103
2104         if (host->board.bus_width_16)   /* 16-bit bus width */
2105                 nand_chip->options |= NAND_BUSWIDTH_16;
2106
2107         nand_chip->read_buf = atmel_read_buf;
2108         nand_chip->write_buf = atmel_write_buf;
2109
2110         platform_set_drvdata(pdev, host);
2111         atmel_nand_enable(host);
2112
2113         if (gpio_is_valid(host->board.det_pin)) {
2114                 res = devm_gpio_request(&pdev->dev,
2115                                 host->board.det_pin, "nand_det");
2116                 if (res < 0) {
2117                         dev_err(&pdev->dev,
2118                                 "can't request det gpio %d\n",
2119                                 host->board.det_pin);
2120                         goto err_no_card;
2121                 }
2122
2123                 res = gpio_direction_input(host->board.det_pin);
2124                 if (res < 0) {
2125                         dev_err(&pdev->dev,
2126                                 "can't request input direction det gpio %d\n",
2127                                 host->board.det_pin);
2128                         goto err_no_card;
2129                 }
2130
2131                 if (gpio_get_value(host->board.det_pin)) {
2132                         dev_info(&pdev->dev, "No SmartMedia card inserted.\n");
2133                         res = -ENXIO;
2134                         goto err_no_card;
2135                 }
2136         }
2137
2138         if (host->board.on_flash_bbt || on_flash_bbt) {
2139                 dev_info(&pdev->dev, "Use On Flash BBT\n");
2140                 nand_chip->bbt_options |= NAND_BBT_USE_FLASH;
2141         }
2142
2143         if (!host->board.has_dma)
2144                 use_dma = 0;
2145
2146         if (use_dma) {
2147                 dma_cap_mask_t mask;
2148
2149                 dma_cap_zero(mask);
2150                 dma_cap_set(DMA_MEMCPY, mask);
2151                 host->dma_chan = dma_request_channel(mask, NULL, NULL);
2152                 if (!host->dma_chan) {
2153                         dev_err(host->dev, "Failed to request DMA channel\n");
2154                         use_dma = 0;
2155                 }
2156         }
2157         if (use_dma)
2158                 dev_info(host->dev, "Using %s for DMA transfers.\n",
2159                                         dma_chan_name(host->dma_chan));
2160         else
2161                 dev_info(host->dev, "No DMA support for NAND access.\n");
2162
2163         /* first scan to find the device and get the page size */
2164         if (nand_scan_ident(mtd, 1, NULL)) {
2165                 res = -ENXIO;
2166                 goto err_scan_ident;
2167         }
2168
2169         if (nand_chip->ecc.mode == NAND_ECC_HW) {
2170                 if (host->has_pmecc)
2171                         res = atmel_pmecc_nand_init_params(pdev, host);
2172                 else
2173                         res = atmel_hw_nand_init_params(pdev, host);
2174
2175                 if (res != 0)
2176                         goto err_hw_ecc;
2177         }
2178
2179         /* initialize the nfc configuration register */
2180         if (host->nfc && host->nfc->use_nfc_sram) {
2181                 res = nfc_sram_init(mtd);
2182                 if (res) {
2183                         host->nfc->use_nfc_sram = false;
2184                         dev_err(host->dev, "Disable use nfc sram for data transfer.\n");
2185                 }
2186         }
2187
2188         /* second phase scan */
2189         if (nand_scan_tail(mtd)) {
2190                 res = -ENXIO;
2191                 goto err_scan_tail;
2192         }
2193
2194         mtd->name = "atmel_nand";
2195         ppdata.of_node = pdev->dev.of_node;
2196         res = mtd_device_parse_register(mtd, NULL, &ppdata,
2197                         host->board.parts, host->board.num_parts);
2198         if (!res)
2199                 return res;
2200
2201 err_scan_tail:
2202         if (host->has_pmecc && host->nand_chip.ecc.mode == NAND_ECC_HW)
2203                 pmecc_writel(host->ecc, CTRL, PMECC_CTRL_DISABLE);
2204 err_hw_ecc:
2205 err_scan_ident:
2206 err_no_card:
2207         atmel_nand_disable(host);
2208         if (host->dma_chan)
2209                 dma_release_channel(host->dma_chan);
2210 err_nand_ioremap:
2211         return res;
2212 }
2213
2214 /*
2215  * Remove a NAND device.
2216  */
2217 static int atmel_nand_remove(struct platform_device *pdev)
2218 {
2219         struct atmel_nand_host *host = platform_get_drvdata(pdev);
2220         struct mtd_info *mtd = &host->mtd;
2221
2222         nand_release(mtd);
2223
2224         atmel_nand_disable(host);
2225
2226         if (host->has_pmecc && host->nand_chip.ecc.mode == NAND_ECC_HW) {
2227                 pmecc_writel(host->ecc, CTRL, PMECC_CTRL_DISABLE);
2228                 pmerrloc_writel(host->pmerrloc_base, ELDIS,
2229                                 PMERRLOC_DISABLE);
2230         }
2231
2232         if (host->dma_chan)
2233                 dma_release_channel(host->dma_chan);
2234
2235         platform_driver_unregister(&atmel_nand_nfc_driver);
2236
2237         return 0;
2238 }
2239
2240 static const struct of_device_id atmel_nand_dt_ids[] = {
2241         { .compatible = "atmel,at91rm9200-nand" },
2242         { /* sentinel */ }
2243 };
2244
2245 MODULE_DEVICE_TABLE(of, atmel_nand_dt_ids);
2246
2247 static int atmel_nand_nfc_probe(struct platform_device *pdev)
2248 {
2249         struct atmel_nfc *nfc = &nand_nfc;
2250         struct resource *nfc_cmd_regs, *nfc_hsmc_regs, *nfc_sram;
2251
2252         nfc_cmd_regs = platform_get_resource(pdev, IORESOURCE_MEM, 0);
2253         nfc->base_cmd_regs = devm_ioremap_resource(&pdev->dev, nfc_cmd_regs);
2254         if (IS_ERR(nfc->base_cmd_regs))
2255                 return PTR_ERR(nfc->base_cmd_regs);
2256
2257         nfc_hsmc_regs = platform_get_resource(pdev, IORESOURCE_MEM, 1);
2258         nfc->hsmc_regs = devm_ioremap_resource(&pdev->dev, nfc_hsmc_regs);
2259         if (IS_ERR(nfc->hsmc_regs))
2260                 return PTR_ERR(nfc->hsmc_regs);
2261
2262         nfc_sram = platform_get_resource(pdev, IORESOURCE_MEM, 2);
2263         if (nfc_sram) {
2264                 nfc->sram_bank0 = devm_ioremap_resource(&pdev->dev, nfc_sram);
2265                 if (IS_ERR(nfc->sram_bank0)) {
2266                         dev_warn(&pdev->dev, "Fail to ioremap the NFC sram with error: %ld. So disable NFC sram.\n",
2267                                         PTR_ERR(nfc->sram_bank0));
2268                 } else {
2269                         nfc->use_nfc_sram = true;
2270                         nfc->sram_bank0_phys = (dma_addr_t)nfc_sram->start;
2271
2272                         if (pdev->dev.of_node)
2273                                 nfc->write_by_sram = of_property_read_bool(
2274                                                 pdev->dev.of_node,
2275                                                 "atmel,write-by-sram");
2276                 }
2277         }
2278
2279         nfc_writel(nfc->hsmc_regs, IDR, 0xffffffff);
2280         nfc_readl(nfc->hsmc_regs, SR);  /* clear the NFC_SR */
2281
2282         nfc->is_initialized = true;
2283         dev_info(&pdev->dev, "NFC is probed.\n");
2284         return 0;
2285 }
2286
2287 static const struct of_device_id atmel_nand_nfc_match[] = {
2288         { .compatible = "atmel,sama5d3-nfc" },
2289         { /* sentinel */ }
2290 };
2291 MODULE_DEVICE_TABLE(of, atmel_nand_nfc_match);
2292
2293 static struct platform_driver atmel_nand_nfc_driver = {
2294         .driver = {
2295                 .name = "atmel_nand_nfc",
2296                 .owner = THIS_MODULE,
2297                 .of_match_table = of_match_ptr(atmel_nand_nfc_match),
2298         },
2299         .probe = atmel_nand_nfc_probe,
2300 };
2301
2302 static struct platform_driver atmel_nand_driver = {
2303         .probe          = atmel_nand_probe,
2304         .remove         = atmel_nand_remove,
2305         .driver         = {
2306                 .name   = "atmel_nand",
2307                 .owner  = THIS_MODULE,
2308                 .of_match_table = of_match_ptr(atmel_nand_dt_ids),
2309         },
2310 };
2311
2312 module_platform_driver(atmel_nand_driver);
2313
2314 MODULE_LICENSE("GPL");
2315 MODULE_AUTHOR("Rick Bronson");
2316 MODULE_DESCRIPTION("NAND/SmartMedia driver for AT91 / AVR32");
2317 MODULE_ALIAS("platform:atmel_nand");