2N/A/* dyncmd.c - support dynamic command */
2N/A/*
2N/A * GRUB -- GRand Unified Bootloader
2N/A * Copyright (C) 2009 Free Software Foundation, Inc.
2N/A *
2N/A * GRUB is free software: you can redistribute it and/or modify
2N/A * it under the terms of the GNU General Public License as published by
2N/A * the Free Software Foundation, either version 3 of the License, or
2N/A * (at your option) any later version.
2N/A *
2N/A * GRUB is distributed in the hope that it will be useful,
2N/A * but WITHOUT ANY WARRANTY; without even the implied warranty of
2N/A * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2N/A * GNU General Public License for more details.
2N/A *
2N/A * You should have received a copy of the GNU General Public License
2N/A * along with GRUB. If not, see <http://www.gnu.org/licenses/>.
2N/A */
2N/A
2N/A#include <grub/dl.h>
2N/A#include <grub/mm.h>
2N/A#include <grub/env.h>
2N/A#include <grub/misc.h>
2N/A#include <grub/command.h>
2N/A#include <grub/normal.h>
2N/A#include <grub/extcmd.h>
2N/A#include <grub/script_sh.h>
2N/A#include <grub/i18n.h>
2N/A
2N/Astatic grub_err_t
2N/Agrub_dyncmd_dispatcher (struct grub_extcmd_context *ctxt,
2N/A int argc, char **args)
2N/A{
2N/A char *modname;
2N/A grub_dl_t mod;
2N/A grub_err_t ret;
2N/A grub_extcmd_t extcmd = ctxt->extcmd;
2N/A grub_command_t cmd = extcmd->cmd;
2N/A
2N/A modname = extcmd->data;
2N/A mod = grub_dl_load (modname);
2N/A if (mod)
2N/A {
2N/A char *name;
2N/A
2N/A grub_free (modname);
2N/A grub_dl_ref (mod);
2N/A
2N/A name = (char *) cmd->name;
2N/A grub_unregister_extcmd (extcmd);
2N/A
2N/A cmd = grub_command_find (name);
2N/A if (cmd)
2N/A {
2N/A if (cmd->flags & GRUB_COMMAND_FLAG_BLOCKS &&
2N/A cmd->flags & GRUB_COMMAND_FLAG_EXTCMD)
2N/A ret = grub_extcmd_dispatcher (cmd, argc, args, ctxt->script);
2N/A else
2N/A ret = (cmd->func) (cmd, argc, args);
2N/A }
2N/A else
2N/A ret = grub_errno;
2N/A
2N/A grub_free (name);
2N/A }
2N/A else
2N/A ret = grub_errno;
2N/A
2N/A return ret;
2N/A}
2N/A
2N/A/* Read the file command.lst for auto-loading. */
2N/Avoid
2N/Aread_command_list (const char *prefix)
2N/A{
2N/A if (prefix)
2N/A {
2N/A char *filename;
2N/A
2N/A filename = grub_xasprintf ("%s/command.lst", prefix);
2N/A if (filename)
2N/A {
2N/A grub_file_t file;
2N/A
2N/A file = grub_file_open (filename);
2N/A if (file)
2N/A {
2N/A char *buf = NULL;
2N/A grub_command_t ptr, last = 0, next;
2N/A
2N/A /* Override previous commands.lst. */
2N/A for (ptr = grub_command_list; ptr; ptr = next)
2N/A {
2N/A next = ptr->next;
2N/A if (ptr->flags & GRUB_COMMAND_FLAG_DYNCMD)
2N/A {
2N/A if (last)
2N/A last->next = ptr->next;
2N/A else
2N/A grub_command_list = ptr->next;
2N/A grub_free (ptr);
2N/A grub_free (ptr->data); /* extcmd struct */
2N/A }
2N/A else
2N/A last = ptr;
2N/A }
2N/A
2N/A for (;; grub_free (buf))
2N/A {
2N/A char *p, *name, *modname;
2N/A grub_extcmd_t cmd;
2N/A int prio = 0;
2N/A
2N/A buf = grub_file_getline (file);
2N/A
2N/A if (! buf)
2N/A break;
2N/A
2N/A name = buf;
2N/A if (*name == '*')
2N/A {
2N/A name++;
2N/A prio++;
2N/A }
2N/A
2N/A if (! grub_isgraph (name[0]))
2N/A continue;
2N/A
2N/A p = grub_strchr (name, ':');
2N/A if (! p)
2N/A continue;
2N/A
2N/A *p = '\0';
2N/A while (*++p == ' ')
2N/A ;
2N/A
2N/A if (! grub_isgraph (*p))
2N/A continue;
2N/A
2N/A if (grub_dl_get (p))
2N/A continue;
2N/A
2N/A name = grub_strdup (name);
2N/A if (! name)
2N/A continue;
2N/A
2N/A modname = grub_strdup (p);
2N/A if (! modname)
2N/A {
2N/A grub_free (name);
2N/A continue;
2N/A }
2N/A
2N/A cmd = grub_register_extcmd_prio (name,
2N/A grub_dyncmd_dispatcher,
2N/A GRUB_COMMAND_FLAG_BLOCKS
2N/A | GRUB_COMMAND_FLAG_EXTCMD
2N/A | GRUB_COMMAND_FLAG_DYNCMD,
2N/A 0, N_("not loaded"), 0,
2N/A prio);
2N/A if (! cmd)
2N/A {
2N/A grub_free (name);
2N/A grub_free (modname);
2N/A continue;
2N/A }
2N/A cmd->data = modname;
2N/A
2N/A /* Update the active flag. */
2N/A grub_command_find (name);
2N/A }
2N/A
2N/A grub_file_close (file);
2N/A }
2N/A
2N/A grub_free (filename);
2N/A }
2N/A }
2N/A
2N/A /* Ignore errors. */
2N/A grub_errno = GRUB_ERR_NONE;
2N/A}