6f32ea403
Baruch Siach
2010-08-25
Subject
nandwrite: new applet
Binary
No binary (build failed)
Architecture & Compiler Flags
—
Build Log
(log yok)
Commit Diff GitHub →
miscutils/nandwrite.c+128 -0
| @@ -0,0 +1,128 @@ | |
| + | /* |
| + | * nandwrite.c - ported to busybox from mtd-utils |
| + | * |
| + | * Author: Baruch Siach <[email protected]>, Orex Computed Radiography |
| + | * |
| + | * Licensed under GPLv2, see file LICENSE in this source tree. |
| + | * |
| + | * TODO: add support for large (>4GB) MTD devices |
| + | */ |
| + | |
| + | //applet:IF_NANDWRITE(APPLET(nandwrite, _BB_DIR_USR_SBIN, _BB_SUID_DROP)) |
| + | |
| + | //kbuild:lib-$(CONFIG_NANDWRITE) += nandwrite.o |
| + | |
| + | //config:config NANDWRITE |
| + | //config: bool "nandwrite" |
| + | //config: default n |
| + | //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> |
| + | |
| + | //usage:#define nandwrite_trivial_usage |
| + | //usage: "[-p] [-s ADDR] MTD_DEVICE [FILE]" |
| + | //usage:#define nandwrite_full_usage "\n\n" |
| + | //usage: "Write to the specified MTD device\n" |
| + | //usage: "\nOptions:" |
| + | //usage: "\n -p Pad to page size" |
| + | //usage: "\n -s ADDR Start address" |
| + | |
| + | 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"); |
| + | 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); |
| + | block_offset += meminfo->erasesize; |
| + | } |
| + | } |
| + | |
| + | int nandwrite_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| + | int nandwrite_main(int argc UNUSED_PARAM, char **argv) |
| + | { |
| + | unsigned opts; |
| + | int fd; |
| + | ssize_t cnt; |
| + | unsigned mtdoffset, meminfo_writesize; |
| + | struct mtd_info_user meminfo; |
| + | 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); |
| + | argv += optind; |
| + | |
| + | if (argv[1]) |
| + | xmove_fd(xopen_stdin(argv[1]), STDIN_FILENO); |
| + | |
| + | fd = xopen(argv[0], O_RDWR); |
| + | xioctl(fd, MEMGETINFO, &meminfo); |
| + | |
| + | mtdoffset = xatou(opt_s); |
| + | |
| + | /* Pull it into a CPU register (hopefully) - smaller code that way */ |
| + | meminfo_writesize = meminfo.writesize; |
| + | |
| + | if (mtdoffset & (meminfo_writesize - 1)) |
| + | bb_error_msg_and_die("start address is not page aligned"); |
| + | |
| + | filebuf = xmalloc(meminfo_writesize); |
| + | |
| + | cnt = -1; |
| + | while (mtdoffset < meminfo.size) { |
| + | unsigned 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); |
| + | } |
| + | /* get some more data from input */ |
| + | cnt = full_read(STDIN_FILENO, 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) |
| + | */ |
| + | break; |
| + | } |
| + | if (cnt < meminfo_writesize) { |
| + | 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); |
| + | mtdoffset += meminfo_writesize; |
| + | if (cnt < meminfo_writesize) |
| + | break; |
| + | } |
| + | |
| + | if (cnt != 0) { |
| + | /* We filled entire MTD, but did we reach EOF on input? */ |
| + | if (full_read(STDIN_FILENO, filebuf, meminfo_writesize) != 0) { |
| + | /* no */ |
| + | bb_error_msg_and_die("not enough space in MTD device"); |
| + | } |
| + | } |
| + | |
| + | if (ENABLE_FEATURE_CLEAN_UP) { |
| + | free(filebuf); |
| + | close(fd); |
| + | } |
| + | |
| + | return EXIT_SUCCESS; |
| + | } |