[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

env_t *env_ptr;

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 void env_ubi_load(void)
{
	ALLOC_CACHE_ALIGN_BUFFER(char, env1_buf, CONFIG_ENV_SIZE);
	ALLOC_CACHE_ALIGN_BUFFER(char, env2_buf, CONFIG_ENV_SIZE);
	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;
	}

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

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

	env_import_redund((char *)tmp_env1, (char *)tmp_env2);
}
#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
static void 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;
	}

	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;
	}

	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	env_t *env_ptr;
	20
	21	DECLARE_GLOBAL_DATA_PTR;
	22
2b74433f	23	#ifdef CONFIG_CMD_SAVEENV
785881f7	24	#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
e5bce247	25	static int env_ubi_save(void)
785881f7 JH	26	{
785881f7 JH	27	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
7ce1526e	28	int ret;
785881f7	29
7ce1526e MV	30	ret = env_export(env_new);
	31	if (ret)
	32	return ret;
785881f7 JH	33
	34	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
	35	printf("\n** Cannot find mtd partition \"%s\"\n",
	36	CONFIG_ENV_UBI_PART);
	37	return 1;
	38	}
	39
203e94f6	40	if (gd->env_valid == ENV_VALID) {
785881f7 JH	41	puts("Writing to redundant UBI... ");
	42	if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME_REDUND,
	43	(void *)env_new, CONFIG_ENV_SIZE)) {
	44	printf("\n Unable to write env to %s:%s \n",
	45	CONFIG_ENV_UBI_PART,
	46	CONFIG_ENV_UBI_VOLUME_REDUND);
	47	return 1;
	48	}
	49	} else {
	50	puts("Writing to UBI... ");
	51	if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME,
	52	(void *)env_new, CONFIG_ENV_SIZE)) {
	53	printf("\n Unable to write env to %s:%s \n",
	54	CONFIG_ENV_UBI_PART,
	55	CONFIG_ENV_UBI_VOLUME);
	56	return 1;
	57	}
	58	}
	59
	60	puts("done\n");
	61
203e94f6	62	gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;
785881f7 JH	63
	64	return 0;
	65	}
	66	#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
e5bce247	67	static int env_ubi_save(void)
2b74433f JH	68	{
2b74433f JH	69	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
7ce1526e	70	int ret;
2b74433f	71
7ce1526e MV	72	ret = env_export(env_new);
	73	if (ret)
	74	return ret;
2b74433f JH	75
	76	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
	77	printf("\n** Cannot find mtd partition \"%s\"\n",
	78	CONFIG_ENV_UBI_PART);
	79	return 1;
	80	}
	81
2b74433f JH	82	if (ubi_volume_write(CONFIG_ENV_UBI_VOLUME, (void *)env_new,
	83	CONFIG_ENV_SIZE)) {
	84	printf("\n Unable to write env to %s:%s \n",
	85	CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
	86	return 1;
	87	}
	88
	89	puts("done\n");
	90	return 0;
	91	}
785881f7	92	#endif /* CONFIG_SYS_REDUNDAND_ENVIRONMENT */
2b74433f JH	93	#endif /* CONFIG_CMD_SAVEENV */
2b74433f JH	94
785881f7	95	#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
e5bce247	96	static void env_ubi_load(void)
785881f7 JH	97	{
	98	ALLOC_CACHE_ALIGN_BUFFER(char, env1_buf, CONFIG_ENV_SIZE);
	99	ALLOC_CACHE_ALIGN_BUFFER(char, env2_buf, CONFIG_ENV_SIZE);
9d364af2	100	env_t tmp_env1, tmp_env2;
785881f7	101
c1f51e0f MN	102	/*
	103	* In case we have restarted u-boot there is a chance that buffer
	104	* contains old environment (from the previous boot).
	105	* If UBI volume is zero size, ubi_volume_read() doesn't modify the
	106	* buffer.
	107	* We need to clear buffer manually here, so the invalid CRC will
	108	* cause setting default environment as expected.
	109	*/
	110	memset(env1_buf, 0x0, CONFIG_ENV_SIZE);
	111	memset(env2_buf, 0x0, CONFIG_ENV_SIZE);
	112
785881f7 JH	113	tmp_env1 = (env_t *)env1_buf;
	114	tmp_env2 = (env_t *)env2_buf;
	115
	116	if (ubi_part(CONFIG_ENV_UBI_PART, NULL)) {
	117	printf("\n** Cannot find mtd partition \"%s\"\n",
	118	CONFIG_ENV_UBI_PART);
	119	set_default_env(NULL);
	120	return;
	121	}
	122
	123	if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME, (void *)tmp_env1,
	124	CONFIG_ENV_SIZE)) {
	125	printf("\n Unable to read env from %s:%s \n",
	126	CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME);
	127	}
	128
	129	if (ubi_volume_read(CONFIG_ENV_UBI_VOLUME_REDUND, (void *)tmp_env2,
	130	CONFIG_ENV_SIZE)) {
	131	printf("\n Unable to read redundant env from %s:%s \n",
	132	CONFIG_ENV_UBI_PART, CONFIG_ENV_UBI_VOLUME_REDUND);
	133	}
	134
9d364af2	135	env_import_redund((char )tmp_env1, (char )tmp_env2);
785881f7 JH	136	}
785881f7 JH	137	#else /* ! CONFIG_SYS_REDUNDAND_ENVIRONMENT */
e5bce247	138	static void 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);
	156	return;
	157	}
	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);
	163	return;
	164	}
	165
	166	env_import(buf, 1);
	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	};