[PATCH v3 2/2] devmapper/getroot: Set up cheated LUKS2 cryptodisk mount from DM parameters

[Josselin Poiret]

This lets a LUKS2 cryptodisk have all the cipher, hash, and sizes
filled out, otherwise they wouldn't be initialized if cheat mounted.
---
 grub-core/osdep/devmapper/getroot.c | 130 +++++++++++++++++++++++++++-
 1 file changed, 129 insertions(+), 1 deletion(-)

diff --git a/grub-core/osdep/devmapper/getroot.c 
b/grub-core/osdep/devmapper/getroot.c
index ad1daf9c8..b8d66029a 100644
--- a/grub-core/osdep/devmapper/getroot.c
+++ b/grub-core/osdep/devmapper/getroot.c
@@ -43,6 +43,8 @@
 #include <grub/osdep/major.h>
 #include <libdevmapper.h>
 
+#define DM_LOG_SECTOR_SIZE 9
+
 #include <grub/types.h>
 #include <grub/util/misc.h>
 
@@ -51,6 +53,8 @@
 #include <grub/emu/misc.h>
 #include <grub/emu/hostdisk.h>
 
+#include <grub/cryptodisk.h>
+
 static int
 grub_util_open_dm (const char *os_dev, struct dm_tree **tree,
                   struct dm_tree_node **node)
@@ -183,7 +187,6 @@ grub_util_pull_devmapper (const char *os_dev)
       && lastsubdev)
     {
       char *grdev = grub_util_get_grub_dev (lastsubdev);
-      dm_tree_free (tree);
       if (grdev)
        {
          grub_err_t err;
@@ -191,7 +194,132 @@ grub_util_pull_devmapper (const char *os_dev)
          if (err)
            grub_util_error (_("can't mount encrypted volume `%s': %s"),
                             lastsubdev, grub_errmsg);
+          if (strncmp (uuid, "CRYPT-LUKS2-", sizeof ("CRYPT-LUKS2-") - 1) == 0)
+            {
+              /* set LUKS2 cipher and size from dm parameters, since it is not
+               * possible to determine the correct ones without unlocking, as
+               * there might be multiple segments.
+               */
+              grub_disk_t source;
+              grub_cryptodisk_t cryptodisk;
+              grub_addr_t start, length;
+              char *target_type;
+              char *params;
+              const char *name;
+              char *cipher, *cipher_mode;
+              struct dm_task *dmt;
+
+              source = grub_disk_open (grdev);
+              cryptodisk = grub_cryptodisk_get_by_source_disk (source);
+              grub_disk_close (source);
+
+              name = dm_tree_node_get_name (node);
+
+              grub_util_info ("populating parameters of cryptomount `%s' from 
DM device `%s'",
+                              uuid, name);
+
+              dmt = dm_task_create (DM_DEVICE_TABLE);
+              if (dmt == 0)
+                grub_util_error (_("can't create dm task DM_DEVICE_TABLE"));
+              if (dm_task_set_name (dmt, name) == 0)
+                grub_util_error (_("can't set dm task name to `%s'"), name);
+              if (dm_task_run (dmt) == 0)
+                grub_util_error (_("can't run dm task for `%s'"), name);
+              /* dm_get_next_target doesn't have any error modes, everything 
has
+               * been handled by dm_task_run.
+               */
+              dm_get_next_target (dmt, NULL, &start, &length,
+                                  &target_type, &params);
+              if (strncmp (target_type, "crypt", sizeof ("crypt")) != 0)
+                grub_util_error (_("dm target of type `%s' is not `crypt'"),
+                                 target_type);
+
+              cryptodisk->total_sectors = length;
+              cryptodisk->log_sector_size = DM_LOG_SECTOR_SIZE;
+
+              /* dm target parameters for dm-crypt is
+               * <cipher> <key> <iv_offset> <device path> <offset> 
[<#opt_params> <opt_param1> ...]
+               */
+              char *seek_head, *c;
+              unsigned int remaining;
+              c = params;
+              remaining = grub_strlen (c);
+
+              /* first, get the cipher name from the cipher */
+              if (!(seek_head = grub_memchr (c, '-', remaining)))
+                grub_util_error (_("can't get cipher from dm-crypt parameters 
`%s'"),
+                                 params);
+              cipher = grub_strndup (c, seek_head - c);
+              remaining -= seek_head - c;
+              c = seek_head + 1;
+
+              /* now, the cipher mode */
+              if (!(seek_head = grub_memchr (c, ' ', remaining)))
+                grub_util_error (_("can't get cipher mode from dm-crypt 
parameters `%s'"),
+                                 params);
+              cipher_mode = grub_strndup (c, seek_head - c);
+              remaining -= seek_head - c;
+              c = seek_head + 1;
+
+              grub_cryptodisk_setcipher (cryptodisk, cipher, cipher_mode);
+
+              grub_free (cipher);
+              grub_free (cipher_mode);
+
+              /* This is the only hash usable by PBKDF2, so set it as such */
+              cryptodisk->hash = grub_crypto_lookup_md_by_name ("sha256");
+
+              /* drop key, iv_offset, device path and offset */
+              for (char dropped_tokens = 0; dropped_tokens < 4; 
dropped_tokens++)
+                {
+                  seek_head = grub_memchr (c, ' ', remaining);
+                  remaining -= seek_head - c;
+                  c = seek_head + 1;
+                }
+
+              /* if we have optional parameters, skip #opt_params */
+              if (seek_head && (seek_head = grub_memchr (c, ' ', remaining)))
+                {
+                  remaining -= seek_head - c;
+                  c = seek_head + 1;
+                  while (remaining > 0)
+                    {
+                      if (strncmp (c, "sector_size:", sizeof ("sector_size:") 
- 1) == 0)
+                        {
+                          char *sector_size_str;
+                          unsigned long long sector_size;
+                          c += sizeof ("sector_size:") - 1;
+                          remaining -= sizeof ("sector_size:") - 1;
+                          seek_head = grub_memchr (c, ' ', remaining);
+                          if (seek_head)
+                            sector_size_str = grub_strndup (c, seek_head - c);
+                          else
+                            sector_size_str = grub_strndup (c, remaining);
+                          sector_size = grub_strtoull (sector_size_str, NULL, 
10);
+                          grub_free (sector_size_str);
+
+                          if (sector_size != 512 && sector_size != 1024 &&
+                              sector_size != 2048 && sector_size != 4096)
+                            grub_util_error (_("dm-crypt parameter sector_size 
`%s' is not a valid LUKS2 sector size"),
+                                             sector_size_str);
+                          cryptodisk->log_sector_size = grub_log2ull 
(sector_size);
+                          cryptodisk->total_sectors = 
cryptodisk->total_sectors /
+                            (sector_size / (1 << DM_LOG_SECTOR_SIZE));
+                          grub_util_info ("set sector_size for cryptodisk `%s' 
to `%d', total_sectors to `%lu'",
+                                          name, 1 << 
cryptodisk->log_sector_size,
+                                          cryptodisk->total_sectors);
+                          break;
+                        }
+                      seek_head = grub_memchr (c, ' ', remaining);
+                      remaining -= seek_head - c;
+                      c = seek_head + 1;
+                    }
+                }
+
+              dm_task_destroy (dmt);
+            }
        }
+      dm_tree_free (tree);
       grub_free (grdev);
     }
   else
-- 
2.36.0