[people/ms/u-boot.git] / drivers / mtd / dataflash.c

/* LowLevel function for ATMEL DataFlash support
 * Author : Hamid Ikdoumi (Atmel)
 *
 * 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
 *
 */
#include <common.h>
#include <config.h>
#include <asm/hardware.h>
#include <dataflash.h>

static AT91S_DataFlash DataFlashInst;

extern void AT91F_SpiInit (void);
extern int AT91F_DataflashProbe (int i, AT91PS_DataflashDesc pDesc);
extern int AT91F_DataFlashRead (AT91PS_DataFlash pDataFlash,
				unsigned long addr,
				unsigned long size, char *buffer);
extern int AT91F_DataFlashWrite( AT91PS_DataFlash pDataFlash,
				unsigned char *src,
				int dest,
				int size );

int AT91F_DataflashInit (void)
{
	int i, j;
	int dfcode;
	int part;
	int last_part;
	int found[CONFIG_SYS_MAX_DATAFLASH_BANKS];
	unsigned char protected;

	AT91F_SpiInit ();

	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
		found[i] = 0;
		dataflash_info[i].Desc.state = IDLE;
		dataflash_info[i].id = 0;
		dataflash_info[i].Device.pages_number = 0;
		dfcode = AT91F_DataflashProbe (cs[i].cs,
				&dataflash_info[i].Desc);

		switch (dfcode) {
		case AT45DB021:
			dataflash_info[i].Device.pages_number = 1024;
			dataflash_info[i].Device.pages_size = 264;
			dataflash_info[i].Device.page_offset = 9;
			dataflash_info[i].Device.byte_mask = 0x300;
			dataflash_info[i].Device.cs = cs[i].cs;
			dataflash_info[i].Desc.DataFlash_state = IDLE;
			dataflash_info[i].logical_address = cs[i].addr;
			dataflash_info[i].id = dfcode;
			found[i] += dfcode;;
			break;

		case AT45DB081:
			dataflash_info[i].Device.pages_number = 4096;
			dataflash_info[i].Device.pages_size = 264;
			dataflash_info[i].Device.page_offset = 9;
			dataflash_info[i].Device.byte_mask = 0x300;
			dataflash_info[i].Device.cs = cs[i].cs;
			dataflash_info[i].Desc.DataFlash_state = IDLE;
			dataflash_info[i].logical_address = cs[i].addr;
			dataflash_info[i].id = dfcode;
			found[i] += dfcode;;
			break;

		case AT45DB161:
			dataflash_info[i].Device.pages_number = 4096;
			dataflash_info[i].Device.pages_size = 528;
			dataflash_info[i].Device.page_offset = 10;
			dataflash_info[i].Device.byte_mask = 0x300;
			dataflash_info[i].Device.cs = cs[i].cs;
			dataflash_info[i].Desc.DataFlash_state = IDLE;
			dataflash_info[i].logical_address = cs[i].addr;
			dataflash_info[i].id = dfcode;
			found[i] += dfcode;;
			break;

		case AT45DB321:
			dataflash_info[i].Device.pages_number = 8192;
			dataflash_info[i].Device.pages_size = 528;
			dataflash_info[i].Device.page_offset = 10;
			dataflash_info[i].Device.byte_mask = 0x300;
			dataflash_info[i].Device.cs = cs[i].cs;
			dataflash_info[i].Desc.DataFlash_state = IDLE;
			dataflash_info[i].logical_address = cs[i].addr;
			dataflash_info[i].id = dfcode;
			found[i] += dfcode;;
			break;

		case AT45DB642:
			dataflash_info[i].Device.pages_number = 8192;
			dataflash_info[i].Device.pages_size = 1056;
			dataflash_info[i].Device.page_offset = 11;
			dataflash_info[i].Device.byte_mask = 0x700;
			dataflash_info[i].Device.cs = cs[i].cs;
			dataflash_info[i].Desc.DataFlash_state = IDLE;
			dataflash_info[i].logical_address = cs[i].addr;
			dataflash_info[i].id = dfcode;
			found[i] += dfcode;;
			break;

		case AT45DB128:
			dataflash_info[i].Device.pages_number = 16384;
			dataflash_info[i].Device.pages_size = 1056;
			dataflash_info[i].Device.page_offset = 11;
			dataflash_info[i].Device.byte_mask = 0x700;
			dataflash_info[i].Device.cs = cs[i].cs;
			dataflash_info[i].Desc.DataFlash_state = IDLE;
			dataflash_info[i].logical_address = cs[i].addr;
			dataflash_info[i].id = dfcode;
			found[i] += dfcode;;
			break;

		default:
			dfcode = 0;
			break;
		}
		/* set the last area end to the dataflash size*/
		area_list[NB_DATAFLASH_AREA -1].end =
				(dataflash_info[i].Device.pages_number *
				dataflash_info[i].Device.pages_size)-1;

		part = 0;
		last_part = 0;
		/* set the area addresses */
		for(j = 0; j<NB_DATAFLASH_AREA; j++) {
			if(found[i]!=0) {
				dataflash_info[i].Device.area_list[j].start =
					area_list[part].start +
					dataflash_info[i].logical_address;
				if(area_list[part].end == 0xffffffff) {
					dataflash_info[i].Device.area_list[j].end =
						dataflash_info[i].end_address +
						dataflash_info	[i].logical_address;
					last_part = 1;
				} else {
					dataflash_info[i].Device.area_list[j].end =
						area_list[part].end +
						dataflash_info[i].logical_address;
				}
				protected = area_list[part].protected;
				/* Set the environment according to the label...*/
				if(protected == FLAG_PROTECT_INVALID) {
					dataflash_info[i].Device.area_list[j].protected =
						FLAG_PROTECT_INVALID;
				} else {
					dataflash_info[i].Device.area_list[j].protected =
						protected;
				}
				strcpy((char*)(dataflash_info[i].Device.area_list[j].label),
						(const char *)area_list[part].label);
			}
			part++;
		}
	}
	return found[0];
}

