cmd_flash.c 19.2 KB
Newer Older
wdenk's avatar
wdenk committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
/*
 * (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>
wdenk's avatar
wdenk committed
29

30 31 32 33
#ifdef CONFIG_HAS_DATAFLASH
#include <dataflash.h>
#endif

34
#if defined(CONFIG_CMD_MTDPARTS)
35 36
#include <jffs2/jffs2.h>

Wolfgang Denk's avatar
Wolfgang Denk committed
37
/* partition handling routines */
38
int mtdparts_init(void);
39
int mtd_id_parse(const char *id, const char **ret_id, u8 *dev_type, u8 *dev_num);
40 41 42 43
int find_dev_and_part(const char *id, struct mtd_device **dev,
		u8 *part_num, struct part_info **part);
#endif

44
#ifndef CONFIG_SYS_NO_FLASH
45 46
#include <flash.h>
#include <mtd/cfi_flash.h>
wdenk's avatar
wdenk committed
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
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
69
abbrev_spec (char *str, flash_info_t ** pinfo, int *psf, int *psl)
wdenk's avatar
wdenk committed
70
{
71 72 73
	flash_info_t *fp;
	int bank, first, last;
	char *p, *ep;
wdenk's avatar
wdenk committed
74

75 76 77
	if ((p = strchr (str, ':')) == NULL)
		return 0;
	*p++ = '\0';
wdenk's avatar
wdenk committed
78

79 80
	bank = simple_strtoul (str, &ep, 10);
	if (ep == str || *ep != '\0' ||
81
		bank < 1 || bank > CONFIG_SYS_MAX_FLASH_BANKS ||
82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
		(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;
	}
wdenk's avatar
wdenk committed
101

102 103 104 105 106 107 108
	*pinfo = fp;
	*psf = first;
	*psl = last;

	return 1;
}

109 110 111 112 113 114 115 116 117 118 119 120
/*
 * Take *addr in Flash and adjust it to fall on the end of its sector
 */
int flash_sect_roundb (ulong *addr)
{
	flash_info_t *info;
	ulong bank, sector_end_addr;
	char found;
	int i;

	/* find the end addr of the sector where the *addr is */
	found = 0;
121
	for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS && !found; ++bank) {
122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141
		info = &flash_info[bank];
		for (i = 0; i < info->sector_count && !found; ++i) {
			/* get the end address of the sector */
			if (i == info->sector_count - 1) {
				sector_end_addr = info->start[0] +
								info->size - 1;
			} else {
				sector_end_addr = info->start[i+1] - 1;
			}

			if (*addr <= sector_end_addr &&
						*addr >= info->start[i]) {
				found = 1;
				/* adjust *addr if necessary */
				if (*addr < sector_end_addr)
					*addr = sector_end_addr;
			} /* sector */
		} /* bank */
	}
	if (!found) {
142
		/* error, address not in flash */
143 144 145 146 147 148 149
		printf("Error: end address (0x%08lx) not in flash!\n", *addr);
		return 1;
	}

	return 0;
}

150 151 152 153 154
/*
 * This function computes the start and end addresses for both
 * erase and protect commands. The range of the addresses on which
 * either of the commands is to operate can be given in two forms:
 * 1. <cmd> start end - operate on <'start',  'end')
Wolfgang Denk's avatar
Wolfgang Denk committed
155
 * 2. <cmd> start +length - operate on <'start', start + length)
156 157 158 159 160 161 162 163 164 165 166 167 168 169 170
 * If the second form is used and the end address doesn't fall on the
 * sector boundary, than it will be adjusted to the next sector boundary.
 * If it isn't in the flash, the function will fail (return -1).
 * Input:
 *    arg1, arg2: address specification (i.e. both command arguments)
 * Output:
 *    addr_first, addr_last: computed address range
 * Return:
 *    1: success
 *   -1: failure (bad format, bad address).
*/
static int
addr_spec(char *arg1, char *arg2, ulong *addr_first, ulong *addr_last)
{
	char *ep;
Heiko Schocher's avatar
Heiko Schocher committed
171
	char len_used; /* indicates if the "start +length" form used */
172

173 174 175 176
	*addr_first = simple_strtoul(arg1, &ep, 16);
	if (ep == arg1 || *ep != '\0')
		return -1;

Heiko Schocher's avatar
Heiko Schocher committed
177
	len_used = 0;
178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200
	if (arg2 && *arg2 == '+'){
		len_used = 1;
		++arg2;
	}

	*addr_last = simple_strtoul(arg2, &ep, 16);
	if (ep == arg2 || *ep != '\0')
		return -1;

	if (len_used){
		/*
		 * *addr_last has the length, compute correct *addr_last
		 * XXX watch out for the integer overflow! Right now it is
		 * checked for in both the callers.
		 */
		*addr_last = *addr_first + *addr_last - 1;

		/*
		 * It may happen that *addr_last doesn't fall on the sector
		 * boundary. We want to round such an address to the next
		 * sector boundary, so that the commands don't fail later on.
		 */

201
		if (flash_sect_roundb(addr_last) > 0)
202 203 204 205 206 207
			return -1;
	} /* "start +length" from used */

	return 1;
}

