[people/ms/u-boot.git] / env / ubi.c

/*
 * (c) Copyright 2012 by National Instruments,
 *        Joe Hershberger <joe.hershberger@ni.com>
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>

#include <command.h>
#include <environment.h>
#include <errno.h>
#include <malloc.h>
#include <memalign.h>
#include <search.h>
#include <ubi_uboot.h>
#undef crc32

DECLARE_GLOBAL_DATA_PTR;

#ifdef CONFIG_CMD_SAVEENV
#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
static int env_ubi_save(void)
{
	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
	int ret;

	ret = env_export(env_new);
	if (ret)
		return ret;

	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
		printf("\n** Cannot find mtd partition \"%s\"\n",
		       CONFIG_ENV_UBI_PART);
		return 1;
	}

	if (gd->env_valid == ENV_VALID) {
		puts("Writing to redundant UBI... ");
		if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME_REDUND,
				     (void *)env_new, CONFIG_ENV_SIZE)) {
			printf("\n** Unable to write env to %s:%s **\n",
			       CONFIG_ENV_UBI_PART,
			       CONFIG_ENV_UBI_VOLUME_REDUND);
			return 1;
		}
	} else {
		puts("Writing to UBI... ");
		if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME,
				     (void *)env_new, CONFIG_ENV_SIZE)) {
			printf("\n** Unable to write env to %s:%s **\n",
			       CONFIG_ENV_UBI_PART,
			       CONFIG_ENV_UBI_VOLUME);
			return 1;
		}
	}

	puts("done\n");

	gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;

	return 0;
}
#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
static int env_ubi_save(void)
{
	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
	int ret;

	ret = env_export(env_new);
	if (ret)
		return ret;

	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
		printf("\n** Cannot find mtd partition \"%s\"\n",
		       CONFIG_ENV_UBI_PART);
		return 1;
	}

	if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME, (void *)env_new,
			     CONFIG_ENV_SIZE)) {
		printf("\n** Unable to write env to %s:%s **\n",
		       CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
		return 1;
	}

	puts("done\n");
	return 0;
}
#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
#endif /* CONFIG_CMD_SAVEENV */

#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
static int env_ubi_load(void)
{
	ALLOC_CACHE_ALIGN_BUFFER(char, env1_buf, CONFIG_ENV_SIZE);
	ALLOC_CACHE_ALIGN_BUFFER(char, env2_buf, CONFIG_ENV_SIZE);
	int read1_fail, read2_fail;
	env_t *tmp_env1, *tmp_env2;

	/*
	 * In case we have restarted u-boot there is a chance that buffer
	 * contains old environment (from the previous boot).
	 * If UBI volume is zero size, ubi_volume_read() doesn't modify the
	 * buffer.
	 * We need to clear buffer manually here, so the invalid CRC will
	 * cause setting default environment as expected.
	 */
	memset(env1_buf, 0x0, CONFIG_ENV_SIZE);
	memset(env2_buf, 0x0, CONFIG_ENV_SIZE);

	tmp_env1 = (env_t *)env1_buf;
	tmp_env2 = (env_t *)env2_buf;

	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
		printf("\n** Cannot find mtd partition \"%s\"\n",
		       CONFIG_ENV_UBI_PART);
		set_default_env(NULL);
		return -EIO;
	}

	read1_fail = ubi_volume_read(CONFIG_ENV_UBI_VOLUME, (void *)tmp_env1,
				     CONFIG_ENV_SIZE);
	if (read1_fail)
		printf("\n** Unable to read env from %s:%s **\n",
		       CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);

	read2_fail = ubi_volume_read(CONFIG_ENV_UBI_VOLUME_REDUND,
				     (void *)tmp_env2, CONFIG_ENV_SIZE);
	if (read2_fail)
		printf("\n** Unable to read redundant env from %s:%s **\n",
		       CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME_REDUND);

	return env_import_redund((char *)tmp_env1, read1_fail, (char *)tmp_env2,
							 read2_fail);
}
#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
static int env_ubi_load(void)
{
	ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);

	/*
	 * In case we have restarted u-boot there is a chance that buffer
	 * contains old environment (from the previous boot).
	 * If UBI volume is zero size, ubi_volume_read() doesn't modify the
	 * buffer.
	 * We need to clear buffer manually here, so the invalid CRC will
	 * cause setting default environment as expected.
	 */
	memset(buf, 0x0, CONFIG_ENV_SIZE);

	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
		printf("\n** Cannot find mtd partition \"%s\"\n",
		       CONFIG_ENV_UBI_PART);
		set_default_env(NULL);
		return -EIO;
	}

	if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME, buf, CONFIG_ENV_SIZE)) {
		printf("\n** Unable to read env from %s:%s **\n",
		       CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
		set_default_env(NULL);
		return -EIO;
	}

	return env_import(buf, 1);
}
#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */

