2 * Copyright (C) 2012 Jo-Philipp Wich <jow@openwrt.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU Lesser General Public License version 2.1
6 * as published by the Free Software Foundation
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
16 #include <libubox/blobmsg.h>
20 #define MODNAME "ubus"
21 #define METANAME MODNAME ".meta"
24 struct ubus_lua_connection {
27 struct ubus_context *ctx;
32 ubus_lua_parse_blob(lua_State *L, struct blob_attr *attr, bool table);
35 ubus_lua_parse_blob_array(lua_State *L, struct blob_attr *attr, int len, bool table)
40 struct blob_attr *pos;
44 __blob_for_each_attr(pos, attr, rem)
46 rv = ubus_lua_parse_blob(L, pos, table);
51 lua_rawseti(L, -2, idx++);
58 ubus_lua_parse_blob(lua_State *L, struct blob_attr *attr, bool table)
65 if (!blobmsg_check_attr(attr, false))
68 if (table && blobmsg_name(attr)[0])
70 lua_pushstring(L, blobmsg_name(attr));
74 data = blobmsg_data(attr);
75 len = blobmsg_data_len(attr);
77 switch (blob_id(attr))
79 case BLOBMSG_TYPE_BOOL:
80 lua_pushboolean(L, *(uint8_t *)data);
83 case BLOBMSG_TYPE_INT16:
84 lua_pushinteger(L, be16_to_cpu(*(uint16_t *)data));
87 case BLOBMSG_TYPE_INT32:
88 lua_pushinteger(L, be32_to_cpu(*(uint32_t *)data));
91 case BLOBMSG_TYPE_INT64:
92 /* NB: Lua cannot handle 64bit, format value as string and push that */
93 sprintf(buf, "%lld", (long long int) be64_to_cpu(*(uint64_t *)data));
94 lua_pushstring(L, buf);
97 case BLOBMSG_TYPE_STRING:
98 lua_pushstring(L, data);
101 case BLOBMSG_TYPE_ARRAY:
102 ubus_lua_parse_blob_array(L, data, len, false);
105 case BLOBMSG_TYPE_TABLE:
106 ubus_lua_parse_blob_array(L, data, len, true);
119 ubus_lua_format_blob_is_array(lua_State *L)
124 /* Find out whether table is array-like */
125 for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1))
128 if (lua_type(L, -2) != LUA_TNUMBER && lua_type(L, -2) != LUA_TINT)
130 if (lua_type(L, -2) != LUA_TNUMBER)
137 cur = lua_tointeger(L, -2);
139 if ((cur - 1) != prv)
152 ubus_lua_format_blob_array(lua_State *L, struct blob_buf *b, bool table);
155 ubus_lua_format_blob(lua_State *L, struct blob_buf *b, bool table)
159 const char *key = table ? lua_tostring(L, -2) : NULL;
161 switch (lua_type(L, -1))
164 blobmsg_add_u8(b, key, (uint8_t)lua_toboolean(L, -1));
171 blobmsg_add_u32(b, key, (uint32_t)lua_tointeger(L, -1));
176 case LUA_TLIGHTUSERDATA:
177 blobmsg_add_string(b, key, lua_tostring(L, -1));
181 if (ubus_lua_format_blob_is_array(L))
183 c = blobmsg_open_array(b, key);
184 rv = ubus_lua_format_blob_array(L, b, false);
185 blobmsg_close_array(b, c);
189 c = blobmsg_open_table(b, key);
190 rv = ubus_lua_format_blob_array(L, b, true);
191 blobmsg_close_table(b, c);
204 ubus_lua_format_blob_array(lua_State *L, struct blob_buf *b, bool table)
206 for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1))
208 if (!ubus_lua_format_blob(L, b, table))
220 ubus_lua_connect(lua_State *L)
222 struct ubus_lua_connection *c;
223 const char *sockpath = luaL_optstring(L, 1, NULL);
224 int timeout = luaL_optint(L, 2, 30);
226 if ((c = lua_newuserdata(L, sizeof(*c))) != NULL &&
227 (c->ctx = ubus_connect(sockpath)) != NULL)
229 ubus_add_uloop(c->ctx);
230 c->timeout = timeout;
231 memset(&c->buf, 0, sizeof(c->buf));
232 luaL_getmetatable(L, METANAME);
233 lua_setmetatable(L, -2);
237 /* NB: no errors from ubus_connect() yet */
239 lua_pushinteger(L, UBUS_STATUS_UNKNOWN_ERROR);
245 ubus_lua_objects_cb(struct ubus_context *c, struct ubus_object_data *o, void *p)
247 lua_State *L = (lua_State *)p;
249 lua_pushstring(L, o->path);
250 lua_rawseti(L, -2, lua_objlen(L, -2) + 1);
254 ubus_lua_objects(lua_State *L)
257 struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
260 rv = ubus_lookup(c->ctx, NULL, ubus_lua_objects_cb, L);
262 if (rv != UBUS_STATUS_OK)
266 lua_pushinteger(L, rv);
275 ubus_lua_signatures_cb(struct ubus_context *c, struct ubus_object_data *o, void *p)
277 lua_State *L = (lua_State *)p;
282 ubus_lua_parse_blob_array(L, blob_data(o->signature), blob_len(o->signature), true);
286 ubus_lua_signatures(lua_State *L)
289 struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
290 const char *path = luaL_checkstring(L, 2);
292 rv = ubus_lookup(c->ctx, path, ubus_lua_signatures_cb, L);
294 if (rv != UBUS_STATUS_OK)
298 lua_pushinteger(L, rv);
307 ubus_lua_call_cb(struct ubus_request *req, int type, struct blob_attr *msg)
309 lua_State *L = (lua_State *)req->priv;
314 ubus_lua_parse_blob_array(L, blob_data(msg), blob_len(msg), true);
318 ubus_lua_call(lua_State *L)
322 struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
323 const char *path = luaL_checkstring(L, 2);
324 const char *func = luaL_checkstring(L, 3);
326 luaL_checktype(L, 4, LUA_TTABLE);
327 blob_buf_init(&c->buf, 0);
329 if (!ubus_lua_format_blob_array(L, &c->buf, true))
332 lua_pushinteger(L, UBUS_STATUS_INVALID_ARGUMENT);
336 rv = ubus_lookup_id(c->ctx, path, &id);
341 lua_pushinteger(L, rv);
345 rv = ubus_invoke(c->ctx, id, func, c->buf.head, ubus_lua_call_cb, L, c->timeout * 1000);
347 if (rv != UBUS_STATUS_OK)
351 lua_pushinteger(L, rv);
360 ubus_lua__gc(lua_State *L)
362 struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
367 memset(c, 0, sizeof(*c));
373 static const luaL_Reg ubus[] = {
374 { "connect", ubus_lua_connect },
375 { "objects", ubus_lua_objects },
376 { "signatures", ubus_lua_signatures },
377 { "call", ubus_lua_call },
378 { "close", ubus_lua__gc },
379 { "__gc", ubus_lua__gc },
383 /* avoid missing prototype warning */
384 int luaopen_ubus(lua_State *L);
387 luaopen_ubus(lua_State *L)
389 /* create metatable */
390 luaL_newmetatable(L, METANAME);
392 /* metatable.__index = metatable */
393 lua_pushvalue(L, -1);
394 lua_setfield(L, -2, "__index");
397 luaL_register(L, NULL, ubus);
401 luaL_register(L, MODNAME, ubus);