[thirdparty/u-boot.git] / cmd / onenand.c

/*
 *  U-Boot command for OneNAND support
 *
 *  Copyright (C) 2005-2008 Samsung Electronics
 *  Kyungmin Park <kyungmin.park@samsung.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include <common.h>
#include <command.h>
#include <malloc.h>
#include <linux/printk.h>

#include <linux/compat.h>
#include <linux/mtd/mtd.h>
#include <linux/mtd/onenand.h>

#include <asm/io.h>

static struct mtd_info *mtd;

static loff_t next_ofs;
static loff_t skip_ofs;

static int arg_off_size_onenand(int argc, char *const argv[], ulong *off,
				size_t *size)
{
	if (argc >= 1) {
		if (!(str2long(argv[0], off))) {
			printf("'%s' is not a number\n", argv[0]);
			return -1;
		}
	} else {
		*off = 0;
	}

	if (argc >= 2) {
		if (!(str2long(argv[1], (ulong *)size))) {
			printf("'%s' is not a number\n", argv[1]);
			return -1;
		}
	} else {
		*size = mtd->size - *off;
	}

	if ((*off + *size) > mtd->size) {
		printf("total chip size (0x%llx) exceeded!\n", mtd->size);
		return -1;
	}

	if (*size == mtd->size)
		puts("whole chip\n");
	else
		printf("offset 0x%lx, size 0x%zx\n", *off, *size);

	return 0;
}

static int onenand_block_read(loff_t from, size_t len,
			      size_t *retlen, u_char *buf, int oob)
{
	struct onenand_chip *this = mtd->priv;
	int blocks = (int) len >> this->erase_shift;
	int blocksize = (1 << this->erase_shift);
	loff_t ofs = from;
	struct mtd_oob_ops ops = {
		.retlen		= 0,
	};
	int ret;

	if (oob)
		ops.ooblen = blocksize;
	else
		ops.len = blocksize;

	while (blocks) {
		ret = mtd_block_isbad(mtd, ofs);
		if (ret) {
			printk("Bad blocks %d at 0x%x\n",
			       (u32)(ofs >> this->erase_shift), (u32)ofs);
			ofs += blocksize;
			continue;
		}

		if (oob)
			ops.oobbuf = buf;
		else
			ops.datbuf = buf;

		ops.retlen = 0;
		ret = mtd_read_oob(mtd, ofs, &ops);
		if (ret) {
			printk("Read failed 0x%x, %d\n", (u32)ofs, ret);
			ofs += blocksize;
			continue;
		}
		ofs += blocksize;
		buf += blocksize;
		blocks--;
		*retlen += ops.retlen;
	}

	return 0;
}

static int onenand_write_oneblock_withoob(loff_t to, const u_char * buf,
					  size_t *retlen)
{
	struct mtd_oob_ops ops = {
		.len = mtd->writesize,
		.ooblen = mtd->oobsize,
		.mode = MTD_OPS_AUTO_OOB,
	};
	int page, ret = 0;
	for (page = 0; page < (mtd->erasesize / mtd->writesize); page ++) {
		ops.datbuf = (u_char *)buf;
		buf += mtd->writesize;
		ops.oobbuf = (u_char *)buf;
		buf += mtd->oobsize;
		ret = mtd_write_oob(mtd, to, &ops);
		if (ret)
			break;
		to += mtd->writesize;
	}

	*retlen = (ret) ? 0 : mtd->erasesize;
	return ret;
}

static int onenand_block_write(loff_t to, size_t len,
			       size_t *retlen, const u_char * buf, int withoob)
{
	struct onenand_chip *this = mtd->priv;
	int blocks = len >> this->erase_shift;
	int blocksize = (1 << this->erase_shift);
	loff_t ofs;
	size_t _retlen = 0;
	int ret;

	if ((to & (mtd->writesize - 1)) != 0) {
		printf("Attempt to write non block-aligned data\n");
		*retlen = 0;
		return 1;
	}

	if (to == next_ofs) {
		next_ofs = to + len;
		to += skip_ofs;
	} else {
		next_ofs = to + len;
		skip_ofs = 0;
	}
	ofs = to;

	while (blocks) {
		ret = mtd_block_isbad(mtd, ofs);
		if (ret) {
			printk("Bad blocks %d at 0x%x\n",
			       (u32)(ofs >> this->erase_shift), (u32)ofs);
			skip_ofs += blocksize;
			goto next;
		}

		if (!withoob)
			ret = mtd_write(mtd, ofs, blocksize, &_retlen, buf);
		else
			ret = onenand_write_oneblock_withoob(ofs, buf, &_retlen);
		if (ret) {
			printk("Write failed 0x%x, %d", (u32)ofs, ret);
			skip_ofs += blocksize;
			goto next;
		}

		buf += blocksize;
		blocks--;
		*retlen += _retlen;
next:
		ofs += blocksize;
	}

	return 0;
}

static int onenand_block_erase(u32 start, u32 size, int force)
{
	struct onenand_chip *this = mtd->priv;
	struct erase_info instr = {};
	loff_t ofs;
	int ret;
	int blocksize = 1 << this->erase_shift;

	for (ofs = start; ofs < (start + size); ofs += blocksize) {
		ret = mtd_block_isbad(mtd, ofs);
		if (ret && !force) {
			printf("Skip erase bad block %d at 0x%x\n",
			       (u32)(ofs >> this->erase_shift), (u32)ofs);
			continue;
		}

		instr.addr = ofs;
		instr.len = blocksize;
		instr.priv = force;
		instr.mtd = mtd;
		ret = mtd_erase(mtd, &instr);
		if (ret) {
			printf("erase failed block %d at 0x%x\n",
			       (u32)(ofs >> this->erase_shift), (u32)ofs);
			continue;
		}
	}

	return 0;
}

static int onenand_block_test(u32 start, u32 size)
{
	struct onenand_chip *this = mtd->priv;
	struct erase_info instr = {};

	int blocks;
	loff_t ofs;
	int blocksize = 1 << this->erase_shift;
	int start_block, end_block;
	size_t retlen;
	u_char *buf;
	u_char *verify_buf;
	int ret;

	buf = malloc(blocksize);
	if (!buf) {
		printf("Not enough malloc space available!\n");
		return -1;
	}

	verify_buf = malloc(blocksize);
	if (!verify_buf) {
		printf("Not enough malloc space available!\n");
		return -1;
	}

	start_block = start >> this->erase_shift;
	end_block = (start + size) >> this->erase_shift;

	/* Protect boot-loader from badblock testing */
	if (start_block < 2)
		start_block = 2;

	if (end_block > (mtd->size >> this->erase_shift))
		end_block = mtd->size >> this->erase_shift;

	blocks = start_block;
	ofs = start;
	while (blocks < end_block) {
		printf("\rTesting block %d at 0x%x", (u32)(ofs >> this->erase_shift), (u32)ofs);

		ret = mtd_block_isbad(mtd, ofs);
		if (ret) {
			printf("Skip erase bad block %d at 0x%x\n",
			       (u32)(ofs >> this->erase_shift), (u32)ofs);
			goto next;
		}

		instr.addr = ofs;
		instr.len = blocksize;
		ret = mtd_erase(mtd, &instr);
		if (ret) {
			printk("Erase failed 0x%x, %d\n", (u32)ofs, ret);
			goto next;
		}

		ret = mtd_write(mtd, ofs, blocksize, &retlen, buf);
		if (ret) {
			printk("Write failed 0x%x, %d\n", (u32)ofs, ret);
			goto next;
		}

		ret = mtd_read(mtd, ofs, blocksize, &retlen, verify_buf);
		if (ret) {
			printk("Read failed 0x%x, %d\n", (u32)ofs, ret);
			goto next;
		}

		if (memcmp(buf, verify_buf, blocksize))
			printk("\nRead/Write test failed at 0x%x\n", (u32)ofs);

next:
		ofs += blocksize;
		blocks++;
	}
	printf("...Done\n");

	free(buf);
	free(verify_buf);

	return 0;
}

