[J-u-boot.git] / common / env_sf.c

/*
 * (C) Copyright 2000-2010
 * Wolfgang Denk, DENX Software Engineering, [email protected].
 *
 * (C) Copyright 2001 Sysgo Real-Time Solutions, GmbH <www.elinos.com>
 * Andreas Heppel <[email protected]>
 *
 * (C) Copyright 2008 Atmel Corporation
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */
#include <common.h>
#include <environment.h>
#include <malloc.h>
#include <spi.h>
#include <spi_flash.h>
#include <search.h>
#include <errno.h>

#ifndef CONFIG_ENV_SPI_BUS
# define CONFIG_ENV_SPI_BUS	0
#endif
#ifndef CONFIG_ENV_SPI_CS
# define CONFIG_ENV_SPI_CS	0
#endif
#ifndef CONFIG_ENV_SPI_MAX_HZ
# define CONFIG_ENV_SPI_MAX_HZ	1000000
#endif
#ifndef CONFIG_ENV_SPI_MODE
# define CONFIG_ENV_SPI_MODE	SPI_MODE_3
#endif

#ifdef CONFIG_ENV_OFFSET_REDUND
static ulong env_offset		= CONFIG_ENV_OFFSET;
static ulong env_new_offset	= CONFIG_ENV_OFFSET_REDUND;

#define ACTIVE_FLAG	1
#define OBSOLETE_FLAG	0
#endif /* CONFIG_ENV_OFFSET_REDUND */

DECLARE_GLOBAL_DATA_PTR;

char *env_name_spec = "SPI Flash";

static struct spi_flash *env_flash;

#if defined(CONFIG_ENV_OFFSET_REDUND)
int saveenv(void)
{
	env_t	env_new;
	char	*saved_buffer = NULL, flag = OBSOLETE_FLAG;
	u32	saved_size, saved_offset, sector = 1;
	int	ret;

	if (!env_flash) {
		env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS,
			CONFIG_ENV_SPI_CS,
			CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
		if (!env_flash) {
			set_default_env("!spi_flash_probe() failed");
			return 1;
		}
	}

	ret = env_export(&env_new);
	if (ret)
		return ret;
	env_new.flags	= ACTIVE_FLAG;

	if (gd->env_valid == 1) {
		env_new_offset = CONFIG_ENV_OFFSET_REDUND;
		env_offset = CONFIG_ENV_OFFSET;
	} else {
		env_new_offset = CONFIG_ENV_OFFSET;
		env_offset = CONFIG_ENV_OFFSET_REDUND;
	}

	/* Is the sector larger than the env (i.e. embedded) */
	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
		saved_size = CONFIG_ENV_SECT_SIZE - CONFIG_ENV_SIZE;
		saved_offset = env_new_offset + CONFIG_ENV_SIZE;
		saved_buffer = memalign(ARCH_DMA_MINALIGN, saved_size);
		if (!saved_buffer) {
			ret = 1;
			goto done;
		}
		ret = spi_flash_read(env_flash, saved_offset,
					saved_size, saved_buffer);
		if (ret)
			goto done;
	}

	if (CONFIG_ENV_SIZE > CONFIG_ENV_SECT_SIZE) {
		sector = CONFIG_ENV_SIZE / CONFIG_ENV_SECT_SIZE;
		if (CONFIG_ENV_SIZE % CONFIG_ENV_SECT_SIZE)
			sector++;
	}

	puts("Erasing SPI flash...");
	ret = spi_flash_erase(env_flash, env_new_offset,
				sector * CONFIG_ENV_SECT_SIZE);
	if (ret)
		goto done;

	puts("Writing to SPI flash...");

	ret = spi_flash_write(env_flash, env_new_offset,
		CONFIG_ENV_SIZE, &env_new);
	if (ret)
		goto done;

	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
		ret = spi_flash_write(env_flash, saved_offset,
					saved_size, saved_buffer);
		if (ret)
			goto done;
	}

	ret = spi_flash_write(env_flash, env_offset + offsetof(env_t, flags),
				sizeof(env_new.flags), &flag);
	if (ret)
		goto done;

	puts("done\n");

	gd->env_valid = gd->env_valid == 2 ? 1 : 2;

	printf("Valid environment: %d\n", (int)gd->env_valid);

 done:
	if (saved_buffer)
		free(saved_buffer);

	return ret;
}

