/*
   * (C) Copyright 2000-2011
   * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
   *

   * SPDX-License-Identifier:	GPL-2.0+

   */
  #include <common.h>
  #include <command.h>

  #include <part.h>

  #if defined(CONFIG_CMD_IDE) || defined(CONFIG_CMD_SCSI) || \
  	defined(CONFIG_USB_STORAGE)

  int common_diskboot(cmd_tbl_t *cmdtp, const char *intf, int argc,
  		    char *const argv[])
  {

  	int dev, part;
  	ulong addr = CONFIG_SYS_LOAD_ADDR;
  	ulong cnt;

  	disk_partition_t info;
  	image_header_t *hdr;
  	block_dev_desc_t *dev_desc;
  
  #if defined(CONFIG_FIT)
  	const void *fit_hdr = NULL;
  #endif
  
  	bootstage_mark(BOOTSTAGE_ID_IDE_START);

  	if (argc > 3) {

  		bootstage_error(BOOTSTAGE_ID_IDE_ADDR);
  		return CMD_RET_USAGE;
  	}
  	bootstage_mark(BOOTSTAGE_ID_IDE_ADDR);

  	if (argc > 1)
  		addr = simple_strtoul(argv[1], NULL, 16);

  	bootstage_mark(BOOTSTAGE_ID_IDE_BOOT_DEVICE);

  	part = get_device_and_partition(intf, (argc == 3) ? argv[2] : NULL,

  					&dev_desc, &info, 1);

  	if (part < 0) {

  		bootstage_error(BOOTSTAGE_ID_IDE_TYPE);
  		return 1;
  	}

  	dev = dev_desc->dev;
  	bootstage_mark(BOOTSTAGE_ID_IDE_TYPE);

  	printf("
  Loading from %s device %d, partition %d: "
  	       "Name: %.32s  Type: %.32s
  ", intf, dev, part, info.name,
  	       info.type);

  	debug("First Block: " LBAFU ",  # of blocks: " LBAFU
  	      ", Block Size: %ld
  ",

  	      info.start, info.size, info.blksz);
  
  	if (dev_desc->block_read(dev, info.start, 1, (ulong *) addr) != 1) {
  		printf("** Read error on %d:%d
  ", dev, part);
  		bootstage_error(BOOTSTAGE_ID_IDE_PART_READ);
  		return 1;
  	}
  	bootstage_mark(BOOTSTAGE_ID_IDE_PART_READ);
  
  	switch (genimg_get_format((void *) addr)) {
  	case IMAGE_FORMAT_LEGACY:
  		hdr = (image_header_t *) addr;
  
  		bootstage_mark(BOOTSTAGE_ID_IDE_FORMAT);
  
  		if (!image_check_hcrc(hdr)) {
  			puts("
  ** Bad Header Checksum **
  ");
  			bootstage_error(BOOTSTAGE_ID_IDE_CHECKSUM);
  			return 1;
  		}
  		bootstage_mark(BOOTSTAGE_ID_IDE_CHECKSUM);
  
  		image_print_contents(hdr);
  
  		cnt = image_get_image_size(hdr);
  		break;
  #if defined(CONFIG_FIT)
  	case IMAGE_FORMAT_FIT:
  		fit_hdr = (const void *) addr;
  		puts("Fit image detected...
  ");
  
  		cnt = fit_get_size(fit_hdr);
  		break;
  #endif
  	default:
  		bootstage_error(BOOTSTAGE_ID_IDE_FORMAT);
  		puts("** Unknown image type
  ");
  		return 1;
  	}
  
  	cnt += info.blksz - 1;
  	cnt /= info.blksz;
  	cnt -= 1;
  
  	if (dev_desc->block_read(dev, info.start + 1, cnt,
  					 (ulong *)(addr + info.blksz)) != cnt) {
  		printf("** Read error on %d:%d
  ", dev, part);
  		bootstage_error(BOOTSTAGE_ID_IDE_READ);
  		return 1;
  	}
  	bootstage_mark(BOOTSTAGE_ID_IDE_READ);
  
  #if defined(CONFIG_FIT)
  	/* This cannot be done earlier,
  	 * we need complete FIT image in RAM first */
  	if (genimg_get_format((void *) addr) == IMAGE_FORMAT_FIT) {
  		if (!fit_check_format(fit_hdr)) {
  			bootstage_error(BOOTSTAGE_ID_IDE_FIT_READ);
  			puts("** Bad FIT image format
  ");
  			return 1;
  		}
  		bootstage_mark(BOOTSTAGE_ID_IDE_FIT_READ_OK);
  		fit_print_contents(fit_hdr);
  	}
  #endif
  
  	flush_cache(addr, (cnt+1)*info.blksz);
  
  	/* Loading ok, update default load address */
  	load_addr = addr;
  
  	return bootm_maybe_autostart(cmdtp, argv[0]);
  }

  #endif