disk: Extend disk_blk_part_validate() with range checking

Check whether access is out of bounds of the partition and
return an error. This way there is no danger of esp. write
or erase outside of the confines of partition.

Signed-off-by: Marek Vasut <marek.vasut+renesas@mailbox.org>

diff --git a/disk/disk-uclass.c b/disk/disk-uclass.c
index 32722cf91760c41c5ceacb7089c179126efb4f1c..f262105375b6ad4dd68c6d59b05306f203e508bf 100644 (file)
--- a/disk/disk-uclass.c
+++ b/disk/disk-uclass.c
@@ -27,9 +27,17 @@
  */
 static int disk_blk_part_validate(struct udevice *dev, lbaint_t start, lbaint_t blkcnt)
 {
+       struct disk_part *part = dev_get_uclass_plat(dev);
+
        if (device_get_uclass_id(dev) != UCLASS_PARTITION)
                return -ENOSYS;
 
+       if (start >= part->gpt_part_info.size)
+               return -E2BIG;
+
+       if ((start + blkcnt) > part->gpt_part_info.size)
+               return -ERANGE;
+
        return 0;
 }