Эх сурвалжийг харах

mmc: add boundary check for mmc operation

Signed-off-by: Lei Wen <leiwen@marvell.com>
Lei Wen 14 жил өмнө
parent
commit
d2bf29e399
1 өөрчлөгдсөн 10 нэмэгдсэн , 0 устгасан
  1. 10 0
      drivers/mmc/mmc.c

+ 10 - 0
drivers/mmc/mmc.c

@@ -92,6 +92,11 @@ mmc_bwrite(int dev_num, ulong start, lbaint_t blkcnt, const void*src)
 
 
 	blklen = mmc->write_bl_len;
 	blklen = mmc->write_bl_len;
 
 
+	if ((start + blkcnt) > mmc->block_dev.lba) {
+		printf("MMC: block number 0x%x exceeds max(0x%x)",
+			start + blkcnt, mmc->block_dev.lba);
+		return 0;
+	}
 	err = mmc_set_blocklen(mmc, mmc->write_bl_len);
 	err = mmc_set_blocklen(mmc, mmc->write_bl_len);
 
 
 	if (err) {
 	if (err) {
@@ -219,6 +224,11 @@ static ulong mmc_bread(int dev_num, ulong start, lbaint_t blkcnt, void *dst)
 	if (!mmc)
 	if (!mmc)
 		return 0;
 		return 0;
 
 
+	if ((start + blkcnt) > mmc->block_dev.lba) {
+		printf("MMC: block number 0x%x exceeds max(0x%x)",
+			start + blkcnt, mmc->block_dev.lba);
+		return 0;
+	}
 	/* We always do full block reads from the card */
 	/* We always do full block reads from the card */
 	err = mmc_set_blocklen(mmc, mmc->read_bl_len);
 	err = mmc_set_blocklen(mmc, mmc->read_bl_len);