208 209 210 211 212 213 214 215 216 217 218
static int
flash_fill_sect_ranges (ulong addr_first, ulong addr_last,
			int *s_first, int *s_last,
			int *s_count )
{
	flash_info_t *info;
	ulong bank;
	int rcode = 0;

	*s_count = 0;

219
	for (bank=0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
220 221 222 223
		s_first[bank] = -1;	/* first sector to erase	*/
		s_last [bank] = -1;	/* last  sector to erase	*/
	}

224
	for (bank=0,info = &flash_info[0];
225
	     (bank < CONFIG_SYS_MAX_FLASH_BANKS) && (addr_first <= addr_last);
226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
	     ++bank, ++info) {
		ulong b_end;
		int sect;
		short s_end;

		if (info->flash_id == FLASH_UNKNOWN) {
			continue;
		}

		b_end = info->start[0] + info->size - 1;	/* bank end addr */
		s_end = info->sector_count - 1;			/* last sector   */


		for (sect=0; sect < info->sector_count; ++sect) {
			ulong end;	/* last address in current sect	*/
wdenk's avatar
wdenk committed
241

242
			end = (sect == s_end) ? b_end : info->start[sect + 1] - 1;
wdenk's avatar
wdenk committed
243

244 245 246 247
			if (addr_first > end)
				continue;
			if (addr_last < info->start[sect])
				continue;
wdenk's avatar
wdenk committed
248

249 250 251 252 253 254 255 256 257 258 259 260
			if (addr_first == info->start[sect]) {
				s_first[bank] = sect;
			}
			if (addr_last  == end) {
				s_last[bank]  = sect;
			}
		}
		if (s_first[bank] >= 0) {
			if (s_last[bank] < 0) {
				if (addr_last > b_end) {
					s_last[bank] = s_end;
				} else {
261
					puts ("Error: end address"
262 263 264 265 266 267
						" not on sector boundary\n");
					rcode = 1;
					break;
				}
			}
			if (s_last[bank] < s_first[bank]) {
268
				puts ("Error: end sector"
269 270 271 272 273 274 275
					" precedes start sector\n");
				rcode = 1;
				break;
			}
			sect = s_last[bank];
			addr_first = (sect == s_end) ? b_end + 1: info->start[sect + 1];
			(*s_count) += s_last[bank] - s_first[bank] + 1;
wdenk's avatar
wdenk committed
276 277 278 279
		} else if (addr_first >= info->start[0] && addr_first < b_end) {
			puts ("Error: start address not on sector boundary\n");
			rcode = 1;
			break;
280
		} else if (s_last[bank] >= 0) {
281
			puts ("Error: cannot span across banks when they are"
282 283 284
			       " mapped in reverse order\n");
			rcode = 1;
			break;
285 286
		}
	}
wdenk's avatar
wdenk committed
287

288
	return rcode;
wdenk's avatar
wdenk committed
289
}
290
#endif /* CONFIG_SYS_NO_FLASH */
291

