[deliverable/linux.git] / drivers / mtd / tests / mtd_subpagetest.c

/*
 * Copyright (C) 2006-2007 Nokia Corporation
 *
 * 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.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; see the file COPYING. If not, write to the Free Software
 * Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 * Test sub-page read and write on MTD device.
 * Author: Adrian Hunter <ext-adrian.hunter@nokia.com>
 *
 */

#include <linux/init.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/err.h>
#include <linux/mtd/mtd.h>
#include <linux/slab.h>
#include <linux/sched.h>

#define PRINT_PREF KERN_INFO "mtd_subpagetest: "

static int dev = -EINVAL;
module_param(dev, int, S_IRUGO);
MODULE_PARM_DESC(dev, "MTD device number to use");

static struct mtd_info *mtd;
static unsigned char *writebuf;
static unsigned char *readbuf;
static unsigned char *bbt;

static int subpgsize;
static int bufsize;
static int ebcnt;
static int pgcnt;
static int errcnt;
static unsigned long next = 1;

static inline unsigned int simple_rand(void)
{
	next = next * 1103515245 + 12345;
	return (unsigned int)((next / 65536) % 32768);
}

static inline void simple_srand(unsigned long seed)
{
	next = seed;
}

static void set_random_data(unsigned char *buf, size_t len)
{
	size_t i;

	for (i = 0; i < len; ++i)
		buf[i] = simple_rand();
}

static inline void clear_data(unsigned char *buf, size_t len)
{
	memset(buf, 0, len);
}

static int erase_eraseblock(int ebnum)
{
	int err;
	struct erase_info ei;
	loff_t addr = ebnum * mtd->erasesize;

	memset(&ei, 0, sizeof(struct erase_info));
	ei.mtd  = mtd;
	ei.addr = addr;
	ei.len  = mtd->erasesize;

	err = mtd_erase(mtd, &ei);
	if (err) {
		printk(PRINT_PREF "error %d while erasing EB %d\n", err, ebnum);
		return err;
	}

	if (ei.state == MTD_ERASE_FAILED) {
		printk(PRINT_PREF "some erase error occurred at EB %d\n",
		       ebnum);
		return -EIO;
	}

	return 0;
}

static int erase_whole_device(void)
{
	int err;
	unsigned int i;

	printk(PRINT_PREF "erasing whole device\n");
	for (i = 0; i < ebcnt; ++i) {
		if (bbt[i])
			continue;
		err = erase_eraseblock(i);
		if (err)
			return err;
		cond_resched();
	}
	printk(PRINT_PREF "erased %u eraseblocks\n", i);
	return 0;
}

static int write_eraseblock(int ebnum)
{
	size_t written;
	int err = 0;
	loff_t addr = ebnum * mtd->erasesize;

	set_random_data(writebuf, subpgsize);
	err = mtd_write(mtd, addr, subpgsize, &written, writebuf);
	if (unlikely(err || written != subpgsize)) {
		printk(PRINT_PREF "error: write failed at %#llx\n",
		       (long long)addr);
		if (written != subpgsize) {
			printk(PRINT_PREF "  write size: %#x\n", subpgsize);
			printk(PRINT_PREF "  written: %#zx\n", written);
		}
		return err ? err : -1;
	}

	addr += subpgsize;

	set_random_data(writebuf, subpgsize);
	err = mtd_write(mtd, addr, subpgsize, &written, writebuf);
	if (unlikely(err || written != subpgsize)) {
		printk(PRINT_PREF "error: write failed at %#llx\n",
		       (long long)addr);
		if (written != subpgsize) {
			printk(PRINT_PREF "  write size: %#x\n", subpgsize);
			printk(PRINT_PREF "  written: %#zx\n", written);
		}
		return err ? err : -1;
	}

	return err;
}

static int write_eraseblock2(int ebnum)
{
	size_t written;
	int err = 0, k;
	loff_t addr = ebnum * mtd->erasesize;

	for (k = 1; k < 33; ++k) {
		if (addr + (subpgsize * k) > (ebnum + 1) * mtd->erasesize)
			break;
		set_random_data(writebuf, subpgsize * k);
		err = mtd_write(mtd, addr, subpgsize * k, &written, writebuf);
		if (unlikely(err || written != subpgsize * k)) {
			printk(PRINT_PREF "error: write failed at %#llx\n",
			       (long long)addr);
			if (written != subpgsize) {
				printk(PRINT_PREF "  write size: %#x\n",
				       subpgsize * k);
				printk(PRINT_PREF "  written: %#08zx\n",
				       written);
			}
			return err ? err : -1;
		}
		addr += subpgsize * k;
	}

	return err;
}

