Added fd_add method for uloop lua binding.

[project/libubox.git] / lua / uloop.c

diff --git a/lua/uloop.c b/lua/uloop.c
index 0f0c878..c71d537 100644 (file)
--- a/lua/uloop.c
+++ b/lua/uloop.c
@@ -22,8 +22,14 @@
 #include <lualib.h>
 #include <lauxlib.h>
 
-#include <libubox/uloop.h>
-#include <libubox/list.h>
+#include "../uloop.h"
+#include "../list.h"
+
+struct lua_uloop_fd {
+       struct uloop_fd fd;
+       int r;
+       int fd_r;
+};
 
 struct lua_uloop_timeout {
        struct uloop_timeout t;
@@ -43,7 +49,10 @@ static void ul_timer_cb(struct uloop_timeout *t)
 
        lua_getglobal(state, "__uloop_cb");
        lua_rawgeti(state, -1, tout->r);
+       lua_remove(state, -2);
+
        lua_call(state, 0, 0);
+
 }
 
 static int ul_timer_set(lua_State *L)
@@ -126,13 +135,97 @@ static int ul_timer(lua_State *L)
        return 1;
 }
 
-static void proc_cb(struct uloop_process *p, int ret)
+static void ul_ufd_cb(struct uloop_fd *fd, unsigned int events)
+{
+       struct lua_uloop_fd *ufd = container_of(fd, struct lua_uloop_fd, fd);
+
+       lua_getglobal(state, "__uloop_cb");
+       lua_rawgeti(state, -1, ufd->r);
+       lua_remove(state, -2);
+
+       /* push fd object */
+       lua_getglobal(state, "__uloop_fds");
+       lua_rawgeti(state, -1, ufd->fd_r);
+       lua_remove(state, -2);
+
+       /* push events */
+       lua_pushinteger(state, events);
+
+       lua_call(state, 2, 0);
+}
+
+
+static int get_sock_fd(lua_State* L, int idx) {
+       int fd;
+       if(lua_isnumber(L, idx)) {
+               fd = lua_tonumber(L, idx);
+       } else {
+               luaL_checktype(L, idx, LUA_TUSERDATA);
+               lua_getfield(L, idx, "getfd");
+               if(lua_isnil(L, -1))
+                       return luaL_error(L, "socket type missing 'getfd' method");
+               lua_pushvalue(L, idx - 1);
+               lua_call(L, 1, 1);
+               fd = lua_tointeger(L, -1);
+               lua_pop(L, 1);
+       }
+       return fd;
+}
+
+static int ul_ufd_add(lua_State *L)
+{
+       struct lua_uloop_fd *ufd;
+       int fd = 0;
+       unsigned int flags = 0;
+       int ref;
+       int fd_ref;
+
+       if (lua_isnumber(L, -1)) {
+               flags = lua_tointeger(L, -1);
+               lua_pop(L, 1);
+       }
+
+       if (!lua_isfunction(L, -1)) {
+               lua_pushstring(L, "invalid arg list");
+               lua_error(L);
+
+               return 0;
+       }
+
+       fd = get_sock_fd(L, -2);
+
+       lua_getglobal(L, "__uloop_cb");
+       lua_pushvalue(L, -2);
+       ref = luaL_ref(L, -2);
+       lua_pop(L, 1);
+
+       lua_getglobal(L, "__uloop_fds");
+       lua_pushvalue(L, -3);
+       fd_ref = luaL_ref(L, -2);
+       lua_pop(L, 1);
+
+       ufd = lua_newuserdata(L, sizeof(*ufd));
+       memset(ufd, 0, sizeof(*ufd));
+
+       ufd->r = ref;
+       ufd->fd.fd = fd;
+       ufd->fd_r = fd_ref;
+       ufd->fd.cb = ul_ufd_cb;
+       if (flags)
+               uloop_fd_add(&ufd->fd, flags);
+
+       return 1;
+}
+
+static void ul_process_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_remove(state, -2);
        lua_pushinteger(state, ret >> 8);
        lua_call(state, 1, 0);
 }
@@ -196,7 +289,7 @@ static int ul_process(lua_State *L)
 
        proc->r = ref;
        proc->p.pid = pid;
-       proc->p.cb = proc_cb;
+       proc->p.cb = ul_process_cb;
        uloop_process_add(&proc->p);
 
        return 1;
@@ -223,9 +316,14 @@ static luaL_reg uloop_func[] = {
        {"run", ul_run},
        {"timer", ul_timer},
        {"process", ul_process},
+       {"fd_add", ul_ufd_add},
        {NULL, NULL},
 };
 
+/* avoid warnings about missing declarations */
+int luaopen_uloop(lua_State *L);
+int luaclose_uloop(lua_State *L);
+
 int luaopen_uloop(lua_State *L)
 {
        state = L;
@@ -233,11 +331,30 @@ int luaopen_uloop(lua_State *L)
        lua_createtable(L, 1, 0);
        lua_setglobal(L, "__uloop_cb");
 
+       lua_createtable(L, 1, 0);
+       lua_setglobal(L, "__uloop_fds");
+
        luaL_openlib(L, "uloop", uloop_func, 0);
        lua_pushstring(L, "_VERSION");
        lua_pushstring(L, "1.0");
        lua_rawset(L, -3);
 
+       lua_pushstring(L, "ULOOP_READ");
+       lua_pushinteger(L, ULOOP_READ);
+       lua_rawset(L, -3);
+
+       lua_pushstring(L, "ULOOP_WRITE");
+       lua_pushinteger(L, ULOOP_WRITE);
+       lua_rawset(L, -3);
+
+       lua_pushstring(L, "ULOOP_EDGE_TRIGGER");
+       lua_pushinteger(L, ULOOP_EDGE_TRIGGER);
+       lua_rawset(L, -3);
+
+       lua_pushstring(L, "ULOOP_BLOCKING");
+       lua_pushinteger(L, ULOOP_BLOCKING);
+       lua_rawset(L, -3);
+
        return 1;
 }