tools/updater/cmd_flash.c - uboot/prism - Git at Google

 /*
  * (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 > 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) {
 		cmd_usage(cmdtp);
 		return 1;
 	}

 	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, &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) {
 		cmd_usage(cmdtp);
 		return 1;
 	}

 	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) {
 		cmd_usage(cmdtp);
 		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 < 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) {
 		cmd_usage(cmdtp);
 		return 1;
 	}

 	if (strcmp(argv[1], "off") == 0)
 		p = 0;
 	else if (strcmp(argv[1], "on") == 0)
 		p = 1;
 	else {
 		cmd_usage(cmdtp);
 		return 1;
 	}

 	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, &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(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) {
 		cmd_usage(cmdtp);
 		return 1;
 	}

 	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) {
 		cmd_usage(cmdtp);
 		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 < 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
	/*
	* (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 > 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) {
	cmd_usage(cmdtp);
	return 1;
	}

	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, &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) {
	cmd_usage(cmdtp);
	return 1;
	}

	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) {
	cmd_usage(cmdtp);
	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 < 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) {
	cmd_usage(cmdtp);
	return 1;
	}

	if (strcmp(argv[1], "off") == 0)
	p = 0;
	else if (strcmp(argv[1], "on") == 0)
	p = 1;
	else {
	cmd_usage(cmdtp);
	return 1;
	}

	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, &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(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) {
	cmd_usage(cmdtp);
	return 1;
	}

	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) {
	cmd_usage(cmdtp);
	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 < 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