Support grub-probe -t drive

diff --git a/grub-core/kern/emu/getroot.c b/grub-core/kern/emu/getroot.c
index f47203cfc3efac61a16d4e77deb387d37d52b9d8..40cbc4998fa817e99a984fb1fb3739f2f746c1d2 100644 (file)
--- a/grub-core/kern/emu/getroot.c
+++ b/grub-core/kern/emu/getroot.c
@@ -640,11 +640,11 @@ grub_guess_root_device (const char *dir)
   return os_dev;
 }
 
-static int
-grub_util_is_lvm (const char *os_dev)
+static char *
+get_dm_uuid (const char *os_dev)
 {
   if ((strncmp ("/dev/mapper/", os_dev, 12) != 0))
-    return 0;
+    return NULL;
   
 #ifdef HAVE_DEVICE_MAPPER
   {
@@ -652,17 +652,18 @@ grub_util_is_lvm (const char *os_dev)
     uint32_t maj, min;
     struct dm_tree_node *node = NULL;
     const char *node_uuid;
+    char *ret;
     struct stat st;
 
     if (stat (os_dev, &st) < 0)
-      return 0;
+      return NULL;
 
     tree = dm_tree_create ();
     if (! tree)
       {
        grub_printf ("Failed to create tree\n");
        grub_dprintf ("hostdisk", "dm_tree_create failed\n");
-       return 0;
+       return NULL;
       }
 
     maj = major (st.st_rdev);
@@ -672,7 +673,7 @@ grub_util_is_lvm (const char *os_dev)
       {
        grub_dprintf ("hostdisk", "dm_tree_add_dev failed\n");
        dm_tree_free (tree);
-       return 0;
+       return NULL;
       }
 
     node = dm_tree_find_node (tree, maj, min);
@@ -680,39 +681,64 @@ grub_util_is_lvm (const char *os_dev)
       {
        grub_dprintf ("hostdisk", "dm_tree_find_node failed\n");
        dm_tree_free (tree);
-       return 0;
+       return NULL;
       }
     node_uuid = dm_tree_node_get_uuid (node);
     if (! node_uuid)
       {
        grub_dprintf ("hostdisk", "%s has no DM uuid\n", os_dev);
        dm_tree_free (tree);
-       return 0;
-      }
-    if (strncmp (node_uuid, "LVM-", 4) != 0)
-      {
-       dm_tree_free (tree);
-       return 0;
+       return NULL;
       }
+
+    ret = grub_strdup (node_uuid);
     dm_tree_free (tree);
-    return 1;
+    return ret;
   }
 #else
-  return 1;
+  return NULL;
 #endif /* HAVE_DEVICE_MAPPER */
 }
 
+static enum grub_dev_abstraction_types
+grub_util_get_dm_abstraction (const char *os_dev)
+{
+  char *uuid;
+  uuid = get_dm_uuid (os_dev);
+
+  if (uuid == NULL)
+    return GRUB_DEV_ABSTRACTION_NONE;
+
+  if (strncmp (uuid, "LVM-", 4) == 0)
+    {
+      grub_free (uuid);
+      return GRUB_DEV_ABSTRACTION_LVM;
+    }
+  if (strncmp (uuid, "CRYPT-LUKS1-", 4) == 0)
+    {
+      grub_free (uuid);
+      return GRUB_DEV_ABSTRACTION_LUKS;
+    }
+
+  grub_free (uuid);
+  return GRUB_DEV_ABSTRACTION_NONE;
+}
+
 int
 grub_util_get_dev_abstraction (const char *os_dev __attribute__((unused)))
 {
 #ifdef __linux__
+  enum grub_dev_abstraction_types ret;
+
   /* User explicitly claims that this drive is visible by BIOS.  */
   if (grub_util_biosdisk_is_present (os_dev))
     return GRUB_DEV_ABSTRACTION_NONE;
 
-  /* Check for LVM.  */
-  if (grub_util_is_lvm (os_dev))
-    return GRUB_DEV_ABSTRACTION_LVM;
+  /* Check for LVM and LUKS.  */
+  ret = grub_util_get_dm_abstraction (os_dev);
+
+  if (ret != GRUB_DEV_ABSTRACTION_NONE)
+    return ret;
 
   /* Check for RAID.  */
   if (!strncmp (os_dev, "/dev/md", 7) && ! grub_util_device_is_mapped (os_dev))
@@ -830,6 +856,19 @@ grub_util_get_grub_dev (const char *os_dev)
 
       break;
 
+    case GRUB_DEV_ABSTRACTION_LUKS:
+      {
+       char *uuid, *dash;
+       uuid = get_dm_uuid (os_dev);
+       dash = grub_strchr (uuid + sizeof ("CRYPT-LUKS1-") - 1, '-');
+       if (dash)
+         *dash = 0;
+       grub_dev = grub_xasprintf ("luksuuid/%s",
+                                  uuid + sizeof ("CRYPT-LUKS1-") - 1);
+       grub_free (uuid);
+      }
+      break;
+
     case GRUB_DEV_ABSTRACTION_RAID:
 
       if (os_dev[7] == '_' && os_dev[8] == 'd')

diff --git a/include/grub/emu/getroot.h b/include/grub/emu/getroot.h
index 581ea8056822caeb710d36292dd9aae26777abcd..14df583b3b3e9ea740a23e00fbeb5dbb97942bac 100644 (file)
--- a/include/grub/emu/getroot.h
+++ b/include/grub/emu/getroot.h
@@ -25,6 +25,7 @@ enum grub_dev_abstraction_types {
   GRUB_DEV_ABSTRACTION_NONE,
   GRUB_DEV_ABSTRACTION_LVM,
   GRUB_DEV_ABSTRACTION_RAID,
+  GRUB_DEV_ABSTRACTION_LUKS,
 };
 
 char *grub_find_device (const char *dir, dev_t dev);