[J-u-boot.git] / 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 <dm.h>
#include <environment.h>
#include <malloc.h>
#include <spi.h>
#include <spi_flash.h>
#include <search.h>
#include <errno.h>
#include <dm/device-internal.h>

#ifndef CONFIG_ENV_SPI_BUS
# define CONFIG_ENV_SPI_BUS	CONFIG_SF_DEFAULT_BUS
#endif
#ifndef CONFIG_ENV_SPI_CS
# define CONFIG_ENV_SPI_CS	CONFIG_SF_DEFAULT_CS
#endif
#ifndef CONFIG_ENV_SPI_MAX_HZ
# define CONFIG_ENV_SPI_MAX_HZ	CONFIG_SF_DEFAULT_SPEED
#endif
#ifndef CONFIG_ENV_SPI_MODE
# define CONFIG_ENV_SPI_MODE	CONFIG_SF_DEFAULT_MODE
#endif

#ifndef CONFIG_SPL_BUILD
#define CMD_SAVEENV
#endif

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

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

DECLARE_GLOBAL_DATA_PTR;

static struct spi_flash *env_flash;

static int setup_flash_device(void)
{
#ifdef CONFIG_DM_SPI_FLASH
	struct udevice *new;
	int	ret;

	/* speed and mode will be read from DT */
	ret = spi_flash_probe_bus_cs(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
				     0, 0, &new);
	if (ret) {
		set_default_env("!spi_flash_probe_bus_cs() failed");
		return ret;
	}

	env_flash = dev_get_uclass_priv(new);
#else

	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 -EIO;
		}
	}
#endif
	return 0;
}

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

	ret = setup_flash_device();
	if (ret)
		return ret;

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

	if (gd->env_valid == ENV_VALID) {
		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 = -ENOMEM;
			goto done;
		}
		ret = spi_flash_read(env_flash, saved_offset,
					saved_size, saved_buffer);
		if (ret)
			goto done;
	}

	sector = DIV_ROUND_UP(CONFIG_ENV_SIZE, CONFIG_ENV_SECT_SIZE);

	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 == ENV_REDUND ? ENV_VALID : ENV_REDUND;

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

 done:
	if (saved_buffer)
		free(saved_buffer);

	return ret;
}
#endif /* CMD_SAVEENV */

static int env_sf_load(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");
		ret = -EIO;
		goto out;
	}

	ret = setup_flash_device();
	if (ret)
		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");
		ret = -EIO;
		goto err_read;
	} else if (crc1_ok && !crc2_ok) {
		gd->env_valid = ENV_VALID;
	} else if (!crc1_ok && crc2_ok) {
		gd->env_valid = ENV_REDUND;
	} else if (tmp_env1->flags == ACTIVE_FLAG &&
		   tmp_env2->flags == OBSOLETE_FLAG) {
		gd->env_valid = ENV_VALID;
	} else if (tmp_env1->flags == OBSOLETE_FLAG &&
		   tmp_env2->flags == ACTIVE_FLAG) {
		gd->env_valid = ENV_REDUND;
	} else if (tmp_env1->flags == tmp_env2->flags) {
		gd->env_valid = ENV_VALID;
	} else if (tmp_env1->flags == 0xFF) {
		gd->env_valid = ENV_VALID;
	} else if (tmp_env2->flags == 0xFF) {
		gd->env_valid = ENV_REDUND;
	} else {
		/*
		 * this differs from code in env_flash.c, but I think a sane
		 * default path is desirable.
		 */
		gd->env_valid = ENV_VALID;
	}

	if (gd->env_valid == ENV_VALID)
		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);

	return ret;
}
#else
#ifdef CMD_SAVEENV
static int env_sf_save(void)
{
	u32	saved_size, saved_offset, sector;
	char	*saved_buffer = NULL;
	int	ret = 1;
	env_t	env_new;

	ret = setup_flash_device();
	if (ret)
		return ret;

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

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

	sector = DIV_ROUND_UP(CONFIG_ENV_SIZE, CONFIG_ENV_SECT_SIZE);

	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;
}
#endif /* CMD_SAVEENV */

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

	buf = (char *)memalign(ARCH_DMA_MINALIGN, CONFIG_ENV_SIZE);
	if (!buf) {
		set_default_env("!malloc() failed");
		return -EIO;
	}

	ret = setup_flash_device();
	if (ret)
		goto out;

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

	ret = env_import(buf, 1);
	if (ret)
		gd->env_valid = ENV_VALID;

