[thirdparty/u-boot.git] / cmd / bootefi.c

/*
 *  EFI application loader
 *
 *  Copyright (c) 2016 Alexander Graf
 *
 *  SPDX-License-Identifier:     GPL-2.0+
 */

#include <common.h>
#include <command.h>
#include <dm.h>
#include <efi_loader.h>
#include <errno.h>
#include <libfdt.h>
#include <libfdt_env.h>
#include <memalign.h>
#include <asm/global_data.h>
#include <asm-generic/sections.h>
#include <linux/linkage.h>

DECLARE_GLOBAL_DATA_PTR;

static uint8_t efi_obj_list_initalized;

static struct efi_device_path *bootefi_image_path;
static struct efi_device_path *bootefi_device_path;

/* Initialize and populate EFI object list */
static void efi_init_obj_list(void)
{
	efi_obj_list_initalized = 1;

	efi_console_register();
#ifdef CONFIG_PARTITIONS
	efi_disk_register();
#endif
#if defined(CONFIG_LCD) || defined(CONFIG_DM_VIDEO)
	efi_gop_register();
#endif
#ifdef CONFIG_NET
	efi_net_register();
#endif
#ifdef CONFIG_GENERATE_SMBIOS_TABLE
	efi_smbios_register();
#endif

	/* Initialize EFI runtime services */
	efi_reset_system_init();
	efi_get_time_init();
}

static void *copy_fdt(void *fdt)
{
	u64 fdt_size = fdt_totalsize(fdt);
	unsigned long fdt_ram_start = -1L, fdt_pages;
	u64 new_fdt_addr;
	void *new_fdt;
	int i;

        for (i = 0; i < CONFIG_NR_DRAM_BANKS; i++) {
                u64 ram_start = gd->bd->bi_dram[i].start;
                u64 ram_size = gd->bd->bi_dram[i].size;

		if (!ram_size)
			continue;

		if (ram_start < fdt_ram_start)
			fdt_ram_start = ram_start;
	}

	/* Give us at least 4kb breathing room */
	fdt_size = ALIGN(fdt_size + 4096, EFI_PAGE_SIZE);
	fdt_pages = fdt_size >> EFI_PAGE_SHIFT;

	/* Safe fdt location is at 128MB */
	new_fdt_addr = fdt_ram_start + (128 * 1024 * 1024) + fdt_size;
	if (efi_allocate_pages(1, EFI_BOOT_SERVICES_DATA, fdt_pages,
			       &new_fdt_addr) != EFI_SUCCESS) {
		/* If we can't put it there, put it somewhere */
		new_fdt_addr = (ulong)memalign(EFI_PAGE_SIZE, fdt_size);
		if (efi_allocate_pages(1, EFI_BOOT_SERVICES_DATA, fdt_pages,
				       &new_fdt_addr) != EFI_SUCCESS) {
			printf("ERROR: Failed to reserve space for FDT\n");
			return NULL;
		}
	}

	new_fdt = (void*)(ulong)new_fdt_addr;
	memcpy(new_fdt, fdt, fdt_totalsize(fdt));
	fdt_set_totalsize(new_fdt, fdt_size);

	return new_fdt;
}

static ulong efi_do_enter(void *image_handle,
			  struct efi_system_table *st,
			  asmlinkage ulong (*entry)(void *image_handle,
				struct efi_system_table *st))
{
	efi_status_t ret = EFI_LOAD_ERROR;

	if (entry)
		ret = entry(image_handle, st);
	st->boottime->exit(image_handle, ret, 0, NULL);
	return ret;
}

#ifdef CONFIG_ARM64
static unsigned long efi_run_in_el2(asmlinkage ulong (*entry)(
			void *image_handle, struct efi_system_table *st),
			void *image_handle, struct efi_system_table *st)
{
	/* Enable caches again */
	dcache_enable();

	return efi_do_enter(image_handle, st, entry);
}
#endif