void AT91F_DataflashSetEnv (void)
{
	int i, j;
	int part;
	unsigned char env;
	unsigned char s[32];	/* Will fit a long int in hex */
	unsigned long start;

	for (i = 0, part= 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
		for(j = 0; j<NB_DATAFLASH_AREA; j++) {
			env = area_list[part].setenv;
			/* Set the environment according to the label...*/
			if((env & FLAG_SETENV) == FLAG_SETENV) {
				start =
				dataflash_info[i].Device.area_list[j].start;
				sprintf((char*) s,"%lX",start);
				setenv((char*) area_list[part].label,(char*) s);
			}
			part++;
		}
	}
}

void dataflash_print_info (void)
{
	int i, j;

	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
		if (dataflash_info[i].id != 0) {
			printf("DataFlash:");
			switch (dataflash_info[i].id) {
			case AT45DB021:
				printf("AT45DB021\n");
				break;
			case AT45DB161:
				printf("AT45DB161\n");
				break;

			case AT45DB321:
				printf("AT45DB321\n");
				break;

			case AT45DB642:
				printf("AT45DB642\n");
				break;
			case AT45DB128:
				printf("AT45DB128\n");
				break;
			}

			printf("Nb pages: %6d\n"
				"Page Size: %6d\n"
				"Size=%8d bytes\n"
				"Logical address: 0x%08X\n",
				(unsigned int) dataflash_info[i].Device.pages_number,
				(unsigned int) dataflash_info[i].Device.pages_size,
				(unsigned int) dataflash_info[i].Device.pages_number *
				dataflash_info[i].Device.pages_size,
				(unsigned int) dataflash_info[i].logical_address);
			for (j=0; j< NB_DATAFLASH_AREA; j++) {
				switch(dataflash_info[i].Device.area_list[j].protected) {
				case	FLAG_PROTECT_SET:
				case	FLAG_PROTECT_CLEAR:
					printf("Area %i:\t%08lX to %08lX %s", j,
						dataflash_info[i].Device.area_list[j].start,
						dataflash_info[i].Device.area_list[j].end,
						(dataflash_info[i].Device.area_list[j].protected==FLAG_PROTECT_SET) ? "(RO)" : "    ");
						printf(" %s\n",	dataflash_info[i].Device.area_list[j].label);
					break;
				case	FLAG_PROTECT_INVALID:
					break;
				}
			}
		}
	}
}

/*---------------------------------------------------------------------------*/
/* Function Name       : AT91F_DataflashSelect				     */
/* Object              : Select the correct device			     */
/*---------------------------------------------------------------------------*/
AT91PS_DataFlash AT91F_DataflashSelect (AT91PS_DataFlash pFlash,
				unsigned long *addr)
{
	char addr_valid = 0;
	int i;

	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++)
		if ( dataflash_info[i].id
			&& ((((int) *addr) & 0xFF000000) ==
			dataflash_info[i].logical_address)) {
			addr_valid = 1;
			break;
		}
	if (!addr_valid) {
		pFlash = (AT91PS_DataFlash) 0;
		return pFlash;
	}
	pFlash->pDataFlashDesc = &(dataflash_info[i].Desc);
	pFlash->pDevice = &(dataflash_info[i].Device);
	*addr -= dataflash_info[i].logical_address;
	return (pFlash);
}