err_read:
	spi_flash_free(env_flash);
	env_flash = NULL;
out:
	free(buf);

	return ret;
}
#endif

U_BOOT_ENV_LOCATION(sf) = {
	.location	= ENVL_SPI_FLASH,
	ENV_NAME("SPI Flash")
	.load		= env_sf_load,
#ifdef CMD_SAVEENV
	.save		= env_save_ptr(env_sf_save),
#endif
};
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>
9d922450	13	#include <dm.h>
8c66497e	14	#include <environment.h>
5b3375ac	15	#include <malloc.h>
843c9e87	16	#include <spi.h>
8c66497e	17	#include <spi_flash.h>
ea882baf WD	18	#include <search.h>
ea882baf WD	19	#include <errno.h>
19c31285	20	#include <dm/device-internal.h>
8c66497e	21
0e8d1586	22	#ifndef CONFIG_ENV_SPI_BUS
2e4e5ad4	23	# define CONFIG_ENV_SPI_BUS CONFIG_SF_DEFAULT_BUS
8c66497e	24	#endif
0e8d1586	25	#ifndef CONFIG_ENV_SPI_CS
2e4e5ad4	26	# define CONFIG_ENV_SPI_CS CONFIG_SF_DEFAULT_CS
8c66497e	27	#endif
0e8d1586	28	#ifndef CONFIG_ENV_SPI_MAX_HZ
2e4e5ad4	29	# define CONFIG_ENV_SPI_MAX_HZ CONFIG_SF_DEFAULT_SPEED
8c66497e	30	#endif
0e8d1586	31	#ifndef CONFIG_ENV_SPI_MODE
2e4e5ad4	32	# define CONFIG_ENV_SPI_MODE CONFIG_SF_DEFAULT_MODE
8c66497e HS	33	#endif
8c66497e HS	34
4415f1d1 SG	35	#ifndef CONFIG_SPL_BUILD
	36	#define CMD_SAVEENV
	37	#endif
	38
7319bcaf	39	#ifdef CONFIG_ENV_OFFSET_REDUND
4415f1d1	40	#ifdef CMD_SAVEENV
eb58a7fc IG	41	static ulong env_offset = CONFIG_ENV_OFFSET;
eb58a7fc IG	42	static ulong env_new_offset = CONFIG_ENV_OFFSET_REDUND;
4415f1d1	43	#endif
7319bcaf	44
eb58a7fc IG	45	#define ACTIVE_FLAG 1
eb58a7fc IG	46	#define OBSOLETE_FLAG 0
a3110f01	47	#endif /* CONFIG_ENV_OFFSET_REDUND */
7319bcaf	48
8c66497e HS	49	DECLARE_GLOBAL_DATA_PTR;
8c66497e HS	50
8c66497e HS	51	static struct spi_flash *env_flash;
8c66497e HS	52
afa81a77	53	static int setup_flash_device(void)
7319bcaf	54	{
19c31285 GQ	55	#ifdef CONFIG_DM_SPI_FLASH
19c31285 GQ	56	struct udevice *new;
afa81a77	57	int ret;
19c31285	58
96907c0f	59	/* speed and mode will be read from DT */
19c31285	60	ret = spi_flash_probe_bus_cs(CONFIG_ENV_SPI_BUS, CONFIG_ENV_SPI_CS,
96907c0f	61	0, 0, &new);
19c31285 GQ	62	if (ret) {
19c31285 GQ	63	set_default_env("!spi_flash_probe_bus_cs() failed");
c5951991	64	return ret;
19c31285 GQ	65	}
	66
	67	env_flash = dev_get_uclass_priv(new);
	68	#else
7319bcaf WW	69
7319bcaf WW	70	if (!env_flash) {
a3110f01 SB	71	env_flash = spi_flash_probe(CONFIG_ENV_SPI_BUS,
	72	CONFIG_ENV_SPI_CS,
	73	CONFIG_ENV_SPI_MAX_HZ, CONFIG_ENV_SPI_MODE);
	74	if (!env_flash) {
	75	set_default_env("!spi_flash_probe() failed");
c5951991	76	return -EIO;
a3110f01	77	}
7319bcaf	78	}
19c31285	79	#endif
afa81a77 AF	80	return 0;
	81	}
	82
	83	#if defined(CONFIG_ENV_OFFSET_REDUND)