/*
 * Load an EFI payload into a newly allocated piece of memory, register all
 * EFI objects it would want to access and jump to it.
 */
static unsigned long do_bootefi_exec(void *efi, void *fdt,
				     struct efi_device_path *device_path,
				     struct efi_device_path *image_path)
{
	struct efi_loaded_image loaded_image_info = {};
	struct efi_object loaded_image_info_obj = {};
	ulong ret;

	ulong (*entry)(void *image_handle, struct efi_system_table *st)
		asmlinkage;
	ulong fdt_pages, fdt_size, fdt_start, fdt_end;
	const efi_guid_t fdt_guid = EFI_FDT_GUID;
	bootm_headers_t img = { 0 };

	/* Initialize and populate EFI object list */
	if (!efi_obj_list_initalized)
		efi_init_obj_list();

	efi_setup_loaded_image(&loaded_image_info, &loaded_image_info_obj,
			       device_path, image_path);

	/*
	 * gd lives in a fixed register which may get clobbered while we execute
	 * the payload. So save it here and restore it on every callback entry
	 */
	efi_save_gd();

	if (fdt && !fdt_check_header(fdt)) {
		/* Prepare fdt for payload */
		fdt = copy_fdt(fdt);

		if (image_setup_libfdt(&img, fdt, 0, NULL)) {
			printf("ERROR: Failed to process device tree\n");
			return -EINVAL;
		}

		/* Link to it in the efi tables */
		efi_install_configuration_table(&fdt_guid, fdt);

		/* And reserve the space in the memory map */
		fdt_start = ((ulong)fdt) & ~EFI_PAGE_MASK;
		fdt_end = ((ulong)fdt) + fdt_totalsize(fdt);
		fdt_size = (fdt_end - fdt_start) + EFI_PAGE_MASK;
		fdt_pages = fdt_size >> EFI_PAGE_SHIFT;
		/* Give a bootloader the chance to modify the device tree */
		fdt_pages += 2;
		efi_add_memory_map(fdt_start, fdt_pages,
				   EFI_BOOT_SERVICES_DATA, true);
	} else {
		printf("WARNING: Invalid device tree, expect boot to fail\n");
		efi_install_configuration_table(&fdt_guid, NULL);
	}

	/* Load the EFI payload */
	entry = efi_load_pe(efi, &loaded_image_info);
	if (!entry) {
		ret = -ENOENT;
		goto exit;
	}

	/* we don't support much: */
	env_set("efi_8be4df61-93ca-11d2-aa0d-00e098032b8c_OsIndicationsSupported",
		"{ro,boot}(blob)0000000000000000");

	/* Call our payload! */
	debug("%s:%d Jumping to 0x%lx\n", __func__, __LINE__, (long)entry);

	if (setjmp(&loaded_image_info.exit_jmp)) {
		ret = loaded_image_info.exit_status;
		goto exit;
	}

#ifdef CONFIG_ARM64
	/* On AArch64 we need to make sure we call our payload in < EL3 */
	if (current_el() == 3) {
		smp_kick_all_cpus();
		dcache_disable();	/* flush cache before switch to EL2 */

		/* Move into EL2 and keep running there */
		armv8_switch_to_el2((ulong)entry, (ulong)&loaded_image_info,
				    (ulong)&systab, 0, (ulong)efi_run_in_el2,
				    ES_TO_AARCH64);

		/* Should never reach here, efi exits with longjmp */
		while (1) { }
	}
#endif

	ret = efi_do_enter(&loaded_image_info, &systab, entry);

exit:
	/* image has returned, loaded-image obj goes *poof*: */
	list_del(&loaded_image_info_obj.link);

	return ret;
}

