183d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
22e192b24SSimon Glass /*
32e192b24SSimon Glass * (C) Copyright 2000-2004
42e192b24SSimon Glass * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
52e192b24SSimon Glass *
62e192b24SSimon Glass * (C) Copyright 2003
72e192b24SSimon Glass * Kai-Uwe Bloem, Auerswald GmbH & Co KG, <linux-development@auerswald.de>
82e192b24SSimon Glass */
92e192b24SSimon Glass
102e192b24SSimon Glass
112e192b24SSimon Glass /*
122e192b24SSimon Glass * Multi Image extract
132e192b24SSimon Glass */
142e192b24SSimon Glass #include <common.h>
152e192b24SSimon Glass #include <command.h>
162e192b24SSimon Glass #include <image.h>
172e192b24SSimon Glass #include <mapmem.h>
182e192b24SSimon Glass #include <watchdog.h>
192e192b24SSimon Glass #if defined(CONFIG_BZIP2)
202e192b24SSimon Glass #include <bzlib.h>
212e192b24SSimon Glass #endif
222e192b24SSimon Glass #include <asm/byteorder.h>
232e192b24SSimon Glass #include <asm/io.h>
242e192b24SSimon Glass
252e192b24SSimon Glass #ifndef CONFIG_SYS_XIMG_LEN
262e192b24SSimon Glass /* use 8MByte as default max gunzip size */
272e192b24SSimon Glass #define CONFIG_SYS_XIMG_LEN 0x800000
282e192b24SSimon Glass #endif
292e192b24SSimon Glass
302e192b24SSimon Glass static int
do_imgextract(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])312e192b24SSimon Glass do_imgextract(cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
322e192b24SSimon Glass {
332e192b24SSimon Glass ulong addr = load_addr;
342e192b24SSimon Glass ulong dest = 0;
352e192b24SSimon Glass ulong data, len;
362e192b24SSimon Glass int verify;
372e192b24SSimon Glass int part = 0;
382e192b24SSimon Glass #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
392e192b24SSimon Glass ulong count;
402e192b24SSimon Glass image_header_t *hdr = NULL;
412e192b24SSimon Glass #endif
422e192b24SSimon Glass #if defined(CONFIG_FIT)
432e192b24SSimon Glass const char *uname = NULL;
442e192b24SSimon Glass const void* fit_hdr;
452e192b24SSimon Glass int noffset;
462e192b24SSimon Glass const void *fit_data;
472e192b24SSimon Glass size_t fit_len;
482e192b24SSimon Glass #endif
492e192b24SSimon Glass #ifdef CONFIG_GZIP
502e192b24SSimon Glass uint unc_len = CONFIG_SYS_XIMG_LEN;
512e192b24SSimon Glass #endif
522e192b24SSimon Glass uint8_t comp;
532e192b24SSimon Glass
54bfebc8c9SSimon Glass verify = env_get_yesno("verify");
552e192b24SSimon Glass
562e192b24SSimon Glass if (argc > 1) {
572e192b24SSimon Glass addr = simple_strtoul(argv[1], NULL, 16);
582e192b24SSimon Glass }
592e192b24SSimon Glass if (argc > 2) {
602e192b24SSimon Glass part = simple_strtoul(argv[2], NULL, 16);
612e192b24SSimon Glass #if defined(CONFIG_FIT)
622e192b24SSimon Glass uname = argv[2];
632e192b24SSimon Glass #endif
642e192b24SSimon Glass }
652e192b24SSimon Glass if (argc > 3) {
662e192b24SSimon Glass dest = simple_strtoul(argv[3], NULL, 16);
672e192b24SSimon Glass }
682e192b24SSimon Glass
692e192b24SSimon Glass switch (genimg_get_format((void *)addr)) {
702e192b24SSimon Glass #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
712e192b24SSimon Glass case IMAGE_FORMAT_LEGACY:
722e192b24SSimon Glass
732e192b24SSimon Glass printf("## Copying part %d from legacy image "
742e192b24SSimon Glass "at %08lx ...\n", part, addr);
752e192b24SSimon Glass
762e192b24SSimon Glass hdr = (image_header_t *)addr;
772e192b24SSimon Glass if (!image_check_magic(hdr)) {
782e192b24SSimon Glass printf("Bad Magic Number\n");
792e192b24SSimon Glass return 1;
802e192b24SSimon Glass }
812e192b24SSimon Glass
822e192b24SSimon Glass if (!image_check_hcrc(hdr)) {
832e192b24SSimon Glass printf("Bad Header Checksum\n");
842e192b24SSimon Glass return 1;
852e192b24SSimon Glass }
862e192b24SSimon Glass #ifdef DEBUG
872e192b24SSimon Glass image_print_contents(hdr);
882e192b24SSimon Glass #endif
892e192b24SSimon Glass
902e192b24SSimon Glass if (!image_check_type(hdr, IH_TYPE_MULTI) &&
912e192b24SSimon Glass !image_check_type(hdr, IH_TYPE_SCRIPT)) {
922e192b24SSimon Glass printf("Wrong Image Type for %s command\n",
932e192b24SSimon Glass cmdtp->name);
942e192b24SSimon Glass return 1;
952e192b24SSimon Glass }
962e192b24SSimon Glass
972e192b24SSimon Glass comp = image_get_comp(hdr);
982e192b24SSimon Glass if ((comp != IH_COMP_NONE) && (argc < 4)) {
992e192b24SSimon Glass printf("Must specify load address for %s command "
1002e192b24SSimon Glass "with compressed image\n",
1012e192b24SSimon Glass cmdtp->name);
1022e192b24SSimon Glass return 1;
1032e192b24SSimon Glass }
1042e192b24SSimon Glass
1052e192b24SSimon Glass if (verify) {
1062e192b24SSimon Glass printf(" Verifying Checksum ... ");
1072e192b24SSimon Glass if (!image_check_dcrc(hdr)) {
1082e192b24SSimon Glass printf("Bad Data CRC\n");
1092e192b24SSimon Glass return 1;
1102e192b24SSimon Glass }
1112e192b24SSimon Glass printf("OK\n");
1122e192b24SSimon Glass }
1132e192b24SSimon Glass
1142e192b24SSimon Glass count = image_multi_count(hdr);
1152e192b24SSimon Glass if (part >= count) {
1162e192b24SSimon Glass printf("Bad Image Part\n");
1172e192b24SSimon Glass return 1;
1182e192b24SSimon Glass }
1192e192b24SSimon Glass
1202e192b24SSimon Glass image_multi_getimg(hdr, part, &data, &len);
1212e192b24SSimon Glass break;
1222e192b24SSimon Glass #endif
1232e192b24SSimon Glass #if defined(CONFIG_FIT)
1242e192b24SSimon Glass case IMAGE_FORMAT_FIT:
1252e192b24SSimon Glass if (uname == NULL) {
1262e192b24SSimon Glass puts("No FIT subimage unit name\n");
1272e192b24SSimon Glass return 1;
1282e192b24SSimon Glass }
1292e192b24SSimon Glass
1302e192b24SSimon Glass printf("## Copying '%s' subimage from FIT image "
1312e192b24SSimon Glass "at %08lx ...\n", uname, addr);
1322e192b24SSimon Glass
1332e192b24SSimon Glass fit_hdr = (const void *)addr;
1342e192b24SSimon Glass if (!fit_check_format(fit_hdr)) {
1352e192b24SSimon Glass puts("Bad FIT image format\n");
1362e192b24SSimon Glass return 1;
1372e192b24SSimon Glass }
1382e192b24SSimon Glass
1392e192b24SSimon Glass /* get subimage node offset */
1402e192b24SSimon Glass noffset = fit_image_get_node(fit_hdr, uname);
1412e192b24SSimon Glass if (noffset < 0) {
1422e192b24SSimon Glass printf("Can't find '%s' FIT subimage\n", uname);
1432e192b24SSimon Glass return 1;
1442e192b24SSimon Glass }
1452e192b24SSimon Glass
146*240e58eeSStefan Theil if (!fit_image_check_comp(fit_hdr, noffset, IH_COMP_NONE)
1472e192b24SSimon Glass && (argc < 4)) {
1482e192b24SSimon Glass printf("Must specify load address for %s command "
1492e192b24SSimon Glass "with compressed image\n",
1502e192b24SSimon Glass cmdtp->name);
1512e192b24SSimon Glass return 1;
1522e192b24SSimon Glass }
1532e192b24SSimon Glass
1542e192b24SSimon Glass /* verify integrity */
1552e192b24SSimon Glass if (verify) {
1562e192b24SSimon Glass if (!fit_image_verify(fit_hdr, noffset)) {
1572e192b24SSimon Glass puts("Bad Data Hash\n");
1582e192b24SSimon Glass return 1;
1592e192b24SSimon Glass }
1602e192b24SSimon Glass }
1612e192b24SSimon Glass
1622e192b24SSimon Glass /* get subimage data address and length */
1632e192b24SSimon Glass if (fit_image_get_data(fit_hdr, noffset,
1642e192b24SSimon Glass &fit_data, &fit_len)) {
1652e192b24SSimon Glass puts("Could not find script subimage data\n");
1662e192b24SSimon Glass return 1;
1672e192b24SSimon Glass }
1682e192b24SSimon Glass
1692e192b24SSimon Glass if (fit_image_get_comp(fit_hdr, noffset, &comp)) {
1702e192b24SSimon Glass puts("Could not find script subimage "
1712e192b24SSimon Glass "compression type\n");
1722e192b24SSimon Glass return 1;
1732e192b24SSimon Glass }
1742e192b24SSimon Glass
1752e192b24SSimon Glass data = (ulong)fit_data;
1762e192b24SSimon Glass len = (ulong)fit_len;
1772e192b24SSimon Glass break;
1782e192b24SSimon Glass #endif
1792e192b24SSimon Glass default:
1802e192b24SSimon Glass puts("Invalid image type for imxtract\n");
1812e192b24SSimon Glass return 1;
1822e192b24SSimon Glass }
1832e192b24SSimon Glass
1842e192b24SSimon Glass if (argc > 3) {
1852e192b24SSimon Glass switch (comp) {
1862e192b24SSimon Glass case IH_COMP_NONE:
1872e192b24SSimon Glass #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
1882e192b24SSimon Glass {
1892e192b24SSimon Glass size_t l = len;
1902e192b24SSimon Glass size_t tail;
1912e192b24SSimon Glass void *to = (void *) dest;
1922e192b24SSimon Glass void *from = (void *)data;
1932e192b24SSimon Glass
1942e192b24SSimon Glass printf(" Loading part %d ... ", part);
1952e192b24SSimon Glass
1962e192b24SSimon Glass while (l > 0) {
1972e192b24SSimon Glass tail = (l > CHUNKSZ) ? CHUNKSZ : l;
1982e192b24SSimon Glass WATCHDOG_RESET();
1992e192b24SSimon Glass memmove(to, from, tail);
2002e192b24SSimon Glass to += tail;
2012e192b24SSimon Glass from += tail;
2022e192b24SSimon Glass l -= tail;
2032e192b24SSimon Glass }
2042e192b24SSimon Glass }
2052e192b24SSimon Glass #else /* !(CONFIG_HW_WATCHDOG || CONFIG_WATCHDOG) */
2062e192b24SSimon Glass printf(" Loading part %d ... ", part);
2072e192b24SSimon Glass memmove((char *) dest, (char *)data, len);
2082e192b24SSimon Glass #endif /* CONFIG_HW_WATCHDOG || CONFIG_WATCHDOG */
2092e192b24SSimon Glass break;
2102e192b24SSimon Glass #ifdef CONFIG_GZIP
2112e192b24SSimon Glass case IH_COMP_GZIP:
2122e192b24SSimon Glass printf(" Uncompressing part %d ... ", part);
2132e192b24SSimon Glass if (gunzip((void *) dest, unc_len,
2142e192b24SSimon Glass (uchar *) data, &len) != 0) {
2152e192b24SSimon Glass puts("GUNZIP ERROR - image not loaded\n");
2162e192b24SSimon Glass return 1;
2172e192b24SSimon Glass }
2182e192b24SSimon Glass break;
2192e192b24SSimon Glass #endif
2202e192b24SSimon Glass #if defined(CONFIG_BZIP2) && defined(CONFIG_IMAGE_FORMAT_LEGACY)
2212e192b24SSimon Glass case IH_COMP_BZIP2:
2222e192b24SSimon Glass {
2232e192b24SSimon Glass int i;
2242e192b24SSimon Glass
2252e192b24SSimon Glass printf(" Uncompressing part %d ... ", part);
2262e192b24SSimon Glass /*
2272e192b24SSimon Glass * If we've got less than 4 MB of malloc()
2282e192b24SSimon Glass * space, use slower decompression algorithm
2292e192b24SSimon Glass * which requires at most 2300 KB of memory.
2302e192b24SSimon Glass */
2312e192b24SSimon Glass i = BZ2_bzBuffToBuffDecompress(
2322e192b24SSimon Glass map_sysmem(ntohl(hdr->ih_load), 0),
2332e192b24SSimon Glass &unc_len, (char *)data, len,
2342e192b24SSimon Glass CONFIG_SYS_MALLOC_LEN < (4096 * 1024),
2352e192b24SSimon Glass 0);
2362e192b24SSimon Glass if (i != BZ_OK) {
2372e192b24SSimon Glass printf("BUNZIP2 ERROR %d - "
2382e192b24SSimon Glass "image not loaded\n", i);
2392e192b24SSimon Glass return 1;
2402e192b24SSimon Glass }
2412e192b24SSimon Glass }
2422e192b24SSimon Glass break;
2432e192b24SSimon Glass #endif /* CONFIG_BZIP2 */
2442e192b24SSimon Glass default:
2452e192b24SSimon Glass printf("Unimplemented compression type %d\n", comp);
2462e192b24SSimon Glass return 1;
2472e192b24SSimon Glass }
2482e192b24SSimon Glass puts("OK\n");
2492e192b24SSimon Glass }
2502e192b24SSimon Glass
2518354aa27SMario Six flush_cache(dest, ALIGN(len, ARCH_DMA_MINALIGN));
2522e192b24SSimon Glass
253018f5303SSimon Glass env_set_hex("fileaddr", data);
254018f5303SSimon Glass env_set_hex("filesize", len);
2552e192b24SSimon Glass
2562e192b24SSimon Glass return 0;
2572e192b24SSimon Glass }
2582e192b24SSimon Glass
2592e192b24SSimon Glass #ifdef CONFIG_SYS_LONGHELP
2602e192b24SSimon Glass static char imgextract_help_text[] =
2612e192b24SSimon Glass "addr part [dest]\n"
2622e192b24SSimon Glass " - extract <part> from legacy image at <addr> and copy to <dest>"
2632e192b24SSimon Glass #if defined(CONFIG_FIT)
2642e192b24SSimon Glass "\n"
2652e192b24SSimon Glass "addr uname [dest]\n"
2662e192b24SSimon Glass " - extract <uname> subimage from FIT image at <addr> and copy to <dest>"
2672e192b24SSimon Glass #endif
2682e192b24SSimon Glass "";
2692e192b24SSimon Glass #endif
2702e192b24SSimon Glass
2712e192b24SSimon Glass U_BOOT_CMD(
2722e192b24SSimon Glass imxtract, 4, 1, do_imgextract,
2732e192b24SSimon Glass "extract a part of a multi-image", imgextract_help_text
2742e192b24SSimon Glass );
275