nanddump: new applet
No binary (build failed)
—
(log yok)
| @@ -1,5 +1,5 @@ | |
| /* | |
| - | * nandwrite.c - ported to busybox from mtd-utils |
| + | * nandwrite and nanddump ported to busybox from mtd-utils |
| * | |
| * Author: Baruch Siach <[email protected]>, 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 */ | |