static int onenand_dump(struct mtd_info *mtd, ulong off, int only_oob)
{
	int i;
	u_char *datbuf, *oobbuf, *p;
	struct mtd_oob_ops ops;
	loff_t addr;

	datbuf = malloc(mtd->writesize + mtd->oobsize);
	oobbuf = malloc(mtd->oobsize);
	if (!datbuf || !oobbuf) {
		puts("No memory for page buffer\n");
		return 1;
	}
	off &= ~(mtd->writesize - 1);
	addr = (loff_t) off;
	memset(&ops, 0, sizeof(ops));
	ops.datbuf = datbuf;
	ops.oobbuf = oobbuf;
	ops.len = mtd->writesize;
	ops.ooblen = mtd->oobsize;
	ops.retlen = 0;
	i = mtd_read_oob(mtd, addr, &ops);
	if (i < 0) {
		printf("Error (%d) reading page %08lx\n", i, off);
		free(datbuf);
		free(oobbuf);
		return 1;
	}
	printf("Page %08lx dump:\n", off);
	i = mtd->writesize >> 4;
	p = datbuf;

	while (i--) {
		if (!only_oob)
			printf("\t%02x %02x %02x %02x %02x %02x %02x %02x"
			       "  %02x %02x %02x %02x %02x %02x %02x %02x\n",
			       p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7],
			       p[8], p[9], p[10], p[11], p[12], p[13], p[14],
			       p[15]);
		p += 16;
	}
	puts("OOB:\n");
	i = mtd->oobsize >> 3;
	p = oobbuf;

	while (i--) {
		printf("\t%02x %02x %02x %02x %02x %02x %02x %02x\n",
		       p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7]);
		p += 8;
	}
	free(datbuf);
	free(oobbuf);

	return 0;
}

static int do_onenand_info(struct cmd_tbl *cmdtp, int flag, int argc,
			   char *const argv[])
{
	printf("%s\n", mtd->name);
	return 0;
}