/*---------------------------------------------------------------------------*/
/* Function Name       : addr_dataflash					     */
/* Object              : Test if address is valid			     */
/*---------------------------------------------------------------------------*/
int addr_dataflash (unsigned long addr)
{
	int addr_valid = 0;
	int i;

	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
		if ((((int) addr) & 0xFF000000) ==
			dataflash_info[i].logical_address) {
			addr_valid = 1;
			break;
		}
	}

	return addr_valid;
}

/*---------------------------------------------------------------------------*/
/* Function Name       : size_dataflash					     */
/* Object              : Test if address is valid regarding the size	     */
/*---------------------------------------------------------------------------*/
int size_dataflash (AT91PS_DataFlash pdataFlash, unsigned long addr,
			unsigned long size)
{
	/* is outside the dataflash */
	if (((int)addr & 0x0FFFFFFF) > (pdataFlash->pDevice->pages_size *
		pdataFlash->pDevice->pages_number)) return 0;
	/* is too large for the dataflash */
	if (size > ((pdataFlash->pDevice->pages_size *
		pdataFlash->pDevice->pages_number) -
		((int)addr & 0x0FFFFFFF))) return 0;

	return 1;
}

/*---------------------------------------------------------------------------*/
/* Function Name       : prot_dataflash					     */
/* Object              : Test if destination area is protected		     */
/*---------------------------------------------------------------------------*/
int prot_dataflash (AT91PS_DataFlash pdataFlash, unsigned long addr)
{
	int area;

	/* find area */
	for (area=0; area < NB_DATAFLASH_AREA; area++) {
		if ((addr >= pdataFlash->pDevice->area_list[area].start) &&
			(addr < pdataFlash->pDevice->area_list[area].end))
			break;
	}
	if (area == NB_DATAFLASH_AREA)
		return -1;

	/*test protection value*/
	if (pdataFlash->pDevice->area_list[area].protected == FLAG_PROTECT_SET)
		return 0;
	if (pdataFlash->pDevice->area_list[area].protected == FLAG_PROTECT_INVALID)
		return 0;

	return 1;
}

/*--------------------------------------------------------------------------*/
/* Function Name       : dataflash_real_protect				    */
/* Object              : protect/unprotect area				    */
/*--------------------------------------------------------------------------*/
int dataflash_real_protect (int flag, unsigned long start_addr,
				unsigned long end_addr)
{
	int i,j, area1, area2, addr_valid = 0;

	/* find dataflash */
	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
		if ((((int) start_addr) & 0xF0000000) ==
			dataflash_info[i].logical_address) {
				addr_valid = 1;
				break;
		}
	}
	if (!addr_valid) {
		return -1;
	}
	/* find start area */
	for (area1=0; area1 < NB_DATAFLASH_AREA; area1++) {
		if (start_addr == dataflash_info[i].Device.area_list[area1].start)
			break;
	}
	if (area1 == NB_DATAFLASH_AREA) return -1;
	/* find end area */
	for (area2=0; area2 < NB_DATAFLASH_AREA; area2++) {
		if (end_addr == dataflash_info[i].Device.area_list[area2].end)
			break;
	}
	if (area2 == NB_DATAFLASH_AREA)
		return -1;

	/*set protection value*/
	for(j = area1; j < area2+1 ; j++)
		if(dataflash_info[i].Device.area_list[j].protected
				!= FLAG_PROTECT_INVALID) {
			if (flag == 0) {
				dataflash_info[i].Device.area_list[j].protected
					= FLAG_PROTECT_CLEAR;
			} else {
				dataflash_info[i].Device.area_list[j].protected
					= FLAG_PROTECT_SET;
			}
		}

	return (area2-area1+1);
}

/*---------------------------------------------------------------------------*/
/* Function Name       : read_dataflash					     */
/* Object              : dataflash memory read				     */
/*---------------------------------------------------------------------------*/
int read_dataflash (unsigned long addr, unsigned long size, char *result)
{
	unsigned long AddrToRead = addr;
	AT91PS_DataFlash pFlash = &DataFlashInst;

	pFlash = AT91F_DataflashSelect (pFlash, &AddrToRead);

	if (pFlash == 0)
		return ERR_UNKNOWN_FLASH_TYPE;

	if (size_dataflash(pFlash,addr,size) == 0)
		return ERR_INVAL;

	return (AT91F_DataFlashRead (pFlash, AddrToRead, size, result));
}

/*---------------------------------------------------------------------------*/
/* Function Name       : write_dataflash				     */
/* Object              : write a block in dataflash			     */
/*---------------------------------------------------------------------------*/
int write_dataflash (unsigned long addr_dest, unsigned long addr_src,
			unsigned long size)
{
	unsigned long AddrToWrite = addr_dest;
	AT91PS_DataFlash pFlash = &DataFlashInst;

	pFlash = AT91F_DataflashSelect (pFlash, &AddrToWrite);

	if (pFlash == 0)
		return ERR_UNKNOWN_FLASH_TYPE;

	if (size_dataflash(pFlash,addr_dest,size) == 0)
		return ERR_INVAL;

	if (prot_dataflash(pFlash,addr_dest) == 0)
		return ERR_PROTECTED;

	if (AddrToWrite == -1)
		return -1;

	return AT91F_DataFlashWrite (pFlash, (uchar *)addr_src,
						AddrToWrite, size);
}

