[J-u-boot.git] / board / exbitgen / flash.c

/*
 * (C) Copyright 2003
 * Wolfgang Denk, DENX Software Engineering, [email protected].
 *
 * See file CREDITS for list of people who contributed to this
 * project.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
 * MA 02111-1307 USA
 */

/*
 * Modified 4/5/2001
 * Wait for completion of each sector erase command issued
 * 4/5/2001
 * Chris Hallinan - DS4.COM, Inc. - [email protected]
 */

#include <common.h>
#include <asm/u-boot.h>
#include <asm/processor.h>
#include <ppc4xx.h>

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

/*-----------------------------------------------------------------------
 * Functions
 */
static ulong flash_get_size (vu_long *addr, flash_info_t *info);
static int write_word (flash_info_t *info, ulong dest, ulong data);

#ifdef MEIGSBOARD_ONBOARD_FLASH /* onboard = 2MB */
#  ifdef CONFIG_EXBITGEN
#     define FLASH_WORD_SIZE unsigned long
#  endif
#else /* Meigsboard socket flash = 512KB */
#  ifdef CONFIG_EXBITGEN
#    define FLASH_WORD_SIZE unsigned char
#  endif
#endif

#ifdef CONFIG_EXBITGEN
#define ADDR0           0x5555
#define ADDR1           0x2aaa
#define FLASH_WORD_SIZE unsigned char
#endif

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

unsigned long flash_init (void)
{
	unsigned long bank_size;
	unsigned long tot_size;
	unsigned long bank_addr;
	int i;

	/* Init: no FLASHes known */
	for (i = 0; i < CONFIG_SYS_MAX_FLASH_BANKS; ++i) {
		flash_info[i].flash_id = FLASH_UNKNOWN;
		flash_info[i].size = 0;
	}

	tot_size = 0;

	/* Detect Boot Flash */
	bank_addr = CONFIG_SYS_FLASH0_BASE;
	bank_size = flash_get_size((vu_long *)bank_addr, &flash_info[0]);
	if (bank_size > 0) {
		(void)flash_protect(FLAG_PROTECT_CLEAR,
			bank_addr,
			bank_addr + bank_size - 1,
			&flash_info[0]);
	}
	if (flash_info[0].flash_id == FLASH_UNKNOWN) {
		printf ("## Unknown FLASH on Boot Flash Bank\n");
	}
	flash_info[0].size = bank_size;
	tot_size += bank_size;

	/* Detect Application Flash */
	bank_addr = CONFIG_SYS_FLASH1_BASE;
	for (i = 1; i < CONFIG_SYS_MAX_FLASH_BANKS; ++i) {
		bank_size = flash_get_size((vu_long *)bank_addr, &flash_info[i]);
		if (flash_info[i].flash_id == FLASH_UNKNOWN) {
			break;
		}
		if (bank_size > 0) {
			(void)flash_protect(FLAG_PROTECT_CLEAR,
				bank_addr,
				bank_addr + bank_size - 1,
				&flash_info[i]);
		}
		flash_info[i].size = bank_size;
		tot_size += bank_size;
		bank_addr += bank_size;
	}
	if (flash_info[1].flash_id == FLASH_UNKNOWN) {
		printf ("## Unknown FLASH on Application Flash Bank\n");
	}

	/* Protect monitor and environment sectors */
#if CONFIG_SYS_MONITOR_BASE >= CONFIG_SYS_FLASH0_BASE
	flash_protect(FLAG_PROTECT_SET,
		CONFIG_SYS_MONITOR_BASE,
		CONFIG_SYS_MONITOR_BASE + monitor_flash_len - 1,
		&flash_info[0]);
#if 0xfffffffc >= CONFIG_SYS_FLASH0_BASE
#if 0xfffffffc <= CONFIG_SYS_FLASH0_BASE + CONFIG_SYS_FLASH0_SIZE - 1
	flash_protect(FLAG_PROTECT_SET,
		0xfffffffc, 0xffffffff,
		&flash_info[0]);
#endif
#endif
#endif

#if defined(CONFIG_ENV_IS_IN_FLASH) && defined(CONFIG_ENV_ADDR)
	flash_protect(FLAG_PROTECT_SET,
		CONFIG_ENV_ADDR,
		CONFIG_ENV_ADDR + CONFIG_ENV_SIZE - 1,
		&flash_info[0]);
#endif

	return tot_size;
}

/*-----------------------------------------------------------------------
 */
