summaryrefslogtreecommitdiffstats
path: root/debian/grub-extras/lua/grub_main.c
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--debian/grub-extras/lua/grub_main.c230
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);
+ }
+}