A Unified Kernel Image is a single UEFI PE file that combines a UEFI boot stub,
a Linux kernel image, an initrd, and further resources. The uki command will
locate where the uki file is and create a GRUB menu entry to load it.

Signed-off-by: Alec Brown <alec.r.br...@oracle.com>
---
 docs/grub.texi              |  26 +++
 grub-core/commands/blsuki.c | 415 ++++++++++++++++++++++++++++++++++--
 include/grub/menu.h         |   2 +
 3 files changed, 428 insertions(+), 15 deletions(-)

diff --git a/docs/grub.texi b/docs/grub.texi
index 19b0cc024..9317b4130 100644
--- a/docs/grub.texi
+++ b/docs/grub.texi
@@ -6491,6 +6491,7 @@ you forget a command, you can run the command 
@command{help}
 * tpm2_key_protector_clear::    Clear the TPM2 key protector
 * true::                        Do nothing, successfully
 * trust::                       Add public key to list of trusted keys
+* uki::                         Load Unified Kernel Image menu entries
 * unset::                       Unset an environment variable
 @comment * vbeinfo::                     List available video modes
 * verify_detached::             Verify detached digital signature
@@ -8164,6 +8165,31 @@ Unset the environment variable @var{envvar}.
 @end deffn
 
 
+@node uki
+@subsection uki
+
+@deffn Command uki [@option{--path} dir] [@option{--show-default}] 
[@option{--show-non-default}] [@option{--entry} file]
+Load Unified Kernel Image entries into the GRUB menu.
+
+The @option{--path} option overrides the default path to the directory 
containing
+the UKI entries. If this option isn't used, the default location is
+/EFI/Linux in the EFI system partition.
+
+The @option{--show-default} option allows the default boot entry to be added 
to the
+GRUB menu from the UKI entries.
+
+The @option{--show-non-default} option allows non-default boot entries to be 
added to
+the GRUB menu from the UKI entries.
+
+The @option{--entry} option allows specific boot entries to be added to the 
GRUB menu
+from the UKI entries.
+
+The @option{--entry}, @option{--show-default}, and @option{--show-non-default} 
options
+are used to filter which UKI entries are added to the GRUB menu. If none are
+used, all entries in the default location or the location specified by 
@option{--path}
+will be added to the GRUB menu.
+@end deffn
+
 @ignore
 @node vbeinfo
 @subsection vbeinfo
diff --git a/grub-core/commands/blsuki.c b/grub-core/commands/blsuki.c
index 12a9a1ed1..bf284e002 100644
--- a/grub-core/commands/blsuki.c
+++ b/grub-core/commands/blsuki.c
@@ -39,9 +39,21 @@
 #define GRUB_BOOT_DEVICE ""
 #endif
 
+#ifdef GRUB_MACHINE_EFI
+#include <grub/efi/efi.h>
+#include <grub/efi/disk.h>
+#include <grub/efi/pe32.h>
+#endif
+
 GRUB_MOD_LICENSE ("GPLv3+");
 
 #define GRUB_BLS_CONFIG_PATH "/loader/entries/"
+#define GRUB_UKI_CONFIG_PATH "/EFI/Linux"
+
+#define GRUB_BLS_CMD 1
+#define GRUB_UKI_CMD 2
+
+static int cmd_type = 0;
 
 static const struct grub_arg_option bls_opt[] =
   {
@@ -52,6 +64,17 @@ static const struct grub_arg_option bls_opt[] =
     {0, 0, 0, 0, 0, 0}
   };
 
