[GRUB PARTUUID PATCH V10 3/4] Add PARTUUID detection support to grub-probe

[Nicholas Vinson]

Add PARTUUID detection support grub-probe for MBR and GPT partition
schemes.

Signed-off-by: Nicholas Vinson <address@hidden>
Reviewed-by: Daniel Kiper <address@hidden>

---
 util/grub-probe.c | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/util/grub-probe.c b/util/grub-probe.c
index 21cb80fbe..e91ae730e 100644
--- a/util/grub-probe.c
+++ b/util/grub-probe.c
@@ -28,6 +28,7 @@
 #include <grub/partition.h>
 #include <grub/msdos_partition.h>
 #include <grub/gpt_partition.h>
+#include <grub/i386/pc/boot.h>
 #include <grub/emu/hostdisk.h>
 #include <grub/emu/getroot.h>
 #include <grub/term.h>
@@ -62,6 +63,7 @@ enum {
   PRINT_DRIVE,
   PRINT_DEVICE,
   PRINT_PARTMAP,
+  PRINT_PARTUUID,
   PRINT_ABSTRACTION,
   PRINT_CRYPTODISK_UUID,
   PRINT_HINT_STR,
@@ -85,6 +87,7 @@ static const char *targets[] =
     [PRINT_DRIVE]              = "drive",
     [PRINT_DEVICE]             = "device",
     [PRINT_PARTMAP]            = "partmap",
+    [PRINT_PARTUUID]           = "partuuid",
     [PRINT_ABSTRACTION]        = "abstraction",
     [PRINT_CRYPTODISK_UUID]    = "cryptodisk_uuid",
     [PRINT_HINT_STR]           = "hints_string",
@@ -181,6 +184,43 @@ probe_partmap (grub_disk_t disk, char delim)
     }
 }
 
+static void
+probe_partuuid (grub_disk_t disk, char delim)
+{
+  grub_partition_t p = disk->partition;
+
+  /*
+   * Nested partitions not supported for now.
+   * Non-nested partitions must have disk->partition->parent == NULL
+   */
+  if (p && p->parent == NULL)
+    {
+      disk->partition = p->parent;
+      if (strcmp(p->partmap->name, "msdos") == 0)
+       {
+           /*
+            * The partition GUID for MSDOS is the partition number (starting
+            * with 1) prepended with the NT disk signature.
+            */
+           grub_uint32_t nt_disk_sig;
+
+           if (grub_disk_read (disk, 0, GRUB_BOOT_MACHINE_WINDOWS_NT_MAGIC,
+                               sizeof(nt_disk_sig), &nt_disk_sig) == 0)
+             grub_printf ("%08x-%02x",
+                          grub_le_to_cpu32(nt_disk_sig), 1 + p->number);
+       }
+      else if (strcmp(p->partmap->name, "gpt") == 0)
+       {
+         struct grub_gpt_partentry gptdata;
+
+         if (grub_disk_read (disk, p->offset, p->index,
+                             sizeof(gptdata), &gptdata) == 0)
+           print_gpt_guid(gptdata.guid);
+       }
+      disk->partition = p;
+    }
+}
+
 static void
 probe_cryptodisk_uuid (grub_disk_t disk, char delim)
 {
@@ -635,6 +675,12 @@ probe (const char *path, char **device_names, char delim)
        /* Check if dev->disk itself is contained in a partmap.  */
        probe_partmap (dev->disk, delim);
 
+      else if (print == PRINT_PARTUUID)
+       {
+         probe_partuuid (dev->disk, delim);
+         putchar (delim);
+       }
+
       else if (print == PRINT_MSDOS_PARTTYPE)
        {
          if (dev->disk->partition
-- 
2.17.0