[thirdparty/u-boot.git] / env / mmc.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * (C) Copyright 2008-2011 Freescale Semiconductor, Inc.
 */

/* #define DEBUG */

#include <asm/global_data.h>

#include <command.h>
#include <env.h>
#include <env_internal.h>
#include <fdtdec.h>
#include <linux/stddef.h>
#include <malloc.h>
#include <memalign.h>
#include <mmc.h>
#include <part.h>
#include <search.h>
#include <errno.h>
#include <dm/ofnode.h>

#define ENV_MMC_INVALID_OFFSET ((s64)-1)

#if defined(CONFIG_ENV_MMC_USE_DT)
/* ENV offset is invalid when not defined in Device Tree */
#define ENV_MMC_OFFSET		ENV_MMC_INVALID_OFFSET
#define ENV_MMC_OFFSET_REDUND	ENV_MMC_INVALID_OFFSET

#else
/* Default ENV offset when not defined in Device Tree */
#define ENV_MMC_OFFSET		CONFIG_ENV_OFFSET

#if defined(CONFIG_ENV_OFFSET_REDUND)
#define ENV_MMC_OFFSET_REDUND	CONFIG_ENV_OFFSET_REDUND
#else
#define ENV_MMC_OFFSET_REDUND	ENV_MMC_INVALID_OFFSET
#endif
#endif

DECLARE_GLOBAL_DATA_PTR;

/*
 * In case the environment is redundant, stored in eMMC hardware boot
 * partition and the environment and redundant environment offsets are
 * identical, store the environment and redundant environment in both
 * eMMC boot partitions, one copy in each.
 * */
#if (defined(CONFIG_SYS_REDUNDAND_ENVIRONMENT) && \
     (CONFIG_SYS_MMC_ENV_PART == 1) && \
     (CONFIG_ENV_OFFSET == CONFIG_ENV_OFFSET_REDUND))
#define ENV_MMC_HWPART_REDUND	1
#endif

#if CONFIG_IS_ENABLED(OF_CONTROL)
static inline int mmc_offset_try_partition(const char *str, int copy, s64 *val)
{
	struct disk_partition info;
	struct blk_desc *desc;
	int len, i, ret;
	char dev_str[4];

	snprintf(dev_str, sizeof(dev_str), "%d", mmc_get_env_dev());
	ret = blk_get_device_by_str("mmc", dev_str, &desc);
	if (ret < 0)
		return (ret);

	for (i = 1;;i++) {
		ret = part_get_info(desc, i, &info);
		if (ret < 0)
			return ret;

		if (str && !strncmp((const char *)info.name, str, sizeof(info.name)))
			break;
#ifdef CONFIG_PARTITION_TYPE_GUID
		if (!str) {
			const efi_guid_t env_guid = PARTITION_U_BOOT_ENVIRONMENT;
			efi_guid_t type_guid;

			uuid_str_to_bin(info.type_guid, type_guid.b, UUID_STR_FORMAT_GUID);
			if (!memcmp(&env_guid, &type_guid, sizeof(efi_guid_t)))
				break;
		}
#endif
	}

	/* round up to info.blksz */
	len = DIV_ROUND_UP(CONFIG_ENV_SIZE, info.blksz);

	/* use the top of the partion for the environment */
	*val = (info.start + info.size - (1 + copy) * len) * info.blksz;

	return 0;
}

static inline s64 mmc_offset(struct mmc *mmc, int copy)
{
	const struct {
		const char *offset_redund;
		const char *partition;
		const char *offset;
	} dt_prop = {
		.offset_redund = "u-boot,mmc-env-offset-redundant",
		.partition = "u-boot,mmc-env-partition",
		.offset = "u-boot,mmc-env-offset",
	};
	s64 val = 0, defvalue;
	const char *propname;
	const char *str;
	int hwpart = 0;
	int err;

	if (IS_ENABLED(CONFIG_SYS_MMC_ENV_PART))
		hwpart = mmc_get_env_part(mmc);

#if defined(CONFIG_ENV_MMC_PARTITION)
	str = CONFIG_ENV_MMC_PARTITION;
#else
	/* look for the partition in mmc CONFIG_SYS_MMC_ENV_DEV */
	str = ofnode_conf_read_str(dt_prop.partition);
#endif

	if (str) {
		/* try to place the environment at end of the partition */
		err = mmc_offset_try_partition(str, copy, &val);
		if (!err)
			return val;
		debug("env partition '%s' not found (%d)", str, err);
	}

	/* try the GPT partition with "U-Boot ENV" TYPE GUID */
	if (IS_ENABLED(CONFIG_PARTITION_TYPE_GUID) && hwpart == 0) {
		err = mmc_offset_try_partition(NULL, copy, &val);
		if (!err)
			return val;
	}

	defvalue = ENV_MMC_OFFSET;
	propname = dt_prop.offset;

	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT) && copy) {
		defvalue = ENV_MMC_OFFSET_REDUND;
		propname = dt_prop.offset_redund;
	}

	return ofnode_conf_read_int(propname, defvalue);
}
#else
static inline s64 mmc_offset(struct mmc *mmc, int copy)
{
	s64 offset = ENV_MMC_OFFSET;

	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT) && copy)
		offset = ENV_MMC_OFFSET_REDUND;

	return offset;
}
#endif