static void print_subpage(unsigned char *p)
{
	int i, j;

	for (i = 0; i < subpgsize; ) {
		for (j = 0; i < subpgsize && j < 32; ++i, ++j)
			printk("%02x", *p++);
		printk("\n");
	}
}

static int verify_eraseblock(int ebnum)
{
	size_t read;
	int err = 0;
	loff_t addr = ebnum * mtd->erasesize;

	set_random_data(writebuf, subpgsize);
	clear_data(readbuf, subpgsize);
	err = mtd_read(mtd, addr, subpgsize, &read, readbuf);
	if (unlikely(err || read != subpgsize)) {
		if (mtd_is_bitflip(err) && read == subpgsize) {
			printk(PRINT_PREF "ECC correction at %#llx\n",
			       (long long)addr);
			err = 0;
		} else {
			printk(PRINT_PREF "error: read failed at %#llx\n",
			       (long long)addr);
			return err ? err : -1;
		}
	}
	if (unlikely(memcmp(readbuf, writebuf, subpgsize))) {
		printk(PRINT_PREF "error: verify failed at %#llx\n",
		       (long long)addr);
		printk(PRINT_PREF "------------- written----------------\n");
		print_subpage(writebuf);
		printk(PRINT_PREF "------------- read ------------------\n");
		print_subpage(readbuf);
		printk(PRINT_PREF "-------------------------------------\n");
		errcnt += 1;
	}

	addr += subpgsize;

	set_random_data(writebuf, subpgsize);
	clear_data(readbuf, subpgsize);
	err = mtd_read(mtd, addr, subpgsize, &read, readbuf);
	if (unlikely(err || read != subpgsize)) {
		if (mtd_is_bitflip(err) && read == subpgsize) {
			printk(PRINT_PREF "ECC correction at %#llx\n",
			       (long long)addr);
			err = 0;
		} else {
			printk(PRINT_PREF "error: read failed at %#llx\n",
			       (long long)addr);
			return err ? err : -1;
		}
	}
	if (unlikely(memcmp(readbuf, writebuf, subpgsize))) {
		printk(PRINT_PREF "error: verify failed at %#llx\n",
		       (long long)addr);
		printk(PRINT_PREF "------------- written----------------\n");
		print_subpage(writebuf);
		printk(PRINT_PREF "------------- read ------------------\n");
		print_subpage(readbuf);
		printk(PRINT_PREF "-------------------------------------\n");
		errcnt += 1;
	}

	return err;
}

static int verify_eraseblock2(int ebnum)
{
	size_t read;
	int err = 0, k;
	loff_t addr = ebnum * mtd->erasesize;

	for (k = 1; k < 33; ++k) {
		if (addr + (subpgsize * k) > (ebnum + 1) * mtd->erasesize)
			break;
		set_random_data(writebuf, subpgsize * k);
		clear_data(readbuf, subpgsize * k);
		err = mtd_read(mtd, addr, subpgsize * k, &read, readbuf);
		if (unlikely(err || read != subpgsize * k)) {
			if (mtd_is_bitflip(err) && read == subpgsize * k) {
				printk(PRINT_PREF "ECC correction at %#llx\n",
				       (long long)addr);
				err = 0;
			} else {
				printk(PRINT_PREF "error: read failed at "
				       "%#llx\n", (long long)addr);
				return err ? err : -1;
			}
		}
		if (unlikely(memcmp(readbuf, writebuf, subpgsize * k))) {
			printk(PRINT_PREF "error: verify failed at %#llx\n",
			       (long long)addr);
			errcnt += 1;
		}
		addr += subpgsize * k;
	}

	return err;
}

static int verify_eraseblock_ff(int ebnum)
{
	uint32_t j;
	size_t read;
	int err = 0;
	loff_t addr = ebnum * mtd->erasesize;

	memset(writebuf, 0xff, subpgsize);
	for (j = 0; j < mtd->erasesize / subpgsize; ++j) {
		clear_data(readbuf, subpgsize);
		err = mtd_read(mtd, addr, subpgsize, &read, readbuf);
		if (unlikely(err || read != subpgsize)) {
			if (mtd_is_bitflip(err) && read == subpgsize) {
				printk(PRINT_PREF "ECC correction at %#llx\n",
				       (long long)addr);
				err = 0;
			} else {
				printk(PRINT_PREF "error: read failed at "
				       "%#llx\n", (long long)addr);
				return err ? err : -1;
			}
		}
		if (unlikely(memcmp(readbuf, writebuf, subpgsize))) {
			printk(PRINT_PREF "error: verify 0xff failed at "
			       "%#llx\n", (long long)addr);
			errcnt += 1;
		}
		addr += subpgsize;
	}

	return err;
}