void env_relocate_spec(void)
{
	int ret;
	int crc1_ok = 0, crc2_ok = 0;
	env_t *tmp_env1 = NULL;
	env_t *tmp_env2 = NULL;
	env_t *ep = NULL;

	tmp_env1 = (env_t *)memalign(ARCH_DMA_MINALIGN,
			CONFIG_ENV_SIZE);
	tmp_env2 = (env_t *)memalign(ARCH_DMA_MINALIGN,
			CONFIG_ENV_SIZE);
	if (!tmp_env1 || !tmp_env2) {
		set_default_env("!malloc() failed");
		goto out;
	}

	env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
			CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
	if (!env_flash) {
		set_default_env("!spi_flash_probe() failed");
		goto out;
	}

	ret = spi_flash_read(env_flash, CONFIG_ENV_OFFSET,
				CONFIG_ENV_SIZE, tmp_env1);
	if (ret) {
		set_default_env("!spi_flash_read() failed");
		goto err_read;
	}

	if (crc32(0, tmp_env1->data, ENV_SIZE) == tmp_env1->crc)
		crc1_ok = 1;

	ret = spi_flash_read(env_flash, CONFIG_ENV_OFFSET_REDUND,
				CONFIG_ENV_SIZE, tmp_env2);
	if (!ret) {
		if (crc32(0, tmp_env2->data, ENV_SIZE) == tmp_env2->crc)
			crc2_ok = 1;
	}

	if (!crc1_ok && !crc2_ok) {
		set_default_env("!bad CRC");
		goto err_read;
	} else if (crc1_ok && !crc2_ok) {
		gd->env_valid = 1;
	} else if (!crc1_ok && crc2_ok) {
		gd->env_valid = 2;
	} else if (tmp_env1->flags == ACTIVE_FLAG &&
		   tmp_env2->flags == OBSOLETE_FLAG) {
		gd->env_valid = 1;
	} else if (tmp_env1->flags == OBSOLETE_FLAG &&
		   tmp_env2->flags == ACTIVE_FLAG) {
		gd->env_valid = 2;
	} else if (tmp_env1->flags == tmp_env2->flags) {
		gd->env_valid = 1;
	} else if (tmp_env1->flags == 0xFF) {
		gd->env_valid = 1;
	} else if (tmp_env2->flags == 0xFF) {
		gd->env_valid = 2;
	} else {
		/*
		 * this differs from code in env_flash.c, but I think a sane
		 * default path is desirable.
		 */
		gd->env_valid = 1;
	}

	if (gd->env_valid == 1)
		ep = tmp_env1;
	else
		ep = tmp_env2;

	ret = env_import((char *)ep, 0);
	if (!ret) {
		error("Cannot import environment: errno = %d\n", errno);
		set_default_env("env_import failed");
	}

err_read:
	spi_flash_free(env_flash);
	env_flash = NULL;
out:
	free(tmp_env1);
	free(tmp_env2);
}
#else
int saveenv(void)
{
	u32	saved_size, saved_offset, sector = 1;
	char	*saved_buffer = NULL;
	int	ret = 1;
	env_t	env_new;

	if (!env_flash) {
		env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS,
			CONFIG_ENV_SPI_CS,
			CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
		if (!env_flash) {
			set_default_env("!spi_flash_probe() failed");
			return 1;
		}
	}

	/* Is the sector larger than the env (i.e. embedded) */
	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
		saved_size = CONFIG_ENV_SECT_SIZE - CONFIG_ENV_SIZE;
		saved_offset = CONFIG_ENV_OFFSET + CONFIG_ENV_SIZE;
		saved_buffer = malloc(saved_size);
		if (!saved_buffer)
			goto done;

		ret = spi_flash_read(env_flash, saved_offset,
			saved_size, saved_buffer);
		if (ret)
			goto done;
	}

	if (CONFIG_ENV_SIZE > CONFIG_ENV_SECT_SIZE) {
		sector = CONFIG_ENV_SIZE / CONFIG_ENV_SECT_SIZE;
		if (CONFIG_ENV_SIZE % CONFIG_ENV_SECT_SIZE)
			sector++;
	}

	ret = env_export(&env_new);
	if (ret)
		goto done;

	puts("Erasing SPI flash...");
	ret = spi_flash_erase(env_flash, CONFIG_ENV_OFFSET,
		sector * CONFIG_ENV_SECT_SIZE);
	if (ret)
		goto done;

	puts("Writing to SPI flash...");
	ret = spi_flash_write(env_flash, CONFIG_ENV_OFFSET,
		CONFIG_ENV_SIZE, &env_new);
	if (ret)
		goto done;

	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
		ret = spi_flash_write(env_flash, saved_offset,
			saved_size, saved_buffer);
		if (ret)
			goto done;
	}

	ret = 0;
	puts("done\n");

 done:
	if (saved_buffer)
		free(saved_buffer);

	return ret;
}

