[people/ms/u-boot.git] / tools / updater / flash.c

/*
 * (C) Copyright 2000-2006
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <flash.h>

extern flash_info_t  flash_info[]; /* info for FLASH chips */

/*-----------------------------------------------------------------------
 * Functions
 */

/*-----------------------------------------------------------------------
 * Set protection status for monitor sectors
 *
 * The monitor is always located in the _first_ Flash bank.
 * If necessary you have to map the second bank at lower addresses.
 */
void
flash_protect (int flag, ulong from, ulong to, flash_info_t *info)
{
	ulong b_end = info->start[0] + info->size - 1;	/* bank end address */
	short s_end = info->sector_count - 1;	/* index of last sector */
	int i;

	/* Do nothing if input data is bad. */
	if (info->sector_count == 0 || info->size == 0 || to < from) {
		return;
	}

	/* There is nothing to do if we have no data about the flash
	 * or the protect range and flash range don't overlap.
	 */
	if (info->flash_id == FLASH_UNKNOWN ||
	    to < info->start[0] || from > b_end) {
		return;
	}

	for (i=0; i<info->sector_count; ++i) {
		ulong end;		/* last address in current sect	*/

		end = (i == s_end) ? b_end : info->start[i + 1] - 1;

		/* Update protection if any part of the sector
		 * is in the specified range.
		 */
		if (from <= end && to >= info->start[i]) {
			if (flag & FLAG_PROTECT_CLEAR) {
#if defined(CONFIG_SYS_FLASH_PROTECTION)
				flash_real_protect(info, i, 0);
#else
				info->protect[i] = 0;
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
			}
			else if (flag & FLAG_PROTECT_SET) {
#if defined(CONFIG_SYS_FLASH_PROTECTION)
				flash_real_protect(info, i, 1);
#else
				info->protect[i] = 1;
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
			}
		}
	}
}

/*-----------------------------------------------------------------------
 */

flash_info_t *
addr2info (ulong addr)
{
#ifndef CONFIG_SPD823TS
	flash_info_t *info;
	int i;

	for (i=0, info = &flash_info[0]; i<CONFIG_SYS_MAX_FLASH_BANKS; ++i, ++info) {
		if (info->flash_id != FLASH_UNKNOWN &&
		    addr >= info->start[0] &&
		    /* WARNING - The '- 1' is needed if the flash
		     * is at the end of the address space, since
		     * info->start[0] + info->size wraps back to 0.
		     * Please don't change this unless you understand this.
		     */
		    addr <= info->start[0] + info->size - 1) {
			return (info);
		}
	}
#endif /* CONFIG_SPD823TS */

	return (NULL);
}

/*-----------------------------------------------------------------------
 * Copy memory to flash.
 * Make sure all target addresses are within Flash bounds,
 * and no protected sectors are hit.
 * Returns:
 * ERR_OK          0 - OK
 * ERR_TIMOUT      1 - write timeout
 * ERR_NOT_ERASED  2 - Flash not erased
 * ERR_PROTECTED   4 - target range includes protected sectors
 * ERR_INVAL       8 - target address not in Flash memory
 * ERR_ALIGN       16 - target address not aligned on boundary
 *			(only some targets require alignment)
 */
int
flash_write (char *src, ulong addr, ulong cnt)
{
#ifdef CONFIG_SPD823TS
	return (ERR_TIMOUT);	/* any other error codes are possible as well */
#else
	int i;
	ulong         end        = addr + cnt - 1;
	flash_info_t *info_first = addr2info (addr);
	flash_info_t *info_last  = addr2info (end );
	flash_info_t *info;
	int j;

	if (cnt == 0) {
		return (ERR_OK);
	}

	if (!info_first || !info_last) {
		return (ERR_INVAL);
	}

	for (info = info_first; info <= info_last; ++info) {
		ulong b_end = info->start[0] + info->size;	/* bank end addr */
		short s_end = info->sector_count - 1;
		for (i=0; i<info->sector_count; ++i) {
			ulong e_addr = (i == s_end) ? b_end : info->start[i + 1];

			if ((end >= info->start[i]) && (addr < e_addr) &&
			    (info->protect[i] != 0) ) {
				return (ERR_PROTECTED);
			}
		}
	}

	printf("\rWriting ");
	for (j=0; j<20; j++) putc(177);
	printf("\rWriting ");

	/* finally write data to flash */
	for (info = info_first; info <= info_last && cnt>0; ++info) {
		ulong len;

		len = info->start[0] + info->size - addr;
		if (len > cnt)
			len = cnt;

		if ((i = write_buff(info, src, addr, len)) != 0) {
			return (i);
		}
		cnt  -= len;
		addr += len;
		src  += len;
	}
	return (ERR_OK);
#endif /* CONFIG_SPD823TS */
}

/*-----------------------------------------------------------------------
 */
Commit	Line	Data
c7de829c	1	/*
6608043f	2	* (C) Copyright 2000-2006
c7de829c WD	3	* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
c7de829c WD	4	*
1a459660	5	* SPDX-License-Identifier: GPL-2.0+
c7de829c WD	6	*/
	7
	8	#include <common.h>
	9	#include <flash.h>
	10