Kim Phillips's avatar
Kim Phillips committed
292
static int do_flinfo(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
wdenk's avatar
wdenk committed
293
{
294
#ifndef CONFIG_SYS_NO_FLASH
wdenk's avatar
wdenk committed
295
	ulong bank;
Stelian Pop's avatar
Stelian Pop committed
296
#endif
wdenk's avatar
wdenk committed
297

298 299 300 301
#ifdef CONFIG_HAS_DATAFLASH
	dataflash_print_info();
#endif

302
#ifndef CONFIG_SYS_NO_FLASH
wdenk's avatar
wdenk committed
303
	if (argc == 1) {	/* print info for all FLASH banks */
304
		for (bank=0; bank <CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
wdenk's avatar
wdenk committed
305 306 307 308 309 310 311 312
			printf ("\nBank # %ld: ", bank+1);

			flash_print_info (&flash_info[bank]);
		}
		return 0;
	}

	bank = simple_strtoul(argv[1], NULL, 16);
313
	if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
wdenk's avatar
wdenk committed
314
		printf ("Only FLASH Banks # 1 ... # %d supported\n",
315
			CONFIG_SYS_MAX_FLASH_BANKS);
wdenk's avatar
wdenk committed
316 317 318 319
		return 1;
	}
	printf ("\nBank # %ld: ", bank);
	flash_print_info (&flash_info[bank-1]);
320
#endif /* CONFIG_SYS_NO_FLASH */
wdenk's avatar
wdenk committed
321 322
	return 0;
}
323

Kim Phillips's avatar
Kim Phillips committed
324
static int do_flerase(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
wdenk's avatar
wdenk committed
325
{
326
#ifndef CONFIG_SYS_NO_FLASH
327
	flash_info_t *info = NULL;
wdenk's avatar
wdenk committed
328
	ulong bank, addr_first, addr_last;
329
	int n, sect_first = 0, sect_last = 0;
330
#if defined(CONFIG_CMD_MTDPARTS)
331 332 333 334
	struct mtd_device *dev;
	struct part_info *part;
	u8 dev_type, dev_num, pnum;
#endif
wdenk's avatar
wdenk committed
335 336
	int rcode = 0;

337
	if (argc < 2)
338
		return CMD_RET_USAGE;
wdenk's avatar
wdenk committed
339 340

	if (strcmp(argv[1], "all") == 0) {
341
		for (bank=1; bank<=CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
wdenk's avatar
wdenk committed
342 343 344 345 346 347 348 349 350
			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) {
351
			puts ("Bad sector specification\n");
wdenk's avatar
wdenk committed
352 353
			return 1;
		}
Stefan Roese's avatar
Stefan Roese committed
354
		printf ("Erase Flash Sectors %d-%d in Bank # %zu ",
wdenk's avatar
wdenk committed
355 356 357 358 359
			sect_first, sect_last, (info-flash_info)+1);
		rcode = flash_erase(info, sect_first, sect_last);
		return rcode;
	}

360
#if defined(CONFIG_CMD_MTDPARTS)
361
	/* erase <part-id> - erase partition */
362
	if ((argc == 2) && (mtd_id_parse(argv[1], NULL, &dev_type, &dev_num) == 0)) {
363 364 365 366 367 368 369 370
		mtdparts_init();
		if (find_dev_and_part(argv[1], &dev, &pnum, &part) == 0) {
			if (dev->id->type == MTD_DEV_TYPE_NOR) {
				bank = dev->id->num;
				info = &flash_info[bank];
				addr_first = part->offset + info->start[0];
				addr_last = addr_first + part->size - 1;

Wolfgang Denk's avatar
Wolfgang Denk committed
371
				printf ("Erase Flash Partition %s, "
372
						"bank %ld, 0x%08lx - 0x%08lx ",
373 374 375 376 377 378 379 380 381 382 383 384 385
						argv[1], bank, addr_first,
						addr_last);

				rcode = flash_sect_erase(addr_first, addr_last);
				return rcode;
			}

			printf("cannot erase, not a NOR device\n");
			return 1;
		}
	}
#endif

386
	if (argc != 3)
387
		return CMD_RET_USAGE;
wdenk's avatar
wdenk committed
388 389 390

	if (strcmp(argv[1], "bank") == 0) {
		bank = simple_strtoul(argv[2], NULL, 16);
391
		if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
wdenk's avatar
wdenk committed
392
			printf ("Only FLASH Banks # 1 ... # %d supported\n",
393
				CONFIG_SYS_MAX_FLASH_BANKS);
wdenk's avatar
wdenk committed
394 395 396 397 398 399 400 401
			return 1;
		}
		printf ("Erase Flash Bank # %ld ", bank);
		info = &flash_info[bank-1];
		rcode = flash_erase (info, 0, info->sector_count-1);
		return rcode;
	}

402 403 404 405
	if (addr_spec(argv[1], argv[2], &addr_first, &addr_last) < 0){
		printf ("Bad address format\n");
		return 1;
	}
wdenk's avatar
wdenk committed
406

407
	if (addr_first >= addr_last)
408
		return CMD_RET_USAGE;
wdenk's avatar
wdenk committed
409 410 411

	rcode = flash_sect_erase(addr_first, addr_last);
	return rcode;
Stelian Pop's avatar
Stelian Pop committed
412 413
#else
	return 0;
414
#endif /* CONFIG_SYS_NO_FLASH */
wdenk's avatar
wdenk committed
415 416
}

