arch/all-pc/boot/grub2-aros/grub-core/efiemu/main.c

   1 /*
   2  *  GRUB  --  GRand Unified Bootloader
   3  *  Copyright (C) 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 /* This is an emulation of EFI runtime services.
  20    This allows a more uniform boot on i386 machines.
  21    As it emulates only runtime service it isn't able
  22    to chainload EFI bootloader on non-EFI system. */
  23
  24
  25 #include <grub/file.h>
  26 #include <grub/err.h>
  27 #include <grub/normal.h>
  28 #include <grub/mm.h>
  29 #include <grub/dl.h>
  30 #include <grub/misc.h>
  31 #include <grub/efiemu/efiemu.h>
  32 #include <grub/command.h>
  33 #include <grub/i18n.h>
  34
  35 GRUB_MOD_LICENSE ("GPLv3+");
  36
  37 /* System table. Two version depending on mode */
  38 grub_efi_system_table32_t *grub_efiemu_system_table32 = 0;
  39 grub_efi_system_table64_t *grub_efiemu_system_table64 = 0;
  40 /* Modules may need to execute some actions after memory allocation happens */
  41 static struct grub_efiemu_prepare_hook *efiemu_prepare_hooks = 0;
  42 /* Linked list of configuration tables */
  43 static struct grub_efiemu_configuration_table *efiemu_config_tables = 0;
  44 static int prepared = 0;
  45
  46 /* Free all allocated space */
  47 grub_err_t
  48 grub_efiemu_unload (void)
  49 {
  50   struct grub_efiemu_configuration_table *cur, *d;
  51   struct grub_efiemu_prepare_hook *curhook, *d2;
  52   grub_efiemu_loadcore_unload ();
  53
  54   grub_efiemu_mm_unload ();
  55
  56   for (cur = efiemu_config_tables; cur;)
  57     {
  58       d = cur->next;
  59       if (cur->unload)
  60         cur->unload (cur->data);
  61       grub_free (cur);
  62       cur = d;
  63     }
  64   efiemu_config_tables = 0;
  65
  66   for (curhook = efiemu_prepare_hooks; curhook;)
  67     {
  68       d2 = curhook->next;
  69       if (curhook->unload)
  70         curhook->unload (curhook->data);
  71       grub_free (curhook);
  72       curhook = d2;
  73     }
  74   efiemu_prepare_hooks = 0;
  75
  76   prepared = 0;
  77
  78   return GRUB_ERR_NONE;
  79 }
  80
  81 /* Remove previously registered table from the list */
  82 grub_err_t
  83 grub_efiemu_unregister_configuration_table (grub_efi_guid_t guid)
  84 {
  85   struct grub_efiemu_configuration_table *cur, *prev;
  86
  87   /* Special treating if head is to remove */
  88   while (efiemu_config_tables
  89          && !grub_memcmp (&(efiemu_config_tables->guid), &guid, sizeof (guid)))
  90     {
  91       if (efiemu_config_tables->unload)
  92           efiemu_config_tables->unload (efiemu_config_tables->data);
  93         cur = efiemu_config_tables->next;
  94         grub_free (efiemu_config_tables);
  95         efiemu_config_tables = cur;
  96     }
  97   if (!efiemu_config_tables)
  98     return GRUB_ERR_NONE;
  99
 100   /* Remove from chain */
 101   for (prev = efiemu_config_tables, cur = prev->next; cur;)
 102     if (grub_memcmp (&(cur->guid), &guid, sizeof (guid)) == 0)
 103       {
 104         if (cur->unload)
 105           cur->unload (cur->data);
 106         prev->next = cur->next;
 107         grub_free (cur);
 108         cur = prev->next;
 109       }
 110     else
 111       {
 112         prev = cur;
 113         cur = cur->next;
 114       }
 115   return GRUB_ERR_NONE;
 116 }
 117
 118 grub_err_t
 119 grub_efiemu_register_prepare_hook (grub_err_t (*hook) (void *data),
 120                                    void (*unload) (void *data),
 121                                    void *data)
 122 {
 123   struct grub_efiemu_prepare_hook *nhook;
 124   nhook = (struct grub_efiemu_prepare_hook *) grub_malloc (sizeof (*nhook));
 125   if (! nhook)
 126     return grub_errno;
 127   nhook->hook = hook;
 128   nhook->unload = unload;
 129   nhook->data = data;
 130   nhook->next = efiemu_prepare_hooks;
 131   efiemu_prepare_hooks = nhook;
 132   return GRUB_ERR_NONE;
 133 }
 134
 135 /* Register a configuration table either supplying the address directly
 136    or with a hook
 137 */
 138 grub_err_t
 139 grub_efiemu_register_configuration_table (grub_efi_guid_t guid,
 140                                           void * (*get_table) (void *data),
 141                                           void (*unload) (void *data),
 142                                           void *data)
 143 {
 144   struct grub_efiemu_configuration_table *tbl;
 145   grub_err_t err;
 146
 147  err = grub_efiemu_unregister_configuration_table (guid);
 148   if (err)
 149     return err;
 150
 151   tbl = (struct grub_efiemu_configuration_table *) grub_malloc (sizeof (*tbl));
 152   if (! tbl)
 153     return grub_errno;
 154
 155   tbl->guid = guid;
 156   tbl->get_table = get_table;
 157   tbl->unload = unload;
 158   tbl->data = data;
 159   tbl->next = efiemu_config_tables;
 160   efiemu_config_tables = tbl;
 161
 162   return GRUB_ERR_NONE;
 163 }
 164
 165 static grub_err_t
 166 grub_cmd_efiemu_unload (grub_command_t cmd __attribute__ ((unused)),
 167                         int argc __attribute__ ((unused)),
 168                         char *args[] __attribute__ ((unused)))
 169 {
 170   return grub_efiemu_unload ();
 171 }
 172
 173 static grub_err_t
 174 grub_cmd_efiemu_prepare (grub_command_t cmd __attribute__ ((unused)),
 175                          int argc __attribute__ ((unused)),
 176                          char *args[] __attribute__ ((unused)))
 177 {
 178   return grub_efiemu_prepare ();
 179 }
 180
 181 \f
 182
 183 /* Load the runtime from the file FILENAME.  */
 184 static grub_err_t
 185 grub_efiemu_load_file (const char *filename)
 186 {
 187   grub_file_t file;
 188   grub_err_t err;
 189
 190   file = grub_file_open (filename);
 191   if (! file)
 192     return grub_errno;
 193
 194   err = grub_efiemu_mm_init ();
 195   if (err)
 196     {
 197       grub_file_close (file);
 198       grub_efiemu_unload ();
 199       return grub_errno;
 200     }
 201
 202   grub_dprintf ("efiemu", "mm initialized\n");
 203
 204   err = grub_efiemu_loadcore_init (file, filename);
 205   if (err)
 206     {
 207       grub_file_close (file);
 208       grub_efiemu_unload ();
 209       return err;
 210     }
 211
 212   grub_file_close (file);
 213
 214   /* For configuration tables entry in system table. */
 215   grub_efiemu_request_symbols (1);
 216
 217   return GRUB_ERR_NONE;
 218 }
 219
 220 grub_err_t
 221 grub_efiemu_autocore (void)
 222 {
 223   const char *prefix;
 224   char *filename;
 225   const char *suffix;
 226   grub_err_t err;
 227
 228   if (grub_efiemu_sizeof_uintn_t () != 0)
 229     return GRUB_ERR_NONE;
 230
 231   prefix = grub_env_get ("prefix");
 232
 233   if (! prefix)
 234     return grub_error (GRUB_ERR_FILE_NOT_FOUND,
 235                        N_("variable `%s' isn't set"), "prefix");
 236
 237   suffix = grub_efiemu_get_default_core_name ();
 238
 239   filename = grub_xasprintf ("%s/" GRUB_TARGET_CPU "-" GRUB_PLATFORM "/%s",
 240                              prefix, suffix);
 241   if (! filename)
 242     return grub_errno;
 243
 244   err = grub_efiemu_load_file (filename);
 245   grub_free (filename);
 246   if (err)
 247     return err;
 248 #ifndef GRUB_MACHINE_EMU
 249   err = grub_machine_efiemu_init_tables ();
 250   if (err)
 251     return err;
 252 #endif
 253
 254   return GRUB_ERR_NONE;
 255 }
 256
 257 grub_err_t
 258 grub_efiemu_prepare (void)
 259 {
 260   grub_err_t err;
 261
 262   if (prepared)
 263     return GRUB_ERR_NONE;
 264
 265   err = grub_efiemu_autocore ();
 266   if (err)
 267     return err;
 268
 269   grub_dprintf ("efiemu", "Preparing %d-bit efiemu\n",
 270                 8 * grub_efiemu_sizeof_uintn_t ());
 271
 272   /* Create NVRAM. */
 273   grub_efiemu_pnvram ();
 274
 275   prepared = 1;
 276
 277   if (grub_efiemu_sizeof_uintn_t () == 4)
 278     return grub_efiemu_prepare32 (efiemu_prepare_hooks, efiemu_config_tables);
 279   else
 280     return grub_efiemu_prepare64 (efiemu_prepare_hooks, efiemu_config_tables);
 281 }
 282
 283
 284 static grub_err_t
 285 grub_cmd_efiemu_load (grub_command_t cmd __attribute__ ((unused)),
 286                       int argc, char *args[])
 287 {
 288   grub_err_t err;
 289
 290   grub_efiemu_unload ();
 291
 292   if (argc != 1)
 293     return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("filename expected"));
 294
 295   err = grub_efiemu_load_file (args[0]);
 296   if (err)
 297     return err;
 298 #ifndef GRUB_MACHINE_EMU
 299   err = grub_machine_efiemu_init_tables ();
 300   if (err)
 301     return err;
 302 #endif
 303   return GRUB_ERR_NONE;
 304 }
 305
 306 static grub_command_t cmd_loadcore, cmd_prepare, cmd_unload;
 307
 308 GRUB_MOD_INIT(efiemu)
 309 {
 310   cmd_loadcore = grub_register_command ("efiemu_loadcore",
 311                                         grub_cmd_efiemu_load,
 312                                         N_("FILE"),
 313                                         N_("Load and initialize EFI emulator."));
 314   cmd_prepare = grub_register_command ("efiemu_prepare",
 315                                        grub_cmd_efiemu_prepare,
 316                                        0,
 317                                        N_("Finalize loading of EFI emulator."));
 318   cmd_unload = grub_register_command ("efiemu_unload", grub_cmd_efiemu_unload,
 319                                       0,
 320                                       N_("Unload EFI emulator."));
 321 }
 322
 323 GRUB_MOD_FINI(efiemu)
 324 {
 325   grub_unregister_command (cmd_loadcore);
 326   grub_unregister_command (cmd_prepare);
 327   grub_unregister_command (cmd_unload);
 328 }