void dataflash_perror (int err)
{
	switch (err) {
	case ERR_OK:
		break;
	case ERR_TIMOUT:
		printf("Timeout writing to DataFlash\n");
		break;
	case ERR_PROTECTED:
		printf("Can't write to protected/invalid DataFlash sectors\n");
		break;
	case ERR_INVAL:
		printf("Outside available DataFlash\n");
		break;
	case ERR_UNKNOWN_FLASH_TYPE:
		printf("Unknown Type of DataFlash\n");
		break;
	case ERR_PROG_ERROR:
		printf("General DataFlash Programming Error\n");
		break;
	default:
		printf("%s[%d] FIXME: rc=%d\n", __FILE__, __LINE__, err);
		break;
	}
}
Commit	Line	Data
2abbe075 WD	1	/* LowLevel function for ATMEL DataFlash support
	2	* Author : Hamid Ikdoumi (Atmel)
	3	*
	4	* This program is free software; you can redistribute it and/or
	5	* modify it under the terms of the GNU General Public License as
	6	* published by the Free Software Foundation; either version 2 of
	7	* the License, or (at your option) any later version.
	8	*
	9	* This program is distributed in the hope that it will be useful,
	10	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	11	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	12	* GNU General Public License for more details.
	13	*
	14	* You should have received a copy of the GNU General Public License
	15	* along with this program; if not, write to the Free Software
	16	* Foundation, Inc., 59 Temple Place, Suite 330, Boston,
	17	* MA 02111-1307 USA
	18	*
	19	*/
	20	#include <common.h>
	21	#include <config.h>
2abbe075 WD	22	#include <asm/hardware.h>
	23	#include <dataflash.h>
	24
2abbe075 WD	25	static AT91S_DataFlash DataFlashInst;
2abbe075 WD	26
2abbe075 WD	27	extern void AT91F_SpiInit (void);
	28	extern int AT91F_DataflashProbe (int i, AT91PS_DataflashDesc pDesc);
	29	extern int AT91F_DataFlashRead (AT91PS_DataFlash pDataFlash,
	30	unsigned long addr,
	31	unsigned long size, char *buffer);
5779d8d9	32	extern int AT91F_DataFlashWrite( AT91PS_DataFlash pDataFlash,
d4fc6012 PP	33	unsigned char *src,
	34	int dest,
	35	int size );
