fix a copy & paste bug
[project/ubus.git] / lua / ubus.c
1 /*
2  * Copyright (C) 2012 Jo-Philipp Wich <jow@openwrt.org>
3  * Copyright (C) 2012 John Crispin <blogic@openwrt.org>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU Lesser General Public License version 2.1
7  * as published by the Free Software Foundation
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  */
14
15 #include <unistd.h>
16 #include <libubus.h>
17 #include <libubox/blobmsg.h>
18 #include <libubox/blobmsg_json.h>
19 #include <lauxlib.h>
20 #include <lua.h>
21
22 #define MODNAME         "ubus"
23 #define METANAME        MODNAME ".meta"
24
25 static lua_State *state;
26
27 struct ubus_lua_connection {
28         int timeout;
29         struct blob_buf buf;
30         struct ubus_context *ctx;
31 };
32
33 struct ubus_lua_object {
34         struct ubus_object o;
35         int r;
36 };
37
38 struct ubus_lua_event {
39         struct ubus_event_handler e;
40         int r;
41 };
42
43 static int
44 ubus_lua_parse_blob(lua_State *L, struct blob_attr *attr, bool table);
45
46 static int
47 ubus_lua_parse_blob_array(lua_State *L, struct blob_attr *attr, int len, bool table)
48 {
49         int rv;
50         int idx = 1;
51         int rem = len;
52         struct blob_attr *pos;
53
54         lua_newtable(L);
55
56         __blob_for_each_attr(pos, attr, rem)
57         {
58                 rv = ubus_lua_parse_blob(L, pos, table);
59
60                 if (rv > 1)
61                         lua_rawset(L, -3);
62                 else if (rv > 0)
63                         lua_rawseti(L, -2, idx++);
64         }
65
66         return 1;
67 }
68
69 static int
70 ubus_lua_parse_blob(lua_State *L, struct blob_attr *attr, bool table)
71 {
72         int len;
73         int off = 0;
74         void *data;
75
76         if (!blobmsg_check_attr(attr, false))
77                 return 0;
78
79         if (table && blobmsg_name(attr)[0])
80         {
81                 lua_pushstring(L, blobmsg_name(attr));
82                 off++;
83         }
84
85         data = blobmsg_data(attr);
86         len = blobmsg_data_len(attr);
87
88         switch (blob_id(attr))
89         {
90         case BLOBMSG_TYPE_BOOL:
91                 lua_pushboolean(L, *(uint8_t *)data);
92                 break;
93
94         case BLOBMSG_TYPE_INT16:
95                 lua_pushinteger(L, be16_to_cpu(*(uint16_t *)data));
96                 break;
97
98         case BLOBMSG_TYPE_INT32:
99                 lua_pushinteger(L, be32_to_cpu(*(uint32_t *)data));
100                 break;
101
102         case BLOBMSG_TYPE_INT64:
103                 lua_pushnumber(L, (double) be64_to_cpu(*(uint64_t *)data));
104                 break;
105
106         case BLOBMSG_TYPE_STRING:
107                 lua_pushstring(L, data);
108                 break;
109
110         case BLOBMSG_TYPE_ARRAY:
111                 ubus_lua_parse_blob_array(L, data, len, false);
112                 break;
113
114         case BLOBMSG_TYPE_TABLE:
115                 ubus_lua_parse_blob_array(L, data, len, true);
116                 break;
117
118         default:
119                 lua_pushnil(L);
120                 break;
121         }
122
123         return off + 1;
124 }
125
126
127 static bool
128 ubus_lua_format_blob_is_array(lua_State *L)
129 {
130         lua_Integer prv = 0;
131         lua_Integer cur = 0;
132
133         /* Find out whether table is array-like */
134         for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1))
135         {
136 #ifdef LUA_TINT
137                 if (lua_type(L, -2) != LUA_TNUMBER && lua_type(L, -2) != LUA_TINT)
138 #else
139                 if (lua_type(L, -2) != LUA_TNUMBER)
140 #endif
141                 {
142                         lua_pop(L, 2);
143                         return false;
144                 }
145
146                 cur = lua_tointeger(L, -2);
147
148                 if ((cur - 1) != prv)
149                 {
150                         lua_pop(L, 2);
151                         return false;
152                 }
153
154                 prv = cur;
155         }
156
157         return true;
158 }
159
160 static int
161 ubus_lua_format_blob_array(lua_State *L, struct blob_buf *b, bool table);
162
163 static int
164 ubus_lua_format_blob(lua_State *L, struct blob_buf *b, bool table)
165 {
166         void *c;
167         bool rv = true;
168         const char *key = table ? lua_tostring(L, -2) : NULL;
169
170         switch (lua_type(L, -1))
171         {
172         case LUA_TBOOLEAN:
173                 blobmsg_add_u8(b, key, (uint8_t)lua_toboolean(L, -1));
174                 break;
175
176 #ifdef LUA_TINT
177         case LUA_TINT:
178 #endif
179         case LUA_TNUMBER:
180                 blobmsg_add_u32(b, key, (uint32_t)lua_tointeger(L, -1));
181                 break;
182
183         case LUA_TSTRING:
184         case LUA_TUSERDATA:
185         case LUA_TLIGHTUSERDATA:
186                 blobmsg_add_string(b, key, lua_tostring(L, -1));
187                 break;
188
189         case LUA_TTABLE:
190                 if (ubus_lua_format_blob_is_array(L))
191                 {
192                         c = blobmsg_open_array(b, key);
193                         rv = ubus_lua_format_blob_array(L, b, false);
194                         blobmsg_close_array(b, c);
195                 }
196                 else
197                 {
198                         c = blobmsg_open_table(b, key);
199                         rv = ubus_lua_format_blob_array(L, b, true);
200                         blobmsg_close_table(b, c);
201                 }
202                 break;
203
204         default:
205                 rv = false;
206                 break;
207         }
208
209         return rv;
210 }
211
212 static int
213 ubus_lua_format_blob_array(lua_State *L, struct blob_buf *b, bool table)
214 {
215         for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1))
216         {
217                 if (!ubus_lua_format_blob(L, b, table))
218                 {
219                         lua_pop(L, 1);
220                         return false;
221                 }
222         }
223
224         return true;
225 }
226
227
228 static int
229 ubus_lua_connect(lua_State *L)
230 {
231         struct ubus_lua_connection *c;
232         const char *sockpath = luaL_optstring(L, 1, NULL);
233         int timeout = luaL_optint(L, 2, 30);
234
235         if ((c = lua_newuserdata(L, sizeof(*c))) != NULL &&
236                 (c->ctx = ubus_connect(sockpath)) != NULL)
237         {
238                 ubus_add_uloop(c->ctx);
239                 c->timeout = timeout;
240                 memset(&c->buf, 0, sizeof(c->buf));
241                 luaL_getmetatable(L, METANAME);
242                 lua_setmetatable(L, -2);
243                 return 1;
244         }
245
246         /* NB: no errors from ubus_connect() yet */
247         lua_pushnil(L);
248         lua_pushinteger(L, UBUS_STATUS_UNKNOWN_ERROR);
249         return 2;
250 }
251
252
253 static void
254 ubus_lua_objects_cb(struct ubus_context *c, struct ubus_object_data *o, void *p)
255 {
256         lua_State *L = (lua_State *)p;
257
258         lua_pushstring(L, o->path);
259         lua_rawseti(L, -2, lua_objlen(L, -2) + 1);
260 }
261
262 static int
263 ubus_lua_objects(lua_State *L)
264 {
265         int rv;
266         struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
267
268         lua_newtable(L);
269         rv = ubus_lookup(c->ctx, NULL, ubus_lua_objects_cb, L);
270
271         if (rv != UBUS_STATUS_OK)
272         {
273                 lua_pop(L, 1);
274                 lua_pushnil(L);
275                 lua_pushinteger(L, rv);
276                 return 2;
277         }
278
279         return 1;
280 }
281
282 static int
283 ubus_method_handler(struct ubus_context *ctx, struct ubus_object *obj,
284                 struct ubus_request_data *req, const char *method,
285                 struct blob_attr *msg)
286 {
287         struct ubus_lua_object *o = container_of(obj, struct ubus_lua_object, o);
288         int rv = 0;
289
290         lua_getglobal(state, "__ubus_cb");
291         lua_rawgeti(state, -1, o->r);
292         lua_getfield(state, -1, method);
293         lua_remove(state, -2);
294         lua_remove(state, -2);
295
296         if (lua_isfunction(state, -1)) {
297                 lua_pushlightuserdata(state, req);
298                 if (!msg)
299                         lua_pushnil(state);
300                 else
301                         ubus_lua_parse_blob_array(state, blob_data(msg), blob_len(msg), true);
302                 lua_call(state, 2, 1);
303                 if (lua_isnumber(state, -1))
304                         rv = lua_tonumber(state, -1);
305         } else
306                 lua_pop(state, 1);
307
308         return rv;
309 }
310
311 static int lua_gettablelen(lua_State *L, int index)
312 {
313         int cnt = 0;
314
315         lua_pushnil(L);
316         index -= 1;
317         while (lua_next(L, index) != 0) {
318                 cnt++;
319                 lua_pop(L, 1);
320         }
321
322         return cnt;
323 }
324
325 static int ubus_lua_reply(lua_State *L)
326 {
327         struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
328         struct ubus_request_data *req;
329
330         luaL_checktype(L, 3, LUA_TTABLE);
331         blob_buf_init(&c->buf, 0);
332
333         if (!ubus_lua_format_blob_array(L, &c->buf, true))
334         {
335                 lua_pushnil(L);
336                 lua_pushinteger(L, UBUS_STATUS_INVALID_ARGUMENT);
337                 return 2;
338         }
339
340         req = lua_touserdata(L, 2);
341         ubus_send_reply(c->ctx, req, c->buf.head);
342
343         return 0;
344 }
345
346 static int ubus_lua_load_methods(lua_State *L, struct ubus_method *m)
347 {
348         struct blobmsg_policy *p;
349         int plen;
350         int pidx = 0;
351
352         /* get the function pointer */
353         lua_pushinteger(L, 1);
354         lua_gettable(L, -2);
355
356         /* get the policy table */
357         lua_pushinteger(L, 2);
358         lua_gettable(L, -3);
359
360         /* check if the method table is valid */
361         if ((lua_type(L, -2) != LUA_TFUNCTION) ||
362                         (lua_type(L, -1) != LUA_TTABLE) ||
363                         lua_objlen(L, -1)) {
364                 lua_pop(L, 2);
365                 return 1;
366         }
367
368         /* store function pointer */
369         lua_pushvalue(L, -2);
370         lua_setfield(L, -6, lua_tostring(L, -5));
371
372         m->name = lua_tostring(L, -4);
373         m->handler = ubus_method_handler;
374
375         plen = lua_gettablelen(L, -1);
376
377         /* exit if policy table is empty */
378         if (!plen) {
379                 lua_pop(L, 2);
380                 return 0;
381         }
382
383         /* setup the policy pointers */
384         p = malloc(sizeof(struct blobmsg_policy) * plen);
385         if (!p)
386                 return 1;
387
388         memset(p, 0, sizeof(struct blobmsg_policy) * plen);
389         m->policy = p;
390         lua_pushnil(L);
391         while (lua_next(L, -2) != 0) {
392                 int val = lua_tointeger(L, -1);
393
394                 /* check if the policy is valid */
395                 if ((lua_type(L, -2) != LUA_TSTRING) ||
396                                 (lua_type(L, -1) != LUA_TNUMBER) ||
397                                 (val < 0) ||
398                                 (val > BLOBMSG_TYPE_LAST)) {
399                         lua_pop(L, 1);
400                         continue;
401                 }
402                 p[pidx].name = lua_tostring(L, -2);
403                 p[pidx].type = val;
404                 lua_pop(L, 1);
405                 pidx++;
406         }
407
408         m->n_policy = pidx;
409         lua_pop(L, 2);
410
411         return 0;
412 }
413
414 static struct ubus_object* ubus_lua_load_object(lua_State *L)
415 {
416         struct ubus_lua_object *obj = NULL;
417         int mlen = lua_gettablelen(L, -1);
418         struct ubus_method *m;
419         int midx = 0;
420
421         /* setup object pointers */
422         obj = malloc(sizeof(struct ubus_lua_object));
423         if (!obj)
424                 return NULL;
425
426         memset(obj, 0, sizeof(struct ubus_lua_object));
427         obj->o.name = lua_tostring(L, -2);
428
429         /* setup method pointers */
430         m = malloc(sizeof(struct ubus_method) * mlen);
431         memset(m, 0, sizeof(struct ubus_method) * mlen);
432         obj->o.methods = m;
433
434         /* setup type pointers */
435         obj->o.type = malloc(sizeof(struct ubus_object_type));
436         if (!obj->o.type) {
437                 free(obj);
438                 return NULL;
439         }
440
441         memset(obj->o.type, 0, sizeof(struct ubus_object_type));
442         obj->o.type->name = lua_tostring(L, -2);
443         obj->o.type->id = 0;
444         obj->o.type->methods = obj->o.methods;
445
446         /* create the callback lookup table */
447         lua_createtable(L, 1, 0);
448         lua_getglobal(L, "__ubus_cb");
449         lua_pushvalue(L, -2);
450         obj->r = luaL_ref(L, -2);
451         lua_pop(L, 1);
452
453         /* scan each method */
454         lua_pushnil(L);
455         while (lua_next(L, -3) != 0) {
456                 /* check if it looks like a method */
457                 if ((lua_type(L, -2) != LUA_TSTRING) ||
458                                 (lua_type(L, -1) != LUA_TTABLE) ||
459                                 !lua_objlen(L, -1)) {
460                         lua_pop(L, 1);
461                         continue;
462                 }
463
464                 if (!ubus_lua_load_methods(L, &m[midx]))
465                         midx++;
466                 lua_pop(L, 1);
467         }
468
469         obj->o.type->n_methods = obj->o.n_methods = midx;
470
471         /* pop the callback table */
472         lua_pop(L, 1);
473
474         return &obj->o;
475 }
476
477 static int ubus_lua_add(lua_State *L)
478 {
479         struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
480
481         /* verify top level object */
482         if (lua_istable(L, 1)) {
483                 lua_pushstring(L, "you need to pass a table");
484                 lua_error(L);
485                 return 0;
486         }
487
488         /* scan each object */
489         lua_pushnil(L);
490         while (lua_next(L, -2) != 0) {
491                 struct ubus_object *obj = NULL;
492
493                 /* check if the object has a table of methods */
494                 if ((lua_type(L, -2) == LUA_TSTRING) && (lua_type(L, -1) == LUA_TTABLE)) {
495                         obj = ubus_lua_load_object(L);
496
497                         if (obj)
498                                 ubus_add_object(c->ctx, obj);
499                 }
500                 lua_pop(L, 1);
501         }
502
503         return 0;
504 }
505
506 static void
507 ubus_lua_signatures_cb(struct ubus_context *c, struct ubus_object_data *o, void *p)
508 {
509         lua_State *L = (lua_State *)p;
510
511         if (!o->signature)
512                 return;
513
514         ubus_lua_parse_blob_array(L, blob_data(o->signature), blob_len(o->signature), true);
515 }
516
517 static int
518 ubus_lua_signatures(lua_State *L)
519 {
520         int rv;
521         struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
522         const char *path = luaL_checkstring(L, 2);
523
524         rv = ubus_lookup(c->ctx, path, ubus_lua_signatures_cb, L);
525
526         if (rv != UBUS_STATUS_OK)
527         {
528                 lua_pop(L, 1);
529                 lua_pushnil(L);
530                 lua_pushinteger(L, rv);
531                 return 2;
532         }
533
534         return 1;
535 }
536
537
538 static void
539 ubus_lua_call_cb(struct ubus_request *req, int type, struct blob_attr *msg)
540 {
541         lua_State *L = (lua_State *)req->priv;
542
543         if (!msg && L)
544                 lua_pushnil(L);
545
546         if (msg && L)
547                 ubus_lua_parse_blob_array(L, blob_data(msg), blob_len(msg), true);
548 }
549
550 static int
551 ubus_lua_call(lua_State *L)
552 {
553         int rv, top;
554         uint32_t id;
555         struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
556         const char *path = luaL_checkstring(L, 2);
557         const char *func = luaL_checkstring(L, 3);
558
559         luaL_checktype(L, 4, LUA_TTABLE);
560         blob_buf_init(&c->buf, 0);
561
562         if (!ubus_lua_format_blob_array(L, &c->buf, true))
563         {
564                 lua_pushnil(L);
565                 lua_pushinteger(L, UBUS_STATUS_INVALID_ARGUMENT);
566                 return 2;
567         }
568
569         rv = ubus_lookup_id(c->ctx, path, &id);
570
571         if (rv)
572         {
573                 lua_pushnil(L);
574                 lua_pushinteger(L, rv);
575                 return 2;
576         }
577
578         top = lua_gettop(L);
579         rv = ubus_invoke(c->ctx, id, func, c->buf.head, ubus_lua_call_cb, L, c->timeout * 1000);
580
581         if (rv != UBUS_STATUS_OK)
582         {
583                 lua_pop(L, 1);
584                 lua_pushnil(L);
585                 lua_pushinteger(L, rv);
586                 return 2;
587         }
588
589         return lua_gettop(L) - top;
590 }
591
592 static void
593 ubus_event_handler(struct ubus_context *ctx, struct ubus_event_handler *ev,
594                         const char *type, struct blob_attr *msg)
595 {
596         struct ubus_lua_event *listener = container_of(ev, struct ubus_lua_event, e);
597
598         lua_getglobal(state, "__ubus_cb_event");
599         lua_rawgeti(state, -1, listener->r);
600
601         if (lua_isfunction(state, -1)) {
602                 ubus_lua_parse_blob_array(state, blob_data(msg), blob_len(msg), true);
603                 lua_call(state, 1, 0);
604         }
605 }
606
607 static struct ubus_event_handler*
608 ubus_lua_load_event(lua_State *L)
609 {
610         struct ubus_lua_event* event = NULL;
611
612         event = malloc(sizeof(struct ubus_lua_event));
613         if (!event)
614                 return NULL;
615
616         memset(event, 0, sizeof(struct ubus_lua_event));
617         event->e.cb = ubus_event_handler;
618
619         /* update the he callback lookup table */
620         lua_getglobal(L, "__ubus_cb_event");
621         lua_pushvalue(L, -2);
622         event->r = luaL_ref(L, -2);
623         lua_setfield(L, -1, lua_tostring(L, -3));
624
625         return &event->e;
626 }
627
628 static int
629 ubus_lua_listen(lua_State *L) {
630         struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
631
632         /* verify top level object */
633         luaL_checktype(L, 2, LUA_TTABLE);
634
635         /* scan each object */
636         lua_pushnil(L);
637         while (lua_next(L, -2) != 0) {
638                 struct ubus_event_handler *listener;
639
640                 /* check if the key is a string and the value is a method */
641                 if ((lua_type(L, -2) == LUA_TSTRING) && (lua_type(L, -1) == LUA_TFUNCTION)) {
642                         listener = ubus_lua_load_event(L);
643                         if(listener != NULL) {
644                                 ubus_register_event_handler(c->ctx, listener, lua_tostring(L, -2));
645                         }
646                 }
647                 lua_pop(L, 1);
648         }
649         return 0;
650 }
651
652 static int
653 ubus_lua_send(lua_State *L)
654 {
655         struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
656         const char *event = luaL_checkstring(L, 2);
657
658         if (*event == 0)
659                 return luaL_argerror(L, 2, "no event name");
660
661         // Event content convert to ubus form
662         luaL_checktype(L, 3, LUA_TTABLE);
663         blob_buf_init(&c->buf, 0);
664
665         if (!ubus_lua_format_blob_array(L, &c->buf, true)) {
666                 lua_pushnil(L);
667                 lua_pushinteger(L, UBUS_STATUS_INVALID_ARGUMENT);
668                 return 2;
669         }
670
671         // Send the event
672         ubus_send_event(c->ctx, event, c->buf.head);
673
674         return 0;
675 }
676
677
678
679 static int
680 ubus_lua__gc(lua_State *L)
681 {
682         struct ubus_lua_connection *c = luaL_checkudata(L, 1, METANAME);
683
684         blob_buf_free(&c->buf);
685         if (c->ctx != NULL)
686         {
687                 ubus_free(c->ctx);
688                 memset(c, 0, sizeof(*c));
689         }
690
691         return 0;
692 }
693
694 static const luaL_Reg ubus[] = {
695         { "connect", ubus_lua_connect },
696         { "objects", ubus_lua_objects },
697         { "add", ubus_lua_add },
698         { "reply", ubus_lua_reply },
699         { "signatures", ubus_lua_signatures },
700         { "call", ubus_lua_call },
701         { "close", ubus_lua__gc },
702         { "listen", ubus_lua_listen },
703         { "send", ubus_lua_send },
704         { "__gc", ubus_lua__gc },
705         { NULL, NULL },
706 };
707
708 /* avoid missing prototype warning */
709 int luaopen_ubus(lua_State *L);
710
711 int
712 luaopen_ubus(lua_State *L)
713 {
714         /* create metatable */
715         luaL_newmetatable(L, METANAME);
716
717         /* metatable.__index = metatable */
718         lua_pushvalue(L, -1);
719         lua_setfield(L, -2, "__index");
720
721         /* fill metatable */
722         luaL_register(L, NULL, ubus);
723         lua_pop(L, 1);
724
725         /* create module */
726         luaL_register(L, MODNAME, ubus);
727
728         /* set some enum defines */
729         lua_pushinteger(L, BLOBMSG_TYPE_ARRAY);
730         lua_setfield(L, -2, "ARRAY");
731         lua_pushinteger(L, BLOBMSG_TYPE_TABLE);
732         lua_setfield(L, -2, "TABLE");
733         lua_pushinteger(L, BLOBMSG_TYPE_STRING);
734         lua_setfield(L, -2, "STRING");
735         lua_pushinteger(L, BLOBMSG_TYPE_INT64);
736         lua_setfield(L, -2, "INT64");
737         lua_pushinteger(L, BLOBMSG_TYPE_INT32);
738         lua_setfield(L, -2, "INT32");
739         lua_pushinteger(L, BLOBMSG_TYPE_INT16);
740         lua_setfield(L, -2, "INT16");
741         lua_pushinteger(L, BLOBMSG_TYPE_INT8);
742         lua_setfield(L, -2, "INT8");
743         lua_pushinteger(L, BLOBMSG_TYPE_BOOL);
744         lua_setfield(L, -2, "BOOLEAN");
745
746         /* used in our callbacks */
747         state = L;
748
749         /* create the callback table */
750         lua_createtable(L, 1, 0);
751         lua_setglobal(L, "__ubus_cb");
752
753         /* create the event table */
754         lua_createtable(L, 1, 0);
755         lua_setglobal(L, "__ubus_cb_event");
756
757         return 0;
758 }