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 c
->timeout
= timeout
;
230 memset(&c
->buf
, 0, sizeof(c
->buf
));
231 luaL_getmetatable(L
, METANAME
);
232 lua_setmetatable(L
, -2);
236 /* NB: no errors from ubus_connect() yet */
238 lua_pushinteger(L
, UBUS_STATUS_UNKNOWN_ERROR
);
244 ubus_lua_objects_cb(struct ubus_context
*c
, struct ubus_object_data
*o
, void *p
)
246 lua_State
*L
= (lua_State
*)p
;
248 lua_pushstring(L
, o
->path
);
249 lua_rawseti(L
, -2, lua_objlen(L
, -2) + 1);
253 ubus_lua_objects(lua_State
*L
)
256 struct ubus_lua_connection
*c
= luaL_checkudata(L
, 1, METANAME
);
259 rv
= ubus_lookup(c
->ctx
, NULL
, ubus_lua_objects_cb
, L
);
261 if (rv
!= UBUS_STATUS_OK
)
265 lua_pushinteger(L
, rv
);
274 ubus_lua_signatures_cb(struct ubus_context
*c
, struct ubus_object_data
*o
, void *p
)
276 lua_State
*L
= (lua_State
*)p
;
281 ubus_lua_parse_blob_array(L
, blob_data(o
->signature
), blob_len(o
->signature
), true);
285 ubus_lua_signatures(lua_State
*L
)
288 struct ubus_lua_connection
*c
= luaL_checkudata(L
, 1, METANAME
);
289 const char *path
= luaL_checkstring(L
, 2);
291 rv
= ubus_lookup(c
->ctx
, path
, ubus_lua_signatures_cb
, L
);
293 if (rv
!= UBUS_STATUS_OK
)
297 lua_pushinteger(L
, rv
);
306 ubus_lua_call_cb(struct ubus_request
*req
, int type
, struct blob_attr
*msg
)
308 lua_State
*L
= (lua_State
*)req
->priv
;
313 ubus_lua_parse_blob_array(L
, blob_data(msg
), blob_len(msg
), true);
317 ubus_lua_call(lua_State
*L
)
321 struct ubus_lua_connection
*c
= luaL_checkudata(L
, 1, METANAME
);
322 const char *path
= luaL_checkstring(L
, 2);
323 const char *func
= luaL_checkstring(L
, 3);
325 luaL_checktype(L
, 4, LUA_TTABLE
);
326 blob_buf_init(&c
->buf
, 0);
328 if (!ubus_lua_format_blob_array(L
, &c
->buf
, true))
331 lua_pushinteger(L
, UBUS_STATUS_INVALID_ARGUMENT
);
335 rv
= ubus_lookup_id(c
->ctx
, path
, &id
);
340 lua_pushinteger(L
, rv
);
344 rv
= ubus_invoke(c
->ctx
, id
, func
, c
->buf
.head
, ubus_lua_call_cb
, L
, c
->timeout
* 1000);
346 if (rv
!= UBUS_STATUS_OK
)
350 lua_pushinteger(L
, rv
);
359 ubus_lua__gc(lua_State
*L
)
361 struct ubus_lua_connection
*c
= luaL_checkudata(L
, 1, METANAME
);
366 memset(c
, 0, sizeof(*c
));
372 static const luaL_Reg ubus
[] = {
373 { "connect", ubus_lua_connect
},
374 { "objects", ubus_lua_objects
},
375 { "signatures", ubus_lua_signatures
},
376 { "call", ubus_lua_call
},
377 { "close", ubus_lua__gc
},
378 { "__gc", ubus_lua__gc
},
382 /* avoid missing prototype warning */
383 int luaopen_ubus(lua_State
*L
);
386 luaopen_ubus(lua_State
*L
)
388 /* create metatable */
389 luaL_newmetatable(L
, METANAME
);
391 /* metatable.__index = metatable */
392 lua_pushvalue(L
, -1);
393 lua_setfield(L
, -2, "__index");
396 luaL_register(L
, NULL
, ubus
);
400 luaL_register(L
, MODNAME
, ubus
);