static int do_onenand_bad(struct cmd_tbl *cmdtp, int flag, int argc,
			  char *const argv[])
{
	ulong ofs;

	mtd = &onenand_mtd;
	/* Currently only one OneNAND device is supported */
	printf("\nDevice %d bad blocks:\n", 0);
	for (ofs = 0; ofs < mtd->size; ofs += mtd->erasesize) {
		if (mtd_block_isbad(mtd, ofs))
			printf("  %08x\n", (u32)ofs);
	}

	return 0;
}

static int do_onenand_read(struct cmd_tbl *cmdtp, int flag, int argc,
			   char *const argv[])
{
	char *s;
	int oob = 0;
	ulong addr, ofs;
	size_t len;
	int ret = 0;
	size_t retlen = 0;

	if (argc < 3)
		return CMD_RET_USAGE;

	s = strchr(argv[0], '.');
	if ((s != NULL) && (!strcmp(s, ".oob")))
		oob = 1;

	addr = (ulong)hextoul(argv[1], NULL);

	printf("\nOneNAND read: ");
	if (arg_off_size_onenand(argc - 2, argv + 2, &ofs, &len) != 0)
		return 1;

	ret = onenand_block_read(ofs, len, &retlen, (u8 *)addr, oob);

	printf(" %zu bytes read: %s\n", retlen, ret ? "ERROR" : "OK");

	return ret == 0 ? 0 : 1;
}

static int do_onenand_write(struct cmd_tbl *cmdtp, int flag, int argc,
			    char *const argv[])
{
	ulong addr, ofs;
	size_t len;
	int ret = 0, withoob = 0;
	size_t retlen = 0;

	if (argc < 3)
		return CMD_RET_USAGE;

	if (strncmp(argv[0] + 6, "yaffs", 5) == 0)
		withoob = 1;

	addr = (ulong)hextoul(argv[1], NULL);

	printf("\nOneNAND write: ");
	if (arg_off_size_onenand(argc - 2, argv + 2, &ofs, &len) != 0)
		return 1;

	ret = onenand_block_write(ofs, len, &retlen, (u8 *)addr, withoob);

	printf(" %zu bytes written: %s\n", retlen, ret ? "ERROR" : "OK");

	return ret == 0 ? 0 : 1;
}

static int do_onenand_erase(struct cmd_tbl *cmdtp, int flag, int argc,
			    char *const argv[])
{
	ulong ofs;
	int ret = 0;
	size_t len;
	int force;

	/*
	 * Syntax is:
	 *   0       1     2       3    4
	 *   onenand erase [force] [off size]
	 */
	argc--;
	argv++;
	if (argc)
	{
		if (!strcmp("force", argv[0]))
		{
			force = 1;
			argc--;
			argv++;
		}
	}
	printf("\nOneNAND erase: ");

	/* skip first two or three arguments, look for offset and size */
	if (arg_off_size_onenand(argc, argv, &ofs, &len) != 0)
		return 1;

	ret = onenand_block_erase(ofs, len, force);

	printf("%s\n", ret ? "ERROR" : "OK");

	return ret == 0 ? 0 : 1;
}

static int do_onenand_test(struct cmd_tbl *cmdtp, int flag, int argc,
			   char *const argv[])
{
	ulong ofs;
	int ret = 0;
	size_t len;

	/*
	 * Syntax is:
	 *   0       1     2       3    4
	 *   onenand test [force] [off size]
	 */

	printf("\nOneNAND test: ");

	/* skip first two or three arguments, look for offset and size */
	if (arg_off_size_onenand(argc - 1, argv + 1, &ofs, &len) != 0)
		return 1;

	ret = onenand_block_test(ofs, len);

	printf("%s\n", ret ? "ERROR" : "OK");

	return ret == 0 ? 0 : 1;
}

static int do_onenand_dump(struct cmd_tbl *cmdtp, int flag, int argc,
			   char *const argv[])
{
	ulong ofs;
	int ret = 0;
	char *s;

	if (argc < 2)
		return CMD_RET_USAGE;

	s = strchr(argv[0], '.');
	ofs = (int)hextoul(argv[1], NULL);

	if (s != NULL && strcmp(s, ".oob") == 0)
		ret = onenand_dump(mtd, ofs, 1);
	else
		ret = onenand_dump(mtd, ofs, 0);

	return ret == 0 ? 1 : 0;
}

static int do_onenand_markbad(struct cmd_tbl *cmdtp, int flag, int argc,
			      char *const argv[])
{
	int ret = 0;
	ulong addr;

	argc -= 2;
	argv += 2;

	if (argc <= 0)
		return CMD_RET_USAGE;

	while (argc > 0) {
		addr = hextoul(*argv, NULL);

		if (mtd_block_markbad(mtd, addr)) {
			printf("block 0x%08lx NOT marked "
				"as bad! ERROR %d\n",
				addr, ret);
			ret = 1;
		} else {
			printf("block 0x%08lx successfully "
				"marked as bad\n",
				addr);
		}
		--argc;
		++argv;
	}
	return ret;
}

