diff options
| -rw-r--r-- | miscutils/nandwrite.c | 155 |
1 files changed, 127 insertions, 28 deletions
diff --git a/miscutils/nandwrite.c b/miscutils/nandwrite.c index f422426..6c85ea3 100644 --- a/miscutils/nandwrite.c +++ b/miscutils/nandwrite.c @@ -1,5 +1,5 @@ /* - * nandwrite.c - ported to busybox from mtd-utils + * nandwrite and nanddump ported to busybox from mtd-utils * * Author: Baruch Siach <baruch@tkos.co.il>, Orex Computed Radiography * @@ -9,8 +9,10 @@ */ //applet:IF_NANDWRITE(APPLET(nandwrite, _BB_DIR_USR_SBIN, _BB_SUID_DROP)) +//applet:IF_NANDWRITE(APPLET_ODDNAME(nanddump, nandwrite, _BB_DIR_USR_SBIN, _BB_SUID_DROP, nanddump)) //kbuild:lib-$(CONFIG_NANDWRITE) += nandwrite.o +//kbuild:lib-$(CONFIG_NANDDUMP) += nandwrite.o //config:config NANDWRITE //config: bool "nandwrite" @@ -18,9 +20,13 @@ //config: depends on PLATFORM_LINUX //config: help //config: Write to the specified MTD device, with bad blocks awareness - -#include "libbb.h" -#include <mtd/mtd-user.h> +//config: +//config:config NANDDUMP +//config: bool "nanddump" +//config: default n +//config: depends on PLATFORM_LINUX +//config: help +//config: Dump the content of raw NAND chip //usage:#define nandwrite_trivial_usage //usage: "[-p] [-s ADDR] MTD_DEVICE [FILE]" @@ -30,18 +36,65 @@ //usage: "\n -p Pad to page size" //usage: "\n -s ADDR Start address" +//usage:#define nanddump_trivial_usage +//usage: "[-o] [-b] [-s ADDR] [-f FILE] MTD_DEVICE" +//usage:#define nanddump_full_usage "\n\n" +//usage: "Dump the sepcified MTD device\n" +//usage: "\nOptions:" +//usage: "\n -o Omit oob data" +//usage: "\n -b Omit bad block from the dump" +//usage: "\n -s ADDR Start address" +//usage: "\n -l LEN Length" +//usage: "\n -f FILE Dump to file ('-' for stdout)" + +#include "libbb.h" +#include <mtd/mtd-user.h> + +#define IS_NANDDUMP (ENABLE_NANDDUMP && (!ENABLE_NANDWRITE || (applet_name[4] == 'd'))) +#define IS_NANDWRITE (ENABLE_NANDWRITE && (!ENABLE_NANDDUMP || (applet_name[4] != 'd'))) + +#define OPT_p (1 << 0) /* nandwrite only */ +#define OPT_o (1 << 0) /* nanddump only */ +#define OPT_s (1 << 1) +#define OPT_b (1 << 2) +#define OPT_f (1 << 3) +#define OPT_l (1 << 4) + +#define NAND_MAX_OOBSIZE 256 +/* helper for writing out 0xff for bad blocks pad */ +static void dump_bad(struct mtd_info_user *meminfo, unsigned len, int oob) +{ + unsigned char buf[meminfo->writesize]; + unsigned count; + + /* round len to the next page */ + len = (len | ~(meminfo->writesize - 1)) + 1; + + memset(buf, 0xff, sizeof(buf)); + for (count = 0; count < len; count += meminfo->writesize) { + xwrite(STDOUT_FILENO, buf, meminfo->writesize); + if (oob) + xwrite(STDOUT_FILENO, buf, meminfo->oobsize); + } +} + static unsigned next_good_eraseblock(int fd, struct mtd_info_user *meminfo, unsigned block_offset) { while (1) { loff_t offs; - if (block_offset >= meminfo->size) - bb_error_msg_and_die("not enough space in MTD device"); + + if (block_offset >= meminfo->size) { + if (IS_NANDWRITE) + bb_error_msg_and_die("not enough space in MTD device"); + return block_offset; /* let the caller exit */ + } offs = block_offset; if (xioctl(fd, MEMGETBADBLOCK, &offs) == 0) return block_offset; /* ioctl returned 1 => "bad block" */ - printf("Skipping bad block at 0x%08x\n", block_offset); + if (IS_NANDWRITE) + printf("Skipping bad block at 0x%08x\n", block_offset); block_offset += meminfo->erasesize; } } @@ -49,31 +102,49 @@ static unsigned next_good_eraseblock(int fd, struct mtd_info_user *meminfo, int nandwrite_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int nandwrite_main(int argc UNUSED_PARAM, char **argv) { + /* Buffer for OOB data */ + unsigned char oobbuf[NAND_MAX_OOBSIZE]; unsigned opts; int fd; ssize_t cnt; - unsigned mtdoffset, meminfo_writesize, blockstart; + unsigned mtdoffset, meminfo_writesize, blockstart, limit; + unsigned end_addr = ~0; struct mtd_info_user meminfo; + struct mtd_oob_buf oob; unsigned char *filebuf; - const char *opt_s = "0"; - enum { - OPT_p = (1 << 0), - OPT_s = (1 << 1), - }; - - opt_complementary = "-1:?2"; - opts = getopt32(argv, "ps:", &opt_s); + const char *opt_s = "0", *opt_f = "-", *opt_l; + + if (IS_NANDDUMP) { + opt_complementary = "=1"; + opts = getopt32(argv, "os:bf:l:", &opt_s, &opt_f, &opt_l); + } else { /* nandwrite */ + opt_complementary = "-1:?2"; + opts = getopt32(argv, "ps:", &opt_s); + } argv += optind; - if (argv[1]) - xmove_fd(xopen_stdin(argv[1]), STDIN_FILENO); + if (IS_NANDWRITE && argv[1]) + opt_f = argv[1]; + if (!LONE_DASH(opt_f)) { + int tmp_fd = xopen(opt_f, + IS_NANDDUMP ? O_WRONLY | O_TRUNC | O_CREAT : O_RDONLY + ); + xmove_fd(tmp_fd, IS_NANDDUMP ? STDOUT_FILENO : STDIN_FILENO); + } fd = xopen(argv[0], O_RDWR); xioctl(fd, MEMGETINFO, &meminfo); - mtdoffset = bb_strtou(opt_s, NULL, 0); - if (errno) - bb_error_msg_and_die("invalid number '%s'", opt_s); + oob.start = 0; + oob.length = meminfo.oobsize; + oob.ptr = oobbuf; + + mtdoffset = xstrtou(opt_s, 0); + if (IS_NANDDUMP && (opts & OPT_l)) { + unsigned length = xstrtou(opt_l, 0); + if (length < meminfo.size - mtdoffset) + end_addr = mtdoffset + length; + } /* Pull it into a CPU register (hopefully) - smaller code that way */ meminfo_writesize = meminfo.writesize; @@ -91,20 +162,39 @@ int nandwrite_main(int argc UNUSED_PARAM, char **argv) * bad. */ tmp = next_good_eraseblock(fd, &meminfo, blockstart); - if (tmp != blockstart) /* bad block(s), advance mtdoffset */ + if (tmp != blockstart) { + /* bad block(s), advance mtdoffset */ + if (IS_NANDDUMP & !(opts & OPT_b)) { + int bad_len = MIN(tmp, end_addr) - mtdoffset; + dump_bad(&meminfo, bad_len, !(opts & OPT_o)); + } mtdoffset = tmp; + } } cnt = -1; - while (mtdoffset < meminfo.size) { + limit = MIN(meminfo.size, end_addr); + while (mtdoffset < limit) { + int input_fd = IS_NANDWRITE ? STDIN_FILENO : fd; + int output_fd = IS_NANDWRITE ? fd : STDOUT_FILENO; + blockstart = mtdoffset & ~(meminfo.erasesize - 1); if (blockstart == mtdoffset) { /* starting a new eraseblock */ mtdoffset = next_good_eraseblock(fd, &meminfo, blockstart); - printf("Writing at 0x%08x\n", mtdoffset); + if (IS_NANDWRITE) + printf("Writing at 0x%08x\n", mtdoffset); + else if (mtdoffset > blockstart) { + int bad_len = MIN(mtdoffset, limit) - blockstart; + dump_bad(&meminfo, bad_len, !(opts & OPT_o)); + } + if (mtdoffset >= limit) + break; } + xlseek(fd, mtdoffset, SEEK_SET); + /* get some more data from input */ - cnt = full_read(STDIN_FILENO, filebuf, meminfo_writesize); + cnt = full_read(input_fd, filebuf, meminfo_writesize); if (cnt == 0) { /* even with -p, we do not pad past the end of input * (-p only zero-pads last incomplete page) @@ -112,20 +202,29 @@ int nandwrite_main(int argc UNUSED_PARAM, char **argv) break; } if (cnt < meminfo_writesize) { + if (IS_NANDDUMP) + bb_error_msg_and_die("short read"); if (!(opts & OPT_p)) bb_error_msg_and_die("input size is not rounded up to page size, " "use -p to zero pad"); /* zero pad to end of write block */ memset(filebuf + cnt, 0, meminfo_writesize - cnt); } - xlseek(fd, mtdoffset, SEEK_SET); - xwrite(fd, filebuf, meminfo_writesize); + xwrite(output_fd, filebuf, meminfo_writesize); + + if (IS_NANDDUMP && !(opts & OPT_o)) { + /* Dump OOB data */ + oob.start = mtdoffset; + xioctl(fd, MEMREADOOB, &oob); + xwrite(output_fd, oobbuf, meminfo.oobsize); + } + mtdoffset += meminfo_writesize; if (cnt < meminfo_writesize) break; } - if (cnt != 0) { + if (IS_NANDWRITE && cnt != 0) { /* We filled entire MTD, but did we reach EOF on input? */ if (full_read(STDIN_FILENO, filebuf, meminfo_writesize) != 0) { /* no */ |