2abbe075 WD	36
	37	int AT91F_DataflashInit (void)
	38	{
	39	int i, j;
	40	int dfcode;
5604e217	41	int part;
d4fc6012	42	int last_part;
6d0f6bcf	43	int found[CONFIG_SYS_MAX_DATAFLASH_BANKS];
d4fc6012	44	unsigned char protected;
2abbe075 WD	45
	46	AT91F_SpiInit ();
	47
6d0f6bcf	48	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
d4fc6012	49	found[i] = 0;
5779d8d9	50	dataflash_info[i].Desc.state = IDLE;
2abbe075 WD	51	dataflash_info[i].id = 0;
2abbe075 WD	52	dataflash_info[i].Device.pages_number = 0;
d49fe4be	53	dfcode = AT91F_DataflashProbe (cs[i].cs,
d4fc6012	54	&dataflash_info[i].Desc);
2abbe075 WD	55
2abbe075 WD	56	switch (dfcode) {
c6457e3b SL	57	case AT45DB021:
c6457e3b SL	58	dataflash_info[i].Device.pages_number = 1024;
f77d92a3	59	dataflash_info[i].Device.pages_size = 264;
c6457e3b SL	60	dataflash_info[i].Device.page_offset = 9;
	61	dataflash_info[i].Device.byte_mask = 0x300;
	62	dataflash_info[i].Device.cs = cs[i].cs;
	63	dataflash_info[i].Desc.DataFlash_state = IDLE;
	64	dataflash_info[i].logical_address = cs[i].addr;
	65	dataflash_info[i].id = dfcode;
	66	found[i] += dfcode;;
	67	break;
f77d92a3 SL	68
	69	case AT45DB081:
	70	dataflash_info[i].Device.pages_number = 4096;
	71	dataflash_info[i].Device.pages_size = 264;
	72	dataflash_info[i].Device.page_offset = 9;
	73	dataflash_info[i].Device.byte_mask = 0x300;
	74	dataflash_info[i].Device.cs = cs[i].cs;
	75	dataflash_info[i].Desc.DataFlash_state = IDLE;
	76	dataflash_info[i].logical_address = cs[i].addr;
	77	dataflash_info[i].id = dfcode;
	78	found[i] += dfcode;;
	79	break;
	80
2abbe075 WD	81	case AT45DB161:
	82	dataflash_info[i].Device.pages_number = 4096;
	83	dataflash_info[i].Device.pages_size = 528;
	84	dataflash_info[i].Device.page_offset = 10;
	85	dataflash_info[i].Device.byte_mask = 0x300;
d49fe4be	86	dataflash_info[i].Device.cs = cs[i].cs;
2abbe075	87	dataflash_info[i].Desc.DataFlash_state = IDLE;
d49fe4be	88	dataflash_info[i].logical_address = cs[i].addr;
2abbe075	89	dataflash_info[i].id = dfcode;
d4fc6012	90	found[i] += dfcode;;
2abbe075 WD	91	break;
	92
	93	case AT45DB321:
	94	dataflash_info[i].Device.pages_number = 8192;
	95	dataflash_info[i].Device.pages_size = 528;
	96	dataflash_info[i].Device.page_offset = 10;
	97	dataflash_info[i].Device.byte_mask = 0x300;
d49fe4be	98	dataflash_info[i].Device.cs = cs[i].cs;
2abbe075	99	dataflash_info[i].Desc.DataFlash_state = IDLE;
d49fe4be	100	dataflash_info[i].logical_address = cs[i].addr;
2abbe075	101	dataflash_info[i].id = dfcode;
d4fc6012	102	found[i] += dfcode;;
2abbe075 WD	103	break;
	104
	105	case AT45DB642:
	106	dataflash_info[i].Device.pages_number = 8192;
	107	dataflash_info[i].Device.pages_size = 1056;
	108	dataflash_info[i].Device.page_offset = 11;
	109	dataflash_info[i].Device.byte_mask = 0x700;
d49fe4be	110	dataflash_info[i].Device.cs = cs[i].cs;
2abbe075	111	dataflash_info[i].Desc.DataFlash_state = IDLE;
d49fe4be	112	dataflash_info[i].logical_address = cs[i].addr;
2abbe075	113	dataflash_info[i].id = dfcode;
d4fc6012	114	found[i] += dfcode;;
2abbe075	115	break;
d4fc6012	116
5779d8d9 WD	117	case AT45DB128:
	118	dataflash_info[i].Device.pages_number = 16384;
	119	dataflash_info[i].Device.pages_size = 1056;
	120	dataflash_info[i].Device.page_offset = 11;
	121	dataflash_info[i].Device.byte_mask = 0x700;
d49fe4be	122	dataflash_info[i].Device.cs = cs[i].cs;
5779d8d9	123	dataflash_info[i].Desc.DataFlash_state = IDLE;
d49fe4be	124	dataflash_info[i].logical_address = cs[i].addr;
5779d8d9	125	dataflash_info[i].id = dfcode;
d4fc6012	126	found[i] += dfcode;;
5779d8d9	127	break;
2abbe075 WD	128
2abbe075 WD	129	default:
d4fc6012	130	dfcode = 0;
2abbe075 WD	131	break;
2abbe075 WD	132	}
5779d8d9 WD	133	/* set the last area end to the dataflash size*/
	134	area_list[NB_DATAFLASH_AREA -1].end =
	135	(dataflash_info[i].Device.pages_number *
	136	dataflash_info[i].Device.pages_size)-1;
	137
5604e217 SP	138	part = 0;
5604e217 SP	139	last_part = 0;
5779d8d9 WD	140	/* set the area addresses */
5779d8d9 WD	141	for(j = 0; j<NB_DATAFLASH_AREA; j++) {
d4fc6012	142	if(found[i]!=0) {
f01dbb54 WD	143	dataflash_info[i].Device.area_list[j].start =
f01dbb54 WD	144	area_list[part].start +
d4fc6012 PP	145	dataflash_info[i].logical_address;
d4fc6012 PP	146	if(area_list[part].end == 0xffffffff) {
f01dbb54 WD	147	dataflash_info[i].Device.area_list[j].end =
f01dbb54 WD	148	dataflash_info[i].end_address +
d4fc6012 PP	149	dataflash_info [i].logical_address;
	150	last_part = 1;
	151	} else {
f01dbb54 WD	152	dataflash_info[i].Device.area_list[j].end =
f01dbb54 WD	153	area_list[part].end +
d4fc6012 PP	154	dataflash_info[i].logical_address;
	155	}
	156	protected = area_list[part].protected;
	157	/* Set the environment according to the label...*/
	158	if(protected == FLAG_PROTECT_INVALID) {
f01dbb54	159	dataflash_info[i].Device.area_list[j].protected =
d4fc6012 PP	160	FLAG_PROTECT_INVALID;
d4fc6012 PP	161	} else {
f01dbb54	162	dataflash_info[i].Device.area_list[j].protected =
d4fc6012 PP	163	protected;
	164	}
	165	strcpy((char*)(dataflash_info[i].Device.area_list[j].label),
	166	(const char *)area_list[part].label);
	167	}
	168	part++;
5779d8d9	169	}
2abbe075	170	}
d4fc6012	171	return found[0];
2abbe075 WD	172	}
2abbe075 WD	173
c3a60cb3	174	void AT91F_DataflashSetEnv (void)
d4fc6012 PP	175	{
	176	int i, j;
	177	int part;
	178	unsigned char env;
	179	unsigned char s[32]; /* Will fit a long int in hex */
	180	unsigned long start;
5604e217	181
6d0f6bcf	182	for (i = 0, part= 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
d4fc6012 PP	183	for(j = 0; j<NB_DATAFLASH_AREA; j++) {
	184	env = area_list[part].setenv;
	185	/* Set the environment according to the label...*/
	186	if((env & FLAG_SETENV) == FLAG_SETENV) {
f01dbb54	187	start =
d4fc6012	188	dataflash_info[i].Device.area_list[j].start;
0a5676be	189	sprintf((char*) s,"%lX",start);
c3a60cb3	190	setenv((char) area_list[part].label,(char) s);
d4fc6012 PP	191	}
	192	part++;
	193	}
	194	}
	195	}
