* grub-core/loader/i386/coreboot/chainloader.c: Support tianocore.

diff --git a/ChangeLog b/ChangeLog
index 7580e1d8a1d6058bc47f5b406de8c084af38323d..245b2702e1936e18752bfff5a668de1086c73070 100644 (file)
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,7 @@
+2013-11-10  Vladimir Serbinenko  <phcoder@gmail.com>
+
+       * grub-core/loader/i386/coreboot/chainloader.c: Support tianocore.
+
 2013-11-10  Vladimir Serbinenko  <phcoder@gmail.com>
 
        * grub-core/loader/i386/coreboot/chainloader.c: Support lzma-compressed

diff --git a/grub-core/loader/i386/coreboot/chainloader.c b/grub-core/loader/i386/coreboot/chainloader.c
index 8df1fa607312b79f2e0f8cca0b66e89f4c187b35..6e1bd9830cb75c06c06d865ea40443867834fdde 100644 (file)
--- a/grub-core/loader/i386/coreboot/chainloader.c
+++ b/grub-core/loader/i386/coreboot/chainloader.c
@@ -31,6 +31,8 @@
 #include <grub/i18n.h>
 #include <grub/cbfs_core.h>
 #include <grub/lib/LzmaDec.h>
+#include <grub/efi/pe32.h>
+#include <grub/i386/cpuid.h>
 
 GRUB_MOD_LICENSE ("GPLv3+");
 
@@ -197,6 +199,157 @@ load_segment (grub_file_t file, const char *filename,
     }
 }
 
+static grub_err_t
+load_tianocore (grub_file_t file)
+{
+  grub_uint16_t header_length;
+  grub_uint32_t section_head;
+  grub_uint8_t mz[2], pe[4];
+  struct grub_pe32_coff_header coff_head;
+  struct file_header
+  {
+    grub_uint8_t unused[18];
+    grub_uint8_t type;
+    grub_uint8_t unused2;
+    grub_uint8_t size[3];
+    grub_uint8_t unused3;
+  } file_head;
+  grub_relocator_chunk_t ch;
+
+  if (grub_file_seek (file, 48) == (grub_off_t) -1
+      || grub_file_read (file, &header_length, sizeof (header_length))
+      != sizeof (header_length)
+      || grub_file_seek (file, header_length) == (grub_off_t) -1)
+    goto fail;
+
+  while (1)
+    {
+      grub_off_t off;
+      if (grub_file_read (file, &file_head, sizeof (file_head))
+         != sizeof (file_head))
+       goto fail;
+      if (file_head.type != 0xf0)
+       break;
+      off = grub_get_unaligned32 (file_head.size) & 0xffffff;
+      if (off < sizeof (file_head))
+       goto fail;
+      if (grub_file_seek (file, grub_file_tell (file) + off
+                         - sizeof (file_head)) == (grub_off_t) -1)
+       goto fail;
+    }
+
+  if (file_head.type != 0x03)
+    goto fail;
+
+  while (1)
+    {
+      if (grub_file_read (file, &section_head, sizeof (section_head))
+         != sizeof (section_head))
+       goto fail;
+      if ((section_head >> 24) != 0x19)
+       break;
+
+      if ((section_head & 0xffffff) < sizeof (section_head))
+       goto fail;
+
+      if (grub_file_seek (file, grub_file_tell (file)
+                         + (section_head & 0xffffff)
+                         - sizeof (section_head)) == (grub_off_t) -1)
+       goto fail;
+    }
+
+  if ((section_head >> 24) != 0x10)
+    goto fail;
+
+  grub_off_t exe_start = grub_file_tell (file);
+
+  if (grub_file_read (file, &mz, sizeof (mz)) != sizeof (mz))
+    goto fail;
+  if (mz[0] != 'M' || mz[1] != 'Z')
+    goto fail;
+
+  if (grub_file_seek (file, grub_file_tell (file) + 0x3a) == (grub_off_t) -1)
+    goto fail;
+
+  if (grub_file_read (file, &section_head, sizeof (section_head))
+      != sizeof (section_head))
+    goto fail;
+  if (section_head < 0x40)
+    goto fail;
+
+  if (grub_file_seek (file, grub_file_tell (file)
+                     + section_head - 0x40) == (grub_off_t) -1)
+    goto fail;
+
+  if (grub_file_read (file, &pe, sizeof (pe))
+      != sizeof (pe))
+    goto fail;
+
+  if (pe[0] != 'P' || pe[1] != 'E' || pe[2] != '\0' || pe[3] != '\0')
+    goto fail;
+
+  if (grub_file_read (file, &coff_head, sizeof (coff_head))
+      != sizeof (coff_head))
+    goto fail;
+
+  grub_uint32_t loadaddr;
+
+  switch (coff_head.machine)
+    {
+    case GRUB_PE32_MACHINE_I386:
+      {
+       struct grub_pe32_optional_header oh;
+       if (grub_file_read (file, &oh, sizeof (oh))
+           != sizeof (oh))
+         goto fail;
+       if (oh.magic != GRUB_PE32_PE32_MAGIC)
+         goto fail;
+       loadaddr = oh.image_base - exe_start;
+       entry = oh.image_base + oh.entry_addr;
+       break;
+      }
+    case GRUB_PE32_MACHINE_X86_64:
+      {
+       struct grub_pe64_optional_header oh;
+       if (! grub_cpuid_has_longmode)
+         {
+           grub_error (GRUB_ERR_BAD_OS, "your CPU does not implement AMD64 architecture");
+           goto fail;
+         }
+
+       if (grub_file_read (file, &oh, sizeof (oh))
+           != sizeof (oh))
+         goto fail;
+       if (oh.magic != GRUB_PE32_PE64_MAGIC)
+         goto fail;
+       loadaddr = oh.image_base - exe_start;
+       entry = oh.image_base + oh.entry_addr;
+       break;
+      }
+    default:
+      goto fail;
+    }
+  if (grub_file_seek (file, 0) == (grub_off_t) -1)
+    goto fail;
+
+  grub_size_t fz = grub_file_size (file);
+
+  if (grub_relocator_alloc_chunk_addr (relocator, &ch,
+                                      loadaddr, fz))
+    goto fail;
+
+  if (grub_file_read (file, get_virtual_current_address (ch), fz)
+      != (grub_ssize_t) fz)
+    goto fail;
+
+  return GRUB_ERR_NONE;
+
+ fail:
+  if (!grub_errno)
+    grub_error (GRUB_ERR_BAD_OS, "fv volume is invalid");
+  return grub_errno;
+}
+
 static grub_err_t
 load_chewed (grub_file_t file, const char *filename)
 {
@@ -320,7 +473,13 @@ grub_cmd_chain (grub_command_t cmd __attribute__ ((unused)),
       break;
 
     default:
-      err = grub_error (GRUB_ERR_BAD_OS, "unrecognised payload type");
+      if (grub_file_seek (file, 40) == (grub_off_t) -1
+         || grub_file_read (file, &head, sizeof (head)) != sizeof (head)
+         || grub_file_seek (file, 0) == (grub_off_t) -1
+         || head != 0x4856465f)
+       err = grub_error (GRUB_ERR_BAD_OS, "unrecognised payload type");
+      else
+       err = load_tianocore (file);
       break;
     }
   grub_file_close (file);