static int verify_all_eraseblocks_ff(void)
{
	int err;
	unsigned int i;

	printk(PRINT_PREF "verifying all eraseblocks for 0xff\n");
	for (i = 0; i < ebcnt; ++i) {
		if (bbt[i])
			continue;
		err = verify_eraseblock_ff(i);
		if (err)
			return err;
		if (i % 256 == 0)
			printk(PRINT_PREF "verified up to eraseblock %u\n", i);
		cond_resched();
	}
	printk(PRINT_PREF "verified %u eraseblocks\n", i);
	return 0;
}

static int is_block_bad(int ebnum)
{
	loff_t addr = ebnum * mtd->erasesize;
	int ret;

	ret = mtd_block_isbad(mtd, addr);
	if (ret)
		printk(PRINT_PREF "block %d is bad\n", ebnum);
	return ret;
}

static int scan_for_bad_eraseblocks(void)
{
	int i, bad = 0;

	bbt = kzalloc(ebcnt, GFP_KERNEL);
	if (!bbt) {
		printk(PRINT_PREF "error: cannot allocate memory\n");
		return -ENOMEM;
	}

	printk(PRINT_PREF "scanning for bad eraseblocks\n");
	for (i = 0; i < ebcnt; ++i) {
		bbt[i] = is_block_bad(i) ? 1 : 0;
		if (bbt[i])
			bad += 1;
		cond_resched();
	}
	printk(PRINT_PREF "scanned %d eraseblocks, %d are bad\n", i, bad);
	return 0;
}

static int __init mtd_subpagetest_init(void)
{
	int err = 0;
	uint32_t i;
	uint64_t tmp;

	printk(KERN_INFO "\n");
	printk(KERN_INFO "=================================================\n");

	if (dev < 0) {
		printk(PRINT_PREF "Please specify a valid mtd-device via module paramter\n");
		printk(KERN_CRIT "CAREFUL: This test wipes all data on the specified MTD device!\n");
		return -EINVAL;
	}

	printk(PRINT_PREF "MTD device: %d\n", dev);

	mtd = get_mtd_device(NULL, dev);
	if (IS_ERR(mtd)) {
		err = PTR_ERR(mtd);
		printk(PRINT_PREF "error: cannot get MTD device\n");
		return err;
	}

	if (mtd->type != MTD_NANDFLASH) {
		printk(PRINT_PREF "this test requires NAND flash\n");
		goto out;
	}

	subpgsize = mtd->writesize >> mtd->subpage_sft;
	tmp = mtd->size;
	do_div(tmp, mtd->erasesize);
	ebcnt = tmp;
	pgcnt = mtd->erasesize / mtd->writesize;

	printk(PRINT_PREF "MTD device size %llu, eraseblock size %u, "
	       "page size %u, subpage size %u, count of eraseblocks %u, "
	       "pages per eraseblock %u, OOB size %u\n",
	       (unsigned long long)mtd->size, mtd->erasesize,
	       mtd->writesize, subpgsize, ebcnt, pgcnt, mtd->oobsize);

	err = -ENOMEM;
	bufsize = subpgsize * 32;
	writebuf = kmalloc(bufsize, GFP_KERNEL);
	if (!writebuf) {
		printk(PRINT_PREF "error: cannot allocate memory\n");
		goto out;
	}
	readbuf = kmalloc(bufsize, GFP_KERNEL);
	if (!readbuf) {
		printk(PRINT_PREF "error: cannot allocate memory\n");
		goto out;
	}

	err = scan_for_bad_eraseblocks();
	if (err)
		goto out;

	err = erase_whole_device();
	if (err)
		goto out;

	printk(PRINT_PREF "writing whole device\n");
	simple_srand(1);
	for (i = 0; i < ebcnt; ++i) {
		if (bbt[i])
			continue;
		err = write_eraseblock(i);
		if (unlikely(err))
			goto out;
		if (i % 256 == 0)
			printk(PRINT_PREF "written up to eraseblock %u\n", i);
		cond_resched();
	}
	printk(PRINT_PREF "written %u eraseblocks\n", i);

	simple_srand(1);
	printk(PRINT_PREF "verifying all eraseblocks\n");
	for (i = 0; i < ebcnt; ++i) {
		if (bbt[i])
			continue;
		err = verify_eraseblock(i);
		if (unlikely(err))
			goto out;
		if (i % 256 == 0)
			printk(PRINT_PREF "verified up to eraseblock %u\n", i);
		cond_resched();
	}
	printk(PRINT_PREF "verified %u eraseblocks\n", i);

	err = erase_whole_device();
	if (err)
		goto out;

	err = verify_all_eraseblocks_ff();
	if (err)
		goto out;

	/* Write all eraseblocks */
	simple_srand(3);
	printk(PRINT_PREF "writing whole device\n");
	for (i = 0; i < ebcnt; ++i) {
		if (bbt[i])
			continue;
		err = write_eraseblock2(i);
		if (unlikely(err))
			goto out;
		if (i % 256 == 0)
			printk(PRINT_PREF "written up to eraseblock %u\n", i);
		cond_resched();
	}
	printk(PRINT_PREF "written %u eraseblocks\n", i);

	/* Check all eraseblocks */
	simple_srand(3);
	printk(PRINT_PREF "verifying all eraseblocks\n");
	for (i = 0; i < ebcnt; ++i) {
		if (bbt[i])
			continue;
		err = verify_eraseblock2(i);
		if (unlikely(err))
			goto out;
		if (i % 256 == 0)
			printk(PRINT_PREF "verified up to eraseblock %u\n", i);
		cond_resched();
	}
	printk(PRINT_PREF "verified %u eraseblocks\n", i);

	err = erase_whole_device();
	if (err)
		goto out;

	err = verify_all_eraseblocks_ff();
	if (err)
		goto out;

	printk(PRINT_PREF "finished with %d errors\n", errcnt);

out:
	kfree(bbt);
	kfree(readbuf);
	kfree(writebuf);
	put_mtd_device(mtd);
	if (err)
		printk(PRINT_PREF "error %d occurred\n", err);
	printk(KERN_INFO "=================================================\n");
	return err;
}
module_init(mtd_subpagetest_init);