void env_relocate_spec(void)
{
	int ret;
	char *buf = NULL;

	buf = (char *)memalign(ARCH_DMA_MINALIGN, CONFIG_ENV_SIZE);
	env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
			CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
	if (!env_flash) {
		set_default_env("!spi_flash_probe() failed");
		if (buf)
			free(buf);
		return;
	}

	ret = spi_flash_read(env_flash,
		CONFIG_ENV_OFFSET, CONFIG_ENV_SIZE, buf);
	if (ret) {
		set_default_env("!spi_flash_read() failed");
		goto out;
	}

	ret = env_import(buf, 1);
	if (ret)
		gd->env_valid = 1;
out:
	spi_flash_free(env_flash);
	if (buf)
		free(buf);
	env_flash = NULL;
}
#endif

int env_init(void)
{
	/* SPI flash isn't usable before relocation */
	gd->env_addr = (ulong)&default_environment[0];
	gd->env_valid = 1;

	return 0;
}
Commit	Line	Data
8c66497e	1	/*
ea882baf	2	* (C) Copyright 2000-2010
8c66497e HS	3	* Wolfgang Denk, DENX Software Engineering, [email protected].
	4	*
	5	* (C) Copyright 2001 Sysgo Real-Time Solutions, GmbH <www.elinos.com>
	6	* Andreas Heppel <[email protected]>
	7	*
	8	* (C) Copyright 2008 Atmel Corporation
	9	*
402ba1e3	10	* SPDX-License-Identifier: GPL-2.0+
8c66497e HS	11	*/
8c66497e HS	12	#include <common.h>
8c66497e	13	#include <environment.h>
5b3375ac	14	#include <malloc.h>
843c9e87	15	#include <spi.h>
8c66497e	16	#include <spi_flash.h>
ea882baf WD	17	#include <search.h>
ea882baf WD	18	#include <errno.h>
8c66497e	19
0e8d1586 JCPV	20	#ifndef CONFIG_ENV_SPI_BUS
0e8d1586 JCPV	21	# define CONFIG_ENV_SPI_BUS 0
8c66497e	22	#endif
0e8d1586	23	#ifndef CONFIG_ENV_SPI_CS
eb58a7fc	24	# define CONFIG_ENV_SPI_CS 0
8c66497e	25	#endif
0e8d1586 JCPV	26	#ifndef CONFIG_ENV_SPI_MAX_HZ
0e8d1586 JCPV	27	# define CONFIG_ENV_SPI_MAX_HZ 1000000
8c66497e	28	#endif
0e8d1586 JCPV	29	#ifndef CONFIG_ENV_SPI_MODE
0e8d1586 JCPV	30	# define CONFIG_ENV_SPI_MODE SPI_MODE_3
8c66497e HS	31	#endif
8c66497e HS	32
7319bcaf	33	#ifdef CONFIG_ENV_OFFSET_REDUND
eb58a7fc IG	34	static ulong env_offset = CONFIG_ENV_OFFSET;
eb58a7fc IG	35	static ulong env_new_offset = CONFIG_ENV_OFFSET_REDUND;
7319bcaf	36
eb58a7fc IG	37	#define ACTIVE_FLAG 1
eb58a7fc IG	38	#define OBSOLETE_FLAG 0
a3110f01	39	#endif /* CONFIG_ENV_OFFSET_REDUND */
7319bcaf	40
8c66497e HS	41	DECLARE_GLOBAL_DATA_PTR;
8c66497e HS	42
eb58a7fc	43	char *env_name_spec = "SPI Flash";
8c66497e HS	44
	45	static struct spi_flash *env_flash;
	46