static int do_bootefi_bootmgr_exec(unsigned long fdt_addr)
{
	struct efi_device_path *device_path, *file_path;
	void *addr;
	efi_status_t r;

	/* Initialize and populate EFI object list */
	if (!efi_obj_list_initalized)
		efi_init_obj_list();

	/*
	 * gd lives in a fixed register which may get clobbered while we execute
	 * the payload. So save it here and restore it on every callback entry
	 */
	efi_save_gd();

	addr = efi_bootmgr_load(&device_path, &file_path);
	if (!addr)
		return 1;

	printf("## Starting EFI application at %p ...\n", addr);
	r = do_bootefi_exec(addr, (void *)fdt_addr, device_path, file_path);
	printf("## Application terminated, r = %lu\n",
	       r & ~EFI_ERROR_MASK);

	if (r != EFI_SUCCESS)
		return 1;

	return 0;
}

/* Interpreter command to boot an arbitrary EFI image from memory */
static int do_bootefi(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
	char *saddr, *sfdt;
	unsigned long addr, fdt_addr = 0;
	unsigned long r;

	if (argc < 2)
		return CMD_RET_USAGE;
#ifdef CONFIG_CMD_BOOTEFI_HELLO
	if (!strcmp(argv[1], "hello")) {
		ulong size = __efi_helloworld_end - __efi_helloworld_begin;

		saddr = env_get("loadaddr");
		if (saddr)
			addr = simple_strtoul(saddr, NULL, 16);
		else
			addr = CONFIG_SYS_LOAD_ADDR;
		memcpy((char *)addr, __efi_helloworld_begin, size);
	} else
#endif
#ifdef CONFIG_CMD_BOOTEFI_SELFTEST
	if (!strcmp(argv[1], "selftest")) {
		struct efi_loaded_image loaded_image_info = {};
		struct efi_object loaded_image_info_obj = {};

		efi_setup_loaded_image(&loaded_image_info,
				       &loaded_image_info_obj,
				       bootefi_device_path, bootefi_image_path);
		/*
		 * gd lives in a fixed register which may get clobbered while we
		 * execute the payload. So save it here and restore it on every
		 * callback entry
		 */
		efi_save_gd();
		/* Initialize and populate EFI object list */
		if (!efi_obj_list_initalized)
			efi_init_obj_list();
		return efi_selftest(&loaded_image_info, &systab);
	} else
#endif
	if (!strcmp(argv[1], "bootmgr")) {
		unsigned long fdt_addr = 0;

		if (argc > 2)
			fdt_addr = simple_strtoul(argv[2], NULL, 16);

		return do_bootefi_bootmgr_exec(fdt_addr);
	} else {
		saddr = argv[1];

		addr = simple_strtoul(saddr, NULL, 16);

		if (argc > 2) {
			sfdt = argv[2];
			fdt_addr = simple_strtoul(sfdt, NULL, 16);
		}
	}

	printf("## Starting EFI application at %08lx ...\n", addr);
	r = do_bootefi_exec((void *)addr, (void *)fdt_addr,
			    bootefi_device_path, bootefi_image_path);
	printf("## Application terminated, r = %lu\n",
	       r & ~EFI_ERROR_MASK);

	if (r != EFI_SUCCESS)
		return 1;
	else
		return 0;
}

#ifdef CONFIG_SYS_LONGHELP
static char bootefi_help_text[] =
	"<image address> [fdt address]\n"
	"  - boot EFI payload stored at address <image address>.\n"
	"    If specified, the device tree located at <fdt address> gets\n"
	"    exposed as EFI configuration table.\n"
#ifdef CONFIG_CMD_BOOTEFI_HELLO
	"bootefi hello\n"
	"  - boot a sample Hello World application stored within U-Boot\n"
#endif
#ifdef CONFIG_CMD_BOOTEFI_SELFTEST
	"bootefi selftest\n"
	"  - boot an EFI selftest application stored within U-Boot\n"
#endif
	"bootmgr [fdt addr]\n"
	"  - load and boot EFI payload based on BootOrder/BootXXXX variables.\n"
	"\n"
	"    If specified, the device tree located at <fdt address> gets\n"
	"    exposed as EFI configuration table.\n";
