Small coding style cleanup
[platform/kernel/u-boot.git] / common / cmd_nand.c
1 /*
2  * Driver for NAND support, Rick Bronson
3  * borrowed heavily from:
4  * (c) 1999 Machine Vision Holdings, Inc.
5  * (c) 1999, 2000 David Woodhouse <dwmw2@infradead.org>
6  *
7  * Added 16-bit nand support
8  * (C) 2004 Texas Instruments
9  */
10
11 #include <common.h>
12
13
14 #ifndef CFG_NAND_LEGACY
15 /*
16  *
17  * New NAND support
18  *
19  */
20 #include <common.h>
21
22 #if (CONFIG_COMMANDS & CFG_CMD_NAND)
23
24 #include <command.h>
25 #include <watchdog.h>
26 #include <malloc.h>
27 #include <asm/byteorder.h>
28
29 #ifdef CONFIG_SHOW_BOOT_PROGRESS
30 # include <status_led.h>
31 # define SHOW_BOOT_PROGRESS(arg)        show_boot_progress(arg)
32 #else
33 # define SHOW_BOOT_PROGRESS(arg)
34 #endif
35
36 #include <jffs2/jffs2.h>
37 #include <nand.h>
38
39 #if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
40
41 /* parition handling routines */
42 int mtdparts_init(void);
43 int id_parse(const char *id, const char **ret_id, u8 *dev_type, u8 *dev_num);
44 int find_dev_and_part(const char *id, struct mtd_device **dev,
45                 u8 *part_num, struct part_info **part);
46 #endif
47
48 extern nand_info_t nand_info[];       /* info for NAND chips */
49
50 static int nand_dump_oob(nand_info_t *nand, ulong off)
51 {
52         return 0;
53 }
54
55 static int nand_dump(nand_info_t *nand, ulong off)
56 {
57         int i;
58         u_char *buf, *p;
59
60         buf = malloc(nand->oobblock + nand->oobsize);
61         if (!buf) {
62                 puts("No memory for page buffer\n");
63                 return 1;
64         }
65         off &= ~(nand->oobblock - 1);
66         i = nand_read_raw(nand, buf, off, nand->oobblock, nand->oobsize);
67         if (i < 0) {
68                 printf("Error (%d) reading page %08x\n", i, off);
69                 free(buf);
70                 return 1;
71         }
72         printf("Page %08x dump:\n", off);
73         i = nand->oobblock >> 4; p = buf;
74         while (i--) {
75                 printf( "\t%02x %02x %02x %02x %02x %02x %02x %02x"
76                         "  %02x %02x %02x %02x %02x %02x %02x %02x\n",
77                         p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7],
78                         p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]);
79                 p += 16;
80         }
81         puts("OOB:\n");
82         i = nand->oobsize >> 3;
83         while (i--) {
84                 printf( "\t%02x %02x %02x %02x %02x %02x %02x %02x\n",
85                         p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7]);
86                 p += 8;
87         }
88         free(buf);
89
90         return 0;
91 }
92
93 /* ------------------------------------------------------------------------- */
94
95 static inline int str2long(char *p, ulong *num)
96 {
97         char *endptr;
98
99         *num = simple_strtoul(p, &endptr, 16);
100         return (*p != '\0' && *endptr == '\0') ? 1 : 0;
101 }
102
103 static int
104 arg_off_size(int argc, char *argv[], nand_info_t *nand, ulong *off, ulong *size)
105 {
106         int idx = nand_curr_device;
107 #if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
108         struct mtd_device *dev;
109         struct part_info *part;
110         u8 pnum;
111
112         if (argc >= 1 && !(str2long(argv[0], off))) {
113                 if ((mtdparts_init() == 0) &&
114                     (find_dev_and_part(argv[0], &dev, &pnum, &part) == 0)) {
115                         if (dev->id->type != MTD_DEV_TYPE_NAND) {
116                                 puts("not a NAND device\n");
117                                 return -1;
118                         }
119                         *off = part->offset;
120                         if (argc >= 2) {
121                                 if (!(str2long(argv[1], size))) {
122                                         printf("'%s' is not a number\n", argv[1]);
123                                         return -1;
124                                 }
125                                 if (*size > part->size)
126                                         *size = part->size;
127                         } else {
128                                 *size = part->size;
129                         }
130                         idx = dev->id->num;
131                         *nand = nand_info[idx];
132                         goto out;
133                 }
134         }
135 #endif
136
137         if (argc >= 1) {
138                 if (!(str2long(argv[0], off))) {
139                         printf("'%s' is not a number\n", argv[0]);
140                         return -1;
141                 }
142         } else {
143                 *off = 0;
144         }
145
146         if (argc >= 2) {
147                 if (!(str2long(argv[1], size))) {
148                         printf("'%s' is not a number\n", argv[1]);
149                         return -1;
150                 }
151         } else {
152                 *size = nand->size - *off;
153         }
154
155 #if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
156 out:
157 #endif
158         printf("device %d ", idx);
159         if (*size == nand->size)
160                 puts("whole chip\n");
161         else
162                 printf("offset 0x%x, size 0x%x\n", *off, *size);
163         return 0;
164 }
165
166 int do_nand(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
167 {
168         int i, dev, ret;
169         ulong addr, off, size;
170         char *cmd, *s;
171         nand_info_t *nand;
172         int quiet = 0;
173         const char *quiet_str = getenv("quiet");
174
175         /* at least two arguments please */
176         if (argc < 2)
177                 goto usage;
178
179         if (quiet_str)
180                 quiet = simple_strtoul(quiet_str, NULL, 0) != 0;
181
182         cmd = argv[1];
183
184         if (strcmp(cmd, "info") == 0) {
185
186                 putc('\n');
187                 for (i = 0; i < CFG_MAX_NAND_DEVICE; i++) {
188                         if (nand_info[i].name)
189                                 printf("Device %d: %s, sector size %lu KiB\n",
190                                         i, nand_info[i].name,
191                                         nand_info[i].erasesize >> 10);
192                 }
193                 return 0;
194         }
195
196         if (strcmp(cmd, "device") == 0) {
197
198                 if (argc < 3) {
199                         if ((nand_curr_device < 0) ||
200                             (nand_curr_device >= CFG_MAX_NAND_DEVICE))
201                                 puts("\nno devices available\n");
202                         else
203                                 printf("\nDevice %d: %s\n", nand_curr_device,
204                                         nand_info[nand_curr_device].name);
205                         return 0;
206                 }
207                 dev = (int)simple_strtoul(argv[2], NULL, 10);
208                 if (dev < 0 || dev >= CFG_MAX_NAND_DEVICE || !nand_info[dev].name) {
209                         puts("No such device\n");
210                         return 1;
211                 }
212                 printf("Device %d: %s", dev, nand_info[dev].name);
213                 puts("... is now current device\n");
214                 nand_curr_device = dev;
215
216 #ifdef CFG_NAND_SELECT_DEVICE
217                 /*
218                  * Select the chip in the board/cpu specific driver
219                  */
220                 board_nand_select_device(nand_info[dev].priv, dev);
221 #endif
222
223                 return 0;
224         }
225
226         if (strcmp(cmd, "bad") != 0 && strcmp(cmd, "erase") != 0 &&
227             strncmp(cmd, "dump", 4) != 0 &&
228             strncmp(cmd, "read", 4) != 0 && strncmp(cmd, "write", 5) != 0 &&
229             strcmp(cmd, "scrub") != 0 && strcmp(cmd, "markbad") != 0 &&
230             strcmp(cmd, "biterr") != 0 &&
231             strcmp(cmd, "lock") != 0 && strcmp(cmd, "unlock") != 0 )
232                 goto usage;
233
234         /* the following commands operate on the current device */
235         if (nand_curr_device < 0 || nand_curr_device >= CFG_MAX_NAND_DEVICE ||
236             !nand_info[nand_curr_device].name) {
237                 puts("\nno devices available\n");
238                 return 1;
239         }
240         nand = &nand_info[nand_curr_device];
241
242         if (strcmp(cmd, "bad") == 0) {
243                 printf("\nDevice %d bad blocks:\n", nand_curr_device);
244                 for (off = 0; off < nand->size; off += nand->erasesize)
245                         if (nand_block_isbad(nand, off))
246                                 printf("  %08x\n", off);
247                 return 0;
248         }
249
250         /*
251          * Syntax is:
252          *   0    1     2       3    4
253          *   nand erase [clean] [off size]
254          */
255         if (strcmp(cmd, "erase") == 0 || strcmp(cmd, "scrub") == 0) {
256                 nand_erase_options_t opts;
257                 /* "clean" at index 2 means request to write cleanmarker */
258                 int clean = argc > 2 && !strcmp("clean", argv[2]);
259                 int o = clean ? 3 : 2;
260                 int scrub = !strcmp(cmd, "scrub");
261
262                 printf("\nNAND %s: ", scrub ? "scrub" : "erase");
263                 /* skip first two or three arguments, look for offset and size */
264                 if (arg_off_size(argc - o, argv + o, nand, &off, &size) != 0)
265                         return 1;
266
267                 memset(&opts, 0, sizeof(opts));
268                 opts.offset = off;
269                 opts.length = size;
270                 opts.jffs2  = clean;
271                 opts.quiet  = quiet;
272
273                 if (scrub) {
274                         puts("Warning: "
275                              "scrub option will erase all factory set "
276                              "bad blocks!\n"
277                              "         "
278                              "There is no reliable way to recover them.\n"
279                              "         "
280                              "Use this command only for testing purposes "
281                              "if you\n"
282                              "         "
283                              "are sure of what you are doing!\n"
284                              "\nReally scrub this NAND flash? <y/N>\n");
285
286                         if (getc() == 'y' && getc() == '\r') {
287                                 opts.scrub = 1;
288                         } else {
289                                 puts("scrub aborted\n");
290                                 return -1;
291                         }
292                 }
293                 ret = nand_erase_opts(nand, &opts);
294                 printf("%s\n", ret ? "ERROR" : "OK");
295
296                 return ret == 0 ? 0 : 1;
297         }
298
299         if (strncmp(cmd, "dump", 4) == 0) {
300                 if (argc < 3)
301                         goto usage;
302
303                 s = strchr(cmd, '.');
304                 off = (int)simple_strtoul(argv[2], NULL, 16);
305
306                 if (s != NULL && strcmp(s, ".oob") == 0)
307                         ret = nand_dump_oob(nand, off);
308                 else
309                         ret = nand_dump(nand, off);
310
311                 return ret == 0 ? 1 : 0;
312
313         }
314
315         /* read write */
316         if (strncmp(cmd, "read", 4) == 0 || strncmp(cmd, "write", 5) == 0) {
317                 int read;
318
319                 if (argc < 4)
320                         goto usage;
321
322                 addr = (ulong)simple_strtoul(argv[2], NULL, 16);
323
324                 read = strncmp(cmd, "read", 4) == 0; /* 1 = read, 0 = write */
325                 printf("\nNAND %s: ", read ? "read" : "write");
326                 if (arg_off_size(argc - 3, argv + 3, nand, &off, &size) != 0)
327                         return 1;
328
329                 s = strchr(cmd, '.');
330                 if (s != NULL &&
331                     (!strcmp(s, ".jffs2") || !strcmp(s, ".e") || !strcmp(s, ".i"))) {
332                         if (read) {
333                                 /* read */
334                                 nand_read_options_t opts;
335                                 memset(&opts, 0, sizeof(opts));
336                                 opts.buffer     = (u_char*) addr;
337                                 opts.length     = size;
338                                 opts.offset     = off;
339                                 opts.quiet      = quiet;
340                                 ret = nand_read_opts(nand, &opts);
341                         } else {
342                                 /* write */
343                                 nand_write_options_t opts;
344                                 memset(&opts, 0, sizeof(opts));
345                                 opts.buffer     = (u_char*) addr;
346                                 opts.length     = size;
347                                 opts.offset     = off;
348                                 /* opts.forcejffs2 = 1; */
349                                 opts.pad        = 1;
350                                 opts.blockalign = 1;
351                                 opts.quiet      = quiet;
352                                 ret = nand_write_opts(nand, &opts);
353                         }
354                 } else {
355                         if (read)
356                                 ret = nand_read(nand, off, &size, (u_char *)addr);
357                         else
358                                 ret = nand_write(nand, off, &size, (u_char *)addr);
359                 }
360
361                 printf(" %d bytes %s: %s\n", size,
362                        read ? "read" : "written", ret ? "ERROR" : "OK");
363
364                 return ret == 0 ? 0 : 1;
365         }
366
367         if (strcmp(cmd, "markbad") == 0) {
368                 addr = (ulong)simple_strtoul(argv[2], NULL, 16);
369
370                 int ret = nand->block_markbad(nand, addr);
371                 if (ret == 0) {
372                         printf("block 0x%08lx successfully marked as bad\n",
373                                (ulong) addr);
374                         return 0;
375                 } else {
376                         printf("block 0x%08lx NOT marked as bad! ERROR %d\n",
377                                (ulong) addr, ret);
378                 }
379                 return 1;
380         }
381         if (strcmp(cmd, "biterr") == 0) {
382                 /* todo */
383                 return 1;
384         }
385
386         if (strcmp(cmd, "lock") == 0) {
387                 int tight  = 0;
388                 int status = 0;
389                 if (argc == 3) {
390                         if (!strcmp("tight", argv[2]))
391                                 tight = 1;
392                         if (!strcmp("status", argv[2]))
393                                 status = 1;
394                 }
395
396                 if (status) {
397                         ulong block_start = 0;
398                         ulong off;
399                         int last_status = -1;
400
401                         struct nand_chip *nand_chip = nand->priv;
402                         /* check the WP bit */
403                         nand_chip->cmdfunc (nand, NAND_CMD_STATUS, -1, -1);
404                         printf("device is %swrite protected\n",
405                                (nand_chip->read_byte(nand) & 0x80 ?
406                                 "NOT " : "" ) );
407
408                         for (off = 0; off < nand->size; off += nand->oobblock) {
409                                 int s = nand_get_lock_status(nand, off);
410
411                                 /* print message only if status has changed
412                                  * or at end of chip
413                                  */
414                                 if (off == nand->size - nand->oobblock
415                                     || (s != last_status && off != 0))  {
416
417                                         printf("%08x - %08x: %8d pages %s%s%s\n",
418                                                block_start,
419                                                off-1,
420                                                (off-block_start)/nand->oobblock,
421                                                ((last_status & NAND_LOCK_STATUS_TIGHT) ? "TIGHT " : ""),
422                                                ((last_status & NAND_LOCK_STATUS_LOCK) ? "LOCK " : ""),
423                                                ((last_status & NAND_LOCK_STATUS_UNLOCK) ? "UNLOCK " : ""));
424                                 }
425
426                                 last_status = s;
427                        }
428                 } else {
429                         if (!nand_lock(nand, tight)) {
430                                 puts("NAND flash successfully locked\n");
431                         } else {
432                                 puts("Error locking NAND flash\n");
433                                 return 1;
434                         }
435                 }
436                 return 0;
437         }
438
439         if (strcmp(cmd, "unlock") == 0) {
440                 if (arg_off_size(argc - 2, argv + 2, nand, &off, &size) < 0)
441                         return 1;
442
443                 if (!nand_unlock(nand, off, size)) {
444                         puts("NAND flash successfully unlocked\n");
445                 } else {
446                         puts("Error unlocking NAND flash, "
447                              "write and erase will probably fail\n");
448                         return 1;
449                 }
450                 return 0;
451         }
452
453 usage:
454         printf("Usage:\n%s\n", cmdtp->usage);
455         return 1;
456 }
457
458 U_BOOT_CMD(nand, 5, 1, do_nand,
459         "nand    - NAND sub-system\n",
460         "info                  - show available NAND devices\n"
461         "nand device [dev]     - show or set current device\n"
462         "nand read[.jffs2]     - addr off|partition size\n"
463         "nand write[.jffs2]    - addr off|partiton size - read/write `size' bytes starting\n"
464         "    at offset `off' to/from memory address `addr'\n"
465         "nand erase [clean] [off size] - erase `size' bytes from\n"
466         "    offset `off' (entire device if not specified)\n"
467         "nand bad - show bad blocks\n"
468         "nand dump[.oob] off - dump page\n"
469         "nand scrub - really clean NAND erasing bad blocks (UNSAFE)\n"
470         "nand markbad off - mark bad block at offset (UNSAFE)\n"
471         "nand biterr off - make a bit error at offset (UNSAFE)\n"
472         "nand lock [tight] [status] - bring nand to lock state or display locked pages\n"
473         "nand unlock [offset] [size] - unlock section\n");
474
475 static int nand_load_image(cmd_tbl_t *cmdtp, nand_info_t *nand,
476                            ulong offset, ulong addr, char *cmd)
477 {
478         int r;
479         char *ep, *s;
480         ulong cnt;
481         image_header_t *hdr;
482         int jffs2 = 0;
483
484         s = strchr(cmd, '.');
485         if (s != NULL &&
486             (!strcmp(s, ".jffs2") || !strcmp(s, ".e") || !strcmp(s, ".i")))
487                 jffs2 = 1;
488
489         printf("\nLoading from %s, offset 0x%lx\n", nand->name, offset);
490
491         cnt = nand->oobblock;
492         if (jffs2) {
493                 nand_read_options_t opts;
494                 memset(&opts, 0, sizeof(opts));
495                 opts.buffer     = (u_char*) addr;
496                 opts.length     = cnt;
497                 opts.offset     = offset;
498                 opts.quiet      = 1;
499                 r = nand_read_opts(nand, &opts);
500         } else {
501                 r = nand_read(nand, offset, &cnt, (u_char *) addr);
502         }
503
504         if (r) {
505                 puts("** Read error\n");
506                 SHOW_BOOT_PROGRESS(-1);
507                 return 1;
508         }
509
510         hdr = (image_header_t *) addr;
511
512         if (ntohl(hdr->ih_magic) != IH_MAGIC) {
513                 printf("\n** Bad Magic Number 0x%x **\n", hdr->ih_magic);
514                 SHOW_BOOT_PROGRESS(-1);
515                 return 1;
516         }
517
518         print_image_hdr(hdr);
519
520         cnt = (ntohl(hdr->ih_size) + sizeof (image_header_t));
521         if (jffs2) {
522                 nand_read_options_t opts;
523                 memset(&opts, 0, sizeof(opts));
524                 opts.buffer     = (u_char*) addr;
525                 opts.length     = cnt;
526                 opts.offset     = offset;
527                 opts.quiet      = 1;
528                 r = nand_read_opts(nand, &opts);
529         } else {
530                 r = nand_read(nand, offset, &cnt, (u_char *) addr);
531         }
532
533         if (r) {
534                 puts("** Read error\n");
535                 SHOW_BOOT_PROGRESS(-1);
536                 return 1;
537         }
538
539         /* Loading ok, update default load address */
540
541         load_addr = addr;
542
543         /* Check if we should attempt an auto-start */
544         if (((ep = getenv("autostart")) != NULL) && (strcmp(ep, "yes") == 0)) {
545                 char *local_args[2];
546                 extern int do_bootm(cmd_tbl_t *, int, int, char *[]);
547
548                 local_args[0] = cmd;
549                 local_args[1] = NULL;
550
551                 printf("Automatic boot of image at addr 0x%08lx ...\n", addr);
552
553                 do_bootm(cmdtp, 0, 1, local_args);
554                 return 1;
555         }
556         return 0;
557 }
558
559 int do_nandboot(cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
560 {
561         char *boot_device = NULL;
562         int idx;
563         ulong addr, offset = 0;
564 #if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
565         struct mtd_device *dev;
566         struct part_info *part;
567         u8 pnum;
568
569         if (argc >= 2) {
570                 char *p = (argc == 2) ? argv[1] : argv[2];
571                 if (!(str2long(p, &addr)) && (mtdparts_init() == 0) &&
572                     (find_dev_and_part(p, &dev, &pnum, &part) == 0)) {
573                         if (dev->id->type != MTD_DEV_TYPE_NAND) {
574                                 puts("Not a NAND device\n");
575                                 return 1;
576                         }
577                         if (argc > 3)
578                                 goto usage;
579                         if (argc == 3)
580                                 addr = simple_strtoul(argv[1], NULL, 16);
581                         else
582                                 addr = CFG_LOAD_ADDR;
583                         return nand_load_image(cmdtp, &nand_info[dev->id->num],
584                                                part->offset, addr, argv[0]);
585                 }
586         }
587 #endif
588
589         switch (argc) {
590         case 1:
591                 addr = CFG_LOAD_ADDR;
592                 boot_device = getenv("bootdevice");
593                 break;
594         case 2:
595                 addr = simple_strtoul(argv[1], NULL, 16);
596                 boot_device = getenv("bootdevice");
597                 break;
598         case 3:
599                 addr = simple_strtoul(argv[1], NULL, 16);
600                 boot_device = argv[2];
601                 break;
602         case 4:
603                 addr = simple_strtoul(argv[1], NULL, 16);
604                 boot_device = argv[2];
605                 offset = simple_strtoul(argv[3], NULL, 16);
606                 break;
607         default:
608 #if (CONFIG_COMMANDS & CFG_CMD_JFFS2) && defined(CONFIG_JFFS2_CMDLINE)
609 usage:
610 #endif
611                 printf("Usage:\n%s\n", cmdtp->usage);
612                 SHOW_BOOT_PROGRESS(-1);
613                 return 1;
614         }
615
616         if (!boot_device) {
617                 puts("\n** No boot device **\n");
618                 SHOW_BOOT_PROGRESS(-1);
619                 return 1;
620         }
621
622         idx = simple_strtoul(boot_device, NULL, 16);
623
624         if (idx < 0 || idx >= CFG_MAX_NAND_DEVICE || !nand_info[idx].name) {
625                 printf("\n** Device %d not available\n", idx);
626                 SHOW_BOOT_PROGRESS(-1);
627                 return 1;
628         }
629
630         return nand_load_image(cmdtp, &nand_info[idx], offset, addr, argv[0]);
631 }
632
633 U_BOOT_CMD(nboot, 4, 1, do_nandboot,
634         "nboot   - boot from NAND device\n",
635         "[.jffs2] [partition] | [[[loadAddr] dev] offset]\n");
636
637 #endif                          /* (CONFIG_COMMANDS & CFG_CMD_NAND) */
638
639 #else /* CFG_NAND_LEGACY */
640 /*
641  *
642  * Legacy NAND support - to be phased out
643  *
644  */
645 #include <command.h>
646 #include <malloc.h>
647 #include <asm/io.h>
648 #include <watchdog.h>
649
650 #ifdef CONFIG_SHOW_BOOT_PROGRESS
651 # include <status_led.h>
652 # define SHOW_BOOT_PROGRESS(arg)        show_boot_progress(arg)
653 #else
654 # define SHOW_BOOT_PROGRESS(arg)
655 #endif
656
657 #if (CONFIG_COMMANDS & CFG_CMD_NAND)
658 #include <linux/mtd/nand_legacy.h>
659 #if 0
660 #include <linux/mtd/nand_ids.h>
661 #include <jffs2/jffs2.h>
662 #endif
663
664 #ifdef CONFIG_OMAP1510
665 void archflashwp(void *archdata, int wp);
666 #endif
667
668 #define ROUND_DOWN(value,boundary)      ((value) & (~((boundary)-1)))
669
670 #undef  NAND_DEBUG
671 #undef  PSYCHO_DEBUG
672
673 /* ****************** WARNING *********************
674  * When ALLOW_ERASE_BAD_DEBUG is non-zero the erase command will
675  * erase (or at least attempt to erase) blocks that are marked
676  * bad. This can be very handy if you are _sure_ that the block
677  * is OK, say because you marked a good block bad to test bad
678  * block handling and you are done testing, or if you have
679  * accidentally marked blocks bad.
680  *
681  * Erasing factory marked bad blocks is a _bad_ idea. If the
682  * erase succeeds there is no reliable way to find them again,
683  * and attempting to program or erase bad blocks can affect
684  * the data in _other_ (good) blocks.
685  */
686 #define  ALLOW_ERASE_BAD_DEBUG 0
687
688 #define CONFIG_MTD_NAND_ECC  /* enable ECC */
689 #define CONFIG_MTD_NAND_ECC_JFFS2
690
691 /* bits for nand_legacy_rw() `cmd'; or together as needed */
692 #define NANDRW_READ     0x01
693 #define NANDRW_WRITE    0x00
694 #define NANDRW_JFFS2    0x02
695 #define NANDRW_JFFS2_SKIP       0x04
696
697 /*
698  * Imports from nand_legacy.c
699  */
700 extern struct nand_chip nand_dev_desc[CFG_MAX_NAND_DEVICE];
701 extern int curr_device;
702 extern int nand_legacy_erase(struct nand_chip *nand, size_t ofs,
703                             size_t len, int clean);
704 extern int nand_legacy_rw(struct nand_chip *nand, int cmd, size_t start,
705                          size_t len, size_t *retlen, u_char *buf);
706 extern void nand_print(struct nand_chip *nand);
707 extern void nand_print_bad(struct nand_chip *nand);
708 extern int nand_read_oob(struct nand_chip *nand, size_t ofs,
709                                size_t len, size_t *retlen, u_char *buf);
710 extern int nand_write_oob(struct nand_chip *nand, size_t ofs,
711                                 size_t len, size_t *retlen, const u_char *buf);
712
713
714 int do_nand (cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
715 {
716         int rcode = 0;
717
718         switch (argc) {
719         case 0:
720         case 1:
721                 printf ("Usage:\n%s\n", cmdtp->usage);
722                 return 1;
723         case 2:
724                 if (strcmp (argv[1], "info") == 0) {
725                         int i;
726
727                         putc ('\n');
728
729                         for (i = 0; i < CFG_MAX_NAND_DEVICE; ++i) {
730                                 if (nand_dev_desc[i].ChipID ==
731                                     NAND_ChipID_UNKNOWN)
732                                         continue;       /* list only known devices */
733                                 printf ("Device %d: ", i);
734                                 nand_print (&nand_dev_desc[i]);
735                         }
736                         return 0;
737
738                 } else if (strcmp (argv[1], "device") == 0) {
739                         if ((curr_device < 0)
740                             || (curr_device >= CFG_MAX_NAND_DEVICE)) {
741                                 puts ("\nno devices available\n");
742                                 return 1;
743                         }
744                         printf ("\nDevice %d: ", curr_device);
745                         nand_print (&nand_dev_desc[curr_device]);
746                         return 0;
747
748                 } else if (strcmp (argv[1], "bad") == 0) {
749                         if ((curr_device < 0)
750                             || (curr_device >= CFG_MAX_NAND_DEVICE)) {
751                                 puts ("\nno devices available\n");
752                                 return 1;
753                         }
754                         printf ("\nDevice %d bad blocks:\n", curr_device);
755                         nand_print_bad (&nand_dev_desc[curr_device]);
756                         return 0;
757
758                 }
759                 printf ("Usage:\n%s\n", cmdtp->usage);
760                 return 1;
761         case 3:
762                 if (strcmp (argv[1], "device") == 0) {
763                         int dev = (int) simple_strtoul (argv[2], NULL, 10);
764
765                         printf ("\nDevice %d: ", dev);
766                         if (dev >= CFG_MAX_NAND_DEVICE) {
767                                 puts ("unknown device\n");
768                                 return 1;
769                         }
770                         nand_print (&nand_dev_desc[dev]);
771                         /*nand_print (dev); */
772
773                         if (nand_dev_desc[dev].ChipID == NAND_ChipID_UNKNOWN) {
774                                 return 1;
775                         }
776
777                         curr_device = dev;
778
779                         puts ("... is now current device\n");
780
781                         return 0;
782                 } else if (strcmp (argv[1], "erase") == 0
783                            && strcmp (argv[2], "clean") == 0) {
784                         struct nand_chip *nand = &nand_dev_desc[curr_device];
785                         ulong off = 0;
786                         ulong size = nand->totlen;
787                         int ret;
788
789                         printf ("\nNAND erase: device %d offset %ld, size %ld ... ", curr_device, off, size);
790
791                         ret = nand_legacy_erase (nand, off, size, 1);
792
793                         printf ("%s\n", ret ? "ERROR" : "OK");
794
795                         return ret;
796                 }
797
798                 printf ("Usage:\n%s\n", cmdtp->usage);
799                 return 1;
800         default:
801                 /* at least 4 args */
802
803                 if (strncmp (argv[1], "read", 4) == 0 ||
804                     strncmp (argv[1], "write", 5) == 0) {
805                         ulong addr = simple_strtoul (argv[2], NULL, 16);
806                         ulong off = simple_strtoul (argv[3], NULL, 16);
807                         ulong size = simple_strtoul (argv[4], NULL, 16);
808                         int cmd = (strncmp (argv[1], "read", 4) == 0) ?
809                                 NANDRW_READ : NANDRW_WRITE;
810                         int ret, total;
811                         char *cmdtail = strchr (argv[1], '.');
812
813                         if (cmdtail && !strncmp (cmdtail, ".oob", 2)) {
814                                 /* read out-of-band data */
815                                 if (cmd & NANDRW_READ) {
816                                         ret = nand_read_oob (nand_dev_desc + curr_device,
817                                                              off, size, (size_t *) & total,
818                                                              (u_char *) addr);
819                                 } else {
820                                         ret = nand_write_oob (nand_dev_desc + curr_device,
821                                                               off, size, (size_t *) & total,
822                                                               (u_char *) addr);
823                                 }
824                                 return ret;
825                         } else if (cmdtail && !strncmp (cmdtail, ".jffs2", 2))
826                                 cmd |= NANDRW_JFFS2;    /* skip bad blocks */
827                         else if (cmdtail && !strncmp (cmdtail, ".jffs2s", 2)) {
828                                 cmd |= NANDRW_JFFS2;    /* skip bad blocks (on read too) */
829                                 if (cmd & NANDRW_READ)
830                                         cmd |= NANDRW_JFFS2_SKIP;       /* skip bad blocks (on read too) */
831                         }
832 #ifdef SXNI855T
833                         /* need ".e" same as ".j" for compatibility with older units */
834                         else if (cmdtail && !strcmp (cmdtail, ".e"))
835                                 cmd |= NANDRW_JFFS2;    /* skip bad blocks */
836 #endif
837 #ifdef CFG_NAND_SKIP_BAD_DOT_I
838                         /* need ".i" same as ".jffs2s" for compatibility with older units (esd) */
839                         /* ".i" for image -> read skips bad block (no 0xff) */
840                         else if (cmdtail && !strcmp (cmdtail, ".i")) {
841                                 cmd |= NANDRW_JFFS2;    /* skip bad blocks (on read too) */
842                                 if (cmd & NANDRW_READ)
843                                         cmd |= NANDRW_JFFS2_SKIP;       /* skip bad blocks (on read too) */
844                         }
845 #endif /* CFG_NAND_SKIP_BAD_DOT_I */
846                         else if (cmdtail) {
847                                 printf ("Usage:\n%s\n", cmdtp->usage);
848                                 return 1;
849                         }
850
851                         printf ("\nNAND %s: device %d offset %ld, size %ld ...\n",
852                                 (cmd & NANDRW_READ) ? "read" : "write",
853                                 curr_device, off, size);
854
855                         ret = nand_legacy_rw (nand_dev_desc + curr_device,
856                                               cmd, off, size,
857                                               (size_t *) & total,
858                                               (u_char *) addr);
859
860                         printf (" %d bytes %s: %s\n", total,
861                                 (cmd & NANDRW_READ) ? "read" : "written",
862                                 ret ? "ERROR" : "OK");
863
864                         return ret;
865                 } else if (strcmp (argv[1], "erase") == 0 &&
866                            (argc == 4 || strcmp ("clean", argv[2]) == 0)) {
867                         int clean = argc == 5;
868                         ulong off =
869                                 simple_strtoul (argv[2 + clean], NULL, 16);
870                         ulong size =
871                                 simple_strtoul (argv[3 + clean], NULL, 16);
872                         int ret;
873
874                         printf ("\nNAND erase: device %d offset %ld, size %ld ...\n",
875                                 curr_device, off, size);
876
877                         ret = nand_legacy_erase (nand_dev_desc + curr_device,
878                                                  off, size, clean);
879
880                         printf ("%s\n", ret ? "ERROR" : "OK");
881
882                         return ret;
883                 } else {
884                         printf ("Usage:\n%s\n", cmdtp->usage);
885                         rcode = 1;
886                 }
887
888                 return rcode;
889         }
890 }
891
892 U_BOOT_CMD(
893         nand,   5,      1,      do_nand,
894         "nand    - legacy NAND sub-system\n",
895         "info  - show available NAND devices\n"
896         "nand device [dev] - show or set current device\n"
897         "nand read[.jffs2[s]]  addr off size\n"
898         "nand write[.jffs2] addr off size - read/write `size' bytes starting\n"
899         "    at offset `off' to/from memory address `addr'\n"
900         "nand erase [clean] [off size] - erase `size' bytes from\n"
901         "    offset `off' (entire device if not specified)\n"
902         "nand bad - show bad blocks\n"
903         "nand read.oob addr off size - read out-of-band data\n"
904         "nand write.oob addr off size - read out-of-band data\n"
905 );
906
907 int do_nandboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
908 {
909         char *boot_device = NULL;
910         char *ep;
911         int dev;
912         ulong cnt;
913         ulong addr;
914         ulong offset = 0;
915         image_header_t *hdr;
916         int rcode = 0;
917         switch (argc) {
918         case 1:
919                 addr = CFG_LOAD_ADDR;
920                 boot_device = getenv ("bootdevice");
921                 break;
922         case 2:
923                 addr = simple_strtoul(argv[1], NULL, 16);
924                 boot_device = getenv ("bootdevice");
925                 break;
926         case 3:
927                 addr = simple_strtoul(argv[1], NULL, 16);
928                 boot_device = argv[2];
929                 break;
930         case 4:
931                 addr = simple_strtoul(argv[1], NULL, 16);
932                 boot_device = argv[2];
933                 offset = simple_strtoul(argv[3], NULL, 16);
934                 break;
935         default:
936                 printf ("Usage:\n%s\n", cmdtp->usage);
937                 SHOW_BOOT_PROGRESS (-1);
938                 return 1;
939         }
940
941         if (!boot_device) {
942                 puts ("\n** No boot device **\n");
943                 SHOW_BOOT_PROGRESS (-1);
944                 return 1;
945         }
946
947         dev = simple_strtoul(boot_device, &ep, 16);
948
949         if ((dev >= CFG_MAX_NAND_DEVICE) ||
950             (nand_dev_desc[dev].ChipID == NAND_ChipID_UNKNOWN)) {
951                 printf ("\n** Device %d not available\n", dev);
952                 SHOW_BOOT_PROGRESS (-1);
953                 return 1;
954         }
955
956         printf ("\nLoading from device %d: %s at 0x%lx (offset 0x%lx)\n",
957                 dev, nand_dev_desc[dev].name, nand_dev_desc[dev].IO_ADDR,
958                 offset);
959
960         if (nand_legacy_rw (nand_dev_desc + dev, NANDRW_READ, offset,
961                         SECTORSIZE, NULL, (u_char *)addr)) {
962                 printf ("** Read error on %d\n", dev);
963                 SHOW_BOOT_PROGRESS (-1);
964                 return 1;
965         }
966
967         hdr = (image_header_t *)addr;
968
969         if (ntohl(hdr->ih_magic) == IH_MAGIC) {
970
971                 print_image_hdr (hdr);
972
973                 cnt = (ntohl(hdr->ih_size) + sizeof(image_header_t));
974                 cnt -= SECTORSIZE;
975         } else {
976                 printf ("\n** Bad Magic Number 0x%x **\n", ntohl(hdr->ih_magic));
977                 SHOW_BOOT_PROGRESS (-1);
978                 return 1;
979         }
980
981         if (nand_legacy_rw (nand_dev_desc + dev, NANDRW_READ,
982                         offset + SECTORSIZE, cnt, NULL,
983                         (u_char *)(addr+SECTORSIZE))) {
984                 printf ("** Read error on %d\n", dev);
985                 SHOW_BOOT_PROGRESS (-1);
986                 return 1;
987         }
988
989         /* Loading ok, update default load address */
990
991         load_addr = addr;
992
993         /* Check if we should attempt an auto-start */
994         if (((ep = getenv("autostart")) != NULL) && (strcmp(ep,"yes") == 0)) {
995                 char *local_args[2];
996                 extern int do_bootm (cmd_tbl_t *, int, int, char *[]);
997
998                 local_args[0] = argv[0];
999                 local_args[1] = NULL;
1000
1001                 printf ("Automatic boot of image at addr 0x%08lx ...\n", addr);
1002
1003                 do_bootm (cmdtp, 0, 1, local_args);
1004                 rcode = 1;
1005         }
1006         return rcode;
1007 }
1008
1009 U_BOOT_CMD(
1010         nboot,  4,      1,      do_nandboot,
1011         "nboot   - boot from NAND device\n",
1012         "loadAddr dev\n"
1013 );
1014
1015 #endif /* (CONFIG_COMMANDS & CFG_CMD_NAND) */
1016
1017 #endif /* CFG_NAND_LEGACY */