2abbe075	196
2abbe075 WD	197	void dataflash_print_info (void)
2abbe075 WD	198	{
5779d8d9	199	int i, j;
2abbe075	200
6d0f6bcf	201	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
2abbe075	202	if (dataflash_info[i].id != 0) {
d4fc6012	203	printf("DataFlash:");
2abbe075	204	switch (dataflash_info[i].id) {
c6457e3b SL	205	case AT45DB021:
	206	printf("AT45DB021\n");
	207	break;
2abbe075	208	case AT45DB161:
d4fc6012	209	printf("AT45DB161\n");
2abbe075 WD	210	break;
	211
	212	case AT45DB321:
d4fc6012	213	printf("AT45DB321\n");
2abbe075 WD	214	break;
	215
	216	case AT45DB642:
d4fc6012	217	printf("AT45DB642\n");
2abbe075	218	break;
5779d8d9	219	case AT45DB128:
d4fc6012	220	printf("AT45DB128\n");
5779d8d9	221	break;
2abbe075 WD	222	}
2abbe075 WD	223
d4fc6012	224	printf("Nb pages: %6d\n"
2abbe075 WD	225	"Page Size: %6d\n"
	226	"Size=%8d bytes\n"
	227	"Logical address: 0x%08X\n",
	228	(unsigned int) dataflash_info[i].Device.pages_number,
	229	(unsigned int) dataflash_info[i].Device.pages_size,
	230	(unsigned int) dataflash_info[i].Device.pages_number *
	231	dataflash_info[i].Device.pages_size,
	232	(unsigned int) dataflash_info[i].logical_address);
5779d8d9	233	for (j=0; j< NB_DATAFLASH_AREA; j++) {
d4fc6012 PP	234	switch(dataflash_info[i].Device.area_list[j].protected) {
	235	case FLAG_PROTECT_SET:
	236	case FLAG_PROTECT_CLEAR:
	237	printf("Area %i:\t%08lX to %08lX %s", j,
	238	dataflash_info[i].Device.area_list[j].start,
	239	dataflash_info[i].Device.area_list[j].end,
	240	(dataflash_info[i].Device.area_list[j].protected==FLAG_PROTECT_SET) ? "(RO)" : " ");
d4fc6012	241	printf(" %s\n", dataflash_info[i].Device.area_list[j].label);
d4fc6012	242	break;
d4fc6012 PP	243	case FLAG_PROTECT_INVALID:
d4fc6012 PP	244	break;
d4fc6012	245	}
5779d8d9	246	}
2abbe075 WD	247	}
	248	}
	249	}
	250
d4fc6012	251	/---------------------------------------------------------------------------/
5604e217	252	/* Function Name : AT91F_DataflashSelect */
d4fc6012 PP	253	/* Object : Select the correct device */
d4fc6012 PP	254	/---------------------------------------------------------------------------/
f01dbb54	255	AT91PS_DataFlash AT91F_DataflashSelect (AT91PS_DataFlash pFlash,
d4fc6012	256	unsigned long *addr)
2abbe075 WD	257	{
	258	char addr_valid = 0;
	259	int i;
	260
6d0f6bcf	261	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++)
d4fc6012	262	if ( dataflash_info[i].id
c77ce474	263	&& ((((int) *addr) & 0xFF000000) ==
d4fc6012	264	dataflash_info[i].logical_address)) {
2abbe075 WD	265	addr_valid = 1;
	266	break;
	267	}
	268	if (!addr_valid) {
	269	pFlash = (AT91PS_DataFlash) 0;
	270	return pFlash;
	271	}
	272	pFlash->pDataFlashDesc = &(dataflash_info[i].Desc);
	273	pFlash->pDevice = &(dataflash_info[i].Device);
	274	*addr -= dataflash_info[i].logical_address;
	275	return (pFlash);
	276	}
	277
