add lua binding for uloop (written by John Crispin)

This commit is contained in:
Felix Fietkau 2012-09-27 13:30:33 +02:00
parent cca2ed6c8c
commit a81cb397ef
3 changed files with 305 additions and 0 deletions

View file

@ -3,6 +3,8 @@ cmake_minimum_required(VERSION 2.6)
PROJECT(ubox C) PROJECT(ubox C)
ADD_DEFINITIONS(-Os -Wall -Werror --std=gnu99 -g3) ADD_DEFINITIONS(-Os -Wall -Werror --std=gnu99 -g3)
OPTION(BUILD_LUA "build Lua plugin" ON)
IF(APPLE) IF(APPLE)
INCLUDE_DIRECTORIES(/opt/local/include) INCLUDE_DIRECTORIES(/opt/local/include)
LINK_DIRECTORIES(/opt/local/lib) LINK_DIRECTORIES(/opt/local/lib)
@ -22,6 +24,8 @@ INSTALL(TARGETS ubox
LIBRARY DESTINATION lib LIBRARY DESTINATION lib
) )
ADD_SUBDIRECTORY(lua)
find_library(json json) find_library(json json)
IF(EXISTS ${json}) IF(EXISTS ${json})
ADD_LIBRARY(blobmsg_json SHARED blobmsg_json.c) ADD_LIBRARY(blobmsg_json SHARED blobmsg_json.c)

52
lua/CMakeLists.txt Normal file
View file

@ -0,0 +1,52 @@
cmake_minimum_required(VERSION 2.6)
PROJECT(uloop C)
SET(CMAKE_INSTALL_PREFIX /)
IF(NOT LUA_CFLAGS)
FIND_PROGRAM(PKG_CONFIG pkg-config)
IF(PKG_CONFIG)
EXECUTE_PROCESS(
COMMAND pkg-config --silence-errors --cflags lua5.1
OUTPUT_VARIABLE LUA_CFLAGS
OUTPUT_STRIP_TRAILING_WHITESPACE
)
ENDIF()
ENDIF()
ADD_DEFINITIONS(-Os -Wall -Werror --std=gnu99 -g3 -I.. ${LUA_CFLAGS})
LINK_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR}/..)
IF(APPLE)
ADD_DEFINITIONS(-I/opt/local/include)
SET(CMAKE_SHARED_MODULE_CREATE_C_FLAGS "${CMAKE_SHARED_MODULE_CREATE_C_FLAGS} -undefined dynamic_lookup")
ENDIF(APPLE)
IF(NOT LUAPATH)
EXECUTE_PROCESS(
COMMAND lua -e "for k in string.gmatch(package.cpath .. \";\", \"([^;]+)/..so;\") do if k:sub(1,1) == \"/\" then print(k) break end end"
OUTPUT_VARIABLE LUAPATH
RESULT_VARIABLE LUA_CHECK_RES
OUTPUT_STRIP_TRAILING_WHITESPACE
)
IF(BUILD_LUA)
IF(NOT ${LUA_CHECK_RES} EQUAL 0 OR "${LUAPATH}" EQUAL "")
MESSAGE(SEND_ERROR "Lua was not found on your system")
ENDIF()
ENDIF()
ENDIF()
IF(BUILD_LUA)
ADD_LIBRARY(uloop_lua MODULE uloop.c)
SET_TARGET_PROPERTIES(uloop_lua PROPERTIES
OUTPUT_NAME uloop
PREFIX ""
)
TARGET_LINK_LIBRARIES(uloop_lua ubox)
INSTALL(TARGETS uloop_lua
LIBRARY DESTINATION ${LUAPATH}
)
ENDIF()

249
lua/uloop.c Normal file
View file