static struct cmd_tbl cmd_onenand_sub[] = {
	U_BOOT_CMD_MKENT(info, 1, 0, do_onenand_info, "", ""),
	U_BOOT_CMD_MKENT(bad, 1, 0, do_onenand_bad, "", ""),
	U_BOOT_CMD_MKENT(read, 4, 0, do_onenand_read, "", ""),
	U_BOOT_CMD_MKENT(write, 4, 0, do_onenand_write, "", ""),
	U_BOOT_CMD_MKENT(write.yaffs, 4, 0, do_onenand_write, "", ""),
	U_BOOT_CMD_MKENT(erase, 3, 0, do_onenand_erase, "", ""),
	U_BOOT_CMD_MKENT(test, 3, 0, do_onenand_test, "", ""),
	U_BOOT_CMD_MKENT(dump, 2, 0, do_onenand_dump, "", ""),
	U_BOOT_CMD_MKENT(markbad, CONFIG_SYS_MAXARGS, 0, do_onenand_markbad, "", ""),
};

static int do_onenand(struct cmd_tbl *cmdtp, int flag, int argc,
		      char *const argv[])
{
	struct cmd_tbl *c;

	if (argc < 2)
		return CMD_RET_USAGE;

	mtd = &onenand_mtd;

	/* Strip off leading 'onenand' command argument */
	argc--;
	argv++;

	c = find_cmd_tbl(argv[0], &cmd_onenand_sub[0], ARRAY_SIZE(cmd_onenand_sub));

	if (c)
		return c->cmd(cmdtp, flag, argc, argv);
	else
		return CMD_RET_USAGE;
}

U_BOOT_CMD(
	onenand,	CONFIG_SYS_MAXARGS,	1,	do_onenand,
	"OneNAND sub-system",
	"info - show available OneNAND devices\n"
	"onenand bad - show bad blocks\n"
	"onenand read[.oob] addr off size\n"
	"onenand write[.yaffs] addr off size\n"
	"    read/write 'size' bytes starting at offset 'off'\n"
	"    to/from memory address 'addr', skipping bad blocks.\n"
	"onenand erase [force] [off size] - erase 'size' bytes from\n"
	"onenand test [off size] - test 'size' bytes from\n"
	"    offset 'off' (entire device if not specified)\n"
	"onenand dump[.oob] off - dump page\n"
	"onenand markbad off [...] - mark bad block(s) at offset (UNSAFE)"
);
Commit	Line	Data
d7e8ce10 KP	1	/*
	2	* U-Boot command for OneNAND support
	3	*
c438ea17	4	* Copyright (C) 2005-2008 Samsung Electronics
d7e8ce10 KP	5	* Kyungmin Park <kyungmin.park@samsung.com>
	6	*
	7	* This program is free software; you can redistribute it and/or modify
	8	* it under the terms of the GNU General Public License version 2 as
	9	* published by the Free Software Foundation.
	10	*/
	11
	12	#include <common.h>
	13	#include <command.h>
c438ea17	14	#include <malloc.h>
1e94b46f	15	#include <linux/printk.h>
d7e8ce10	16
7b15e2bb	17	#include <linux/compat.h>
d7e8ce10 KP	18	#include <linux/mtd/mtd.h>
	19	#include <linux/mtd/onenand.h>
	20
	21	#include <asm/io.h>
	22
c438ea17 SR	23	static struct mtd_info *mtd;
	24
	25	static loff_t next_ofs;
	26	static loff_t skip_ofs;
	27
09140113	28	static int arg_off_size_onenand(int argc, char const argv[], ulong off,
09c32807	29	size_t *size)
c438ea17 SR	30	{
	31	if (argc >= 1) {
	32	if (!(str2long(argv[0], off))) {
	33	printf("'%s' is not a number\n", argv[0]);
	34	return -1;
	35	}
	36	} else {
	37	*off = 0;
	38	}
	39
	40	if (argc >= 2) {
	41	if (!(str2long(argv[1], (ulong *)size))) {
	42	printf("'%s' is not a number\n", argv[1]);
	43	return -1;
	44	}
	45	} else {
	46	size = mtd->size - off;
	47	}
	48
	49	if ((off + size) > mtd->size) {
8d2effea	50	printf("total chip size (0x%llx) exceeded!\n", mtd->size);
c438ea17 SR	51	return -1;
	52	}
	53
	54	if (*size == mtd->size)
	55	puts("whole chip\n");
	56	else
6d813197	57	printf("offset 0x%lx, size 0x%zx\n", off, size);
c438ea17 SR	58
	59	return 0;
	60	}
	61
	62	static int onenand_block_read(loff_t from, size_t len,
	63	size_t retlen, u_char buf, int oob)
	64	{
	65	struct onenand_chip *this = mtd->priv;
	66	int blocks = (int) len >> this->erase_shift;
	67	int blocksize = (1 << this->erase_shift);
	68	loff_t ofs = from;
	69	struct mtd_oob_ops ops = {
	70	.retlen = 0,
	71	};
	72	int ret;
	73
	74	if (oob)
	75	ops.ooblen = blocksize;
	76	else
	77	ops.len = blocksize;
	78
	79	while (blocks) {
dfe64e2c	80	ret = mtd_block_isbad(mtd, ofs);
c438ea17 SR	81	if (ret) {
	82	printk("Bad blocks %d at 0x%x\n",
	83	(u32)(ofs >> this->erase_shift), (u32)ofs);
	84	ofs += blocksize;
	85	continue;
	86	}
	87
	88	if (oob)
	89	ops.oobbuf = buf;
	90	else
	91	ops.datbuf = buf;
	92
	93	ops.retlen = 0;
dfe64e2c	94	ret = mtd_read_oob(mtd, ofs, &ops);
c438ea17 SR	95	if (ret) {
	96	printk("Read failed 0x%x, %d\n", (u32)ofs, ret);
	97	ofs += blocksize;
	98	continue;
	99	}
	100	ofs += blocksize;
	101	buf += blocksize;
	102	blocks--;
	103	*retlen += ops.retlen;
	104	}
	105
	106	return 0;
	107	}
	108
