arch/all-pc/boot/grub2-aros/grub-core/normal/dyncmd.c

   1 /* dyncmd.c - support dynamic command */
   2 /*
   3  *  GRUB  --  GRand Unified Bootloader
   4  *  Copyright (C) 2009  Free Software Foundation, Inc.
   5  *
   6  *  GRUB is free software: you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation, either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  GRUB is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #include <grub/dl.h>
  21 #include <grub/mm.h>
  22 #include <grub/env.h>
  23 #include <grub/misc.h>
  24 #include <grub/command.h>
  25 #include <grub/normal.h>
  26 #include <grub/extcmd.h>
  27 #include <grub/script_sh.h>
  28 #include <grub/i18n.h>
  29
  30 grub_command_t
  31 grub_dyncmd_get_cmd (grub_command_t cmd)
  32 {
  33   grub_extcmd_t extcmd = cmd->data;
  34   char *modname;
  35   char *name;
  36   grub_dl_t mod;
  37
  38   modname = extcmd->data;
  39   mod = grub_dl_load (modname);
  40   if (!mod)
  41     return NULL;
  42
  43   grub_free (modname);
  44   grub_dl_ref (mod);
  45
  46   name = (char *) cmd->name;
  47   grub_unregister_extcmd (extcmd);
  48
  49   cmd = grub_command_find (name);
  50
  51   grub_free (name);
  52
  53   return cmd;
  54 }
  55
  56 static grub_err_t
  57 grub_dyncmd_dispatcher (struct grub_extcmd_context *ctxt,
  58                         int argc, char **args)
  59 {
  60   char *modname;
  61   grub_dl_t mod;
  62   grub_err_t ret;
  63   grub_extcmd_t extcmd = ctxt->extcmd;
  64   grub_command_t cmd = extcmd->cmd;
  65   char *name;
  66
  67   modname = extcmd->data;
  68   mod = grub_dl_load (modname);
  69   if (!mod)
  70     return grub_errno;
  71
  72   grub_free (modname);
  73   grub_dl_ref (mod);
  74
  75   name = (char *) cmd->name;
  76   grub_unregister_extcmd (extcmd);
  77
  78   cmd = grub_command_find (name);
  79   if (cmd)
  80     {
  81       if (cmd->flags & GRUB_COMMAND_FLAG_BLOCKS &&
  82           cmd->flags & GRUB_COMMAND_FLAG_EXTCMD)
  83         ret = grub_extcmd_dispatcher (cmd, argc, args, ctxt->script);
  84       else
  85         ret = (cmd->func) (cmd, argc, args);
  86     }
  87   else
  88     ret = grub_errno;
  89
  90   grub_free (name);
  91
  92   return ret;
  93 }
  94
  95 /* Read the file command.lst for auto-loading.  */
  96 void
  97 read_command_list (const char *prefix)
  98 {
  99   if (prefix)
 100     {
 101       char *filename;
 102
 103       filename = grub_xasprintf ("%s/" GRUB_TARGET_CPU "-" GRUB_PLATFORM
 104                                  "/command.lst", prefix);
 105       if (filename)
 106         {
 107           grub_file_t file;
 108
 109           file = grub_file_open (filename);
 110           if (file)
 111             {
 112               char *buf = NULL;
 113               grub_command_t ptr, last = 0, next;
 114
 115               /* Override previous commands.lst.  */
 116               for (ptr = grub_command_list; ptr; ptr = next)
 117                 {
 118                   next = ptr->next;
 119                   if (ptr->flags & GRUB_COMMAND_FLAG_DYNCMD)
 120                     {
 121                       if (last)
 122                         last->next = ptr->next;
 123                       else
 124                         grub_command_list = ptr->next;
 125                       grub_free (ptr->data); /* extcmd struct */
 126                       grub_free (ptr);
 127                     }
 128                   else
 129                     last = ptr;
 130                 }
 131
 132               for (;; grub_free (buf))
 133                 {
 134                   char *p, *name, *modname;
 135                   grub_extcmd_t cmd;
 136                   int prio = 0;
 137
 138                   buf = grub_file_getline (file);
 139
 140                   if (! buf)
 141                     break;
 142
 143                   name = buf;
 144                   while (grub_isspace (name[0]))
 145                     name++;
 146
 147                   if (*name == '*')
 148                     {
 149                       name++;
 150                       prio++;
 151                     }
 152
 153                   if (! grub_isgraph (name[0]))
 154                     continue;
 155
 156                   p = grub_strchr (name, ':');
 157                   if (! p)
 158                     continue;
 159
 160                   *p = '\0';
 161                   p++;
 162                   while (*p == ' ' || *p == '\t')
 163                     p++;
 164
 165                   if (! grub_isgraph (*p))
 166                     continue;
 167
 168                   if (grub_dl_get (p))
 169                     continue;
 170
 171                   name = grub_strdup (name);
 172                   if (! name)
 173                     continue;
 174
 175                   modname = grub_strdup (p);
 176                   if (! modname)
 177                     {
 178                       grub_free (name);
 179                       continue;
 180                     }
 181
 182                   cmd = grub_register_extcmd_prio (name,
 183                                                    grub_dyncmd_dispatcher,
 184                                                    GRUB_COMMAND_FLAG_BLOCKS
 185                                                    | GRUB_COMMAND_FLAG_EXTCMD
 186                                                    | GRUB_COMMAND_FLAG_DYNCMD,
 187                                                    0, N_("module isn't loaded"),
 188                                                    0, prio);
 189                   if (! cmd)
 190                     {
 191                       grub_free (name);
 192                       grub_free (modname);
 193                       continue;
 194                     }
 195                   cmd->data = modname;
 196
 197                   /* Update the active flag.  */
 198                   grub_command_find (name);
 199                 }
 200
 201               grub_file_close (file);
 202             }
 203
 204           grub_free (filename);
 205         }
 206     }
 207
 208   /* Ignore errors.  */
 209   grub_errno = GRUB_ERR_NONE;
 210 }