Create a new API spl_usb_load() that takes the filename as a parameter
instead of taking the default U-boot PAYLOAD_NAME

Signed-off-by: Faiz Abbas <faiz_ab...@ti.com>
---
 common/spl/spl_usb.c | 20 +++++++++++++-------
 include/spl.h        | 14 ++++++++++++++
 2 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/common/spl/spl_usb.c b/common/spl/spl_usb.c
index 08837b38fc..92ae96f66e 100644
--- a/common/spl/spl_usb.c
+++ b/common/spl/spl_usb.c
@@ -18,8 +18,9 @@
 
 static int usb_stor_curr_dev = -1; /* current device */
 
-static int spl_usb_load_image(struct spl_image_info *spl_image,
-                             struct spl_boot_device *bootdev)
+int spl_usb_load(struct spl_image_info *spl_image,
+                struct spl_boot_device *bootdev, int partition,
+                const char *filename)
 {
        int err;
        struct blk_desc *stor_dev;
@@ -43,13 +44,10 @@ static int spl_usb_load_image(struct spl_image_info 
*spl_image,
 
 #ifdef CONFIG_SPL_OS_BOOT
        if (spl_start_uboot() ||
-           spl_load_image_fat_os(spl_image, stor_dev,
-                                 CONFIG_SYS_USB_FAT_BOOT_PARTITION))
+           spl_load_image_fat_os(spl_image, stor_dev, partition))
 #endif
        {
-               err = spl_load_image_fat(spl_image, stor_dev,
-                                       CONFIG_SYS_USB_FAT_BOOT_PARTITION,
-                                       CONFIG_SPL_FS_LOAD_PAYLOAD_NAME);
+               err = spl_load_image_fat(spl_image, stor_dev, partition, 
filename);
        }
 
        if (err) {
@@ -59,4 +57,12 @@ static int spl_usb_load_image(struct spl_image_info 
*spl_image,
 
        return 0;
 }
+
+static int spl_usb_load_image(struct spl_image_info *spl_image,
+                             struct spl_boot_device *bootdev)
+{
+       return spl_usb_load(spl_image, bootdev,
+                           CONFIG_SYS_USB_FAT_BOOT_PARTITION,
+                           CONFIG_SPL_FS_LOAD_PAYLOAD_NAME);
+}
 SPL_LOAD_IMAGE_METHOD("USB", 0, BOOT_DEVICE_USB, spl_usb_load_image);
diff --git a/include/spl.h b/include/spl.h
index b31c9bb4ab..9c050efcf9 100644
--- a/include/spl.h
+++ b/include/spl.h
@@ -503,6 +503,20 @@ int spl_mmc_load(struct spl_image_info *spl_image,
                 int raw_part,
                 unsigned long raw_sect);
 
+/**
+ * spl_usb_load() - Load an image file from USB mass storage
+ *
+ * @param spl_image    Image data filled in by loading process
+ * @param bootdev      Describes which device to load from
+ * @param raw_part     Fat partition to load from
+ * @param filename     Name of file to load
+ *
+ * @return 0 on success, otherwise error code
+ */
+int spl_usb_load(struct spl_image_info *spl_image,
+                struct spl_boot_device *bootdev,
+                int partition, const char *filename);
+
 int spl_ymodem_load_image(struct spl_image_info *spl_image,
                          struct spl_boot_device *bootdev);
 
-- 
2.17.1

Reply via email to