e6f2e902	11	extern flash_info_t flash_info[]; /* info for FLASH chips */
c7de829c WD	12
	13	/*-----------------------------------------------------------------------
	14	* Functions
	15	*/
	16
	17	/*-----------------------------------------------------------------------
	18	* Set protection status for monitor sectors
	19	*
	20	* The monitor is always located in the _first_ Flash bank.
	21	* If necessary you have to map the second bank at lower addresses.
	22	*/
	23	void
	24	flash_protect (int flag, ulong from, ulong to, flash_info_t *info)
	25	{
	26	ulong b_end = info->start[0] + info->size - 1; /* bank end address */
	27	short s_end = info->sector_count - 1; /* index of last sector */
	28	int i;
	29
	30	/* Do nothing if input data is bad. */
	31	if (info->sector_count == 0 \|\| info->size == 0 \|\| to < from) {
	32	return;
	33	}
	34
	35	/* There is nothing to do if we have no data about the flash
	36	* or the protect range and flash range don't overlap.
	37	*/
	38	if (info->flash_id == FLASH_UNKNOWN \|\|
	39	to < info->start[0] \|\| from > b_end) {
	40	return;
	41	}
	42
	43	for (i=0; i<info->sector_count; ++i) {
	44	ulong end; /* last address in current sect */
	45
	46	end = (i == s_end) ? b_end : info->start[i + 1] - 1;
	47
	48	/* Update protection if any part of the sector
	49	* is in the specified range.
	50	*/
	51	if (from <= end && to >= info->start[i]) {
	52	if (flag & FLAG_PROTECT_CLEAR) {
6d0f6bcf	53	#if defined(CONFIG_SYS_FLASH_PROTECTION)
c7de829c WD	54	flash_real_protect(info, i, 0);
	55	#else
	56	info->protect[i] = 0;
6d0f6bcf	57	#endif /* CONFIG_SYS_FLASH_PROTECTION */
c7de829c WD	58	}
c7de829c WD	59	else if (flag & FLAG_PROTECT_SET) {
6d0f6bcf	60	#if defined(CONFIG_SYS_FLASH_PROTECTION)
c7de829c WD	61	flash_real_protect(info, i, 1);
	62	#else
	63	info->protect[i] = 1;
6d0f6bcf	64	#endif /* CONFIG_SYS_FLASH_PROTECTION */
c7de829c WD	65	}
	66	}
	67	}
	68	}
	69
	70	/*-----------------------------------------------------------------------
	71	*/
	72
	73	flash_info_t *
	74	addr2info (ulong addr)
	75	{
	76	#ifndef CONFIG_SPD823TS
	77	flash_info_t *info;
	78	int i;
	79
6d0f6bcf	80	for (i=0, info = &flash_info[0]; i<CONFIG_SYS_MAX_FLASH_BANKS; ++i, ++info) {
c7de829c WD	81	if (info->flash_id != FLASH_UNKNOWN &&
	82	addr >= info->start[0] &&
	83	/* WARNING - The '- 1' is needed if the flash
	84	* is at the end of the address space, since
	85	* info->start[0] + info->size wraps back to 0.
	86	* Please don't change this unless you understand this.
	87	*/
	88	addr <= info->start[0] + info->size - 1) {
	89	return (info);
	90	}
	91	}
	92	#endif /* CONFIG_SPD823TS */
	93
	94	return (NULL);
	95	}
	96
	97	/*-----------------------------------------------------------------------
	98	* Copy memory to flash.
	99	* Make sure all target addresses are within Flash bounds,
	100	* and no protected sectors are hit.
	101	* Returns:
	102	* ERR_OK 0 - OK
	103	* ERR_TIMOUT 1 - write timeout
	104	* ERR_NOT_ERASED 2 - Flash not erased
	105	* ERR_PROTECTED 4 - target range includes protected sectors
	106	* ERR_INVAL 8 - target address not in Flash memory
	107	* ERR_ALIGN 16 - target address not aligned on boundary
	108	* (only some targets require alignment)
	109	*/
	110	int
6608043f	111	flash_write (char *src, ulong addr, ulong cnt)
c7de829c WD	112	{
	113	#ifdef CONFIG_SPD823TS
	114	return (ERR_TIMOUT); /* any other error codes are possible as well */
	115	#else
	116	int i;
	117	ulong end = addr + cnt - 1;
	118	flash_info_t *info_first = addr2info (addr);
	119	flash_info_t *info_last = addr2info (end );
	120	flash_info_t *info;
	121	int j;
	122
	123	if (cnt == 0) {
	124	return (ERR_OK);
	125	}
	126
	127	if (!info_first \|\| !info_last) {
	128	return (ERR_INVAL);
	129	}
	130
	131	for (info = info_first; info <= info_last; ++info) {
	132	ulong b_end = info->start[0] + info->size; /* bank end addr */
	133	short s_end = info->sector_count - 1;
	134	for (i=0; i<info->sector_count; ++i) {
	135	ulong e_addr = (i == s_end) ? b_end : info->start[i + 1];
	136
	137	if ((end >= info->start[i]) && (addr < e_addr) &&
	138	(info->protect[i] != 0) ) {
	139	return (ERR_PROTECTED);
	140	}
	141	}
	142	}
	143
27b207fd WD	144	printf("\rWriting ");
	145	for (j=0; j<20; j++) putc(177);
	146	printf("\rWriting ");
c7de829c WD	147
	148	/* finally write data to flash */
	149	for (info = info_first; info <= info_last && cnt>0; ++info) {
	150	ulong len;
	151
	152	len = info->start[0] + info->size - addr;
	153	if (len > cnt)
	154	len = cnt;
	155
	156	if ((i = write_buff(info, src, addr, len)) != 0) {
	157	return (i);
	158	}
	159	cnt -= len;
	160	addr += len;
	161	src += len;
	162	}
	163	return (ERR_OK);
	164	#endif /* CONFIG_SPD823TS */
	165	}
	166
	167	/*-----------------------------------------------------------------------
	168	*/