[U-Boot] [PATCH] env: Update env_addr for mmc environment driver

[Pankit Garg]

Signed-off-by: Pankit Garg <pankit.garg at nxp.com>
---
 env/mmc.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/env/mmc.c b/env/mmc.c
index 9f1878def1..832e11cde6 100644
--- a/env/mmc.c
+++ b/env/mmc.c
@@ -362,6 +362,7 @@ static int env_mmc_load(void)
 	int ret;
 	int dev = mmc_get_env_dev();
 	const char *errmsg;
+	env_t *ep = NULL;
 
 	mmc = find_mmc_device(dev);
 
@@ -383,6 +384,10 @@ static int env_mmc_load(void)
 	}
 
 	ret = env_import(buf, 1);
+	if (!ret) {
+		ep = (env_t *)buf;
+		gd->env_addr    = (ulong)&(ep->data);
+	}
 
 fini:
 	fini_mmc_for_env(mmc);
-- 
2.17.1