2 * libuci plugin for Lua
3 * Copyright (C) 2008 Felix Fietkau <nbd@openwrt.org>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2
7 * as published by the Free Software Foundation
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.
15 #include <sys/types.h>
28 #define METANAME MODNAME ".meta"
32 #define DPRINTF(...) fprintf(stderr, __VA_ARGS__)
34 #define DPRINTF(...) do {} while (0)
37 static struct uci_context *global_ctx = NULL;
39 static struct uci_context *
40 find_context(lua_State *L, int *offset)
42 struct uci_context **ctx;
43 if (!lua_isuserdata(L, 1)) {
45 global_ctx = uci_alloc_context();
47 luaL_error(L, "failed to allocate UCI context");
55 ctx = luaL_checkudata(L, 1, METANAME);
57 luaL_error(L, "failed to get UCI context");
62 static struct uci_package *
63 find_package(lua_State *L, struct uci_context *ctx, const char *str, bool al)
65 struct uci_package *p = NULL;
66 struct uci_element *e;
70 sep = strchr(str, '.');
72 name = malloc(1 + sep - str);
74 luaL_error(L, "out of memory");
75 strncpy(name, str, sep - str);
80 uci_foreach_element(&ctx->root, e) {
81 if (strcmp(e->name, name) != 0)
84 p = uci_to_package(e);
89 uci_load(ctx, name, &p);
91 uci_load(ctx, name, &p);
101 lookup_args(lua_State *L, struct uci_context *ctx, int offset, struct uci_ptr *ptr, char **buf)
107 luaL_checkstring(L, 1 + offset);
108 s = strdup(lua_tostring(L, 1 + offset));
112 memset(ptr, 0, sizeof(struct uci_ptr));
113 if (!find_package(L, ctx, s, true))
116 switch (n - offset) {
119 ptr->option = luaL_checkstring(L, 3 + offset);
122 ptr->section = luaL_checkstring(L, 2 + offset);
123 ptr->package = luaL_checkstring(L, 1 + offset);
124 if (uci_lookup_ptr(ctx, ptr, NULL, false) != UCI_OK)
128 if (uci_lookup_ptr(ctx, ptr, s, false) != UCI_OK)
132 luaL_error(L, "invalid argument count");
146 uci_push_status(lua_State *L, struct uci_context *ctx, bool hasarg)
151 lua_pushboolean(L, (ctx->err == UCI_OK));
153 uci_get_errorstr(ctx, &str, MODNAME);
155 lua_pushstring(L, str);
164 uci_push_option(lua_State *L, struct uci_option *o)
166 struct uci_element *e;
170 case UCI_TYPE_STRING:
171 lua_pushstring(L, o->v.string);
175 uci_foreach_element(&o->v.list, e) {
177 lua_pushstring(L, e->name);
178 lua_rawseti(L, -2, i);
188 uci_push_section(lua_State *L, struct uci_section *s)
190 struct uci_element *e;
193 lua_pushboolean(L, s->anonymous);
194 lua_setfield(L, -2, ".anonymous");
195 lua_pushstring(L, s->type);
196 lua_setfield(L, -2, ".type");
197 lua_pushstring(L, s->e.name);
198 lua_setfield(L, -2, ".name");
200 uci_foreach_element(&s->options, e) {
201 struct uci_option *o = uci_to_option(e);
202 uci_push_option(L, o);
203 lua_setfield(L, -2, o->e.name);
208 uci_push_package(lua_State *L, struct uci_package *p)
210 struct uci_element *e;
214 uci_foreach_element(&p->sections, e) {
216 uci_push_section(L, uci_to_section(e));
217 lua_setfield(L, -2, e->name);
222 uci_lua_unload(lua_State *L)
224 struct uci_context *ctx;
225 struct uci_package *p;
229 ctx = find_context(L, &offset);
230 luaL_checkstring(L, 1 + offset);
231 s = lua_tostring(L, 1 + offset);
232 p = find_package(L, ctx, s, false);
235 return uci_push_status(L, ctx, false);
237 lua_pushboolean(L, 0);
243 uci_lua_load(lua_State *L)
245 struct uci_context *ctx;
246 struct uci_package *p = NULL;
250 ctx = find_context(L, &offset);
252 lua_pop(L, 1); /* bool ret value of unload */
253 s = lua_tostring(L, -1);
255 uci_load(ctx, s, &p);
256 return uci_push_status(L, ctx, false);
261 uci_lua_foreach(lua_State *L)
263 struct uci_context *ctx;
264 struct uci_package *p;
265 struct uci_element *e;
266 const char *package, *type;
270 ctx = find_context(L, &offset);
271 package = luaL_checkstring(L, 1 + offset);
276 type = luaL_checkstring(L, 2 + offset);
278 if (!lua_isfunction(L, 3 + offset) || !package)
279 luaL_error(L, "Invalid argument");
281 p = find_package(L, ctx, package, true);
285 uci_foreach_element(&p->sections, e) {
286 struct uci_section *s = uci_to_section(e);
288 if (type && (strcmp(s->type, type) != 0))
291 lua_pushvalue(L, 3 + offset); /* iterator function */
292 uci_push_section(L, s);
293 if (lua_pcall(L, 1, 0, 0) == 0)
298 lua_pushboolean(L, ret);
303 uci_lua_get_any(lua_State *L, bool all)
305 struct uci_context *ctx;
306 struct uci_element *e = NULL;
310 int err = UCI_ERR_NOTFOUND;
312 ctx = find_context(L, &offset);
314 if (lookup_args(L, ctx, offset, &ptr, &s))
317 uci_lookup_ptr(ctx, &ptr, NULL, false);
318 if (!all && !ptr.s) {
322 if (!(ptr.flags & UCI_LOOKUP_COMPLETE)) {
323 err = UCI_ERR_NOTFOUND;
330 case UCI_TYPE_PACKAGE:
331 uci_push_package(L, ptr.p);
333 case UCI_TYPE_SECTION:
335 uci_push_section(L, ptr.s);
337 lua_pushstring(L, ptr.s->type);
339 case UCI_TYPE_OPTION:
340 uci_push_option(L, ptr.o);
354 return uci_push_status(L, ctx, true);
358 uci_lua_get(lua_State *L)
360 return uci_lua_get_any(L, false);
364 uci_lua_get_all(lua_State *L)
366 return uci_lua_get_any(L, true);
370 uci_lua_add(lua_State *L)
372 struct uci_context *ctx;
373 struct uci_section *s = NULL;
374 struct uci_package *p;
377 const char *name = NULL;
380 ctx = find_context(L, &offset);
381 package = luaL_checkstring(L, 1 + offset);
382 type = luaL_checkstring(L, 2 + offset);
383 p = find_package(L, ctx, package, true);
387 if (uci_add_section(ctx, p, type, &s) || !s)
391 lua_pushstring(L, name);
396 return uci_push_status(L, ctx, true);
400 uci_lua_delete(lua_State *L)
402 struct uci_context *ctx;
406 int err = UCI_ERR_NOTFOUND;
408 ctx = find_context(L, &offset);
410 if (lookup_args(L, ctx, offset, &ptr, &s))
413 err = uci_delete(ctx, &ptr);
418 return uci_push_status(L, ctx, false);
422 uci_lua_set(lua_State *L)
424 struct uci_context *ctx;
426 bool istable = false;
427 int err = UCI_ERR_MEM;
429 int i, nargs, offset = 0;
431 ctx = find_context(L, &offset);
432 nargs = lua_gettop(L);
433 if (lookup_args(L, ctx, offset, &ptr, &s))
436 switch(nargs - offset) {
438 /* Format: uci.set("p.s.o=v") or uci.set("p.s=v") */
441 /* Format: uci.set("p", "s", "o", "v") */
442 if (lua_istable(L, nargs)) {
443 if (lua_objlen(L, nargs) < 1)
444 luaL_error(L, "Cannot set an uci option to an empty table value");
445 lua_rawgeti(L, nargs, 1);
446 ptr.value = luaL_checkstring(L, -1);
450 ptr.value = luaL_checkstring(L, nargs);
454 /* Format: uci.set("p", "s", "v") */
455 ptr.value = ptr.option;
463 err = uci_lookup_ptr(ctx, &ptr, NULL, false);
467 if (((ptr.s == NULL) && (ptr.option != NULL)) || (ptr.value == NULL)) {
472 err = uci_set(ctx, &ptr);
477 for (i = 2; i <= lua_objlen(L, nargs); i++) {
478 lua_rawgeti(L, nargs, i);
479 ptr.value = luaL_checkstring(L, -1);
480 err = uci_add_list(ctx, &ptr);
488 return uci_push_status(L, ctx, false);
498 uci_lua_package_cmd(lua_State *L, enum pkg_cmd cmd)
500 struct uci_context *ctx;
501 struct uci_element *e, *tmp;
505 int nargs, offset = 0;
507 ctx = find_context(L, &offset);
508 nargs = lua_gettop(L);
509 if ((cmd != CMD_REVERT) && (nargs - offset > 1))
512 if (lookup_args(L, ctx, offset, &ptr, &s))
515 uci_lookup_ptr(ctx, &ptr, NULL, false);
517 uci_foreach_element_safe(&ctx->root, tmp, e) {
518 struct uci_package *p = uci_to_package(e);
519 int ret = UCI_ERR_INVAL;
521 if (ptr.p && (ptr.p != p))
527 ret = uci_commit(ctx, &p, false);
530 ret = uci_save(ctx, p);
533 ret = uci_revert(ctx, &ptr);
542 return uci_push_status(L, ctx, false);
546 uci_lua_save(lua_State *L)
548 return uci_lua_package_cmd(L, CMD_SAVE);
552 uci_lua_commit(lua_State *L)
554 return uci_lua_package_cmd(L, CMD_COMMIT);
558 uci_lua_revert(lua_State *L)
560 return uci_lua_package_cmd(L, CMD_REVERT);
564 uci_lua_add_change(lua_State *L, struct uci_element *e)
566 struct uci_history *h;
569 h = uci_to_history(e);
573 lua_getfield(L, -1, h->section);
574 if (lua_isnil(L, -1)) {
577 lua_pushvalue(L, -1); /* copy for setfield */
578 lua_setfield(L, -3, h->section);
581 name = (h->e.name ? h->e.name : ".type");
583 lua_pushstring(L, h->value);
585 lua_pushstring(L, "");
586 lua_setfield(L, -2, name);
591 uci_lua_changes_pkg(lua_State *L, struct uci_context *ctx, const char *package)
593 struct uci_package *p = NULL;
594 struct uci_element *e;
595 bool autoload = false;
597 p = find_package(L, ctx, package, false);
600 p = find_package(L, ctx, package, true);
605 if (uci_list_empty(&p->history) && uci_list_empty(&p->saved_history))
609 uci_foreach_element(&p->saved_history, e) {
610 uci_lua_add_change(L, e);
612 uci_foreach_element(&p->history, e) {
613 uci_lua_add_change(L, e);
615 lua_setfield(L, -2, p->e.name);
623 uci_lua_changes(lua_State *L)
625 struct uci_context *ctx;
626 const char *package = NULL;
627 char **config = NULL;
631 ctx = find_context(L, &offset);
632 nargs = lua_gettop(L);
633 switch(nargs - offset) {
635 package = luaL_checkstring(L, 1 + offset);
639 luaL_error(L, "invalid argument count");
644 uci_lua_changes_pkg(L, ctx, package);
646 if (uci_list_configs(ctx, &config) != 0)
649 for(i = 0; config[i] != NULL; i++) {
650 uci_lua_changes_pkg(L, ctx, config[i]);
659 uci_lua_get_confdir(lua_State *L)
661 struct uci_context *ctx = find_context(L, NULL);
662 lua_pushstring(L, ctx->confdir);
667 uci_lua_set_confdir(lua_State *L)
669 struct uci_context *ctx;
672 ctx = find_context(L, &offset);
673 luaL_checkstring(L, 1 + offset);
674 ret = uci_set_confdir(ctx, lua_tostring(L, -1));
675 return uci_push_status(L, ctx, false);
679 uci_lua_get_savedir(lua_State *L)
681 struct uci_context *ctx = find_context(L, NULL);
682 lua_pushstring(L, ctx->savedir);
687 uci_lua_set_savedir(lua_State *L)
689 struct uci_context *ctx;
692 ctx = find_context(L, &offset);
693 luaL_checkstring(L, 1 + offset);
694 ret = uci_set_savedir(ctx, lua_tostring(L, -1));
695 return uci_push_status(L, ctx, false);
699 uci_lua_gc(lua_State *L)
701 struct uci_context *ctx = find_context(L, NULL);
702 uci_free_context(ctx);
707 uci_lua_cursor(lua_State *L)
709 struct uci_context **u;
710 int argc = lua_gettop(L);
712 u = lua_newuserdata(L, sizeof(struct uci_context *));
713 luaL_getmetatable(L, METANAME);
714 lua_setmetatable(L, -2);
716 *u = uci_alloc_context();
718 luaL_error(L, "Cannot allocate UCI context");
721 if (lua_isstring(L, 2) &&
722 (uci_set_savedir(*u, luaL_checkstring(L, 2)) != UCI_OK))
723 luaL_error(L, "Unable to set savedir");
726 if (lua_isstring(L, 1) &&
727 (uci_set_confdir(*u, luaL_checkstring(L, 1)) != UCI_OK))
728 luaL_error(L, "Unable to set savedir");
736 static const luaL_Reg uci[] = {
737 { "__gc", uci_lua_gc },
738 { "cursor", uci_lua_cursor },
739 { "load", uci_lua_load },
740 { "unload", uci_lua_unload },
741 { "get", uci_lua_get },
742 { "get_all", uci_lua_get_all },
743 { "add", uci_lua_add },
744 { "set", uci_lua_set },
745 { "save", uci_lua_save },
746 { "delete", uci_lua_delete },
747 { "commit", uci_lua_commit },
748 { "revert", uci_lua_revert },
749 { "changes", uci_lua_changes },
750 { "foreach", uci_lua_foreach },
751 { "get_confdir", uci_lua_get_confdir },
752 { "set_confdir", uci_lua_set_confdir },
753 { "get_savedir", uci_lua_get_savedir },
754 { "set_savedir", uci_lua_set_savedir },
760 luaopen_uci(lua_State *L)
762 /* create metatable */
763 luaL_newmetatable(L, METANAME);
765 /* metatable.__index = metatable */
766 lua_pushvalue(L, -1);
767 lua_setfield(L, -2, "__index");
770 luaL_register(L, NULL, uci);
774 luaL_register(L, MODNAME, uci);