cmd/mvebu/bubt.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Copyright (C) 2016 Marvell International Ltd.
   4  * https://spdx.org/licenses
   5  */
   6
   7 #include <config.h>
   8 #include <common.h>
   9 #include <command.h>
  10 #include <vsprintf.h>
  11 #include <errno.h>
  12 #include <dm.h>
  13
  14 #include <spi_flash.h>
  15 #include <spi.h>
  16 #include <nand.h>
  17 #include <usb.h>
  18 #include <fs.h>
  19 #include <mmc.h>
  20 #ifdef CONFIG_BLK
  21 #include <blk.h>
  22 #endif
  23 #include <u-boot/sha1.h>
  24 #include <u-boot/sha256.h>
  25
  26 #ifndef CONFIG_SYS_MMC_ENV_DEV
  27 #define CONFIG_SYS_MMC_ENV_DEV  0
  28 #endif
  29
  30 #if defined(CONFIG_ARMADA_8K)
  31 #define MAIN_HDR_MAGIC          0xB105B002
  32
  33 struct mvebu_image_header {
  34         u32     magic;                  /*  0-3  */
  35         u32     prolog_size;            /*  4-7  */
  36         u32     prolog_checksum;        /*  8-11 */
  37         u32     boot_image_size;        /* 12-15 */
  38         u32     boot_image_checksum;    /* 16-19 */
  39         u32     rsrvd0;                 /* 20-23 */
  40         u32     load_addr;              /* 24-27 */
  41         u32     exec_addr;              /* 28-31 */
  42         u8      uart_cfg;               /*  32   */
  43         u8      baudrate;               /*  33   */
  44         u8      ext_count;              /*  34   */
  45         u8      aux_flags;              /*  35   */
  46         u32     io_arg_0;               /* 36-39 */
  47         u32     io_arg_1;               /* 40-43 */
  48         u32     io_arg_2;               /* 43-47 */
  49         u32     io_arg_3;               /* 48-51 */
  50         u32     rsrvd1;                 /* 52-55 */
  51         u32     rsrvd2;                 /* 56-59 */
  52         u32     rsrvd3;                 /* 60-63 */
  53 };
  54 #elif defined(CONFIG_ARMADA_3700)       /* A3700 */
  55 #define HASH_SUM_LEN            16
  56 #define IMAGE_VERSION_3_6_0     0x030600
  57 #define IMAGE_VERSION_3_5_0     0x030500
  58
  59 struct common_tim_data {
  60         u32     version;
  61         u32     identifier;
  62         u32     trusted;
  63         u32     issue_date;
  64         u32     oem_unique_id;
  65         u32     reserved[5];            /* Reserve 20 bytes */
  66         u32     boot_flash_sign;
  67         u32     num_images;
  68         u32     num_keys;
  69         u32     size_of_reserved;
  70 };
  71
  72 struct mvebu_image_info {
  73         u32     image_id;
  74         u32     next_image_id;
  75         u32     flash_entry_addr;
  76         u32     load_addr;
  77         u32     image_size;
  78         u32     image_size_to_hash;
  79         u32     hash_algorithm_id;
  80         u32     hash[HASH_SUM_LEN];     /* Reserve 512 bits for the hash */
  81         u32     partition_number;
  82         u32     enc_algorithm_id;
  83         u32     encrypt_start_offset;
  84         u32     encrypt_size;
  85 };
  86 #endif /* CONFIG_ARMADA_XXX */
  87
  88 struct bubt_dev {
  89         char name[8];
  90         size_t (*read)(const char *file_name);
  91         int (*write)(size_t image_size);
  92         int (*active)(void);
  93 };
  94
  95 static ulong get_load_addr(void)
  96 {
  97         const char *addr_str;
  98         unsigned long addr;
  99
 100         addr_str = env_get("loadaddr");
 101         if (addr_str)
 102                 addr = simple_strtoul(addr_str, NULL, 16);
 103         else
 104                 addr = CONFIG_SYS_LOAD_ADDR;
 105
 106         return addr;
 107 }
 108
 109 /********************************************************************
 110  *     eMMC services
 111  ********************************************************************/
 112 #if CONFIG_IS_ENABLED(DM_MMC) && CONFIG_IS_ENABLED(MMC_WRITE)
 113 static int mmc_burn_image(size_t image_size)
 114 {
 115         struct mmc      *mmc;
 116         lbaint_t        start_lba;
 117         lbaint_t        blk_count;
 118         ulong           blk_written;
 119         int             err;
 120         const u8        mmc_dev_num = CONFIG_SYS_MMC_ENV_DEV;
 121 #ifdef CONFIG_BLK
 122         struct blk_desc *blk_desc;
 123 #endif
 124         mmc = find_mmc_device(mmc_dev_num);
 125         if (!mmc) {
 126                 printf("No SD/MMC/eMMC card found\n");
 127                 return -ENOMEDIUM;
 128         }
 129
 130         err = mmc_init(mmc);
 131         if (err) {
 132                 printf("%s(%d) init failed\n", IS_SD(mmc) ? "SD" : "MMC",
 133                        mmc_dev_num);
 134                 return err;
 135         }
 136
 137 #ifdef CONFIG_SYS_MMC_ENV_PART
 138         if (mmc->part_num != CONFIG_SYS_MMC_ENV_PART) {
 139                 err = mmc_switch_part(mmc_dev_num, CONFIG_SYS_MMC_ENV_PART);
 140                 if (err) {
 141                         printf("MMC partition switch failed\n");
 142                         return err;
 143                 }
 144         }
 145 #endif
 146
 147         /* SD reserves LBA-0 for MBR and boots from LBA-1,
 148          * MMC/eMMC boots from LBA-0
 149          */
 150         start_lba = IS_SD(mmc) ? 1 : 0;
 151 #ifdef CONFIG_BLK
 152         blk_count = image_size / mmc->write_bl_len;
 153         if (image_size % mmc->write_bl_len)
 154                 blk_count += 1;
 155
 156         blk_desc = mmc_get_blk_desc(mmc);
 157         if (!blk_desc) {
 158                 printf("Error - failed to obtain block descriptor\n");
 159                 return -ENODEV;
 160         }
 161         blk_written = blk_dwrite(blk_desc, start_lba, blk_count,
 162                                  (void *)get_load_addr());
 163 #else
 164         blk_count = image_size / mmc->block_dev.blksz;
 165         if (image_size % mmc->block_dev.blksz)
 166                 blk_count += 1;
 167
 168         blk_written = mmc->block_dev.block_write(mmc_dev_num,
 169                                                  start_lba, blk_count,
 170                                                  (void *)get_load_addr());
 171 #endif /* CONFIG_BLK */
 172         if (blk_written != blk_count) {
 173                 printf("Error - written %#lx blocks\n", blk_written);
 174                 return -ENOSPC;
 175         }
 176         printf("Done!\n");
 177
 178 #ifdef CONFIG_SYS_MMC_ENV_PART
 179         if (mmc->part_num != CONFIG_SYS_MMC_ENV_PART)
 180                 mmc_switch_part(mmc_dev_num, mmc->part_num);
 181 #endif
 182
 183         return 0;
 184 }
 185
 186 static size_t mmc_read_file(const char *file_name)
 187 {
 188         loff_t          act_read = 0;
 189         int             rc;
 190         struct mmc      *mmc;
 191         const u8        mmc_dev_num = CONFIG_SYS_MMC_ENV_DEV;
 192
 193         mmc = find_mmc_device(mmc_dev_num);
 194         if (!mmc) {
 195                 printf("No SD/MMC/eMMC card found\n");
 196                 return 0;
 197         }
 198
 199         if (mmc_init(mmc)) {
 200                 printf("%s(%d) init failed\n", IS_SD(mmc) ? "SD" : "MMC",
 201                        mmc_dev_num);
 202                 return 0;
 203         }
 204
 205         /* Load from data partition (0) */
 206         if (fs_set_blk_dev("mmc", "0", FS_TYPE_ANY)) {
 207                 printf("Error: MMC 0 not found\n");
 208                 return 0;
 209         }
 210
 211         /* Perfrom file read */
 212         rc = fs_read(file_name, get_load_addr(), 0, 0, &act_read);
 213         if (rc)
 214                 return 0;
 215
 216         return act_read;
 217 }
 218
 219 static int is_mmc_active(void)
 220 {
 221         return 1;
 222 }
 223 #else /* CONFIG_DM_MMC */
 224 static int mmc_burn_image(size_t image_size)
 225 {
 226         return -ENODEV;
 227 }
 228
 229 static size_t mmc_read_file(const char *file_name)
 230 {
 231         return 0;
 232 }
 233
 234 static int is_mmc_active(void)
 235 {
 236         return 0;
 237 }
 238 #endif /* CONFIG_DM_MMC */
 239
 240 /********************************************************************
 241  *     SPI services
 242  ********************************************************************/
 243 #ifdef CONFIG_SPI_FLASH
 244 static int spi_burn_image(size_t image_size)
 245 {
 246         int ret;
 247         struct spi_flash *flash;
 248         u32 erase_bytes;
 249
 250         /* Probe the SPI bus to get the flash device */
 251         flash = spi_flash_probe(CONFIG_ENV_SPI_BUS,
 252                                 CONFIG_ENV_SPI_CS,
 253                                 CONFIG_SF_DEFAULT_SPEED,
 254                                 CONFIG_SF_DEFAULT_MODE);
 255         if (!flash) {
 256                 printf("Failed to probe SPI Flash\n");
 257                 return -ENOMEDIUM;
 258         }
 259
 260 #ifdef CONFIG_SPI_FLASH_PROTECTION
 261         spi_flash_protect(flash, 0);
 262 #endif
 263         erase_bytes = image_size +
 264                 (flash->erase_size - image_size % flash->erase_size);
 265         printf("Erasing %d bytes (%d blocks) at offset 0 ...",
 266                erase_bytes, erase_bytes / flash->erase_size);
 267         ret = spi_flash_erase(flash, 0, erase_bytes);
 268         if (ret)
 269                 printf("Error!\n");
 270         else
 271                 printf("Done!\n");
 272
 273         printf("Writing %d bytes from 0x%lx to offset 0 ...",
 274                (int)image_size, get_load_addr());
 275         ret = spi_flash_write(flash, 0, image_size, (void *)get_load_addr());
 276         if (ret)
 277                 printf("Error!\n");
 278         else
 279                 printf("Done!\n");
 280
 281 #ifdef CONFIG_SPI_FLASH_PROTECTION
 282         spi_flash_protect(flash, 1);
 283 #endif
 284
 285         return ret;
 286 }
 287
 288 static int is_spi_active(void)
 289 {
 290         return 1;
 291 }
 292
 293 #else /* CONFIG_SPI_FLASH */
 294 static int spi_burn_image(size_t image_size)
 295 {
 296         return -ENODEV;
 297 }
 298
 299 static int is_spi_active(void)
 300 {
 301         return 0;
 302 }
 303 #endif /* CONFIG_SPI_FLASH */
 304
 305 /********************************************************************
 306  *     NAND services
 307  ********************************************************************/
 308 #ifdef CONFIG_CMD_NAND
 309 static int nand_burn_image(size_t image_size)
 310 {
 311         int ret;
 312         uint32_t block_size;
 313         struct mtd_info *mtd;
 314
 315         mtd = get_nand_dev_by_index(nand_curr_device);
 316         if (!mtd) {
 317                 puts("\nno devices available\n");
 318                 return -ENOMEDIUM;
 319         }
 320         block_size = mtd->erasesize;
 321
 322         /* Align U-Boot size to currently used blocksize */
 323         image_size = ((image_size + (block_size - 1)) & (~(block_size - 1)));
 324
 325         /* Erase the U-BOOT image space */
 326         printf("Erasing 0x%x - 0x%x:...", 0, (int)image_size);
 327         ret = nand_erase(mtd, 0, image_size);
 328         if (ret) {
 329                 printf("Error!\n");
 330                 goto error;
 331         }
 332         printf("Done!\n");
 333
 334         /* Write the image to flash */
 335         printf("Writing %d bytes from 0x%lx to offset 0 ... ",
 336                (int)image_size, get_load_addr());
 337         ret = nand_write(mtd, 0, &image_size, (void *)get_load_addr());
 338         if (ret)
 339                 printf("Error!\n");
 340         else
 341                 printf("Done!\n");
 342
 343 error:
 344         return ret;
 345 }
 346
 347 static int is_nand_active(void)
 348 {
 349         return 1;
 350 }
 351
 352 #else /* CONFIG_CMD_NAND */
 353 static int nand_burn_image(size_t image_size)
 354 {
 355         return -ENODEV;
 356 }
 357
 358 static int is_nand_active(void)
 359 {
 360         return 0;
 361 }
 362 #endif /* CONFIG_CMD_NAND */
 363
 364 /********************************************************************
 365  *     USB services
 366  ********************************************************************/
 367 #if defined(CONFIG_USB_STORAGE) && defined(CONFIG_BLK)
 368 static size_t usb_read_file(const char *file_name)
 369 {
 370         loff_t act_read = 0;
 371         struct udevice *dev;
 372         int rc;
 373
 374         usb_stop();
 375
 376         if (usb_init() < 0) {
 377                 printf("Error: usb_init failed\n");
 378                 return 0;
 379         }
 380
 381         /* Try to recognize storage devices immediately */
 382         blk_first_device(IF_TYPE_USB, &dev);
 383         if (!dev) {
 384                 printf("Error: USB storage device not found\n");
 385                 return 0;
 386         }
 387
 388         /* Always load from usb 0 */
 389         if (fs_set_blk_dev("usb", "0", FS_TYPE_ANY)) {
 390                 printf("Error: USB 0 not found\n");
 391                 return 0;
 392         }
 393
 394         /* Perfrom file read */
 395         rc = fs_read(file_name, get_load_addr(), 0, 0, &act_read);
 396         if (rc)
 397                 return 0;
 398
 399         return act_read;
 400 }
 401
 402 static int is_usb_active(void)
 403 {
 404         return 1;
 405 }
 406
 407 #else /* defined(CONFIG_USB_STORAGE) && defined (CONFIG_BLK) */
 408 static size_t usb_read_file(const char *file_name)
 409 {
 410         return 0;
 411 }
 412
 413 static int is_usb_active(void)
 414 {
 415         return 0;
 416 }
 417 #endif /* defined(CONFIG_USB_STORAGE) && defined (CONFIG_BLK) */
 418
 419 /********************************************************************
 420  *     Network services
 421  ********************************************************************/
 422 #ifdef CONFIG_CMD_NET
 423 static size_t tftp_read_file(const char *file_name)
 424 {
 425         /* update global variable load_addr before tftp file from network */
 426         load_addr = get_load_addr();
 427         return net_loop(TFTPGET);
 428 }
 429
 430 static int is_tftp_active(void)
 431 {
 432         return 1;
 433 }
 434
 435 #else
 436 static size_t tftp_read_file(const char *file_name)
 437 {
 438         return 0;
 439 }
 440
 441 static int is_tftp_active(void)
 442 {
 443         return 0;
 444 }
 445 #endif /* CONFIG_CMD_NET */
 446
 447 enum bubt_devices {
 448         BUBT_DEV_NET = 0,
 449         BUBT_DEV_USB,
 450         BUBT_DEV_MMC,
 451         BUBT_DEV_SPI,
 452         BUBT_DEV_NAND,
 453
 454         BUBT_MAX_DEV
 455 };
 456
 457 struct bubt_dev bubt_devs[BUBT_MAX_DEV] = {
 458         {"tftp", tftp_read_file, NULL, is_tftp_active},
 459         {"usb",  usb_read_file,  NULL, is_usb_active},
 460         {"mmc",  mmc_read_file,  mmc_burn_image, is_mmc_active},
 461         {"spi",  NULL, spi_burn_image,  is_spi_active},
 462         {"nand", NULL, nand_burn_image, is_nand_active},
 463 };
 464
 465 static int bubt_write_file(struct bubt_dev *dst, size_t image_size)
 466 {
 467         if (!dst->write) {
 468                 printf("Error: Write not supported on device %s\n", dst->name);
 469                 return -ENOTSUPP;
 470         }
 471
 472         return dst->write(image_size);
 473 }
 474
 475 #if defined(CONFIG_ARMADA_8K)
 476 u32 do_checksum32(u32 *start, int32_t len)
 477 {
 478         u32 sum = 0;
 479         u32 *startp = start;
 480
 481         do {
 482                 sum += *startp;
 483                 startp++;
 484                 len -= 4;
 485         } while (len > 0);
 486
 487         return sum;
 488 }
 489
 490 static int check_image_header(void)
 491 {
 492         struct mvebu_image_header *hdr =
 493                         (struct mvebu_image_header *)get_load_addr();
 494         u32 header_len = hdr->prolog_size;
 495         u32 checksum;
 496         u32 checksum_ref = hdr->prolog_checksum;
 497
 498         /*
 499          * For now compare checksum, and magic. Later we can
 500          * verify more stuff on the header like interface type, etc
 501          */
 502         if (hdr->magic != MAIN_HDR_MAGIC) {
 503                 printf("ERROR: Bad MAGIC 0x%08x != 0x%08x\n",
 504                        hdr->magic, MAIN_HDR_MAGIC);
 505                 return -ENOEXEC;
 506         }
 507
 508         /* The checksum value is discarded from checksum calculation */
 509         hdr->prolog_checksum = 0;
 510
 511         checksum = do_checksum32((u32 *)hdr, header_len);
 512         if (checksum != checksum_ref) {
 513                 printf("Error: Bad Image checksum. 0x%x != 0x%x\n",
 514                        checksum, checksum_ref);
 515                 return -ENOEXEC;
 516         }
 517
 518         /* Restore the checksum before writing */
 519         hdr->prolog_checksum = checksum_ref;
 520         printf("Image checksum...OK!\n");
 521
 522         return 0;
 523 }
 524 #elif defined(CONFIG_ARMADA_3700) /* Armada 3700 */
 525 static int check_image_header(void)
 526 {
 527         struct common_tim_data *hdr = (struct common_tim_data *)get_load_addr();
 528         int image_num;
 529         u8 hash_160_output[SHA1_SUM_LEN];
 530         u8 hash_256_output[SHA256_SUM_LEN];
 531         sha1_context hash1_text;
 532         sha256_context hash256_text;
 533         u8 *hash_output;
 534         u32 hash_algorithm_id;
 535         u32 image_size_to_hash;
 536         u32 flash_entry_addr;
 537         u32 *hash_value;
 538         u32 internal_hash[HASH_SUM_LEN];
 539         const u8 *buff;
 540         u32 num_of_image = hdr->num_images;
 541         u32 version = hdr->version;
 542         u32 trusted = hdr->trusted;
 543
 544         /* bubt checksum validation only supports nontrusted images */
 545         if (trusted == 1) {
 546                 printf("bypass image validation, ");
 547                 printf("only untrusted image is supported now\n");
 548                 return 0;
 549         }
 550         /* only supports image version 3.5 and 3.6 */
 551         if (version != IMAGE_VERSION_3_5_0 && version != IMAGE_VERSION_3_6_0) {
 552                 printf("Error: Unsupported Image version = 0x%08x\n", version);
 553                 return -ENOEXEC;
 554         }
 555         /* validate images hash value */
 556         for (image_num = 0; image_num < num_of_image; image_num++) {
 557                 struct mvebu_image_info *info =
 558                                 (struct mvebu_image_info *)(get_load_addr() +
 559                                 sizeof(struct common_tim_data) +
 560                                 image_num * sizeof(struct mvebu_image_info));
 561                 hash_algorithm_id = info->hash_algorithm_id;
 562                 image_size_to_hash = info->image_size_to_hash;
 563                 flash_entry_addr = info->flash_entry_addr;
 564                 hash_value = info->hash;
 565                 buff = (const u8 *)(get_load_addr() + flash_entry_addr);
 566
 567                 if (image_num == 0) {
 568                         /*
 569                          * The first image includes hash values in its content.
 570                          * For hash calculation, we need to save the original
 571                          * hash values to a local variable that will be
 572                          * copied back for comparsion and set all zeros to
 573                          * the orignal hash values for calculating new value.
 574                          * First image original format :
 575                          * x...x (datum1) x...x(orig. hash values) x...x(datum2)
 576                          * Replaced first image format :
 577                          * x...x (datum1) 0...0(hash values) x...x(datum2)
 578                          */
 579                         memcpy(internal_hash, hash_value,
 580                                sizeof(internal_hash));
 581                         memset(hash_value, 0, sizeof(internal_hash));
 582                 }
 583                 if (image_size_to_hash == 0) {
 584                         printf("Warning: Image_%d hash checksum is disabled, ",
 585                                image_num);
 586                         printf("skip the image validation.\n");
 587                         continue;
 588                 }
 589                 switch (hash_algorithm_id) {
 590                 case SHA1_SUM_LEN:
 591                         sha1_starts(&hash1_text);
 592                         sha1_update(&hash1_text, buff, image_size_to_hash);
 593                         sha1_finish(&hash1_text, hash_160_output);
 594                         hash_output = hash_160_output;
 595                         break;
 596                 case SHA256_SUM_LEN:
 597                         sha256_starts(&hash256_text);
 598                         sha256_update(&hash256_text, buff, image_size_to_hash);
 599                         sha256_finish(&hash256_text, hash_256_output);
 600                         hash_output = hash_256_output;
 601                         break;
 602                 default:
 603                         printf("Error: Unsupported hash_algorithm_id = %d\n",
 604                                hash_algorithm_id);
 605                         return -ENOEXEC;
 606                 }
 607                 if (image_num == 0)
 608                         memcpy(hash_value, internal_hash,
 609                                sizeof(internal_hash));
 610                 if (memcmp(hash_value, hash_output, hash_algorithm_id) != 0) {
 611                         printf("Error: Image_%d checksum is not correct\n",
 612                                image_num);
 613                         return -ENOEXEC;
 614                 }
 615         }
 616         printf("Image checksum...OK!\n");
 617
 618         return 0;
 619 }
 620
 621 #else /* Not ARMADA? */
 622 static int check_image_header(void)
 623 {
 624         printf("bubt cmd does not support this SoC device or family!\n");
 625         return -ENOEXEC;
 626 }
 627 #endif
 628
 629 static int bubt_verify(size_t image_size)
 630 {
 631         int err;
 632
 633         /* Check a correct image header exists */
 634         err = check_image_header();
 635         if (err) {
 636                 printf("Error: Image header verification failed\n");
 637                 return err;
 638         }
 639
 640         return 0;
 641 }
 642
 643 static int bubt_read_file(struct bubt_dev *src)
 644 {
 645         size_t image_size;
 646
 647         if (!src->read) {
 648                 printf("Error: Read not supported on device \"%s\"\n",
 649                        src->name);
 650                 return 0;
 651         }
 652
 653         image_size = src->read(net_boot_file_name);
 654         if (image_size <= 0) {
 655                 printf("Error: Failed to read file %s from %s\n",
 656                        net_boot_file_name, src->name);
 657                 return 0;
 658         }
 659
 660         return image_size;
 661 }
 662
 663 static int bubt_is_dev_active(struct bubt_dev *dev)
 664 {
 665         if (!dev->active) {
 666                 printf("Device \"%s\" not supported by U-BOOT image\n",
 667                        dev->name);
 668                 return 0;
 669         }
 670
 671         if (!dev->active()) {
 672                 printf("Device \"%s\" is inactive\n", dev->name);
 673                 return 0;
 674         }
 675
 676         return 1;
 677 }
 678
 679 struct bubt_dev *find_bubt_dev(char *dev_name)
 680 {
 681         int dev;
 682
 683         for (dev = 0; dev < BUBT_MAX_DEV; dev++) {
 684                 if (strcmp(bubt_devs[dev].name, dev_name) == 0)
 685                         return &bubt_devs[dev];
 686         }
 687
 688         return 0;
 689 }
 690
 691 #define DEFAULT_BUBT_SRC "tftp"
 692
 693 #ifndef DEFAULT_BUBT_DST
 694 #ifdef CONFIG_MVEBU_SPI_BOOT
 695 #define DEFAULT_BUBT_DST "spi"
 696 #elif defined(CONFIG_MVEBU_NAND_BOOT)
 697 #define DEFAULT_BUBT_DST "nand"
 698 #elif defined(CONFIG_MVEBU_MMC_BOOT)
 699 #define DEFAULT_BUBT_DST "mmc"
 700 else
 701 #define DEFAULT_BUBT_DST "error"
 702 #endif
 703 #endif /* DEFAULT_BUBT_DST */
 704
 705 int do_bubt_cmd(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 706 {
 707         struct bubt_dev *src, *dst;
 708         size_t image_size;
 709         char src_dev_name[8];
 710         char dst_dev_name[8];
 711         char *name;
 712         int  err;
 713
 714         if (argc < 2)
 715                 copy_filename(net_boot_file_name,
 716                               CONFIG_MVEBU_UBOOT_DFLT_NAME,
 717                               sizeof(net_boot_file_name));
 718         else
 719                 copy_filename(net_boot_file_name, argv[1],
 720                               sizeof(net_boot_file_name));
 721
 722         if (argc >= 3) {
 723                 strncpy(dst_dev_name, argv[2], 8);
 724         } else {
 725                 name = DEFAULT_BUBT_DST;
 726                 strncpy(dst_dev_name, name, 8);
 727         }
 728
 729         if (argc >= 4)
 730                 strncpy(src_dev_name, argv[3], 8);
 731         else
 732                 strncpy(src_dev_name, DEFAULT_BUBT_SRC, 8);
 733
 734         /* Figure out the destination device */
 735         dst = find_bubt_dev(dst_dev_name);
 736         if (!dst) {
 737                 printf("Error: Unknown destination \"%s\"\n", dst_dev_name);
 738                 return -EINVAL;
 739         }
 740
 741         if (!bubt_is_dev_active(dst))
 742                 return -ENODEV;
 743
 744         /* Figure out the source device */
 745         src = find_bubt_dev(src_dev_name);
 746         if (!src) {
 747                 printf("Error: Unknown source \"%s\"\n", src_dev_name);
 748                 return 1;
 749         }
 750
 751         if (!bubt_is_dev_active(src))
 752                 return -ENODEV;
 753
 754         printf("Burning U-BOOT image \"%s\" from \"%s\" to \"%s\"\n",
 755                net_boot_file_name, src->name, dst->name);
 756
 757         image_size = bubt_read_file(src);
 758         if (!image_size)
 759                 return -EIO;
 760
 761         err = bubt_verify(image_size);
 762         if (err)
 763                 return err;
 764
 765         err = bubt_write_file(dst, image_size);
 766         if (err)
 767                 return err;
 768
 769         return 0;
 770 }
 771
 772 U_BOOT_CMD(
 773         bubt, 4, 0, do_bubt_cmd,
 774         "Burn a u-boot image to flash",
 775         "[file-name] [destination [source]]\n"
 776         "\t-file-name     The image file name to burn. Default = flash-image.bin\n"
 777         "\t-destination   Flash to burn to [spi, nand, mmc]. Default = active boot device\n"
 778         "\t-source        The source to load image from [tftp, usb, mmc]. Default = tftp\n"
 779         "Examples:\n"
 780         "\tbubt - Burn flash-image.bin from tftp to active boot device\n"
 781         "\tbubt flash-image-new.bin nand - Burn flash-image-new.bin from tftp to NAND flash\n"
 782         "\tbubt backup-flash-image.bin mmc usb - Burn backup-flash-image.bin from usb to MMC\n"
 783
 784 );