__weak int mmc_get_env_addr(struct mmc *mmc, int copy, u32 *env_addr)
{
	s64 offset = mmc_offset(mmc, copy);

	if (offset == ENV_MMC_INVALID_OFFSET) {
		printf("Invalid ENV offset in MMC, copy=%d\n", copy);
		return -ENOENT;
	}

	if (offset < 0)
		offset += mmc->capacity;

	*env_addr = offset;

	return 0;
}

#ifdef CONFIG_SYS_MMC_ENV_PART
__weak uint mmc_get_env_part(struct mmc *mmc)
{
	return CONFIG_SYS_MMC_ENV_PART;
}

static unsigned char env_mmc_orig_hwpart;

static int mmc_set_env_part(struct mmc *mmc, uint part)
{
	int dev = mmc_get_env_dev();
	int ret = 0;

	ret = blk_select_hwpart_devnum(UCLASS_MMC, dev, part);
	if (ret)
		puts("MMC partition switch failed\n");

	return ret;
}

static bool mmc_set_env_part_init(struct mmc *mmc)
{
	env_mmc_orig_hwpart = mmc_get_blk_desc(mmc)->hwpart;
	if (mmc_set_env_part(mmc, mmc_get_env_part(mmc)))
		return false;

	return true;
}

static int mmc_set_env_part_restore(struct mmc *mmc)
{
	return mmc_set_env_part(mmc, env_mmc_orig_hwpart);
}
#else
static inline int mmc_set_env_part(struct mmc *mmc, uint part) {return 0; };
static bool mmc_set_env_part_init(struct mmc *mmc) {return true; }
static inline int mmc_set_env_part_restore(struct mmc *mmc) {return 0; };
#endif

static const char *init_mmc_for_env(struct mmc *mmc)
{
	if (!mmc)
		return "No MMC card found";

#if CONFIG_IS_ENABLED(BLK)
	struct udevice *dev;

	if (blk_get_from_parent(mmc->dev, &dev))
		return "No block device";
#else
	if (mmc_init(mmc))
		return "MMC init failed";
#endif
	if (!mmc_set_env_part_init(mmc))
		return "MMC partition switch failed";

	return NULL;
}

static void fini_mmc_for_env(struct mmc *mmc)
{
	mmc_set_env_part_restore(mmc);
}

#if defined(CONFIG_CMD_SAVEENV) && !defined(CONFIG_SPL_BUILD)
static inline int write_env(struct mmc *mmc, unsigned long size,
			    unsigned long offset, const void *buffer)
{
	uint blk_start, blk_cnt, n;
	struct blk_desc *desc = mmc_get_blk_desc(mmc);

	blk_start	= ALIGN(offset, mmc->write_bl_len) / mmc->write_bl_len;
	blk_cnt		= ALIGN(size, mmc->write_bl_len) / mmc->write_bl_len;

	n = blk_dwrite(desc, blk_start, blk_cnt, (u_char *)buffer);

	return (n == blk_cnt) ? 0 : -1;
}

static int env_mmc_save(void)
{
	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
	int dev = mmc_get_env_dev();
	struct mmc *mmc = find_mmc_device(dev);
	u32	offset;
	int	ret, copy = 0;
	const char *errmsg;

	errmsg = init_mmc_for_env(mmc);
	if (errmsg) {
		printf("%s\n", errmsg);
		return 1;
	}

	ret = env_export(env_new);
	if (ret)
		goto fini;

	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT)) {
		if (gd->env_valid == ENV_VALID)
			copy = 1;

		if (IS_ENABLED(ENV_MMC_HWPART_REDUND)) {
			ret = mmc_set_env_part(mmc, copy + 1);
			if (ret)
				goto fini;
		}
	}

	if (mmc_get_env_addr(mmc, copy, &offset)) {
		ret = 1;
		goto fini;
	}

	printf("Writing to %sMMC(%d)... ", copy ? "redundant " : "", dev);
	if (write_env(mmc, CONFIG_ENV_SIZE, offset, (u_char *)env_new)) {
		puts("failed\n");
		ret = 1;
		goto fini;
	}

	ret = 0;

	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT))
		gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;

fini:
	fini_mmc_for_env(mmc);

	return ret;
}

