+static int
+ubus_method_handler(struct ubus_context *ctx, struct ubus_object *obj,
+ struct ubus_request_data *req, const char *method,
+ 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);
+ if (!msg)
+ lua_pushnil(state);
+ else
+ ubus_lua_parse_blob_array(state, blob_data(msg), blob_len(msg), true);
+ 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)
+{
+ int cnt = 0;
+
+ lua_pushnil(L);
+ index -= 1;
+ while (lua_next(L, index) != 0) {
+ cnt++;
+ lua_pop(L, 1);
+ }
+
+ return cnt;
+}
+
+static int ubus_lua_reply(lua_State *L)
+{
+ struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
+ struct ubus_request_data *req;
+
+ 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;
+ }
+
+ req = lua_touserdata(L, 2);
+ ubus_send_reply(c->ctx, req, c->buf.head);
+
+ return 0;
+}
+
+static int ubus_lua_load_methods(lua_State *L, struct ubus_method *m)
+{
+ struct blobmsg_policy *p;
+ int plen;
+ int pidx = 0;
+
+ /* get the function pointer */
+ lua_pushinteger(L, 1);
+ lua_gettable(L, -2);
+
+ /* get the policy table */
+ lua_pushinteger(L, 2);
+ lua_gettable(L, -3);
+
+ /* check if the method table is valid */
+ if ((lua_type(L, -2) != LUA_TFUNCTION) ||
+ (lua_type(L, -1) != LUA_TTABLE) ||
+ lua_objlen(L, -1)) {
+ lua_pop(L, 2);
+ return 1;
+ }
+
+ /* store function pointer */
+ 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);
+ if (!p)
+ return 1;
+
+ memset(p, 0, sizeof(struct blobmsg_policy) * plen);
+ m->policy = p;
+ lua_pushnil(L);
+ while (lua_next(L, -2) != 0) {
+ int val = lua_tointeger(L, -1);
+
+ /* check if the policy is valid */
+ if ((lua_type(L, -2) != LUA_TSTRING) ||
+ (lua_type(L, -1) != LUA_TNUMBER) ||
+ (val < 0) ||
+ (val > BLOBMSG_TYPE_LAST)) {
+ lua_pop(L, 1);
+ continue;
+ }
+ p[pidx].name = lua_tostring(L, -2);
+ p[pidx].type = val;
+ lua_pop(L, 1);
+ pidx++;
+ }
+
+ m->n_policy = pidx;
+ lua_pop(L, 2);
+
+ return 0;
+}
+
+static struct ubus_object* ubus_lua_load_object(lua_State *L)
+{
+ struct ubus_lua_object *obj = NULL;
+ int mlen = lua_gettablelen(L, -1);
+ struct ubus_method *m;
+ int midx = 0;
+
+ /* setup object pointers */
+ obj = malloc(sizeof(struct ubus_lua_object));
+ if (!obj)
+ return NULL;
+
+ memset(obj, 0, sizeof(struct ubus_lua_object));
+ obj->o.name = lua_tostring(L, -2);
+
+ /* setup method pointers */
+ m = malloc(sizeof(struct ubus_method) * mlen);
+ memset(m, 0, sizeof(struct ubus_method) * mlen);
+ obj->o.methods = m;
+
+ /* setup type pointers */
+ obj->o.type = malloc(sizeof(struct ubus_object_type));
+ if (!obj->o.type) {
+ free(obj);
+ return NULL;
+ }
+
+ memset(obj->o.type, 0, sizeof(struct ubus_object_type));
+ obj->o.type->name = lua_tostring(L, -2);
+ obj->o.type->id = 0;
+ obj->o.type->methods = obj->o.methods;
+
+ /* create the callback lookup table */
+ lua_createtable(L, 1, 0);
+ lua_getglobal(L, "__ubus_cb");
+ lua_pushvalue(L, -2);
+ obj->r = luaL_ref(L, -2);
+ lua_pop(L, 1);
+
+ /* scan each method */
+ lua_pushnil(L);
+ while (lua_next(L, -3) != 0) {
+ /* check if it looks like a method */
+ if ((lua_type(L, -2) != LUA_TSTRING) ||
+ (lua_type(L, -1) != LUA_TTABLE) ||
+ !lua_objlen(L, -1)) {
+ lua_pop(L, 1);
+ continue;
+ }
+
+ if (!ubus_lua_load_methods(L, &m[midx]))
+ midx++;
+ lua_pop(L, 1);
+ }
+
+ obj->o.type->n_methods = obj->o.n_methods = midx;
+
+ /* pop the callback table */
+ lua_pop(L, 1);
+
+ return &obj->o;
+}
+
+static int ubus_lua_add(lua_State *L)
+{
+ struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
+
+ /* verify top level object */
+ if (lua_istable(L, 1)) {
+ lua_pushstring(L, "you need to pass a table");
+ lua_error(L);
+ return 0;
+ }
+
+ /* scan each object */
+ lua_pushnil(L);
+ while (lua_next(L, -2) != 0) {
+ struct ubus_object *obj = NULL;
+
+ /* check if the object has a table of methods */
+ if ((lua_type(L, -2) == LUA_TSTRING) && (lua_type(L, -1) == LUA_TTABLE)) {
+ obj = ubus_lua_load_object(L);
+
+ if (obj)
+ ubus_add_object(c->ctx, obj);
+ }
+ lua_pop(L, 1);
+ }
+
+ return 0;
+}