static void __exit mtd_subpagetest_exit(void)
{
	return;
}
module_exit(mtd_subpagetest_exit);

MODULE_DESCRIPTION("Subpage test module");
MODULE_AUTHOR("Adrian Hunter");
MODULE_LICENSE("GPL");
Commit	Line	Data
bf60862a AB	1	/*
	2	* Copyright (C) 2006-2007 Nokia Corporation
	3	*
	4	* This program is free software; you can redistribute it and/or modify it
	5	* under the terms of the GNU General Public License version 2 as published by
	6	* the Free Software Foundation.
	7	*
	8	* This program is distributed in the hope that it will be useful, but WITHOUT
	9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	10	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
	11	* more details.
	12	*
	13	* You should have received a copy of the GNU General Public License along with
	14	* this program; see the file COPYING. If not, write to the Free Software
	15	* Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
	16	*
	17	* Test sub-page read and write on MTD device.
	18	* Author: Adrian Hunter <ext-adrian.hunter@nokia.com>
	19	*
	20	*/
	21
	22	#include <linux/init.h>
	23	#include <linux/module.h>
	24	#include <linux/moduleparam.h>
	25	#include <linux/err.h>
	26	#include <linux/mtd/mtd.h>
5a0e3ad6	27	#include <linux/slab.h>
bf60862a AB	28	#include <linux/sched.h>
	29
	30	#define PRINT_PREF KERN_INFO "mtd_subpagetest: "
	31
