Merge tag 'u-boot-atmel-fixes-2020.07-a' of https://gitlab.denx.de/u-boot/custodians...

[oweals/u-boot.git] / arch / arm / mach-uniphier / mmc-first-dev.c

diff --git a/arch/arm/mach-uniphier/mmc-first-dev.c b/arch/arm/mach-uniphier/mmc-first-dev.c
index 2f1c109b9e2c18696f796856b80ad1e8fb45cbf2..45bb10f8eb1f9c85a28fe40bd70d08960a1fbe01 100644 (file)
--- a/arch/arm/mach-uniphier/mmc-first-dev.c
+++ b/arch/arm/mach-uniphier/mmc-first-dev.c
@@ -4,17 +4,19 @@
  *   Author: Masahiro Yamada <yamada.masahiro@socionext.com>
  */
 
-#include <common.h>
+#include <command.h>
+#include <env.h>
 #include <mmc.h>
 #include <linux/errno.h>
 
-static int find_first_mmc_device(void)
+static int find_first_mmc_device(bool is_sd)
 {
        struct mmc *mmc;
        int i;
 
        for (i = 0; (mmc = find_mmc_device(i)); i++) {
-               if (!mmc_init(mmc) && IS_MMC(mmc))
+               if (!mmc_init(mmc) &&
+                   ((is_sd && IS_SD(mmc)) || (!is_sd && IS_MMC(mmc))))
                        return i;
        }
 
@@ -23,14 +25,15 @@ static int find_first_mmc_device(void)
 
 int mmc_get_env_dev(void)
 {
-       return find_first_mmc_device();
+       return find_first_mmc_device(false);
 }
 
-static int do_mmcsetn(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
+static int do_mmcsetn(struct cmd_tbl *cmdtp, int flag, int argc,
+                     char *const argv[])
 {
        int dev;
 
-       dev = find_first_mmc_device();
+       dev = find_first_mmc_device(false);
        if (dev < 0)
                return CMD_RET_FAILURE;
 
@@ -43,3 +46,22 @@ U_BOOT_CMD(
        "Set the first MMC (not SD) dev number to \"mmc_first_dev\" environment",
        ""
 );
+
+static int do_sdsetn(struct cmd_tbl *cmdtp, int flag, int argc,
+                    char *const argv[])
+{
+       int dev;
+
+       dev = find_first_mmc_device(true);
+       if (dev < 0)
+               return CMD_RET_FAILURE;
+
+       env_set_ulong("sd_first_dev", dev);
+       return CMD_RET_SUCCESS;
+}
+
+U_BOOT_CMD(
+       sdsetn, 1,      1,      do_sdsetn,
+       "Set the first SD dev number to \"sd_first_dev\" environment",
+       ""
+);