static inline int erase_env(struct mmc *mmc, unsigned long size,
			    unsigned long offset)
{
	uint blk_start, blk_cnt, n;
	struct blk_desc *desc = mmc_get_blk_desc(mmc);
	u32 erase_size;

	erase_size = mmc->erase_grp_size * desc->blksz;
	blk_start = ALIGN_DOWN(offset, erase_size) / desc->blksz;
	blk_cnt = ALIGN(size, erase_size) / desc->blksz;

	n = blk_derase(desc, blk_start, blk_cnt);
	printf("%d blocks erased at 0x%x: %s\n", n, blk_start,
	       (n == blk_cnt) ? "OK" : "ERROR");

	return (n == blk_cnt) ? 0 : 1;
}

static int env_mmc_erase(void)
{
	int dev = mmc_get_env_dev();
	struct mmc *mmc = find_mmc_device(dev);
	int	ret, copy = 0;
	u32	offset;
	const char *errmsg;

	errmsg = init_mmc_for_env(mmc);
	if (errmsg) {
		printf("%s\n", errmsg);
		return 1;
	}

	if (mmc_get_env_addr(mmc, copy, &offset)) {
		ret = CMD_RET_FAILURE;
		goto fini;
	}

	printf("\n");
	ret = erase_env(mmc, CONFIG_ENV_SIZE, offset);

	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT)) {
		copy = 1;

		if (IS_ENABLED(ENV_MMC_HWPART_REDUND)) {
			ret = mmc_set_env_part(mmc, copy + 1);
			if (ret)
				goto fini;
		}

		if (mmc_get_env_addr(mmc, copy, &offset)) {
			ret = CMD_RET_FAILURE;
			goto fini;
		}

		ret |= erase_env(mmc, CONFIG_ENV_SIZE, offset);
	}

fini:
	fini_mmc_for_env(mmc);
	return ret;
}
#endif /* CONFIG_CMD_SAVEENV && !CONFIG_SPL_BUILD */

static inline int read_env(struct mmc *mmc, unsigned long size,
			   unsigned long offset, const void *buffer)
{
	uint blk_start, blk_cnt, n;
	struct blk_desc *desc = mmc_get_blk_desc(mmc);

	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 = blk_dread(desc, blk_start, blk_cnt, (uchar *)buffer);

	return (n == blk_cnt) ? 0 : -1;
}

#if defined(ENV_IS_EMBEDDED)
static int env_mmc_load(void)
{
	return 0;
}
#elif defined(CONFIG_SYS_REDUNDAND_ENVIRONMENT)
static int env_mmc_load(void)
{
	struct mmc *mmc;
	u32 offset1, offset2;
	int read1_fail = 0, read2_fail = 0;
	int ret;
	int dev = mmc_get_env_dev();
	const char *errmsg = NULL;

	ALLOC_CACHE_ALIGN_BUFFER(env_t, tmp_env1, 1);
	ALLOC_CACHE_ALIGN_BUFFER(env_t, tmp_env2, 1);

	mmc_initialize(NULL);

	mmc = find_mmc_device(dev);

	errmsg = init_mmc_for_env(mmc);
	if (errmsg) {
		ret = -EIO;
		goto err;
	}

	if (mmc_get_env_addr(mmc, 0, &offset1) ||
	    mmc_get_env_addr(mmc, 1, &offset2)) {
		ret = -EIO;
		goto fini;
	}

	if (IS_ENABLED(ENV_MMC_HWPART_REDUND)) {
		ret = mmc_set_env_part(mmc, 1);
		if (ret)
			goto fini;
	}

	read1_fail = read_env(mmc, CONFIG_ENV_SIZE, offset1, tmp_env1);

	if (IS_ENABLED(ENV_MMC_HWPART_REDUND)) {
		ret = mmc_set_env_part(mmc, 2);
		if (ret)
			goto fini;
	}

	read2_fail = read_env(mmc, CONFIG_ENV_SIZE, offset2, tmp_env2);

	ret = env_import_redund((char *)tmp_env1, read1_fail, (char *)tmp_env2,
				read2_fail, H_EXTERNAL);
	printf("Reading from %sMMC(%d)... ", gd->env_valid == ENV_REDUND ? "redundant " : "", dev);

fini:
	fini_mmc_for_env(mmc);
err:
	if (ret)
		env_set_default(errmsg, 0);

	return ret;
}
#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
static int env_mmc_load(void)
{
	ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);
	struct mmc *mmc;
	u32 offset;
	int ret;
	int dev = mmc_get_env_dev();
	const char *errmsg;
	env_t *ep = NULL;

	mmc = find_mmc_device(dev);

	errmsg = init_mmc_for_env(mmc);
	if (errmsg) {
		ret = -EIO;
		goto err;
	}

	if (mmc_get_env_addr(mmc, 0, &offset)) {
		ret = -EIO;
		goto fini;
	}

	if (read_env(mmc, CONFIG_ENV_SIZE, offset, buf)) {
		errmsg = "!read failed";
		ret = -EIO;
		goto fini;
	}

	printf("Reading from MMC(%d)... ", dev);

	ret = env_import(buf, 1, H_EXTERNAL);
	if (!ret) {
		ep = (env_t *)buf;
		gd->env_addr = (ulong)&ep->data;
	}

