1
/* execute.c -- Execute a GRUB script. */
3
* GRUB -- GRand Unified Bootloader
4
* Copyright (C) 2005 Free Software Foundation, Inc.
6
* This program 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 2 of the License, or
9
* (at your option) any later version.
11
* This program 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.
16
* You should have received a copy of the GNU General Public License
17
* along with this program; if not, write to the Free Software
18
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21
#include <grub/misc.h>
23
#include <grub/normal.h>
26
#include <grub/script.h>
29
grub_script_execute_cmd (struct grub_script_cmd *cmd)
38
/* Parse ARG and return the textual representation. Add strings are
39
concatenated and all values of the variables are filled in. */
41
grub_script_execute_argument_to_string (struct grub_script_arg *arg)
46
struct grub_script_arg *argi;
48
/* First determine the size of the argument. */
49
for (argi = arg; argi; argi = argi->next)
53
val = grub_env_get (argi->str);
54
size += grub_strlen (val);
57
size += grub_strlen (argi->str);
60
/* Create the argument. */
61
chararg = grub_malloc (size + 1);
66
/* First determine the size of the argument. */
67
for (argi = arg; argi; argi = argi->next)
71
val = grub_env_get (argi->str);
72
grub_strcat (chararg, val);
75
grub_strcat (chararg, argi->str);
81
/* Execute a single command line. */
83
grub_script_execute_cmdline (struct grub_script_cmd *cmd)
85
struct grub_script_cmdline *cmdline = (struct grub_script_cmdline *) cmd;
86
struct grub_script_arglist *arglist;
89
grub_command_t grubcmd;
90
struct grub_arg_list *state;
91
struct grub_arg_option *parser;
93
char **parsed_arglist;
97
grub_script_function_t func = 0;
100
/* Lookup the command. */
101
grubcmd = grub_command_find (cmdline->cmdname);
104
/* It's not a GRUB command, try all functions. */
105
func = grub_script_function_find (cmdline->cmdname);
110
if (cmdline->arglist)
112
argcount = cmdline->arglist->argcount;
114
/* Create argv from the arguments. */
115
args = grub_malloc (sizeof (char *) * argcount);
116
for (arglist = cmdline->arglist; arglist; arglist = arglist->next)
119
str = grub_script_execute_argument_to_string (arglist->arg);
124
/* Execute the GRUB command or function. */
127
/* Count the amount of options the command has. */
128
parser = (struct grub_arg_option *) grubcmd->options;
129
while (parser && (parser++)->doc)
132
/* Set up the option state. */
133
state = grub_malloc (sizeof (struct grub_arg_list) * maxargs);
134
grub_memset (state, 0, sizeof (struct grub_arg_list) * maxargs);
136
/* Start the command. */
137
if (! (grubcmd->flags & GRUB_COMMAND_FLAG_NO_ARG_PARSE))
139
if (grub_arg_parse (grubcmd, argcount, args, state, &parsed_arglist, &numargs))
140
ret = (grubcmd->func) (state, numargs, parsed_arglist);
143
ret = (grubcmd->func) (state, argcount, args);
148
ret = grub_script_function_call (func, argcount, args);
150
/* Free arguments. */
151
for (i = 0; i < argcount; i++)
155
grub_sprintf (errnobuf, "%d", ret);
156
grub_env_set ("?", errnobuf);
161
/* Execute a block of one or more commands. */
163
grub_script_execute_cmdblock (struct grub_script_cmd *cmd)
165
struct grub_script_cmdblock *cmdblock = (struct grub_script_cmdblock *) cmd;
167
/* Loop over every command and execute it. */
168
for (cmd = cmdblock->cmdlist; cmd; cmd = cmd->next)
169
grub_script_execute_cmd (cmd);
174
/* Execute an if statement. */
176
grub_script_execute_cmdif (struct grub_script_cmd *cmd)
178
struct grub_script_cmdif *cmdif = (struct grub_script_cmdif *) cmd;
181
/* Check if the commands results in a true or a false. The value is
182
read from the env variable `RESULT'. */
183
grub_script_execute_cmd (cmdif->bool);
184
bool = grub_env_get ("?");
186
/* Execute the `if' or the `else' part depending on the value of
188
if (bool && ! grub_strcmp (bool, "0"))
189
return grub_script_execute_cmd (cmdif->true);
191
return grub_script_execute_cmd (cmdif->false);
196
/* Execute any GRUB pre-parsed command or script. */
198
grub_script_execute (struct grub_script *script)
203
return grub_script_execute_cmd (script->cmd);