sfdisk: validate partition number is positive

Signed-off-by: lijian <lijian01@kylinos.cn>

diff --git a/disk-utils/sfdisk.c b/disk-utils/sfdisk.c
index 0a66bc5fcf3bfcc3f70b9da928bd3e555e4da303..d35962d9078af7ae6e70d2bf27cdd84af5b3c8ad 100644 (file)
--- a/disk-utils/sfdisk.c
+++ b/disk-utils/sfdisk.c
@@ -1139,6 +1139,8 @@ static void assign_device_partition(struct sfdisk *sf,
        if (partno > n)
                errx(EXIT_FAILURE, _("%s: partition %zu: partition table contains "
                                     "only %zu partitions"), devname, partno, n);
+       if (partno == 0)
+               errx(EXIT_FAILURE, _("%s: partition number must be a positive number"), devname);
        if (!fdisk_is_partition_used(sf->cxt, partno - 1))
                errx(EXIT_FAILURE, _("%s: partition %zu: partition is unused"),
                                devname, partno);