41c86240 LW	109	static int onenand_write_oneblock_withoob(loff_t to, const u_char * buf,
	110	size_t *retlen)
	111	{
	112	struct mtd_oob_ops ops = {
	113	.len = mtd->writesize,
	114	.ooblen = mtd->oobsize,
dfe64e2c	115	.mode = MTD_OPS_AUTO_OOB,
41c86240 LW	116	};
	117	int page, ret = 0;
	118	for (page = 0; page < (mtd->erasesize / mtd->writesize); page ++) {
	119	ops.datbuf = (u_char *)buf;
	120	buf += mtd->writesize;
	121	ops.oobbuf = (u_char *)buf;
	122	buf += mtd->oobsize;
dfe64e2c	123	ret = mtd_write_oob(mtd, to, &ops);
41c86240 LW	124	if (ret)
	125	break;
	126	to += mtd->writesize;
	127	}
	128
	129	*retlen = (ret) ? 0 : mtd->erasesize;
	130	return ret;
	131	}
	132
c438ea17	133	static int onenand_block_write(loff_t to, size_t len,
41c86240	134	size_t retlen, const u_char buf, int withoob)
c438ea17 SR	135	{
	136	struct onenand_chip *this = mtd->priv;
	137	int blocks = len >> this->erase_shift;
	138	int blocksize = (1 << this->erase_shift);
	139	loff_t ofs;
	140	size_t _retlen = 0;
	141	int ret;
	142
9c00d982 LM	143	if ((to & (mtd->writesize - 1)) != 0) {
	144	printf("Attempt to write non block-aligned data\n");
	145	*retlen = 0;
	146	return 1;
	147	}
	148
c438ea17 SR	149	if (to == next_ofs) {
	150	next_ofs = to + len;
	151	to += skip_ofs;
	152	} else {
	153	next_ofs = to + len;
	154	skip_ofs = 0;
	155	}
	156	ofs = to;
	157
	158	while (blocks) {
dfe64e2c	159	ret = mtd_block_isbad(mtd, ofs);
c438ea17 SR	160	if (ret) {
	161	printk("Bad blocks %d at 0x%x\n",
	162	(u32)(ofs >> this->erase_shift), (u32)ofs);
	163	skip_ofs += blocksize;
	164	goto next;
	165	}
	166
41c86240	167	if (!withoob)
dfe64e2c	168	ret = mtd_write(mtd, ofs, blocksize, &_retlen, buf);
41c86240 LW	169	else
41c86240 LW	170	ret = onenand_write_oneblock_withoob(ofs, buf, &_retlen);
c438ea17 SR	171	if (ret) {
	172	printk("Write failed 0x%x, %d", (u32)ofs, ret);
	173	skip_ofs += blocksize;
	174	goto next;
	175	}
	176
	177	buf += blocksize;
	178	blocks--;
	179	*retlen += _retlen;
	180	next:
	181	ofs += blocksize;
	182	}
	183
	184	return 0;
	185	}
	186
	187	static int onenand_block_erase(u32 start, u32 size, int force)
	188	{
	189	struct onenand_chip *this = mtd->priv;
0d1ecc99	190	struct erase_info instr = {};
c438ea17 SR	191	loff_t ofs;
	192	int ret;
	193	int blocksize = 1 << this->erase_shift;
	194
	195	for (ofs = start; ofs < (start + size); ofs += blocksize) {
dfe64e2c	196	ret = mtd_block_isbad(mtd, ofs);
c438ea17 SR	197	if (ret && !force) {
	198	printf("Skip erase bad block %d at 0x%x\n",
	199	(u32)(ofs >> this->erase_shift), (u32)ofs);
	200	continue;
	201	}
	202
	203	instr.addr = ofs;
	204	instr.len = blocksize;
	205	instr.priv = force;
	206	instr.mtd = mtd;
dfe64e2c	207	ret = mtd_erase(mtd, &instr);
c438ea17 SR	208	if (ret) {
	209	printf("erase failed block %d at 0x%x\n",
	210	(u32)(ofs >> this->erase_shift), (u32)ofs);
	211	continue;
	212	}
	213	}
	214
	215	return 0;
	216	}
	217
	218	static int onenand_block_test(u32 start, u32 size)
	219	{
	220	struct onenand_chip *this = mtd->priv;
0d1ecc99	221	struct erase_info instr = {};
c438ea17 SR	222
	223	int blocks;
	224	loff_t ofs;
	225	int blocksize = 1 << this->erase_shift;
	226	int start_block, end_block;
	227	size_t retlen;
	228	u_char *buf;
	229	u_char *verify_buf;
	230	int ret;
	231
	232	buf = malloc(blocksize);
	233	if (!buf) {
	234	printf("Not enough malloc space available!\n");
	235	return -1;
	236	}
	237
	238	verify_buf = malloc(blocksize);
	239	if (!verify_buf) {
	240	printf("Not enough malloc space available!\n");
	241	return -1;
	242	}
	243
	244	start_block = start >> this->erase_shift;
	245	end_block = (start + size) >> this->erase_shift;
	246
	247	/* Protect boot-loader from badblock testing */
	248	if (start_block < 2)
	249	start_block = 2;
	250
	251	if (end_block > (mtd->size >> this->erase_shift))
	252	end_block = mtd->size >> this->erase_shift;
	253
	254	blocks = start_block;
	255	ofs = start;
	256	while (blocks < end_block) {
	257	printf("\rTesting block %d at 0x%x", (u32)(ofs >> this->erase_shift), (u32)ofs);
	258
dfe64e2c	259	ret = mtd_block_isbad(mtd, ofs);
c438ea17 SR	260	if (ret) {
	261	printf("Skip erase bad block %d at 0x%x\n",
	262	(u32)(ofs >> this->erase_shift), (u32)ofs);
	263	goto next;
	264	}
	265
	266	instr.addr = ofs;
	267	instr.len = blocksize;
dfe64e2c	268	ret = mtd_erase(mtd, &instr);
c438ea17 SR	269	if (ret) {
	270	printk("Erase failed 0x%x, %d\n", (u32)ofs, ret);
	271	goto next;
	272	}
	273
dfe64e2c	274	ret = mtd_write(mtd, ofs, blocksize, &retlen, buf);
c438ea17 SR	275	if (ret) {
	276	printk("Write failed 0x%x, %d\n", (u32)ofs, ret);
	277	goto next;
	278	}
	279
dfe64e2c	280	ret = mtd_read(mtd, ofs, blocksize, &retlen, verify_buf);
c438ea17 SR	281	if (ret) {
	282	printk("Read failed 0x%x, %d\n", (u32)ofs, ret);
	283	goto next;
	284	}
	285
	286	if (memcmp(buf, verify_buf, blocksize))
	287	printk("\nRead/Write test failed at 0x%x\n", (u32)ofs);
	288
	289	next:
	290	ofs += blocksize;
	291	blocks++;
	292	}
	293	printf("...Done\n");
	294
	295	free(buf);
	296	free(verify_buf);
	297
	298	return 0;
	299	}
	300
	301	static int onenand_dump(struct mtd_info *mtd, ulong off, int only_oob)
	302	{
	303	int i;
	304	u_char datbuf, oobbuf, *p;
	305	struct mtd_oob_ops ops;
	306	loff_t addr;
	307
	308	datbuf = malloc(mtd->writesize + mtd->oobsize);
	309	oobbuf = malloc(mtd->oobsize);
	310	if (!datbuf \|\| !oobbuf) {
	311	puts("No memory for page buffer\n");
	312	return 1;
	313	}
	314	off &= ~(mtd->writesize - 1);
	315	addr = (loff_t) off;
	316	memset(&ops, 0, sizeof(ops));
	317	ops.datbuf = datbuf;
a430b137	318	ops.oobbuf = oobbuf;
c438ea17 SR	319	ops.len = mtd->writesize;
	320	ops.ooblen = mtd->oobsize;
	321	ops.retlen = 0;
dfe64e2c	322	i = mtd_read_oob(mtd, addr, &ops);
c438ea17 SR	323	if (i < 0) {
	324	printf("Error (%d) reading page %08lx\n", i, off);
	325	free(datbuf);
	326	free(oobbuf);
	327	return 1;
	328	}
	329	printf("Page %08lx dump:\n", off);
	330	i = mtd->writesize >> 4;
	331	p = datbuf;
	332
	333	while (i--) {
	334	if (!only_oob)
	335	printf("\t%02x %02x %02x %02x %02x %02x %02x %02x"
	336	" %02x %02x %02x %02x %02x %02x %02x %02x\n",
	337	p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7],
	338	p[8], p[9], p[10], p[11], p[12], p[13], p[14],
	339	p[15]);
	340	p += 16;
	341	}
	342	puts("OOB:\n");
	343	i = mtd->oobsize >> 3;
