[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_Dataflashenv_set(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);
				env_set((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;
51855e89	52	found[i] += dfcode;
c6457e3b	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;
51855e89	64	found[i] += dfcode;
f77d92a3 SL	65	break;
f77d92a3 SL	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;
51855e89	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;
51855e89	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;
51855e89	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;
51855e89	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
382bee57	158	void AT91F_Dataflashenv_set(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;
382bee57 SG	172	sprintf((char *)s, "%lX", start);
	173	env_set((char *)area_list[part].label,
	174	(char *)s);
d4fc6012 PP	175	}
	176	part++;
	177	}
	178	}
	179	}
2abbe075	180
2abbe075 WD	181	void dataflash_print_info (void)
2abbe075 WD	182	{
5779d8d9	183	int i, j;
2abbe075	184
6d0f6bcf	185	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
2abbe075	186	if (dataflash_info[i].id != 0) {
d4fc6012	187	printf("DataFlash:");
2abbe075	188	switch (dataflash_info[i].id) {
c6457e3b SL	189	case AT45DB021:
	190	printf("AT45DB021\n");
	191	break;
2abbe075	192	case AT45DB161:
d4fc6012	193	printf("AT45DB161\n");
2abbe075 WD	194	break;
	195
	196	case AT45DB321:
d4fc6012	197	printf("AT45DB321\n");
2abbe075 WD	198	break;
	199
	200	case AT45DB642:
d4fc6012	201	printf("AT45DB642\n");
2abbe075	202	break;
5779d8d9	203	case AT45DB128:
d4fc6012	204	printf("AT45DB128\n");
5779d8d9	205	break;
2abbe075 WD	206	}
2abbe075 WD	207
d4fc6012	208	printf("Nb pages: %6d\n"
2abbe075 WD	209	"Page Size: %6d\n"
	210	"Size=%8d bytes\n"
	211	"Logical address: 0x%08X\n",
	212	(unsigned int) dataflash_info[i].Device.pages_number,
	213	(unsigned int) dataflash_info[i].Device.pages_size,
	214	(unsigned int) dataflash_info[i].Device.pages_number *
	215	dataflash_info[i].Device.pages_size,
	216	(unsigned int) dataflash_info[i].logical_address);
03f79779	217	for (j = 0; j < NB_DATAFLASH_AREA; j++) {
d4fc6012 PP	218	switch(dataflash_info[i].Device.area_list[j].protected) {
	219	case FLAG_PROTECT_SET:
	220	case FLAG_PROTECT_CLEAR:
	221	printf("Area %i:\t%08lX to %08lX %s", j,
	222	dataflash_info[i].Device.area_list[j].start,
	223	dataflash_info[i].Device.area_list[j].end,
	224	(dataflash_info[i].Device.area_list[j].protected==FLAG_PROTECT_SET) ? "(RO)" : " ");
d4fc6012	225	printf(" %s\n", dataflash_info[i].Device.area_list[j].label);
d4fc6012	226	break;
d4fc6012 PP	227	case FLAG_PROTECT_INVALID:
d4fc6012 PP	228	break;
d4fc6012	229	}
5779d8d9	230	}
2abbe075 WD	231	}
	232	}
	233	}
	234
d4fc6012	235	/---------------------------------------------------------------------------/
5604e217	236	/* Function Name : AT91F_DataflashSelect */
d4fc6012 PP	237	/* Object : Select the correct device */
d4fc6012 PP	238	/---------------------------------------------------------------------------/
f01dbb54	239	AT91PS_DataFlash AT91F_DataflashSelect (AT91PS_DataFlash pFlash,
d4fc6012	240	unsigned long *addr)
2abbe075 WD	241	{
	242	char addr_valid = 0;
	243	int i;
	244
6d0f6bcf	245	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++)
d4fc6012	246	if ( dataflash_info[i].id
c77ce474	247	&& ((((int) *addr) & 0xFF000000) ==
d4fc6012	248	dataflash_info[i].logical_address)) {
2abbe075 WD	249	addr_valid = 1;
	250	break;
	251	}
	252	if (!addr_valid) {
	253	pFlash = (AT91PS_DataFlash) 0;
	254	return pFlash;
	255	}
	256	pFlash->pDataFlashDesc = &(dataflash_info[i].Desc);
	257	pFlash->pDevice = &(dataflash_info[i].Device);
	258	*addr -= dataflash_info[i].logical_address;
	259	return (pFlash);
	260	}
	261
