diff options
Diffstat (limited to 'debian/grub-extras/lua/grub_main.c')
| -rw-r--r-- | debian/grub-extras/lua/grub_main.c | 230 |
1 files changed, 230 insertions, 0 deletions
diff --git a/debian/grub-extras/lua/grub_main.c b/debian/grub-extras/lua/grub_main.c new file mode 100644 index 0000000..cbb1790 --- /dev/null +++ b/debian/grub-extras/lua/grub_main.c @@ -0,0 +1,230 @@ +/* + * GRUB -- GRand Unified Bootloader + * Copyright (C) 2009 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 "lua.h" +#include "lauxlib.h" +#include "lualib.h" +#include "grub_lib.h" + +#include <grub/dl.h> +#include <grub/command.h> +#include <grub/i18n.h> +#include <grub/misc.h> +#include <grub/normal.h> + +GRUB_MOD_LICENSE("GPLv3+"); + +static const char * +scan_str (const char *s1, const char *s2) +{ + while (*s1) + { + const char *p = s2; + + while (*p) + { + if (*s1 == *p) + return s1; + p++; + } + + s1++; + } + + return s1; +} + +int +strcspn (const char *s1, const char *s2) +{ + const char *r; + + r = scan_str (s1, s2); + return r - s1; +} + +char * +strpbrk (const char *s1, const char *s2) +{ + const char *r; + + r = scan_str (s1, s2); + return (*r) ? (char *) r : 0; +} + +static lua_State *state; + +/* Call `grub_error' to report a Lua error. The error message string must be + on the top of the Lua stack (at index -1). The error message is popped off + the Lua stack before this function returns. */ +static void +handle_lua_error (const char *error_type) +{ + const char *error_msg; + error_msg = lua_tostring (state, -1); + if (error_msg == NULL) + error_msg = "(error message not a string)"; + grub_error (GRUB_ERR_BAD_ARGUMENT, "%s: %s", error_type, error_msg); + /* Pop the error message. */ + lua_pop (state, 1); +} + +/* Taken from lua.c */ +static int +incomplete (lua_State * L, int status) +{ + if (status == LUA_ERRSYNTAX) + { + size_t lmsg; + const char *msg = lua_tolstring (L, -1, &lmsg); + const char *tp = msg + lmsg - (sizeof (LUA_QL ("<eof>")) - 1); + if (strstr (msg, LUA_QL ("<eof>")) == tp) + { + lua_pop (L, 1); + return 1; + } + } + return 0; /* else... */ +} + +static grub_err_t +interactive (void) +{ + const char *ps1 = "lua> "; + const char *ps2 = "lua>> "; + const char *prompt = ps1; + char *line; + char *chunk = NULL; + size_t len; + size_t oldlen = 0; + int r; + + grub_printf ("%s", N_ ("Welcome to lua, press the escape key to exit.")); + + while ((line = grub_cmdline_get (prompt)) != NULL) + { + /* len = lenght of chunk + line + newline character */ + len = oldlen + grub_strlen (line) + 1; + chunk = grub_realloc (chunk, len + 1); + grub_strcpy (chunk + oldlen , line); + chunk[len - 1] = '\n'; + chunk[len] = '\0'; + grub_free (line); + + r = luaL_loadbuffer (state, chunk, len, "stdin"); + if (!r) + { + /* No error: Execute this chunk and prepare to read another */ + r = lua_pcall (state, 0, 0, 0); + if (r) + { + handle_lua_error ("Lua"); + grub_print_error (); + } + + grub_free (chunk); + chunk = NULL; + len = 0; + prompt = ps1; + } + else if (incomplete (state, r)) + { + /* Chunk is incomplete, try reading another line */ + prompt = ps2; + } + else if (r == LUA_ERRSYNTAX) + { + handle_lua_error ("Lua"); + grub_print_error (); + + /* This chunk is garbage, try starting another one */ + grub_free (chunk); + chunk = NULL; + len = 0; + prompt = ps1; + } + else + { + /* Handle errors other than syntax errors (out of memory, etc.) */ + grub_free (chunk); + handle_lua_error ("Lua parser failed"); + return grub_errno; + } + + oldlen = len; + } + + grub_free (chunk); + lua_gc (state, LUA_GCCOLLECT, 0); + + return grub_errno; +} + +static grub_err_t +grub_cmd_lua (grub_command_t cmd __attribute__ ((unused)), + int argc, char **args) +{ + if (argc == 1) + { + if (luaL_loadfile (state, args[0])) + { + handle_lua_error ("Lua"); + } + else if (lua_pcall (state, 0, 0, 0)) + { + handle_lua_error ("Lua"); + } + } + else if (argc == 0) + { + return interactive (); + } + else + { + return grub_error (GRUB_ERR_BAD_ARGUMENT, "1 or 0 arguments expected"); + } + + return grub_errno; +} + +static grub_command_t cmd; + +GRUB_MOD_INIT (lua) +{ + (void) mod; + + state = lua_open (); + if (state) + { + lua_gc (state, LUA_GCSTOP, 0); + luaL_openlibs (state); + luaL_register (state, "grub", grub_lua_lib); + lua_gc (state, LUA_GCRESTART, 0); + cmd = grub_register_command ("lua", grub_cmd_lua, N_("[FILE]"), + N_ ("Run lua script FILE or start interactive lua shell")); + } +} + +GRUB_MOD_FINI (lua) +{ + if (state) + { + lua_close (state); + grub_unregister_command (cmd); + } +} |