a430b137 LW	344	p = oobbuf;
a430b137 LW	345
c438ea17 SR	346	while (i--) {
	347	printf("\t%02x %02x %02x %02x %02x %02x %02x %02x\n",
	348	p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7]);
	349	p += 8;
	350	}
	351	free(datbuf);
	352	free(oobbuf);
	353
	354	return 0;
	355	}
d7e8ce10	356
09140113 SG	357	static int do_onenand_info(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	358	char *const argv[])
d7e8ce10	359	{
8cd85282 FM	360	printf("%s\n", mtd->name);
	361	return 0;
	362	}
	363
09140113 SG	364	static int do_onenand_bad(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	365	char *const argv[])
8cd85282 FM	366	{
	367	ulong ofs;
	368
	369	mtd = &onenand_mtd;
	370	/* Currently only one OneNAND device is supported */
	371	printf("\nDevice %d bad blocks:\n", 0);
	372	for (ofs = 0; ofs < mtd->size; ofs += mtd->erasesize) {
dfe64e2c	373	if (mtd_block_isbad(mtd, ofs))
8cd85282 FM	374	printf(" %08x\n", (u32)ofs);
	375	}
	376
	377	return 0;
	378	}
	379
09140113 SG	380	static int do_onenand_read(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	381	char *const argv[])
8cd85282 FM	382	{
	383	char *s;
	384	int oob = 0;
c438ea17	385	ulong addr, ofs;
8cd85282	386	size_t len;
8360b66b	387	int ret = 0;
8cd85282	388	size_t retlen = 0;
c438ea17	389
8cd85282	390	if (argc < 3)
4c12eeb8	391	return CMD_RET_USAGE;
c438ea17	392
8cd85282 FM	393	s = strchr(argv[0], '.');
	394	if ((s != NULL) && (!strcmp(s, ".oob")))
	395	oob = 1;
d7e8ce10	396
7e5f460e	397	addr = (ulong)hextoul(argv[1], NULL);
d7e8ce10	398
8cd85282	399	printf("\nOneNAND read: ");
09c32807	400	if (arg_off_size_onenand(argc - 2, argv + 2, &ofs, &len) != 0)
8cd85282	401	return 1;
c438ea17	402
8cd85282	403	ret = onenand_block_read(ofs, len, &retlen, (u8 *)addr, oob);
c438ea17	404
6d813197	405	printf(" %zu bytes read: %s\n", retlen, ret ? "ERROR" : "OK");
d7e8ce10	406
8cd85282 FM	407	return ret == 0 ? 0 : 1;
8cd85282 FM	408	}
a9da2b41	409
09140113 SG	410	static int do_onenand_write(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	411	char *const argv[])
8cd85282 FM	412	{
	413	ulong addr, ofs;
	414	size_t len;
41c86240	415	int ret = 0, withoob = 0;
8cd85282	416	size_t retlen = 0;
d7e8ce10	417
8cd85282	418	if (argc < 3)
4c12eeb8	419	return CMD_RET_USAGE;
d7e8ce10	420
41c86240 LW	421	if (strncmp(argv[0] + 6, "yaffs", 5) == 0)
	422	withoob = 1;
	423
7e5f460e	424	addr = (ulong)hextoul(argv[1], NULL);
8cd85282 FM	425
8cd85282 FM	426	printf("\nOneNAND write: ");
09c32807	427	if (arg_off_size_onenand(argc - 2, argv + 2, &ofs, &len) != 0)
8cd85282	428	return 1;
d7e8ce10	429
41c86240	430	ret = onenand_block_write(ofs, len, &retlen, (u8 *)addr, withoob);
d7e8ce10	431
6d813197	432	printf(" %zu bytes written: %s\n", retlen, ret ? "ERROR" : "OK");
c438ea17	433
8cd85282 FM	434	return ret == 0 ? 0 : 1;
	435	}
	436
