f36cc10867
Update a few copyright years which we forgot to do in 2008 (only for files whose changes made in 2008 were copyright-significant) * Makefile.in: Add 2008 to Copyright line. * disk/ieee1275/ofdisk.c: Likewise. * disk/efi/efidisk.c: Likewise. * kern/dl.c: Likewise. * kern/sparc64/ieee1275/init.c: Likewise. * kern/mm.c: Likewise. * kern/efi/mm.c: Likewise. * boot/i386/pc/boot.S: Likewise. * genfslist.sh: Likewise. * fs/iso9660.c: Likewise. * fs/hfs.c: Likewise. * fs/jfs.c: Likewise. * fs/minix.c: Likewise. * fs/ufs.c: Likewise. * gensymlist.sh.in: Likewise. * genkernsyms.sh.in: Likewise. * include/grub/misc.h: Likewise. * include/grub/types.h: Likewise. * include/grub/symbol.h: Likewise. * include/grub/elf.h: Likewise. * include/grub/kernel.h: Likewise. * include/grub/disk.h: Likewise. * include/grub/dl.h: Likewise. * include/grub/i386/linux.h: Likewise. * include/grub/i386/pc/biosdisk.h: Likewise. * include/grub/efi/api.h: Likewise. * include/grub/efi/pe32.h: Likewise. * include/grub/util/misc.h: Likewise. * normal/execute.c: Likewise. * normal/arg.c: Likewise. * normal/completion.c: Likewise. * normal/lexer.c: Likewise. * normal/parser.y: Likewise. * normal/misc.c: Likewise. * commands/i386/pc/vbeinfo.c: Likewise. * commands/hexdump.c: Likewise. * commands/terminal.c: Likewise. * commands/ls.c: Likewise. * commands/help.c: Likewise. * partmap/pc.c: Likewise. * loader/efi/chainloader.c: Likewise. * loader/multiboot_loader.c: Likewise. * loader/i386/pc/multiboot2.c: Likewise. * term/efi/console.c: Likewise. * term/i386/pc/serial.c: Likewise. * util/lvm.c: Likewise. * util/console.c: Likewise. * util/i386/efi/grub-mkimage.c: Likewise. * util/raid.c: Likewise.
254 lines
6.5 KiB
C
254 lines
6.5 KiB
C
/* execute.c -- Execute a GRUB script. */
|
||
/*
|
||
* GRUB -- GRand Unified Bootloader
|
||
* Copyright (C) 2005,2007,2008 Free Software Foundation, Inc.
|
||
*
|
||
* GRUB is free software: you can redistribute it and/or modify
|
||
* it under the terms of the GNU General Public License as published by
|
||
* the Free Software Foundation, either version 3 of the License, or
|
||
* (at your option) any later version.
|
||
*
|
||
* GRUB is distributed in the hope that it will be useful,
|
||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
* GNU General Public License for more details.
|
||
*
|
||
* You should have received a copy of the GNU General Public License
|
||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||
*/
|
||
|
||
#include <grub/misc.h>
|
||
#include <grub/mm.h>
|
||
#include <grub/normal.h>
|
||
#include <grub/arg.h>
|
||
#include <grub/env.h>
|
||
#include <grub/script.h>
|
||
|
||
static grub_err_t
|
||
grub_script_execute_cmd (struct grub_script_cmd *cmd)
|
||
{
|
||
if (cmd == 0)
|
||
return 0;
|
||
|
||
return cmd->exec (cmd);
|
||
}
|
||
|
||
/* Parse ARG and return the textual representation. Add strings are
|
||
concatenated and all values of the variables are filled in. */
|
||
static char *
|
||
grub_script_execute_argument_to_string (struct grub_script_arg *arg)
|
||
{
|
||
int size = 0;
|
||
char *val;
|
||
char *chararg;
|
||
struct grub_script_arg *argi;
|
||
|
||
/* First determine the size of the argument. */
|
||
for (argi = arg; argi; argi = argi->next)
|
||
{
|
||
if (argi->type == 1)
|
||
{
|
||
val = grub_env_get (argi->str);
|
||
if (val)
|
||
size += grub_strlen (val);
|
||
}
|
||
else
|
||
size += grub_strlen (argi->str);
|
||
}
|
||
|
||
/* Create the argument. */
|
||
chararg = grub_malloc (size + 1);
|
||
if (! chararg)
|
||
return 0;
|
||
|
||
*chararg = '\0';
|
||
/* First determine the size of the argument. */
|
||
for (argi = arg; argi; argi = argi->next)
|
||
{
|
||
if (argi->type == 1)
|
||
{
|
||
val = grub_env_get (argi->str);
|
||
if (val)
|
||
grub_strcat (chararg, val);
|
||
}
|
||
else
|
||
grub_strcat (chararg, argi->str);
|
||
}
|
||
|
||
return chararg;
|
||
}
|
||
|
||
/* Execute a single command line. */
|
||
grub_err_t
|
||
grub_script_execute_cmdline (struct grub_script_cmd *cmd)
|
||
{
|
||
struct grub_script_cmdline *cmdline = (struct grub_script_cmdline *) cmd;
|
||
struct grub_script_arglist *arglist;
|
||
char **args = 0;
|
||
int i = 0;
|
||
grub_command_t grubcmd;
|
||
struct grub_arg_list *state;
|
||
struct grub_arg_option *parser;
|
||
int maxargs = 0;
|
||
char **parsed_arglist;
|
||
int numargs;
|
||
grub_err_t ret = 0;
|
||
int argcount = 0;
|
||
grub_script_function_t func = 0;
|
||
char errnobuf[6];
|
||
|
||
/* Lookup the command. */
|
||
grubcmd = grub_command_find (cmdline->cmdname);
|
||
if (! grubcmd)
|
||
{
|
||
/* Ignore errors. */
|
||
grub_errno = GRUB_ERR_NONE;
|
||
|
||
/* It's not a GRUB command, try all functions. */
|
||
func = grub_script_function_find (cmdline->cmdname);
|
||
if (! func)
|
||
{
|
||
/* As a last resort, try if it is an assignment. */
|
||
char *assign = grub_strdup (cmdline->cmdname);
|
||
char *eq = grub_strchr (assign, '=');
|
||
|
||
if (eq)
|
||
{
|
||
/* Create two strings and set the variable. */
|
||
*eq = '\0';
|
||
eq++;
|
||
grub_env_set (assign, eq);
|
||
|
||
/* This was set because the command was not found. */
|
||
grub_errno = GRUB_ERR_NONE;
|
||
}
|
||
grub_free (assign);
|
||
return 0;
|
||
}
|
||
}
|
||
|
||
if (cmdline->arglist)
|
||
{
|
||
argcount = cmdline->arglist->argcount;
|
||
|
||
/* Create argv from the arguments. */
|
||
args = grub_malloc (sizeof (char *) * argcount);
|
||
for (arglist = cmdline->arglist; arglist; arglist = arglist->next)
|
||
{
|
||
char *str;
|
||
str = grub_script_execute_argument_to_string (arglist->arg);
|
||
args[i++] = str;
|
||
}
|
||
}
|
||
|
||
/* Execute the GRUB command or function. */
|
||
if (grubcmd)
|
||
{
|
||
/* Count the amount of options the command has. */
|
||
parser = (struct grub_arg_option *) grubcmd->options;
|
||
while (parser && (parser++)->doc)
|
||
maxargs++;
|
||
|
||
/* Set up the option state. */
|
||
state = grub_malloc (sizeof (struct grub_arg_list) * maxargs);
|
||
grub_memset (state, 0, sizeof (struct grub_arg_list) * maxargs);
|
||
|
||
/* Start the command. */
|
||
if (! (grubcmd->flags & GRUB_COMMAND_FLAG_NO_ARG_PARSE))
|
||
{
|
||
if (grub_arg_parse (grubcmd, argcount, args, state, &parsed_arglist, &numargs))
|
||
ret = (grubcmd->func) (state, numargs, parsed_arglist);
|
||
}
|
||
else
|
||
ret = (grubcmd->func) (state, argcount, args);
|
||
|
||
grub_free (state);
|
||
}
|
||
else
|
||
ret = grub_script_function_call (func, argcount, args);
|
||
|
||
/* Free arguments. */
|
||
for (i = 0; i < argcount; i++)
|
||
grub_free (args[i]);
|
||
grub_free (args);
|
||
|
||
grub_sprintf (errnobuf, "%d", ret);
|
||
grub_env_set ("?", errnobuf);
|
||
|
||
return ret;
|
||
}
|
||
|
||
/* Execute a block of one or more commands. */
|
||
grub_err_t
|
||
grub_script_execute_cmdblock (struct grub_script_cmd *cmd)
|
||
{
|
||
struct grub_script_cmdblock *cmdblock = (struct grub_script_cmdblock *) cmd;
|
||
|
||
/* Loop over every command and execute it. */
|
||
for (cmd = cmdblock->cmdlist; cmd; cmd = cmd->next)
|
||
grub_script_execute_cmd (cmd);
|
||
|
||
return 0;
|
||
}
|
||
|
||
/* Execute an if statement. */
|
||
grub_err_t
|
||
grub_script_execute_cmdif (struct grub_script_cmd *cmd)
|
||
{
|
||
struct grub_script_cmdif *cmdif = (struct grub_script_cmdif *) cmd;
|
||
char *result;
|
||
|
||
/* Check if the commands results in a true or a false. The value is
|
||
read from the env variable `?'. */
|
||
grub_script_execute_cmd (cmdif->exec_to_evaluate);
|
||
result = grub_env_get ("?");
|
||
|
||
/* Execute the `if' or the `else' part depending on the value of
|
||
`?'. */
|
||
if (result && ! grub_strcmp (result, "0"))
|
||
return grub_script_execute_cmd (cmdif->exec_on_true);
|
||
else
|
||
return grub_script_execute_cmd (cmdif->exec_on_false);
|
||
}
|
||
|
||
/* Execute the menu entry generate statement. */
|
||
grub_err_t
|
||
grub_script_execute_menuentry (struct grub_script_cmd *cmd)
|
||
{
|
||
struct grub_script_cmd_menuentry *cmd_menuentry;
|
||
char *title;
|
||
struct grub_script *script;
|
||
|
||
cmd_menuentry = (struct grub_script_cmd_menuentry *) cmd;
|
||
|
||
/* The title can contain variables, parse them and generate a string
|
||
from it. */
|
||
title = grub_script_execute_argument_to_string (cmd_menuentry->title);
|
||
if (! title)
|
||
return grub_errno;
|
||
|
||
/* Parse the menu entry *again*. */
|
||
script = grub_script_parse ((char *) cmd_menuentry->sourcecode, 0);
|
||
|
||
if (! script)
|
||
{
|
||
grub_free (title);
|
||
return grub_errno;
|
||
}
|
||
|
||
/* XXX: When this fails, the memory should be freed? */
|
||
return grub_normal_menu_addentry (title, script,
|
||
cmd_menuentry->sourcecode);
|
||
}
|
||
|
||
|
||
|
||
/* Execute any GRUB pre-parsed command or script. */
|
||
grub_err_t
|
||
grub_script_execute (struct grub_script *script)
|
||
{
|
||
if (script == 0)
|
||
return 0;
|
||
|
||
return grub_script_execute_cmd (script->cmd);
|
||
}
|