fini:
	fini_mmc_for_env(mmc);
err:
	if (ret)
		env_set_default(errmsg, 0);

	return ret;
}
#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */

U_BOOT_ENV_LOCATION(mmc) = {
	.location	= ENVL_MMC,
	ENV_NAME("MMC")
	.load		= env_mmc_load,
#if defined(CONFIG_CMD_SAVEENV) && !defined(CONFIG_SPL_BUILD)
	.save		= env_save_ptr(env_mmc_save),
	.erase		= ENV_ERASE_PTR(env_mmc_erase)
#endif
};
Commit	Line	Data
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
a8060359	2	/*
97039ab9	3	* (C) Copyright 2008-2011 Freescale Semiconductor, Inc.
a8060359 TL	4	*/
	5
	6	/* #define DEBUG */
	7
401d1c4f	8	#include <asm/global_data.h>
a8060359 TL	9
a8060359 TL	10	#include <command.h>
0ac7d722	11	#include <env.h>
f3998fdc	12	#include <env_internal.h>
f8b8a554	13	#include <fdtdec.h>
a8060359 TL	14	#include <linux/stddef.h>
a8060359 TL	15	#include <malloc.h>
cf92e05c	16	#include <memalign.h>
a8060359	17	#include <mmc.h>
c9e87ba6	18	#include <part.h>
6d1d51b3	19	#include <search.h>
e79f4839	20	#include <errno.h>
7de8bd03	21	#include <dm/ofnode.h>
a8060359	22
f7e07a7e PD	23	#define ENV_MMC_INVALID_OFFSET ((s64)-1)
f7e07a7e PD	24
d2103e20 PD	25	#if defined(CONFIG_ENV_MMC_USE_DT)
	26	/* ENV offset is invalid when not defined in Device Tree */
	27	#define ENV_MMC_OFFSET ENV_MMC_INVALID_OFFSET
	28	#define ENV_MMC_OFFSET_REDUND ENV_MMC_INVALID_OFFSET
	29
	30	#else
f7e07a7e PD	31	/* Default ENV offset when not defined in Device Tree */
	32	#define ENV_MMC_OFFSET CONFIG_ENV_OFFSET
	33
	34	#if defined(CONFIG_ENV_OFFSET_REDUND)
	35	#define ENV_MMC_OFFSET_REDUND CONFIG_ENV_OFFSET_REDUND
	36	#else
	37	#define ENV_MMC_OFFSET_REDUND ENV_MMC_INVALID_OFFSET
	38	#endif
d2103e20	39	#endif
f7e07a7e	40
a8060359 TL	41	DECLARE_GLOBAL_DATA_PTR;
a8060359 TL	42
d11d1bec MV	43	/*
	44	* In case the environment is redundant, stored in eMMC hardware boot
	45	* partition and the environment and redundant environment offsets are
	46	* identical, store the environment and redundant environment in both
	47	* eMMC boot partitions, one copy in each.
	48	* */
	49	#if (defined(CONFIG_SYS_REDUNDAND_ENVIRONMENT) && \
	50	(CONFIG_SYS_MMC_ENV_PART == 1) && \
	51	(CONFIG_ENV_OFFSET == CONFIG_ENV_OFFSET_REDUND))
46c9016b	52	#define ENV_MMC_HWPART_REDUND 1
d11d1bec MV	53	#endif
d11d1bec MV	54
f8b8a554	55	#if CONFIG_IS_ENABLED(OF_CONTROL)
5d4f7b4e	56	static inline int mmc_offset_try_partition(const char str, int copy, s64 val)
c9e87ba6	57	{
0528979f	58	struct disk_partition info;
c9e87ba6 JRO	59	struct blk_desc *desc;
c9e87ba6 JRO	60	int len, i, ret;
2b2f7275	61	char dev_str[4];
c9e87ba6	62
2b2f7275 PD	63	snprintf(dev_str, sizeof(dev_str), "%d", mmc_get_env_dev());
2b2f7275 PD	64	ret = blk_get_device_by_str("mmc", dev_str, &desc);
c9e87ba6 JRO	65	if (ret < 0)
	66	return (ret);
	67
	68	for (i = 1;;i++) {
	69	ret = part_get_info(desc, i, &info);
	70	if (ret < 0)
	71	return ret;
	72
80105d8f	73	if (str && !strncmp((const char *)info.name, str, sizeof(info.name)))
c9e87ba6	74	break;
80105d8f PD	75	#ifdef CONFIG_PARTITION_TYPE_GUID
	76	if (!str) {
	77	const efi_guid_t env_guid = PARTITION_U_BOOT_ENVIRONMENT;
	78	efi_guid_t type_guid;
	79
	80	uuid_str_to_bin(info.type_guid, type_guid.b, UUID_STR_FORMAT_GUID);
	81	if (!memcmp(&env_guid, &type_guid, sizeof(efi_guid_t)))
	82	break;
	83	}
	84	#endif
c9e87ba6 JRO	85	}
	86
	87	/* round up to info.blksz */
76b640c3	88	len = DIV_ROUND_UP(CONFIG_ENV_SIZE, info.blksz);
c9e87ba6 JRO	89
c9e87ba6 JRO	90	/* use the top of the partion for the environment */
5d4f7b4e	91	val = (info.start + info.size - (1 + copy) len) * info.blksz;
c9e87ba6 JRO	92
	93	return 0;
	94	}
	95