U_BOOT_ENV_LOCATION(ubi) = {
	.location	= ENVL_UBI,
	.load		= env_ubi_load,
	.save		= env_save_ptr(env_ubi_save),
};
Commit	Line	Data
2b74433f JH	1	/*
	2	* (c) Copyright 2012 by National Instruments,
	3	* Joe Hershberger <joe.hershberger@ni.com>
	4	*
3765b3e7	5	* SPDX-License-Identifier: GPL-2.0+
2b74433f JH	6	*/
	7
	8	#include <common.h>
	9
	10	#include <command.h>
	11	#include <environment.h>
	12	#include <errno.h>
	13	#include <malloc.h>
cf92e05c	14	#include <memalign.h>
2b74433f JH	15	#include <search.h>
	16	#include <ubi_uboot.h>
	17	#undef crc32
	18
2b74433f JH	19	DECLARE_GLOBAL_DATA_PTR;
2b74433f JH	20
2b74433f	21	#ifdef CONFIG_CMD_SAVEENV
785881f7	22	#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
e5bce247	23	static int env_ubi_save(void)
785881f7 JH	24	{
785881f7 JH	25	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
7ce1526e	26	int ret;
785881f7	27
7ce1526e MV	28	ret = env_export(env_new);
	29	if (ret)
	30	return ret;
785881f7 JH	31
	32	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
	33	printf("\n** Cannot find mtd partition \"%s\"\n",
	34	CONFIG_ENV_UBI_PART);
	35	return 1;
	36	}
	37
203e94f6	38	if (gd->env_valid == ENV_VALID) {
785881f7 JH	39	puts("Writing to redundant UBI... ");
	40	if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME_REDUND,
	41	(void *)env_new, CONFIG_ENV_SIZE)) {
	42	printf("\n Unable to write env to %s:%s \n",
	43	CONFIG_ENV_UBI_PART,
	44	CONFIG_ENV_UBI_VOLUME_REDUND);
	45	return 1;
	46	}
	47	} else {
	48	puts("Writing to UBI... ");
	49	if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME,
	50	(void *)env_new, CONFIG_ENV_SIZE)) {
	51	printf("\n Unable to write env to %s:%s \n",
	52	CONFIG_ENV_UBI_PART,
	53	CONFIG_ENV_UBI_VOLUME);
	54	return 1;
	55	}
	56	}
	57
	58	puts("done\n");
	59
203e94f6	60	gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;
785881f7 JH	61
	62	return 0;
	63	}
	64	#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
e5bce247	65	static int env_ubi_save(void)
2b74433f JH	66	{
2b74433f JH	67	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
7ce1526e	68	int ret;
2b74433f	69
7ce1526e MV	70	ret = env_export(env_new);
	71	if (ret)
	72	return ret;
2b74433f JH	73
	74	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
	75	printf("\n** Cannot find mtd partition \"%s\"\n",
	76	CONFIG_ENV_UBI_PART);
	77	return 1;
	78	}
	79
2b74433f JH	80	if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME, (void *)env_new,
	81	CONFIG_ENV_SIZE)) {
	82	printf("\n Unable to write env to %s:%s \n",
	83	CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
	84	return 1;
	85	}
	86
	87	puts("done\n");
	88	return 0;
	89	}
