| /* | 
 |  * (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 defined(CONFIG_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 int | 
 | abbrev_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, §_first, §_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, §_first, §_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 |