d4fc6012	278	/---------------------------------------------------------------------------/
5604e217	279	/* Function Name : addr_dataflash */
d4fc6012 PP	280	/* Object : Test if address is valid */
d4fc6012 PP	281	/---------------------------------------------------------------------------/
2abbe075 WD	282	int addr_dataflash (unsigned long addr)
	283	{
	284	int addr_valid = 0;
	285	int i;
	286
6d0f6bcf	287	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
2abbe075 WD	288	if ((((int) addr) & 0xFF000000) ==
	289	dataflash_info[i].logical_address) {
	290	addr_valid = 1;
	291	break;
	292	}
	293	}
	294
	295	return addr_valid;
	296	}
5604e217	297
d4fc6012	298	/---------------------------------------------------------------------------/
5604e217	299	/* Function Name : size_dataflash */
d4fc6012 PP	300	/* Object : Test if address is valid regarding the size */
d4fc6012 PP	301	/---------------------------------------------------------------------------/
f01dbb54	302	int size_dataflash (AT91PS_DataFlash pdataFlash, unsigned long addr,
d4fc6012	303	unsigned long size)
5779d8d9 WD	304	{
	305	/* is outside the dataflash */
	306	if (((int)addr & 0x0FFFFFFF) > (pdataFlash->pDevice->pages_size *
	307	pdataFlash->pDevice->pages_number)) return 0;
	308	/* is too large for the dataflash */
	309	if (size > ((pdataFlash->pDevice->pages_size *
f01dbb54	310	pdataFlash->pDevice->pages_number) -
d4fc6012	311	((int)addr & 0x0FFFFFFF))) return 0;
5779d8d9 WD	312
	313	return 1;
	314	}
5604e217	315
d4fc6012	316	/---------------------------------------------------------------------------/
5604e217	317	/* Function Name : prot_dataflash */
d4fc6012 PP	318	/* Object : Test if destination area is protected */
d4fc6012 PP	319	/---------------------------------------------------------------------------/
5779d8d9 WD	320	int prot_dataflash (AT91PS_DataFlash pdataFlash, unsigned long addr)
5779d8d9 WD	321	{
5604e217 SP	322	int area;
5604e217 SP	323
5779d8d9 WD	324	/* find area */
	325	for (area=0; area < NB_DATAFLASH_AREA; area++) {
	326	if ((addr >= pdataFlash->pDevice->area_list[area].start) &&
	327	(addr < pdataFlash->pDevice->area_list[area].end))
	328	break;
	329	}
f01dbb54	330	if (area == NB_DATAFLASH_AREA)
d4fc6012 PP	331	return -1;
d4fc6012 PP	332
5779d8d9	333	/test protection value/
f01dbb54	334	if (pdataFlash->pDevice->area_list[area].protected == FLAG_PROTECT_SET)
d4fc6012	335	return 0;
f01dbb54	336	if (pdataFlash->pDevice->area_list[area].protected == FLAG_PROTECT_INVALID)
d4fc6012	337	return 0;
5779d8d9 WD	338
	339	return 1;
	340	}
5604e217	341
d4fc6012 PP	342	/--------------------------------------------------------------------------/
	343	/* Function Name : dataflash_real_protect */
	344	/* Object : protect/unprotect area */
	345	/--------------------------------------------------------------------------/
f01dbb54	346	int dataflash_real_protect (int flag, unsigned long start_addr,
d4fc6012	347	unsigned long end_addr)
5779d8d9	348	{
5604e217 SP	349	int i,j, area1, area2, addr_valid = 0;
5604e217 SP	350
5779d8d9	351	/* find dataflash */
6d0f6bcf	352	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
5779d8d9 WD	353	if ((((int) start_addr) & 0xF0000000) ==
	354	dataflash_info[i].logical_address) {
	355	addr_valid = 1;
	356	break;
	357	}
	358	}
	359	if (!addr_valid) {
	360	return -1;
	361	}
	362	/* find start area */
	363	for (area1=0; area1 < NB_DATAFLASH_AREA; area1++) {
f01dbb54	364	if (start_addr == dataflash_info[i].Device.area_list[area1].start)
d4fc6012	365	break;
5779d8d9 WD	366	}
	367	if (area1 == NB_DATAFLASH_AREA) return -1;
	368	/* find end area */
	369	for (area2=0; area2 < NB_DATAFLASH_AREA; area2++) {
f01dbb54	370	if (end_addr == dataflash_info[i].Device.area_list[area2].end)
d4fc6012	371	break;
5779d8d9	372	}
f01dbb54	373	if (area2 == NB_DATAFLASH_AREA)
d4fc6012	374	return -1;
5779d8d9 WD	375
	376	/set protection value/
	377	for(j = area1; j < area2+1 ; j++)