#endif

U_BOOT_CMD(
	bootefi, 3, 0, do_bootefi,
	"Boots an EFI payload from memory",
	bootefi_help_text
);

static int parse_partnum(const char *devnr)
{
	const char *str = strchr(devnr, ':');
	if (str) {
		str++;
		return simple_strtoul(str, NULL, 16);
	}
	return 0;
}

void efi_set_bootdev(const char *dev, const char *devnr, const char *path)
{
	char filename[32] = { 0 }; /* dp->str is u16[32] long */
	char *s;

	if (strcmp(dev, "Net")) {
		struct blk_desc *desc;
		int part;

		desc = blk_get_dev(dev, simple_strtol(devnr, NULL, 10));
		part = parse_partnum(devnr);

		bootefi_device_path = efi_dp_from_part(desc, part);
	} else {
#ifdef CONFIG_NET
		bootefi_device_path = efi_dp_from_eth();
#endif
	}

	if (!path)
		return;

	if (strcmp(dev, "Net")) {
		/* Add leading / to fs paths, because they're absolute */
		snprintf(filename, sizeof(filename), "/%s", path);
	} else {
		snprintf(filename, sizeof(filename), "%s", path);
	}
	/* DOS style file path: */
	s = filename;
	while ((s = strchr(s, '/')))
		*s++ = '\\';
	bootefi_image_path = efi_dp_from_file(NULL, 0, filename);
}
Commit	Line	Data
b9939336 AG	1	/*
	2	* EFI application loader
	3	*
	4	* Copyright (c) 2016 Alexander Graf
	5	*
	6	* SPDX-License-Identifier: GPL-2.0+
	7	*/
	8
	9	#include <common.h>
	10	#include <command.h>
9d922450	11	#include <dm.h>
b9939336 AG	12	#include <efi_loader.h>
	13	#include <errno.h>
	14	#include <libfdt.h>
	15	#include <libfdt_env.h>
ad0c1a3d	16	#include <memalign.h>
0d9d501f	17	#include <asm/global_data.h>
e275458c SG	18	#include <asm-generic/sections.h>
e275458c SG	19	#include <linux/linkage.h>
0d9d501f AG	20
0d9d501f AG	21	DECLARE_GLOBAL_DATA_PTR;
b9939336	22
7cbc1241 HS	23	static uint8_t efi_obj_list_initalized;
7cbc1241 HS	24
95c5553e RC	25	static struct efi_device_path *bootefi_image_path;
95c5553e RC	26	static struct efi_device_path *bootefi_device_path;
b9939336	27
7cbc1241 HS	28	/* Initialize and populate EFI object list */
	29	static void efi_init_obj_list(void)
	30	{
	31	efi_obj_list_initalized = 1;
	32
7cbc1241 HS	33	efi_console_register();
	34	#ifdef CONFIG_PARTITIONS
	35	efi_disk_register();
	36	#endif
	37	#if defined(CONFIG_LCD) \|\| defined(CONFIG_DM_VIDEO)
	38	efi_gop_register();
	39	#endif
	40	#ifdef CONFIG_NET
95c5553e	41	efi_net_register();
7cbc1241 HS	42	#endif
	43	#ifdef CONFIG_GENERATE_SMBIOS_TABLE
	44	efi_smbios_register();
	45	#endif
	46
	47	/* Initialize EFI runtime services */
	48	efi_reset_system_init();
	49	efi_get_time_init();
	50	}
	51
