/* execute.c -- Execute a GRUB script. */ /* * GRUB -- GRand Unified Bootloader * Copyright (C) 2005 Free Software Foundation, Inc. * * This program 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 2 of the License, or * (at your option) any later version. * * This program 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 this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ #include #include #include #include #include #include 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); 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); 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) { /* 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 *bool; /* Check if the commands results in a true or a false. The value is read from the env variable `RESULT'. */ grub_script_execute_cmd (cmdif->bool); bool = grub_env_get ("?"); /* Execute the `if' or the `else' part depending on the value of `RESULT'. */ if (bool && ! grub_strcmp (bool, "0")) return grub_script_execute_cmd (cmdif->true); else return grub_script_execute_cmd (cmdif->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 free'ed? */ 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); }