X-Git-Url: https://git.archive.openwrt.org/?p=project%2Fuci.git;a=blobdiff_plain;f=lua%2Fuci.c;h=3f8f3a57e79d11e5a393117ec506edde3f74835e;hp=0a751d677a7d6193d1aad1368de93c6535cf8736;hb=946c51e93303ff71dd5cc44db8a1c2d1d5914625;hpb=4837364113f30c793b373073d27508fec6234ed3

diff --git a/lua/uci.c b/lua/uci.c
index 0a751d6..3f8f3a5 100644
--- a/lua/uci.c
+++ b/lua/uci.c
@@ -14,6 +14,7 @@
 
 #include <sys/types.h>
 #include <sys/time.h>
+#include <stdbool.h>
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
@@ -33,19 +34,58 @@
 #endif
 
 static struct uci_context *ctx = NULL;
+enum autoload {
+	AUTOLOAD_OFF = 0,
+	AUTOLOAD_ON = 1,
+	AUTOLOAD_FORCE = 2
+};
 
 static struct uci_package *
-find_package(const char *name)
+find_package(lua_State *L, const char *str, enum autoload al)
 {
 	struct uci_package *p = NULL;
 	struct uci_element *e;
+	char *sep;
+	char *name;
+
+	sep = strchr(str, '.');
+	if (sep) {
+		name = malloc(1 + sep - str);
+		if (!name)
+			luaL_error(L, "out of memory");
+		strncpy(name, str, sep - str);
+		name[sep - str] = 0;
+	} else
+		name = (char *) str;
+
 	uci_foreach_element(&ctx->root, e) {
 		if (strcmp(e->name, name) != 0)
 			continue;
 
 		p = uci_to_package(e);
-		break;
+		goto done;
+	}
+
+	if (al == AUTOLOAD_FORCE)
+		uci_load(ctx, name, &p);
+	else if (al) {
+		do {
+			lua_getfield(L, LUA_GLOBALSINDEX, "uci");
+			lua_getfield(L, -1, "autoload");
+			if (!lua_isboolean(L, -1))
+				break;
+
+			if (!lua_toboolean(L, -1))
+				break;
+
+			uci_load(ctx, name, &p);
+		} while (0);
+		lua_pop(L, 2);
 	}
+
+done:
+	if (name != str)
+		free(name);
 	return p;
 }
 
@@ -63,14 +103,114 @@ done:
 }
 
 static int
