109 lines
2.0 KiB
C
109 lines
2.0 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(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
|