785881f7	90	#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
2b74433f JH	91	#endif /* CONFIG_CMD_SAVEENV */
2b74433f JH	92
785881f7	93	#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
c5951991	94	static int env_ubi_load(void)
785881f7 JH	95	{
	96	ALLOC_CACHE_ALIGN_BUFFER(char, env1_buf, CONFIG_ENV_SIZE);
	97	ALLOC_CACHE_ALIGN_BUFFER(char, env2_buf, CONFIG_ENV_SIZE);
31f044bd	98	int read1_fail, read2_fail;
9d364af2	99	env_t tmp_env1, tmp_env2;
785881f7	100
c1f51e0f MN	101	/*
	102	* In case we have restarted u-boot there is a chance that buffer
	103	* contains old environment (from the previous boot).
	104	* If UBI volume is zero size, ubi_volume_read() doesn't modify the
	105	* buffer.
	106	* We need to clear buffer manually here, so the invalid CRC will
	107	* cause setting default environment as expected.
	108	*/
	109	memset(env1_buf, 0x0, CONFIG_ENV_SIZE);
	110	memset(env2_buf, 0x0, CONFIG_ENV_SIZE);
	111
785881f7 JH	112	tmp_env1 = (env_t *)env1_buf;
	113	tmp_env2 = (env_t *)env2_buf;
	114
	115	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
	116	printf("\n** Cannot find mtd partition \"%s\"\n",
	117	CONFIG_ENV_UBI_PART);
	118	set_default_env(NULL);
c5951991	119	return -EIO;
785881f7 JH	120	}
785881f7 JH	121
31f044bd SG	122	read1_fail = ubi_volume_read(CONFIG_ENV_UBI_VOLUME, (void *)tmp_env1,
	123	CONFIG_ENV_SIZE);
	124	if (read1_fail)
785881f7 JH	125	printf("\n Unable to read env from %s:%s \n",
785881f7 JH	126	CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
785881f7	127
31f044bd SG	128	read2_fail = ubi_volume_read(CONFIG_ENV_UBI_VOLUME_REDUND,
	129	(void *)tmp_env2, CONFIG_ENV_SIZE);
	130	if (read2_fail)
785881f7 JH	131	printf("\n Unable to read redundant env from %s:%s \n",
785881f7 JH	132	CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME_REDUND);
785881f7	133
31f044bd SG	134	return env_import_redund((char )tmp_env1, read1_fail, (char )tmp_env2,
31f044bd SG	135	read2_fail);
785881f7 JH	136	}
785881f7 JH	137	#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
c5951991	138	static int env_ubi_load(void)
2b74433f JH	139	{
	140	ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);
	141
c1f51e0f MN	142	/*
	143	* In case we have restarted u-boot there is a chance that buffer
	144	* contains old environment (from the previous boot).
	145	* If UBI volume is zero size, ubi_volume_read() doesn't modify the
	146	* buffer.
	147	* We need to clear buffer manually here, so the invalid CRC will
	148	* cause setting default environment as expected.
	149	*/
	150	memset(buf, 0x0, CONFIG_ENV_SIZE);
	151
2b74433f JH	152	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
	153	printf("\n** Cannot find mtd partition \"%s\"\n",
	154	CONFIG_ENV_UBI_PART);
	155	set_default_env(NULL);
c5951991	156	return -EIO;
2b74433f JH	157	}
2b74433f JH	158
a7c06cd3	159	if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME, buf, CONFIG_ENV_SIZE)) {
2b74433f JH	160	printf("\n Unable to read env from %s:%s \n",
	161	CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
	162	set_default_env(NULL);
c5951991	163	return -EIO;
2b74433f JH	164	}
2b74433f JH	165
2166ebf7	166	return env_import(buf, 1);
2b74433f	167	}
785881f7	168	#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
4415f1d1 SG	169
	170	U_BOOT_ENV_LOCATION(ubi) = {
	171	.location = ENVL_UBI,
e5bce247 SG	172	.load = env_ubi_load,
e5bce247 SG	173	.save = env_save_ptr(env_ubi_save),
4415f1d1	174	};