grub-core/loader/multiboot_elfxx.c

   1 /*
   2  *  GRUB  --  GRand Unified Bootloader
   3  *  Copyright (C) 1999,2000,2001,2002,2003,2004,2005,2007,2008,2009  Free Software Foundation, Inc.
   4  *
   5  *  GRUB is free software: you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation, either version 3 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  GRUB is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #if defined(MULTIBOOT_LOAD_ELF32)
  20 # define XX             32
  21 # define E_MACHINE      MULTIBOOT_ELF32_MACHINE
  22 # define ELFCLASSXX     ELFCLASS32
  23 # define Elf_Ehdr       Elf32_Ehdr
  24 # define Elf_Phdr       Elf32_Phdr
  25 # define Elf_Shdr       Elf32_Shdr
  26 #elif defined(MULTIBOOT_LOAD_ELF64)
  27 # define XX             64
  28 # define E_MACHINE      MULTIBOOT_ELF64_MACHINE
  29 # define ELFCLASSXX     ELFCLASS64
  30 # define Elf_Ehdr       Elf64_Ehdr
  31 # define Elf_Phdr       Elf64_Phdr
  32 # define Elf_Shdr       Elf64_Shdr
  33 #else
  34 #error "I'm confused"
  35 #endif
  36
  37 #include <grub/i386/relocator.h>
  38
  39 #define CONCAT(a,b)     CONCAT_(a, b)
  40 #define CONCAT_(a,b)    a ## b
  41
  42 #pragma GCC diagnostic ignored "-Wcast-align"
  43
  44 /* Check if BUFFER contains ELF32 (or ELF64).  */
  45 static int
  46 CONCAT(grub_multiboot_is_elf, XX) (void *buffer)
  47 {
  48   Elf_Ehdr *ehdr = (Elf_Ehdr *) buffer;
  49
  50   return ehdr->e_ident[EI_CLASS] == ELFCLASSXX;
  51 }
  52
  53 static grub_err_t
  54 CONCAT(grub_multiboot_load_elf, XX) (mbi_load_data_t *mld)
  55 {
  56   Elf_Ehdr *ehdr = (Elf_Ehdr *) mld->buffer;
  57   char *phdr_base;
  58   grub_err_t err;
  59   grub_relocator_chunk_t ch;
  60   grub_uint32_t load_offset, load_size;
  61   int i;
  62   void *source;
  63
  64   if (ehdr->e_ident[EI_MAG0] != ELFMAG0
  65       || ehdr->e_ident[EI_MAG1] != ELFMAG1
  66       || ehdr->e_ident[EI_MAG2] != ELFMAG2
  67       || ehdr->e_ident[EI_MAG3] != ELFMAG3
  68       || ehdr->e_ident[EI_DATA] != ELFDATA2LSB)
  69     return grub_error(GRUB_ERR_UNKNOWN_OS, N_("invalid arch-independent ELF magic"));
  70
  71   if (ehdr->e_ident[EI_CLASS] != ELFCLASSXX || ehdr->e_machine != E_MACHINE
  72       || ehdr->e_version != EV_CURRENT)
  73     return grub_error (GRUB_ERR_UNKNOWN_OS, N_("invalid arch-dependent ELF magic"));
  74
  75   if (ehdr->e_type != ET_EXEC && ehdr->e_type != ET_DYN)
  76     return grub_error (GRUB_ERR_UNKNOWN_OS, N_("this ELF file is not of the right type"));
  77
  78   /* FIXME: Should we support program headers at strange locations?  */
  79   if (ehdr->e_phoff + (grub_uint32_t) ehdr->e_phnum * ehdr->e_phentsize > MULTIBOOT_SEARCH)
  80     return grub_error (GRUB_ERR_BAD_OS, "program header at a too high offset");
  81
  82   phdr_base = (char *) mld->buffer + ehdr->e_phoff;
  83 #define phdr(i)                 ((Elf_Phdr *) (phdr_base + (i) * ehdr->e_phentsize))
  84
  85   mld->link_base_addr = ~0;
  86
  87   /* Calculate lowest and highest load address.  */
  88   for (i = 0; i < ehdr->e_phnum; i++)
  89     if (phdr(i)->p_type == PT_LOAD)
  90       {
  91         mld->link_base_addr = grub_min (mld->link_base_addr, phdr(i)->p_paddr);
  92         highest_load = grub_max (highest_load, phdr(i)->p_paddr + phdr(i)->p_memsz);
  93       }
  94
  95 #ifdef MULTIBOOT_LOAD_ELF64
  96   if (highest_load >= 0x100000000)
  97     return grub_error (GRUB_ERR_BAD_OS, "segment crosses 4 GiB border");
  98 #endif
  99
 100   load_size = highest_load - mld->link_base_addr;
 101
 102   if (mld->relocatable)
 103     {
 104       if (load_size > mld->max_addr || mld->min_addr > mld->max_addr - load_size)
 105         return grub_error (GRUB_ERR_BAD_OS, "invalid min/max address and/or load size");
 106
 107       err = grub_relocator_alloc_chunk_align (grub_multiboot_relocator, &ch,
 108                                               mld->min_addr, mld->max_addr - load_size,
 109                                               load_size, mld->align ? mld->align : 1,
 110                                               mld->preference, mld->avoid_efi_boot_services);
 111     }
 112   else
 113     err = grub_relocator_alloc_chunk_addr (grub_multiboot_relocator, &ch,
 114                                            mld->link_base_addr, load_size);
 115
 116   if (err)
 117     {
 118       grub_dprintf ("multiboot_loader", "Cannot allocate memory for OS image\n");
 119       return err;
 120     }
 121
 122   mld->load_base_addr = get_physical_target_address (ch);
 123   source = get_virtual_current_address (ch);
 124
 125   grub_dprintf ("multiboot_loader", "link_base_addr=0x%x, load_base_addr=0x%x, "
 126                 "load_size=0x%x, relocatable=%d\n", mld->link_base_addr,
 127                 mld->load_base_addr, load_size, mld->relocatable);
 128
 129   if (mld->relocatable)
 130     grub_dprintf ("multiboot_loader", "align=0x%lx, preference=0x%x, avoid_efi_boot_services=%d\n",
 131                   (long) mld->align, mld->preference, mld->avoid_efi_boot_services);
 132
 133   /* Load every loadable segment in memory.  */
 134   for (i = 0; i < ehdr->e_phnum; i++)
 135     {
 136       if (phdr(i)->p_type == PT_LOAD)
 137         {
 138
 139           grub_dprintf ("multiboot_loader", "segment %d: paddr=0x%lx, memsz=0x%lx, vaddr=0x%lx\n",
 140                         i, (long) phdr(i)->p_paddr, (long) phdr(i)->p_memsz, (long) phdr(i)->p_vaddr);
 141
 142           load_offset = phdr(i)->p_paddr - mld->link_base_addr;
 143
 144           if (phdr(i)->p_filesz != 0)
 145             {
 146               if (grub_file_seek (mld->file, (grub_off_t) phdr(i)->p_offset)
 147                   == (grub_off_t) -1)
 148                 return grub_errno;
 149
 150               if (grub_file_read (mld->file, (grub_uint8_t *) source + load_offset, phdr(i)->p_filesz)
 151                   != (grub_ssize_t) phdr(i)->p_filesz)
 152                 {
 153                   if (!grub_errno)
 154                     grub_error (GRUB_ERR_FILE_READ_ERROR, N_("premature end of file %s"),
 155                                 mld->filename);
 156                   return grub_errno;
 157                 }
 158             }
 159
 160           if (phdr(i)->p_filesz < phdr(i)->p_memsz)
 161             grub_memset ((grub_uint8_t *) source + load_offset + phdr(i)->p_filesz, 0,
 162                          phdr(i)->p_memsz - phdr(i)->p_filesz);
 163         }
 164     }
 165
 166   for (i = 0; i < ehdr->e_phnum; i++)
 167     if (phdr(i)->p_vaddr <= ehdr->e_entry
 168         && phdr(i)->p_vaddr + phdr(i)->p_memsz > ehdr->e_entry)
 169       {
 170         grub_multiboot_payload_eip = (ehdr->e_entry - phdr(i)->p_vaddr)
 171           + phdr(i)->p_paddr;
 172 #ifdef MULTIBOOT_LOAD_ELF64
 173 # ifdef __mips
 174   /* We still in 32-bit mode.  */
 175   if ((ehdr->e_entry - phdr(i)->p_vaddr)
 176       + phdr(i)->p_paddr < 0xffffffff80000000ULL)
 177     return grub_error (GRUB_ERR_BAD_OS, "invalid entry point for ELF64");
 178 # else
 179   /* We still in 32-bit mode.  */
 180   if ((ehdr->e_entry - phdr(i)->p_vaddr)
 181       + phdr(i)->p_paddr > 0xffffffff)
 182     return grub_error (GRUB_ERR_BAD_OS, "invalid entry point for ELF64");
 183 # endif
 184 #endif
 185         break;
 186       }
 187
 188   if (i == ehdr->e_phnum)
 189     return grub_error (GRUB_ERR_BAD_OS, "entry point isn't in a segment");
 190
 191 #if defined (__i386__) || defined (__x86_64__)
 192
 193 #elif defined (__mips)
 194   grub_multiboot_payload_eip |= 0x80000000;
 195 #else
 196 #error Please complete this
 197 #endif
 198
 199   if (ehdr->e_shnum)
 200     {
 201       grub_uint8_t *shdr, *shdrptr;
 202
 203       shdr = grub_malloc ((grub_uint32_t) ehdr->e_shnum * ehdr->e_shentsize);
 204       if (!shdr)
 205         return grub_errno;
 206
 207       if (grub_file_seek (mld->file, ehdr->e_shoff) == (grub_off_t) -1)
 208         {
 209           grub_free (shdr);
 210           return grub_errno;
 211         }
 212
 213       if (grub_file_read (mld->file, shdr, (grub_uint32_t) ehdr->e_shnum * ehdr->e_shentsize)
 214               != (grub_ssize_t) ehdr->e_shnum * ehdr->e_shentsize)
 215         {
 216           if (!grub_errno)
 217             grub_error (GRUB_ERR_FILE_READ_ERROR, N_("premature end of file %s"),
 218                         mld->filename);
 219           return grub_errno;
 220         }
 221
 222       for (shdrptr = shdr, i = 0; i < ehdr->e_shnum;
 223            shdrptr += ehdr->e_shentsize, i++)
 224         {
 225           Elf_Shdr *sh = (Elf_Shdr *) shdrptr;
 226           void *src;
 227           grub_addr_t target;
 228
 229           if (mld->mbi_ver >= 2 && (sh->sh_type == SHT_REL || sh->sh_type == SHT_RELA))
 230             return grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET, "ELF files with relocs are not supported yet");
 231
 232           /* This section is a loaded section,
 233              so we don't care.  */
 234           if (sh->sh_addr != 0)
 235             continue;
 236
 237           /* This section is empty, so we don't care.  */
 238           if (sh->sh_size == 0)
 239             continue;
 240
 241           err = grub_relocator_alloc_chunk_align (grub_multiboot_relocator, &ch, 0,
 242                                                   (0xffffffff - sh->sh_size) + 1,
 243                                                   sh->sh_size, sh->sh_addralign,
 244                                                   GRUB_RELOCATOR_PREFERENCE_NONE,
 245                                                   mld->avoid_efi_boot_services);
 246           if (err)
 247             {
 248               grub_dprintf ("multiboot_loader", "Error loading shdr %d\n", i);
 249               return err;
 250             }
 251           src = get_virtual_current_address (ch);
 252           target = get_physical_target_address (ch);
 253
 254           if (grub_file_seek (mld->file, sh->sh_offset) == (grub_off_t) -1)
 255             return grub_errno;
 256
 257           if (grub_file_read (mld->file, src, sh->sh_size)
 258               != (grub_ssize_t) sh->sh_size)
 259             {
 260               if (!grub_errno)
 261                 grub_error (GRUB_ERR_FILE_READ_ERROR, N_("premature end of file %s"),
 262                             mld->filename);
 263               return grub_errno;
 264             }
 265           sh->sh_addr = target;
 266         }
 267       grub_multiboot_add_elfsyms (ehdr->e_shnum, ehdr->e_shentsize,
 268                                   ehdr->e_shstrndx, shdr);
 269     }
 270
 271 #undef phdr
 272
 273   return grub_errno;
 274 }
 275
 276 #undef XX
 277 #undef E_MACHINE
 278 #undef ELFCLASSXX
 279 #undef Elf_Ehdr
 280 #undef Elf_Phdr
 281 #undef Elf_Shdr