| // SPDX-License-Identifier: GPL-2.0-only |
| /* |
| * Copyright (C) 2006-2008 Artem Bityutskiy |
| * Copyright (C) 2006-2008 Jarkko Lavinen |
| * Copyright (C) 2006-2008 Adrian Hunter |
| * |
| * Authors: Artem Bityutskiy, Jarkko Lavinen, Adria Hunter |
| * |
| * WARNING: this test program may kill your flash and your device. Do not |
| * use it unless you know what you do. Authors are not responsible for any |
| * damage caused by this program. |
| */ |
| |
| #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt |
| |
| #include <linux/init.h> |
| #include <linux/ktime.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> |
| #include "mtd_test.h" |
| |
| #define RETRIES 3 |
| |
| static int eb = 8; |
| module_param(eb, int, S_IRUGO); |
| MODULE_PARM_DESC(eb, "eraseblock number within the selected MTD device"); |
| |
| static int ebcnt = 32; |
| module_param(ebcnt, int, S_IRUGO); |
| MODULE_PARM_DESC(ebcnt, "number of consecutive eraseblocks to torture"); |
| |
| static int pgcnt; |
| module_param(pgcnt, int, S_IRUGO); |
| MODULE_PARM_DESC(pgcnt, "number of pages per eraseblock to torture (0 => all)"); |
| |
| static int dev = -EINVAL; |
| module_param(dev, int, S_IRUGO); |
| MODULE_PARM_DESC(dev, "MTD device number to use"); |
| |
| static int gran = 512; |
| module_param(gran, int, S_IRUGO); |
| MODULE_PARM_DESC(gran, "how often the status information should be printed"); |
| |
| static int check = 1; |
| module_param(check, int, S_IRUGO); |
| MODULE_PARM_DESC(check, "if the written data should be checked"); |
| |
| static unsigned int cycles_count; |
| module_param(cycles_count, uint, S_IRUGO); |
| MODULE_PARM_DESC(cycles_count, "how many erase cycles to do " |
| "(infinite by default)"); |
| |
| static struct mtd_info *mtd; |
| |
| /* This buffer contains 0x555555...0xAAAAAA... pattern */ |
| static unsigned char *patt_5A5; |
| /* This buffer contains 0xAAAAAA...0x555555... pattern */ |
| static unsigned char *patt_A5A; |
| /* This buffer contains all 0xFF bytes */ |
| static unsigned char *patt_FF; |
| /* This a temporary buffer is use when checking data */ |
| static unsigned char *check_buf; |
| /* How many erase cycles were done */ |
| static unsigned int erase_cycles; |
| |
| static int pgsize; |
| static ktime_t start, finish; |
| |
| static void report_corrupt(unsigned char *read, unsigned char *written); |
| |
| static inline void start_timing(void) |
| { |
| start = ktime_get(); |
| } |
| |
| static inline void stop_timing(void) |
| { |
| finish = ktime_get(); |
| } |
| |
| /* |
| * Check that the contents of eraseblock number @enbum is equivalent to the |
| * @buf buffer. |
| */ |
| static inline int check_eraseblock(int ebnum, unsigned char *buf) |
| { |
| int err, retries = 0; |
| size_t read; |
| loff_t addr = (loff_t)ebnum * mtd->erasesize; |
| size_t len = mtd->erasesize; |
| |
| if (pgcnt) { |
| addr = (loff_t)(ebnum + 1) * mtd->erasesize - pgcnt * pgsize; |
| len = pgcnt * pgsize; |
| } |
| |
| retry: |
| err = mtd_read(mtd, addr, len, &read, check_buf); |
| if (mtd_is_bitflip(err)) |
| pr_err("single bit flip occurred at EB %d " |
| "MTD reported that it was fixed.\n", ebnum); |
| else if (err) { |
| pr_err("error %d while reading EB %d, " |
| "read %zd\n", err, ebnum, read); |
| return err; |
| } |
| |
| if (read != len) { |
| pr_err("failed to read %zd bytes from EB %d, " |
| "read only %zd, but no error reported\n", |
| len, ebnum, read); |
| return -EIO; |
| } |
| |
| if (memcmp(buf, check_buf, len)) { |
| pr_err("read wrong data from EB %d\n", ebnum); |
| report_corrupt(check_buf, buf); |
| |
| if (retries++ < RETRIES) { |
| /* Try read again */ |
| yield(); |
| pr_info("re-try reading data from EB %d\n", |
| ebnum); |
| goto retry; |
| } else { |
| pr_info("retried %d times, still errors, " |
| "give-up\n", RETRIES); |
| return -EINVAL; |
| } |
| } |
| |
| if (retries != 0) |
| pr_info("only attempt number %d was OK (!!!)\n", |
| retries); |
| |
| return 0; |
| } |
| |
| static inline int write_pattern(int ebnum, void *buf) |
| { |
| int err; |
| size_t written; |
| loff_t addr = (loff_t)ebnum * mtd->erasesize; |
| size_t len = mtd->erasesize; |
| |
| if (pgcnt) { |
| addr = (loff_t)(ebnum + 1) * mtd->erasesize - pgcnt * pgsize; |
| len = pgcnt * pgsize; |
| } |
| err = mtd_write(mtd, addr, len, &written, buf); |
| if (err) { |
| pr_err("error %d while writing EB %d, written %zd" |
| " bytes\n", err, ebnum, written); |
| return err; |
| } |
| if (written != len) { |
| pr_info("written only %zd bytes of %zd, but no error" |
| " reported\n", written, len); |
| return -EIO; |
| } |
| |
| return 0; |
| } |
| |
| static int __init tort_init(void) |
| { |
| int err = 0, i, infinite = !cycles_count; |
| unsigned char *bad_ebs; |
| |
| printk(KERN_INFO "\n"); |
| printk(KERN_INFO "=================================================\n"); |
| pr_info("Warning: this program is trying to wear out your " |
| "flash, stop it if this is not wanted.\n"); |
| |
| if (dev < 0) { |
| pr_info("Please specify a valid mtd-device via module parameter\n"); |
| pr_crit("CAREFUL: This test wipes all data on the specified MTD device!\n"); |
| return -EINVAL; |
| } |
| |
| pr_info("MTD device: %d\n", dev); |
| pr_info("torture %d eraseblocks (%d-%d) of mtd%d\n", |
| ebcnt, eb, eb + ebcnt - 1, dev); |
| if (pgcnt) |
| pr_info("torturing just %d pages per eraseblock\n", |
| pgcnt); |
| pr_info("write verify %s\n", check ? "enabled" : "disabled"); |
| |
| mtd = get_mtd_device(NULL, dev); |
| if (IS_ERR(mtd)) { |
| err = PTR_ERR(mtd); |
| pr_err("error: cannot get MTD device\n"); |
| return err; |
| } |
| |
| if (mtd->writesize == 1) { |
| pr_info("not NAND flash, assume page size is 512 " |
| "bytes.\n"); |
| pgsize = 512; |
| } else |
| pgsize = mtd->writesize; |
| |
| if (pgcnt && (pgcnt > mtd->erasesize / pgsize || pgcnt < 0)) { |
| pr_err("error: invalid pgcnt value %d\n", pgcnt); |
| goto out_mtd; |
| } |
| |
| err = -ENOMEM; |
| patt_5A5 = kmalloc(mtd->erasesize, GFP_KERNEL); |
| if (!patt_5A5) |
| goto out_mtd; |
| |
| patt_A5A = kmalloc(mtd->erasesize, GFP_KERNEL); |
| if (!patt_A5A) |
| goto out_patt_5A5; |
| |
| patt_FF = kmalloc(mtd->erasesize, GFP_KERNEL); |
| if (!patt_FF) |
| goto out_patt_A5A; |
| |
| check_buf = kmalloc(mtd->erasesize, GFP_KERNEL); |
| if (!check_buf) |
| goto out_patt_FF; |
| |
| bad_ebs = kzalloc(ebcnt, GFP_KERNEL); |
| if (!bad_ebs) |
| goto out_check_buf; |
| |
| /* Initialize patterns */ |
| memset(patt_FF, 0xFF, mtd->erasesize); |
| for (i = 0; i < mtd->erasesize / pgsize; i++) { |
| if (!(i & 1)) { |
| memset(patt_5A5 + i * pgsize, 0x55, pgsize); |
| memset(patt_A5A + i * pgsize, 0xAA, pgsize); |
| } else { |
| memset(patt_5A5 + i * pgsize, 0xAA, pgsize); |
| memset(patt_A5A + i * pgsize, 0x55, pgsize); |
| } |
| } |
| |
| err = mtdtest_scan_for_bad_eraseblocks(mtd, bad_ebs, eb, ebcnt); |
| if (err) |
| goto out; |
| |
| start_timing(); |
| while (1) { |
| int i; |
| void *patt; |
| |
| err = mtdtest_erase_good_eraseblocks(mtd, bad_ebs, eb, ebcnt); |
| if (err) |
| goto out; |
| |
| /* Check if the eraseblocks contain only 0xFF bytes */ |
| if (check) { |
| for (i = eb; i < eb + ebcnt; i++) { |
| if (bad_ebs[i - eb]) |
| continue; |
| err = check_eraseblock(i, patt_FF); |
| if (err) { |
| pr_info("verify failed" |
| " for 0xFF... pattern\n"); |
| goto out; |
| } |
| |
| err = mtdtest_relax(); |
| if (err) |
| goto out; |
| } |
| } |
| |
| /* Write the pattern */ |
| for (i = eb; i < eb + ebcnt; i++) { |
| if (bad_ebs[i - eb]) |
| continue; |
| if ((eb + erase_cycles) & 1) |
| patt = patt_5A5; |
| else |
| patt = patt_A5A; |
| err = write_pattern(i, patt); |
| if (err) |
| goto out; |
| |
| err = mtdtest_relax(); |
| if (err) |
| goto out; |
| } |
| |
| /* Verify what we wrote */ |
| if (check) { |
| for (i = eb; i < eb + ebcnt; i++) { |
| if (bad_ebs[i - eb]) |
| continue; |
| if ((eb + erase_cycles) & 1) |
| patt = patt_5A5; |
| else |
| patt = patt_A5A; |
| err = check_eraseblock(i, patt); |
| if (err) { |
| pr_info("verify failed for %s" |
| " pattern\n", |
| ((eb + erase_cycles) & 1) ? |
| "0x55AA55..." : "0xAA55AA..."); |
| goto out; |
| } |
| |
| err = mtdtest_relax(); |
| if (err) |
| goto out; |
| } |
| } |
| |
| erase_cycles += 1; |
| |
| if (erase_cycles % gran == 0) { |
| long ms; |
| |
| stop_timing(); |
| ms = ktime_ms_delta(finish, start); |
| pr_info("%08u erase cycles done, took %lu " |
| "milliseconds (%lu seconds)\n", |
| erase_cycles, ms, ms / 1000); |
| start_timing(); |
| } |
| |
| if (!infinite && --cycles_count == 0) |
| break; |
| } |
| out: |
| |
| pr_info("finished after %u erase cycles\n", |
| erase_cycles); |
| kfree(bad_ebs); |
| out_check_buf: |
| kfree(check_buf); |
| out_patt_FF: |
| kfree(patt_FF); |
| out_patt_A5A: |
| kfree(patt_A5A); |
| out_patt_5A5: |
| kfree(patt_5A5); |
| out_mtd: |
| put_mtd_device(mtd); |
| if (err) |
| pr_info("error %d occurred during torturing\n", err); |
| printk(KERN_INFO "=================================================\n"); |
| return err; |
| } |
| module_init(tort_init); |
| |
| static void __exit tort_exit(void) |
| { |
| return; |
| } |
| module_exit(tort_exit); |
| |
| static int countdiffs(unsigned char *buf, unsigned char *check_buf, |
| unsigned offset, unsigned len, unsigned *bytesp, |
| unsigned *bitsp); |
| static void print_bufs(unsigned char *read, unsigned char *written, int start, |
| int len); |
| |
| /* |
| * Report the detailed information about how the read EB differs from what was |
| * written. |
| */ |
| static void report_corrupt(unsigned char *read, unsigned char *written) |
| { |
| int i; |
| int bytes, bits, pages, first; |
| int offset, len; |
| size_t check_len = mtd->erasesize; |
| |
| if (pgcnt) |
| check_len = pgcnt * pgsize; |
| |
| bytes = bits = pages = 0; |
| for (i = 0; i < check_len; i += pgsize) |
| if (countdiffs(written, read, i, pgsize, &bytes, |
| &bits) >= 0) |
| pages++; |
| |
| pr_info("verify fails on %d pages, %d bytes/%d bits\n", |
| pages, bytes, bits); |
| pr_info("The following is a list of all differences between" |
| " what was read from flash and what was expected\n"); |
| |
| for (i = 0; i < check_len; i += pgsize) { |
| cond_resched(); |
| bytes = bits = 0; |
| first = countdiffs(written, read, i, pgsize, &bytes, |
| &bits); |
| if (first < 0) |
| continue; |
| |
| printk("-------------------------------------------------------" |
| "----------------------------------\n"); |
| |
| pr_info("Page %zd has %d bytes/%d bits failing verify," |
| " starting at offset 0x%x\n", |
| (mtd->erasesize - check_len + i) / pgsize, |
| bytes, bits, first); |
| |
| offset = first & ~0x7; |
| len = ((first + bytes) | 0x7) + 1 - offset; |
| |
| print_bufs(read, written, offset, len); |
| } |
| } |
| |
| static void print_bufs(unsigned char *read, unsigned char *written, int start, |
| int len) |
| { |
| int i = 0, j1, j2; |
| char *diff; |
| |
| printk("Offset Read Written\n"); |
| while (i < len) { |
| printk("0x%08x: ", start + i); |
| diff = " "; |
| for (j1 = 0; j1 < 8 && i + j1 < len; j1++) { |
| printk(" %02x", read[start + i + j1]); |
| if (read[start + i + j1] != written[start + i + j1]) |
| diff = "***"; |
| } |
| |
| while (j1 < 8) { |
| printk(" "); |
| j1 += 1; |
| } |
| |
| printk(" %s ", diff); |
| |
| for (j2 = 0; j2 < 8 && i + j2 < len; j2++) |
| printk(" %02x", written[start + i + j2]); |
| printk("\n"); |
| i += 8; |
| } |
| } |
| |
| /* |
| * Count the number of differing bytes and bits and return the first differing |
| * offset. |
| */ |
| static int countdiffs(unsigned char *buf, unsigned char *check_buf, |
| unsigned offset, unsigned len, unsigned *bytesp, |
| unsigned *bitsp) |
| { |
| unsigned i, bit; |
| int first = -1; |
| |
| for (i = offset; i < offset + len; i++) |
| if (buf[i] != check_buf[i]) { |
| first = i; |
| break; |
| } |
| |
| while (i < offset + len) { |
| if (buf[i] != check_buf[i]) { |
| (*bytesp)++; |
| bit = 1; |
| while (bit < 256) { |
| if ((buf[i] & bit) != (check_buf[i] & bit)) |
| (*bitsp)++; |
| bit <<= 1; |
| } |
| } |
| i++; |
| } |
| |
| return first; |
| } |
| |
| MODULE_DESCRIPTION("Eraseblock torturing module"); |
| MODULE_AUTHOR("Artem Bityutskiy, Jarkko Lavinen, Adrian Hunter"); |
| MODULE_LICENSE("GPL"); |