2009-07-05 Bean <bean123ch@gmail.com>
* conf/common.rmk (lua_mode_SOURCES): Add script/lua/lstrlib.c. * script/lua/grub_lib.c (grub_lua_run): Check input parameter. (grub_lua_getenv): Likewise. (grub_lua_setenv): Likewise. (save_errno): New function. (push_result): Likewise. (grub_lua_enum_device): Likewise. (grub_lua_enum_file): Likewise. (grub_lua_file_open): Likewise. (grub_lua_file_close): Likewise. (grub_lua_file_seek): Likewise. (grub_lua_file_read): Likewise. (grub_lua_file_getline): Likewise. (grub_lua_file_getsize): Likewise. (grub_lua_file_getpos): Likewise. (grub_lua_file_eof): Likewise. (grub_lua_file_exist): Likewise. (grub_lua_add_menu): Likewise. * script/lua/grub_lua.h (isupper): New inline function. (islower): Likewise. (ispunct): Likewise. (isxdigit): Likewise. (strcspn): Change to normal function. (strpbkr): New function declaration. (memchr): Likewise. * script/lua/grub_main.c (scan_str): New function. (strcspn): Likewise. (strpbrk): Likewise. (memchr): Likewise. * script/lua/linit.c (lualibs): Enable the string library. * util/osdetect.lua: New file.
This commit is contained in:
parent
2da9229550
commit
b4a1dc7990
7 changed files with 581 additions and 41 deletions
39
ChangeLog
39
ChangeLog
|
@ -1,3 +1,42 @@
|
||||||
|
2009-07-05 Bean <bean123ch@gmail.com>
|
||||||
|
|
||||||
|
* conf/common.rmk (lua_mode_SOURCES): Add script/lua/lstrlib.c.
|
||||||
|
|
||||||
|
* script/lua/grub_lib.c (grub_lua_run): Check input parameter.
|
||||||
|
(grub_lua_getenv): Likewise.
|
||||||
|
(grub_lua_setenv): Likewise.
|
||||||
|
(save_errno): New function.
|
||||||
|
(push_result): Likewise.
|
||||||
|
(grub_lua_enum_device): Likewise.
|
||||||
|
(grub_lua_enum_file): Likewise.
|
||||||
|
(grub_lua_file_open): Likewise.
|
||||||
|
(grub_lua_file_close): Likewise.
|
||||||
|
(grub_lua_file_seek): Likewise.
|
||||||
|
(grub_lua_file_read): Likewise.
|
||||||
|
(grub_lua_file_getline): Likewise.
|
||||||
|
(grub_lua_file_getsize): Likewise.
|
||||||
|
(grub_lua_file_getpos): Likewise.
|
||||||
|
(grub_lua_file_eof): Likewise.
|
||||||
|
(grub_lua_file_exist): Likewise.
|
||||||
|
(grub_lua_add_menu): Likewise.
|
||||||
|
|
||||||
|
* script/lua/grub_lua.h (isupper): New inline function.
|
||||||
|
(islower): Likewise.
|
||||||
|
(ispunct): Likewise.
|
||||||
|
(isxdigit): Likewise.
|
||||||
|
(strcspn): Change to normal function.
|
||||||
|
(strpbkr): New function declaration.
|
||||||
|
(memchr): Likewise.
|
||||||
|
|
||||||
|
* script/lua/grub_main.c (scan_str): New function.
|
||||||
|
(strcspn): Likewise.
|
||||||
|
(strpbrk): Likewise.
|
||||||
|
(memchr): Likewise.
|
||||||
|
|
||||||
|
* script/lua/linit.c (lualibs): Enable the string library.
|
||||||
|
|
||||||
|
* util/osdetect.lua: New file.
|
||||||
|
|
||||||
2009-07-04 Robert Millan <rmh.grub@aybabtu.com>
|
2009-07-04 Robert Millan <rmh.grub@aybabtu.com>
|
||||||
|
|
||||||
* include/grub/i386/linux.h (struct linux_kernel_params): Add
|
* include/grub/i386/linux.h (struct linux_kernel_params): Add
|
||||||
|
|
|
@ -519,15 +519,14 @@ lua_mod_SOURCES = script/lua/lapi.c script/lua/lcode.c script/lua/ldebug.c \
|
||||||
script/lua/lstate.c script/lua/lstring.c script/lua/ltable.c \
|
script/lua/lstate.c script/lua/lstring.c script/lua/ltable.c \
|
||||||
script/lua/ltm.c script/lua/lundump.c script/lua/lvm.c \
|
script/lua/ltm.c script/lua/lundump.c script/lua/lvm.c \
|
||||||
script/lua/lzio.c script/lua/lauxlib.c script/lua/lbaselib.c \
|
script/lua/lzio.c script/lua/lauxlib.c script/lua/lbaselib.c \
|
||||||
script/lua/linit.c script/lua/ltablib.c \
|
script/lua/linit.c script/lua/ltablib.c script/lua/lstrlib.c \
|
||||||
script/lua/grub_main.c script/lua/grub_lib.c
|
script/lua/grub_main.c script/lua/grub_lib.c
|
||||||
lua_mod_CFLAGS = $(COMMON_CFLAGS)
|
lua_mod_CFLAGS = $(COMMON_CFLAGS)
|
||||||
lua_mod_LDFLAGS = $(COMMON_LDFLAGS)
|
lua_mod_LDFLAGS = $(COMMON_LDFLAGS)
|
||||||
|
|
||||||
# Extra libraries for lua
|
# Extra libraries for lua
|
||||||
# script/lua/lmathlib.c script/lua/loslib.c script/lua/liolib.c
|
# script/lua/lmathlib.c script/lua/loslib.c script/lua/liolib.c
|
||||||
# script/lua/lstrlib.c script/lua/ldblib.c script/lua/ltablib.c
|
# script/lua/ldblib.c script/lua/loadlib.c
|
||||||
# script/lua/loadlib.c
|
|
||||||
|
|
||||||
# Common Video Subsystem specific modules.
|
# Common Video Subsystem specific modules.
|
||||||
pkglib_MODULES += video.mod videotest.mod bitmap.mod tga.mod jpeg.mod \
|
pkglib_MODULES += video.mod videotest.mod bitmap.mod tga.mod jpeg.mod \
|
||||||
|
|
|
@ -24,18 +24,47 @@
|
||||||
#include <grub/env.h>
|
#include <grub/env.h>
|
||||||
#include <grub/parser.h>
|
#include <grub/parser.h>
|
||||||
#include <grub/command.h>
|
#include <grub/command.h>
|
||||||
|
#include <grub/normal.h>
|
||||||
|
#include <grub/file.h>
|
||||||
|
#include <grub/device.h>
|
||||||
|
|
||||||
|
static int
|
||||||
|
save_errno (lua_State *state)
|
||||||
|
{
|
||||||
|
int saved_errno;
|
||||||
|
|
||||||
|
saved_errno = grub_errno;
|
||||||
|
grub_errno = 0;
|
||||||
|
|
||||||
|
lua_pushinteger (state, saved_errno);
|
||||||
|
lua_setfield (state, LUA_GLOBALSINDEX, "grub_errno");
|
||||||
|
|
||||||
|
if (saved_errno)
|
||||||
|
lua_pushstring (state, grub_errmsg);
|
||||||
|
else
|
||||||
|
lua_pushnil (state);
|
||||||
|
|
||||||
|
lua_setfield (state, LUA_GLOBALSINDEX, "grub_errmsg");
|
||||||
|
|
||||||
|
return saved_errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
push_result (lua_State *state)
|
||||||
|
{
|
||||||
|
lua_pushinteger (state, save_errno (state));
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
static int
|
static int
|
||||||
grub_lua_run (lua_State *state)
|
grub_lua_run (lua_State *state)
|
||||||
{
|
{
|
||||||
int n;
|
int n;
|
||||||
char **args;
|
char **args;
|
||||||
grub_err_t result;
|
const char *s;
|
||||||
|
|
||||||
if (! lua_gettop(state))
|
s = luaL_checkstring (state, 1);
|
||||||
return 0;
|
if ((! grub_parser_split_cmdline (s, 0, &n, &args))
|
||||||
|
|
||||||
if ((! grub_parser_split_cmdline (lua_tostring (state, 1), 0, &n, &args))
|
|
||||||
&& (n >= 0))
|
&& (n >= 0))
|
||||||
{
|
{
|
||||||
grub_command_t cmd;
|
grub_command_t cmd;
|
||||||
|
@ -50,15 +79,7 @@ grub_lua_run (lua_State *state)
|
||||||
grub_free (args);
|
grub_free (args);
|
||||||
}
|
}
|
||||||
|
|
||||||
result = grub_errno;
|
return push_result (state);
|
||||||
grub_errno = 0;
|
|
||||||
|
|
||||||
lua_pushinteger(state, result);
|
|
||||||
|
|
||||||
if (result)
|
|
||||||
lua_pushstring (state, grub_errmsg);
|
|
||||||
|
|
||||||
return (result) ? 2 : 1;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static int
|
static int
|
||||||
|
@ -71,7 +92,7 @@ grub_lua_getenv (lua_State *state)
|
||||||
{
|
{
|
||||||
const char *name, *value;
|
const char *name, *value;
|
||||||
|
|
||||||
name = lua_tostring (state, i);
|
name = luaL_checkstring (state, i);
|
||||||
value = grub_env_get (name);
|
value = grub_env_get (name);
|
||||||
if (value)
|
if (value)
|
||||||
lua_pushstring (state, value);
|
lua_pushstring (state, value);
|
||||||
|
@ -87,11 +108,8 @@ grub_lua_setenv (lua_State *state)
|
||||||
{
|
{
|
||||||
const char *name, *value;
|
const char *name, *value;
|
||||||
|
|
||||||
if (lua_gettop(state) != 2)
|
name = luaL_checkstring (state, 1);
|
||||||
return 0;
|
value = luaL_checkstring (state, 2);
|
||||||
|
|
||||||
name = lua_tostring (state, 1);
|
|
||||||
value = lua_tostring (state, 2);
|
|
||||||
|
|
||||||
if (name[0])
|
if (name[0])
|
||||||
grub_env_set (name, value);
|
grub_env_set (name, value);
|
||||||
|
@ -99,10 +117,322 @@ grub_lua_setenv (lua_State *state)
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_enum_device (lua_State *state)
|
||||||
|
{
|
||||||
|
auto int enum_device (const char *name);
|
||||||
|
int enum_device (const char *name)
|
||||||
|
{
|
||||||
|
int result;
|
||||||
|
grub_device_t dev;
|
||||||
|
|
||||||
|
result = 0;
|
||||||
|
dev = grub_device_open (name);
|
||||||
|
if (dev)
|
||||||
|
{
|
||||||
|
grub_fs_t fs;
|
||||||
|
|
||||||
|
fs = grub_fs_probe (dev);
|
||||||
|
if (fs)
|
||||||
|
{
|
||||||
|
lua_pushvalue (state, 1);
|
||||||
|
lua_pushstring (state, name);
|
||||||
|
lua_pushstring (state, fs->name);
|
||||||
|
if (! fs->uuid)
|
||||||
|
lua_pushnil (state);
|
||||||
|
else
|
||||||
|
{
|
||||||
|
int err;
|
||||||
|
char *uuid;
|
||||||
|
|
||||||
|
err = fs->uuid (dev, &uuid);
|
||||||
|
if (err)
|
||||||
|
{
|
||||||
|
grub_errno = 0;
|
||||||
|
lua_pushnil (state);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
lua_pushstring (state, uuid);
|
||||||
|
grub_free (uuid);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
lua_call (state, 3, 1);
|
||||||
|
result = lua_tointeger (state, -1);
|
||||||
|
lua_pop (state, 1);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
grub_errno = 0;
|
||||||
|
grub_device_close (dev);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
grub_errno = 0;
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TFUNCTION);
|
||||||
|
grub_device_iterate (enum_device);
|
||||||
|
return push_result (state);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_enum_file (lua_State *state)
|
||||||
|
{
|
||||||
|
char *device_name;
|
||||||
|
const char *arg;
|
||||||
|
grub_device_t dev;
|
||||||
|
|
||||||
|
auto int enum_file (const char *name, const struct grub_dirhook_info *info);
|
||||||
|
int enum_file (const char *name, const struct grub_dirhook_info *info)
|
||||||
|
{
|
||||||
|
int result;
|
||||||
|
|
||||||
|
lua_pushvalue (state, 1);
|
||||||
|
lua_pushstring (state, name);
|
||||||
|
lua_pushinteger (state, info->dir != 0);
|
||||||
|
lua_call (state, 2, 1);
|
||||||
|
result = lua_tointeger (state, -1);
|
||||||
|
lua_pop (state, 1);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TFUNCTION);
|
||||||
|
arg = luaL_checkstring (state, 2);
|
||||||
|
device_name = grub_file_get_device_name (arg);
|
||||||
|
dev = grub_device_open (device_name);
|
||||||
|
if (dev)
|
||||||
|
{
|
||||||
|
grub_fs_t fs;
|
||||||
|
const char *path;
|
||||||
|
|
||||||
|
fs = grub_fs_probe (dev);
|
||||||
|
path = grub_strchr (arg, ')');
|
||||||
|
if (! path)
|
||||||
|
path = arg;
|
||||||
|
else
|
||||||
|
path++;
|
||||||
|
|
||||||
|
if (fs)
|
||||||
|
{
|
||||||
|
(fs->dir) (dev, path, enum_file);
|
||||||
|
}
|
||||||
|
|
||||||
|
grub_device_close (dev);
|
||||||
|
}
|
||||||
|
|
||||||
|
grub_free (device_name);
|
||||||
|
|
||||||
|
return push_result (state);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_open (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
const char *name;
|
||||||
|
|
||||||
|
name = luaL_checkstring (state, 1);
|
||||||
|
file = grub_file_open (name);
|
||||||
|
save_errno (state);
|
||||||
|
|
||||||
|
if (! file)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
lua_pushlightuserdata (state, file);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_close (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TLIGHTUSERDATA);
|
||||||
|
file = lua_touserdata (state, 1);
|
||||||
|
grub_file_close (file);
|
||||||
|
|
||||||
|
return push_result (state);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_seek (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
grub_off_t offset;
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TLIGHTUSERDATA);
|
||||||
|
file = lua_touserdata (state, 1);
|
||||||
|
offset = luaL_checkinteger (state, 2);
|
||||||
|
|
||||||
|
offset = grub_file_seek (file, offset);
|
||||||
|
save_errno (state);
|
||||||
|
|
||||||
|
lua_pushinteger (state, offset);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_read (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
luaL_Buffer b;
|
||||||
|
int n;
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TLIGHTUSERDATA);
|
||||||
|
file = lua_touserdata (state, 1);
|
||||||
|
n = luaL_checkinteger (state, 2);
|
||||||
|
|
||||||
|
luaL_buffinit (state, &b);
|
||||||
|
while (n)
|
||||||
|
{
|
||||||
|
char *p;
|
||||||
|
int nr;
|
||||||
|
|
||||||
|
nr = (n > LUAL_BUFFERSIZE) ? LUAL_BUFFERSIZE : n;
|
||||||
|
p = luaL_prepbuffer (&b);
|
||||||
|
|
||||||
|
nr = grub_file_read (file, p, nr);
|
||||||
|
if (nr <= 0)
|
||||||
|
break;
|
||||||
|
|
||||||
|
luaL_addsize (&b, nr);
|
||||||
|
n -= nr;
|
||||||
|
}
|
||||||
|
|
||||||
|
save_errno (state);
|
||||||
|
luaL_pushresult (&b);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_getline (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
char *line;
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TLIGHTUSERDATA);
|
||||||
|
file = lua_touserdata (state, 1);
|
||||||
|
|
||||||
|
line = grub_file_getline (file);
|
||||||
|
save_errno (state);
|
||||||
|
|
||||||
|
if (! line)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
lua_pushstring (state, line);
|
||||||
|
grub_free (line);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_getsize (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TLIGHTUSERDATA);
|
||||||
|
file = lua_touserdata (state, 1);
|
||||||
|
|
||||||
|
lua_pushinteger (state, file->size);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_getpos (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TLIGHTUSERDATA);
|
||||||
|
file = lua_touserdata (state, 1);
|
||||||
|
|
||||||
|
lua_pushinteger (state, file->offset);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_eof (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
|
||||||
|
luaL_checktype (state, 1, LUA_TLIGHTUSERDATA);
|
||||||
|
file = lua_touserdata (state, 1);
|
||||||
|
|
||||||
|
lua_pushboolean (state, file->offset >= file->size);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_file_exist (lua_State *state)
|
||||||
|
{
|
||||||
|
grub_file_t file;
|
||||||
|
const char *name;
|
||||||
|
int result;
|
||||||
|
|
||||||
|
result = 0;
|
||||||
|
name = luaL_checkstring (state, 1);
|
||||||
|
file = grub_file_open (name);
|
||||||
|
if (file)
|
||||||
|
{
|
||||||
|
result++;
|
||||||
|
grub_file_close (file);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
grub_errno = 0;
|
||||||
|
|
||||||
|
lua_pushboolean (state, result);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
grub_lua_add_menu (lua_State *state)
|
||||||
|
{
|
||||||
|
int n;
|
||||||
|
const char *source;
|
||||||
|
|
||||||
|
source = luaL_checklstring (state, 1, 0);
|
||||||
|
n = lua_gettop (state) - 1;
|
||||||
|
if (n > 0)
|
||||||
|
{
|
||||||
|
const char *args[sizeof (char *) * n];
|
||||||
|
char *p;
|
||||||
|
int i;
|
||||||
|
|
||||||
|
for (i = 0; i < n; i++)
|
||||||
|
args[i] = luaL_checkstring (state, 2 + i);
|
||||||
|
|
||||||
|
p = grub_strdup (source);
|
||||||
|
if (! p)
|
||||||
|
return push_result (state);
|
||||||
|
|
||||||
|
grub_normal_add_menu_entry (n, args, p);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
lua_pushstring (state, "not enough parameter");
|
||||||
|
lua_error (state);
|
||||||
|
}
|
||||||
|
|
||||||
|
return push_result (state);
|
||||||
|
}
|
||||||
|
|
||||||
luaL_Reg grub_lua_lib[] =
|
luaL_Reg grub_lua_lib[] =
|
||||||
{
|
{
|
||||||
{"run", grub_lua_run},
|
{"run", grub_lua_run},
|
||||||
{"getenv", grub_lua_getenv},
|
{"getenv", grub_lua_getenv},
|
||||||
{"setenv", grub_lua_setenv},
|
{"setenv", grub_lua_setenv},
|
||||||
|
{"enum_device", grub_lua_enum_device},
|
||||||
|
{"enum_file", grub_lua_enum_file},
|
||||||
|
{"file_open", grub_lua_file_open},
|
||||||
|
{"file_close", grub_lua_file_close},
|
||||||
|
{"file_seek", grub_lua_file_seek},
|
||||||
|
{"file_read", grub_lua_file_read},
|
||||||
|
{"file_getline", grub_lua_file_getline},
|
||||||
|
{"file_getsize", grub_lua_file_getsize},
|
||||||
|
{"file_getpos", grub_lua_file_getpos},
|
||||||
|
{"file_eof", grub_lua_file_eof},
|
||||||
|
{"file_exist", grub_lua_file_exist},
|
||||||
|
{"add_menu", grub_lua_add_menu},
|
||||||
{0, 0}
|
{0, 0}
|
||||||
};
|
};
|
||||||
|
|
|
@ -77,26 +77,28 @@ iscntrl (int c)
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int
|
static inline int
|
||||||
strcspn(const char *s1, const char *s2)
|
isupper (int c)
|
||||||
{
|
{
|
||||||
int size = 0;
|
return ((c >= 'A') && (c <= 'Z'));
|
||||||
|
|
||||||
while (*s1)
|
|
||||||
{
|
|
||||||
const char *p = s2;
|
|
||||||
|
|
||||||
while (*p)
|
|
||||||
{
|
|
||||||
if (*s1 == *p)
|
|
||||||
return size;
|
|
||||||
p++;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
s1++;
|
static inline int
|
||||||
size++;
|
islower (int c)
|
||||||
|
{
|
||||||
|
return ((c >= 'a') && (c <= 'z'));
|
||||||
}
|
}
|
||||||
|
|
||||||
return size;
|
static inline int
|
||||||
|
ispunct (int c)
|
||||||
|
{
|
||||||
|
return ((! isspace (c)) && (! isalnum (c)));
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline int
|
||||||
|
isxdigit (int c)
|
||||||
|
{
|
||||||
|
return (isdigit (c) || ((c >= 'a') && (c <= 'f')) ||
|
||||||
|
((c >= 'A') && (c <= 'F')));
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline int
|
static inline int
|
||||||
|
@ -105,4 +107,8 @@ abs (int c)
|
||||||
return (c >= 0) ? : -c;
|
return (c >= 0) ? : -c;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int strcspn (const char *s1, const char *s2);
|
||||||
|
char *strpbrk (const char *s1, const char *s2);
|
||||||
|
void *memchr (const void *s, int c, size_t n);
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -24,6 +24,61 @@
|
||||||
#include <grub/dl.h>
|
#include <grub/dl.h>
|
||||||
#include <grub/parser.h>
|
#include <grub/parser.h>
|
||||||
|
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
|
void *
|
||||||
|
memchr (const void *s, int c, size_t n)
|
||||||
|
{
|
||||||
|
const unsigned char *p = s;
|
||||||
|
|
||||||
|
while (n)
|
||||||
|
{
|
||||||
|
if (*p == c)
|
||||||
|
return (void *) p;
|
||||||
|
|
||||||
|
n--;
|
||||||
|
p++;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
static lua_State *state;
|
static lua_State *state;
|
||||||
|
|
||||||
/* Call `grub_error' to report a Lua error. The error message string must be
|
/* Call `grub_error' to report a Lua error. The error message string must be
|
||||||
|
|
|
@ -20,7 +20,7 @@ static const luaL_Reg lualibs[] = {
|
||||||
{LUA_TABLIBNAME, luaopen_table},
|
{LUA_TABLIBNAME, luaopen_table},
|
||||||
// {LUA_IOLIBNAME, luaopen_io},
|
// {LUA_IOLIBNAME, luaopen_io},
|
||||||
// {LUA_OSLIBNAME, luaopen_os},
|
// {LUA_OSLIBNAME, luaopen_os},
|
||||||
// {LUA_STRLIBNAME, luaopen_string},
|
{LUA_STRLIBNAME, luaopen_string},
|
||||||
// {LUA_MATHLIBNAME, luaopen_math},
|
// {LUA_MATHLIBNAME, luaopen_math},
|
||||||
// {LUA_DBLIBNAME, luaopen_debug},
|
// {LUA_DBLIBNAME, luaopen_debug},
|
||||||
{NULL, NULL}
|
{NULL, NULL}
|
||||||
|
|
111
util/osdetect.lua
Normal file
111
util/osdetect.lua
Normal file
|
@ -0,0 +1,111 @@
|
||||||
|
#!lua
|
||||||
|
--
|
||||||
|
-- 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/>.
|
||||||
|
--
|
||||||
|
|
||||||
|
function enum_device (device, fs, uuid)
|
||||||
|
local root
|
||||||
|
local title
|
||||||
|
local source
|
||||||
|
local kernels = {}
|
||||||
|
local kernel_num = 0
|
||||||
|
|
||||||
|
local function enum_file (name)
|
||||||
|
local version
|
||||||
|
|
||||||
|
version = string.match (name, "vmlinuz%-(.*)")
|
||||||
|
if (version ~= nil) then
|
||||||
|
table.insert (kernels, version)
|
||||||
|
kernel_num = kernel_num + 1
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
local function sort_kernel (first, second)
|
||||||
|
local a1, a2, a3, a4, b1, b2, b3, b4
|
||||||
|
|
||||||
|
a1, a2, a3, a4 = string.match (first, "(%d+)%.?(%d*).?(%d*)%-?(%d*)")
|
||||||
|
b1, b2, b3, b4 = string.match (second, "(%d+)%.?(%d*).?(%d*)%-?(%d*)")
|
||||||
|
return (a1 > b1) or (a2 > b2) or (a3 > b3) or (a4 < b4);
|
||||||
|
end
|
||||||
|
|
||||||
|
root = "(" .. device .. ")/"
|
||||||
|
source = "root (" .. device .. ")\nchainloader +1"
|
||||||
|
title = nil
|
||||||
|
if (grub.file_exist (root .. "bootmgr") and
|
||||||
|
grub.file_exist (root .. "boot/bcd")) then
|
||||||
|
title = "Windows Vista bootmgr"
|
||||||
|
elseif (grub.file_exist (root .. "ntldr") and
|
||||||
|
grub.file_exist (root .. "ntdetect.com") and
|
||||||
|
grub.file_exist (root .. "boot.ini")) then
|
||||||
|
title = "Windows NT/2000/XP loader"
|
||||||
|
elseif (grub.file_exist (root .. "windows/win.com")) then
|
||||||
|
title = "Windows 98/ME"
|
||||||
|
elseif (grub.file_exist (root .. "io.sys") and
|
||||||
|
grub.file_exist (root .. "command.com")) then
|
||||||
|
title = "MS-DOS"
|
||||||
|
elseif (grub.file_exist (root .. "kernel.sys")) then
|
||||||
|
title = "FreeDOS"
|
||||||
|
elseif (grub.file_exist (root .. "boot/loader") and
|
||||||
|
grub.file_exist (root .. "boot/devices.hints")) then
|
||||||
|
source = "root (" .. device .. ")\nfreebsd /boot/loader" ..
|
||||||
|
"\nfreebsd_loadenv /boot/device.hints"
|
||||||
|
title = "FreeBSD"
|
||||||
|
else
|
||||||
|
grub.enum_file (enum_file, root .. "boot")
|
||||||
|
if kernel_num ~= 0 then
|
||||||
|
table.sort (kernels, sort_kernel)
|
||||||
|
for i = 1, kernel_num do
|
||||||
|
local initrd
|
||||||
|
|
||||||
|
title = "Linux " .. kernels[i]
|
||||||
|
source = "root (" .. device ..
|
||||||
|
")\nlinux /boot/vmlinuz-" .. kernels[i] ..
|
||||||
|
" root=UUID=" .. uuid .. " ro"
|
||||||
|
|
||||||
|
if grub.file_exist (root .. "boot/initrd-" ..
|
||||||
|
kernels[i] .. ".img") then
|
||||||
|
initrd = "\ninitrd /boot/initrd-" .. kernels[i] .. ".img"
|
||||||
|
elseif grub.file_exist (root .. "boot/initrd.img-" .. kernels[i]) then
|
||||||
|
initrd = "\ninitrd /boot/initrd.img-" .. kernels[i]
|
||||||
|
elseif grub.file_exist (root .. "boot/initrd-" .. kernels[i]) then
|
||||||
|
initrd = "\ninitrd /boot/initrd-" .. kernels[i]
|
||||||
|
else
|
||||||
|
initrd = ""
|
||||||
|
end
|
||||||
|
|
||||||
|
grub.add_menu (source .. initrd, title)
|
||||||
|
grub.add_menu (source .. " single" .. initrd,
|
||||||
|
title .. " (single-user mode)")
|
||||||
|
end
|
||||||
|
return 0
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
if title == nil then
|
||||||
|
local partition = string.match (device, ".*,(%d+)")
|
||||||
|
|
||||||
|
if (partition ~= nil) and (tonumber (partition) > 4) then
|
||||||
|
return 0
|
||||||
|
end
|
||||||
|
|
||||||
|
title = "Other OS"
|
||||||
|
end
|
||||||
|
|
||||||
|
grub.add_menu (source, title)
|
||||||
|
return 0
|
||||||
|
end
|
||||||
|
|
||||||
|
grub.enum_device (enum_device)
|
Loading…
Reference in a new issue