5b4acb0f	96	static inline s64 mmc_offset(struct mmc *mmc, int copy)
f8b8a554	97	{
c9e87ba6 JRO	98	const struct {
	99	const char *offset_redund;
	100	const char *partition;
	101	const char *offset;
	102	} dt_prop = {
	103	.offset_redund = "u-boot,mmc-env-offset-redundant",
	104	.partition = "u-boot,mmc-env-partition",
	105	.offset = "u-boot,mmc-env-offset",
	106	};
fd374665	107	s64 val = 0, defvalue;
c9e87ba6 JRO	108	const char *propname;
c9e87ba6 JRO	109	const char *str;
5b4acb0f	110	int hwpart = 0;
c9e87ba6 JRO	111	int err;
c9e87ba6 JRO	112
5b4acb0f MV	113	if (IS_ENABLED(CONFIG_SYS_MMC_ENV_PART))
	114	hwpart = mmc_get_env_part(mmc);
	115
9e70676c EDF	116	#if defined(CONFIG_ENV_MMC_PARTITION)
	117	str = CONFIG_ENV_MMC_PARTITION;
	118	#else
c9e87ba6	119	/* look for the partition in mmc CONFIG_SYS_MMC_ENV_DEV */
7de8bd03	120	str = ofnode_conf_read_str(dt_prop.partition);
9e70676c EDF	121	#endif
9e70676c EDF	122
c9e87ba6 JRO	123	if (str) {
c9e87ba6 JRO	124	/* try to place the environment at end of the partition */
5d4f7b4e	125	err = mmc_offset_try_partition(str, copy, &val);
c9e87ba6 JRO	126	if (!err)
c9e87ba6 JRO	127	return val;
52e9aa3c	128	debug("env partition '%s' not found (%d)", str, err);
c9e87ba6 JRO	129	}
c9e87ba6 JRO	130
80105d8f	131	/* try the GPT partition with "U-Boot ENV" TYPE GUID */
5b4acb0f	132	if (IS_ENABLED(CONFIG_PARTITION_TYPE_GUID) && hwpart == 0) {
80105d8f PD	133	err = mmc_offset_try_partition(NULL, copy, &val);
	134	if (!err)
	135	return val;
	136	}
	137
f7e07a7e	138	defvalue = ENV_MMC_OFFSET;
c9e87ba6	139	propname = dt_prop.offset;
f8b8a554	140
46c9016b	141	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT) && copy) {
f7e07a7e	142	defvalue = ENV_MMC_OFFSET_REDUND;
c9e87ba6	143	propname = dt_prop.offset_redund;
f8b8a554	144	}
46c9016b	145
7de8bd03	146	return ofnode_conf_read_int(propname, defvalue);
f8b8a554 PT	147	}
f8b8a554 PT	148	#else
5b4acb0f	149	static inline s64 mmc_offset(struct mmc *mmc, int copy)
97039ab9	150	{
f7e07a7e	151	s64 offset = ENV_MMC_OFFSET;
5c088ee8	152
46c9016b	153	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT) && copy)
f7e07a7e	154	offset = ENV_MMC_OFFSET_REDUND;
46c9016b	155
f8b8a554 PT	156	return offset;
	157	}
	158	#endif
	159
	160	__weak int mmc_get_env_addr(struct mmc mmc, int copy, u32 env_addr)
	161	{
5b4acb0f	162	s64 offset = mmc_offset(mmc, copy);
5c088ee8	163
f7e07a7e PD	164	if (offset == ENV_MMC_INVALID_OFFSET) {
	165	printf("Invalid ENV offset in MMC, copy=%d\n", copy);
	166	return -ENOENT;
	167	}
	168
5c088ee8 SW	169	if (offset < 0)
	170	offset += mmc->capacity;
	171
	172	*env_addr = offset;
	173
97039ab9 MH	174	return 0;
97039ab9 MH	175	}
97039ab9	176
b9c8ccab	177	#ifdef CONFIG_SYS_MMC_ENV_PART
6e7b7df4 DL	178	__weak uint mmc_get_env_part(struct mmc *mmc)
	179	{
	180	return CONFIG_SYS_MMC_ENV_PART;
	181	}
	182
