Root/target/linux/ar71xx/files/drivers/mtd/nand/rb4xx_nand.c

1/*
2 * NAND flash driver for the MikroTik RouterBoard 4xx series
3 *
4 * Copyright (C) 2008 Gabor Juhos <juhosg@openwrt.org>
5 * Copyright (C) 2008 Imre Kaloz <kaloz@openwrt.org>
6 *
7 * This file was based on the driver for Linux 2.6.22 published by
8 * MikroTik for their RouterBoard 4xx series devices.
9 *
10 * This program is free software; you can redistribute it and/or modify it
11 * under the terms of the GNU General Public License version 2 as published
12 * by the Free Software Foundation.
13 */
14
15#include <linux/init.h>
16#include <linux/mtd/nand.h>
17#include <linux/mtd/mtd.h>
18#include <linux/mtd/partitions.h>
19#include <linux/platform_device.h>
20#include <linux/delay.h>
21#include <linux/io.h>
22#include <linux/gpio.h>
23
24#include <asm/mach-ar71xx/ar71xx.h>
25
26#define DRV_NAME "rb4xx-nand"
27#define DRV_VERSION "0.1.10"
28#define DRV_DESC "NAND flash driver for RouterBoard 4xx series"
29
30#define USE_FAST_READ 1
31#define USE_FAST_WRITE 1
32#undef RB4XX_NAND_DEBUG
33
34#ifdef RB4XX_NAND_DEBUG
35#define DBG(fmt, arg...) printk(KERN_DEBUG DRV_NAME ": " fmt, ## arg)
36#else
37#define DBG(fmt, arg...) do {} while (0)
38#endif
39
40#define RB4XX_NAND_GPIO_RDY 5
41#define RB4XX_FLASH_HZ 33333334
42#define RB4XX_NAND_HZ 33333334
43
44#define SPI_CTRL_FASTEST 0x40
45#define SPI_CTRL_SAFE 0x43 /* 25 MHz for AHB 200 MHz */
46#define SBIT_IOC_BASE SPI_IOC_CS1
47#define SBIT_IOC_DO_SHIFT 0
48#define SBIT_IOC_DO (1u << SBIT_IOC_DO_SHIFT)
49#define SBIT_IOC_DO2_SHIFT 18
50#define SBIT_IOC_DO2 (1u << SBIT_IOC_DO2_SHIFT)
51
52#define CPLD_CMD_WRITE_MULT 0x08 /* send cmd, n x send data, read data */
53#define CPLD_CMD_WRITE_CFG 0x09 /* send cmd, n x send cfg */
54#define CPLD_CMD_READ_MULT 0x0a /* send cmd, send idle, n x read data */
55#define CPLD_CMD_READ_FAST 0x0b /* send cmd, 4 x idle, n x read data */
56
57#define CFG_BIT_nCE 0x80
58#define CFG_BIT_CLE 0x40
59#define CFG_BIT_ALE 0x20
60#define CFG_BIT_FAN 0x10
61#define CFG_BIT_nLED4 0x08
62#define CFG_BIT_nLED3 0x04
63#define CFG_BIT_nLED2 0x02
64#define CFG_BIT_nLED1 0x01
65
66#define CFG_BIT_nLEDS \
67    (CFG_BIT_nLED1 | CFG_BIT_nLED2 | CFG_BIT_nLED3 | CFG_BIT_nLED4)
68
69struct rb4xx_nand_info {
70    struct nand_chip chip;
71    struct mtd_info mtd;
72};
73
74/*
75 * We need to use the OLD Yaffs-1 OOB layout, otherwise the RB bootloader
76 * will not be able to find the kernel that we load.
77 */
78static struct nand_ecclayout rb4xx_nand_ecclayout = {
79    .eccbytes = 6,
80    .eccpos = { 8, 9, 10, 13, 14, 15 },
81    .oobavail = 9,
82    .oobfree = { { 0, 4 }, { 6, 2 }, { 11, 2 }, { 4, 1 } }
83};
84
85static struct mtd_partition rb4xx_nand_partitions[] = {
86    {
87        .name = "booter",
88        .offset = 0,
89        .size = (256 * 1024),
90        .mask_flags = MTD_WRITEABLE,
91    },
92    {
93        .name = "kernel",
94        .offset = (256 * 1024),
95        .size = (4 * 1024 * 1024) - (256 * 1024),
96    },
97    {
98        .name = "rootfs",
99        .offset = MTDPART_OFS_NXTBLK,
100        .size = MTDPART_SIZ_FULL,
101    },
102};
103
104#if USE_FAST_READ
105#define SPI_NDATA_BASE 0x00800000
106static unsigned spi_ctrl_fread = SPI_CTRL_SAFE;
107static unsigned spi_ctrl_flash = SPI_CTRL_SAFE;
108extern unsigned mips_hpt_frequency;
109#endif
110
111static inline unsigned rb4xx_spi_rreg(unsigned r)
112{
113    return __raw_readl((void * __iomem)(KSEG1ADDR(AR71XX_SPI_BASE) + r));
114}
115
116static inline void rb4xx_spi_wreg(unsigned r, unsigned v)
117{
118    __raw_writel(v, (void * __iomem)(KSEG1ADDR(AR71XX_SPI_BASE) + r));
119}
120
121static inline void do_spi_clk(int bit)
122{
123    unsigned bval = SBIT_IOC_BASE | (bit & 1);
124
125    rb4xx_spi_wreg(SPI_REG_IOC, bval);
126    rb4xx_spi_wreg(SPI_REG_IOC, bval | SPI_IOC_CLK);
127}
128
129static void do_spi_byte(uint8_t byte)
130{
131    do_spi_clk(byte >> 7);
132    do_spi_clk(byte >> 6);
133    do_spi_clk(byte >> 5);
134    do_spi_clk(byte >> 4);
135    do_spi_clk(byte >> 3);
136    do_spi_clk(byte >> 2);
137    do_spi_clk(byte >> 1);
138    do_spi_clk(byte);
139
140    DBG("spi_byte sent 0x%02x got 0x%x\n",
141                    byte, rb4xx_spi_rreg(SPI_REG_RDS));
142}
143
144#if USE_FAST_WRITE
145static inline void do_spi_clk_fast(int bit1, int bit2)
146{
147    unsigned bval = (SBIT_IOC_BASE |
148            ((bit1 << SBIT_IOC_DO_SHIFT) & SBIT_IOC_DO) |
149            ((bit2 << SBIT_IOC_DO2_SHIFT) & SBIT_IOC_DO2));
150
151    rb4xx_spi_wreg(SPI_REG_IOC, bval);
152    rb4xx_spi_wreg(SPI_REG_IOC, bval | SPI_IOC_CLK);
153}
154
155static inline void do_spi_byte_fast(uint8_t byte)
156{
157    do_spi_clk_fast(byte >> 7, byte >> 6);
158    do_spi_clk_fast(byte >> 5, byte >> 4);
159    do_spi_clk_fast(byte >> 3, byte >> 2);
160    do_spi_clk_fast(byte >> 1, byte >> 0);
161
162    DBG("spi_byte_fast sent 0x%02x got 0x%x\n",
163                    byte, rb4xx_spi_rreg(SPI_REG_RDS));
164}
165#else
166static inline void do_spi_byte_fast(uint8_t byte)
167{
168    do_spi_byte(byte);
169}
170#endif /* USE_FAST_WRITE */
171
172static int do_spi_cmd(unsigned cmd, unsigned sendCnt, const uint8_t *sendData,
173        unsigned recvCnt, uint8_t *recvData,
174        const uint8_t *verifyData, int fastWrite)
175{
176    unsigned i;
177
178    DBG("SPI cmd 0x%x send %u recv %u\n", cmd, sendCnt, recvCnt);
179
180    rb4xx_spi_wreg(SPI_REG_FS, SPI_FS_GPIO);
181    rb4xx_spi_wreg(SPI_REG_CTRL, SPI_CTRL_FASTEST);
182
183    do_spi_byte(cmd);
184#if 0
185    if (cmd == CPLD_CMD_READ_FAST) {
186        do_spi_byte(0x80);
187        do_spi_byte(0);
188        do_spi_byte(0);
189    }
190#endif
191    for (i = 0; i < sendCnt; ++i) {
192        if (fastWrite)
193            do_spi_byte_fast(sendData[i]);
194        else
195            do_spi_byte(sendData[i]);
196    }
197
198    for (i = 0; i < recvCnt; ++i) {
199        if (fastWrite)
200            do_spi_byte_fast(0);
201        else
202            do_spi_byte(0);
203
204        if (recvData) {
205            recvData[i] = rb4xx_spi_rreg(SPI_REG_RDS) & 0xff;
206        } else if (verifyData) {
207            if (verifyData[i] != (rb4xx_spi_rreg(SPI_REG_RDS)
208                             & 0xff))
209                break;
210        }
211    }
212
213    rb4xx_spi_wreg(SPI_REG_IOC, SBIT_IOC_BASE | SPI_IOC_CS0);
214    rb4xx_spi_wreg(SPI_REG_CTRL, spi_ctrl_flash);
215    rb4xx_spi_wreg(SPI_REG_FS, 0);
216
217    return i == recvCnt;
218}
219
220static int got_write = 1;
221
222static void rb4xx_nand_write_data(const uint8_t *byte, unsigned cnt)
223{
224    do_spi_cmd(CPLD_CMD_WRITE_MULT, cnt, byte, 1, NULL, NULL, 1);
225    got_write = 1;
226}
227
228static void rb4xx_nand_write_byte(uint8_t byte)
229{
230    rb4xx_nand_write_data(&byte, 1);
231}
232
233#if USE_FAST_READ
234static uint8_t *rb4xx_nand_read_getaddr(unsigned cnt)
235{
236    static unsigned nboffset = 0x100000;
237    unsigned addr;
238
239    if (got_write) {
240        nboffset = (nboffset + 31) & ~31;
241        if (nboffset >= 0x100000) /* 1MB */
242            nboffset = 0;
243
244        got_write = 0;
245        rb4xx_spi_wreg(SPI_REG_FS, SPI_FS_GPIO);
246        rb4xx_spi_wreg(SPI_REG_CTRL, spi_ctrl_fread);
247        rb4xx_spi_wreg(SPI_REG_FS, 0);
248    }
249
250    addr = KSEG1ADDR(AR71XX_SPI_BASE + SPI_NDATA_BASE) + nboffset;
251    DBG("rb4xx_nand_read_getaddr 0x%x cnt 0x%x\n", addr, cnt);
252
253    nboffset += cnt;
254    return (uint8_t *)addr;
255}
256
257static void rb4xx_nand_read_data(uint8_t *buf, unsigned cnt)
258{
259    unsigned size32 = cnt & ~31;
260    unsigned remain = cnt & 31;
261
262    if (size32) {
263        uint8_t *addr = rb4xx_nand_read_getaddr(size32);
264        memcpy(buf, (void *)addr, size32);
265    }
266
267    if (remain) {
268        do_spi_cmd(CPLD_CMD_READ_MULT, 1, buf, remain,
269               buf + size32, NULL, 0);
270    }
271}
272
273static int rb4xx_nand_verify_data(const uint8_t *buf, unsigned cnt)
274{
275    unsigned size32 = cnt & ~31;
276    unsigned remain = cnt & 31;
277
278    if (size32) {
279        uint8_t *addr = rb4xx_nand_read_getaddr(size32);
280        if (memcmp(buf, (void *)addr, size32) != 0)
281            return 0;
282    }
283
284    if (remain) {
285        return do_spi_cmd(CPLD_CMD_READ_MULT, 1, buf, remain,
286                  NULL, buf + size32, 0);
287    }
288    return 1;
289}
290#else /* USE_FAST_READ */
291static void rb4xx_nand_read_data(uint8_t *buf, unsigned cnt)
292{
293    do_spi_cmd(CPLD_CMD_READ_MULT, 1, buf, cnt, buf, NULL, 0);
294}
295
296static int rb4xx_nand_verify_data(const uint8_t *buf, unsigned cnt)
297{
298    return do_spi_cmd(CPLD_CMD_READ_MULT, 1, buf, cnt, NULL, buf, 0);
299}
300#endif /* USE_FAST_READ */
301
302static void rb4xx_nand_write_cfg(uint8_t byte)
303{
304    do_spi_cmd(CPLD_CMD_WRITE_CFG, 1, &byte, 0, NULL, NULL, 0);
305    got_write = 1;
306}
307
308static int rb4xx_nand_dev_ready(struct mtd_info *mtd)
309{
310    return gpio_get_value(RB4XX_NAND_GPIO_RDY);
311}
312
313static void rb4xx_nand_cmd_ctrl(struct mtd_info *mtd, int cmd,
314                unsigned int ctrl)
315{
316    if (ctrl & NAND_CTRL_CHANGE) {
317        uint8_t cfg = CFG_BIT_nLEDS;
318
319        cfg |= (ctrl & NAND_CLE) ? CFG_BIT_CLE : 0;
320        cfg |= (ctrl & NAND_ALE) ? CFG_BIT_ALE : 0;
321        cfg |= (ctrl & NAND_NCE) ? 0 : CFG_BIT_nCE;
322
323        rb4xx_nand_write_cfg(cfg);
324    }
325
326    if (cmd != NAND_CMD_NONE)
327        rb4xx_nand_write_byte(cmd);
328}
329
330static uint8_t rb4xx_nand_read_byte(struct mtd_info *mtd)
331{
332    uint8_t byte = 0;
333
334    rb4xx_nand_read_data(&byte, 1);
335    return byte;
336}
337
338static void rb4xx_nand_write_buf(struct mtd_info *mtd, const uint8_t *buf,
339                int len)
340{
341    rb4xx_nand_write_data(buf, len);
342}
343
344static void rb4xx_nand_read_buf(struct mtd_info *mtd, uint8_t *buf,
345                int len)
346{
347    rb4xx_nand_read_data(buf, len);
348}
349
350static int rb4xx_nand_verify_buf(struct mtd_info *mtd, const uint8_t *buf,
351                int len)
352{
353    if (!rb4xx_nand_verify_data(buf, len))
354        return -EFAULT;
355
356    return 0;
357}
358
359static unsigned get_spi_ctrl(unsigned hz_max, const char *name)
360{
361    unsigned div;
362
363    div = (ar71xx_ahb_freq - 1) / (2 * hz_max);
364    /*
365     * CPU has a bug at (div == 0) - first bit read is random
366     */
367    if (div == 0)
368        ++div;
369
370    if (name) {
371        unsigned ahb_khz = (ar71xx_ahb_freq + 500) / 1000;
372        unsigned div_real = 2 * (div + 1);
373        printk(KERN_INFO "%s SPI clock %u kHz (AHB %u kHz / %u)\n",
374               name,
375               ahb_khz / div_real,
376               ahb_khz, div_real);
377    }
378
379    return SPI_CTRL_FASTEST + div;
380}
381
382static int __init rb4xx_nand_probe(struct platform_device *pdev)
383{
384    struct rb4xx_nand_info *info;
385    int ret;
386
387    printk(KERN_INFO DRV_DESC " version " DRV_VERSION "\n");
388
389    ret = gpio_request(RB4XX_NAND_GPIO_RDY, "NAND RDY");
390    if (ret) {
391        printk(KERN_ERR "rb4xx-nand: gpio request failed\n");
392        return ret;
393    }
394
395    ret = gpio_direction_input(RB4XX_NAND_GPIO_RDY);
396    if (ret) {
397        printk(KERN_ERR "rb4xx-nand: unable to set input mode "
398                    "on gpio%d\n", RB4XX_NAND_GPIO_RDY);
399        goto err_free_gpio;
400    }
401
402    info = kzalloc(sizeof(*info), GFP_KERNEL);
403    if (!info) {
404        printk(KERN_ERR "rb4xx-nand: no memory for private data\n");
405        ret = -ENOMEM;
406        goto err_free_gpio;
407    }
408
409#if USE_FAST_READ
410    spi_ctrl_fread = get_spi_ctrl(RB4XX_NAND_HZ, "NAND");
411#endif
412    spi_ctrl_flash = get_spi_ctrl(RB4XX_FLASH_HZ, "FLASH");
413
414    rb4xx_nand_write_cfg(CFG_BIT_nLEDS | CFG_BIT_nCE);
415
416    info->chip.priv = &info;
417    info->mtd.priv = &info->chip;
418    info->mtd.owner = THIS_MODULE;
419
420    info->chip.cmd_ctrl = rb4xx_nand_cmd_ctrl;
421    info->chip.dev_ready = rb4xx_nand_dev_ready;
422    info->chip.read_byte = rb4xx_nand_read_byte;
423    info->chip.write_buf = rb4xx_nand_write_buf;
424    info->chip.read_buf = rb4xx_nand_read_buf;
425    info->chip.verify_buf = rb4xx_nand_verify_buf;
426
427    info->chip.chip_delay = 25;
428    info->chip.ecc.mode = NAND_ECC_SOFT;
429    info->chip.options |= NAND_NO_AUTOINCR;
430
431    platform_set_drvdata(pdev, info);
432
433    ret = nand_scan_ident(&info->mtd, 1);
434    if (ret) {
435        ret = -ENXIO;
436        goto err_free_info;
437    }
438
439    if (info->mtd.writesize == 512)
440        info->chip.ecc.layout = &rb4xx_nand_ecclayout;
441
442    ret = nand_scan_tail(&info->mtd);
443    if (ret) {
444        return -ENXIO;
445        goto err_set_drvdata;
446    }
447
448#ifdef CONFIG_MTD_PARTITIONS
449    ret = add_mtd_partitions(&info->mtd, rb4xx_nand_partitions,
450                ARRAY_SIZE(rb4xx_nand_partitions));
451#else
452    ret = add_mtd_device(&info->mtd);
453#endif
454    if (ret)
455        goto err_release_nand;
456
457    return 0;
458
459err_release_nand:
460    nand_release(&info->mtd);
461err_set_drvdata:
462    platform_set_drvdata(pdev, NULL);
463err_free_info:
464    kfree(info);
465err_free_gpio:
466    gpio_free(RB4XX_NAND_GPIO_RDY);
467    return ret;
468}
469
470static int __devexit rb4xx_nand_remove(struct platform_device *pdev)
471{
472    struct rb4xx_nand_info *info = platform_get_drvdata(pdev);
473
474    nand_release(&info->mtd);
475    platform_set_drvdata(pdev, NULL);
476    kfree(info);
477
478    return 0;
479}
480
481static struct platform_driver rb4xx_nand_driver = {
482    .probe = rb4xx_nand_probe,
483    .remove = __devexit_p(rb4xx_nand_remove),
484    .driver = {
485        .name = DRV_NAME,
486        .owner = THIS_MODULE,
487    },
488};
489
490static int __init rb4xx_nand_init(void)
491{
492    return platform_driver_register(&rb4xx_nand_driver);
493}
494
495static void __exit rb4xx_nand_exit(void)
496{
497    platform_driver_unregister(&rb4xx_nand_driver);
498}
499
500module_init(rb4xx_nand_init);
501module_exit(rb4xx_nand_exit);
502
503MODULE_DESCRIPTION(DRV_DESC);
504MODULE_VERSION(DRV_VERSION);
505MODULE_AUTHOR("Gabor Juhos <juhosg@openwrt.org>");
506MODULE_AUTHOR("Imre Kaloz <kaloz@openwrt.org>");
507MODULE_LICENSE("GPL v2");
508

Archive Download this file

Branches:
backfire
history
master
release_2010-11-17
release_2010-12-14
release_2011-02-23
release_2011-05-28
release_2011-08-27
release_2011-11-13
release_2012-03-18
release_2012-04-09
release_2012-10-11
uclibc_nptl



interactive