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
rand_eb(void)43a995c792SAkinobu Mita static int rand_eb(void)
44a995c792SAkinobu Mita {
45a995c792SAkinobu Mita unsigned int eb;
46a995c792SAkinobu Mita
47a995c792SAkinobu Mita again:
48a995c792SAkinobu Mita /* Read or write up 2 eraseblocks at a time - hence 'ebcnt - 1' */
49*8032bf12SJason A. Donenfeld eb = get_random_u32_below(ebcnt - 1);
50a995c792SAkinobu Mita if (bbt[eb])
51a995c792SAkinobu Mita goto again;
52a995c792SAkinobu Mita return eb;
53a995c792SAkinobu Mita }
54a995c792SAkinobu Mita
rand_offs(void)55a995c792SAkinobu Mita static int rand_offs(void)
56a995c792SAkinobu Mita {
57*8032bf12SJason A. Donenfeld return get_random_u32_below(bufsize);
58a995c792SAkinobu Mita }
59a995c792SAkinobu Mita
rand_len(int offs)60a995c792SAkinobu Mita static int rand_len(int offs)
61a995c792SAkinobu Mita {
62*8032bf12SJason A. Donenfeld return get_random_u32_below(bufsize - offs);
63a995c792SAkinobu Mita }
64a995c792SAkinobu Mita
do_read(void)65a995c792SAkinobu Mita static int do_read(void)
66a995c792SAkinobu Mita {
67a995c792SAkinobu Mita int eb = rand_eb();
68a995c792SAkinobu Mita int offs = rand_offs();
69abc173adSAkinobu Mita int len = rand_len(offs);
70a995c792SAkinobu Mita loff_t addr;
71a995c792SAkinobu Mita
72a995c792SAkinobu Mita if (bbt[eb + 1]) {
73a995c792SAkinobu Mita if (offs >= mtd->erasesize)
74a995c792SAkinobu Mita offs -= mtd->erasesize;
75a995c792SAkinobu Mita if (offs + len > mtd->erasesize)
76a995c792SAkinobu Mita len = mtd->erasesize - offs;
77a995c792SAkinobu Mita }
78b9da8baeSBrian Norris addr = (loff_t)eb * mtd->erasesize + offs;
79abc173adSAkinobu Mita return mtdtest_read(mtd, addr, len, readbuf);
80a995c792SAkinobu Mita }
81a995c792SAkinobu Mita
do_write(void)82a995c792SAkinobu Mita static int do_write(void)
83a995c792SAkinobu Mita {
84a995c792SAkinobu Mita int eb = rand_eb(), offs, err, len;
85a995c792SAkinobu Mita loff_t addr;
86a995c792SAkinobu Mita
87a995c792SAkinobu Mita offs = offsets[eb];
88a995c792SAkinobu Mita if (offs >= mtd->erasesize) {
895a78df69SAkinobu Mita err = mtdtest_erase_eraseblock(mtd, eb);
90a995c792SAkinobu Mita if (err)
91a995c792SAkinobu Mita return err;
92a995c792SAkinobu Mita offs = offsets[eb] = 0;
93a995c792SAkinobu Mita }
94a995c792SAkinobu Mita len = rand_len(offs);
95a995c792SAkinobu Mita len = ((len + pgsize - 1) / pgsize) * pgsize;
96a995c792SAkinobu Mita if (offs + len > mtd->erasesize) {
97a995c792SAkinobu Mita if (bbt[eb + 1])
98a995c792SAkinobu Mita len = mtd->erasesize - offs;
99a995c792SAkinobu Mita else {
1005a78df69SAkinobu Mita err = mtdtest_erase_eraseblock(mtd, eb + 1);
101a995c792SAkinobu Mita if (err)
102a995c792SAkinobu Mita return err;
103a995c792SAkinobu Mita offsets[eb + 1] = 0;
104a995c792SAkinobu Mita }
105a995c792SAkinobu Mita }
106b9da8baeSBrian Norris addr = (loff_t)eb * mtd->erasesize + offs;
1075a78df69SAkinobu Mita err = mtdtest_write(mtd, addr, len, writebuf);
1088a9f4aa3SAkinobu Mita if (unlikely(err))
109a995c792SAkinobu Mita return err;
110a995c792SAkinobu Mita offs += len;
111a995c792SAkinobu Mita while (offs > mtd->erasesize) {
112a995c792SAkinobu Mita offsets[eb++] = mtd->erasesize;
113a995c792SAkinobu Mita offs -= mtd->erasesize;
114a995c792SAkinobu Mita }
115a995c792SAkinobu Mita offsets[eb] = offs;
116a995c792SAkinobu Mita return 0;
117a995c792SAkinobu Mita }
118a995c792SAkinobu Mita
do_operation(void)119a995c792SAkinobu Mita static int do_operation(void)
120a995c792SAkinobu Mita {
121*8032bf12SJason A. Donenfeld if (get_random_u32_below(2))
122a995c792SAkinobu Mita return do_read();
123a995c792SAkinobu Mita else
124a995c792SAkinobu Mita return do_write();
125a995c792SAkinobu Mita }
126a995c792SAkinobu Mita
mtd_stresstest_init(void)127a995c792SAkinobu Mita static int __init mtd_stresstest_init(void)
128a995c792SAkinobu Mita {
129a995c792SAkinobu Mita int err;
130a995c792SAkinobu Mita int i, op;
131a995c792SAkinobu Mita uint64_t tmp;
132a995c792SAkinobu Mita
133a995c792SAkinobu Mita printk(KERN_INFO "\n");
134a995c792SAkinobu Mita printk(KERN_INFO "=================================================\n");
135a995c792SAkinobu Mita
136a995c792SAkinobu Mita if (dev < 0) {
137a995c792SAkinobu Mita pr_info("Please specify a valid mtd-device via module parameter\n");
138a995c792SAkinobu Mita pr_crit("CAREFUL: This test wipes all data on the specified MTD device!\n");
139a995c792SAkinobu Mita return -EINVAL;
140a995c792SAkinobu Mita }
141a995c792SAkinobu Mita
142a995c792SAkinobu Mita pr_info("MTD device: %d\n", dev);
143a995c792SAkinobu Mita
144a995c792SAkinobu Mita mtd = get_mtd_device(NULL, dev);
145a995c792SAkinobu Mita if (IS_ERR(mtd)) {
146a995c792SAkinobu Mita err = PTR_ERR(mtd);
147a995c792SAkinobu Mita pr_err("error: cannot get MTD device\n");
148a995c792SAkinobu Mita return err;
149a995c792SAkinobu Mita }
150a995c792SAkinobu Mita
151a995c792SAkinobu Mita if (mtd->writesize == 1) {
152a995c792SAkinobu Mita pr_info("not NAND flash, assume page size is 512 "
153a995c792SAkinobu Mita "bytes.\n");
154a995c792SAkinobu Mita pgsize = 512;
155a995c792SAkinobu Mita } else
156a995c792SAkinobu Mita pgsize = mtd->writesize;
157a995c792SAkinobu Mita
158a995c792SAkinobu Mita tmp = mtd->size;
159a995c792SAkinobu Mita do_div(tmp, mtd->erasesize);
160a995c792SAkinobu Mita ebcnt = tmp;
161a995c792SAkinobu Mita pgcnt = mtd->erasesize / pgsize;
162a995c792SAkinobu Mita
163a995c792SAkinobu Mita pr_info("MTD device size %llu, eraseblock size %u, "
164a995c792SAkinobu Mita "page size %u, count of eraseblocks %u, pages per "
165a995c792SAkinobu Mita "eraseblock %u, OOB size %u\n",
166a995c792SAkinobu Mita (unsigned long long)mtd->size, mtd->erasesize,
167a995c792SAkinobu Mita pgsize, ebcnt, pgcnt, mtd->oobsize);
168a995c792SAkinobu Mita
169a995c792SAkinobu Mita if (ebcnt < 2) {
170a995c792SAkinobu Mita pr_err("error: need at least 2 eraseblocks\n");
171a995c792SAkinobu Mita err = -ENOSPC;
172a995c792SAkinobu Mita goto out_put_mtd;
173a995c792SAkinobu Mita }
174a995c792SAkinobu Mita
175a995c792SAkinobu Mita /* Read or write up 2 eraseblocks at a time */
176a995c792SAkinobu Mita bufsize = mtd->erasesize * 2;
177a995c792SAkinobu Mita
178a995c792SAkinobu Mita err = -ENOMEM;
179a995c792SAkinobu Mita readbuf = vmalloc(bufsize);
180a995c792SAkinobu Mita writebuf = vmalloc(bufsize);
1816da2ec56SKees Cook offsets = kmalloc_array(ebcnt, sizeof(int), GFP_KERNEL);
182a995c792SAkinobu Mita if (!readbuf || !writebuf || !offsets)
183a995c792SAkinobu Mita goto out;
184a995c792SAkinobu Mita for (i = 0; i < ebcnt; i++)
185a995c792SAkinobu Mita offsets[i] = mtd->erasesize;
186197173dbSJason A. Donenfeld get_random_bytes(writebuf, bufsize);
187a995c792SAkinobu Mita
1885a78df69SAkinobu Mita bbt = kzalloc(ebcnt, GFP_KERNEL);
1895a78df69SAkinobu Mita if (!bbt)
1905a78df69SAkinobu Mita goto out;
1915a78df69SAkinobu Mita err = mtdtest_scan_for_bad_eraseblocks(mtd, bbt, 0, ebcnt);
192a995c792SAkinobu Mita if (err)
193a995c792SAkinobu Mita goto out;
194a995c792SAkinobu Mita
195a995c792SAkinobu Mita /* Do operations */
196a995c792SAkinobu Mita pr_info("doing operations\n");
197a995c792SAkinobu Mita for (op = 0; op < count; op++) {
198a995c792SAkinobu Mita if ((op & 1023) == 0)
199a995c792SAkinobu Mita pr_info("%d operations done\n", op);
200a995c792SAkinobu Mita err = do_operation();
201a995c792SAkinobu Mita if (err)
202a995c792SAkinobu Mita goto out;
2032a6a28e7SRichard Weinberger
2042a6a28e7SRichard Weinberger err = mtdtest_relax();
2052a6a28e7SRichard Weinberger if (err)
2062a6a28e7SRichard Weinberger goto out;
207a995c792SAkinobu Mita }
208a995c792SAkinobu Mita pr_info("finished, %d operations done\n", op);
209a995c792SAkinobu Mita
210a995c792SAkinobu Mita out:
211a995c792SAkinobu Mita kfree(offsets);
212a995c792SAkinobu Mita kfree(bbt);
213a995c792SAkinobu Mita vfree(writebuf);
214a995c792SAkinobu Mita vfree(readbuf);
215a995c792SAkinobu Mita out_put_mtd:
216a995c792SAkinobu Mita put_mtd_device(mtd);
217a995c792SAkinobu Mita if (err)
218a995c792SAkinobu Mita pr_info("error %d occurred\n", err);
219a995c792SAkinobu Mita printk(KERN_INFO "=================================================\n");
220a995c792SAkinobu Mita return err;
221a995c792SAkinobu Mita }
222a995c792SAkinobu Mita module_init(mtd_stresstest_init);
223a995c792SAkinobu Mita
mtd_stresstest_exit(void)224a995c792SAkinobu Mita static void __exit mtd_stresstest_exit(void)
225a995c792SAkinobu Mita {
226a995c792SAkinobu Mita return;
227a995c792SAkinobu Mita }
228a995c792SAkinobu Mita module_exit(mtd_stresstest_exit);
229a995c792SAkinobu Mita
230a995c792SAkinobu Mita MODULE_DESCRIPTION("Stress test module");
231a995c792SAkinobu Mita MODULE_AUTHOR("Adrian Hunter");
232a995c792SAkinobu Mita MODULE_LICENSE("GPL");
233