xref: /openbmc/u-boot/cmd/mvebu/bubt.c (revision 3ba98ed8)
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 );
785