417
#ifndef CONFIG_SYS_NO_FLASH
wdenk's avatar
wdenk committed
418 419 420 421
int flash_sect_erase (ulong addr_first, ulong addr_last)
{
	flash_info_t *info;
	ulong bank;
422
	int s_first[CONFIG_SYS_MAX_FLASH_BANKS], s_last[CONFIG_SYS_MAX_FLASH_BANKS];
423 424
	int erased = 0;
	int planned;
wdenk's avatar
wdenk committed
425 426
	int rcode = 0;

427 428 429 430
	rcode = flash_fill_sect_ranges (addr_first, addr_last,
					s_first, s_last, &planned );

	if (planned && (rcode == 0)) {
431
		for (bank=0,info = &flash_info[0];
432
		     (bank < CONFIG_SYS_MAX_FLASH_BANKS) && (rcode == 0);
433 434 435
		     ++bank, ++info) {
			if (s_first[bank]>=0) {
				erased += s_last[bank] - s_first[bank] + 1;
wdenk's avatar
wdenk committed
436
				debug ("Erase Flash from 0x%08lx to 0x%08lx "
437 438 439 440 441 442 443
					"in Bank # %ld ",
					info->start[s_first[bank]],
					(s_last[bank] == info->sector_count) ?
						info->start[0] + info->size - 1:
						info->start[s_last[bank]+1] - 1,
					bank+1);
				rcode = flash_erase (info, s_first[bank], s_last[bank]);
wdenk's avatar
wdenk committed
444 445
			}
		}
446 447
		if (rcode == 0)
			printf("Erased %d sectors\n", erased);
448
	} else if (rcode == 0) {
449
		puts ("Error: start and/or end address"
wdenk's avatar
wdenk committed
450 451 452 453 454
			" not on sector boundary\n");
		rcode = 1;
	}
	return rcode;
}
455
#endif /* CONFIG_SYS_NO_FLASH */
wdenk's avatar
wdenk committed
456

