From e53a39d9ec28b24ea0d408f1500e987d005cd651 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Wed, 19 Apr 2017 15:34:04 +0200 Subject: Lua shell :) --- src/sysbin/lx/lxinit.c | 48 ++++++++++ src/sysbin/lx/lxlib.h | 15 ++++ src/sysbin/lx/lxsyslib.c | 221 +++++++++++++++++++++++++++++++++++++++++++++++ src/sysbin/lx/main.c | 50 +---------- 4 files changed, 288 insertions(+), 46 deletions(-) create mode 100644 src/sysbin/lx/lxinit.c create mode 100644 src/sysbin/lx/lxlib.h create mode 100644 src/sysbin/lx/lxsyslib.c (limited to 'src/sysbin/lx') diff --git a/src/sysbin/lx/lxinit.c b/src/sysbin/lx/lxinit.c new file mode 100644 index 0000000..c0059c7 --- /dev/null +++ b/src/sysbin/lx/lxinit.c @@ -0,0 +1,48 @@ + + +#define lxinit_c +#define LUA_LIB + +/* +** If you embed Lua in your program and need to open the standard +** libraries, call luaL_openlibs in your program. If you need a +** different set of libraries, copy this file to your project and edit +** it to suit your needs. +** +** You can also *preload* libraries, so that a later 'require' can +** open the library, which is already linked to the application. +** For that, do the following code: +** +** luaL_getsubtable(L, LUA_REGISTRYINDEX, "_PRELOAD"); +** lua_pushcfunction(L, luaopen_modname); +** lua_setfield(L, -2, modname); +** lua_pop(L, 1); // remove _PRELOAD table +*/ + +#include + + +#include + +#include + +#include +#include + +#include "lxlib.h" + + +static const luaL_Reg loadedlibs[] = { + {LX_SYSLIBNAME, lx_open_sys}, + {NULL, NULL} +}; + +LUALIB_API void lx_openlibs (lua_State *L) { + const luaL_Reg *lib; + /* "require" functions from 'loadedlibs' and set results to global table */ + for (lib = loadedlibs; lib->func; lib++) { + luaL_requiref(L, lib->name, lib->func, 1); + lua_pop(L, 1); /* remove lib */ + } +} + diff --git a/src/sysbin/lx/lxlib.h b/src/sysbin/lx/lxlib.h new file mode 100644 index 0000000..e65ab6f --- /dev/null +++ b/src/sysbin/lx/lxlib.h @@ -0,0 +1,15 @@ +/* +** Lua eXtended libraries +*/ +#pragma once + +#include + + +#define LX_SYSLIBNAME "lx.sys" +LUAMOD_API int (lx_open_sys) (lua_State *L); + + +/* open all previous libraries */ +LUALIB_API void (lx_openlibs) (lua_State *L); + diff --git a/src/sysbin/lx/lxsyslib.c b/src/sysbin/lx/lxsyslib.c new file mode 100644 index 0000000..5bfec3e --- /dev/null +++ b/src/sysbin/lx/lxsyslib.c @@ -0,0 +1,221 @@ +/* +** Lua eXtended system library +*/ + +#define lxsyslib_c +#define LUA_LIB + +#include + + +#include +#include +#include +#include +#include + +#include + +#include +#include + +#include "lxlib.h" + + + +static void setintfield (lua_State *L, const char *key, int value) { + lua_pushinteger(L, value); + lua_setfield(L, -2, key); +} + +static void setstrfield (lua_State *L, const char *key, const char* value) { + lua_pushstring(L, value); + lua_setfield(L, -2, key); +} + + +static int sys_dbg_print(lua_State *L) { + const char *str = luaL_checkstring(L, 1); + sc_dbg_print(str); + return 0; +} + +static int sys_yield(lua_State *L) { + sc_yield(); + return 0; +} + +static int sys_exit(lua_State *L) { + int code = luaL_checkinteger(L, 1); + sc_exit(code); + return 0; +} + +static int sys_usleep(lua_State *L) { + int usecs = luaL_checkinteger(L, 1); + sc_usleep(usecs); + return 0; +} + +/* + Missing syscalls : + - sys_new_thread, exit_thread -> into specific threading library + - mmap, mmap_file, mchmap, munmap -> into specific memory-related library ?? +*/ + +static int sys_create(lua_State *L) { + const char *name = luaL_checkstring(L, 1); + int type = luaL_checkinteger(L, 2); + lua_pushboolean(L, sc_create(name, type)); + return 1; +} + +static int sys_delete(lua_State *L) { + const char *name = luaL_checkstring(L, 1); + lua_pushboolean(L, sc_delete(name)); + return 1; +} + +static int sys_move(lua_State *L) { + const char *oldname = luaL_checkstring(L, 1); + const char *newname = luaL_checkstring(L, 2); + lua_pushboolean(L, sc_move(oldname, newname)); + return 1; +} + +static int sys_stat(lua_State *L) { + const char* name = luaL_checkstring(L, 1); + stat_t s; + if (sc_stat(name, &s)) { + lua_createtable(L, 0, 3); + setintfield(L, "type", s.type); + setintfield(L, "access", s.access); + setintfield(L, "size", s.size); + } else { + lua_pushnil(L); + } + return 1; +} + + +static int sys_open(lua_State *L) { + const char* name = luaL_checkstring(L, 1); + int mode = luaL_checkinteger(L, 2); + fd_t fh = sc_open(name, mode); + if (fh) { + lua_pushinteger(L, fh); + } else { + lua_pushnil(L); + } + return 1; +} + +static int sys_close(lua_State *L) { + int fd = luaL_checkinteger(L, 1); + sc_close(fd); + return 0; +} + +static int sys_read(lua_State *L) { + int fd = luaL_checkinteger(L, 1); + int offset = luaL_checkinteger(L, 2); + int len = luaL_checkinteger(L, 3); + + luaL_Buffer b; + char* bufptr = luaL_buffinitsize(L, &b, len); + int ret = sc_read(fd, offset, len, bufptr); + + luaL_pushresultsize(&b, ret); + lua_pushinteger(L, ret); + return 2; +} + +static int sys_write(lua_State *L) { + int fd = luaL_checkinteger(L, 1); + int offset = luaL_checkinteger(L, 2); + const char* str = luaL_checkstring(L, 3); + int len = luaL_len(L, 3); + + int ret = sc_write(fd, offset, len, str); + lua_pushinteger(L, ret); + return 1; +} + +static int sys_readdir(lua_State *L) { + int fd = luaL_checkinteger(L, 1); + int ent_no = luaL_checkinteger(L, 2); + + dirent_t d; + bool ret = sc_readdir(fd, ent_no, &d); + if (ret) { + lua_createtable(L, 0, 4); + setstrfield(L, "name", d.name); + setintfield(L, "type", d.st.type); + setintfield(L, "access", d.st.access); + setintfield(L, "size", d.st.size); + } else { + lua_pushnil(L); + } + return 1; +} + +static int sys_stat_open(lua_State *L) { + int fd = luaL_checkinteger(L, 1); + stat_t s; + if (sc_stat_open(fd, &s)) { + lua_createtable(L, 0, 3); + setintfield(L, "type", s.type); + setintfield(L, "access", s.access); + setintfield(L, "size", s.size); + } else { + lua_pushnil(L); + } + return 1; +} + +static int sys_ioctl(lua_State *L) { + // TODO + return 0; +} + +static int sys_fctl(lua_State *L) { + // TODO + return 0; +} + +static int sys_select(lua_State *L) { + // TODO + return 0; +} + + +static const luaL_Reg syslib[] = { + {"dbg_print", sys_dbg_print}, + {"yield", sys_yield}, + {"exit", sys_exit}, + {"usleep", sys_usleep}, + {"create", sys_create}, + {"delete", sys_delete}, + {"move", sys_move}, + {"stat", sys_stat}, + {"open", sys_open}, + {"close", sys_close}, + {"read", sys_read}, + {"write", sys_write}, + {"readdir", sys_readdir}, + {"stat_open", sys_stat_open}, + + {NULL, NULL} +}; + +/* }====================================================== */ + + + +LUAMOD_API int lx_open_sys (lua_State *L) { + luaL_newlib(L, syslib); + return 1; +} + + +/* vim: set sts=2 ts=2 sw=2 tw=0 et :*/ diff --git a/src/sysbin/lx/main.c b/src/sysbin/lx/main.c index ecfecd8..b7e08d9 100644 --- a/src/sysbin/lx/main.c +++ b/src/sysbin/lx/main.c @@ -16,6 +16,8 @@ #include #include +#include "lxlib.h" + #define LUA_PROMPT "> " #define LUA_PROMPT2 ">> " @@ -183,20 +185,6 @@ int dostring (lua_State *L, const char *s, const char *name) { } -/* -** Calls 'require(name)' and stores the result in a global variable -** with the given name. -*/ -int dolibrary (lua_State *L, const char *name) { - int status; - lua_getglobal(L, "require"); - lua_pushstring(L, name); - status = docall(L, 1, 1); /* call 'require(name)' */ - if (status == LUA_OK) - lua_setglobal(L, name); /* global[name] = require return */ - return report(L, status); -} - /* ** Returns the string to be used as a prompt by the interpreter. @@ -349,36 +337,6 @@ void doREPL (lua_State *L) { } -/* -** Push on the stack the contents of table 'arg' from 1 to #arg -*/ -int pushargs (lua_State *L) { - int i, n; - if (lua_getglobal(L, "arg") != LUA_TTABLE) - luaL_error(L, "'arg' is not a table"); - n = (int)luaL_len(L, -1); - luaL_checkstack(L, n + 3, "too many arguments to script"); - for (i = 1; i <= n; i++) - lua_rawgeti(L, -i, i); - lua_remove(L, -i); /* remove table from the stack */ - return n; -} - - -int handle_script (lua_State *L, char **argv) { - int status; - const char *fname = argv[0]; - if (strcmp(fname, "-") == 0 && strcmp(argv[-1], "--") != 0) - fname = NULL; /* stdin */ - status = luaL_loadfile(L, fname); - if (status == LUA_OK) { - int n = pushargs(L); /* push arguments to script */ - status = docall(L, n, LUA_MULTRET); - } - return report(L, status); -} - - int handle_luainit (lua_State *L) { const char *name = "=" LUA_INITVARVERSION; @@ -395,10 +353,11 @@ int handle_luainit (lua_State *L) { int pmain (lua_State *L) { int status; luaL_checkversion(L); /* check that interpreter has correct version */ - + print_version(); luaL_openlibs(L); /* open standard libraries */ + lx_openlibs(L); if (handle_luainit(L) != LUA_OK) /* run LUA_INIT */ return 0; /* error running LUA_INIT */ @@ -409,7 +368,6 @@ int pmain (lua_State *L) { if (status != LUA_OK) { // no main, launch a REPL if (lua_stdin_is_tty()) { /* running in interactive mode? */ - print_version(); doREPL(L); /* do read-eval-print loop */ } else { dofile(L, NULL); /* executes stdin as a file */ -- cgit v1.2.3