#include <libubox/blobmsg.h>
#include <libubox/blobmsg_json.h>
#include <lauxlib.h>
+#include <lua.h>
-
-#define MODNAME "ubus"
-#define METANAME MODNAME ".meta"
+#define MODNAME "ubus"
+#define METANAME MODNAME ".meta"
static lua_State *state;
int r;
};
+struct ubus_lua_event {
+ struct ubus_event_handler e;
+ int r;
+};
+
static int
ubus_lua_parse_blob(lua_State *L, struct blob_attr *attr, bool table);
int len;
int off = 0;
void *data;
- char buf[32];
if (!blobmsg_check_attr(attr, false))
return 0;
break;
case BLOBMSG_TYPE_INT64:
- /* NB: Lua cannot handle 64bit, format value as string and push that */
- sprintf(buf, "%lld", (long long int) be64_to_cpu(*(uint64_t *)data));
- lua_pushstring(L, buf);
+ lua_pushnumber(L, (double) be64_to_cpu(*(uint64_t *)data));
break;
case BLOBMSG_TYPE_STRING:
prv = cur;
}
- lua_pop(L, 1);
return true;
}
struct blob_attr *msg)
{
struct ubus_lua_object *o = container_of(obj, struct ubus_lua_object, o);
+ int rv = 0;
lua_getglobal(state, "__ubus_cb");
lua_rawgeti(state, -1, o->r);
lua_getfield(state, -1, method);
+ lua_remove(state, -2);
+ lua_remove(state, -2);
if (lua_isfunction(state, -1)) {
lua_pushlightuserdata(state, req);
lua_pushnil(state);
else
ubus_lua_parse_blob_array(state, blob_data(msg), blob_len(msg), true);
- lua_call(state, 2, 0);
- }
- return 0;
+ lua_call(state, 2, 1);
+ if (lua_isnumber(state, -1))
+ rv = lua_tonumber(state, -1);
+ } else
+ lua_pop(state, 1);
+
+ return rv;
}
static int lua_gettablelen(lua_State *L, int index)
/* get the policy table */
lua_pushinteger(L, 2);
lua_gettable(L, -3);
- plen = lua_gettablelen(L, -1);
/* check if the method table is valid */
if ((lua_type(L, -2) != LUA_TFUNCTION) ||
(lua_type(L, -1) != LUA_TTABLE) ||
- lua_objlen(L, -1) || !plen) {
+ lua_objlen(L, -1)) {
lua_pop(L, 2);
return 1;
}
lua_pushvalue(L, -2);
lua_setfield(L, -6, lua_tostring(L, -5));
+ m->name = lua_tostring(L, -4);
+ m->handler = ubus_method_handler;
+
+ plen = lua_gettablelen(L, -1);
+
+ /* exit if policy table is empty */
+ if (!plen) {
+ lua_pop(L, 2);
+ return 0;
+ }
+
/* setup the policy pointers */
p = malloc(sizeof(struct blobmsg_policy) * plen);
memset(p, 0, sizeof(struct blobmsg_policy) * plen);
}
m->n_policy = pidx;
- m->name = lua_tostring(L, -4);
- m->handler = ubus_method_handler;
lua_pop(L, 2);
return 0;
obj->o.type->id = 0;
obj->o.type->methods = obj->o.methods;
- /* create the he callback lookup table */
+ /* create the callback lookup table */
lua_createtable(L, 1, 0);
lua_getglobal(L, "__ubus_cb");
lua_pushvalue(L, -2);
return lua_gettop(L) - top;
}
+static void
+ubus_event_handler(struct ubus_context *ctx, struct ubus_event_handler *ev,
+ const char *type, struct blob_attr *msg)
+{
+ struct ubus_lua_event *listener = container_of(ev, struct ubus_lua_event, e);
+
+ lua_getglobal(state, "__ubus_cb_event");
+ lua_rawgeti(state, -1, listener->r);
+
+ if (lua_isfunction(state, -1)) {
+ ubus_lua_parse_blob_array(state, blob_data(msg), blob_len(msg), true);
+ lua_call(state, 1, 0);
+ }
+}
+
+static struct ubus_event_handler*
+ubus_lua_load_event(lua_State *L)
+{
+ struct ubus_lua_event* event = NULL;
+
+ event = malloc(sizeof(struct ubus_lua_event));
+ memset(event, 0, sizeof(struct ubus_lua_event));
+ event->e.cb = ubus_event_handler;
+
+ /* update the he callback lookup table */
+ lua_getglobal(L, "__ubus_cb_event");
+ lua_pushvalue(L, -2);
+ event->r = luaL_ref(L, -2);
+ lua_setfield(L, -1, lua_tostring(L, -3));
+
+ return &event->e;
+}
+
+static int
+ubus_lua_listen(lua_State *L) {
+ struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
+
+ /* verify top level object */
+ luaL_checktype(L, 2, LUA_TTABLE);
+
+ /* scan each object */
+ lua_pushnil(L);
+ while (lua_next(L, -2) != 0) {
+ struct ubus_event_handler *listener;
+
+ /* check if the key is a string and the value is a method */
+ if ((lua_type(L, -2) == LUA_TSTRING) && (lua_type(L, -1) == LUA_TFUNCTION)) {
+ listener = ubus_lua_load_event(L);
+ if(listener != NULL) {
+ ubus_register_event_handler(c->ctx, listener, lua_tostring(L, -2));
+ }
+ }
+ lua_pop(L, 1);
+ }
+ return 0;
+}
+
+static int
+ubus_lua_send(lua_State *L)
+{
+ struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
+ const char *event = luaL_checkstring(L, 2);
+
+ if (*event == 0)
+ return luaL_argerror(L, 2, "no event name");
+
+ // Event content convert to ubus form
+ luaL_checktype(L, 3, LUA_TTABLE);
+ blob_buf_init(&c->buf, 0);
+
+ if (!ubus_lua_format_blob_array(L, &c->buf, true)) {
+ lua_pushnil(L);
+ lua_pushinteger(L, UBUS_STATUS_INVALID_ARGUMENT);
+ return 2;
+ }
+
+ // Send the event
+ ubus_send_event(c->ctx, event, c->buf.head);
+
+ return 0;
+}
+
+
static int
ubus_lua__gc(lua_State *L)
{ "signatures", ubus_lua_signatures },
{ "call", ubus_lua_call },
{ "close", ubus_lua__gc },
+ { "listen", ubus_lua_listen },
+ { "send", ubus_lua_send },
{ "__gc", ubus_lua__gc },
{ NULL, NULL },
};
lua_createtable(L, 1, 0);
lua_setglobal(L, "__ubus_cb");
+ /* create the event table */
+ lua_createtable(L, 1, 0);
+ lua_setglobal(L, "__ubus_cb_event");
+
return 0;
}