873cc1d7 SW	183	static unsigned char env_mmc_orig_hwpart;
873cc1d7 SW	184
d11d1bec	185	static int mmc_set_env_part(struct mmc *mmc, uint part)
6e7b7df4	186	{
e92029c0	187	int dev = mmc_get_env_dev();
6e7b7df4	188	int ret = 0;
b9c8ccab	189
e33a5c6b	190	ret = blk_select_hwpart_devnum(UCLASS_MMC, dev, part);
873cc1d7 SW	191	if (ret)
873cc1d7 SW	192	puts("MMC partition switch failed\n");
6e7b7df4 DL	193
	194	return ret;
	195	}
46c9016b PD	196
	197	static bool mmc_set_env_part_init(struct mmc *mmc)
	198	{
	199	env_mmc_orig_hwpart = mmc_get_blk_desc(mmc)->hwpart;
	200	if (mmc_set_env_part(mmc, mmc_get_env_part(mmc)))
	201	return false;
	202
	203	return true;
	204	}
	205
	206	static int mmc_set_env_part_restore(struct mmc *mmc)
	207	{
	208	return mmc_set_env_part(mmc, env_mmc_orig_hwpart);
	209	}
6e7b7df4	210	#else
d11d1bec	211	static inline int mmc_set_env_part(struct mmc *mmc, uint part) {return 0; };
46c9016b PD	212	static bool mmc_set_env_part_init(struct mmc *mmc) {return true; }
46c9016b PD	213	static inline int mmc_set_env_part_restore(struct mmc *mmc) {return 0; };
b9c8ccab TR	214	#endif
b9c8ccab TR	215
c75648d7	216	static const char init_mmc_for_env(struct mmc mmc)
6e7b7df4	217	{
c75648d7	218	if (!mmc)
c5d548a9	219	return "No MMC card found";
a8060359	220
d48b8d11	221	#if CONFIG_IS_ENABLED(BLK)
01b73fe6 SG	222	struct udevice *dev;
	223
	224	if (blk_get_from_parent(mmc->dev, &dev))
c5d548a9	225	return "No block device";
01b73fe6	226	#else
c75648d7	227	if (mmc_init(mmc))
c5d548a9	228	return "MMC init failed";
e7017a3c	229	#endif
46c9016b	230	if (!mmc_set_env_part_init(mmc))
c5d548a9	231	return "MMC partition switch failed";
a8060359	232
c75648d7	233	return NULL;
a8060359 TL	234	}
a8060359 TL	235
9404a5fc SW	236	static void fini_mmc_for_env(struct mmc *mmc)
9404a5fc SW	237	{
46c9016b	238	mmc_set_env_part_restore(mmc);
9404a5fc SW	239	}
9404a5fc SW	240
e5bce247	241	#if defined(CONFIG_CMD_SAVEENV) && !defined(CONFIG_SPL_BUILD)
e8db8f71 IG	242	static inline int write_env(struct mmc *mmc, unsigned long size,
e8db8f71 IG	243	unsigned long offset, const void *buffer)
a8060359 TL	244	{
a8060359 TL	245	uint blk_start, blk_cnt, n;
5461acba	246	struct blk_desc *desc = mmc_get_blk_desc(mmc);
a8060359	247
e8db8f71 IG	248	blk_start = ALIGN(offset, mmc->write_bl_len) / mmc->write_bl_len;
e8db8f71 IG	249	blk_cnt = ALIGN(size, mmc->write_bl_len) / mmc->write_bl_len;
a8060359	250
5461acba	251	n = blk_dwrite(desc, blk_start, blk_cnt, (u_char *)buffer);
a8060359 TL	252
	253	return (n == blk_cnt) ? 0 : -1;
	254	}
	255
