14cd10358SThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only 2a995c792SAkinobu Mita /* 3a995c792SAkinobu Mita * Copyright (C) 2006-2008 Nokia Corporation 4a995c792SAkinobu Mita * 5a995c792SAkinobu Mita * Test random reads, writes and erases on MTD device. 6a995c792SAkinobu Mita * 7a995c792SAkinobu Mita * Author: Adrian Hunter <ext-adrian.hunter@nokia.com> 8a995c792SAkinobu Mita */ 9a995c792SAkinobu Mita 10a995c792SAkinobu Mita #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 11a995c792SAkinobu Mita 12a995c792SAkinobu Mita #include <linux/init.h> 13a995c792SAkinobu Mita #include <linux/module.h> 14a995c792SAkinobu Mita #include <linux/moduleparam.h> 15a995c792SAkinobu Mita #include <linux/err.h> 16a995c792SAkinobu Mita #include <linux/mtd/mtd.h> 17a995c792SAkinobu Mita #include <linux/slab.h> 18a995c792SAkinobu Mita #include <linux/sched.h> 19a995c792SAkinobu Mita #include <linux/vmalloc.h> 20a995c792SAkinobu Mita #include <linux/random.h> 21a995c792SAkinobu Mita 225a78df69SAkinobu Mita #include "mtd_test.h" 235a78df69SAkinobu Mita 24a995c792SAkinobu Mita static int dev = -EINVAL; 25a995c792SAkinobu Mita module_param(dev, int, S_IRUGO); 26a995c792SAkinobu Mita MODULE_PARM_DESC(dev, "MTD device number to use"); 27a995c792SAkinobu Mita 28a995c792SAkinobu Mita static int count = 10000; 29a995c792SAkinobu Mita module_param(count, int, S_IRUGO); 30a995c792SAkinobu Mita MODULE_PARM_DESC(count, "Number of operations to do (default is 10000)"); 31a995c792SAkinobu Mita 32a995c792SAkinobu Mita static struct mtd_info *mtd; 33a995c792SAkinobu Mita static unsigned char *writebuf; 34a995c792SAkinobu Mita static unsigned char *readbuf; 35a995c792SAkinobu Mita static unsigned char *bbt; 36a995c792SAkinobu Mita static int *offsets; 37a995c792SAkinobu Mita 38a995c792SAkinobu Mita static int pgsize; 39a995c792SAkinobu Mita static int bufsize; 40a995c792SAkinobu Mita static int ebcnt; 41a995c792SAkinobu Mita static int pgcnt; 42a995c792SAkinobu Mita 43a995c792SAkinobu Mita static int rand_eb(void) 44a995c792SAkinobu Mita { 45a995c792SAkinobu Mita unsigned int eb; 46a995c792SAkinobu Mita 47a995c792SAkinobu Mita again: 48a995c792SAkinobu Mita eb = prandom_u32(); 49a995c792SAkinobu Mita /* Read or write up 2 eraseblocks at a time - hence 'ebcnt - 1' */ 50a995c792SAkinobu Mita eb %= (ebcnt - 1); 51a995c792SAkinobu Mita if (bbt[eb]) 52a995c792SAkinobu Mita goto again; 53a995c792SAkinobu Mita return eb; 54a995c792SAkinobu Mita } 55a995c792SAkinobu Mita 56a995c792SAkinobu Mita static int rand_offs(void) 57a995c792SAkinobu Mita { 58a995c792SAkinobu Mita unsigned int offs; 59a995c792SAkinobu Mita 60a995c792SAkinobu Mita offs = prandom_u32(); 61a995c792SAkinobu Mita offs %= bufsize; 62a995c792SAkinobu Mita return offs; 63a995c792SAkinobu Mita } 64a995c792SAkinobu Mita 65a995c792SAkinobu Mita static int rand_len(int offs) 66a995c792SAkinobu Mita { 67a995c792SAkinobu Mita unsigned int len; 68a995c792SAkinobu Mita 69a995c792SAkinobu Mita len = prandom_u32(); 70a995c792SAkinobu Mita len %= (bufsize - offs); 71a995c792SAkinobu Mita return len; 72a995c792SAkinobu Mita } 73a995c792SAkinobu Mita 74a995c792SAkinobu Mita static int do_read(void) 75a995c792SAkinobu Mita { 76a995c792SAkinobu Mita int eb = rand_eb(); 77a995c792SAkinobu Mita int offs = rand_offs(); 78abc173adSAkinobu Mita int len = rand_len(offs); 79a995c792SAkinobu Mita loff_t addr; 80a995c792SAkinobu Mita 81a995c792SAkinobu Mita if (bbt[eb + 1]) { 82a995c792SAkinobu Mita if (offs >= mtd->erasesize) 83a995c792SAkinobu Mita offs -= mtd->erasesize; 84a995c792SAkinobu Mita if (offs + len > mtd->erasesize) 85a995c792SAkinobu Mita len = mtd->erasesize - offs; 86a995c792SAkinobu Mita } 87b9da8baeSBrian Norris addr = (loff_t)eb * mtd->erasesize + offs; 88abc173adSAkinobu Mita return mtdtest_read(mtd, addr, len, readbuf); 89a995c792SAkinobu Mita } 90a995c792SAkinobu Mita 91a995c792SAkinobu Mita static int do_write(void) 92a995c792SAkinobu Mita { 93a995c792SAkinobu Mita int eb = rand_eb(), offs, err, len; 94a995c792SAkinobu Mita loff_t addr; 95a995c792SAkinobu Mita 96a995c792SAkinobu Mita offs = offsets[eb]; 97a995c792SAkinobu Mita if (offs >= mtd->erasesize) { 985a78df69SAkinobu Mita err = mtdtest_erase_eraseblock(mtd, eb); 99a995c792SAkinobu Mita if (err) 100a995c792SAkinobu Mita return err; 101a995c792SAkinobu Mita offs = offsets[eb] = 0; 102a995c792SAkinobu Mita } 103a995c792SAkinobu Mita len = rand_len(offs); 104a995c792SAkinobu Mita len = ((len + pgsize - 1) / pgsize) * pgsize; 105a995c792SAkinobu Mita if (offs + len > mtd->erasesize) { 106a995c792SAkinobu Mita if (bbt[eb + 1]) 107a995c792SAkinobu Mita len = mtd->erasesize - offs; 108a995c792SAkinobu Mita else { 1095a78df69SAkinobu Mita err = mtdtest_erase_eraseblock(mtd, eb + 1); 110a995c792SAkinobu Mita if (err) 111a995c792SAkinobu Mita return err; 112a995c792SAkinobu Mita offsets[eb + 1] = 0; 113a995c792SAkinobu Mita } 114a995c792SAkinobu Mita } 115b9da8baeSBrian Norris addr = (loff_t)eb * mtd->erasesize + offs; 1165a78df69SAkinobu Mita err = mtdtest_write(mtd, addr, len, writebuf); 1178a9f4aa3SAkinobu Mita if (unlikely(err)) 118a995c792SAkinobu Mita return err; 119a995c792SAkinobu Mita offs += len; 120a995c792SAkinobu Mita while (offs > mtd->erasesize) { 121a995c792SAkinobu Mita offsets[eb++] = mtd->erasesize; 122a995c792SAkinobu Mita offs -= mtd->erasesize; 123a995c792SAkinobu Mita } 124a995c792SAkinobu Mita offsets[eb] = offs; 125a995c792SAkinobu Mita return 0; 126a995c792SAkinobu Mita } 127a995c792SAkinobu Mita 128a995c792SAkinobu Mita static int do_operation(void) 129a995c792SAkinobu Mita { 130a995c792SAkinobu Mita if (prandom_u32() & 1) 131a995c792SAkinobu Mita return do_read(); 132a995c792SAkinobu Mita else 133a995c792SAkinobu Mita return do_write(); 134a995c792SAkinobu Mita } 135a995c792SAkinobu Mita 136a995c792SAkinobu Mita static int __init mtd_stresstest_init(void) 137a995c792SAkinobu Mita { 138a995c792SAkinobu Mita int err; 139a995c792SAkinobu Mita int i, op; 140a995c792SAkinobu Mita uint64_t tmp; 141a995c792SAkinobu Mita 142a995c792SAkinobu Mita printk(KERN_INFO "\n"); 143a995c792SAkinobu Mita printk(KERN_INFO "=================================================\n"); 144a995c792SAkinobu Mita 145a995c792SAkinobu Mita if (dev < 0) { 146a995c792SAkinobu Mita pr_info("Please specify a valid mtd-device via module parameter\n"); 147a995c792SAkinobu Mita pr_crit("CAREFUL: This test wipes all data on the specified MTD device!\n"); 148a995c792SAkinobu Mita return -EINVAL; 149a995c792SAkinobu Mita } 150a995c792SAkinobu Mita 151a995c792SAkinobu Mita pr_info("MTD device: %d\n", dev); 152a995c792SAkinobu Mita 153a995c792SAkinobu Mita mtd = get_mtd_device(NULL, dev); 154a995c792SAkinobu Mita if (IS_ERR(mtd)) { 155a995c792SAkinobu Mita err = PTR_ERR(mtd); 156a995c792SAkinobu Mita pr_err("error: cannot get MTD device\n"); 157a995c792SAkinobu Mita return err; 158a995c792SAkinobu Mita } 159a995c792SAkinobu Mita 160a995c792SAkinobu Mita if (mtd->writesize == 1) { 161a995c792SAkinobu Mita pr_info("not NAND flash, assume page size is 512 " 162a995c792SAkinobu Mita "bytes.\n"); 163a995c792SAkinobu Mita pgsize = 512; 164a995c792SAkinobu Mita } else 165a995c792SAkinobu Mita pgsize = mtd->writesize; 166a995c792SAkinobu Mita 167a995c792SAkinobu Mita tmp = mtd->size; 168a995c792SAkinobu Mita do_div(tmp, mtd->erasesize); 169a995c792SAkinobu Mita ebcnt = tmp; 170a995c792SAkinobu Mita pgcnt = mtd->erasesize / pgsize; 171a995c792SAkinobu Mita 172a995c792SAkinobu Mita pr_info("MTD device size %llu, eraseblock size %u, " 173a995c792SAkinobu Mita "page size %u, count of eraseblocks %u, pages per " 174a995c792SAkinobu Mita "eraseblock %u, OOB size %u\n", 175a995c792SAkinobu Mita (unsigned long long)mtd->size, mtd->erasesize, 176a995c792SAkinobu Mita pgsize, ebcnt, pgcnt, mtd->oobsize); 177a995c792SAkinobu Mita 178a995c792SAkinobu Mita if (ebcnt < 2) { 179a995c792SAkinobu Mita pr_err("error: need at least 2 eraseblocks\n"); 180a995c792SAkinobu Mita err = -ENOSPC; 181a995c792SAkinobu Mita goto out_put_mtd; 182a995c792SAkinobu Mita } 183a995c792SAkinobu Mita 184a995c792SAkinobu Mita /* Read or write up 2 eraseblocks at a time */ 185a995c792SAkinobu Mita bufsize = mtd->erasesize * 2; 186a995c792SAkinobu Mita 187a995c792SAkinobu Mita err = -ENOMEM; 188a995c792SAkinobu Mita readbuf = vmalloc(bufsize); 189a995c792SAkinobu Mita writebuf = vmalloc(bufsize); 1906da2ec56SKees Cook offsets = kmalloc_array(ebcnt, sizeof(int), GFP_KERNEL); 191a995c792SAkinobu Mita if (!readbuf || !writebuf || !offsets) 192a995c792SAkinobu Mita goto out; 193a995c792SAkinobu Mita for (i = 0; i < ebcnt; i++) 194a995c792SAkinobu Mita offsets[i] = mtd->erasesize; 195a995c792SAkinobu Mita prandom_bytes(writebuf, bufsize); 196a995c792SAkinobu Mita 1975a78df69SAkinobu Mita bbt = kzalloc(ebcnt, GFP_KERNEL); 1985a78df69SAkinobu Mita if (!bbt) 1995a78df69SAkinobu Mita goto out; 2005a78df69SAkinobu Mita err = mtdtest_scan_for_bad_eraseblocks(mtd, bbt, 0, ebcnt); 201a995c792SAkinobu Mita if (err) 202a995c792SAkinobu Mita goto out; 203a995c792SAkinobu Mita 204a995c792SAkinobu Mita /* Do operations */ 205a995c792SAkinobu Mita pr_info("doing operations\n"); 206a995c792SAkinobu Mita for (op = 0; op < count; op++) { 207a995c792SAkinobu Mita if ((op & 1023) == 0) 208a995c792SAkinobu Mita pr_info("%d operations done\n", op); 209a995c792SAkinobu Mita err = do_operation(); 210a995c792SAkinobu Mita if (err) 211a995c792SAkinobu Mita goto out; 2122a6a28e7SRichard Weinberger 2132a6a28e7SRichard Weinberger err = mtdtest_relax(); 2142a6a28e7SRichard Weinberger if (err) 2152a6a28e7SRichard Weinberger goto out; 216a995c792SAkinobu Mita } 217a995c792SAkinobu Mita pr_info("finished, %d operations done\n", op); 218a995c792SAkinobu Mita 219a995c792SAkinobu Mita out: 220a995c792SAkinobu Mita kfree(offsets); 221a995c792SAkinobu Mita kfree(bbt); 222a995c792SAkinobu Mita vfree(writebuf); 223a995c792SAkinobu Mita vfree(readbuf); 224a995c792SAkinobu Mita out_put_mtd: 225a995c792SAkinobu Mita put_mtd_device(mtd); 226a995c792SAkinobu Mita if (err) 227a995c792SAkinobu Mita pr_info("error %d occurred\n", err); 228a995c792SAkinobu Mita printk(KERN_INFO "=================================================\n"); 229a995c792SAkinobu Mita return err; 230a995c792SAkinobu Mita } 231a995c792SAkinobu Mita module_init(mtd_stresstest_init); 232a995c792SAkinobu Mita 233a995c792SAkinobu Mita static void __exit mtd_stresstest_exit(void) 234a995c792SAkinobu Mita { 235a995c792SAkinobu Mita return; 236a995c792SAkinobu Mita } 237a995c792SAkinobu Mita module_exit(mtd_stresstest_exit); 238a995c792SAkinobu Mita 239a995c792SAkinobu Mita MODULE_DESCRIPTION("Stress test module"); 240a995c792SAkinobu Mita MODULE_AUTHOR("Adrian Hunter"); 241a995c792SAkinobu Mita MODULE_LICENSE("GPL"); 242