barebox/commands/uimage.c
Sascha Hauer 1d1618a11b add uimage command
The uimage command superseeds the iminfo command. Unlike
the iminfo command the uimage command can also be used
to verify the data crc on demand and to extract uImages
to files.

Signed-off-by: Sascha Hauer <s.hauer@pengutronix.de>
2011-12-15 11:07:15 +01:00

109 lines
2 KiB
C

#include <common.h>
#include <command.h>
#include <image.h>
#include <libbb.h>
#include <fcntl.h>
#include <fs.h>
#include <malloc.h>
#include <errno.h>
#include <getopt.h>
static int uimage_fd;
static int uimage_flush(void *buf, unsigned int len)
{
int ret;
ret = write_full(uimage_fd, buf, len);
return ret;
}
static int do_uimage(struct command *cmdtp, int argc, char *argv[])
{
struct uimage_handle *handle;
int ret;
int verify = 0;
int fd;
int opt;
char *extract = NULL;
int info = 0;
int image_no = 0;
while ((opt = getopt(argc, argv, "ve:in:")) > 0) {
switch (opt) {
case 'v':
verify = 1;
break;
case 'i':
info = 1;
break;
case 'e':
extract = optarg;
break;
case 'n':
image_no = simple_strtoul(optarg, NULL, 0);
break;
}
}
if (optind == argc)
return COMMAND_ERROR_USAGE;
handle = uimage_open(argv[optind]);
if (!handle)
return 1;
if (info) {
printf("Image at %s:\n", argv[optind]);
uimage_print_contents(handle);
}
if (verify) {
printf("verifying data crc... ");
ret = uimage_verify(handle);
if (ret) {
goto err;
printf("Bad Data CRC\n");
} else {
printf("ok\n");
}
}
if (extract) {
fd = open(extract, O_WRONLY | O_CREAT | O_TRUNC);
if (fd < 0) {
perror("open");
ret = fd;
goto err;
}
uimage_fd = fd;
ret = uimage_load(handle, image_no, uimage_flush);
if (ret) {
printf("loading uImage failed with %d\n", ret);
close(fd);
goto err;
}
close(fd);
}
err:
uimage_close(handle);
return ret ? 1 : 0;
}
BAREBOX_CMD_HELP_START(uimage)
BAREBOX_CMD_HELP_USAGE("uimage [OPTIONS] <file>\n")
BAREBOX_CMD_HELP_OPT ("-i", "show information about image\n")
BAREBOX_CMD_HELP_OPT ("-v", "verify image\n")
BAREBOX_CMD_HELP_OPT ("-e <outfile>", "extract image to <outfile>\n")
BAREBOX_CMD_HELP_OPT ("-n <no>", "use image number <no> in multifile images\n")
BAREBOX_CMD_HELP_END
BAREBOX_CMD_START(uimage)
.cmd = do_uimage,
.usage = "extract/verify uImage",
BAREBOX_CMD_HELP(cmd_uimage_help)
BAREBOX_CMD_END