e5bce247	256	static int env_mmc_save(void)
a8060359	257	{
cd0f4fa1	258	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
e92029c0 CG	259	int dev = mmc_get_env_dev();
e92029c0 CG	260	struct mmc *mmc = find_mmc_device(dev);
e8db8f71	261	u32 offset;
d196bd88	262	int ret, copy = 0;
c75648d7	263	const char *errmsg;
a8060359	264
c75648d7 TH	265	errmsg = init_mmc_for_env(mmc);
	266	if (errmsg) {
	267	printf("%s\n", errmsg);
97039ab9	268	return 1;
c75648d7	269	}
97039ab9	270
7ce1526e MV	271	ret = env_export(env_new);
7ce1526e MV	272	if (ret)
9404a5fc	273	goto fini;
d196bd88	274
46c9016b PD	275	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT)) {
	276	if (gd->env_valid == ENV_VALID)
	277	copy = 1;
d11d1bec	278
46c9016b PD	279	if (IS_ENABLED(ENV_MMC_HWPART_REDUND)) {
	280	ret = mmc_set_env_part(mmc, copy + 1);
	281	if (ret)
	282	goto fini;
	283	}
ccd0542a	284	}
d196bd88	285
ccd0542a YL	286	if (mmc_get_env_addr(mmc, copy, &offset)) {
	287	ret = 1;
	288	goto fini;
d196bd88 MH	289	}
d196bd88 MH	290
e92029c0	291	printf("Writing to %sMMC(%d)... ", copy ? "redundant " : "", dev);
4036b630	292	if (write_env(mmc, CONFIG_ENV_SIZE, offset, (u_char *)env_new)) {
a8060359	293	puts("failed\n");
9404a5fc SW	294	ret = 1;
9404a5fc SW	295	goto fini;
a8060359 TL	296	}
a8060359 TL	297
9404a5fc SW	298	ret = 0;
9404a5fc SW	299
46c9016b PD	300	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT))
46c9016b PD	301	gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;
d196bd88	302
9404a5fc SW	303	fini:
9404a5fc SW	304	fini_mmc_for_env(mmc);
46c9016b	305
9404a5fc	306	return ret;
a8060359	307	}
34853925	308
34853925 FW	309	static inline int erase_env(struct mmc *mmc, unsigned long size,
	310	unsigned long offset)
	311	{
	312	uint blk_start, blk_cnt, n;
	313	struct blk_desc *desc = mmc_get_blk_desc(mmc);
d7226704	314	u32 erase_size;
34853925	315
d7226704 PD	316	erase_size = mmc->erase_grp_size * desc->blksz;
	317	blk_start = ALIGN_DOWN(offset, erase_size) / desc->blksz;
	318	blk_cnt = ALIGN(size, erase_size) / desc->blksz;
34853925 FW	319
34853925 FW	320	n = blk_derase(desc, blk_start, blk_cnt);
d7226704 PD	321	printf("%d blocks erased at 0x%x: %s\n", n, blk_start,
d7226704 PD	322	(n == blk_cnt) ? "OK" : "ERROR");
34853925 FW	323
	324	return (n == blk_cnt) ? 0 : 1;
	325	}
	326
	327	static int env_mmc_erase(void)
	328	{
	329	int dev = mmc_get_env_dev();
	330	struct mmc *mmc = find_mmc_device(dev);
	331	int ret, copy = 0;
	332	u32 offset;
	333	const char *errmsg;
	334
	335	errmsg = init_mmc_for_env(mmc);
	336	if (errmsg) {
	337	printf("%s\n", errmsg);
	338	return 1;
	339	}
	340
f47f87f2 MV	341	if (mmc_get_env_addr(mmc, copy, &offset)) {
	342	ret = CMD_RET_FAILURE;
	343	goto fini;
	344	}
34853925	345
d7226704	346	printf("\n");
34853925 FW	347	ret = erase_env(mmc, CONFIG_ENV_SIZE, offset);
34853925 FW	348
46c9016b PD	349	if (IS_ENABLED(CONFIG_SYS_REDUNDAND_ENVIRONMENT)) {
46c9016b PD	350	copy = 1;
34853925	351
46c9016b PD	352	if (IS_ENABLED(ENV_MMC_HWPART_REDUND)) {
	353	ret = mmc_set_env_part(mmc, copy + 1);
	354	if (ret)
	355	goto fini;
	356	}
d11d1bec	357
46c9016b PD	358	if (mmc_get_env_addr(mmc, copy, &offset)) {
	359	ret = CMD_RET_FAILURE;
	360	goto fini;
	361	}
34853925	362
46c9016b PD	363	ret \|= erase_env(mmc, CONFIG_ENV_SIZE, offset);
46c9016b PD	364	}
34853925	365
f47f87f2 MV	366	fini:
f47f87f2 MV	367	fini_mmc_for_env(mmc);
34853925 FW	368	return ret;
34853925 FW	369	}
e5bce247	370	#endif /* CONFIG_CMD_SAVEENV && !CONFIG_SPL_BUILD */
a8060359	371
e8db8f71 IG	372	static inline int read_env(struct mmc *mmc, unsigned long size,
e8db8f71 IG	373	unsigned long offset, const void *buffer)
a8060359 TL	374	{
a8060359 TL	375	uint blk_start, blk_cnt, n;
5461acba	376	struct blk_desc *desc = mmc_get_blk_desc(mmc);
a8060359	377
e8db8f71 IG	378	blk_start = ALIGN(offset, mmc->read_bl_len) / mmc->read_bl_len;
e8db8f71 IG	379	blk_cnt = ALIGN(size, mmc->read_bl_len) / mmc->read_bl_len;
a8060359	380
5461acba	381	n = blk_dread(desc, blk_start, blk_cnt, (uchar *)buffer);
a8060359 TL	382
	383	return (n == blk_cnt) ? 0 : -1;
	384	}
	385
8566050e PD	386	#if defined(ENV_IS_EMBEDDED)
	387	static int env_mmc_load(void)
	388	{
	389	return 0;
	390	}
	391	#elif defined(CONFIG_SYS_REDUNDAND_ENVIRONMENT)