4415f1d1	84	#ifdef CMD_SAVEENV
e5bce247	85	static int env_sf_save(void)
afa81a77 AF	86	{
	87	env_t env_new;
	88	char *saved_buffer = NULL, flag = OBSOLETE_FLAG;
0b2e5bbe	89	u32 saved_size, saved_offset, sector;
afa81a77 AF	90	int ret;
	91
	92	ret = setup_flash_device();
	93	if (ret)
	94	return ret;
7319bcaf	95
7ce1526e MV	96	ret = env_export(&env_new);
7ce1526e MV	97	if (ret)
c5951991	98	return -EIO;
cd0f4fa1	99	env_new.flags = ACTIVE_FLAG;
ea882baf	100
203e94f6	101	if (gd->env_valid == ENV_VALID) {
a3110f01 SB	102	env_new_offset = CONFIG_ENV_OFFSET_REDUND;
	103	env_offset = CONFIG_ENV_OFFSET;
	104	} else {
	105	env_new_offset = CONFIG_ENV_OFFSET;
	106	env_offset = CONFIG_ENV_OFFSET_REDUND;
	107	}
	108
7319bcaf WW	109	/* Is the sector larger than the env (i.e. embedded) */
	110	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
	111	saved_size = CONFIG_ENV_SECT_SIZE - CONFIG_ENV_SIZE;
	112	saved_offset = env_new_offset + CONFIG_ENV_SIZE;
7dd01744	113	saved_buffer = memalign(ARCH_DMA_MINALIGN, saved_size);
7319bcaf	114	if (!saved_buffer) {
c5951991	115	ret = -ENOMEM;
7319bcaf WW	116	goto done;
	117	}
	118	ret = spi_flash_read(env_flash, saved_offset,
	119	saved_size, saved_buffer);
	120	if (ret)
	121	goto done;
	122	}
	123
0b2e5bbe	124	sector = DIV_ROUND_UP(CONFIG_ENV_SIZE, CONFIG_ENV_SECT_SIZE);
7319bcaf WW	125
	126	puts("Erasing SPI flash...");
	127	ret = spi_flash_erase(env_flash, env_new_offset,
	128	sector * CONFIG_ENV_SECT_SIZE);
	129	if (ret)
	130	goto done;
	131
	132	puts("Writing to SPI flash...");
7319bcaf	133
a3110f01	134	ret = spi_flash_write(env_flash, env_new_offset,
cd0f4fa1	135	CONFIG_ENV_SIZE, &env_new);
7319bcaf WW	136	if (ret)
	137	goto done;
	138
	139	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
	140	ret = spi_flash_write(env_flash, saved_offset,
	141	saved_size, saved_buffer);
	142	if (ret)
	143	goto done;
	144	}
	145
eb58a7fc	146	ret = spi_flash_write(env_flash, env_offset + offsetof(env_t, flags),
cd0f4fa1	147	sizeof(env_new.flags), &flag);
a3110f01 SB	148	if (ret)
a3110f01 SB	149	goto done;
7319bcaf	150
7319bcaf WW	151	puts("done\n");
7319bcaf WW	152
203e94f6	153	gd->env_valid = gd->env_valid == ENV_REDUND ? ENV_VALID : ENV_REDUND;
a3110f01	154
2dc55d9e	155	printf("Valid environment: %d\n", (int)gd->env_valid);
a3110f01	156
7319bcaf WW	157	done:
	158	if (saved_buffer)
	159	free(saved_buffer);