d4fc6012	262	/---------------------------------------------------------------------------/
5604e217	263	/* Function Name : addr_dataflash */
d4fc6012 PP	264	/* Object : Test if address is valid */
d4fc6012 PP	265	/---------------------------------------------------------------------------/
2abbe075 WD	266	int addr_dataflash (unsigned long addr)
	267	{
	268	int addr_valid = 0;
	269	int i;
	270
6d0f6bcf	271	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
2abbe075 WD	272	if ((((int) addr) & 0xFF000000) ==
	273	dataflash_info[i].logical_address) {
	274	addr_valid = 1;
	275	break;
	276	}
	277	}
	278
	279	return addr_valid;
	280	}
5604e217	281
d4fc6012	282	/---------------------------------------------------------------------------/
5604e217	283	/* Function Name : size_dataflash */
d4fc6012 PP	284	/* Object : Test if address is valid regarding the size */
d4fc6012 PP	285	/---------------------------------------------------------------------------/
f01dbb54	286	int size_dataflash (AT91PS_DataFlash pdataFlash, unsigned long addr,
d4fc6012	287	unsigned long size)
5779d8d9 WD	288	{
	289	/* is outside the dataflash */
	290	if (((int)addr & 0x0FFFFFFF) > (pdataFlash->pDevice->pages_size *
	291	pdataFlash->pDevice->pages_number)) return 0;
	292	/* is too large for the dataflash */
	293	if (size > ((pdataFlash->pDevice->pages_size *
f01dbb54	294	pdataFlash->pDevice->pages_number) -
d4fc6012	295	((int)addr & 0x0FFFFFFF))) return 0;
5779d8d9 WD	296
	297	return 1;
	298	}
5604e217	299
d4fc6012	300	/---------------------------------------------------------------------------/
5604e217	301	/* Function Name : prot_dataflash */
d4fc6012 PP	302	/* Object : Test if destination area is protected */
d4fc6012 PP	303	/---------------------------------------------------------------------------/
5779d8d9 WD	304	int prot_dataflash (AT91PS_DataFlash pdataFlash, unsigned long addr)
5779d8d9 WD	305	{
5604e217 SP	306	int area;
5604e217 SP	307
5779d8d9	308	/* find area */
03f79779	309	for (area = 0; area < NB_DATAFLASH_AREA; area++) {
5779d8d9 WD	310	if ((addr >= pdataFlash->pDevice->area_list[area].start) &&
	311	(addr < pdataFlash->pDevice->area_list[area].end))
	312	break;
	313	}
f01dbb54	314	if (area == NB_DATAFLASH_AREA)
d4fc6012 PP	315	return -1;
d4fc6012 PP	316
5779d8d9	317	/test protection value/
f01dbb54	318	if (pdataFlash->pDevice->area_list[area].protected == FLAG_PROTECT_SET)
d4fc6012	319	return 0;
f01dbb54	320	if (pdataFlash->pDevice->area_list[area].protected == FLAG_PROTECT_INVALID)
d4fc6012	321	return 0;
5779d8d9 WD	322
	323	return 1;
	324	}
5604e217	325
d4fc6012 PP	326	/--------------------------------------------------------------------------/
	327	/* Function Name : dataflash_real_protect */
	328	/* Object : protect/unprotect area */
	329	/--------------------------------------------------------------------------/
