diff options
| -rw-r--r-- | flash_unlock.c | 73 |
1 files changed, 36 insertions, 37 deletions
diff --git a/flash_unlock.c b/flash_unlock.c index 759a5bc..3bb3381 100644 --- a/flash_unlock.c +++ b/flash_unlock.c @@ -22,48 +22,51 @@ #include <sys/mount.h> #include <string.h> +#include "common.h" #include <mtd/mtd-user.h> +static void usage(int status) +{ + fprintf(status ? stderr : stdout, + "Usage: %s <mtd device> [offset] [block count]\n\n" + "If offset is not specified, it defaults to 0.\n" + "If block count is not specified, it defaults to all blocks.\n", + PROGRAM_NAME); + exit(status); +} + int main(int argc, char *argv[]) { int fd, request; struct mtd_info_user mtdInfo; struct erase_info_user mtdLockInfo; int count; + const char *dev; + + /* Parse command line options */ + if (argc < 2 || argc > 4) + usage(1); + if (!strcmp(argv[1], "-h") || !strcmp(argv[1], "--help")) + usage(0); - /* - * Parse command line options - */ - if (argc < 2) { - fprintf(stderr, "USAGE: %s <mtd device> <offset> <block count>\n", PROGRAM_NAME); - exit(1); - } else if (strncmp(argv[1], "/dev/mtd", 8) != 0) { - fprintf(stderr, "'%s' is not a MTD device. Must specify mtd device: /dev/mtd?\n", argv[1]); - exit(1); - } + dev = argv[1]; - fd = open(argv[1], O_RDWR); - if (fd < 0) { - fprintf(stderr, "Could not open mtd device: %s\n", argv[1]); - exit(1); - } + /* Get the device info to compare to command line sizes */ + fd = open(dev, O_RDWR); + if (fd < 0) + sys_errmsg_die("could not open: %s", dev); - if (ioctl(fd, MEMGETINFO, &mtdInfo)) { - fprintf(stderr, "Could not get MTD device info from %s\n", argv[1]); - close(fd); - exit(1); - } + if (ioctl(fd, MEMGETINFO, &mtdInfo)) + sys_errmsg_die("could not get mtd info: %s", dev); + /* Make sure user options are valid */ if (argc > 2) mtdLockInfo.start = strtol(argv[2], NULL, 0); else mtdLockInfo.start = 0; - if (mtdLockInfo.start > mtdInfo.size) { - fprintf(stderr, "%#x is beyond device size %#x\n", + if (mtdLockInfo.start > mtdInfo.size) + errmsg_die("%#x is beyond device size %#x", mtdLockInfo.start, mtdInfo.size); - close(fd); - exit(1); - } if (argc > 3) { count = strtol(argv[3], NULL, 0); @@ -71,21 +74,17 @@ int main(int argc, char *argv[]) mtdLockInfo.length = mtdInfo.size - mtdInfo.erasesize; else mtdLockInfo.length = mtdInfo.erasesize * count; - } else { + } else mtdLockInfo.length = mtdInfo.size - mtdInfo.erasesize; - } - if (mtdLockInfo.start + mtdLockInfo.length > mtdInfo.size) { - fprintf(stderr, "%s range is more than device supports\n", FLASH_MSG); - exit(1); - } + if (mtdLockInfo.start + mtdLockInfo.length > mtdInfo.size) + errmsg_die("range is more than device supports: %#x + %#x > %#x", + mtdLockInfo.start, mtdLockInfo.length, mtdInfo.size); + /* Finally do the operation */ request = FLASH_UNLOCK ? MEMUNLOCK : MEMLOCK; - if (ioctl(fd, request, &mtdLockInfo)) { - fprintf(stderr, "Could not %s MTD device: %s\n", - FLASH_MSG, argv[1]); - close(fd); - exit(1); - } + if (ioctl(fd, request, &mtdLockInfo)) + sys_errmsg_die("could not %s device: %s\n", + FLASH_MSG, dev); return 0; } |