+lookup_args(lua_State *L, struct uci_ptr *ptr, char **buf)
+{
+	char *s = NULL;
+	int n;
+
+	n = lua_gettop(L);
+	luaL_checkstring(L, 1);
+	s = strdup(lua_tostring(L, 1));
+	if (!s)
+		goto error;
+
+	memset(ptr, 0, sizeof(struct uci_ptr));
+	if (!find_package(L, s, AUTOLOAD_ON))
+		goto error;
+
+	switch (n) {
+	case 4:
+	case 3:
+		ptr->option = luaL_checkstring(L, 3);
+		/* fall through */
+	case 2:
+		ptr->section = luaL_checkstring(L, 2);
+		ptr->package = luaL_checkstring(L, 1);
+		if (uci_lookup_ptr(ctx, ptr, NULL, false) != UCI_OK)
+			goto error;
+		break;
+	case 1:
+		if (uci_lookup_ptr(ctx, ptr, s, false) != UCI_OK)
+			goto error;
+		break;
+	default:
+		luaL_error(L, "invalid argument count");
+		goto error;
+	}
+
+	*buf = s;
+	return 0;
+
+error:
+	if (s)
+		free(s);
+	return 1;
+}
+
+static void uci_push_option(lua_State *L, struct uci_option *o)
+{
+	struct uci_element *e;
+	int i = 0;
+
+	switch(o->type) {
+	case UCI_TYPE_STRING:
+		lua_pushstring(L, o->v.string);
+		break;
+	case UCI_TYPE_LIST:
+		lua_newtable(L);
+		uci_foreach_element(&o->v.list, e) {
+			i++;
+			lua_pushstring(L, e->name);
+			lua_rawseti(L, -2, i);
+		}
+		break;
+	default:
+		lua_pushnil(L);
+		break;
+	}
+}
+
+static void uci_push_section(lua_State *L, struct uci_section *s)
+{
+	struct uci_element *e;
+
+	lua_newtable(L);
+	lua_pushboolean(L, s->anonymous);
+	lua_setfield(L, -2, ".anonymous");
+	lua_pushstring(L, s->type);
+	lua_setfield(L, -2, ".type");
+	lua_pushstring(L, s->e.name);
+	lua_setfield(L, -2, ".name");
+
+	uci_foreach_element(&s->options, e) {
+		struct uci_option *o = uci_to_option(e);
+		uci_push_option(L, o);
+		lua_setfield(L, -2, o->e.name);
+	}
+}
+
+static void uci_push_package(lua_State *L, struct uci_package *p)
+{
+	struct uci_element *e;
+	int i = 0;
+
+	lua_newtable(L);
+	uci_foreach_element(&p->sections, e) {
+		i++;
+		uci_push_section(L, uci_to_section(e));
+		lua_setfield(L, -2, e->name);
+	}
+}
+
+static int
 uci_lua_unload(lua_State *L)
 {
 	struct uci_package *p;
 	const char *s;
 
-	luaL_checkany(L, 1);
+	luaL_checkstring(L, 1);
 	s = lua_tostring(L, -1);
-	p = find_package(s);
+	p = find_package(L, s, AUTOLOAD_OFF);
 	if (p) {
 		uci_unload(ctx, p);
 		lua_pushboolean(L, 1);
@@ -100,46 +240,81 @@ uci_lua_load(lua_State *L)
 	return 1;
 }
 
+
 static int
-uci_lua_get(lua_State *L)
+uci_lua_foreach(lua_State *L)
 {
-	struct uci_lua_context *f;
-	struct uci_element *e = NULL;
-	struct uci_package *p = NULL;
-	char *package = NULL;
-	char *section = NULL;
-	char *option = NULL;
-	char *s;
-	int err = UCI_ERR_MEM;
+	struct uci_package *p;
+	struct uci_element *e;
+	const char *package, *type;
+	bool ret = false;
 
-	luaL_checkany(L, 1);
-	s = strdup(lua_tostring(L, -1));
-	if (!s)
-		goto error;
+	package = luaL_checkstring(L, 1);
+
+	if (lua_isnil(L, 2))
+		type = NULL;
+	else
+		type = luaL_checkstring(L, 2);
+
+	if (!lua_isfunction(L, 3) || !package)
+		luaL_error(L, "Invalid argument");
+
+	p = find_package(L, package, AUTOLOAD_ON);
+	if (!p)
+		goto done;
+
+	uci_foreach_element(&p->sections, e) {
+		struct uci_section *s = uci_to_section(e);
+
+		if (type && (strcmp(s->type, type) != 0))
+			continue;
+
+		lua_pushvalue(L, 3); /* iterator function */
+		uci_push_section(L, s);
+		if (lua_pcall(L, 1, 0, 0) == 0)
+			ret = true;
+	}
 
-	if ((err = uci_parse_tuple(ctx, s, &package, &section, &option, NULL)))
+done:
+	lua_pushboolean(L, ret);
+	return 1;
+}
+
+static int
+uci_lua_get_any(lua_State *L, bool all)
+{
+	struct uci_element *e = NULL;
+	struct uci_ptr ptr;
+	char *s = NULL;
+	int err = UCI_ERR_NOTFOUND;
+
+	if (lookup_args(L, &ptr, &s))
 		goto error;
 
-	if (section == NULL) {
+	uci_lookup_ptr(ctx, &ptr, NULL, false);
+	if (!all && !ptr.s) {
 		err = UCI_ERR_INVAL;
 		goto error;
 	}
-
-	p = find_package(package);
-	if (!p) {
+	if (!(ptr.flags & UCI_LOOKUP_COMPLETE)) {
 		err = UCI_ERR_NOTFOUND;
 		goto error;
 	}
 
-	if ((err = uci_lookup(ctx, &e, p, section, option)))
-		goto error;
-
+	err = UCI_OK;
+	e = ptr.last;
 	switch(e->type) {
+		case UCI_TYPE_PACKAGE:
+			uci_push_package(L, ptr.p);
+			break;
 		case UCI_TYPE_SECTION:
-			lua_pushstring(L, uci_to_section(e)->type);
+			if (all)
+				uci_push_section(L, ptr.s);
+			else
+				lua_pushstring(L, ptr.s->type);
 			break;
 		case UCI_TYPE_OPTION:
-			lua_pushstring(L, uci_to_option(e)->value);
+			uci_push_option(L, ptr.o);
 			break;
 		default:
 			err = UCI_ERR_INVAL;
@@ -162,11 +337,354 @@ error:
 	}
 }
 
+static int
+uci_lua_get(lua_State *L)
+{
+	return uci_lua_get_any(L, false);
+}
+
+static int
+uci_lua_get_all(lua_State *L)
+{
+	return uci_lua_get_any(L, true);
+}
+
+static int
+uci_lua_add(lua_State *L)
+{
+	struct uci_section *s = NULL;
+	struct uci_package *p;
+	const char *package;
+	const char *type;
+	const char *name = NULL;
+
+	do {
+		package = luaL_checkstring(L, 1);
+		type = luaL_checkstring(L, 2);
+		p = find_package(L, package, AUTOLOAD_ON);
+		if (!p)
+			break;
+
+		if (uci_add_section(ctx, p, type, &s) || !s)
+			break;
+
+		name = s->e.name;
+	} while (0);
+
+	lua_pushstring(L, name);
+	return 1;
+}
+
+static int
+uci_lua_delete(lua_State *L)
+{
+	struct uci_ptr ptr;
+	char *s = NULL;
+	int err = UCI_ERR_NOTFOUND;
+
+	if (lookup_args(L, &ptr, &s))
+		goto error;
+
+	err = uci_delete(ctx, &ptr);
+
+error:
+	if (s)
+		free(s);
+	if (err)
+		uci_lua_perror(L, "uci.delete");
+	lua_pushboolean(L, (err == 0));
+	return 1;
+}
+
+static int
+uci_lua_set(lua_State *L)
+{
+	bool istable = false;
+	struct uci_ptr ptr;
+	int err = UCI_ERR_MEM;
+	char *s = NULL;
+	int i, nargs;
+
+	nargs = lua_gettop(L);
+	if (lookup_args(L, &ptr, &s))
+		goto error;
+
+	switch(nargs) {
+	case 1:
+		/* Format: uci.set("p.s.o=v") or uci.set("p.s=v") */
+		break;
+	case 4:
+		/* Format: uci.set("p", "s", "o", "v") */
+		if (lua_istable(L, nargs)) {
+			if (lua_objlen(L, nargs) < 1)
+				luaL_error(L, "Cannot set an uci option to an empty table value");
+			lua_rawgeti(L, nargs, 1);
+			ptr.value = luaL_checkstring(L, -1);
+			lua_pop(L, 1);
+			istable = true;
+		} else {
+			ptr.value = luaL_checkstring(L, nargs);
+		}
+		break;
+	case 3:
+		/* Format: uci.set("p", "s", "v") */
+		ptr.value = ptr.option;
+		ptr.option = NULL;
+		break;
+	default:
+		err = UCI_ERR_INVAL;
+		goto error;
+	}
+
+	err = uci_lookup_ptr(ctx, &ptr, NULL, false);
+	if (err)
+		goto error;
+
+	if (((ptr.s == NULL) && (ptr.option != NULL)) || (ptr.value == NULL)) {
+		err = UCI_ERR_INVAL;
+		goto error;
+	}
+
+	err = uci_set(ctx, &ptr);
+	if (err)
+		goto error;
+
+	if (istable) {
+		for (i = 2; i <= lua_objlen(L, nargs); i++) {
+			lua_rawgeti(L, nargs, i);
+			ptr.value = luaL_checkstring(L, -1);
+			err = uci_add_list(ctx, &ptr);
+			lua_pop(L, 1);
+			if (err)
+				goto error;
+		}
+	}
+
+error:
+	if (err)
+		uci_lua_perror(L, "uci.set");
+	lua_pushboolean(L, (err == 0));
+	return 1;
+}
+
+enum pkg_cmd {
+	CMD_SAVE,
+	CMD_COMMIT,
+	CMD_REVERT
+};
+
+static int
+uci_lua_package_cmd(lua_State *L, enum pkg_cmd cmd)
+{
+	struct uci_element *e, *tmp;
+	struct uci_ptr ptr;
+	char *s = NULL;
+	int failed = 0;
+	int nargs;
+
+	nargs = lua_gettop(L);
+	if ((cmd != CMD_REVERT) && (nargs > 1))
+		goto err;
+
+	if (lookup_args(L, &ptr, &s))
+		goto err;
+
+	uci_lookup_ptr(ctx, &ptr, NULL, false);
+
+	uci_foreach_element_safe(&ctx->root, tmp, e) {
+		struct uci_package *p = uci_to_package(e);
+		int ret = UCI_ERR_INVAL;
+
+		if (ptr.p && (ptr.p != p))
+			continue;
+
+		ptr.p = p;
+		switch(cmd) {
+		case CMD_COMMIT:
+			ret = uci_commit(ctx, &p, false);
+			break;
+		case CMD_SAVE:
+			ret = uci_save(ctx, p);
+			break;
+		case CMD_REVERT:
+			ret = uci_revert(ctx, &ptr);
+			break;
+		}
+
+		if (ret != 0)
+			failed = 1;
+	}
+
+err:
+	lua_pushboolean(L, !failed);
+	return 1;
+}
+
+static int
+uci_lua_save(lua_State *L)
+{
+	return uci_lua_package_cmd(L, CMD_SAVE);
+}
+
+static int
+uci_lua_commit(lua_State *L)
+{
+	return uci_lua_package_cmd(L, CMD_COMMIT);
+}
+
+static int
+uci_lua_revert(lua_State *L)
+{
+	return uci_lua_package_cmd(L, CMD_REVERT);
+}
+
+static void
+uci_lua_add_change(lua_State *L, struct uci_element *e)
+{
+	struct uci_history *h;
+	const char *name;
+
+	h = uci_to_history(e);
+	if (!h->section)
+		return;
+
+	lua_getfield(L, -1, h->section);
+	if (lua_isnil(L, -1)) {
+		lua_pop(L, 1);
+		lua_newtable(L);
+		lua_pushvalue(L, -1); /* copy for setfield */
+		lua_setfield(L, -3, h->section);
+	}
+
+	name = (h->e.name ? h->e.name : ".type");
+	if (h->value)
+		lua_pushstring(L, h->value);
+	else
+		lua_pushstring(L, "");
+	lua_setfield(L, -2, name);
+	lua_pop(L, 1);
+}
+
+static void
+uci_lua_changes_pkg(lua_State *L, const char *package)
+{
+	struct uci_package *p = NULL;
+	struct uci_element *e;
+	bool autoload = false;
+
+	p = find_package(L, package, AUTOLOAD_OFF);
+	if (!p) {
+		autoload = true;
+		p = find_package(L, package, AUTOLOAD_FORCE);
+		if (!p)
+			return;
+	}
+
+	if (uci_list_empty(&p->history) && uci_list_empty(&p->saved_history))
+		goto done;
+
+	lua_newtable(L);
+	uci_foreach_element(&p->saved_history, e) {
+		uci_lua_add_change(L, e);
+	}
+	uci_foreach_element(&p->history, e) {
+		uci_lua_add_change(L, e);
+	}
+	lua_setfield(L, -2, p->e.name);
+
+done:
+	if (autoload)
+		uci_unload(ctx, p);
+}
+
+static int
+uci_lua_changes(lua_State *L)
+{
+	const char *package = NULL;
+	char **config = NULL;
+	int nargs;
+	int i;
+
+	nargs = lua_gettop(L);
+	switch(nargs) {
+	case 1:
+		package = luaL_checkstring(L, 1);
+	case 0:
+		break;
+	default:
+		luaL_error(L, "invalid argument count");
+	}
+
+	lua_newtable(L);
+	if (package) {
+		uci_lua_changes_pkg(L, package);
+	} else {
+		if (uci_list_configs(ctx, &config) != 0)
+			goto done;
+
+		for(i = 0; config[i] != NULL; i++) {
+			uci_lua_changes_pkg(L, config[i]);
+		}
+	}
+
+done:
+	return 1;
+}
+
+static int
+uci_lua_get_confdir(lua_State *L)
+{
+	lua_pushstring(L, ctx->confdir);
+	return 1;
+}
+
+static int
+uci_lua_set_confdir(lua_State *L)
+{
+	int ret;
+
+	luaL_checkstring(L, 1);
+	ret = uci_set_confdir(ctx, lua_tostring(L, -1));
+	lua_pushboolean(L, (ret == 0));
+	return 1;
+}
+
+static int
+uci_lua_get_savedir(lua_State *L)
+{
+	lua_pushstring(L, ctx->savedir);
+	return 1;
+}
+
+static int
+uci_lua_set_savedir(lua_State *L)
+{
+	int ret;
+
+	luaL_checkstring(L, 1);
+	ret = uci_set_savedir(ctx, lua_tostring(L, -1));
+	lua_pushboolean(L, (ret == 0));
+
+	return 1;
+}
 
 static const luaL_Reg uci[] = {
 	{ "load", uci_lua_load },
 	{ "unload", uci_lua_unload },
 	{ "get", uci_lua_get },
+	{ "get_all", uci_lua_get_all },
+	{ "add", uci_lua_add },
+	{ "set", uci_lua_set },
+	{ "save", uci_lua_save },
+	{ "delete", uci_lua_delete },
+	{ "commit", uci_lua_commit },
+	{ "revert", uci_lua_revert },
+	{ "changes", uci_lua_changes },
+	{ "foreach", uci_lua_foreach },
+	{ "get_confdir", uci_lua_get_confdir },
+	{ "set_confdir", uci_lua_set_confdir },
+	{ "get_savedir", uci_lua_get_savedir },
+	{ "set_savedir", uci_lua_set_savedir },
 	{ NULL, NULL },
 };
 
@@ -178,5 +696,12 @@ luaopen_uci(lua_State *L)
 	if (!ctx)
 		luaL_error(L, "Cannot allocate UCI context\n");
 	luaL_register(L, MODNAME, uci);
+
+	/* enable autoload by default */
+	lua_getfield(L, LUA_GLOBALSINDEX, "uci");
+	lua_pushboolean(L, 1);
+	lua_setfield(L, -2, "autoload");
+	lua_pop(L, 1);
+
 	return 0;
 }