flash.c

U-Boot1.1.2是最为常用的嵌入式系统Bootloader

		goto done;
	}

	if (len == 0) {
		rv = 1;
		goto done;
	}

	if (dev->bank == FLASH0_BANK) {
		addr = pos;
		words = len;

		while (words--) {
			if (FLASH0_READ (dev, addr) != 0xff) {
				rv = 0;
				goto done;
			}
			addr++;
		}
	}

	rv = 1;

  done:
	PRINTF ("flashWrite: rv=%d\n", rv);
	return rv;
}


/*
 * NOTE: the below code cannot run from FLASH!!!
 */
/***********************************************************************
 *
 * Flash Diagnostics
 *
 ***********************************************************************/

STATUS flashDiag (flash_dev_t * dev)
{
	unsigned int *buf = 0;
	int i, len, sector;
	int rv = ERROR;

	if (flashCheck (dev) == ERROR)
		return ERROR;

	printf ("flashDiag: Testing device %d, "
		"base: 0x%x, %d sectors @ %d kB = %d kB\n",
		DEV_NO (dev), dev->base,
		dev->sectors,
		1 << (dev->lgSectorSize - 10),
		dev->sectors << (dev->lgSectorSize - 10));

	len = 1 << dev->lgSectorSize;

	printf ("flashDiag: Erasing\n");

	if (flashErase (dev) == ERROR) {
		printf ("flashDiag: Erase failed\n");
		goto done;
	}
	printf ("%d bytes requested ...\n", len);
	buf = malloc (len);
	printf ("allocated %d bytes ...\n", len);
	if (buf == 0) {
		printf ("flashDiag: Out of memory\n");
		goto done;
	}

	/*
	 * Write unique counting pattern to each sector
	 */

	for (sector = 0; sector < dev->sectors; sector++) {
		printf ("flashDiag: Write sector %d\n", sector);

		for (i = 0; i < len / 4; i++)
			buf[i] = sector << 24 | i;

		if (flashWrite (dev,
				sector << dev->lgSectorSize,
				(char *) buf, len) == ERROR) {
			printf ("flashDiag: Write failed (dev: %d, sector: %d)\n",
				DEV_NO (dev), sector);
			goto done;
		}
	}

	/*
	 * Verify
	 */

	for (sector = 0; sector < dev->sectors; sector++) {
		printf ("flashDiag: Verify sector %d\n", sector);

		if (flashRead (dev,
				   sector << dev->lgSectorSize,
				   (char *) buf, len) == ERROR) {
			printf ("flashDiag: Read failed (dev: %d, sector: %d)\n",
				DEV_NO (dev), sector);
			goto done;
		}

		for (i = 0; i < len / 4; i++) {
			if (buf[i] != (sector << 24 | i)) {
				printf ("flashDiag: Verify error "
					"(dev: %d, sector: %d, offset: 0x%x)\n",
					DEV_NO (dev), sector, i);
				printf ("flashDiag: Expected 0x%08x, got 0x%08x\n",
					sector << 24 | i, buf[i]);

				goto done;
			}
		}
	}

	printf ("flashDiag: Erasing\n");

	if (flashErase (dev) == ERROR) {
		printf ("flashDiag: Final erase failed\n");
		goto done;
	}

	rv = OK;

  done:
	if (buf)
		free (buf);

	if (rv == OK)
		printf ("flashDiag: Device %d passed\n", DEV_NO (dev));
	else
		printf ("flashDiag: Device %d failed\n", DEV_NO (dev));

	return rv;
}

STATUS flashDiagAll (void)
{
	int i;
	int rv = OK;

	PRINTF ("flashDiagAll: devices=%d\n", flashDevCount);

	for (i = 0; i < flashDevCount; i++) {
		flash_dev_t *dev = &flashDev[i];

		if (dev->found && flashDiag (dev) == ERROR)
			rv = ERROR;
	}

	if (rv == OK)
		printf ("flashDiagAll: Passed\n");
	else
		printf ("flashDiagAll: Failed because of earlier errors\n");

	return OK;
}


/*-----------------------------------------------------------------------
 */
unsigned long flash_init (void)
{
	unsigned long size = 0;
	flash_dev_t *dev = NULL;

	flashLibInit ();

	/*
	 * Provide info for FLASH (up to 960K) of Kernel Image data.
	 */
	dev = FLASH_DEV_BANK0_LOW;
	flash_info[FLASH_BANK_KERNEL].flash_id =
			(dev->vendorID << 16) | dev->deviceID;
	flash_info[FLASH_BANK_KERNEL].sector_count = dev->sectors;
	flash_info[FLASH_BANK_KERNEL].size =
			flash_info[FLASH_BANK_KERNEL].sector_count * FLASH_SECTOR_SIZE;
	flash_info[FLASH_BANK_KERNEL].start[FIRST_SECTOR] = dev->base;
	size += flash_info[FLASH_BANK_KERNEL].size;

	/*
	 * Provide info for 512K PLCC FLASH ROM (U-Boot)
	 */
	dev = FLASH_DEV_BANK0_BOOT;
	flash_info[FLASH_BANK_BOOT].flash_id =
			(dev->vendorID << 16) | dev->deviceID;
	flash_info[FLASH_BANK_BOOT].sector_count = dev->sectors;
	flash_info[FLASH_BANK_BOOT].size =
			flash_info[FLASH_BANK_BOOT].sector_count * FLASH_SECTOR_SIZE;
	flash_info[FLASH_BANK_BOOT].start[FIRST_SECTOR] = dev->base;
	size += flash_info[FLASH_BANK_BOOT].size;


	/*
	 * Provide info for 512K FLASH0 segment (U-Boot)
	 */
	dev = FLASH_DEV_BANK0_HIGH;
	flash_info[FLASH_BANK_AUX].flash_id =
			(dev->vendorID << 16) | dev->deviceID;
	flash_info[FLASH_BANK_AUX].sector_count = dev->sectors;
	flash_info[FLASH_BANK_AUX].size =
			flash_info[FLASH_BANK_AUX].sector_count * FLASH_SECTOR_SIZE;
	flash_info[FLASH_BANK_AUX].start[FIRST_SECTOR] = dev->base;
	size += flash_info[FLASH_BANK_AUX].size;


	return size;
}