eb58a7fc	160
7319bcaf WW	161	return ret;
7319bcaf WW	162	}
4415f1d1	163	#endif /* CMD_SAVEENV */
7319bcaf	164
c5951991	165	static int env_sf_load(void)
7319bcaf WW	166	{
	167	int ret;
	168	int crc1_ok = 0, crc2_ok = 0;
	169	env_t *tmp_env1 = NULL;
	170	env_t *tmp_env2 = NULL;
a3110f01	171	env_t *ep = NULL;
7319bcaf	172
7dd01744 RB	173	tmp_env1 = (env_t *)memalign(ARCH_DMA_MINALIGN,
	174	CONFIG_ENV_SIZE);
	175	tmp_env2 = (env_t *)memalign(ARCH_DMA_MINALIGN,
	176	CONFIG_ENV_SIZE);
ea882baf	177	if (!tmp_env1 \|\| !tmp_env2) {
ea882baf	178	set_default_env("!malloc() failed");
c5951991	179	ret = -EIO;
2dc55d9e	180	goto out;
7319bcaf WW	181	}
7319bcaf WW	182
8fee8845 AF	183	ret = setup_flash_device();
8fee8845 AF	184	if (ret)
2dc55d9e	185	goto out;
7319bcaf WW	186
	187	ret = spi_flash_read(env_flash, CONFIG_ENV_OFFSET,
	188	CONFIG_ENV_SIZE, tmp_env1);
ea882baf WD	189	if (ret) {
ea882baf WD	190	set_default_env("!spi_flash_read() failed");
7319bcaf	191	goto err_read;
ea882baf	192	}
7319bcaf WW	193
	194	if (crc32(0, tmp_env1->data, ENV_SIZE) == tmp_env1->crc)
	195	crc1_ok = 1;
7319bcaf WW	196
	197	ret = spi_flash_read(env_flash, CONFIG_ENV_OFFSET_REDUND,
	198	CONFIG_ENV_SIZE, tmp_env2);
	199	if (!ret) {
	200	if (crc32(0, tmp_env2->data, ENV_SIZE) == tmp_env2->crc)
	201	crc2_ok = 1;
7319bcaf WW	202	}
7319bcaf WW	203
a3110f01	204	if (!crc1_ok && !crc2_ok) {
a3110f01	205	set_default_env("!bad CRC");
c5951991	206	ret = -EIO;
2dc55d9e	207	goto err_read;
a3110f01	208	} else if (crc1_ok && !crc2_ok) {
203e94f6	209	gd->env_valid = ENV_VALID;
7319bcaf	210	} else if (!crc1_ok && crc2_ok) {
203e94f6	211	gd->env_valid = ENV_REDUND;
2dc55d9e	212	} else if (tmp_env1->flags == ACTIVE_FLAG &&
2dc55d9e	213	tmp_env2->flags == OBSOLETE_FLAG) {
203e94f6	214	gd->env_valid = ENV_VALID;
2dc55d9e	215	} else if (tmp_env1->flags == OBSOLETE_FLAG &&
2dc55d9e	216	tmp_env2->flags == ACTIVE_FLAG) {
203e94f6	217	gd->env_valid = ENV_REDUND;
2dc55d9e	218	} else if (tmp_env1->flags == tmp_env2->flags) {
203e94f6	219	gd->env_valid = ENV_VALID;
2dc55d9e	220	} else if (tmp_env1->flags == 0xFF) {
203e94f6	221	gd->env_valid = ENV_VALID;
5e8397dd	222	} else if (tmp_env2->flags == 0xFF) {
203e94f6	223	gd->env_valid = ENV_REDUND;
7319bcaf WW	224	} else {
	225	/*
	226	* this differs from code in env_flash.c, but I think a sane
	227	* default path is desirable.
	228	*/
203e94f6	229	gd->env_valid = ENV_VALID;
7319bcaf	230	}
ea882baf	231
203e94f6	232	if (gd->env_valid == ENV_VALID)
a3110f01 SB	233	ep = tmp_env1;
	234	else
	235	ep = tmp_env2;
	236
	237	ret = env_import((char *)ep, 0);
	238	if (!ret) {
	239	error("Cannot import environment: errno = %d\n", errno);
9a5d993e	240	set_default_env("!env_import failed");
7319bcaf	241	}
7319bcaf WW	242
	243	err_read:
	244	spi_flash_free(env_flash);
	245	env_flash = NULL;
7319bcaf	246	out:
ea882baf WD	247	free(tmp_env1);
ea882baf WD	248	free(tmp_env2);
c5951991 SG	249
c5951991 SG	250	return ret;
7319bcaf WW	251	}
7319bcaf WW	252	#else
4415f1d1	253	#ifdef CMD_SAVEENV
e5bce247	254	static int env_sf_save(void)
8c66497e	255	{
0b2e5bbe	256	u32 saved_size, saved_offset, sector;
7ce1526e	257	char *saved_buffer = NULL;
eb58a7fc	258	int ret = 1;
cd0f4fa1	259	env_t env_new;
19c31285	260
afa81a77 AF	261	ret = setup_flash_device();
	262	if (ret)
	263	return ret;
8c66497e	264
5b3375ac MF	265	/* Is the sector larger than the env (i.e. embedded) */
	266	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
	267	saved_size = CONFIG_ENV_SECT_SIZE - CONFIG_ENV_SIZE;
	268	saved_offset = CONFIG_ENV_OFFSET + CONFIG_ENV_SIZE;
	269	saved_buffer = malloc(saved_size);
eb58a7fc	270	if (!saved_buffer)
5b3375ac	271	goto done;
eb58a7fc	272
a3110f01 SB	273	ret = spi_flash_read(env_flash, saved_offset,
a3110f01 SB	274	saved_size, saved_buffer);
5b3375ac MF	275	if (ret)
	276	goto done;
	277	}
	278