7319bcaf	47	#if defined(CONFIG_ENV_OFFSET_REDUND)
7319bcaf WW	48	int saveenv(void)
7319bcaf WW	49	{
cd0f4fa1	50	env_t env_new;
7ce1526e	51	char *saved_buffer = NULL, flag = OBSOLETE_FLAG;
eb58a7fc	52	u32 saved_size, saved_offset, sector = 1;
ea882baf	53	int ret;
7319bcaf WW	54
7319bcaf WW	55	if (!env_flash) {
a3110f01 SB	56	env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS,
	57	CONFIG_ENV_SPI_CS,
	58	CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
	59	if (!env_flash) {
	60	set_default_env("!spi_flash_probe() failed");
	61	return 1;
	62	}
7319bcaf WW	63	}
7319bcaf WW	64
7ce1526e MV	65	ret = env_export(&env_new);
	66	if (ret)
	67	return ret;
cd0f4fa1	68	env_new.flags = ACTIVE_FLAG;
ea882baf	69
a3110f01 SB	70	if (gd->env_valid == 1) {
	71	env_new_offset = CONFIG_ENV_OFFSET_REDUND;
	72	env_offset = CONFIG_ENV_OFFSET;
	73	} else {
	74	env_new_offset = CONFIG_ENV_OFFSET;
	75	env_offset = CONFIG_ENV_OFFSET_REDUND;
	76	}
	77
7319bcaf WW	78	/* Is the sector larger than the env (i.e. embedded) */
	79	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
	80	saved_size = CONFIG_ENV_SECT_SIZE - CONFIG_ENV_SIZE;
	81	saved_offset = env_new_offset + CONFIG_ENV_SIZE;
7dd01744	82	saved_buffer = memalign(ARCH_DMA_MINALIGN, saved_size);
7319bcaf WW	83	if (!saved_buffer) {
	84	ret = 1;
	85	goto done;
	86	}
	87	ret = spi_flash_read(env_flash, saved_offset,
	88	saved_size, saved_buffer);
	89	if (ret)
	90	goto done;
	91	}
	92
	93	if (CONFIG_ENV_SIZE > CONFIG_ENV_SECT_SIZE) {
	94	sector = CONFIG_ENV_SIZE / CONFIG_ENV_SECT_SIZE;
	95	if (CONFIG_ENV_SIZE % CONFIG_ENV_SECT_SIZE)
	96	sector++;
	97	}
	98
	99	puts("Erasing SPI flash...");
	100	ret = spi_flash_erase(env_flash, env_new_offset,
	101	sector * CONFIG_ENV_SECT_SIZE);
	102	if (ret)
	103	goto done;
	104
	105	puts("Writing to SPI flash...");
7319bcaf	106
a3110f01	107	ret = spi_flash_write(env_flash, env_new_offset,
cd0f4fa1	108	CONFIG_ENV_SIZE, &env_new);
7319bcaf WW	109	if (ret)
	110	goto done;
	111
	112	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
	113	ret = spi_flash_write(env_flash, saved_offset,
	114	saved_size, saved_buffer);
	115	if (ret)
	116	goto done;
	117	}
	118
eb58a7fc	119	ret = spi_flash_write(env_flash, env_offset + offsetof(env_t, flags),
cd0f4fa1	120	sizeof(env_new.flags), &flag);
a3110f01 SB	121	if (ret)
a3110f01 SB	122	goto done;
7319bcaf	123
7319bcaf WW	124	puts("done\n");
7319bcaf WW	125
eb58a7fc	126	gd->env_valid = gd->env_valid == 2 ? 1 : 2;
a3110f01	127
2dc55d9e	128	printf("Valid environment: %d\n", (int)gd->env_valid);
a3110f01	129
7319bcaf WW	130	done:
	131	if (saved_buffer)
	132	free(saved_buffer);