09140113 SG	437	static int do_onenand_erase(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	438	char *const argv[])
8cd85282 FM	439	{
	440	ulong ofs;
	441	int ret = 0;
	442	size_t len;
	443	int force;
	444
	445	/*
	446	* Syntax is:
	447	* 0 1 2 3 4
	448	* onenand erase [force] [off size]
	449	*/
	450	argc--;
	451	argv++;
	452	if (argc)
	453	{
	454	if (!strcmp("force", argv[0]))
	455	{
	456	force = 1;
	457	argc--;
	458	argv++;
d7e8ce10	459	}
8cd85282 FM	460	}
8cd85282 FM	461	printf("\nOneNAND erase: ");
d7e8ce10	462
8cd85282	463	/* skip first two or three arguments, look for offset and size */
09c32807	464	if (arg_off_size_onenand(argc, argv, &ofs, &len) != 0)
8cd85282	465	return 1;
bfd7f386	466
8cd85282	467	ret = onenand_block_erase(ofs, len, force);
bfd7f386	468
8cd85282	469	printf("%s\n", ret ? "ERROR" : "OK");
d7e8ce10	470
8cd85282 FM	471	return ret == 0 ? 0 : 1;
8cd85282 FM	472	}
d7e8ce10	473
09140113 SG	474	static int do_onenand_test(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	475	char *const argv[])
8cd85282 FM	476	{
	477	ulong ofs;
	478	int ret = 0;
	479	size_t len;
d7e8ce10	480
8cd85282 FM	481	/*
	482	* Syntax is:
	483	* 0 1 2 3 4
	484	* onenand test [force] [off size]
	485	*/
d7e8ce10	486
8cd85282	487	printf("\nOneNAND test: ");
d7e8ce10	488
8cd85282	489	/* skip first two or three arguments, look for offset and size */
09c32807	490	if (arg_off_size_onenand(argc - 1, argv + 1, &ofs, &len) != 0)
8cd85282	491	return 1;
d7e8ce10	492
8cd85282	493	ret = onenand_block_test(ofs, len);
d7e8ce10	494
8cd85282	495	printf("%s\n", ret ? "ERROR" : "OK");
d7e8ce10	496
8cd85282 FM	497	return ret == 0 ? 0 : 1;
8cd85282 FM	498	}
bfd7f386	499
09140113 SG	500	static int do_onenand_dump(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	501	char *const argv[])
8cd85282 FM	502	{
	503	ulong ofs;
	504	int ret = 0;
	505	char *s;
d7e8ce10	506
8cd85282	507	if (argc < 2)
4c12eeb8	508	return CMD_RET_USAGE;
8cd85282 FM	509
8cd85282 FM	510	s = strchr(argv[0], '.');
7e5f460e	511	ofs = (int)hextoul(argv[1], NULL);
8cd85282 FM	512
	513	if (s != NULL && strcmp(s, ".oob") == 0)
	514	ret = onenand_dump(mtd, ofs, 1);
	515	else
	516	ret = onenand_dump(mtd, ofs, 0);
d7e8ce10	517
8cd85282 FM	518	return ret == 0 ? 1 : 0;
	519	}
	520