Kim Phillips's avatar
Kim Phillips committed
457
static int do_protect(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
wdenk's avatar
wdenk committed
458
{
459
	int rcode = 0;
460
#ifndef CONFIG_SYS_NO_FLASH
461
	flash_info_t *info = NULL;
Stelian Pop's avatar
Stelian Pop committed
462
	ulong bank;
463
	int i, n, sect_first = 0, sect_last = 0;
464
#if defined(CONFIG_CMD_MTDPARTS)
465 466 467 468
	struct mtd_device *dev;
	struct part_info *part;
	u8 dev_type, dev_num, pnum;
#endif
469
#endif /* CONFIG_SYS_NO_FLASH */
470 471 472
#ifdef CONFIG_HAS_DATAFLASH
	int status;
#endif
473
#if !defined(CONFIG_SYS_NO_FLASH) || defined(CONFIG_HAS_DATAFLASH)
474
	int p;
475 476
	ulong addr_first, addr_last;
#endif
477

478
	if (argc < 3)
479
		return CMD_RET_USAGE;
wdenk's avatar
wdenk committed
480

481
#if !defined(CONFIG_SYS_NO_FLASH) || defined(CONFIG_HAS_DATAFLASH)
482
	if (strcmp(argv[1], "off") == 0)
wdenk's avatar
wdenk committed
483
		p = 0;
484
	else if (strcmp(argv[1], "on") == 0)
wdenk's avatar
wdenk committed
485
		p = 1;
486
	else
487
		return CMD_RET_USAGE;
488
#endif
wdenk's avatar
wdenk committed
489

490 491 492 493 494 495 496 497
#ifdef CONFIG_HAS_DATAFLASH
	if ((strcmp(argv[2], "all") != 0) && (strcmp(argv[2], "bank") != 0)) {
		addr_first = simple_strtoul(argv[2], NULL, 16);
		addr_last  = simple_strtoul(argv[3], NULL, 16);

		if (addr_dataflash(addr_first) && addr_dataflash(addr_last)) {
			status = dataflash_real_protect(p,addr_first,addr_last);
			if (status < 0){
498
				puts ("Bad DataFlash sector specification\n");
wdenk's avatar
wdenk committed
499 500 501 502
				return 1;
			}
			printf("%sProtect %d DataFlash Sectors\n",
				p ? "" : "Un-", status);
503 504 505 506
			return 0;
		}
	}
#endif
wdenk's avatar
wdenk committed
507

508
#ifndef CONFIG_SYS_NO_FLASH
wdenk's avatar
wdenk committed
509
	if (strcmp(argv[2], "all") == 0) {
510
		for (bank=1; bank<=CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
wdenk's avatar
wdenk committed
511 512 513 514 515 516 517 518
			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) {
519
#if defined(CONFIG_SYS_FLASH_PROTECTION)
wdenk's avatar
wdenk committed
520 521 522 523 524
				if (flash_real_protect(info, i, p))
					rcode = 1;
				putc ('.');
#else
				info->protect[i] = p;
525
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
wdenk's avatar
wdenk committed
526
			}
527
#if defined(CONFIG_SYS_FLASH_PROTECTION)
Heiko Schocher's avatar
Heiko Schocher committed
528
			if (!rcode) puts (" done\n");
529
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
Heiko Schocher's avatar
Heiko Schocher committed
530
		}
wdenk's avatar
wdenk committed
531 532 533 534 535
		return rcode;
	}

	if ((n = abbrev_spec(argv[2], &info, &sect_first, &sect_last)) != 0) {
		if (n < 0) {
536
			puts ("Bad sector specification\n");
wdenk's avatar
wdenk committed
537 538
			return 1;
		}
Stefan Roese's avatar
Stefan Roese committed
539
		printf("%sProtect Flash Sectors %d-%d in Bank # %zu\n",
wdenk's avatar
wdenk committed
540 541 542
			p ? "" : "Un-", sect_first, sect_last,
			(info-flash_info)+1);
		for (i = sect_first; i <= sect_last; i++) {
543
#if defined(CONFIG_SYS_FLASH_PROTECTION)
wdenk's avatar
wdenk committed
544 545 546 547 548
			if (flash_real_protect(info, i, p))
				rcode =  1;
			putc ('.');
#else
			info->protect[i] = p;
549
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
wdenk's avatar
wdenk committed
550 551
		}

552
#if defined(CONFIG_SYS_FLASH_PROTECTION)
wdenk's avatar
wdenk committed
553
		if (!rcode) puts (" done\n");
554
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
wdenk's avatar
wdenk committed
555 556 557

		return rcode;
	}
558

559
#if defined(CONFIG_CMD_MTDPARTS)
560
	/* protect on/off <part-id> */
561
	if ((argc == 3) && (mtd_id_parse(argv[2], NULL, &dev_type, &dev_num) == 0)) {
562 563 564 565 566 567 568 569
		mtdparts_init();
		if (find_dev_and_part(argv[2], &dev, &pnum, &part) == 0) {
			if (dev->id->type == MTD_DEV_TYPE_NOR) {
				bank = dev->id->num;
				info = &flash_info[bank];
				addr_first = part->offset + info->start[0];
				addr_last = addr_first + part->size - 1;

Wolfgang Denk's avatar
Wolfgang Denk committed
570
				printf ("%sProtect Flash Partition %s, "
571
						"bank %ld, 0x%08lx - 0x%08lx\n",
572 573 574 575 576 577 578 579 580 581 582 583 584
						p ? "" : "Un", argv[1],
						bank, addr_first, addr_last);

				rcode = flash_sect_protect (p, addr_first, addr_last);
				return rcode;
			}

			printf("cannot %sprotect, not a NOR device\n",
					p ? "" : "un");
			return 1;
		}
	}
#endif
wdenk's avatar
wdenk committed
585

586
	if (argc != 4)
587
		return CMD_RET_USAGE;
wdenk's avatar
wdenk committed
588 589 590

	if (strcmp(argv[2], "bank") == 0) {
		bank = simple_strtoul(argv[3], NULL, 16);
591
		if ((bank < 1) || (bank > CONFIG_SYS_MAX_FLASH_BANKS)) {
wdenk's avatar
wdenk committed
592
			printf ("Only FLASH Banks # 1 ... # %d supported\n",
593
				CONFIG_SYS_MAX_FLASH_BANKS);
wdenk's avatar
wdenk committed
594 595 596 597 598 599 600
			return 1;
		}
		printf ("%sProtect Flash Bank # %ld\n",
			p ? "" : "Un-", bank);
		info = &flash_info[bank-1];

		if (info->flash_id == FLASH_UNKNOWN) {
601
			puts ("missing or unknown FLASH type\n");
wdenk's avatar
wdenk committed
602 603 604
			return 1;
		}
		for (i=0; i<info->sector_count; ++i) {
605
#if defined(CONFIG_SYS_FLASH_PROTECTION)
wdenk's avatar
wdenk committed
606 607 608 609 610
			if (flash_real_protect(info, i, p))
				rcode =  1;
			putc ('.');
#else
			info->protect[i] = p;
611
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
wdenk's avatar
wdenk committed
612 613
		}

614
#if defined(CONFIG_SYS_FLASH_PROTECTION)
wdenk's avatar
wdenk committed
615
		if (!rcode) puts (" done\n");
616
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
wdenk's avatar
wdenk committed
617 618 619 620

		return rcode;
	}

621 622 623 624
	if (addr_spec(argv[2], argv[3], &addr_first, &addr_last) < 0){
		printf("Bad address format\n");
		return 1;
	}
wdenk's avatar
wdenk committed
625

626
	if (addr_first >= addr_last)
627
		return CMD_RET_USAGE;
628

wdenk's avatar
wdenk committed
629
	rcode = flash_sect_protect (p, addr_first, addr_last);
630
#endif /* CONFIG_SYS_NO_FLASH */
wdenk's avatar
wdenk committed
631 632 633
	return rcode;
}

