diff options
Diffstat (limited to 'drivers')
| -rw-r--r-- | drivers/dfu/dfu_nand.c | 18 | ||||
| -rw-r--r-- | drivers/mtd/nand/nand_util.c | 4 | 
2 files changed, 20 insertions, 2 deletions
| diff --git a/drivers/dfu/dfu_nand.c b/drivers/dfu/dfu_nand.c index 7dc89b2f2..07dee8981 100644 --- a/drivers/dfu/dfu_nand.c +++ b/drivers/dfu/dfu_nand.c @@ -63,12 +63,26 @@ static int nand_block_op(enum dfu_nand_op op, struct dfu_entity *dfu,  	nand = &nand_info[nand_curr_device]; -	if (op == DFU_OP_READ) +	if (op == DFU_OP_READ) {  		ret = nand_read_skip_bad(nand, start, &count, &actual,  				lim, buf); -	else +	} else { +		nand_erase_options_t opts; + +		memset(&opts, 0, sizeof(opts)); +		opts.offset = start; +		opts.length = count; +		opts.spread = 1; +		opts.quiet = 1; +		opts.lim = lim; +		/* first erase */ +		ret = nand_erase_opts(nand, &opts); +		if (ret) +			return ret; +		/* then write */  		ret = nand_write_skip_bad(nand, start, &count, &actual,  				lim, buf, 0); +	}  	if (ret != 0) {  		printf("%s: nand_%s_skip_bad call failed at %llx!\n", diff --git a/drivers/mtd/nand/nand_util.c b/drivers/mtd/nand/nand_util.c index d81972ca2..1d22b5240 100644 --- a/drivers/mtd/nand/nand_util.c +++ b/drivers/mtd/nand/nand_util.c @@ -120,6 +120,10 @@ int nand_erase_opts(nand_info_t *meminfo, const nand_erase_options_t *opts)  		WATCHDOG_RESET(); +		if (opts->lim && (erase.addr >= (opts->offset + opts->lim))) { +			puts("Size of erase exceeds limit\n"); +			return -EFBIG; +		}  		if (!opts->scrub && bbtest) {  			int ret = mtd_block_isbad(meminfo, erase.addr);  			if (ret > 0) { |