0d9d501f AG	52	static void copy_fdt(void fdt)
	53	{
	54	u64 fdt_size = fdt_totalsize(fdt);
ad0c1a3d AG	55	unsigned long fdt_ram_start = -1L, fdt_pages;
ad0c1a3d AG	56	u64 new_fdt_addr;
0d9d501f	57	void *new_fdt;
ad0c1a3d	58	int i;
0d9d501f	59
ad0c1a3d AG	60	for (i = 0; i < CONFIG_NR_DRAM_BANKS; i++) {
	61	u64 ram_start = gd->bd->bi_dram[i].start;
	62	u64 ram_size = gd->bd->bi_dram[i].size;
0d9d501f	63
ad0c1a3d AG	64	if (!ram_size)
	65	continue;
	66
	67	if (ram_start < fdt_ram_start)
	68	fdt_ram_start = ram_start;
	69	}
	70
	71	/* Give us at least 4kb breathing room */
a44bffcc	72	fdt_size = ALIGN(fdt_size + 4096, EFI_PAGE_SIZE);
ad0c1a3d AG	73	fdt_pages = fdt_size >> EFI_PAGE_SHIFT;
	74
	75	/* Safe fdt location is at 128MB */
	76	new_fdt_addr = fdt_ram_start + (128 * 1024 * 1024) + fdt_size;
	77	if (efi_allocate_pages(1, EFI_BOOT_SERVICES_DATA, fdt_pages,
	78	&new_fdt_addr) != EFI_SUCCESS) {
	79	/* If we can't put it there, put it somewhere */
a44bffcc	80	new_fdt_addr = (ulong)memalign(EFI_PAGE_SIZE, fdt_size);
85a6e9b3 AG	81	if (efi_allocate_pages(1, EFI_BOOT_SERVICES_DATA, fdt_pages,
	82	&new_fdt_addr) != EFI_SUCCESS) {
	83	printf("ERROR: Failed to reserve space for FDT\n");
	84	return NULL;
	85	}
ad0c1a3d	86	}
85a6e9b3	87
ad0c1a3d	88	new_fdt = (void*)(ulong)new_fdt_addr;
0d9d501f AG	89	memcpy(new_fdt, fdt, fdt_totalsize(fdt));
	90	fdt_set_totalsize(new_fdt, fdt_size);
	91
	92	return new_fdt;
	93	}
	94
b06d8ac3 HS	95	static ulong efi_do_enter(void *image_handle,
	96	struct efi_system_table *st,
	97	asmlinkage ulong (entry)(void image_handle,
	98	struct efi_system_table *st))
	99	{
	100	efi_status_t ret = EFI_LOAD_ERROR;
	101
	102	if (entry)
	103	ret = entry(image_handle, st);
	104	st->boottime->exit(image_handle, ret, 0, NULL);
	105	return ret;
	106	}
	107
ec6617c3	108	#ifdef CONFIG_ARM64
b06d8ac3 HS	109	static unsigned long efi_run_in_el2(asmlinkage ulong (*entry)(
	110	void image_handle, struct efi_system_table st),
	111	void image_handle, struct efi_system_table st)
ec6617c3 AW	112	{
	113	/* Enable caches again */
	114	dcache_enable();
	115
b06d8ac3	116	return efi_do_enter(image_handle, st, entry);
ec6617c3 AW	117	}
	118	#endif
	119
b9939336 AG	120	/*
	121	* Load an EFI payload into a newly allocated piece of memory, register all
	122	* EFI objects it would want to access and jump to it.
	123	*/
95c5553e RC	124	static unsigned long do_bootefi_exec(void efi, void fdt,
	125	struct efi_device_path *device_path,
	126	struct efi_device_path *image_path)
