// SPDX-License-Identifier: GPL-2.0+

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

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

  #include <irq_func.h>

  #include <lmb.h>
  #include <mapmem.h>

  #include <linux/kernel.h>
  #include <linux/sizes.h>

  /*
   * Image booting support
   */
  static int booti_start(cmd_tbl_t *cmdtp, int flag, int argc,
  			char * const argv[], bootm_headers_t *images)
  {
  	int ret;

  	ulong ld;
  	ulong relocated_addr;
  	ulong image_size;

  
  	ret = do_bootm_states(cmdtp, flag, argc, argv, BOOTM_STATE_START,
  			      images, 1);
  
  	/* Setup Linux kernel Image entry point */
  	if (!argc) {

  		ld = image_load_addr;

  		debug("*  kernel: default image load address = 0x%08lx
  ",

  				image_load_addr);

  	} else {

  		ld = simple_strtoul(argv[0], NULL, 16);

  		debug("*  kernel: cmdline image address = 0x%08lx
  ", ld);

  	}

  	ret = booti_setup(ld, &relocated_addr, &image_size, false);

  	if (ret != 0)
  		return 1;

  #if defined(CONFIG_IMX_HAB) && !defined(CONFIG_AVB_SUPPORT)

  	extern int authenticate_image(
  		uint32_t ddr_start, uint32_t raw_image_size);
  	if (authenticate_image(ld, image_size) != 0) {
  		printf("Authenticate Image Fail, Please check
  ");
  		return 1;
  	}
  
  #endif

  	/* Handle BOOTM_STATE_LOADOS */
  	if (relocated_addr != ld) {
  		debug("Moving Image from 0x%lx to 0x%lx
  ", ld, relocated_addr);
  		memmove((void *)relocated_addr, (void *)ld, image_size);
  	}

  	images->ep = relocated_addr;

  	images->os.start = relocated_addr;
  	images->os.end = relocated_addr + image_size;

  	lmb_reserve(&images->lmb, images->ep, le32_to_cpu(image_size));

  
  	/*
  	 * Handle the BOOTM_STATE_FINDOTHER state ourselves as we do not
  	 * have a header that provide this informaiton.
  	 */
  	if (bootm_find_images(flag, argc, argv))
  		return 1;
  
  	return 0;
  }
  
  int do_booti(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  {
  	int ret;
  
  	/* Consume 'booti' */
  	argc--; argv++;
  
  	if (booti_start(cmdtp, flag, argc, argv, &images))
  		return 1;
  
  	/*
  	 * We are doing the BOOTM_STATE_LOADOS state ourselves, so must
  	 * disable interrupts ourselves
  	 */
  	bootm_disable_interrupts();
  
  	images.os.os = IH_OS_LINUX;

  #ifdef CONFIG_RISCV_SMODE
  	images.os.arch = IH_ARCH_RISCV;
  #elif CONFIG_ARM64

  	images.os.arch = IH_ARCH_ARM64;

  #endif

  	ret = do_bootm_states(cmdtp, flag, argc, argv,

  #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
  			      BOOTM_STATE_RAMDISK |
  #endif

  			      BOOTM_STATE_OS_PREP | BOOTM_STATE_OS_FAKE_GO |
  			      BOOTM_STATE_OS_GO,
  			      &images, 1);
  
  	return ret;
  }
  
  #ifdef CONFIG_SYS_LONGHELP
  static char booti_help_text[] =
  	"[addr [initrd[:size]] [fdt]]
  "

  	"    - boot Linux 'Image' stored at 'addr'
  "

  	"\tThe argument 'initrd' is optional and specifies the address
  "
  	"\tof an initrd in memory. The optional parameter ':size' allows
  "
  	"\tspecifying the size of a RAW initrd.
  "
  #if defined(CONFIG_OF_LIBFDT)
  	"\tSince booting a Linux kernel requires a flat device-tree, a
  "
  	"\tthird argument providing the address of the device-tree blob
  "
  	"\tis required. To boot a kernel with a device-tree blob but
  "
  	"\twithout an initrd image, use a '-' for the initrd argument.
  "
  #endif
  	"";
  #endif
  
  U_BOOT_CMD(
  	booti,	CONFIG_SYS_MAXARGS,	1,	do_booti,

  	"boot Linux kernel 'Image' format from memory", booti_help_text

  );