void flash_print_info  (flash_info_t *info)
{
	int i;

	if (info->flash_id == FLASH_UNKNOWN) {
		printf ("missing or unknown FLASH type\n");
		return;
	}

	switch (info->flash_id & FLASH_VENDMASK) {
	case FLASH_MAN_AMD:	printf ("AMD ");		break;
	case FLASH_MAN_FUJ:	printf ("FUJITSU ");		break;
	case FLASH_MAN_SST:	printf ("SST ");		break;
	default:		printf ("Unknown Vendor ");	break;
	}

	switch (info->flash_id & FLASH_TYPEMASK) {
	case FLASH_AM040:	printf ("AM29F040 (512 Kbit, uniform sector size)\n");
				break;
	case FLASH_AM400B:	printf ("AM29LV400B (4 Mbit, bottom boot sect)\n");
				break;
	case FLASH_AM400T:	printf ("AM29LV400T (4 Mbit, top boot sector)\n");
				break;
	case FLASH_AM800B:	printf ("AM29LV800B (8 Mbit, bottom boot sect)\n");
				break;
	case FLASH_AM800T:	printf ("AM29LV800T (8 Mbit, top boot sector)\n");
				break;
	case FLASH_AM160B:	printf ("AM29LV160B (16 Mbit, bottom boot sect)\n");
				break;
	case FLASH_AM160T:	printf ("AM29LV160T (16 Mbit, top boot sector)\n");
				break;
	case FLASH_AM320B:	printf ("AM29LV320B (32 Mbit, bottom boot sect)\n");
				break;
	case FLASH_AM320T:	printf ("AM29LV320T (32 Mbit, top boot sector)\n");
				break;
	case FLASH_AMDLV033C:	printf ("AM29LV033C (32 Mbit, uniform sector size)\n");
				break;
	case FLASH_AMDLV065D:	printf ("AM29LV065D (64 Mbit, uniform sector size)\n");
				break;
	case FLASH_SST800A:	printf ("SST39LF/VF800 (8 Mbit, uniform sector size)\n");
				break;
	case FLASH_SST160A:	printf ("SST39LF/VF160 (16 Mbit, uniform sector size)\n");
				break;
	case FLASH_SST040:	printf ("SST39LF/VF040 (4 Mbit, uniform sector size)\n");
				break;
	default:		printf ("Unknown Chip Type\n");
				break;
	}

	printf ("  Size: %ld KB in %d Sectors\n",
		info->size >> 10, info->sector_count);

	printf ("  Sector Start Addresses:");
	for (i=0; i<info->sector_count; ++i) {
		if ((i % 5) == 0)
			printf ("\n   ");
		printf (" %08lX%s",
			info->start[i],
			info->protect[i] ? " (RO)" : "     "
		);
	}
	printf ("\n");
}

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


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

/*
 * The following code cannot be run from FLASH!
 */