b9939336	127	{
95c5553e RC	128	struct efi_loaded_image loaded_image_info = {};
	129	struct efi_object loaded_image_info_obj = {};
	130	ulong ret;
	131
e275458c SG	132	ulong (entry)(void image_handle, struct efi_system_table *st)
e275458c SG	133	asmlinkage;
b9939336	134	ulong fdt_pages, fdt_size, fdt_start, fdt_end;
f4f9993f	135	const efi_guid_t fdt_guid = EFI_FDT_GUID;
dea2174d	136	bootm_headers_t img = { 0 };
b9939336	137
95c5553e RC	138	/* Initialize and populate EFI object list */
	139	if (!efi_obj_list_initalized)
	140	efi_init_obj_list();
	141
	142	efi_setup_loaded_image(&loaded_image_info, &loaded_image_info_obj,
	143	device_path, image_path);
	144
b9939336 AG	145	/*
	146	* gd lives in a fixed register which may get clobbered while we execute
	147	* the payload. So save it here and restore it on every callback entry
	148	*/
	149	efi_save_gd();
	150
1c39809b	151	if (fdt && !fdt_check_header(fdt)) {
dea2174d	152	/* Prepare fdt for payload */
0d9d501f AG	153	fdt = copy_fdt(fdt);
	154
	155	if (image_setup_libfdt(&img, fdt, 0, NULL)) {
dea2174d AG	156	printf("ERROR: Failed to process device tree\n");
	157	return -EINVAL;
	158	}
	159
	160	/* Link to it in the efi tables */
f4f9993f	161	efi_install_configuration_table(&fdt_guid, fdt);
b9939336 AG	162
b9939336 AG	163	/* And reserve the space in the memory map */
0d9d501f AG	164	fdt_start = ((ulong)fdt) & ~EFI_PAGE_MASK;
0d9d501f AG	165	fdt_end = ((ulong)fdt) + fdt_totalsize(fdt);
b9939336 AG	166	fdt_size = (fdt_end - fdt_start) + EFI_PAGE_MASK;
	167	fdt_pages = fdt_size >> EFI_PAGE_SHIFT;
	168	/* Give a bootloader the chance to modify the device tree */
	169	fdt_pages += 2;
	170	efi_add_memory_map(fdt_start, fdt_pages,
	171	EFI_BOOT_SERVICES_DATA, true);
b9939336	172	} else {
1c39809b	173	printf("WARNING: Invalid device tree, expect boot to fail\n");
f4f9993f	174	efi_install_configuration_table(&fdt_guid, NULL);
b9939336 AG	175	}
	176
	177	/* Load the EFI payload */
	178	entry = efi_load_pe(efi, &loaded_image_info);
95c5553e RC	179	if (!entry) {
	180	ret = -ENOENT;
	181	goto exit;
	182	}
80a4800e	183
ad644e7c RC	184	/* we don't support much: */
	185	env_set("efi_8be4df61-93ca-11d2-aa0d-00e098032b8c_OsIndicationsSupported",
	186	"{ro,boot}(blob)0000000000000000");
	187
b9939336	188	/* Call our payload! */
edcef3ba	189	debug("%s:%d Jumping to 0x%lx\n", __func__, __LINE__, (long)entry);
a86aeaf2 AG	190
a86aeaf2 AG	191	if (setjmp(&loaded_image_info.exit_jmp)) {
95c5553e	192	ret = loaded_image_info.exit_status;
95c5553e	193	goto exit;
a86aeaf2 AG	194	}
a86aeaf2 AG	195
69bd459d AG	196	#ifdef CONFIG_ARM64
	197	/* On AArch64 we need to make sure we call our payload in < EL3 */
	198	if (current_el() == 3) {
	199	smp_kick_all_cpus();
	200	dcache_disable(); /* flush cache before switch to EL2 */
ec6617c3 AW	201
	202	/* Move into EL2 and keep running there */
	203	armv8_switch_to_el2((ulong)entry, (ulong)&loaded_image_info,
7c5e1feb	204	(ulong)&systab, 0, (ulong)efi_run_in_el2,
ec6617c3 AW	205	ES_TO_AARCH64);
	206
	207	/* Should never reach here, efi exits with longjmp */
	208	while (1) { }
69bd459d AG	209	}
	210	#endif
	211
95c5553e RC	212	ret = efi_do_enter(&loaded_image_info, &systab, entry);
	213
	214	exit:
	215	/* image has returned, loaded-image obj goes poof: */
	216	list_del(&loaded_image_info_obj.link);
	217
	218	return ret;
b9939336 AG	219	}
b9939336 AG	220
9975fe96 RC	221	static int do_bootefi_bootmgr_exec(unsigned long fdt_addr)
	222	{
	223	struct efi_device_path device_path, file_path;
	224	void *addr;
	225	efi_status_t r;
	226
	227	/* Initialize and populate EFI object list */
	228	if (!efi_obj_list_initalized)
	229	efi_init_obj_list();
	230
	231	/*
	232	* gd lives in a fixed register which may get clobbered while we execute
	233	* the payload. So save it here and restore it on every callback entry
	234	*/
	235	efi_save_gd();
	236
	237	addr = efi_bootmgr_load(&device_path, &file_path);
	238	if (!addr)
	239	return 1;
	240
	241	printf("## Starting EFI application at %p ...\n", addr);
	242	r = do_bootefi_exec(addr, (void *)fdt_addr, device_path, file_path);
	243	printf("## Application terminated, r = %lu\n",
	244	r & ~EFI_ERROR_MASK);
	245
	246	if (r != EFI_SUCCESS)
	247	return 1;
	248
	249	return 0;
	250	}
	251
