1a995c792SAkinobu Mita /* 2a995c792SAkinobu Mita * Copyright (C) 2006-2008 Nokia Corporation 3a995c792SAkinobu Mita * 4a995c792SAkinobu Mita * This program is free software; you can redistribute it and/or modify it 5a995c792SAkinobu Mita * under the terms of the GNU General Public License version 2 as published by 6a995c792SAkinobu Mita * the Free Software Foundation. 7a995c792SAkinobu Mita * 8a995c792SAkinobu Mita * This program is distributed in the hope that it will be useful, but WITHOUT 9a995c792SAkinobu Mita * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 10a995c792SAkinobu Mita * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for 11a995c792SAkinobu Mita * more details. 12a995c792SAkinobu Mita * 13a995c792SAkinobu Mita * You should have received a copy of the GNU General Public License along with 14a995c792SAkinobu Mita * this program; see the file COPYING. If not, write to the Free Software 15a995c792SAkinobu Mita * Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. 16a995c792SAkinobu Mita * 17a995c792SAkinobu Mita * Test random reads, writes and erases on MTD device. 18a995c792SAkinobu Mita * 19a995c792SAkinobu Mita * Author: Adrian Hunter <ext-adrian.hunter@nokia.com> 20a995c792SAkinobu Mita */ 21a995c792SAkinobu Mita 22a995c792SAkinobu Mita #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 23a995c792SAkinobu Mita 24a995c792SAkinobu Mita #include <linux/init.h> 25a995c792SAkinobu Mita #include <linux/module.h> 26a995c792SAkinobu Mita #include <linux/moduleparam.h> 27a995c792SAkinobu Mita #include <linux/err.h> 28a995c792SAkinobu Mita #include <linux/mtd/mtd.h> 29a995c792SAkinobu Mita #include <linux/slab.h> 30a995c792SAkinobu Mita #include <linux/sched.h> 31a995c792SAkinobu Mita #include <linux/vmalloc.h> 32a995c792SAkinobu Mita #include <linux/random.h> 33a995c792SAkinobu Mita 345a78df69SAkinobu Mita #include "mtd_test.h" 355a78df69SAkinobu Mita 36a995c792SAkinobu Mita static int dev = -EINVAL; 37a995c792SAkinobu Mita module_param(dev, int, S_IRUGO); 38a995c792SAkinobu Mita MODULE_PARM_DESC(dev, "MTD device number to use"); 39a995c792SAkinobu Mita 40a995c792SAkinobu Mita static int count = 10000; 41a995c792SAkinobu Mita module_param(count, int, S_IRUGO); 42a995c792SAkinobu Mita MODULE_PARM_DESC(count, "Number of operations to do (default is 10000)"); 43a995c792SAkinobu Mita 44a995c792SAkinobu Mita static struct mtd_info *mtd; 45a995c792SAkinobu Mita static unsigned char *writebuf; 46a995c792SAkinobu Mita static unsigned char *readbuf; 47a995c792SAkinobu Mita static unsigned char *bbt; 48a995c792SAkinobu Mita static int *offsets; 49a995c792SAkinobu Mita 50a995c792SAkinobu Mita static int pgsize; 51a995c792SAkinobu Mita static int bufsize; 52a995c792SAkinobu Mita static int ebcnt; 53a995c792SAkinobu Mita static int pgcnt; 54a995c792SAkinobu Mita 55a995c792SAkinobu Mita static int rand_eb(void) 56a995c792SAkinobu Mita { 57a995c792SAkinobu Mita unsigned int eb; 58a995c792SAkinobu Mita 59a995c792SAkinobu Mita again: 60a995c792SAkinobu Mita eb = prandom_u32(); 61a995c792SAkinobu Mita /* Read or write up 2 eraseblocks at a time - hence 'ebcnt - 1' */ 62a995c792SAkinobu Mita eb %= (ebcnt - 1); 63a995c792SAkinobu Mita if (bbt[eb]) 64a995c792SAkinobu Mita goto again; 65a995c792SAkinobu Mita return eb; 66a995c792SAkinobu Mita } 67a995c792SAkinobu Mita 68a995c792SAkinobu Mita static int rand_offs(void) 69a995c792SAkinobu Mita { 70a995c792SAkinobu Mita unsigned int offs; 71a995c792SAkinobu Mita 72a995c792SAkinobu Mita offs = prandom_u32(); 73a995c792SAkinobu Mita offs %= bufsize; 74a995c792SAkinobu Mita return offs; 75a995c792SAkinobu Mita } 76a995c792SAkinobu Mita 77a995c792SAkinobu Mita static int rand_len(int offs) 78a995c792SAkinobu Mita { 79a995c792SAkinobu Mita unsigned int len; 80a995c792SAkinobu Mita 81a995c792SAkinobu Mita len = prandom_u32(); 82a995c792SAkinobu Mita len %= (bufsize - offs); 83a995c792SAkinobu Mita return len; 84a995c792SAkinobu Mita } 85a995c792SAkinobu Mita 86a995c792SAkinobu Mita static int do_read(void) 87a995c792SAkinobu Mita { 88a995c792SAkinobu Mita int eb = rand_eb(); 89a995c792SAkinobu Mita int offs = rand_offs(); 90abc173adSAkinobu Mita int len = rand_len(offs); 91a995c792SAkinobu Mita loff_t addr; 92a995c792SAkinobu Mita 93a995c792SAkinobu Mita if (bbt[eb + 1]) { 94a995c792SAkinobu Mita if (offs >= mtd->erasesize) 95a995c792SAkinobu Mita offs -= mtd->erasesize; 96a995c792SAkinobu Mita if (offs + len > mtd->erasesize) 97a995c792SAkinobu Mita len = mtd->erasesize - offs; 98a995c792SAkinobu Mita } 99b9da8baeSBrian Norris addr = (loff_t)eb * mtd->erasesize + offs; 100abc173adSAkinobu Mita return mtdtest_read(mtd, addr, len, readbuf); 101a995c792SAkinobu Mita } 102a995c792SAkinobu Mita 103a995c792SAkinobu Mita static int do_write(void) 104a995c792SAkinobu Mita { 105a995c792SAkinobu Mita int eb = rand_eb(), offs, err, len; 106a995c792SAkinobu Mita loff_t addr; 107a995c792SAkinobu Mita 108a995c792SAkinobu Mita offs = offsets[eb]; 109a995c792SAkinobu Mita if (offs >= mtd->erasesize) { 1105a78df69SAkinobu Mita err = mtdtest_erase_eraseblock(mtd, eb); 111a995c792SAkinobu Mita if (err) 112a995c792SAkinobu Mita return err; 113a995c792SAkinobu Mita offs = offsets[eb] = 0; 114a995c792SAkinobu Mita } 115a995c792SAkinobu Mita len = rand_len(offs); 116a995c792SAkinobu Mita len = ((len + pgsize - 1) / pgsize) * pgsize; 117a995c792SAkinobu Mita if (offs + len > mtd->erasesize) { 118a995c792SAkinobu Mita if (bbt[eb + 1]) 119a995c792SAkinobu Mita len = mtd->erasesize - offs; 120a995c792SAkinobu Mita else { 1215a78df69SAkinobu Mita err = mtdtest_erase_eraseblock(mtd, eb + 1); 122a995c792SAkinobu Mita if (err) 123a995c792SAkinobu Mita return err; 124a995c792SAkinobu Mita offsets[eb + 1] = 0; 125a995c792SAkinobu Mita } 126a995c792SAkinobu Mita } 127b9da8baeSBrian Norris addr = (loff_t)eb * mtd->erasesize + offs; 1285a78df69SAkinobu Mita err = mtdtest_write(mtd, addr, len, writebuf); 1298a9f4aa3SAkinobu Mita if (unlikely(err)) 130a995c792SAkinobu Mita return err; 131a995c792SAkinobu Mita offs += len; 132a995c792SAkinobu Mita while (offs > mtd->erasesize) { 133a995c792SAkinobu Mita offsets[eb++] = mtd->erasesize; 134a995c792SAkinobu Mita offs -= mtd->erasesize; 135a995c792SAkinobu Mita } 136a995c792SAkinobu Mita offsets[eb] = offs; 137a995c792SAkinobu Mita return 0; 138a995c792SAkinobu Mita } 139a995c792SAkinobu Mita 140a995c792SAkinobu Mita static int do_operation(void) 141a995c792SAkinobu Mita { 142a995c792SAkinobu Mita if (prandom_u32() & 1) 143a995c792SAkinobu Mita return do_read(); 144a995c792SAkinobu Mita else 145a995c792SAkinobu Mita return do_write(); 146a995c792SAkinobu Mita } 147a995c792SAkinobu Mita 148a995c792SAkinobu Mita static int __init mtd_stresstest_init(void) 149a995c792SAkinobu Mita { 150a995c792SAkinobu Mita int err; 151a995c792SAkinobu Mita int i, op; 152a995c792SAkinobu Mita uint64_t tmp; 153a995c792SAkinobu Mita 154a995c792SAkinobu Mita printk(KERN_INFO "\n"); 155a995c792SAkinobu Mita printk(KERN_INFO "=================================================\n"); 156a995c792SAkinobu Mita 157a995c792SAkinobu Mita if (dev < 0) { 158a995c792SAkinobu Mita pr_info("Please specify a valid mtd-device via module parameter\n"); 159a995c792SAkinobu Mita pr_crit("CAREFUL: This test wipes all data on the specified MTD device!\n"); 160a995c792SAkinobu Mita return -EINVAL; 161a995c792SAkinobu Mita } 162a995c792SAkinobu Mita 163a995c792SAkinobu Mita pr_info("MTD device: %d\n", dev); 164a995c792SAkinobu Mita 165a995c792SAkinobu Mita mtd = get_mtd_device(NULL, dev); 166a995c792SAkinobu Mita if (IS_ERR(mtd)) { 167a995c792SAkinobu Mita err = PTR_ERR(mtd); 168a995c792SAkinobu Mita pr_err("error: cannot get MTD device\n"); 169a995c792SAkinobu Mita return err; 170a995c792SAkinobu Mita } 171a995c792SAkinobu Mita 172a995c792SAkinobu Mita if (mtd->writesize == 1) { 173a995c792SAkinobu Mita pr_info("not NAND flash, assume page size is 512 " 174a995c792SAkinobu Mita "bytes.\n"); 175a995c792SAkinobu Mita pgsize = 512; 176a995c792SAkinobu Mita } else 177a995c792SAkinobu Mita pgsize = mtd->writesize; 178a995c792SAkinobu Mita 179a995c792SAkinobu Mita tmp = mtd->size; 180a995c792SAkinobu Mita do_div(tmp, mtd->erasesize); 181a995c792SAkinobu Mita ebcnt = tmp; 182a995c792SAkinobu Mita pgcnt = mtd->erasesize / pgsize; 183a995c792SAkinobu Mita 184a995c792SAkinobu Mita pr_info("MTD device size %llu, eraseblock size %u, " 185a995c792SAkinobu Mita "page size %u, count of eraseblocks %u, pages per " 186a995c792SAkinobu Mita "eraseblock %u, OOB size %u\n", 187a995c792SAkinobu Mita (unsigned long long)mtd->size, mtd->erasesize, 188a995c792SAkinobu Mita pgsize, ebcnt, pgcnt, mtd->oobsize); 189a995c792SAkinobu Mita 190a995c792SAkinobu Mita if (ebcnt < 2) { 191a995c792SAkinobu Mita pr_err("error: need at least 2 eraseblocks\n"); 192a995c792SAkinobu Mita err = -ENOSPC; 193a995c792SAkinobu Mita goto out_put_mtd; 194a995c792SAkinobu Mita } 195a995c792SAkinobu Mita 196a995c792SAkinobu Mita /* Read or write up 2 eraseblocks at a time */ 197a995c792SAkinobu Mita bufsize = mtd->erasesize * 2; 198a995c792SAkinobu Mita 199a995c792SAkinobu Mita err = -ENOMEM; 200a995c792SAkinobu Mita readbuf = vmalloc(bufsize); 201a995c792SAkinobu Mita writebuf = vmalloc(bufsize); 202a995c792SAkinobu Mita offsets = kmalloc(ebcnt * sizeof(int), GFP_KERNEL); 203a995c792SAkinobu Mita if (!readbuf || !writebuf || !offsets) 204a995c792SAkinobu Mita goto out; 205a995c792SAkinobu Mita for (i = 0; i < ebcnt; i++) 206a995c792SAkinobu Mita offsets[i] = mtd->erasesize; 207a995c792SAkinobu Mita prandom_bytes(writebuf, bufsize); 208a995c792SAkinobu Mita 2095a78df69SAkinobu Mita bbt = kzalloc(ebcnt, GFP_KERNEL); 2105a78df69SAkinobu Mita if (!bbt) 2115a78df69SAkinobu Mita goto out; 2125a78df69SAkinobu Mita err = mtdtest_scan_for_bad_eraseblocks(mtd, bbt, 0, ebcnt); 213a995c792SAkinobu Mita if (err) 214a995c792SAkinobu Mita goto out; 215a995c792SAkinobu Mita 216a995c792SAkinobu Mita /* Do operations */ 217a995c792SAkinobu Mita pr_info("doing operations\n"); 218a995c792SAkinobu Mita for (op = 0; op < count; op++) { 219a995c792SAkinobu Mita if ((op & 1023) == 0) 220a995c792SAkinobu Mita pr_info("%d operations done\n", op); 221a995c792SAkinobu Mita err = do_operation(); 222a995c792SAkinobu Mita if (err) 223a995c792SAkinobu Mita goto out; 224a995c792SAkinobu Mita cond_resched(); 225a995c792SAkinobu Mita } 226a995c792SAkinobu Mita pr_info("finished, %d operations done\n", op); 227a995c792SAkinobu Mita 228a995c792SAkinobu Mita out: 229a995c792SAkinobu Mita kfree(offsets); 230a995c792SAkinobu Mita kfree(bbt); 231a995c792SAkinobu Mita vfree(writebuf); 232a995c792SAkinobu Mita vfree(readbuf); 233a995c792SAkinobu Mita out_put_mtd: 234a995c792SAkinobu Mita put_mtd_device(mtd); 235a995c792SAkinobu Mita if (err) 236a995c792SAkinobu Mita pr_info("error %d occurred\n", err); 237a995c792SAkinobu Mita printk(KERN_INFO "=================================================\n"); 238a995c792SAkinobu Mita return err; 239a995c792SAkinobu Mita } 240a995c792SAkinobu Mita module_init(mtd_stresstest_init); 241a995c792SAkinobu Mita 242a995c792SAkinobu Mita static void __exit mtd_stresstest_exit(void) 243a995c792SAkinobu Mita { 244a995c792SAkinobu Mita return; 245a995c792SAkinobu Mita } 246a995c792SAkinobu Mita module_exit(mtd_stresstest_exit); 247a995c792SAkinobu Mita 248a995c792SAkinobu Mita MODULE_DESCRIPTION("Stress test module"); 249a995c792SAkinobu Mita MODULE_AUTHOR("Adrian Hunter"); 250a995c792SAkinobu Mita MODULE_LICENSE("GPL"); 251