@ -0,0 +1,249 @@
/*
* Copyright (C) 2012 John Crispin <blogic@openwrt.org>
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <lua.h>
#include <lualib.h>
#include <lauxlib.h>
#include <libubox/uloop.h>
#include <libubox/list.h>
struct lua_uloop_timeout {
struct uloop_timeout t;
int r;
};
struct lua_uloop_process {
struct uloop_process p;
int r;
};
static lua_State *state;
static void ul_timer_cb(struct uloop_timeout *t)
{
struct lua_uloop_timeout *tout = container_of(t, struct lua_uloop_timeout, t);
lua_getglobal(state, "__uloop_cb");
lua_rawgeti(state, -1, tout->r);
lua_call(state, 0, 0);
}
static int ul_timer_set(lua_State *L)
{
struct lua_uloop_timeout *tout;
double set;
if (!lua_isnumber(L, -1)) {
lua_pushstring(L, "invalid arg list");
lua_error(L);
return 0;
}
set = lua_tointeger(L, -1);
tout = lua_touserdata(L, 1);
uloop_timeout_set(&tout->t, set);
return 1;
}
static int ul_timer_free(lua_State *L)
{
struct lua_uloop_timeout *tout = lua_touserdata(L, 1);
uloop_timeout_cancel(&tout->t);
lua_getglobal(state, "__uloop_cb");
luaL_unref(L, -1, tout->r);
return 1;
}
static const luaL_Reg timer_m[] = {
{ "set", ul_timer_set },
{ "cancel", ul_timer_free },
{ NULL, NULL }
};
static int ul_timer(lua_State *L)
{
struct lua_uloop_timeout *tout;
int set = 0;
int ref;
if (lua_isnumber(L, -1)) {
set = lua_tointeger(L, -1);
lua_pop(L, 1);
}
if (!lua_isfunction(L, -1)) {
lua_pushstring(L, "invalid arg list");
lua_error(L);
return 0;
}
lua_getglobal(L, "__uloop_cb");
lua_pushvalue(L, -2);
ref = luaL_ref(L, -2);
tout = lua_newuserdata(L, sizeof(*tout));
lua_createtable(L, 0, 2);
lua_pushvalue(L, -1);
lua_setfield(L, -2, "__index");
lua_pushcfunction(L, ul_timer_free);
lua_setfield(L, -2, "__gc");
lua_pushvalue(L, -1);
lua_setmetatable(L, -3);
lua_pushvalue(L, -2);
luaI_openlib(L, NULL, timer_m, 1);
lua_pushvalue(L, -2);
memset(tout, 0, sizeof(*tout));
tout->r = ref;
tout->t.cb = ul_timer_cb;
if (set)
uloop_timeout_set(&tout->t, set);
return 1;
}
static void proc_cb(struct uloop_process *p, int ret)
{
struct lua_uloop_process *proc = container_of(p, struct lua_uloop_process, p);
lua_getglobal(state, "__uloop_cb");
lua_rawgeti(state, -1, proc->r);
luaL_unref(state, -2, proc->r);
lua_pushinteger(state, ret >> 8);
lua_call(state, 1, 0);
}
static int ul_process(lua_State *L)
{
struct lua_uloop_process *proc;
pid_t pid;
int ref;
if (!lua_isfunction(L, -1) || !lua_istable(L, -2) ||
!lua_istable(L, -3) || !lua_isstring(L, -4)) {
lua_pushstring(L, "invalid arg list");
lua_error(L);
return 0;
}
pid = fork();
if (pid == -1) {
lua_pushstring(L, "failed to fork");
lua_error(L);
return 0;
}
if (pid == 0) {
/* child */
int argn = lua_objlen(L, -3);
int envn = lua_objlen(L, -2);
char** argp = malloc(sizeof(char*) * (argn + 2));
char** envp = malloc(sizeof(char*) * envn + 1);
int i = 1;
argp[0] = (char*) lua_tostring(L, -4);
for (i = 1; i <= argn; i++) {
lua_rawgeti(L, -3, i);
argp[i] = (char*) lua_tostring(L, -1);
lua_pop(L, 1);
}
argp[i] = NULL;
for (i = 1; i <= envn; i++) {
lua_rawgeti(L, -2, i);
envp[i - 1] = (char*) lua_tostring(L, -1);
lua_pop(L, 1);
}
envp[i - 1] = NULL;
execve(*argp, argp, envp);
exit(-1);
}
lua_getglobal(L, "__uloop_cb");
lua_pushvalue(L, -2);
ref = luaL_ref(L, -2);
proc = lua_newuserdata(L, sizeof(*proc));
memset(proc, 0, sizeof(*proc));
proc->r = ref;
proc->p.pid = pid;
proc->p.cb = proc_cb;
uloop_process_add(&proc->p);
return 1;
}
static int ul_init(lua_State *L)
{
uloop_init();
lua_pushboolean(L, 1);
return 1;
}
static int ul_run(lua_State *L)
{
uloop_run();
lua_pushboolean(L, 1);
return 1;
}
static luaL_reg uloop_func[] = {
{"init", ul_init},
{"run", ul_run},
{"timer", ul_timer},
{"process", ul_process},
{NULL, NULL},
};
int luaopen_uloop(lua_State *L)
{
state = L;
lua_createtable(L, 1, 0);
lua_setglobal(L, "__uloop_cb");
luaL_openlib(L, "uloop", uloop_func, 0);
lua_pushstring(L, "_VERSION");
lua_pushstring(L, "1.0");
lua_rawset(L, -3);
return 1;
}
int luaclose_uloop(lua_State *L)
{
lua_pushstring(L, "Called");
return 1;
}