b9939336 AG	252	/* Interpreter command to boot an arbitrary EFI image from memory */
	253	static int do_bootefi(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
	254	{
1c39809b AG	255	char saddr, sfdt;
1c39809b AG	256	unsigned long addr, fdt_addr = 0;
1da1bac4	257	unsigned long r;
b9939336 AG	258
b9939336 AG	259	if (argc < 2)
3c1dcef6	260	return CMD_RET_USAGE;
c7ae3dfd SG	261	#ifdef CONFIG_CMD_BOOTEFI_HELLO
c7ae3dfd SG	262	if (!strcmp(argv[1], "hello")) {
5e44489b	263	ulong size = __efi_helloworld_end - __efi_helloworld_begin;
b9939336	264
51c533fd HS	265	saddr = env_get("loadaddr");
	266	if (saddr)
	267	addr = simple_strtoul(saddr, NULL, 16);
	268	else
	269	addr = CONFIG_SYS_LOAD_ADDR;
5e44489b	270	memcpy((char *)addr, __efi_helloworld_begin, size);
c7ae3dfd	271	} else
623b3a57 HS	272	#endif
	273	#ifdef CONFIG_CMD_BOOTEFI_SELFTEST
	274	if (!strcmp(argv[1], "selftest")) {
7aca68ca HS	275	struct efi_loaded_image loaded_image_info = {};
	276	struct efi_object loaded_image_info_obj = {};
	277
	278	efi_setup_loaded_image(&loaded_image_info,
	279	&loaded_image_info_obj,
	280	bootefi_device_path, bootefi_image_path);
623b3a57 HS	281	/*
	282	* gd lives in a fixed register which may get clobbered while we
	283	* execute the payload. So save it here and restore it on every
	284	* callback entry
	285	*/
	286	efi_save_gd();
	287	/* Initialize and populate EFI object list */
	288	if (!efi_obj_list_initalized)
	289	efi_init_obj_list();
623b3a57 HS	290	return efi_selftest(&loaded_image_info, &systab);
623b3a57 HS	291	} else
c7ae3dfd	292	#endif
9975fe96 RC	293	if (!strcmp(argv[1], "bootmgr")) {
	294	unsigned long fdt_addr = 0;
	295
	296	if (argc > 2)
	297	fdt_addr = simple_strtoul(argv[2], NULL, 16);
	298
	299	return do_bootefi_bootmgr_exec(fdt_addr);
	300	} else {
c7ae3dfd	301	saddr = argv[1];
b9939336	302
c7ae3dfd SG	303	addr = simple_strtoul(saddr, NULL, 16);
	304
	305	if (argc > 2) {
	306	sfdt = argv[2];
	307	fdt_addr = simple_strtoul(sfdt, NULL, 16);
	308	}
1c39809b AG	309	}
1c39809b AG	310
5ee31baf	311	printf("## Starting EFI application at %08lx ...\n", addr);
95c5553e RC	312	r = do_bootefi_exec((void )addr, (void )fdt_addr,
95c5553e RC	313	bootefi_device_path, bootefi_image_path);
1da1bac4 HS	314	printf("## Application terminated, r = %lu\n",
1da1bac4 HS	315	r & ~EFI_ERROR_MASK);
b9939336	316
1da1bac4 HS	317	if (r != EFI_SUCCESS)
	318	return 1;
	319	else
	320	return 0;
b9939336 AG	321	}
	322
	323	#ifdef CONFIG_SYS_LONGHELP
	324	static char bootefi_help_text[] =
1c39809b AG	325	"<image address> [fdt address]\n"
	326	" - boot EFI payload stored at address <image address>.\n"
	327	" If specified, the device tree located at <fdt address> gets\n"
c7ae3dfd SG	328	" exposed as EFI configuration table.\n"
c7ae3dfd SG	329	#ifdef CONFIG_CMD_BOOTEFI_HELLO
623b3a57 HS	330	"bootefi hello\n"
	331	" - boot a sample Hello World application stored within U-Boot\n"
	332	#endif
	333	#ifdef CONFIG_CMD_BOOTEFI_SELFTEST
	334	"bootefi selftest\n"
	335	" - boot an EFI selftest application stored within U-Boot\n"
c7ae3dfd	336	#endif
9975fe96 RC	337	"bootmgr [fdt addr]\n"
	338	" - load and boot EFI payload based on BootOrder/BootXXXX variables.\n"
	339	"\n"
	340	" If specified, the device tree located at <fdt address> gets\n"
	341	" exposed as EFI configuration table.\n";
b9939336 AG	342	#endif
	343
	344	U_BOOT_CMD(
1c39809b	345	bootefi, 3, 0, do_bootefi,
92dfd922	346	"Boots an EFI payload from memory",
b9939336 AG	347	bootefi_help_text
b9939336 AG	348	);
0f4060eb	349
95c5553e	350	static int parse_partnum(const char *devnr)
0f4060eb	351	{
95c5553e RC	352	const char *str = strchr(devnr, ':');
	353	if (str) {
	354	str++;
	355	return simple_strtoul(str, NULL, 16);
f9d334bd	356	}
95c5553e RC	357	return 0;
95c5553e RC	358	}
f9d334bd	359
95c5553e RC	360	void efi_set_bootdev(const char dev, const char devnr, const char *path)
	361	{
	362	char filename[32] = { 0 }; /* dp->str is u16[32] long */
	363	char *s;
f9d334bd	364
95c5553e RC	365	if (strcmp(dev, "Net")) {
	366	struct blk_desc *desc;
	367	int part;
8c3df0bf	368
95c5553e RC	369	desc = blk_get_dev(dev, simple_strtol(devnr, NULL, 10));
95c5553e RC	370	part = parse_partnum(devnr);
0f4060eb	371
95c5553e RC	372	bootefi_device_path = efi_dp_from_part(desc, part);
	373	} else {
	374	#ifdef CONFIG_NET
	375	bootefi_device_path = efi_dp_from_eth();
	376	#endif
	377	}
c07ad7c0	378
9975fe96 RC	379	if (!path)
	380	return;
	381
49271666 AG	382	if (strcmp(dev, "Net")) {
49271666 AG	383	/* Add leading / to fs paths, because they're absolute */
95c5553e	384	snprintf(filename, sizeof(filename), "/%s", path);
49271666	385	} else {
95c5553e	386	snprintf(filename, sizeof(filename), "%s", path);
49271666	387	}
3e433e96	388	/* DOS style file path: */
95c5553e	389	s = filename;
3e433e96 RC	390	while ((s = strchr(s, '/')))
3e433e96 RC	391	*s++ = '\\';
95c5553e	392	bootefi_image_path = efi_dp_from_file(NULL, 0, filename);
0f4060eb	393	}