static ulong flash_get_size (vu_long *addr, flash_info_t *info)
{
	short i;
	FLASH_WORD_SIZE value;
	ulong base = (ulong)addr;
	volatile FLASH_WORD_SIZE *addr2 = (FLASH_WORD_SIZE *)addr;

	/* Write auto select command: read Manufacturer ID */
	addr2[ADDR0] = (FLASH_WORD_SIZE)0x00AA00AA;
	addr2[ADDR1] = (FLASH_WORD_SIZE)0x00550055;
	addr2[ADDR0] = (FLASH_WORD_SIZE)0x00900090;

	value = addr2[0];

	switch (value) {
	case (FLASH_WORD_SIZE)AMD_MANUFACT:
		info->flash_id = FLASH_MAN_AMD;
		break;
	case (FLASH_WORD_SIZE)FUJ_MANUFACT:
		info->flash_id = FLASH_MAN_FUJ;
		break;
	case (FLASH_WORD_SIZE)SST_MANUFACT:
		info->flash_id = FLASH_MAN_SST;
		break;
	default:
		info->flash_id = FLASH_UNKNOWN;
		info->sector_count = 0;
		info->size = 0;
		return (0);			/* no or unknown flash	*/
	}

	value = addr2[1];			/* device ID		*/

	switch (value) {
	case (FLASH_WORD_SIZE)AMD_ID_F040B:
		info->flash_id += FLASH_AM040;
		info->sector_count = 8;
		info->size = 0x0080000; /* => 512 ko */
		break;
	case (FLASH_WORD_SIZE)AMD_ID_LV400T:
		info->flash_id += FLASH_AM400T;
		info->sector_count = 11;
		info->size = 0x00080000;
		break;				/* => 0.5 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV400B:
		info->flash_id += FLASH_AM400B;
		info->sector_count = 11;
		info->size = 0x00080000;
		break;				/* => 0.5 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV800T:
		info->flash_id += FLASH_AM800T;
		info->sector_count = 19;
		info->size = 0x00100000;
		break;				/* => 1 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV800B:
		info->flash_id += FLASH_AM800B;
		info->sector_count = 19;
		info->size = 0x00100000;
		break;				/* => 1 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV160T:
		info->flash_id += FLASH_AM160T;
		info->sector_count = 35;
		info->size = 0x00200000;
		break;				/* => 2 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV160B:
		info->flash_id += FLASH_AM160B;
		info->sector_count = 35;
		info->size = 0x00200000;
		break;				/* => 2 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV033C:
		info->flash_id += FLASH_AMDLV033C;
		info->sector_count = 64;
		info->size = 0x00400000;
		break;				/* => 4 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV065D:
		info->flash_id += FLASH_AMDLV065D;
		info->sector_count = 128;
		info->size = 0x00800000;
		break;				/* => 8 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV320T:
		info->flash_id += FLASH_AM320T;
		info->sector_count = 67;
		info->size = 0x00400000;
		break;				/* => 4 MB		*/

	case (FLASH_WORD_SIZE)AMD_ID_LV320B:
		info->flash_id += FLASH_AM320B;
		info->sector_count = 67;
		info->size = 0x00400000;
		break;				/* => 4 MB		*/

	case (FLASH_WORD_SIZE)SST_ID_xF800A:
		info->flash_id += FLASH_SST800A;
		info->sector_count = 16;
		info->size = 0x00100000;
		break;				/* => 1 MB		*/

	case (FLASH_WORD_SIZE)SST_ID_xF160A:
		info->flash_id += FLASH_SST160A;
		info->sector_count = 32;
		info->size = 0x00200000;
		break;				/* => 2 MB		*/
	case (FLASH_WORD_SIZE)SST_ID_xF040:
		info->flash_id += FLASH_SST040;
		info->sector_count = 128;
		info->size = 0x00080000;
		break;				/* => 512KB		*/

	default:
		info->flash_id = FLASH_UNKNOWN;
		return (0);			/* => no or unknown flash */

	}

	/* set up sector start address table */
	if (((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST) ||
		(info->flash_id  == FLASH_AM040) ||
		(info->flash_id == FLASH_AMDLV033C) ||
		(info->flash_id == FLASH_AMDLV065D)) {
		ulong sectsize = info->size / info->sector_count;
		for (i = 0; i < info->sector_count; i++)
			info->start[i] = base + (i * sectsize);
	} else {
	    if (info->flash_id & FLASH_BTYPE) {
		/* set sector offsets for bottom boot block type	*/
		info->start[0] = base + 0x00000000;
		info->start[1] = base + 0x00004000;
		info->start[2] = base + 0x00006000;
		info->start[3] = base + 0x00008000;
		for (i = 4; i < info->sector_count; i++) {
			info->start[i] = base + (i * 0x00010000) - 0x00030000;
		}
	    } else {
		/* set sector offsets for top boot block type		*/
		i = info->sector_count - 1;
		info->start[i--] = base + info->size - 0x00004000;
		info->start[i--] = base + info->size - 0x00006000;
		info->start[i--] = base + info->size - 0x00008000;
		for (; i >= 0; i--) {
			info->start[i] = base + i * 0x00010000;
		}
	    }
	}

	/* check for protected sectors */
	for (i = 0; i < info->sector_count; i++) {
		/* read sector protection at sector address, (A7 .. A0) = 0x02 */
		/* D0 = 1 if protected */

		addr2 = (volatile FLASH_WORD_SIZE *)(info->start[i]);
		if ((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST)
			info->protect[i] = 0;
		else
			info->protect[i] = addr2[2] & 1;
	}

	/* switch to the read mode */
	if (info->flash_id != FLASH_UNKNOWN) {
		addr2 = (FLASH_WORD_SIZE *)info->start[0];
		*addr2 = (FLASH_WORD_SIZE)0x00F000F0;	/* reset bank */
	}

	return (info->size);
}

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

int flash_erase (flash_info_t *info, int s_first, int s_last)
{
	volatile FLASH_WORD_SIZE *addr = (FLASH_WORD_SIZE *)(info->start[0]);
	volatile FLASH_WORD_SIZE *addr2;
	int flag, prot, sect;
	ulong start, now, last;

	if ((s_first < 0) || (s_first > s_last)) {
		if (info->flash_id == FLASH_UNKNOWN) {
			printf ("- missing\n");
		} else {
			printf ("- no sectors to erase\n");
		}
		return 1;
	}

	if (info->flash_id == FLASH_UNKNOWN) {
		printf ("Can't erase unknown flash type - aborted\n");
		return 1;
	}

	prot = 0;
	for (sect=s_first; sect<=s_last; ++sect) {
		if (info->protect[sect]) {
			prot++;
		}
	}

	if (prot) {
		printf ("- Warning: %d protected sectors will not be erased!\n",
			prot);
	} else {
		printf ("\n");
	}

	start = get_timer (0);
	last  = start;
	/* Start erase on unprotected sectors */
	for (sect = s_first; sect <= s_last; sect++) {
		if (info->protect[sect] == 0) {	/* not protected */
			addr2 = (FLASH_WORD_SIZE *)(info->start[sect]);

			/* Disable interrupts which might cause a timeout here */
			flag = disable_interrupts();

			addr[ADDR0] = (FLASH_WORD_SIZE)0x00AA00AA;
			addr[ADDR1] = (FLASH_WORD_SIZE)0x00550055;
			addr[ADDR0] = (FLASH_WORD_SIZE)0x00800080;
			addr[ADDR0] = (FLASH_WORD_SIZE)0x00AA00AA;
			addr[ADDR1] = (FLASH_WORD_SIZE)0x00550055;
			addr2[0] = (FLASH_WORD_SIZE)0x00300030;

			/* re-enable interrupts if necessary */
			if (flag)
				enable_interrupts();

			/* wait at least 80us - let's wait 1 ms */
			udelay (1000);

			while ((addr2[0] & 0x00800080) !=
				(FLASH_WORD_SIZE) 0x00800080) {
				if ((now=get_timer(start)) >
					   CONFIG_SYS_FLASH_ERASE_TOUT) {
					printf ("Timeout\n");
					addr[0] = (FLASH_WORD_SIZE)0x00F000F0;
					return 1;
				}

				/* show that we're waiting */
				if ((now - last) > 1000) {  /* every second  */
					putc ('.');
					last = now;
				}
			}

			addr[0] = (FLASH_WORD_SIZE)0x00F000F0;
		}
	}

	printf (" done\n");

	return 0;
}

/*-----------------------------------------------------------------------
 * Copy memory to flash, returns:
 * 0 - OK
 * 1 - write timeout
 * 2 - Flash not erased
 */

int write_buff (flash_info_t *info, uchar *src, ulong addr, ulong cnt)
{
	ulong cp, wp, data;
	int i, l, rc;

	wp = (addr & ~3);	/* get lower word aligned address */

	/*
	 * handle unaligned start bytes
	 */
	if ((l = addr - wp) != 0) {
		data = 0;
		for (i=0, cp=wp; i<l; ++i, ++cp) {
			data = (data << 8) | (*(uchar *)cp);
		}
		for (; i<4 && cnt>0; ++i) {
			data = (data << 8) | *src++;
			--cnt;
			++cp;
		}
		for (; cnt==0 && i<4; ++i, ++cp) {
			data = (data << 8) | (*(uchar *)cp);
		}

		if ((rc = write_word(info, wp, data)) != 0) {
			return (rc);
		}
		wp += 4;
	}

	/*
	 * handle word aligned part
	 */
	while (cnt >= 4) {
		data = 0;
		for (i=0; i<4; ++i) {
			data = (data << 8) | *src++;
		}
		if ((rc = write_word(info, wp, data)) != 0) {
			return (rc);
		}
		wp  += 4;
		cnt -= 4;
	}

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

	/*
	 * handle unaligned tail bytes
	 */
	data = 0;
	for (i=0, cp=wp; i<4 && cnt>0; ++i, ++cp) {
		data = (data << 8) | *src++;
		--cnt;
	}
	for (; i<4; ++i, ++cp) {
		data = (data << 8) | (*(uchar *)cp);
	}

	return (write_word(info, wp, data));
}

/*-----------------------------------------------------------------------
 * Write a word to Flash, returns:
 * 0 - OK
 * 1 - write timeout
 * 2 - Flash not erased
 */
static int write_word (flash_info_t *info, ulong dest, ulong data)
{
	volatile FLASH_WORD_SIZE *addr2 = (FLASH_WORD_SIZE *)(info->start[0]);
	volatile FLASH_WORD_SIZE *dest2 = (FLASH_WORD_SIZE *)dest;
	volatile FLASH_WORD_SIZE *data2 = (FLASH_WORD_SIZE *)&data;
	ulong start;
	int flag;
	int i;

	/* Check if Flash is (sufficiently) erased */
	if ((*((volatile ulong *)dest) & data) != data) {
		printf("dest = %08lx, *dest = %08lx, data = %08lx\n",
			dest, *(volatile ulong *)dest, data);
		return 2;
	}

	for (i=0; i < 4/sizeof(FLASH_WORD_SIZE); i++) {
		/* Disable interrupts which might cause a timeout here */
		flag = disable_interrupts();

		addr2[ADDR0] = (FLASH_WORD_SIZE)0x00AA00AA;
		addr2[ADDR1] = (FLASH_WORD_SIZE)0x00550055;
		addr2[ADDR0] = (FLASH_WORD_SIZE)0x00A000A0;
		dest2[i] = data2[i];

		/* re-enable interrupts if necessary */
		if (flag)
			enable_interrupts();

		/* data polling for D7 */
		start = get_timer (0);
		while ((dest2[i] & 0x00800080) != (data2[i] & 0x00800080)) {
			if (get_timer(start) > CONFIG_SYS_FLASH_WRITE_TOUT) {
				addr2[0] = (FLASH_WORD_SIZE)0x00F000F0;
				return (1);
			}
		}
	}

	addr2[0] = (FLASH_WORD_SIZE)0x00F000F0;

	return (0);
}

/*-----------------------------------------------------------------------
 */
Commit	Line	Data
d1cbe85b WD	1	/*
	2	* (C) Copyright 2003
	3	* Wolfgang Denk, DENX Software Engineering, [email protected].
	4	*
	5	* See file CREDITS for list of people who contributed to this
	6	* project.
	7	*
	8	* This program is free software; you can redistribute it and/or
	9	* modify it under the terms of the GNU General Public License as
	10	* published by the Free Software Foundation; either version 2 of
	11	* the License, or (at your option) any later version.
	12	*
	13	* This program is distributed in the hope that it will be useful,
	14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	16	* GNU General Public License for more details.
	17	*
	18	* You should have received a copy of the GNU General Public License
	19	* along with this program; if not, write to the Free Software
	20	* Foundation, Inc., 59 Temple Place, Suite 330, Boston,
	21	* MA 02111-1307 USA
	22	*/
	23
	24	/*
	25	* Modified 4/5/2001
	26	* Wait for completion of each sector erase command issued
	27	* 4/5/2001
	28	* Chris Hallinan - DS4.COM, Inc. - [email protected]
	29	*/
	30
4ff170a8	31	#include <common.h>
d1cbe85b WD	32	#include <asm/u-boot.h>
	33	#include <asm/processor.h>
	34	#include <ppc4xx.h>
d1cbe85b	35
6d0f6bcf	36	flash_info_t flash_info[CONFIG_SYS_MAX_FLASH_BANKS]; /* info for FLASH chips */
d1cbe85b WD	37
	38	/*-----------------------------------------------------------------------
	39	* Functions
	40	*/
	41	static ulong flash_get_size (vu_long addr, flash_info_t info);
	42	static int write_word (flash_info_t *info, ulong dest, ulong data);
	43
	44	#ifdef MEIGSBOARD_ONBOARD_FLASH /* onboard = 2MB */
	45	# ifdef CONFIG_EXBITGEN
	46	# define FLASH_WORD_SIZE unsigned long
	47	# endif
	48	#else /* Meigsboard socket flash = 512KB */
	49	# ifdef CONFIG_EXBITGEN
	50	# define FLASH_WORD_SIZE unsigned char
	51	# endif
	52	#endif
	53
	54	#ifdef CONFIG_EXBITGEN
	55	#define ADDR0 0x5555
	56	#define ADDR1 0x2aaa
	57	#define FLASH_WORD_SIZE unsigned char
	58	#endif
	59
	60	/*-----------------------------------------------------------------------
	61	*/
	62
	63	unsigned long flash_init (void)
	64	{
	65	unsigned long bank_size;
	66	unsigned long tot_size;
	67	unsigned long bank_addr;
	68	int i;
	69
	70	/* Init: no FLASHes known */
6d0f6bcf	71	for (i = 0; i < CONFIG_SYS_MAX_FLASH_BANKS; ++i) {
d1cbe85b WD	72	flash_info[i].flash_id = FLASH_UNKNOWN;
	73	flash_info[i].size = 0;
	74	}
	75
	76	tot_size = 0;
	77
	78	/* Detect Boot Flash */
6d0f6bcf	79	bank_addr = CONFIG_SYS_FLASH0_BASE;
d1cbe85b WD	80	bank_size = flash_get_size((vu_long *)bank_addr, &flash_info[0]);
	81	if (bank_size > 0) {
	82	(void)flash_protect(FLAG_PROTECT_CLEAR,
	83	bank_addr,
	84	bank_addr + bank_size - 1,
	85	&flash_info[0]);
	86	}
	87	if (flash_info[0].flash_id == FLASH_UNKNOWN) {
	88	printf ("## Unknown FLASH on Boot Flash Bank\n");
	89	}
	90	flash_info[0].size = bank_size;
	91	tot_size += bank_size;
	92
	93	/* Detect Application Flash */
6d0f6bcf JCPV	94	bank_addr = CONFIG_SYS_FLASH1_BASE;
6d0f6bcf JCPV	95	for (i = 1; i < CONFIG_SYS_MAX_FLASH_BANKS; ++i) {
d1cbe85b WD	96	bank_size = flash_get_size((vu_long *)bank_addr, &flash_info[i]);
	97	if (flash_info[i].flash_id == FLASH_UNKNOWN) {
	98	break;
	99	}
	100	if (bank_size > 0) {
	101	(void)flash_protect(FLAG_PROTECT_CLEAR,
	102	bank_addr,
	103	bank_addr + bank_size - 1,
	104	&flash_info[i]);
	105	}
	106	flash_info[i].size = bank_size;
	107	tot_size += bank_size;
	108	bank_addr += bank_size;
	109	}
	110	if (flash_info[1].flash_id == FLASH_UNKNOWN) {
	111	printf ("## Unknown FLASH on Application Flash Bank\n");
	112	}
	113
	114	/* Protect monitor and environment sectors */
6d0f6bcf	115	#if CONFIG_SYS_MONITOR_BASE >= CONFIG_SYS_FLASH0_BASE
d1cbe85b	116	flash_protect(FLAG_PROTECT_SET,
6d0f6bcf JCPV	117	CONFIG_SYS_MONITOR_BASE,
6d0f6bcf JCPV	118	CONFIG_SYS_MONITOR_BASE + monitor_flash_len - 1,
d1cbe85b	119	&flash_info[0]);
6d0f6bcf JCPV	120	#if 0xfffffffc >= CONFIG_SYS_FLASH0_BASE
6d0f6bcf JCPV	121	#if 0xfffffffc <= CONFIG_SYS_FLASH0_BASE + CONFIG_SYS_FLASH0_SIZE - 1
d1cbe85b WD	122	flash_protect(FLAG_PROTECT_SET,
	123	0xfffffffc, 0xffffffff,
	124	&flash_info[0]);
	125	#endif
	126	#endif
	127	#endif
	128
0e8d1586	129	#if defined(CONFIG_ENV_IS_IN_FLASH) && defined(CONFIG_ENV_ADDR)
d1cbe85b	130	flash_protect(FLAG_PROTECT_SET,
0e8d1586 JCPV	131	CONFIG_ENV_ADDR,
0e8d1586 JCPV	132	CONFIG_ENV_ADDR + CONFIG_ENV_SIZE - 1,
d1cbe85b WD	133	&flash_info[0]);
	134	#endif
	135
	136	return tot_size;
	137	}
	138
	139	/*-----------------------------------------------------------------------
	140	*/
	141	void flash_print_info (flash_info_t *info)
	142	{
	143	int i;
	144
	145	if (info->flash_id == FLASH_UNKNOWN) {
	146	printf ("missing or unknown FLASH type\n");
	147	return;
	148	}
	149
	150	switch (info->flash_id & FLASH_VENDMASK) {
	151	case FLASH_MAN_AMD: printf ("AMD "); break;
	152	case FLASH_MAN_FUJ: printf ("FUJITSU "); break;
	153	case FLASH_MAN_SST: printf ("SST "); break;
	154	default: printf ("Unknown Vendor "); break;
	155	}
	156
	157	switch (info->flash_id & FLASH_TYPEMASK) {
	158	case FLASH_AM040: printf ("AM29F040 (512 Kbit, uniform sector size)\n");
	159	break;
	160	case FLASH_AM400B: printf ("AM29LV400B (4 Mbit, bottom boot sect)\n");
	161	break;
	162	case FLASH_AM400T: printf ("AM29LV400T (4 Mbit, top boot sector)\n");
	163	break;
	164	case FLASH_AM800B: printf ("AM29LV800B (8 Mbit, bottom boot sect)\n");
	165	break;
	166	case FLASH_AM800T: printf ("AM29LV800T (8 Mbit, top boot sector)\n");
	167	break;
	168	case FLASH_AM160B: printf ("AM29LV160B (16 Mbit, bottom boot sect)\n");
	169	break;
	170	case FLASH_AM160T: printf ("AM29LV160T (16 Mbit, top boot sector)\n");
	171	break;
	172	case FLASH_AM320B: printf ("AM29LV320B (32 Mbit, bottom boot sect)\n");
	173	break;
	174	case FLASH_AM320T: printf ("AM29LV320T (32 Mbit, top boot sector)\n");
	175	break;
	176	case FLASH_AMDLV033C: printf ("AM29LV033C (32 Mbit, uniform sector size)\n");
	177	break;
	178	case FLASH_AMDLV065D: printf ("AM29LV065D (64 Mbit, uniform sector size)\n");
	179	break;
	180	case FLASH_SST800A: printf ("SST39LF/VF800 (8 Mbit, uniform sector size)\n");
	181	break;
	182	case FLASH_SST160A: printf ("SST39LF/VF160 (16 Mbit, uniform sector size)\n");
	183	break;
	184	case FLASH_SST040: printf ("SST39LF/VF040 (4 Mbit, uniform sector size)\n");
	185	break;
	186	default: printf ("Unknown Chip Type\n");
	187	break;
	188	}
	189
	190	printf (" Size: %ld KB in %d Sectors\n",
	191	info->size >> 10, info->sector_count);
	192
	193	printf (" Sector Start Addresses:");
	194	for (i=0; i<info->sector_count; ++i) {
	195	if ((i % 5) == 0)
	196	printf ("\n ");
197	printf (" %08lX%s",
198	info->start[i],
199	info->protect[i] ? " (RO)" : " "
200	);
201	}
202	printf ("\n");
203	}
204
205	/*-----------------------------------------------------------------------
206	*/
207
208
209	/*-----------------------------------------------------------------------
210	*/
211
212	/*
213	* The following code cannot be run from FLASH!
214	*/
215	static ulong flash_get_size (vu_long addr, flash_info_t info)
216	{
217	short i;
218	FLASH_WORD_SIZE value;
219	ulong base = (ulong)addr;
945af8d7	220	volatile FLASH_WORD_SIZE addr2 = (FLASH_WORD_SIZE )addr;
d1cbe85b WD	221
	222	/* Write auto select command: read Manufacturer ID */
	223	addr2[ADDR0] = (FLASH_WORD_SIZE)0x00AA00AA;
	224	addr2[ADDR1] = (FLASH_WORD_SIZE)0x00550055;
	225	addr2[ADDR0] = (FLASH_WORD_SIZE)0x00900090;
	226
	227	value = addr2[0];
	228
	229	switch (value) {
	230	case (FLASH_WORD_SIZE)AMD_MANUFACT:
	231	info->flash_id = FLASH_MAN_AMD;
	232	break;
	233	case (FLASH_WORD_SIZE)FUJ_MANUFACT:
	234	info->flash_id = FLASH_MAN_FUJ;
	235	break;
	236	case (FLASH_WORD_SIZE)SST_MANUFACT:
	237	info->flash_id = FLASH_MAN_SST;
	238	break;
	239	default:
	240	info->flash_id = FLASH_UNKNOWN;
	241	info->sector_count = 0;
	242	info->size = 0;
	243	return (0); /* no or unknown flash */
	244	}
	245
	246	value = addr2[1]; /* device ID */
	247
	248	switch (value) {
	249	case (FLASH_WORD_SIZE)AMD_ID_F040B:
945af8d7	250	info->flash_id += FLASH_AM040;
d1cbe85b WD	251	info->sector_count = 8;
	252	info->size = 0x0080000; /* => 512 ko */
	253	break;
	254	case (FLASH_WORD_SIZE)AMD_ID_LV400T:
	255	info->flash_id += FLASH_AM400T;
	256	info->sector_count = 11;
	257	info->size = 0x00080000;
	258	break; /* => 0.5 MB */
	259
	260	case (FLASH_WORD_SIZE)AMD_ID_LV400B:
	261	info->flash_id += FLASH_AM400B;
	262	info->sector_count = 11;
	263	info->size = 0x00080000;
	264	break; /* => 0.5 MB */
	265
	266	case (FLASH_WORD_SIZE)AMD_ID_LV800T:
	267	info->flash_id += FLASH_AM800T;
	268	info->sector_count = 19;
	269	info->size = 0x00100000;
	270	break; /* => 1 MB */
	271
	272	case (FLASH_WORD_SIZE)AMD_ID_LV800B:
	273	info->flash_id += FLASH_AM800B;
	274	info->sector_count = 19;
	275	info->size = 0x00100000;
	276	break; /* => 1 MB */
	277
	278	case (FLASH_WORD_SIZE)AMD_ID_LV160T:
	279	info->flash_id += FLASH_AM160T;
	280	info->sector_count = 35;
	281	info->size = 0x00200000;
	282	break; /* => 2 MB */
	283
	284	case (FLASH_WORD_SIZE)AMD_ID_LV160B:
	285	info->flash_id += FLASH_AM160B;
	286	info->sector_count = 35;
	287	info->size = 0x00200000;
	288	break; /* => 2 MB */
	289
	290	case (FLASH_WORD_SIZE)AMD_ID_LV033C:
	291	info->flash_id += FLASH_AMDLV033C;
	292	info->sector_count = 64;
	293	info->size = 0x00400000;
	294	break; /* => 4 MB */
	295
	296	case (FLASH_WORD_SIZE)AMD_ID_LV065D:
	297	info->flash_id += FLASH_AMDLV065D;
	298	info->sector_count = 128;
	299	info->size = 0x00800000;
	300	break; /* => 8 MB */
	301
	302	case (FLASH_WORD_SIZE)AMD_ID_LV320T:
	303	info->flash_id += FLASH_AM320T;
	304	info->sector_count = 67;
	305	info->size = 0x00400000;
	306	break; /* => 4 MB */
	307
	308	case (FLASH_WORD_SIZE)AMD_ID_LV320B:
	309	info->flash_id += FLASH_AM320B;
	310	info->sector_count = 67;
	311	info->size = 0x00400000;
	312	break; /* => 4 MB */
	313
	314	case (FLASH_WORD_SIZE)SST_ID_xF800A:
315	info->flash_id += FLASH_SST800A;
316	info->sector_count = 16;
317	info->size = 0x00100000;
318	break; /* => 1 MB */
319
320	case (FLASH_WORD_SIZE)SST_ID_xF160A:
321	info->flash_id += FLASH_SST160A;
322	info->sector_count = 32;
323	info->size = 0x00200000;
324	break; /* => 2 MB */
325	case (FLASH_WORD_SIZE)SST_ID_xF040:
326	info->flash_id += FLASH_SST040;
327	info->sector_count = 128;
328	info->size = 0x00080000;
329	break; /* => 512KB */
330
331	default:
332	info->flash_id = FLASH_UNKNOWN;
333	return (0); /* => no or unknown flash */
334
335	}
336
337	/* set up sector start address table */
945af8d7	338	if (((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST) \|\|
d1cbe85b WD	339	(info->flash_id == FLASH_AM040) \|\|
	340	(info->flash_id == FLASH_AMDLV033C) \|\|
	341	(info->flash_id == FLASH_AMDLV065D)) {
	342	ulong sectsize = info->size / info->sector_count;
	343	for (i = 0; i < info->sector_count; i++)
	344	info->start[i] = base + (i * sectsize);
945af8d7	345	} else {
d1cbe85b WD	346	if (info->flash_id & FLASH_BTYPE) {
	347	/* set sector offsets for bottom boot block type */
	348	info->start[0] = base + 0x00000000;
	349	info->start[1] = base + 0x00004000;
	350	info->start[2] = base + 0x00006000;
	351	info->start[3] = base + 0x00008000;
	352	for (i = 4; i < info->sector_count; i++) {
	353	info->start[i] = base + (i * 0x00010000) - 0x00030000;
	354	}
	355	} else {
	356	/* set sector offsets for top boot block type */
	357	i = info->sector_count - 1;
	358	info->start[i--] = base + info->size - 0x00004000;
	359	info->start[i--] = base + info->size - 0x00006000;
	360	info->start[i--] = base + info->size - 0x00008000;
	361	for (; i >= 0; i--) {
	362	info->start[i] = base + i * 0x00010000;
	363	}
	364	}
	365	}
	366
	367	/* check for protected sectors */
	368	for (i = 0; i < info->sector_count; i++) {
	369	/* read sector protection at sector address, (A7 .. A0) = 0x02 */
	370	/* D0 = 1 if protected */
	371
	372	addr2 = (volatile FLASH_WORD_SIZE *)(info->start[i]);
945af8d7	373	if ((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST)
d1cbe85b	374	info->protect[i] = 0;
945af8d7	375	else
d1cbe85b WD	376	info->protect[i] = addr2[2] & 1;
	377	}
	378
	379	/* switch to the read mode */
	380	if (info->flash_id != FLASH_UNKNOWN) {
	381	addr2 = (FLASH_WORD_SIZE *)info->start[0];
	382	addr2 = (FLASH_WORD_SIZE)0x00F000F0; / reset bank */
	383	}
	384
	385	return (info->size);
	386	}
	387
	388	/*-----------------------------------------------------------------------
	389	*/
	390
	391	int flash_erase (flash_info_t *info, int s_first, int s_last)
	392	{
	393	volatile FLASH_WORD_SIZE addr = (FLASH_WORD_SIZE )(info->start[0]);
	394	volatile FLASH_WORD_SIZE *addr2;
	395	int flag, prot, sect;
	396	ulong start, now, last;
	397
	398	if ((s_first < 0) \|\| (s_first > s_last)) {
	399	if (info->flash_id == FLASH_UNKNOWN) {
	400	printf ("- missing\n");
	401	} else {
	402	printf ("- no sectors to erase\n");
	403	}
	404	return 1;
	405	}
	406
	407	if (info->flash_id == FLASH_UNKNOWN) {
	408	printf ("Can't erase unknown flash type - aborted\n");
	409	return 1;
	410	}
	411
	412	prot = 0;
	413	for (sect=s_first; sect<=s_last; ++sect) {
	414	if (info->protect[sect]) {
	415	prot++;
	416	}
	417	}
	418
	419	if (prot) {
	420	printf ("- Warning: %d protected sectors will not be erased!\n",
	421	prot);
	422	} else {
	423	printf ("\n");
	424	}
	425
	426	start = get_timer (0);
	427	last = start;
	428	/* Start erase on unprotected sectors */
	429	for (sect = s_first; sect <= s_last; sect++) {
	430	if (info->protect[sect] == 0) { /* not protected */
	431	addr2 = (FLASH_WORD_SIZE *)(info->start[sect]);
	432
	433	/* Disable interrupts which might cause a timeout here */
	434	flag = disable_interrupts();
	435
	436	addr[ADDR0] = (FLASH_WORD_SIZE)0x00AA00AA;
	437	addr[ADDR1] = (FLASH_WORD_SIZE)0x00550055;
	438	addr[ADDR0] = (FLASH_WORD_SIZE)0x00800080;
	439	addr[ADDR0] = (FLASH_WORD_SIZE)0x00AA00AA;
440	addr[ADDR1] = (FLASH_WORD_SIZE)0x00550055;
441	addr2[0] = (FLASH_WORD_SIZE)0x00300030;
442
443	/* re-enable interrupts if necessary */
444	if (flag)
445	enable_interrupts();
446
447	/* wait at least 80us - let's wait 1 ms */
448	udelay (1000);
449
450	while ((addr2[0] & 0x00800080) !=
451	(FLASH_WORD_SIZE) 0x00800080) {
452	if ((now=get_timer(start)) >
6d0f6bcf	453	CONFIG_SYS_FLASH_ERASE_TOUT) {
d1cbe85b WD	454	printf ("Timeout\n");
	455	addr[0] = (FLASH_WORD_SIZE)0x00F000F0;
	456	return 1;
	457	}
	458
	459	/* show that we're waiting */
	460	if ((now - last) > 1000) { /* every second */
	461	putc ('.');
	462	last = now;
	463	}
	464	}
	465
	466	addr[0] = (FLASH_WORD_SIZE)0x00F000F0;
	467	}
	468	}
	469
	470	printf (" done\n");
	471
	472	return 0;
	473	}
	474
	475	/*-----------------------------------------------------------------------
	476	* Copy memory to flash, returns:
	477	* 0 - OK
	478	* 1 - write timeout
	479	* 2 - Flash not erased
	480	*/
	481
	482	int write_buff (flash_info_t info, uchar src, ulong addr, ulong cnt)
	483	{
	484	ulong cp, wp, data;
	485	int i, l, rc;
	486
	487	wp = (addr & ~3); /* get lower word aligned address */
	488
	489	/*
	490	* handle unaligned start bytes
	491	*/
	492	if ((l = addr - wp) != 0) {
	493	data = 0;
	494	for (i=0, cp=wp; i<l; ++i, ++cp) {
	495	data = (data << 8) \| ((uchar )cp);
	496	}
	497	for (; i<4 && cnt>0; ++i) {
	498	data = (data << 8) \| *src++;
	499	--cnt;
	500	++cp;
	501	}
	502	for (; cnt==0 && i<4; ++i, ++cp) {
	503	data = (data << 8) \| ((uchar )cp);
	504	}
	505
	506	if ((rc = write_word(info, wp, data)) != 0) {
	507	return (rc);
	508	}
	509	wp += 4;
	510	}
	511
	512	/*
	513	* handle word aligned part
	514	*/
	515	while (cnt >= 4) {
	516	data = 0;
	517	for (i=0; i<4; ++i) {
518	data = (data << 8) \| *src++;
519	}
520	if ((rc = write_word(info, wp, data)) != 0) {
521	return (rc);
522	}
523	wp += 4;
524	cnt -= 4;
525	}
526
527	if (cnt == 0) {
528	return (0);
529	}
530
531	/*
532	* handle unaligned tail bytes
533	*/
534	data = 0;
535	for (i=0, cp=wp; i<4 && cnt>0; ++i, ++cp) {
536	data = (data << 8) \| *src++;
537	--cnt;
538	}
539	for (; i<4; ++i, ++cp) {
540	data = (data << 8) \| ((uchar )cp);
541	}
542
543	return (write_word(info, wp, data));
544	}
545
546	/*-----------------------------------------------------------------------
547	* Write a word to Flash, returns:
548	* 0 - OK
549	* 1 - write timeout
550	* 2 - Flash not erased
551	*/
552	static int write_word (flash_info_t *info, ulong dest, ulong data)
553	{
945af8d7 WD	554	volatile FLASH_WORD_SIZE addr2 = (FLASH_WORD_SIZE )(info->start[0]);
	555	volatile FLASH_WORD_SIZE dest2 = (FLASH_WORD_SIZE )dest;
	556	volatile FLASH_WORD_SIZE data2 = (FLASH_WORD_SIZE )&data;
d1cbe85b WD	557	ulong start;
d1cbe85b WD	558	int flag;
945af8d7	559	int i;
d1cbe85b WD	560
	561	/* Check if Flash is (sufficiently) erased */
	562	if ((((volatile ulong )dest) & data) != data) {
	563	printf("dest = %08lx, *dest = %08lx, data = %08lx\n",
	564	dest, (volatile ulong )dest, data);
	565	return 2;
	566	}
	567
945af8d7	568	for (i=0; i < 4/sizeof(FLASH_WORD_SIZE); i++) {
d1cbe85b WD	569	/* Disable interrupts which might cause a timeout here */
	570	flag = disable_interrupts();
	571
	572	addr2[ADDR0] = (FLASH_WORD_SIZE)0x00AA00AA;
	573	addr2[ADDR1] = (FLASH_WORD_SIZE)0x00550055;
	574	addr2[ADDR0] = (FLASH_WORD_SIZE)0x00A000A0;
	575	dest2[i] = data2[i];
	576
	577	/* re-enable interrupts if necessary */
	578	if (flag)
	579	enable_interrupts();
	580
	581	/* data polling for D7 */
	582	start = get_timer (0);
	583	while ((dest2[i] & 0x00800080) != (data2[i] & 0x00800080)) {
6d0f6bcf	584	if (get_timer(start) > CONFIG_SYS_FLASH_WRITE_TOUT) {
d1cbe85b WD	585	addr2[0] = (FLASH_WORD_SIZE)0x00F000F0;
	586	return (1);
	587	}
	588	}
	589	}
	590
	591	addr2[0] = (FLASH_WORD_SIZE)0x00F000F0;
	592
	593	return (0);
	594	}
	595
	596	/*-----------------------------------------------------------------------
	597	*/