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