eb58a7fc	133
7319bcaf WW	134	return ret;
	135	}
	136
	137	void env_relocate_spec(void)
	138	{
	139	int ret;
	140	int crc1_ok = 0, crc2_ok = 0;
	141	env_t *tmp_env1 = NULL;
	142	env_t *tmp_env2 = NULL;
a3110f01	143	env_t *ep = NULL;
7319bcaf	144
7dd01744 RB	145	tmp_env1 = (env_t *)memalign(ARCH_DMA_MINALIGN,
	146	CONFIG_ENV_SIZE);
	147	tmp_env2 = (env_t *)memalign(ARCH_DMA_MINALIGN,
	148	CONFIG_ENV_SIZE);
ea882baf	149	if (!tmp_env1 \|\| !tmp_env2) {
ea882baf	150	set_default_env("!malloc() failed");
2dc55d9e	151	goto out;
7319bcaf WW	152	}
	153
	154	env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
	155	CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
ea882baf WD	156	if (!env_flash) {
ea882baf WD	157	set_default_env("!spi_flash_probe() failed");
2dc55d9e	158	goto out;
ea882baf	159	}
7319bcaf WW	160
	161	ret = spi_flash_read(env_flash, CONFIG_ENV_OFFSET,
	162	CONFIG_ENV_SIZE, tmp_env1);
ea882baf WD	163	if (ret) {
ea882baf WD	164	set_default_env("!spi_flash_read() failed");
7319bcaf	165	goto err_read;
ea882baf	166	}
7319bcaf WW	167
	168	if (crc32(0, tmp_env1->data, ENV_SIZE) == tmp_env1->crc)
	169	crc1_ok = 1;
7319bcaf WW	170
	171	ret = spi_flash_read(env_flash, CONFIG_ENV_OFFSET_REDUND,
	172	CONFIG_ENV_SIZE, tmp_env2);
	173	if (!ret) {
	174	if (crc32(0, tmp_env2->data, ENV_SIZE) == tmp_env2->crc)
	175	crc2_ok = 1;
7319bcaf WW	176	}
7319bcaf WW	177
a3110f01	178	if (!crc1_ok && !crc2_ok) {
a3110f01	179	set_default_env("!bad CRC");
2dc55d9e	180	goto err_read;
a3110f01	181	} else if (crc1_ok && !crc2_ok) {
7319bcaf	182	gd->env_valid = 1;
7319bcaf	183	} else if (!crc1_ok && crc2_ok) {
2dc55d9e	184	gd->env_valid = 2;
	185	} else if (tmp_env1->flags == ACTIVE_FLAG &&
	186	tmp_env2->flags == OBSOLETE_FLAG) {
7319bcaf	187	gd->env_valid = 1;
2dc55d9e	188	} else if (tmp_env1->flags == OBSOLETE_FLAG &&
2dc55d9e	189	tmp_env2->flags == ACTIVE_FLAG) {
a3110f01	190	gd->env_valid = 2;
2dc55d9e	191	} else if (tmp_env1->flags == tmp_env2->flags) {
5e8397dd	192	gd->env_valid = 1;
2dc55d9e	193	} else if (tmp_env1->flags == 0xFF) {
5e8397dd MS	194	gd->env_valid = 1;
5e8397dd MS	195	} else if (tmp_env2->flags == 0xFF) {
7319bcaf	196	gd->env_valid = 2;
7319bcaf WW	197	} else {
	198	/*
	199	* this differs from code in env_flash.c, but I think a sane
	200	* default path is desirable.
	201	*/
5e8397dd	202	gd->env_valid = 1;
7319bcaf	203	}
ea882baf	204
a3110f01 SB	205	if (gd->env_valid == 1)
	206	ep = tmp_env1;
	207	else
	208	ep = tmp_env2;
	209
	210	ret = env_import((char *)ep, 0);
	211	if (!ret) {
	212	error("Cannot import environment: errno = %d\n", errno);
	213	set_default_env("env_import failed");
7319bcaf	214	}
7319bcaf WW	215
	216	err_read:
	217	spi_flash_free(env_flash);
	218	env_flash = NULL;
7319bcaf	219	out:
ea882baf WD	220	free(tmp_env1);
ea882baf WD	221	free(tmp_env2);
7319bcaf WW	222	}
7319bcaf WW	223	#else
8c66497e HS	224	int saveenv(void)
8c66497e HS	225	{
eb58a7fc	226	u32 saved_size, saved_offset, sector = 1;
7ce1526e	227	char *saved_buffer = NULL;
eb58a7fc	228	int ret = 1;
cd0f4fa1	229	env_t env_new;
07efc9e3	230
8c66497e	231	if (!env_flash) {
a3110f01 SB	232	env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS,
	233	CONFIG_ENV_SPI_CS,
	234	CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
	235	if (!env_flash) {
	236	set_default_env("!spi_flash_probe() failed");
	237	return 1;
	238	}
8c66497e HS	239	}
8c66497e HS	240
5b3375ac MF	241	/* Is the sector larger than the env (i.e. embedded) */
	242	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
	243	saved_size = CONFIG_ENV_SECT_SIZE - CONFIG_ENV_SIZE;
	244	saved_offset = CONFIG_ENV_OFFSET + CONFIG_ENV_SIZE;
	245	saved_buffer = malloc(saved_size);
eb58a7fc	246	if (!saved_buffer)
5b3375ac	247	goto done;
eb58a7fc	248
a3110f01 SB	249	ret = spi_flash_read(env_flash, saved_offset,
a3110f01 SB	250	saved_size, saved_buffer);
5b3375ac MF	251	if (ret)
	252	goto done;
	253	}
	254