+#ifdef GRUB_MACHINE_EFI
+static const struct grub_arg_option uki_opt[] =
+  {
+    {"path", 'p', 0, N_("Specify path to find UKI entries."), N_("DIR"), 
ARG_TYPE_PATHNAME},
+    {"show-default", 'd', 0, N_("Allow the default UKI entry to be added to 
the GRUB menu."), 0, ARG_TYPE_NONE},
+    {"show-non-default", 'n', 0, N_("Allow the non-default UKI entries to be 
added to the GRUB menu."), 0, ARG_TYPE_NONE},
+    {"entry", 'e', 0, N_("Allow specificUKII entries to be added to the GRUB 
menu."), N_("FILE"), ARG_TYPE_FILE},
+    {0, 0, 0, 0, 0, 0}
+  };
+#endif
+
 struct keyval
 {
   const char *key;
@@ -288,6 +311,206 @@ bls_read_entry (grub_file_t f, grub_blsuki_entry_t *entry)
   return err;
 }
 
+#ifdef GRUB_MACHINE_EFI
+static grub_err_t
+uki_read_entry (grub_file_t f, grub_blsuki_entry_t *entry)
+{
+  struct grub_msdos_image_header *dos = NULL;
+  struct grub_pe_image_header *pe = NULL;
+  grub_off_t section_offset = 0;
+  struct grub_pe32_section_table *section = NULL;
+  struct grub_pe32_coff_header *coff_header = NULL;
+  char *val = NULL;
+  char *key = NULL;
+  const char *target[] = {".cmdline", ".osrel", ".linux", NULL};
+  bool has_linux = false;
+  grub_err_t err = GRUB_ERR_NONE;
+
+  dos = grub_zalloc (sizeof (*dos));
+  if (dos == NULL)
+    return grub_errno;
+  if (grub_file_read (f, dos, sizeof (*dos)) < (grub_ssize_t) sizeof (*dos))
+    {
+      err = grub_error (GRUB_ERR_FILE_READ_ERROR, "failed to read UKI image 
header");
+      goto fail;
+    }
+  if (dos->msdos_magic != GRUB_PE32_MAGIC)
+    {
+      err = grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET, N_("plain image kernel 
not supported"));
+      goto fail;
+    }
+
+  grub_dprintf ("blsuki", "PE/COFF header @ %08x\n", 
dos->pe_image_header_offset);
+  pe = grub_zalloc (sizeof (*pe));
+  if (pe == NULL)
+    {
+      err = grub_errno;
+      goto fail;
+    }
+  if (grub_file_seek (f, dos->pe_image_header_offset) == (grub_off_t) -1
+      || grub_file_read (f, pe, sizeof (*pe)) != sizeof (*pe))
+    {
+      err = grub_error (GRUB_ERR_FILE_READ_ERROR, "failed to read COFF image 
header");
+      goto fail;
+    }
+  if (pe->optional_header.magic != GRUB_PE32_NATIVE_MAGIC)
+    {
+      err = grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET, "non-native image not 
supported");
+      goto fail;
+    }
+
+  coff_header = &(pe->coff_header);
+  section_offset = dos->pe_image_header_offset + sizeof (*pe);
+
+  for (int i = 0; i < coff_header->num_sections; i++)
+    {
+      key = NULL;
+      val = NULL;
+      section = grub_zalloc (sizeof (*section));
+      if (section == NULL)
+       {
+         err = grub_errno;
+         goto fail;
+       }
+
+      grub_file_seek (f, section_offset);
+      if (grub_file_read (f, section, sizeof (*section)) != sizeof (*section))
+       {
+         err = grub_error (GRUB_ERR_FILE_READ_ERROR, "failed to read section 
header");
+         goto fail;
+       }
+
+      key = grub_strndup (section->name, 8);
+      if (key == NULL)
+       {
+         err = grub_errno;
+         goto fail;
+       }
+
+      for (int j = 0; target[j] != NULL; j++)
+       {
+         if (grub_strcmp (key, target[j]) == 0)
+           {
+             /*
+              * We don't need to read the contents of the .linux PE section, 
but we
+              * should verify that the section exists.
+              */
+             if (grub_strcmp (key, ".linux") == 0)
+               {
+                 has_linux = true;
+                 break;
+               }
+
+             val = grub_zalloc (section->raw_data_size);
+             if (val == NULL)
+               {
+                 err = grub_errno;
+                 goto fail;
+               }
+
+             grub_file_seek (f, section->raw_data_offset);
+             if (grub_file_read (f, val, section->raw_data_size) != 
(grub_ssize_t) section->raw_data_size)
+               {
+                 err = grub_error (GRUB_ERR_FILE_READ_ERROR, "failed to read 
section");
+                 goto fail;
+               }
+
+             err = blsuki_add_keyval (entry, key, val);
+             if (err != GRUB_ERR_NONE)
+               goto fail;
+
+             break;
+           }
+       }
+
+      section_offset += sizeof (*section);
+      grub_free (section);
+      grub_free (val);
+      grub_free (key);
+    }
+
+  if (has_linux == false)
+    err = grub_error (GRUB_ERR_NO_KERNEL, "UKI is missing the '.linux' 
section");
+
+  grub_free (dos);
+  grub_free (pe);
+  return err;
+
+ fail:
+  grub_free (dos);
+  grub_free (pe);
+  grub_free (section);
+  grub_free (val);
+  grub_free (key);
+  return err;
+}
+#endif
+
+static char *
+uki_read_osrel (char *content, grub_off_t *pos, char **key_ret, char **val_ret)
+{
+  char *line;
+  char *value;
+  grub_size_t linelen;
+
+ skip:
+  line = content + *pos;
+  if (*line == '\0')
+    return NULL;
+
+  linelen = 0;
+  while (line[linelen] != '\0' && !grub_strchr ("\n\r", line[linelen]))
+    linelen++;
+
+  /* Move pos to the next line */
+  *pos += linelen;
+  if (content[*pos] != '\0')
+    (*pos)++;
+
+  /* Skip empty line */
+  if (linelen == 0)
+    goto skip;
+
+  line[linelen] = '\0';
+
+  /* Remove leading white space */
+  while (grub_strchr (" \t", *line))
+    {
+      line++;
+      linelen--;
+    }
+
+  /* Remove trailing whitespace */
+  while (linelen > 0 && grub_strchr ("=", line[linelen - 1]))
+    linelen--;
+  line[linelen] = '\0';
+
+  if (*line == '#')
+    goto skip;
+
+  /* Split key/value */
+  value = line;
+  while (*value != '\0' && !grub_strchr ("=", *value))
+    value++;
+  if (*value == '\0')
+    goto skip;
+  *value = '\0';
+  value++;
+  while (*value != '\0' && grub_strchr ("=", *value))
+    value++;
+
+  /* Remove quotes from value */
+  if (value[0] == '\"' && line[linelen - 1] == '\"')
+    {
+      value++;
+      line[linelen - 1] = '\0';
+    }
+
+  *key_ret = line;
+  *val_ret = value;
+  return line;
+}
+
 struct read_entry_info
 {
   const char *devid;
@@ -301,9 +524,11 @@ blsuki_read_entry (const char *filename,
                   void *data)
 {
   grub_size_t m = 0, n, ext_len = 5;
-  grub_err_t err;
+  grub_err_t err = GRUB_ERR_NONE;
   char *p = NULL;
+  const char *ext = NULL;
   grub_file_t f = NULL;
+  enum grub_file_type file_type = 0;
   grub_blsuki_entry_t *entry;
   struct read_entry_info *info = (struct read_entry_info *) data;
 
@@ -311,6 +536,18 @@ blsuki_read_entry (const char *filename,
 
   n = grub_strlen (filename);
 
+  if (cmd_type == GRUB_BLS_CMD)
+    {
+      ext = ".conf";
+      file_type = GRUB_FILE_TYPE_CONFIG;
+    }
+  else if (cmd_type == GRUB_UKI_CMD)
+    {
+      ext = ".efi";
+      file_type = GRUB_FILE_TYPE_EFI_CHAINLOADED_IMAGE;
+    }
+  ext_len = grub_strlen (ext);
+
   if (info->file != NULL)
     {
       f = info->file;
@@ -320,15 +557,14 @@ blsuki_read_entry (const char *filename,
       if (filename[0] == '.')
        return 0;
 
-      if (n <= 5)
+      if (n <= ext_len)
        return 0;
 
-      if (grub_strcmp (filename + n - ext_len, ".conf") != 0)
+      if (grub_strcmp (filename + n - ext_len, ext) != 0)
        return 0;
 
       p = grub_xasprintf ("(%s)%s/%s", info->devid, info->dirname, filename);
-
-      f = grub_file_open (p, GRUB_FILE_TYPE_CONFIG);
+      f = grub_file_open (p, file_type);
       grub_free (p);
       if (f == NULL)
        goto finish;
@@ -365,7 +601,26 @@ blsuki_read_entry (const char *filename,
       goto finish;
     }
 
-  err = bls_read_entry (f, entry);
+  entry->dirname = grub_strdup (info->dirname);
+  if (entry->dirname == NULL)
+    {
+      grub_free (entry);
+      goto finish;
+    }
+
+  entry->devid = grub_strdup (info->devid);
+  if (entry->devid == NULL)
+    {
+      grub_free (entry);
+      goto finish;
+    }
+
+  if (cmd_type == GRUB_BLS_CMD)
+    err = bls_read_entry (f, entry);
+#ifdef GRUB_MACHINE_EFI
+  else if (cmd_type == GRUB_UKI_CMD)
+    err = uki_read_entry (f, entry);
+#endif
 
   if (err == GRUB_ERR_NONE)
     blsuki_add_entry (entry);
@@ -810,6 +1065,75 @@ bls_create_entry (grub_blsuki_entry_t *entry)
   grub_free (src);
 }
 
+static void
+uki_create_entry (grub_blsuki_entry_t *entry)
+{
+  int argc = 0;
+  const char **argv = NULL;
+  char *id = entry->filename;
+  char *title = NULL;
+  char *options = NULL;
+  char *osrel = NULL;
+  char *line;
+  char *key = NULL;
+  char *value = NULL;
+  char *src = NULL;
+  grub_size_t size = 0;
+  grub_off_t pos = 0;
+
+  options = blsuki_get_val (entry, ".cmdline", NULL);
+  if (options == NULL)
+    {
+      grub_dprintf ("blsuki", "Skipping file %s with no '.cmdline' key.\n", 
entry->filename);
+      goto finish;
+    }
+  osrel = blsuki_get_val (entry, ".osrel", NULL);
+  if (osrel == NULL)
+    {
+      grub_dprintf ("blsuki", "Skipping file %s with no '.osrel' key.\n", 
entry->filename);
+      goto finish;
+    }
+
+  line = osrel;
+  while ((line = uki_read_osrel (osrel, &pos, &key, &value)))
+    {
+      if (grub_strcmp ("PRETTY_NAME", key) == 0)
+       {
+         title = value;
+         break;
+       }
+    }
+
+  argc += 1;
+  if (grub_mul (argc + 1, sizeof (char *), &size))
+    {
+      grub_error (GRUB_ERR_OUT_OF_RANGE, N_("overflow detected creating argv 
list"));
+      goto finish;
+    }
+  argv = grub_malloc (size);
+  if (argv == NULL)
+    {
+      grub_error (GRUB_ERR_OUT_OF_MEMORY, "failed to allocate argv list");
+      goto finish;
+    }
+  argv[0] = title;
+  argv[argc] = NULL;
+
+  src = grub_xasprintf ("insmod chain\n"
+                       "chainloader (%s)%s/%s%s%s\n",
+                       entry->devid, entry->dirname,
+                       entry->filename, options ? " " : "", options ? options 
: "");
+
+
+  grub_normal_add_menu_entry (argc, argv, NULL, id, NULL, NULL, NULL, src, 0, 
entry);
+
+ finish:
+  grub_free (argv);
+  grub_free (src);
+  grub_free (options);
+  grub_free (osrel);
+}
+
 struct find_entry_info
 {
   const char *dirname;
@@ -830,7 +1154,12 @@ blsuki_find_entry (struct find_entry_info *info)
   int r = 0;
 
   if (dir == NULL)
-    dir = GRUB_BLS_CONFIG_PATH;
+    {
+      if (cmd_type == GRUB_BLS_CMD)
+       dir = GRUB_BLS_CONFIG_PATH;
+      else if (cmd_type == GRUB_UKI_CMD)
+       dir = GRUB_UKI_CONFIG_PATH;
+    }
 
   read_entry_info.file = NULL;
   read_entry_info.dirname = dir;
@@ -853,11 +1182,17 @@ blsuki_find_entry (struct find_entry_info *info)
   /*
    * If we aren't able to find BLS entries in the directory given by 
info->dirname,
    * we can fallback to the default location "/boot/loader/entries/" and see 
if we
-   * can find the files there.
+   * can find the files there. If we can't find UKI entries, fallback to
+   * "/boot/efi/EFI/Linux".
    */
   if (r != 0 && info->dirname == NULL && fallback == 0)
     {
-      read_entry_info.dirname = "/boot" GRUB_BLS_CONFIG_PATH;
+      if (cmd_type == GRUB_BLS_CMD)
+       read_entry_info.dirname = "/boot" GRUB_BLS_CONFIG_PATH;
+      else if (cmd_type == GRUB_UKI_CMD)
+       {
+         read_entry_info.dirname = GRUB_UKI_CONFIG_PATH;
+       }
       grub_dprintf ("blsuki", "Entries weren't found in %s, fallback to %s\n",
                    dir, read_entry_info.dirname);
       fallback = 1;
@@ -869,11 +1204,13 @@ static grub_err_t
 blsuki_load_entries (char *path)
 {
   grub_size_t len;
+  grub_size_t ext_len;
   grub_fs_t fs;
   grub_device_t dev;
   static grub_err_t r;
   const char *devid = NULL;
   char *dir = NULL;
+  const char *ext = NULL;
   struct find_entry_info info = {
       .dev = NULL,
       .fs = NULL,
@@ -886,8 +1223,14 @@ blsuki_load_entries (char *path)
 
   if (path != NULL)
     {
+      if (cmd_type == GRUB_BLS_CMD)
+       ext = ".conf";
+      else if (cmd_type == GRUB_UKI_CMD)
+       ext = ".efi";
+
       len = grub_strlen (path);
-      if (grub_strcmp (path + len - 5, ".conf") == 0)
+      ext_len = grub_strlen (ext);
+      if (grub_strcmp (path + len - ext_len, ext) == 0)
        {
          rei.file = grub_file_open (path, GRUB_FILE_TYPE_CONFIG);
          if (rei.file == NULL)
@@ -916,11 +1259,22 @@ blsuki_load_entries (char *path)
 
   if (devid == NULL)
     {
+      if (cmd_type == GRUB_BLS_CMD)
+       {
 #ifdef GRUB_MACHINE_EMU
-      devid = "host";
+         devid = "host";
 #else
-      devid = grub_env_get ("root");
+         devid = grub_env_get ("root");
 #endif
+       }
+      else if (cmd_type == GRUB_UKI_CMD)
+       {
+#ifdef GRUB_MACHINE_EFI
+         grub_efi_loaded_image_t *image;
+         image = grub_efi_get_loaded_image (grub_efi_image_handle);
+         devid = grub_efidisk_get_device_name (image->device_handle);
+#endif
+       }
       if (devid == NULL)
        return grub_error (GRUB_ERR_FILE_NOT_FOUND, N_("variable `%s' isn't 
set"), "root");
     }
@@ -1001,7 +1355,10 @@ blsuki_create_entries (bool show_default, bool 
show_non_default, char *entry_id)
          (show_non_default == true && blsuki_is_default_entry (def_entry, 
entry, idx) == false) ||
          (entry_id != NULL && grub_strcmp (entry_id, entry->filename) == 0))
        {
-         bls_create_entry (entry);
+         if (cmd_type == GRUB_BLS_CMD)
+           bls_create_entry (entry);
+         else if (cmd_type == GRUB_UKI_CMD)
+           uki_create_entry (entry);
          entry->visible = 1;
        }
       idx++;
@@ -1011,8 +1368,7 @@ blsuki_create_entries (bool show_default, bool 
show_non_default, char *entry_id)
 }
 
 static grub_err_t
-grub_cmd_blscfg (grub_extcmd_context_t ctxt, int argc __attribute__ ((unused)),
-                char **args __attribute__ ((unused)))
+blsuki_cmd (grub_extcmd_context_t ctxt)
 {
   grub_err_t err;
   struct grub_arg_list *state = ctxt->state;
@@ -1021,6 +1377,7 @@ grub_cmd_blscfg (grub_extcmd_context_t ctxt, int argc 
__attribute__ ((unused)),
   bool show_default = false;
   bool show_non_default = false;
   bool all = true;
+  entries = NULL;
 
   if (state[0].set)
     path = state[0].arg;
@@ -1052,17 +1409,45 @@ grub_cmd_blscfg (grub_extcmd_context_t ctxt, int argc 
__attribute__ ((unused)),
   return blsuki_create_entries (show_default, show_non_default, entry_id);
 }
 
+static grub_err_t
+grub_cmd_blscfg (grub_extcmd_context_t ctxt, int argc __attribute__ ((unused)),
+                char **args __attribute__ ((unused)))
+{
+  cmd_type = GRUB_BLS_CMD;
+  return blsuki_cmd (ctxt);
+}
+
 static grub_extcmd_t bls_cmd;
 
+#ifdef GRUB_MACHINE_EFI
+static grub_err_t
+grub_cmd_uki (grub_extcmd_context_t ctxt, int argc __attribute__ ((unused)),
+                char **args __attribute__ ((unused)))
+{
+  cmd_type = GRUB_UKI_CMD;
+  return blsuki_cmd (ctxt);
+}
+
+static grub_extcmd_t uki_cmd;
+#endif
+
 GRUB_MOD_INIT(blsuki)
 {
   bls_cmd = grub_register_extcmd ("blscfg", grub_cmd_blscfg, 0,
                                  N_("[-p|--path] DIR [-d|--show-default] 
[-n|--show-non-default] [-e|--entry] FILE"),
                                  N_("Import Boot Loader Specification 
snippets."),
                                  bls_opt);
+#ifdef GRUB_MACHINE_EFI
+  uki_cmd = grub_register_extcmd ("uki", grub_cmd_uki, 0,
+                                 N_("[-p|--path] DIR [-d|--show-default] 
[-n|--show-non-default] [-e|--entry] FILE"),
+                                 N_("Import Unified Kernel Images"), uki_opt);
+#endif
 }
 
 GRUB_MOD_FINI(blsuki)
 {
   grub_unregister_extcmd (bls_cmd);
+#ifdef GRUB_MACHINE_EFI
+  grub_unregister_extcmd (uki_cmd);
+#endif
 }
diff --git a/include/grub/menu.h b/include/grub/menu.h
index c25a0d16d..907373625 100644
--- a/include/grub/menu.h
+++ b/include/grub/menu.h
@@ -28,6 +28,8 @@ struct grub_blsuki_entry
   grub_size_t keyvals_size;
   int nkeyvals;
   char *filename;
+  char *dirname;
+  char *devid;
   int visible;
 };
 typedef struct grub_blsuki_entry grub_blsuki_entry_t;
-- 
2.27.0


_______________________________________________
Grub-devel mailing list
Grub-devel@gnu.org
https://lists.gnu.org/mailman/listinfo/grub-devel

Reply via email to