qiurui
/
rv1126-uboot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mtd: mtd_blk: Check map table block address overflow 

1.Check map table block address overflow
2.Reinit map table original value

Change-Id: I4450b5a6856e38e2624da9db31d5eb98de7f5696
Signed-off-by: Jon Lin <jon.lin@rock-chips.com>
This commit is contained in:
Jon Lin 2020-06-15 15:24:48 +08:00 committed by Dingqiang Lin
parent f9976ae435
commit 0f1dc4879a
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
drivers/mtd/mtd_blk.c
View File

@ -49,13 +49,17 @@ int mtd_blk_map_table_init(struct blk_desc *desc,
} else {
blk_total = (mtd->size + mtd->erasesize - 1) >> mtd->erasesize_shift;
if (!mtd_map_blk_table) {
mtd_map_blk_table = (int *)malloc(blk_total * 4);
memset(mtd_map_blk_table, MTD_BLK_TABLE_BLOCK_UNKNOWN,
blk_total * 4);
mtd_map_blk_table = (int *)malloc(blk_total * sizeof(int));
for (i = 0; i < blk_total; i++)
mtd_map_blk_table[i] = MTD_BLK_TABLE_BLOCK_UNKNOWN;
}
blk_begin = (u32)offset >> mtd->erasesize_shift;
blk_cnt = ((u32)((offset & mtd->erasesize_mask) + length) >> mtd->erasesize_shift);
if (blk_begin >= blk_total) {
pr_err("map table blk begin[%d] overflow\n", blk_begin);
return -EINVAL;
}
if ((blk_begin + blk_cnt) > blk_total)
blk_cnt = blk_total - blk_begin;