0e8d1586 JCPV	255	if (CONFIG_ENV_SIZE > CONFIG_ENV_SECT_SIZE) {
	256	sector = CONFIG_ENV_SIZE / CONFIG_ENV_SECT_SIZE;
	257	if (CONFIG_ENV_SIZE % CONFIG_ENV_SECT_SIZE)
07efc9e3 TL	258	sector++;
	259	}
	260
7ce1526e MV	261	ret = env_export(&env_new);
7ce1526e MV	262	if (ret)
a3110f01	263	goto done;
a3110f01	264
8c66497e	265	puts("Erasing SPI flash...");
a3110f01 SB	266	ret = spi_flash_erase(env_flash, CONFIG_ENV_OFFSET,
a3110f01 SB	267	sector * CONFIG_ENV_SECT_SIZE);
5b3375ac MF	268	if (ret)
5b3375ac MF	269	goto done;
8c66497e HS	270
8c66497e HS	271	puts("Writing to SPI flash...");
a3110f01	272	ret = spi_flash_write(env_flash, CONFIG_ENV_OFFSET,
cd0f4fa1	273	CONFIG_ENV_SIZE, &env_new);
5b3375ac MF	274	if (ret)
5b3375ac MF	275	goto done;
8c66497e	276
5b3375ac	277	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
a3110f01 SB	278	ret = spi_flash_write(env_flash, saved_offset,
a3110f01 SB	279	saved_size, saved_buffer);
5b3375ac MF	280	if (ret)
	281	goto done;
	282	}
	283
	284	ret = 0;
8c66497e	285	puts("done\n");
5b3375ac MF	286
	287	done:
	288	if (saved_buffer)
	289	free(saved_buffer);
eb58a7fc	290
5b3375ac	291	return ret;
8c66497e HS	292	}
	293
	294	void env_relocate_spec(void)
	295	{
	296	int ret;
5a89fa92	297	char *buf = NULL;
8c66497e	298
7dd01744	299	buf = (char *)memalign(ARCH_DMA_MINALIGN, CONFIG_ENV_SIZE);
0e8d1586 JCPV	300	env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
0e8d1586 JCPV	301	CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
ea882baf WD	302	if (!env_flash) {
ea882baf WD	303	set_default_env("!spi_flash_probe() failed");
5a89fa92 YZ	304	if (buf)
5a89fa92 YZ	305	free(buf);
ea882baf WD	306	return;
ea882baf WD	307	}
8c66497e	308
ea882baf WD	309	ret = spi_flash_read(env_flash,
	310	CONFIG_ENV_OFFSET, CONFIG_ENV_SIZE, buf);
	311	if (ret) {
	312	set_default_env("!spi_flash_read() failed");
	313	goto out;
	314	}
8c66497e	315
ea882baf	316	ret = env_import(buf, 1);
ea882baf WD	317	if (ret)
	318	gd->env_valid = 1;
	319	out:
8c66497e	320	spi_flash_free(env_flash);
5a89fa92 YZ	321	if (buf)
5a89fa92 YZ	322	free(buf);
8c66497e	323	env_flash = NULL;
8c66497e	324	}
7319bcaf	325	#endif
8c66497e HS	326
	327	int env_init(void)
	328	{
	329	/* SPI flash isn't usable before relocation */
	330	gd->env_addr = (ulong)&default_environment[0];
	331	gd->env_valid = 1;
	332
	333	return 0;
	334	}