7406060e	32	static int dev = -EINVAL;
bf60862a AB	33	module_param(dev, int, S_IRUGO);
	34	MODULE_PARM_DESC(dev, "MTD device number to use");
	35
	36	static struct mtd_info *mtd;
	37	static unsigned char *writebuf;
	38	static unsigned char *readbuf;
	39	static unsigned char *bbt;
	40
	41	static int subpgsize;
	42	static int bufsize;
	43	static int ebcnt;
	44	static int pgcnt;
	45	static int errcnt;
	46	static unsigned long next = 1;
	47
	48	static inline unsigned int simple_rand(void)
	49	{
	50	next = next * 1103515245 + 12345;
	51	return (unsigned int)((next / 65536) % 32768);
	52	}
	53
	54	static inline void simple_srand(unsigned long seed)
	55	{
	56	next = seed;
	57	}
	58
	59	static void set_random_data(unsigned char *buf, size_t len)
	60	{
	61	size_t i;
	62
	63	for (i = 0; i < len; ++i)
	64	buf[i] = simple_rand();
	65	}
	66
	67	static inline void clear_data(unsigned char *buf, size_t len)
	68	{
	69	memset(buf, 0, len);
	70	}
	71
	72	static int erase_eraseblock(int ebnum)
	73	{
	74	int err;
	75	struct erase_info ei;
	76	loff_t addr = ebnum * mtd->erasesize;
	77
	78	memset(&ei, 0, sizeof(struct erase_info));
	79	ei.mtd = mtd;
	80	ei.addr = addr;
	81	ei.len = mtd->erasesize;
	82
7e1f0dc0	83	err = mtd_erase(mtd, &ei);
bf60862a AB	84	if (err) {
	85	printk(PRINT_PREF "error %d while erasing EB %d\n", err, ebnum);
	86	return err;
	87	}
	88
	89	if (ei.state == MTD_ERASE_FAILED) {
	90	printk(PRINT_PREF "some erase error occurred at EB %d\n",
	91	ebnum);
	92	return -EIO;
	93	}
	94
	95	return 0;
	96	}
	97
	98	static int erase_whole_device(void)
	99	{
	100	int err;
	101	unsigned int i;
	102
	103	printk(PRINT_PREF "erasing whole device\n");
	104	for (i = 0; i < ebcnt; ++i) {
	105	if (bbt[i])
	106	continue;
	107	err = erase_eraseblock(i);
	108	if (err)
	109	return err;
	110	cond_resched();
	111	}
	112	printk(PRINT_PREF "erased %u eraseblocks\n", i);
	113	return 0;
	114	}
	115
	116	static int write_eraseblock(int ebnum)
	117	{
30fa9848	118	size_t written;
bf60862a AB	119	int err = 0;
	120	loff_t addr = ebnum * mtd->erasesize;
	121
	122	set_random_data(writebuf, subpgsize);
eda95cbf	123	err = mtd_write(mtd, addr, subpgsize, &written, writebuf);
bf60862a AB	124	if (unlikely(err \|\| written != subpgsize)) {
	125	printk(PRINT_PREF "error: write failed at %#llx\n",
	126	(long long)addr);
	127	if (written != subpgsize) {
	128	printk(PRINT_PREF " write size: %#x\n", subpgsize);
647b0d38	129	printk(PRINT_PREF " written: %#zx\n", written);
bf60862a AB	130	}
	131	return err ? err : -1;
	132	}
	133
	134	addr += subpgsize;
	135
	136	set_random_data(writebuf, subpgsize);
eda95cbf	137	err = mtd_write(mtd, addr, subpgsize, &written, writebuf);
bf60862a AB	138	if (unlikely(err \|\| written != subpgsize)) {
	139	printk(PRINT_PREF "error: write failed at %#llx\n",
	140	(long long)addr);
	141	if (written != subpgsize) {
	142	printk(PRINT_PREF " write size: %#x\n", subpgsize);
647b0d38	143	printk(PRINT_PREF " written: %#zx\n", written);
bf60862a AB	144	}
	145	return err ? err : -1;
	146	}
	147
	148	return err;
	149	}
	150
	151	static int write_eraseblock2(int ebnum)
	152	{
30fa9848	153	size_t written;
bf60862a AB	154	int err = 0, k;
	155	loff_t addr = ebnum * mtd->erasesize;
	156
	157	for (k = 1; k < 33; ++k) {
	158	if (addr + (subpgsize * k) > (ebnum + 1) * mtd->erasesize)
	159	break;
	160	set_random_data(writebuf, subpgsize * k);
eda95cbf	161	err = mtd_write(mtd, addr, subpgsize * k, &written, writebuf);
bf60862a AB	162	if (unlikely(err \|\| written != subpgsize * k)) {
	163	printk(PRINT_PREF "error: write failed at %#llx\n",
	164	(long long)addr);
	165	if (written != subpgsize) {
	166	printk(PRINT_PREF " write size: %#x\n",
	167	subpgsize * k);
647b0d38	168	printk(PRINT_PREF " written: %#08zx\n",
bf60862a AB	169	written);
	170	}
	171	return err ? err : -1;
	172	}
	173	addr += subpgsize * k;
	174	}
	175
	176	return err;
	177	}
	178
	179	static void print_subpage(unsigned char *p)
	180	{
	181	int i, j;
	182
	183	for (i = 0; i < subpgsize; ) {
	184	for (j = 0; i < subpgsize && j < 32; ++i, ++j)
	185	printk("%02x", *p++);
	186	printk("\n");
	187	}
	188	}
	189
	190	static int verify_eraseblock(int ebnum)
	191	{
30fa9848	192	size_t read;
bf60862a AB	193	int err = 0;
	194	loff_t addr = ebnum * mtd->erasesize;
	195
	196	set_random_data(writebuf, subpgsize);
	197	clear_data(readbuf, subpgsize);
329ad399	198	err = mtd_read(mtd, addr, subpgsize, &read, readbuf);
bf60862a	199	if (unlikely(err \|\| read != subpgsize)) {
d57f4054	200	if (mtd_is_bitflip(err) && read == subpgsize) {
bf60862a AB	201	printk(PRINT_PREF "ECC correction at %#llx\n",
	202	(long long)addr);
	203	err = 0;
	204	} else {
	205	printk(PRINT_PREF "error: read failed at %#llx\n",
	206	(long long)addr);
	207	return err ? err : -1;
	208	}
	209	}
	210	if (unlikely(memcmp(readbuf, writebuf, subpgsize))) {
	211	printk(PRINT_PREF "error: verify failed at %#llx\n",
	212	(long long)addr);
	213	printk(PRINT_PREF "------------- written----------------\n");
	214	print_subpage(writebuf);
	215	printk(PRINT_PREF "------------- read ------------------\n");
	216	print_subpage(readbuf);
	217	printk(PRINT_PREF "-------------------------------------\n");
	218	errcnt += 1;
	219	}
	220
	221	addr += subpgsize;
	222
	223	set_random_data(writebuf, subpgsize);
	224	clear_data(readbuf, subpgsize);
329ad399	225	err = mtd_read(mtd, addr, subpgsize, &read, readbuf);
bf60862a	226	if (unlikely(err \|\| read != subpgsize)) {
d57f4054	227	if (mtd_is_bitflip(err) && read == subpgsize) {
bf60862a AB	228	printk(PRINT_PREF "ECC correction at %#llx\n",
	229	(long long)addr);
	230	err = 0;
	231	} else {
	232	printk(PRINT_PREF "error: read failed at %#llx\n",
	233	(long long)addr);
	234	return err ? err : -1;
	235	}
	236	}
	237	if (unlikely(memcmp(readbuf, writebuf, subpgsize))) {
	238	printk(PRINT_PREF "error: verify failed at %#llx\n",
	239	(long long)addr);
	240	printk(PRINT_PREF "------------- written----------------\n");
	241	print_subpage(writebuf);
	242	printk(PRINT_PREF "------------- read ------------------\n");
	243	print_subpage(readbuf);
	244	printk(PRINT_PREF "-------------------------------------\n");
	245	errcnt += 1;
	246	}
	247
	248	return err;
	249	}
	250
	251	static int verify_eraseblock2(int ebnum)
	252	{
30fa9848	253	size_t read;
bf60862a AB	254	int err = 0, k;
	255	loff_t addr = ebnum * mtd->erasesize;
	256
	257	for (k = 1; k < 33; ++k) {
	258	if (addr + (subpgsize * k) > (ebnum + 1) * mtd->erasesize)
	259	break;
	260	set_random_data(writebuf, subpgsize * k);
	261	clear_data(readbuf, subpgsize * k);
329ad399	262	err = mtd_read(mtd, addr, subpgsize * k, &read, readbuf);
bf60862a	263	if (unlikely(err \|\| read != subpgsize * k)) {
d57f4054	264	if (mtd_is_bitflip(err) && read == subpgsize * k) {
bf60862a AB	265	printk(PRINT_PREF "ECC correction at %#llx\n",
	266	(long long)addr);
	267	err = 0;
	268	} else {
	269	printk(PRINT_PREF "error: read failed at "
	270	"%#llx\n", (long long)addr);
	271	return err ? err : -1;
	272	}
	273	}
	274	if (unlikely(memcmp(readbuf, writebuf, subpgsize * k))) {
	275	printk(PRINT_PREF "error: verify failed at %#llx\n",
	276	(long long)addr);
	277	errcnt += 1;
	278	}
	279	addr += subpgsize * k;
	280	}
	281
	282	return err;
	283	}
	284
	285	static int verify_eraseblock_ff(int ebnum)
	286	{
	287	uint32_t j;
30fa9848	288	size_t read;
bf60862a AB	289	int err = 0;
	290	loff_t addr = ebnum * mtd->erasesize;
	291
	292	memset(writebuf, 0xff, subpgsize);
	293	for (j = 0; j < mtd->erasesize / subpgsize; ++j) {
	294	clear_data(readbuf, subpgsize);
329ad399	295	err = mtd_read(mtd, addr, subpgsize, &read, readbuf);
bf60862a	296	if (unlikely(err \|\| read != subpgsize)) {
d57f4054	297	if (mtd_is_bitflip(err) && read == subpgsize) {
bf60862a AB	298	printk(PRINT_PREF "ECC correction at %#llx\n",
	299	(long long)addr);
	300	err = 0;
	301	} else {
	302	printk(PRINT_PREF "error: read failed at "
	303	"%#llx\n", (long long)addr);
	304	return err ? err : -1;
	305	}
	306	}
	307	if (unlikely(memcmp(readbuf, writebuf, subpgsize))) {
	308	printk(PRINT_PREF "error: verify 0xff failed at "
	309	"%#llx\n", (long long)addr);
	310	errcnt += 1;
	311	}
	312	addr += subpgsize;
	313	}
	314
	315	return err;
	316	}
	317
	318	static int verify_all_eraseblocks_ff(void)
	319	{
	320	int err;
	321	unsigned int i;
	322
	323	printk(PRINT_PREF "verifying all eraseblocks for 0xff\n");
	324	for (i = 0; i < ebcnt; ++i) {
	325	if (bbt[i])
	326	continue;
	327	err = verify_eraseblock_ff(i);
	328	if (err)
	329	return err;
	330	if (i % 256 == 0)
	331	printk(PRINT_PREF "verified up to eraseblock %u\n", i);
	332	cond_resched();
	333	}
	334	printk(PRINT_PREF "verified %u eraseblocks\n", i);
	335	return 0;
	336	}
	337
	338	static int is_block_bad(int ebnum)
	339	{
	340	loff_t addr = ebnum * mtd->erasesize;
	341	int ret;
	342
7086c19d	343	ret = mtd_block_isbad(mtd, addr);
bf60862a AB	344	if (ret)
	345	printk(PRINT_PREF "block %d is bad\n", ebnum);
	346	return ret;
	347	}
	348
	349	static int scan_for_bad_eraseblocks(void)
	350	{
	351	int i, bad = 0;
	352
2bfefa4c	353	bbt = kzalloc(ebcnt, GFP_KERNEL);
bf60862a AB	354	if (!bbt) {
	355	printk(PRINT_PREF "error: cannot allocate memory\n");
	356	return -ENOMEM;
	357	}
bf60862a AB	358
	359	printk(PRINT_PREF "scanning for bad eraseblocks\n");
	360	for (i = 0; i < ebcnt; ++i) {
	361	bbt[i] = is_block_bad(i) ? 1 : 0;
	362	if (bbt[i])
	363	bad += 1;
	364	cond_resched();
	365	}
	366	printk(PRINT_PREF "scanned %d eraseblocks, %d are bad\n", i, bad);
	367	return 0;
	368	}
	369
	370	static int __init mtd_subpagetest_init(void)
	371	{
	372	int err = 0;
	373	uint32_t i;
	374	uint64_t tmp;
	375
	376	printk(KERN_INFO "\n");
	377	printk(KERN_INFO "=================================================\n");
7406060e WS	378
	379	if (dev < 0) {
	380	printk(PRINT_PREF "Please specify a valid mtd-device via module paramter\n");
	381	printk(KERN_CRIT "CAREFUL: This test wipes all data on the specified MTD device!\n");
	382	return -EINVAL;
	383	}
	384
bf60862a AB	385	printk(PRINT_PREF "MTD device: %d\n", dev);
	386
	387	mtd = get_mtd_device(NULL, dev);
	388	if (IS_ERR(mtd)) {
	389	err = PTR_ERR(mtd);
	390	printk(PRINT_PREF "error: cannot get MTD device\n");
	391	return err;
	392	}
	393
	394	if (mtd->type != MTD_NANDFLASH) {
	395	printk(PRINT_PREF "this test requires NAND flash\n");
	396	goto out;
	397	}
	398
	399	subpgsize = mtd->writesize >> mtd->subpage_sft;
7b7e905e RT	400	tmp = mtd->size;
	401	do_div(tmp, mtd->erasesize);
	402	ebcnt = tmp;
	403	pgcnt = mtd->erasesize / mtd->writesize;
	404
bf60862a AB	405	printk(PRINT_PREF "MTD device size %llu, eraseblock size %u, "
	406	"page size %u, subpage size %u, count of eraseblocks %u, "
	407	"pages per eraseblock %u, OOB size %u\n",
	408	(unsigned long long)mtd->size, mtd->erasesize,
	409	mtd->writesize, subpgsize, ebcnt, pgcnt, mtd->oobsize);
	410
	411	err = -ENOMEM;
	412	bufsize = subpgsize * 32;
	413	writebuf = kmalloc(bufsize, GFP_KERNEL);
	414	if (!writebuf) {
	415	printk(PRINT_PREF "error: cannot allocate memory\n");
	416	goto out;
	417	}
	418	readbuf = kmalloc(bufsize, GFP_KERNEL);
	419	if (!readbuf) {
	420	printk(PRINT_PREF "error: cannot allocate memory\n");
	421	goto out;
	422	}
	423
bf60862a AB	424	err = scan_for_bad_eraseblocks();
	425	if (err)
	426	goto out;
	427
	428	err = erase_whole_device();
	429	if (err)
	430	goto out;
	431
	432	printk(PRINT_PREF "writing whole device\n");
	433	simple_srand(1);
	434	for (i = 0; i < ebcnt; ++i) {
	435	if (bbt[i])
	436	continue;
	437	err = write_eraseblock(i);
	438	if (unlikely(err))
	439	goto out;
	440	if (i % 256 == 0)
	441	printk(PRINT_PREF "written up to eraseblock %u\n", i);
	442	cond_resched();
	443	}
	444	printk(PRINT_PREF "written %u eraseblocks\n", i);
	445
	446	simple_srand(1);
	447	printk(PRINT_PREF "verifying all eraseblocks\n");
	448	for (i = 0; i < ebcnt; ++i) {
	449	if (bbt[i])
	450	continue;
	451	err = verify_eraseblock(i);
	452	if (unlikely(err))
	453	goto out;
	454	if (i % 256 == 0)
	455	printk(PRINT_PREF "verified up to eraseblock %u\n", i);
	456	cond_resched();
	457	}
	458	printk(PRINT_PREF "verified %u eraseblocks\n", i);
	459
	460	err = erase_whole_device();
	461	if (err)
	462	goto out;
	463
	464	err = verify_all_eraseblocks_ff();
	465	if (err)
	466	goto out;
	467
	468	/* Write all eraseblocks */
	469	simple_srand(3);
	470	printk(PRINT_PREF "writing whole device\n");
	471	for (i = 0; i < ebcnt; ++i) {
	472	if (bbt[i])
	473	continue;
	474	err = write_eraseblock2(i);
	475	if (unlikely(err))
	476	goto out;
	477	if (i % 256 == 0)
	478	printk(PRINT_PREF "written up to eraseblock %u\n", i);
	479	cond_resched();
	480	}
	481	printk(PRINT_PREF "written %u eraseblocks\n", i);
	482
	483	/* Check all eraseblocks */
	484	simple_srand(3);
	485	printk(PRINT_PREF "verifying all eraseblocks\n");
	486	for (i = 0; i < ebcnt; ++i) {
	487	if (bbt[i])
488	continue;
489	err = verify_eraseblock2(i);
490	if (unlikely(err))
491	goto out;
492	if (i % 256 == 0)
493	printk(PRINT_PREF "verified up to eraseblock %u\n", i);
494	cond_resched();
495	}
496	printk(PRINT_PREF "verified %u eraseblocks\n", i);
497
498	err = erase_whole_device();
499	if (err)
500	goto out;
501
502	err = verify_all_eraseblocks_ff();
503	if (err)
504	goto out;
505
506	printk(PRINT_PREF "finished with %d errors\n", errcnt);
507
508	out:
509	kfree(bbt);
510	kfree(readbuf);
511	kfree(writebuf);
512	put_mtd_device(mtd);
513	if (err)
514	printk(PRINT_PREF "error %d occurred\n", err);
515	printk(KERN_INFO "=================================================\n");
516	return err;
517	}
518	module_init(mtd_subpagetest_init);
519
520	static void __exit mtd_subpagetest_exit(void)
521	{
522	return;
523	}
524	module_exit(mtd_subpagetest_exit);
525
526	MODULE_DESCRIPTION("Subpage test module");
527	MODULE_AUTHOR("Adrian Hunter");
528	MODULE_LICENSE("GPL");