c5951991	392	static int env_mmc_load(void)
d196bd88	393	{
b9c8ccab	394	struct mmc *mmc;
d196bd88 MH	395	u32 offset1, offset2;
d196bd88 MH	396	int read1_fail = 0, read2_fail = 0;
d196bd88	397	int ret;
e92029c0	398	int dev = mmc_get_env_dev();
c75648d7	399	const char *errmsg = NULL;
d196bd88	400
452a2722 MN	401	ALLOC_CACHE_ALIGN_BUFFER(env_t, tmp_env1, 1);
	402	ALLOC_CACHE_ALIGN_BUFFER(env_t, tmp_env2, 1);
	403
26862b4a FA	404	mmc_initialize(NULL);
26862b4a FA	405
b9c8ccab TR	406	mmc = find_mmc_device(dev);
b9c8ccab TR	407
c75648d7 TH	408	errmsg = init_mmc_for_env(mmc);
c75648d7 TH	409	if (errmsg) {
c5951991	410	ret = -EIO;
d196bd88 MH	411	goto err;
	412	}
	413
	414	if (mmc_get_env_addr(mmc, 0, &offset1) \|\|
	415	mmc_get_env_addr(mmc, 1, &offset2)) {
c5951991	416	ret = -EIO;
d196bd88 MH	417	goto fini;
	418	}
	419
46c9016b PD	420	if (IS_ENABLED(ENV_MMC_HWPART_REDUND)) {
	421	ret = mmc_set_env_part(mmc, 1);
	422	if (ret)
	423	goto fini;
	424	}
d11d1bec	425
d196bd88	426	read1_fail = read_env(mmc, CONFIG_ENV_SIZE, offset1, tmp_env1);
d11d1bec	427
46c9016b PD	428	if (IS_ENABLED(ENV_MMC_HWPART_REDUND)) {
	429	ret = mmc_set_env_part(mmc, 2);
	430	if (ret)
	431	goto fini;
	432	}
d11d1bec	433
d196bd88 MH	434	read2_fail = read_env(mmc, CONFIG_ENV_SIZE, offset2, tmp_env2);
d196bd88 MH	435
31f044bd	436	ret = env_import_redund((char )tmp_env1, read1_fail, (char )tmp_env2,
890feeca	437	read2_fail, H_EXTERNAL);
97b34f6a	438	printf("Reading from %sMMC(%d)... ", gd->env_valid == ENV_REDUND ? "redundant " : "", dev);
d196bd88 MH	439
	440	fini:
	441	fini_mmc_for_env(mmc);
	442	err:
	443	if (ret)
0ac7d722	444	env_set_default(errmsg, 0);
c5951991	445
c5951991	446	return ret;
d196bd88	447	}
46c9016b	448	#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
c5951991	449	static int env_mmc_load(void)
a8060359	450	{
cd0f4fa1	451	ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);
b9c8ccab	452	struct mmc *mmc;
97039ab9	453	u32 offset;
9404a5fc	454	int ret;
e92029c0	455	int dev = mmc_get_env_dev();
c75648d7	456	const char *errmsg;
0536b440	457	env_t *ep = NULL;
b9c8ccab	458
b9c8ccab	459	mmc = find_mmc_device(dev);
a8060359	460
c75648d7 TH	461	errmsg = init_mmc_for_env(mmc);
c75648d7 TH	462	if (errmsg) {
c5951991	463	ret = -EIO;
9404a5fc SW	464	goto err;
9404a5fc SW	465	}
a8060359	466
d196bd88	467	if (mmc_get_env_addr(mmc, 0, &offset)) {
c5951991	468	ret = -EIO;
9404a5fc SW	469	goto fini;
	470	}
	471
cd0f4fa1	472	if (read_env(mmc, CONFIG_ENV_SIZE, offset, buf)) {
c75648d7	473	errmsg = "!read failed";
c5951991	474	ret = -EIO;
9404a5fc SW	475	goto fini;
9404a5fc SW	476	}
a8060359	477
97b34f6a QS	478	printf("Reading from MMC(%d)... ", dev);
97b34f6a QS	479
890feeca	480	ret = env_import(buf, 1, H_EXTERNAL);
0536b440 PG	481	if (!ret) {
	482	ep = (env_t *)buf;
	483	gd->env_addr = (ulong)&ep->data;
	484	}
9404a5fc SW	485
	486	fini:
	487	fini_mmc_for_env(mmc);
	488	err:
	489	if (ret)
0ac7d722	490	env_set_default(errmsg, 0);
8566050e	491
c5951991	492	return ret;
a8060359	493	}
46c9016b	494	#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
4415f1d1 SG	495
	496	U_BOOT_ENV_LOCATION(mmc) = {
	497	.location = ENVL_MMC,
ac358beb	498	ENV_NAME("MMC")
e5bce247	499	.load = env_mmc_load,
f7fac5e7	500	#if defined(CONFIG_CMD_SAVEENV) && !defined(CONFIG_SPL_BUILD)
e5bce247	501	.save = env_save_ptr(env_mmc_save),
1af031ac	502	.erase = ENV_ERASE_PTR(env_mmc_erase)
4415f1d1	503	#endif
4415f1d1	504	};