From e6effcac61152880451da25142cf93931834ffa9 Mon Sep 17 00:00:00 2001 From: Karel Zak Date: Thu, 10 Jan 2019 12:17:19 +0100 Subject: fdisk: make partition types uses more robust * report failed partition type parsing * make sure partition types code (from user reply) for MBR is hex Reported-by: Jonny Grant Signed-off-by: Karel Zak --- disk-utils/fdisk.c | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'disk-utils') diff --git a/disk-utils/fdisk.c b/disk-utils/fdisk.c index 89fec2629..3e8ebf4a3 100644 --- a/disk-utils/fdisk.c +++ b/disk-utils/fdisk.c @@ -493,8 +493,13 @@ static struct fdisk_parttype *ask_partition_type(struct fdisk_context *cxt, int if (buf[1] == '\0' && toupper(*buf) == 'L') list_partition_types(cxt); - else if (*buf) - return fdisk_label_parse_parttype(lb, buf); + else if (*buf) { + struct fdisk_parttype *t = fdisk_label_parse_parttype(lb, buf); + + if (!t) + fdisk_info(cxt, _("Failed to parse '%s' partition type."), buf); + return t; + } } while (1); return NULL; @@ -621,7 +626,7 @@ void change_partition_type(struct fdisk_context *cxt) break; } while (!t); - if (canceled == 0 && fdisk_set_partition_type(cxt, i, t) == 0) + if (canceled == 0 && t && fdisk_set_partition_type(cxt, i, t) == 0) fdisk_info(cxt, _("Changed type of partition '%s' to '%s'."), old, t ? fdisk_parttype_get_name(t) : _("Unknown")); -- cgit v1.2.3-55-g7522