/* * (C) Copyright 2000 * Wolfgang Denk, DENX Software Engineering, wd@denx.de. * * See file CREDITS for list of people who contributed to this * project. * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License as * published by the Free Software Foundation; either version 2 of * the License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, * MA 02111-1307 USA *//* * FLASH support */#include <common.h>#include <command.h>#include <flash.h>#if (CONFIG_COMMANDS & CFG_CMD_FLASH)extern flash_info_t flash_info[];	/* info for FLASH chips *//* * The user interface starts numbering for Flash banks with 1 * for historical reasons. *//* * this routine looks for an abbreviated flash range specification. * the syntax is B:SF[-SL], where B is the bank number, SF is the first * sector to erase, and SL is the last sector to erase (defaults to SF). * bank numbers start at 1 to be consistent with other specs, sector numbers * start at zero. * * returns:	1	- correct spec; *pinfo, *psf and *psl are *			  set appropriately *		0	- doesn't look like an abbreviated spec *		-1	- looks like an abbreviated spec, but got *			  a parsing error, a number out of range, *			  or an invalid flash bank. */static intabbrev_spec(char *str, flash_info_t **pinfo, int *psf, int *psl){    flash_info_t *fp;    int bank, first, last;    char *p, *ep;    if ((p = strchr(str, ':')) == NULL)	return 0;    *p++ = '\0';    bank = simple_strtoul(str, &ep, 10);    if (ep == str || *ep != '\0' ||      bank < 1 || bank > CFG_MAX_FLASH_BANKS ||      (fp = &flash_info[bank - 1])->flash_id == FLASH_UNKNOWN)	return -1;    str = p;    if ((p = strchr(str, '-')) != NULL)	*p++ = '\0';    first = simple_strtoul(str, &ep, 10);    if (ep == str || *ep != '\0' || first >= fp->sector_count)	return -1;    if (p != NULL) {	last = simple_strtoul(p, &ep, 10);	if (ep == p || *ep != '\0' ||	  last < first || last >= fp->sector_count)	    return -1;    }    else	last = first;    *pinfo = fp;    *psf = first;    *psl = last;    return 1;}int do_flinfo (cmd_tbl_t *cmdtp, bd_t *bd, int flag, int argc, char *argv[]){	ulong bank;	if (argc == 1) {	/* print info for all FLASH banks */		for (bank=0; bank <CFG_MAX_FLASH_BANKS; ++bank) {			printf ("\nBank # %ld: ", bank+1);			flash_print_info (&flash_info[bank]);		}		return 0;	}	bank = simple_strtoul(argv[1], NULL, 16);	if ((bank < 1) || (bank > CFG_MAX_FLASH_BANKS)) {		printf ("Only FLASH Banks # 1 ... # %d supported\n",			CFG_MAX_FLASH_BANKS);		return 1;	}	printf ("\nBank # %ld: ", bank);	flash_print_info (&flash_info[bank-1]);	return 0;}int do_flerase(cmd_tbl_t *cmdtp, bd_t *bd, int flag, int argc, char *argv[]){	flash_info_t *info;	ulong bank, addr_first, addr_last;	int n, sect_first, sect_last;	int rcode = 0;	if (argc < 2) {		printf ("Usage:\n%s\n", cmdtp->usage);		return 1;	}	if (strcmp(argv[1], "all") == 0) {		for (bank=1; bank<=CFG_MAX_FLASH_BANKS; ++bank) {			printf ("Erase Flash Bank # %ld ", bank);			info = &flash_info[bank-1];			rcode = flash_erase (info, 0, info->sector_count-1);		}		return rcode;	}	if ((n = abbrev_spec(argv[1], &info, &sect_first, &sect_last)) != 0) {		if (n < 0) {			printf("Bad sector specification\n");			return 1;		}		printf ("Erase Flash Sectors %d-%d in Bank # %d ",			sect_first, sect_last, (info-flash_info)+1);		rcode = flash_erase(info, sect_first, sect_last);		return rcode;	}	if (argc != 3) {		printf ("Usage:\n%s\n", cmdtp->usage);		return 1;	}	if (strcmp(argv[1], "bank") == 0) {		bank = simple_strtoul(argv[2], NULL, 16);		if ((bank < 1) || (bank > CFG_MAX_FLASH_BANKS)) {			printf ("Only FLASH Banks # 1 ... # %d supported\n",				CFG_MAX_FLASH_BANKS);			return 1;		}		printf ("Erase Flash Bank # %ld ", bank);		info = &flash_info[bank-1];		rcode = flash_erase (info, 0, info->sector_count-1);		return rcode;	}	addr_first = simple_strtoul(argv[1], NULL, 16);	addr_last  = simple_strtoul(argv[2], NULL, 16);	if (addr_first >= addr_last) {		printf ("Usage:\n%s\n", cmdtp->usage);		return 1;	}	printf ("Erase Flash from 0x%08lx to 0x%08lx ", addr_first, addr_last);	rcode = flash_sect_erase(addr_first, addr_last);	return rcode;}int flash_sect_erase (ulong addr_first, ulong addr_last){	flash_info_t *info;	ulong bank;	int s_first, s_last;	int erased;	int rcode = 0;	erased = 0;	for (bank=0,info=&flash_info[0]; bank < CFG_MAX_FLASH_BANKS; ++bank, ++info) {		ulong b_end;		int sect;		if (info->flash_id == FLASH_UNKNOWN) {			continue;		}		b_end = info->start[0] + info->size - 1; /* bank end addr */		s_first = -1;		/* first sector to erase	*/		s_last  = -1;		/* last  sector to erase	*/		for (sect=0; sect < info->sector_count; ++sect) {			ulong end;		/* last address in current sect	*/			short s_end;			s_end = info->sector_count - 1;			end = (sect == s_end) ? b_end : info->start[sect + 1] - 1;			if (addr_first > end)				continue;			if (addr_last < info->start[sect])				continue;			if (addr_first == info->start[sect]) {				s_first = sect;			}			if (addr_last  == end) {				s_last  = sect;			}		}		if (s_first>=0 && s_first<=s_last) {			erased += s_last - s_first + 1;			rcode = flash_erase (info, s_first, s_last);		}	}	if (erased) {	    /*	printf ("Erased %d sectors\n", erased); */	} else {		printf ("Error: start and/or end address"			" not on sector boundary\n");		rcode = 1;	}	return rcode;}int do_protect(cmd_tbl_t *cmdtp, bd_t *bd, int flag, int argc, char *argv[]){	flash_info_t *info;	ulong bank, addr_first, addr_last;	int i, p, n, sect_first, sect_last;	int rcode = 0;	if (argc < 3) {		printf ("Usage:\n%s\n", cmdtp->usage);		return 1;	}	if (strcmp(argv[1], "off") == 0)		p = 0;	else if (strcmp(argv[1], "on") == 0)		p = 1;	else {		printf ("Usage:\n%s\n", cmdtp->usage);		return 1;	}	if (strcmp(argv[2], "all") == 0) {		for (bank=1; bank<=CFG_MAX_FLASH_BANKS; ++bank) {			info = &flash_info[bank-1];			if (info->flash_id == FLASH_UNKNOWN) {				continue;			}			/*printf ("%sProtect Flash Bank # %ld\n", */			/*	p ? "" : "Un-", bank); */			for (i=0; i<info->sector_count; ++i) {#if defined(CFG_FLASH_PROTECTION)				if (flash_real_protect(info, i, p))					rcode = 1;				putc ('.');#else				info->protect[i] = p;#endif	/* CFG_FLASH_PROTECTION */			}		}#if defined(CFG_FLASH_PROTECTION)		if (!rcode) puts (" done\n");#endif	/* CFG_FLASH_PROTECTION */		return rcode;	}	if ((n = abbrev_spec(argv[2], &info, &sect_first, &sect_last)) != 0) {		if (n < 0) {			printf("Bad sector specification\n");			return 1;		}		/*printf("%sProtect Flash Sectors %d-%d in Bank # %d\n", */		/*	p ? "" : "Un-", sect_first, sect_last, */		/*	(info-flash_info)+1); */		for (i = sect_first; i <= sect_last; i++) {#if defined(CFG_FLASH_PROTECTION)			if (flash_real_protect(info, i, p))				rcode =  1;			putc ('.');#else			info->protect[i] = p;#endif	/* CFG_FLASH_PROTECTION */		}#if defined(CFG_FLASH_PROTECTION)		if (!rcode) puts (" done\n");#endif	/* CFG_FLASH_PROTECTION */		return rcode;	}	if (argc != 4) {		printf ("Usage:\n%s\n", cmdtp->usage);		return 1;	}	if (strcmp(argv[2], "bank") == 0) {		bank = simple_strtoul(argv[3], NULL, 16);		if ((bank < 1) || (bank > CFG_MAX_FLASH_BANKS)) {			printf ("Only FLASH Banks # 1 ... # %d supported\n",				CFG_MAX_FLASH_BANKS);			return 1;		}		printf ("%sProtect Flash Bank # %ld\n",			p ? "" : "Un-", bank);		info = &flash_info[bank-1];		if (info->flash_id == FLASH_UNKNOWN) {			printf ("missing or unknown FLASH type\n");			return 1;		}		for (i=0; i<info->sector_count; ++i) {#if defined(CFG_FLASH_PROTECTION)			if (flash_real_protect(info, i, p))				rcode =  1;			putc ('.');#else			info->protect[i] = p;#endif	/* CFG_FLASH_PROTECTION */		}#if defined(CFG_FLASH_PROTECTION)		if (!rcode) puts (" done\n");#endif	/* CFG_FLASH_PROTECTION */		return rcode;	}	addr_first = simple_strtoul(argv[2], NULL, 16);	addr_last  = simple_strtoul(argv[3], NULL, 16);	if (addr_first >= addr_last) {		printf ("Usage:\n%s\n", cmdtp->usage);		return 1;	}	rcode = flash_sect_protect (p, addr_first, addr_last);	return rcode;}int flash_sect_protect (int p, ulong addr_first, ulong addr_last){	flash_info_t *info;	ulong bank;	int s_first, s_last;	int protected, i;	int rcode = 0;	protected = 0;	for (bank=0,info=&flash_info[0]; bank < CFG_MAX_FLASH_BANKS; ++bank, ++info) {		ulong b_end;		int sect;		if (info->flash_id == FLASH_UNKNOWN) {			continue;		}		b_end = info->start[0] + info->size - 1; /* bank end addr */		s_first = -1;		/* first sector to erase	*/		s_last  = -1;		/* last  sector to erase	*/		for (sect=0; sect < info->sector_count; ++sect) {			ulong end;		/* last address in current sect	*/			short s_end;			s_end = info->sector_count - 1;			end = (sect == s_end) ? b_end : info->start[sect + 1] - 1;			if (addr_first > end)				continue;			if (addr_last < info->start[sect])				continue;			if (addr_first == info->start[sect]) {				s_first = sect;			}			if (addr_last  == end) {				s_last  = sect;			}		}		if (s_first>=0 && s_first<=s_last) {			protected += s_last - s_first + 1;			for (i=s_first; i<=s_last; ++i) {#if defined(CFG_FLASH_PROTECTION)				if (flash_real_protect(info, i, p))					rcode = 1;				putc ('.');#else				info->protect[i] = p;#endif	/* CFG_FLASH_PROTECTION */			}		}#if defined(CFG_FLASH_PROTECTION)		if (!rcode) putc ('\n');#endif	/* CFG_FLASH_PROTECTION */	}	if (protected) {	    /*	printf ("%sProtected %d sectors\n", */	    /*	p ? "" : "Un-", protected); */	} else {	    printf ("Error: start and/or end address"			" not on sector boundary\n");		rcode = 1;	}	return rcode;}#endif	/* CFG_CMD_FLASH */