f01dbb54	330	int dataflash_real_protect (int flag, unsigned long start_addr,
d4fc6012	331	unsigned long end_addr)
5779d8d9	332	{
5604e217 SP	333	int i,j, area1, area2, addr_valid = 0;
5604e217 SP	334
5779d8d9	335	/* find dataflash */
6d0f6bcf	336	for (i = 0; i < CONFIG_SYS_MAX_DATAFLASH_BANKS; i++) {
5779d8d9 WD	337	if ((((int) start_addr) & 0xF0000000) ==
	338	dataflash_info[i].logical_address) {
	339	addr_valid = 1;
	340	break;
	341	}
	342	}
	343	if (!addr_valid) {
	344	return -1;
	345	}
	346	/* find start area */
03f79779	347	for (area1 = 0; area1 < NB_DATAFLASH_AREA; area1++) {
f01dbb54	348	if (start_addr == dataflash_info[i].Device.area_list[area1].start)
d4fc6012	349	break;
5779d8d9 WD	350	}
	351	if (area1 == NB_DATAFLASH_AREA) return -1;
	352	/* find end area */
03f79779	353	for (area2 = 0; area2 < NB_DATAFLASH_AREA; area2++) {
f01dbb54	354	if (end_addr == dataflash_info[i].Device.area_list[area2].end)
d4fc6012	355	break;
5779d8d9	356	}
f01dbb54	357	if (area2 == NB_DATAFLASH_AREA)
d4fc6012	358	return -1;
5779d8d9 WD	359
5779d8d9 WD	360	/set protection value/
03f79779	361	for(j = area1; j < area2 + 1 ; j++)
f01dbb54	362	if(dataflash_info[i].Device.area_list[j].protected
d4fc6012 PP	363	!= FLAG_PROTECT_INVALID) {
d4fc6012 PP	364	if (flag == 0) {
f01dbb54	365	dataflash_info[i].Device.area_list[j].protected
d4fc6012 PP	366	= FLAG_PROTECT_CLEAR;
d4fc6012 PP	367	} else {
f01dbb54	368	dataflash_info[i].Device.area_list[j].protected
d4fc6012 PP	369	= FLAG_PROTECT_SET;
	370	}
	371	}
5779d8d9	372
03f79779	373	return (area2 - area1 + 1);
5779d8d9	374	}
2abbe075	375
d4fc6012	376	/---------------------------------------------------------------------------/
5604e217	377	/* Function Name : read_dataflash */
d4fc6012 PP	378	/* Object : dataflash memory read */
d4fc6012 PP	379	/---------------------------------------------------------------------------/
2abbe075 WD	380	int read_dataflash (unsigned long addr, unsigned long size, char *result)
2abbe075 WD	381	{
d52fb7e3	382	unsigned long AddrToRead = addr;
2abbe075 WD	383	AT91PS_DataFlash pFlash = &DataFlashInst;
	384
	385	pFlash = AT91F_DataflashSelect (pFlash, &AddrToRead);
5779d8d9	386
2abbe075	387	if (pFlash == 0)
5779d8d9 WD	388	return ERR_UNKNOWN_FLASH_TYPE;
	389
	390	if (size_dataflash(pFlash,addr,size) == 0)
	391	return ERR_INVAL;
2abbe075 WD	392
	393	return (AT91F_DataFlashRead (pFlash, AddrToRead, size, result));
	394	}
	395
d4fc6012	396	/---------------------------------------------------------------------------/
5604e217	397	/* Function Name : write_dataflash */
d4fc6012 PP	398	/* Object : write a block in dataflash */
d4fc6012 PP	399	/---------------------------------------------------------------------------/
2abbe075	400	int write_dataflash (unsigned long addr_dest, unsigned long addr_src,
d4fc6012	401	unsigned long size)
2abbe075	402	{
d52fb7e3	403	unsigned long AddrToWrite = addr_dest;
2abbe075 WD	404	AT91PS_DataFlash pFlash = &DataFlashInst;
	405
	406	pFlash = AT91F_DataflashSelect (pFlash, &AddrToWrite);
5779d8d9 WD	407
	408	if (pFlash == 0)
	409	return ERR_UNKNOWN_FLASH_TYPE;
	410
	411	if (size_dataflash(pFlash,addr_dest,size) == 0)
	412	return ERR_INVAL;
	413
	414	if (prot_dataflash(pFlash,addr_dest) == 0)
	415	return ERR_PROTECTED;
	416
2abbe075 WD	417	if (AddrToWrite == -1)
	418	return -1;
	419
f01dbb54	420	return AT91F_DataFlashWrite (pFlash, (uchar *)addr_src,
d4fc6012	421	AddrToWrite, size);
2abbe075 WD	422	}
2abbe075 WD	423
2abbe075 WD	424	void dataflash_perror (int err)
	425	{
	426	switch (err) {
	427	case ERR_OK:
	428	break;
	429	case ERR_TIMOUT:
d4fc6012	430	printf("Timeout writing to DataFlash\n");
2abbe075 WD	431	break;
2abbe075 WD	432	case ERR_PROTECTED:
d4fc6012	433	printf("Can't write to protected/invalid DataFlash sectors\n");
2abbe075 WD	434	break;
2abbe075 WD	435	case ERR_INVAL:
d4fc6012	436	printf("Outside available DataFlash\n");
2abbe075 WD	437	break;
2abbe075 WD	438	case ERR_UNKNOWN_FLASH_TYPE:
d4fc6012	439	printf("Unknown Type of DataFlash\n");
2abbe075 WD	440	break;
2abbe075 WD	441	case ERR_PROG_ERROR:
d4fc6012	442	printf("General DataFlash Programming Error\n");
2abbe075 WD	443	break;
2abbe075 WD	444	default:
d4fc6012	445	printf("%s[%d] FIXME: rc=%d\n", __FILE__, __LINE__, err);
2abbe075 WD	446	break;
	447	}
	448	}