7ce1526e MV	279	ret = env_export(&env_new);
7ce1526e MV	280	if (ret)
a3110f01	281	goto done;
a3110f01	282
0b2e5bbe AF	283	sector = DIV_ROUND_UP(CONFIG_ENV_SIZE, CONFIG_ENV_SECT_SIZE);
0b2e5bbe AF	284
8c66497e	285	puts("Erasing SPI flash...");
a3110f01 SB	286	ret = spi_flash_erase(env_flash, CONFIG_ENV_OFFSET,
a3110f01 SB	287	sector * CONFIG_ENV_SECT_SIZE);
5b3375ac MF	288	if (ret)
5b3375ac MF	289	goto done;
8c66497e HS	290
8c66497e HS	291	puts("Writing to SPI flash...");
a3110f01	292	ret = spi_flash_write(env_flash, CONFIG_ENV_OFFSET,
cd0f4fa1	293	CONFIG_ENV_SIZE, &env_new);
5b3375ac MF	294	if (ret)
5b3375ac MF	295	goto done;
8c66497e	296
5b3375ac	297	if (CONFIG_ENV_SECT_SIZE > CONFIG_ENV_SIZE) {
a3110f01 SB	298	ret = spi_flash_write(env_flash, saved_offset,
a3110f01 SB	299	saved_size, saved_buffer);
5b3375ac MF	300	if (ret)
	301	goto done;
	302	}
	303
	304	ret = 0;
8c66497e	305	puts("done\n");
5b3375ac MF	306
	307	done:
	308	if (saved_buffer)
	309	free(saved_buffer);
eb58a7fc	310
5b3375ac	311	return ret;
8c66497e	312	}
4415f1d1	313	#endif /* CMD_SAVEENV */
8c66497e	314
c5951991	315	static int env_sf_load(void)
8c66497e HS	316	{
8c66497e HS	317	int ret;
5a89fa92	318	char *buf = NULL;
8c66497e	319
7dd01744	320	buf = (char *)memalign(ARCH_DMA_MINALIGN, CONFIG_ENV_SIZE);
c041c60c AF	321	if (!buf) {
c041c60c AF	322	set_default_env("!malloc() failed");
c5951991	323	return -EIO;
ea882baf	324	}
8c66497e	325
c041c60c AF	326	ret = setup_flash_device();
	327	if (ret)
	328	goto out;
	329
ea882baf WD	330	ret = spi_flash_read(env_flash,
	331	CONFIG_ENV_OFFSET, CONFIG_ENV_SIZE, buf);
	332	if (ret) {
	333	set_default_env("!spi_flash_read() failed");
c041c60c	334	goto err_read;
ea882baf	335	}
8c66497e	336
ea882baf	337	ret = env_import(buf, 1);
ea882baf	338	if (ret)
203e94f6	339	gd->env_valid = ENV_VALID;
c041c60c AF	340
c041c60c AF	341	err_read:
8c66497e HS	342	spi_flash_free(env_flash);
8c66497e HS	343	env_flash = NULL;
c041c60c AF	344	out:
c041c60c AF	345	free(buf);
c5951991 SG	346
c5951991 SG	347	return ret;
8c66497e	348	}
7319bcaf	349	#endif
8c66497e	350
4415f1d1 SG	351	U_BOOT_ENV_LOCATION(sf) = {
4415f1d1 SG	352	.location = ENVL_SPI_FLASH,
ac358beb	353	ENV_NAME("SPI Flash")
e5bce247	354	.load = env_sf_load,
4415f1d1	355	#ifdef CMD_SAVEENV
e5bce247	356	.save = env_save_ptr(env_sf_save),
4415f1d1	357	#endif
4415f1d1	358	};