|  | /* | 
|  | * (C) Copyright 2000 | 
|  | * Wolfgang Denk, DENX Software Engineering, wd@denx.de. | 
|  | * | 
|  | * SPDX-License-Identifier:	GPL-2.0+ | 
|  | */ | 
|  |  | 
|  | /* | 
|  | * 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 > CONFIG_SYS_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 <CONFIG_SYS_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 > CONFIG_SYS_MAX_FLASH_BANKS)) { | 
|  | printf ("Only FLASH Banks # 1 ... # %d supported\n", | 
|  | CONFIG_SYS_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) | 
|  | return cmd_usage(cmdtp); | 
|  |  | 
|  | if (strcmp(argv[1], "all") == 0) { | 
|  | for (bank=1; bank<=CONFIG_SYS_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) | 
|  | return cmd_usage(cmdtp); | 
|  |  | 
|  | if (strcmp(argv[1], "bank") == 0) { | 
|  | bank = simple_strtoul(argv[2], NULL, 16); | 
|  | if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) { | 
|  | printf ("Only FLASH Banks # 1 ... # %d supported\n", | 
|  | CONFIG_SYS_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) | 
|  | return cmd_usage(cmdtp); | 
|  |  | 
|  | 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 < CONFIG_SYS_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) | 
|  | return cmd_usage(cmdtp); | 
|  |  | 
|  | if (strcmp(argv[1], "off") == 0) | 
|  | p = 0; | 
|  | else if (strcmp(argv[1], "on") == 0) | 
|  | p = 1; | 
|  | else | 
|  | return cmd_usage(cmdtp); | 
|  |  | 
|  | if (strcmp(argv[2], "all") == 0) { | 
|  | for (bank=1; bank<=CONFIG_SYS_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(CONFIG_SYS_FLASH_PROTECTION) | 
|  | if (flash_real_protect(info, i, p)) | 
|  | rcode = 1; | 
|  | putc ('.'); | 
|  | #else | 
|  | info->protect[i] = p; | 
|  | #endif	/* CONFIG_SYS_FLASH_PROTECTION */ | 
|  | } | 
|  | } | 
|  |  | 
|  | #if defined(CONFIG_SYS_FLASH_PROTECTION) | 
|  | if (!rcode) puts (" done\n"); | 
|  | #endif	/* CONFIG_SYS_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(CONFIG_SYS_FLASH_PROTECTION) | 
|  | if (flash_real_protect(info, i, p)) | 
|  | rcode =  1; | 
|  | putc ('.'); | 
|  | #else | 
|  | info->protect[i] = p; | 
|  | #endif	/* CONFIG_SYS_FLASH_PROTECTION */ | 
|  | } | 
|  |  | 
|  | #if defined(CONFIG_SYS_FLASH_PROTECTION) | 
|  | if (!rcode) puts (" done\n"); | 
|  | #endif	/* CONFIG_SYS_FLASH_PROTECTION */ | 
|  |  | 
|  | return rcode; | 
|  | } | 
|  |  | 
|  | if (argc != 4) | 
|  | return cmd_usage(cmdtp); | 
|  |  | 
|  | if (strcmp(argv[2], "bank") == 0) { | 
|  | bank = simple_strtoul(argv[3], NULL, 16); | 
|  | if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) { | 
|  | printf ("Only FLASH Banks # 1 ... # %d supported\n", | 
|  | CONFIG_SYS_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(CONFIG_SYS_FLASH_PROTECTION) | 
|  | if (flash_real_protect(info, i, p)) | 
|  | rcode =  1; | 
|  | putc ('.'); | 
|  | #else | 
|  | info->protect[i] = p; | 
|  | #endif	/* CONFIG_SYS_FLASH_PROTECTION */ | 
|  | } | 
|  |  | 
|  | #if defined(CONFIG_SYS_FLASH_PROTECTION) | 
|  | if (!rcode) | 
|  | puts(" done\n"); | 
|  | #endif	/* CONFIG_SYS_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) | 
|  | return cmd_usage(cmdtp); | 
|  |  | 
|  | return flash_sect_protect (p, addr_first, addr_last); | 
|  | } | 
|  | 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 < CONFIG_SYS_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(CONFIG_SYS_FLASH_PROTECTION) | 
|  | if (flash_real_protect(info, i, p)) | 
|  | rcode = 1; | 
|  | putc ('.'); | 
|  | #else | 
|  | info->protect[i] = p; | 
|  | #endif	/* CONFIG_SYS_FLASH_PROTECTION */ | 
|  | } | 
|  | } | 
|  | #if defined(CONFIG_SYS_FLASH_PROTECTION) | 
|  | if (!rcode) putc ('\n'); | 
|  | #endif	/* CONFIG_SYS_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 |