/*
 * Get flash device from U-Boot flash info.
 */
flash_dev_t *getFlashDevFromInfo (flash_info_t * info)
{
	int i;

	if (!info)
		return NULL;

	for (i = 0; i < flashDevCount; i++) {
		flash_dev_t *dev = &flashDev[i];

		if (dev->found && (dev->base == info->start[0]))
			return dev;
	}
	printf ("ERROR: notice, no FLASH mapped at address 0x%x\n",
			(unsigned int) info->start[0]);
	return NULL;
}

ulong flash_get_size (vu_long * addr, flash_info_t * info)
{
	int i;

	for (i = 0; i < flashDevCount; i++) {
		flash_dev_t *dev = &flashDev[i];

		if (dev->found) {
			if (dev->base == (unsigned int) addr) {
				info->flash_id = (dev->vendorID << 16) | dev->deviceID;
				info->sector_count = dev->sectors;
				info->size = info->sector_count * FLASH_SECTOR_SIZE;
				return dev->sectors * FLASH_SECTOR_SIZE;
			}
		}
	}
	return 0;
}

void flash_print_info (flash_info_t * info)
{
	int i;
	unsigned int chip;

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

	switch ((info->flash_id >> 16) & 0xff) {
	case 0x1:
		printf ("AMD ");
		break;
	default:
		printf ("Unknown Vendor ");
		break;
	}
	chip = (unsigned int) info->flash_id & 0x000000ff;

	switch (chip) {

	case AMD_ID_F040B:
		printf ("AM29F040B (4 Mbit)\n");
		break;

	case AMD_ID_LV160B:
	case FLASH_AM160LV:
	case 0x49:
		printf ("AM29LV160B (16 Mbit / 2M x 8bit)\n");
		break;

	default:
		printf ("Unknown Chip Type:0x%x\n", chip);
		break;
	}

	printf ("  Size: %ld bytes in %d Sectors\n",
		info->size, 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[FIRST_SECTOR] + i * FLASH_SECTOR_SIZE,
			info->protect[i] ? " (RO)" : "     ");
	}
	printf ("\n");
}


/*
 * Erase a range of flash sectors.
 */
int flash_erase (flash_info_t * info, int s_first, int s_last)
{
	vu_long *addr = (vu_long *) (info->start[0]);
	int prot, sect, l_sect;
	flash_dev_t *dev = NULL;

	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;
	}

	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");
	}

	l_sect = -1;

	/* Start erase on unprotected sectors */
	dev = getFlashDevFromInfo (info);
	if (dev) {
		printf ("Erase FLASH[%s] -%d sectors:", dev->name, dev->sectors);
		for (sect = s_first; sect <= s_last; sect++) {
			if (info->protect[sect] == 0) {	/* not protected */
				addr = (vu_long *) (dev->base);
				/*   printf("erase_sector: sector=%d, addr=0x%x\n",
				   sect, addr); */
				printf (".");
				if (ERROR == flashEraseSector (dev, sect)) {
					printf ("ERROR: could not erase sector %d on FLASH[%s]\n", sect, dev->name);
					return 1;
				}
			}
		}
	}
	printf (" done\n");
	return 0;
}

/*-----------------------------------------------------------------------
 * 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)
{

	flash_dev_t *dev = getFlashDevFromInfo (info);
	int addr = dest - info->start[0];

	if (!dev)
		return 1;

	if (OK != flashWrite (dev, addr, (char *) &data, sizeof (ulong))) {
		printf ("ERROR: could not write to addr=0x%x, data=0x%x\n",
			(unsigned int) addr, (unsigned) data);
		return 1;
	}

	if ((addr % FLASH_SECTOR_SIZE) == 0)
		printf (".");


	PRINTF ("write_word:0x%x, base=0x%x, addr=0x%x, data=0x%x\n",
		(unsigned) info->start[0],
		(unsigned) dest,
		(unsigned) (dest - info->start[0]), (unsigned) data);

	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;
	flash_dev_t *dev = getFlashDevFromInfo (info);

	if (dev) {
		printf ("FLASH[%s]:", dev->name);
		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));
	}
	return 1;
}

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