634
#ifndef CONFIG_SYS_NO_FLASH
wdenk's avatar
wdenk committed
635 636 637 638
int flash_sect_protect (int p, ulong addr_first, ulong addr_last)
{
	flash_info_t *info;
	ulong bank;
639
	int s_first[CONFIG_SYS_MAX_FLASH_BANKS], s_last[CONFIG_SYS_MAX_FLASH_BANKS];
wdenk's avatar
wdenk committed
640
	int protected, i;
641 642
	int planned;
	int rcode;
wdenk's avatar
wdenk committed
643

644
	rcode = flash_fill_sect_ranges( addr_first, addr_last, s_first, s_last, &planned );
wdenk's avatar
wdenk committed
645

646
	protected = 0;
wdenk's avatar
wdenk committed
647

648
	if (planned && (rcode == 0)) {
649
		for (bank=0,info = &flash_info[0]; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank, ++info) {
650
			if (info->flash_id == FLASH_UNKNOWN) {
wdenk's avatar
wdenk committed
651 652
				continue;
			}
653 654

			if (s_first[bank]>=0 && s_first[bank]<=s_last[bank]) {
stroese's avatar
stroese committed
655 656
				debug ("%sProtecting sectors %d..%d in bank %ld\n",
					p ? "" : "Un-",
657 658 659
					s_first[bank], s_last[bank], bank+1);
				protected += s_last[bank] - s_first[bank] + 1;
				for (i=s_first[bank]; i<=s_last[bank]; ++i) {
660
#if defined(CONFIG_SYS_FLASH_PROTECTION)
661 662 663
					if (flash_real_protect(info, i, p))
						rcode = 1;
					putc ('.');
wdenk's avatar
wdenk committed
664
#else
665
					info->protect[i] = p;
666
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
667
				}
wdenk's avatar
wdenk committed
668
			}
669
		}
670
#if defined(CONFIG_SYS_FLASH_PROTECTION)
671
		puts (" done\n");
672
#endif	/* CONFIG_SYS_FLASH_PROTECTION */
wdenk's avatar
wdenk committed
673 674 675

		printf ("%sProtected %d sectors\n",
			p ? "" : "Un-", protected);
676
	} else if (rcode == 0) {
677
		puts ("Error: start and/or end address"
wdenk's avatar
wdenk committed
678 679 680 681 682
			" not on sector boundary\n");
		rcode = 1;
	}
	return rcode;
}
683
#endif /* CONFIG_SYS_NO_FLASH */
wdenk's avatar
wdenk committed
684

