efi_loader: Pass proper device path in on boot

[oweals/u-boot.git] / cmd / fs.c

diff --git a/cmd/fs.c b/cmd/fs.c
index 8f8f1b2bfcf108988bd4ac46d32b7a7c64380910..be8f2899e93aa052be21a4cd5e154823575b8177 100644 (file)
--- a/cmd/fs.c
+++ b/cmd/fs.c
@@ -9,6 +9,7 @@
 #include <common.h>
 #include <command.h>
 #include <fs.h>
+#include <efi_loader.h>
 
 static int do_size_wrapper(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 {
@@ -26,6 +27,7 @@ U_BOOT_CMD(
 static int do_load_wrapper(cmd_tbl_t *cmdtp, int flag, int argc,
                                char * const argv[])
 {
+       efi_set_bootdev(argv[1], (argc > 2) ? argv[2] : "");
        return do_load(cmdtp, flag, argc, argv, FS_TYPE_ANY);
 }