2 * Copyright (c) 2011 Wojciech A. Koszek <wkoszek@FreeBSD.org>
3 * Copyright (c) 2014 Pedro Souza <pedrosouza@freebsd.org>
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 #include <sys/cdefs.h>
29 __FBSDID("$FreeBSD$");
32 #include "bootstrap.h"
47 struct interp_lua_softc {
51 static struct interp_lua_softc lua_softc;
54 #define LDBG(...) do { \
55 printf("%s(%d): ", __func__, __LINE__); \
56 printf(__VA_ARGS__); \
63 #define LOADER_LUA LUA_PATH "/loader.lua"
68 interp_lua_realloc(void *ud __unused, void *ptr, size_t osize __unused, size_t nsize)
75 return realloc(ptr, nsize);
79 * The libraries commented out below either lack the proper
80 * support from libsa, or they are unlikely to be useful
81 * in the bootloader, so have been commented out.
83 static const luaL_Reg loadedlibs[] = {
85 {LUA_LOADLIBNAME, luaopen_package},
86 // {LUA_COLIBNAME, luaopen_coroutine},
87 // {LUA_TABLIBNAME, luaopen_table},
88 {LUA_STRLIBNAME, luaopen_string},
89 // {LUA_IOLIBNAME, luaopen_io},
90 // {LUA_OSLIBNAME, luaopen_os},
91 // {LUA_MATHLIBNAME, luaopen_math},
92 // {LUA_UTF8LIBNAME, luaopen_utf8},
93 // {LUA_DBLIBNAME, luaopen_debug},
94 {"errno", luaopen_errno},
97 {"loader", luaopen_loader},
98 {"pager", luaopen_pager},
106 struct interp_lua_softc *softc = &lua_softc;
107 const char *filename;
112 setenv("script.lang", "lua", 1);
113 LDBG("creating context");
115 luap = lua_newstate(interp_lua_realloc, NULL);
117 printf("problem initializing the Lua interpreter\n");
122 /* "require" functions from 'loadedlibs' and set results to global table */
123 for (lib = loadedlibs; lib->func; lib++) {
124 luaL_requiref(luap, lib->name, lib->func, 1);
125 lua_pop(luap, 1); /* remove lib */
128 filename = LOADER_LUA;
129 if (interp_include(filename) != 0) {
130 const char *errstr = lua_tostring(luap, -1);
131 errstr = errstr == NULL ? "unknown" : errstr;
132 printf("ERROR: %s.\n", errstr);
134 setenv("autoboot_delay", "NO", 1);
141 interp_run(const char *line)
146 struct interp_lua_softc *softc = &lua_softc;
151 LDBG("executing line...");
152 if ((status = luaL_dostring(luap, line)) != 0) {
155 * The line wasn't executable as lua; run it through parse to
156 * to get consistent parsing of command line arguments, then
157 * run it through cli_execute. If that fails, then we'll try it
160 command_errmsg = NULL;
161 if (parse(&argc, &argv, line) == 0) {
162 lua_getglobal(luap, "cli_execute");
163 for (nargc = 0; nargc < argc; ++nargc) {
164 lua_pushstring(luap, argv[nargc]);
166 status = lua_pcall(luap, argc, 1, 0);
167 ret = lua_tointeger(luap, 1);
169 if (status != 0 || ret != 0) {
171 * Lua cli_execute will pass the function back
172 * through loader.command, which is a proxy to
173 * interp_builtin_cmd. If we failed to interpret
174 * the command, though, then there's a chance
175 * that didn't happen. Call interp_builtin_cmd
176 * directly if our lua_pcall was not successful.
178 status = interp_builtin_cmd(argc, argv);
181 if (command_errmsg != NULL)
182 printf("%s\n", command_errmsg);
184 printf("Command failed\n");
189 printf("Failed to parse \'%s\'\n", line);
195 return (status == 0 ? CMD_OK : CMD_ERROR);
199 interp_include(const char *filename)
201 struct interp_lua_softc *softc = &lua_softc;
203 LDBG("loading file %s", filename);
205 return (luaL_dofile(softc->luap, filename));