wdenk's avatar
wdenk committed
685 686

/**************************************************/
687
#if defined(CONFIG_CMD_MTDPARTS)
688 689 690 691 692 693 694 695
# define TMP_ERASE	"erase <part-id>\n    - erase partition\n"
# define TMP_PROT_ON	"protect on <part-id>\n    - protect partition\n"
# define TMP_PROT_OFF	"protect off <part-id>\n    - make partition writable\n"
#else
# define TMP_ERASE	/* empty */
# define TMP_PROT_ON	/* empty */
# define TMP_PROT_OFF	/* empty */
#endif
wdenk's avatar
wdenk committed
696

697 698
U_BOOT_CMD(
	flinfo,    2,    1,    do_flinfo,
Peter Tyser's avatar
Peter Tyser committed
699
	"print FLASH memory information",
wdenk's avatar
wdenk committed
700
	"\n    - print information for all FLASH memory banks\n"
Wolfgang Denk's avatar
Wolfgang Denk committed
701
	"flinfo N\n    - print information for FLASH memory bank # N"
wdenk's avatar
wdenk committed
702 703
);

704
U_BOOT_CMD(
705
	erase,   3,   0,  do_flerase,
Peter Tyser's avatar
Peter Tyser committed
706
	"erase FLASH memory",
wdenk's avatar
wdenk committed
707 708
	"start end\n"
	"    - erase FLASH from addr 'start' to addr 'end'\n"
709 710 711
	"erase start +len\n"
	"    - erase FLASH from addr 'start' to the end of sect "
	"w/addr 'start'+'len'-1\n"
wdenk's avatar
wdenk committed
712 713
	"erase N:SF[-SL]\n    - erase sectors SF-SL in FLASH bank # N\n"
	"erase bank N\n    - erase FLASH bank # N\n"
714
	TMP_ERASE
Wolfgang Denk's avatar
Wolfgang Denk committed
715
	"erase all\n    - erase all FLASH banks"
wdenk's avatar
wdenk committed
716 717
);

718
U_BOOT_CMD(
719
	protect,  4,  0,   do_protect,
Peter Tyser's avatar
Peter Tyser committed
720
	"enable or disable FLASH write protection",
wdenk's avatar
wdenk committed
721 722
	"on  start end\n"
	"    - protect FLASH from addr 'start' to addr 'end'\n"
723 724 725
	"protect on start +len\n"
	"    - protect FLASH from addr 'start' to end of sect "
	"w/addr 'start'+'len'-1\n"
wdenk's avatar
wdenk committed
726 727 728
	"protect on  N:SF[-SL]\n"
	"    - protect sectors SF-SL in FLASH bank # N\n"
	"protect on  bank N\n    - protect FLASH bank # N\n"
729
	TMP_PROT_ON
wdenk's avatar
wdenk committed
730 731 732
	"protect on  all\n    - protect all FLASH banks\n"
	"protect off start end\n"
	"    - make FLASH from addr 'start' to addr 'end' writable\n"
733 734 735
	"protect off start +len\n"
	"    - make FLASH from addr 'start' to end of sect "
	"w/addr 'start'+'len'-1 wrtable\n"
wdenk's avatar
wdenk committed
736 737 738
	"protect off N:SF[-SL]\n"
	"    - make sectors SF-SL writable in FLASH bank # N\n"
	"protect off bank N\n    - make FLASH bank # N writable\n"
739
	TMP_PROT_OFF
Wolfgang Denk's avatar
Wolfgang Denk committed
740
	"protect off all\n    - make all FLASH banks writable"
wdenk's avatar
wdenk committed
741 742
);

743 744 745
#undef	TMP_ERASE
#undef	TMP_PROT_ON
#undef	TMP_PROT_OFF