x86: Add ramboot and nfsboot commands in x86-common.h

[karo-tx-uboot.git] / common / env_mmc.c

diff --git a/common/env_mmc.c b/common/env_mmc.c
index c99fc750fa0a0b95db379b673423866d71b77a82..14648e37bdf284ccb683740b5ce3d05df47e3779 100644 (file)
--- a/common/env_mmc.c
+++ b/common/env_mmc.c
@@ -62,6 +62,34 @@ int env_init(void)
        return 0;
 }
 
+#ifdef CONFIG_SYS_MMC_ENV_PART
+__weak uint mmc_get_env_part(struct mmc *mmc)
+{
+       return CONFIG_SYS_MMC_ENV_PART;
+}
+
+static int mmc_set_env_part(struct mmc *mmc)
+{
+       uint part = mmc_get_env_part(mmc);
+       int dev = CONFIG_SYS_MMC_ENV_DEV;
+       int ret = 0;
+
+#ifdef CONFIG_SPL_BUILD
+       dev = 0;
+#endif
+
+       if (part != mmc->part_num) {
+               ret = mmc_switch_part(dev, part);
+               if (ret)
+                       puts("MMC partition switch failed\n");
+       }
+
+       return ret;
+}
+#else
+static inline int mmc_set_env_part(struct mmc *mmc) {return 0; };
+#endif
+
 static int init_mmc_for_env(struct mmc *mmc)
 {
        if (!mmc) {
@@ -74,25 +102,19 @@ static int init_mmc_for_env(struct mmc *mmc)
                return -1;
        }
 
-#ifdef CONFIG_SYS_MMC_ENV_PART
-       if (CONFIG_SYS_MMC_ENV_PART != mmc->part_num) {
-               if (mmc_switch_part(CONFIG_SYS_MMC_ENV_DEV,
-                                   CONFIG_SYS_MMC_ENV_PART)) {
-                       puts("MMC partition switch failed\n");
-                       return -1;
-               }
-       }
-#endif
-
-       return 0;
+       return mmc_set_env_part(mmc);
 }
 
 static void fini_mmc_for_env(struct mmc *mmc)
 {
 #ifdef CONFIG_SYS_MMC_ENV_PART
-       if (CONFIG_SYS_MMC_ENV_PART != mmc->part_num)
-               mmc_switch_part(CONFIG_SYS_MMC_ENV_DEV,
-                               mmc->part_num);
+       int dev = CONFIG_SYS_MMC_ENV_DEV;
+
+#ifdef CONFIG_SPL_BUILD
+       dev = 0;
+#endif
+       if (mmc_get_env_part(mmc) != mmc->part_num)
+               mmc_switch_part(dev, mmc->part_num);
 #endif
 }
 
@@ -166,12 +188,16 @@ static inline int read_env(struct mmc *mmc, unsigned long size,
                           unsigned long offset, const void *buffer)
 {
        uint blk_start, blk_cnt, n;
+       int dev = CONFIG_SYS_MMC_ENV_DEV;
+
+#ifdef CONFIG_SPL_BUILD
+       dev = 0;
+#endif
 
        blk_start       = ALIGN(offset, mmc->read_bl_len) / mmc->read_bl_len;
        blk_cnt         = ALIGN(size, mmc->read_bl_len) / mmc->read_bl_len;
 
-       n = mmc->block_dev.block_read(CONFIG_SYS_MMC_ENV_DEV, blk_start,
-                                       blk_cnt, (uchar *)buffer);
+       n = mmc->block_dev.block_read(dev, blk_start, blk_cnt, (uchar *)buffer);
 
        return (n == blk_cnt) ? 0 : -1;
 }
@@ -180,21 +206,22 @@ static inline int read_env(struct mmc *mmc, unsigned long size,
 void env_relocate_spec(void)
 {
 #if !defined(ENV_IS_EMBEDDED)
-       struct mmc *mmc = find_mmc_device(CONFIG_SYS_MMC_ENV_DEV);
+       struct mmc *mmc;
        u32 offset1, offset2;
        int read1_fail = 0, read2_fail = 0;
        int crc1_ok = 0, crc2_ok = 0;
        env_t *ep;
        int ret;
+       int dev = CONFIG_SYS_MMC_ENV_DEV;
 
        ALLOC_CACHE_ALIGN_BUFFER(env_t, tmp_env1, 1);
        ALLOC_CACHE_ALIGN_BUFFER(env_t, tmp_env2, 1);
 
-       if (tmp_env1 == NULL || tmp_env2 == NULL) {
-               puts("Can't allocate buffers for environment\n");
-               ret = 1;
-               goto err;
-       }
+#ifdef CONFIG_SPL_BUILD
+       dev = 0;
+#endif
+
+       mmc = find_mmc_device(dev);
 
        if (init_mmc_for_env(mmc)) {
                ret = 1;
@@ -266,9 +293,16 @@ void env_relocate_spec(void)
 {
 #if !defined(ENV_IS_EMBEDDED)
        ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);
-       struct mmc *mmc = find_mmc_device(CONFIG_SYS_MMC_ENV_DEV);
+       struct mmc *mmc;
        u32 offset;
        int ret;
+       int dev = CONFIG_SYS_MMC_ENV_DEV;
+
+#ifdef CONFIG_SPL_BUILD
+       dev = 0;
+#endif
+
+       mmc = find_mmc_device(dev);
 
        if (init_mmc_for_env(mmc)) {
                ret = 1;