[people/ms/u-boot.git] / common / 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

char *env_name_spec = "UBI";

env_t *env_ptr;

DECLARE_GLOBAL_DATA_PTR;

int env_init(void)
{
	/* use default */
	gd->env_addr = (ulong)&default_environment[0];
	gd->env_valid = 1;

	return 0;
}

#ifdef CONFIG_CMD_SAVEENV
#ifdef CONFIG_SYS_REDUNDAND_ENVIRONMENT
int saveenv(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 == 1) {
		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 == 2 ? 1 : 2;

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