f01dbb54	378	if(dataflash_info[i].Device.area_list[j].protected
d4fc6012 PP	379	!= FLAG_PROTECT_INVALID) {
d4fc6012 PP	380	if (flag == 0) {
f01dbb54	381	dataflash_info[i].Device.area_list[j].protected
d4fc6012 PP	382	= FLAG_PROTECT_CLEAR;
d4fc6012 PP	383	} else {
f01dbb54	384	dataflash_info[i].Device.area_list[j].protected
d4fc6012 PP	385	= FLAG_PROTECT_SET;
	386	}
	387	}
5779d8d9 WD	388
	389	return (area2-area1+1);
	390	}
2abbe075	391
d4fc6012	392	/---------------------------------------------------------------------------/
5604e217	393	/* Function Name : read_dataflash */
d4fc6012 PP	394	/* Object : dataflash memory read */
d4fc6012 PP	395	/---------------------------------------------------------------------------/
2abbe075 WD	396	int read_dataflash (unsigned long addr, unsigned long size, char *result)
2abbe075 WD	397	{
d52fb7e3	398	unsigned long AddrToRead = addr;
2abbe075 WD	399	AT91PS_DataFlash pFlash = &DataFlashInst;
	400
	401	pFlash = AT91F_DataflashSelect (pFlash, &AddrToRead);
5779d8d9	402
2abbe075	403	if (pFlash == 0)
5779d8d9 WD	404	return ERR_UNKNOWN_FLASH_TYPE;
	405
	406	if (size_dataflash(pFlash,addr,size) == 0)
	407	return ERR_INVAL;
2abbe075 WD	408
	409	return (AT91F_DataFlashRead (pFlash, AddrToRead, size, result));
	410	}
	411
d4fc6012	412	/---------------------------------------------------------------------------/
5604e217	413	/* Function Name : write_dataflash */
d4fc6012 PP	414	/* Object : write a block in dataflash */
d4fc6012 PP	415	/---------------------------------------------------------------------------/
2abbe075	416	int write_dataflash (unsigned long addr_dest, unsigned long addr_src,
d4fc6012	417	unsigned long size)
2abbe075	418	{
d52fb7e3	419	unsigned long AddrToWrite = addr_dest;
2abbe075 WD	420	AT91PS_DataFlash pFlash = &DataFlashInst;
	421
	422	pFlash = AT91F_DataflashSelect (pFlash, &AddrToWrite);
5779d8d9 WD	423
	424	if (pFlash == 0)
	425	return ERR_UNKNOWN_FLASH_TYPE;
	426
	427	if (size_dataflash(pFlash,addr_dest,size) == 0)
	428	return ERR_INVAL;
	429
	430	if (prot_dataflash(pFlash,addr_dest) == 0)
	431	return ERR_PROTECTED;
	432
2abbe075 WD	433	if (AddrToWrite == -1)
	434	return -1;
	435
f01dbb54	436	return AT91F_DataFlashWrite (pFlash, (uchar *)addr_src,
d4fc6012	437	AddrToWrite, size);
2abbe075 WD	438	}
2abbe075 WD	439
2abbe075 WD	440	void dataflash_perror (int err)
	441	{
	442	switch (err) {
	443	case ERR_OK:
	444	break;
	445	case ERR_TIMOUT:
d4fc6012	446	printf("Timeout writing to DataFlash\n");
2abbe075 WD	447	break;
2abbe075 WD	448	case ERR_PROTECTED:
d4fc6012	449	printf("Can't write to protected/invalid DataFlash sectors\n");
2abbe075 WD	450	break;
2abbe075 WD	451	case ERR_INVAL:
d4fc6012	452	printf("Outside available DataFlash\n");
2abbe075 WD	453	break;
2abbe075 WD	454	case ERR_UNKNOWN_FLASH_TYPE:
d4fc6012	455	printf("Unknown Type of DataFlash\n");
2abbe075 WD	456	break;
2abbe075 WD	457	case ERR_PROG_ERROR:
d4fc6012	458	printf("General DataFlash Programming Error\n");
2abbe075 WD	459	break;
2abbe075 WD	460	default:
d4fc6012	461	printf("%s[%d] FIXME: rc=%d\n", __FILE__, __LINE__, err);
2abbe075 WD	462	break;
	463	}
	464	}