sfdisk: add --part-attrs

Signed-off-by: Karel Zak <kzak@redhat.com>

diff --git a/disk-utils/sfdisk.8 b/disk-utils/sfdisk.8
index f0eea948cdd4ea276ce15187950b55ec57b00749..d2de5101f0c757ae65fd05ab6a55c1790f71f624 100644 (file)
--- a/disk-utils/sfdisk.8
+++ b/disk-utils/sfdisk.8
@@ -66,6 +66,13 @@ List geometry of all or specified devices.
 List partitions of all or specified devices. This command can be used
 together with \fB\-\-verify\fR.
 .TP
+.BR \-\-part\-attrs " " \fIdevice\fR " " \fIpartno\fR " "[\fIattrs\fR]
+Change GPT partition attribute bits. If \fIattrs\fR no specified then print the current
+partition setting. The \fIattrs\fR is comma delimited list of bits. The currently
+supported attribute bits are: RequiredPartiton, NoBlockIOProtocol, LegacyBIOSBootable
+and GUID specific bits in range from 48 to 63. For example string 
+"RequiredPartiton,50,51" sets three bits.
+.TP
 .BR \-\-part\-label " " \fIdevice\fR " " \fIpartno\fR " "[\fIlabel\fR]
 Change GPT partition name (label). If \fIlabel\fR no specified then print the current
 partition label.
@@ -255,7 +262,8 @@ in bytes and the size is aligned according to device I/O limits.
 Mark partition as bootable.
 .TP
 .B attrs=<string>
-Partition attributes.
+Partition attributes, usually GPT partition attribute bits. See --part-attrs for
+more details about GPT bits string format.
 .TP
 .B uuid=<string>
 GPT partition UUID.

diff --git a/disk-utils/sfdisk.c b/disk-utils/sfdisk.c
index b9ae1b35c3c28c7a9e25c2167e7791523de3d8fc..754bd6f2eac04cd86883a96c071b22319c2883d1 100644 (file)
--- a/disk-utils/sfdisk.c
+++ b/disk-utils/sfdisk.c
@@ -72,6 +72,7 @@ enum {
        ACT_PARTTYPE,
        ACT_PARTUUID,
        ACT_PARTLABEL,
+       ACT_PARTATTRS,
 };
 
 struct sfdisk {
@@ -821,6 +822,60 @@ static int command_partlabel(struct sfdisk *sf, int argc, char **argv)
        return write_changes(sf);
 }
 
+/*
+ * sfdisk --part-attrs <device> <partno> [<attrs>]
+ */
+static int command_partattrs(struct sfdisk *sf, int argc, char **argv)
+{
+       size_t partno;
+       struct fdisk_partition *pa = NULL;
+       const char *devname = NULL, *attrs = NULL;
+
+       if (!argc)
+               errx(EXIT_FAILURE, _("no disk device specified"));
+       devname = argv[0];
+
+       if (argc < 2)
+               errx(EXIT_FAILURE, _("no partition number specified"));
+       partno = strtou32_or_err(argv[1], _("failed to parse partition number"));
+
+       if (argc == 3)
+               attrs = argv[2];
+       else if (argc > 3)
+               errx(EXIT_FAILURE, _("uneexpected arguments"));
+
+       /* read-only if name not given */
+       assign_device_partition(sf, devname, partno, !attrs);
+
+       /* print partition name */
+       if (!attrs) {
+               const char *str;
+
+               if (fdisk_get_partition(sf->cxt, partno - 1, &pa) == 0)
+                       str = fdisk_partition_get_attrs(pa);
+               if (str)
+                       printf("%s\n", str);
+               fdisk_unref_partition(pa);
+               fdisk_deassign_device(sf->cxt, 1);
+               return 0;
+       }
+
+       if (sf->backup)
+               backup_partition_table(sf, devname);
+
+       pa = fdisk_new_partition();
+       if (!pa)
+               err(EXIT_FAILURE, _("failed to allocate partition object"));
+
+       if (fdisk_partition_set_attrs(pa, attrs) != 0 ||
+           fdisk_set_partition(sf->cxt, partno - 1, pa) != 0)
+               errx(EXIT_FAILURE, _("%s: partition %zu: failed to set partition attributes"),
+                               devname, partno);
+
+       fdisk_unref_partition(pa);
+       return write_changes(sf);
+}
+
 static void sfdisk_print_partition(struct sfdisk *sf, size_t n)
 {
        struct fdisk_partition *pa = NULL;
@@ -1240,6 +1295,7 @@ static void __attribute__ ((__noreturn__)) usage(FILE *out)
        fputs(_(" --part-label <dev> <part> [<str>] print or change partition label\n"), out);
        fputs(_(" --part-type <dev> <part> [<type>] print or change partition type\n"), out);
        fputs(_(" --part-uuid <dev> <part> [<uuid>] print or change partition uuid\n"), out);
+       fputs(_(" --part-attrs <dev> <part> [<str>] print or change partition attributes\n"), out);
 
        fputs(USAGE_SEPARATOR, out);
        fputs(_(" <dev>                     device (usually disk) path\n"), out);
@@ -1284,6 +1340,7 @@ int main(int argc, char *argv[])
                OPT_PARTUUID,
                OPT_PARTLABEL,
                OPT_PARTTYPE,
+               OPT_PARTATTRS,
        };
 
        static const struct option longopts[] = {
@@ -1309,6 +1366,7 @@ int main(int argc, char *argv[])
                { "part-uuid",  no_argument,    NULL, OPT_PARTUUID },
                { "part-label", no_argument,    NULL, OPT_PARTLABEL },
                { "part-type",  no_argument,    NULL, OPT_PARTTYPE },
+               { "part-attrs", no_argument,    NULL, OPT_PARTATTRS },
 
                { "unit",    required_argument, NULL, 'u' },            /* deprecated */
                { "Linux",   no_argument,       NULL, 'L' },            /* deprecated */
@@ -1410,6 +1468,9 @@ int main(int argc, char *argv[])
                case OPT_PARTLABEL:
                        sf->act = ACT_PARTLABEL;
                        break;
+               case OPT_PARTATTRS:
+                       sf->act = ACT_PARTATTRS;
+                       break;
                case OPT_NOREREAD:
                        sf->noreread = 1;
                        break;
@@ -1471,6 +1532,10 @@ int main(int argc, char *argv[])
                rc = command_partlabel(sf, argc - optind, argv + optind);
                break;
 
+       case ACT_PARTATTRS:
+               rc = command_partattrs(sf, argc - optind, argv + optind);
+               break;
+
        }
 
        sfdisk_deinit(sf);