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

/*
 * LowLevel function for ATMEL DataFlash support
 * Author : Hamid Ikdoumi (Atmel)
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */
#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 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*/
		dataflash_info[i].end_address =
				(dataflash_info[i].Device.pages_number *
				dataflash_info[i].Device.pages_size) - 1;

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