09140113 SG	521	static int do_onenand_markbad(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	522	char *const argv[])
8cd85282 FM	523	{
	524	int ret = 0;
	525	ulong addr;
	526
	527	argc -= 2;
	528	argv += 2;
	529
	530	if (argc <= 0)
4c12eeb8	531	return CMD_RET_USAGE;
d7e8ce10	532
8cd85282	533	while (argc > 0) {
7e5f460e	534	addr = hextoul(*argv, NULL);
8cd85282	535
dfe64e2c	536	if (mtd_block_markbad(mtd, addr)) {
8cd85282 FM	537	printf("block 0x%08lx NOT marked "
	538	"as bad! ERROR %d\n",
	539	addr, ret);
	540	ret = 1;
	541	} else {
	542	printf("block 0x%08lx successfully "
	543	"marked as bad\n",
	544	addr);
	545	}
	546	--argc;
	547	++argv;
	548	}
	549	return ret;
	550	}
	551
09140113	552	static struct cmd_tbl cmd_onenand_sub[] = {
8cd85282 FM	553	U_BOOT_CMD_MKENT(info, 1, 0, do_onenand_info, "", ""),
	554	U_BOOT_CMD_MKENT(bad, 1, 0, do_onenand_bad, "", ""),
	555	U_BOOT_CMD_MKENT(read, 4, 0, do_onenand_read, "", ""),
	556	U_BOOT_CMD_MKENT(write, 4, 0, do_onenand_write, "", ""),
41c86240	557	U_BOOT_CMD_MKENT(write.yaffs, 4, 0, do_onenand_write, "", ""),
8cd85282 FM	558	U_BOOT_CMD_MKENT(erase, 3, 0, do_onenand_erase, "", ""),
	559	U_BOOT_CMD_MKENT(test, 3, 0, do_onenand_test, "", ""),
	560	U_BOOT_CMD_MKENT(dump, 2, 0, do_onenand_dump, "", ""),
	561	U_BOOT_CMD_MKENT(markbad, CONFIG_SYS_MAXARGS, 0, do_onenand_markbad, "", ""),
	562	};
	563
09140113 SG	564	static int do_onenand(struct cmd_tbl *cmdtp, int flag, int argc,
09140113 SG	565	char *const argv[])
8cd85282	566	{
09140113	567	struct cmd_tbl *c;
8cd85282	568
57ff9f24	569	if (argc < 2)
4c12eeb8	570	return CMD_RET_USAGE;
57ff9f24	571
8cd85282 FM	572	mtd = &onenand_mtd;
	573
	574	/* Strip off leading 'onenand' command argument */
	575	argc--;
	576	argv++;
c438ea17	577
8cd85282 FM	578	c = find_cmd_tbl(argv[0], &cmd_onenand_sub[0], ARRAY_SIZE(cmd_onenand_sub));
8cd85282 FM	579
47e26b1b WD	580	if (c)
	581	return c->cmd(cmdtp, flag, argc, argv);
	582	else
4c12eeb8	583	return CMD_RET_USAGE;
d7e8ce10 KP	584	}
	585
	586	U_BOOT_CMD(
8360b66b	587	onenand, CONFIG_SYS_MAXARGS, 1, do_onenand,
2fb2604d	588	"OneNAND sub-system",
c438ea17 SR	589	"info - show available OneNAND devices\n"
	590	"onenand bad - show bad blocks\n"
	591	"onenand read[.oob] addr off size\n"
41c86240	592	"onenand write[.yaffs] addr off size\n"
c438ea17 SR	593	" read/write 'size' bytes starting at offset 'off'\n"
	594	" to/from memory address 'addr', skipping bad blocks.\n"
	595	"onenand erase [force] [off size] - erase 'size' bytes from\n"
	596	"onenand test [off size] - test 'size' bytes from\n"
	597	" offset 'off' (entire device if not specified)\n"
	598	"onenand dump[.oob] off - dump page\n"
a89c33db	599	"onenand markbad off [...] - mark bad block(s) at offset (UNSAFE)"
d7e8ce10	600	);