mirror of https://github.com/hak5/openwrt-owl.git
add ucitrigger: a uci plugin, command line tool and lua interface for automatically applying uci config changes
SVN-Revision: 16375owl
parent
2cda2ff575
commit
e7cbce1f9e
|
@ -36,10 +36,17 @@ define Package/uci
|
|||
TITLE:=Utility for the Unified Configuration Interface (UCI)
|
||||
endef
|
||||
|
||||
define Package/ucitrigger
|
||||
SECTION:=base
|
||||
CATEGORY:=Base system
|
||||
DEPENDS:=+libuci-lua
|
||||
TITLE:=Automatic triggers for applying system config changes
|
||||
endef
|
||||
|
||||
define Package/libuci-lua
|
||||
SECTION=libs
|
||||
CATEGORY=Libraries
|
||||
DEPENDS:=+libuci +lua
|
||||
DEPENDS:=+libuci +liblua
|
||||
TITLE:=Lua plugin for UCI
|
||||
endef
|
||||
|
||||
|
@ -63,6 +70,8 @@ endif
|
|||
define Build/Compile
|
||||
$(MAKE) -C $(PKG_BUILD_DIR) $(UCI_MAKEOPTS)
|
||||
$(MAKE) -C $(PKG_BUILD_DIR)/lua $(UCI_MAKEOPTS)
|
||||
$(MAKE) -C $(PKG_BUILD_DIR)/trigger $(UCI_MAKEOPTS) \
|
||||
LIBS="$(TARGET_LDFLAGS) -L$(PKG_BUILD_DIR) -luci -llua -lcrypt -lm"
|
||||
endef
|
||||
|
||||
define Package/libuci/install
|
||||
|
@ -75,6 +84,14 @@ define Package/libuci-lua/install
|
|||
$(CP) $(PKG_BUILD_DIR)/lua/uci.so $(1)/usr/lib/lua/
|
||||
endef
|
||||
|
||||
define Package/ucitrigger/install
|
||||
$(INSTALL_DIR) $(1)/usr/lib/lua/uci $(1)/lib/config/trigger $(1)/usr/sbin
|
||||
$(INSTALL_DATA) ./trigger/lib/trigger.lua $(1)/usr/lib/lua/uci/
|
||||
$(INSTALL_DATA) ./trigger/modules/*.lua $(1)/lib/config/trigger/
|
||||
$(INSTALL_DATA) $(PKG_BUILD_DIR)/trigger/uci_trigger.so $(1)/usr/lib/
|
||||
$(INSTALL_BIN) ./trigger/apply_config $(1)/usr/sbin/
|
||||
endef
|
||||
|
||||
define Package/uci/install
|
||||
$(INSTALL_DIR) $(1)/etc/uci-defaults
|
||||
$(INSTALL_DIR) $(1)/sbin
|
||||
|
@ -94,3 +111,4 @@ endef
|
|||
$(eval $(call BuildPackage,uci))
|
||||
$(eval $(call BuildPackage,libuci))
|
||||
$(eval $(call BuildPackage,libuci-lua))
|
||||
$(eval $(call BuildPackage,ucitrigger))
|
||||
|
|
|
@ -0,0 +1,423 @@
|
|||
--- a/Makefile
|
||||
+++ b/Makefile
|
||||
@@ -7,7 +7,7 @@ DEBUG_TYPECAST=0
|
||||
|
||||
include Makefile.inc
|
||||
|
||||
-LIBS=-lc
|
||||
+LIBS=-lc -ldl
|
||||
SHLIB_FILE=libuci.$(SHLIB_EXT).$(VERSION)
|
||||
|
||||
define add_feature
|
||||
@@ -23,6 +23,7 @@ ucimap.o: ucimap.c uci.h uci_config.h uc
|
||||
|
||||
uci_config.h: FORCE
|
||||
@rm -f "$@.tmp"
|
||||
+ @echo "#define UCI_PREFIX \"$(prefix)\"" > "$@.tmp"
|
||||
$(call add_feature,PLUGIN_SUPPORT)
|
||||
$(call add_feature,DEBUG)
|
||||
$(call add_feature,DEBUG_TYPECAST)
|
||||
@@ -33,10 +34,10 @@ uci_config.h: FORCE
|
||||
fi
|
||||
|
||||
uci: cli.o libuci.$(SHLIB_EXT)
|
||||
- $(CC) -o $@ $< -L. -luci
|
||||
+ $(CC) -o $@ $< -L. -luci $(LIBS)
|
||||
|
||||
uci-static: cli.o libuci.a
|
||||
- $(CC) $(CFLAGS) -o $@ $^
|
||||
+ $(CC) $(CFLAGS) -o $@ $^ $(LIBS)
|
||||
|
||||
libuci-static.o: libuci.c $(LIBUCI_DEPS)
|
||||
$(CC) $(CFLAGS) -c -o $@ $<
|
||||
--- a/cli.c
|
||||
+++ b/cli.c
|
||||
@@ -27,6 +27,7 @@ static enum {
|
||||
CLI_FLAG_NOCOMMIT = (1 << 2),
|
||||
CLI_FLAG_BATCH = (1 << 3),
|
||||
CLI_FLAG_SHOW_EXT = (1 << 4),
|
||||
+ CLI_FLAG_NOPLUGINS= (1 << 5),
|
||||
} flags;
|
||||
|
||||
static FILE *input;
|
||||
@@ -136,6 +137,7 @@ static void uci_usage(void)
|
||||
"\t-c <path> set the search path for config files (default: /etc/config)\n"
|
||||
"\t-d <str> set the delimiter for list values in uci show\n"
|
||||
"\t-f <file> use <file> as input instead of stdin\n"
|
||||
+ "\t-L do not load any plugins\n"
|
||||
"\t-m when importing, merge data into an existing package\n"
|
||||
"\t-n name unnamed sections on export (default)\n"
|
||||
"\t-N don't name unnamed sections\n"
|
||||
@@ -603,7 +605,7 @@ int main(int argc, char **argv)
|
||||
return 1;
|
||||
}
|
||||
|
||||
- while((c = getopt(argc, argv, "c:d:f:mnNp:P:sSqX")) != -1) {
|
||||
+ while((c = getopt(argc, argv, "c:d:f:LmnNp:P:sSqX")) != -1) {
|
||||
switch(c) {
|
||||
case 'c':
|
||||
uci_set_confdir(ctx, optarg);
|
||||
@@ -618,6 +620,9 @@ int main(int argc, char **argv)
|
||||
return 1;
|
||||
}
|
||||
break;
|
||||
+ case 'L':
|
||||
+ flags |= CLI_FLAG_NOPLUGINS;
|
||||
+ break;
|
||||
case 'm':
|
||||
flags |= CLI_FLAG_MERGE;
|
||||
break;
|
||||
@@ -662,6 +667,10 @@ int main(int argc, char **argv)
|
||||
uci_usage();
|
||||
return 0;
|
||||
}
|
||||
+
|
||||
+ if (!(flags & CLI_FLAG_NOPLUGINS))
|
||||
+ uci_load_plugins(ctx, NULL);
|
||||
+
|
||||
ret = uci_cmd(argc - 1, argv + 1);
|
||||
if (input != stdin)
|
||||
fclose(input);
|
||||
--- a/history.c
|
||||
+++ b/history.c
|
||||
@@ -406,6 +406,17 @@ int uci_save(struct uci_context *ctx, st
|
||||
if ((asprintf(&filename, "%s/%s", ctx->savedir, p->e.name) < 0) || !filename)
|
||||
UCI_THROW(ctx, UCI_ERR_MEM);
|
||||
|
||||
+ uci_foreach_element(&ctx->hooks, tmp) {
|
||||
+ struct uci_hook *hook = uci_to_hook(tmp);
|
||||
+
|
||||
+ if (!hook->ops->set)
|
||||
+ continue;
|
||||
+
|
||||
+ uci_foreach_element(&p->history, e) {
|
||||
+ hook->ops->set(hook->ops, p, uci_to_history(e));
|
||||
+ }
|
||||
+ }
|
||||
+
|
||||
ctx->err = 0;
|
||||
UCI_TRAP_SAVE(ctx, done);
|
||||
f = uci_open_stream(ctx, filename, SEEK_END, true, true);
|
||||
--- a/libuci.c
|
||||
+++ b/libuci.c
|
||||
@@ -22,6 +22,8 @@
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
+#include <dlfcn.h>
|
||||
+#include <glob.h>
|
||||
#include "uci.h"
|
||||
|
||||
static const char *uci_confdir = UCI_CONFDIR;
|
||||
@@ -39,6 +41,7 @@ static const char *uci_errstr[] = {
|
||||
};
|
||||
|
||||
static void uci_cleanup(struct uci_context *ctx);
|
||||
+static void uci_unload_plugin(struct uci_context *ctx, struct uci_plugin *p);
|
||||
|
||||
#include "uci_internal.h"
|
||||
#include "util.c"
|
||||
@@ -56,6 +59,8 @@ struct uci_context *uci_alloc_context(vo
|
||||
uci_list_init(&ctx->root);
|
||||
uci_list_init(&ctx->history_path);
|
||||
uci_list_init(&ctx->backends);
|
||||
+ uci_list_init(&ctx->hooks);
|
||||
+ uci_list_init(&ctx->plugins);
|
||||
ctx->flags = UCI_FLAG_STRICT | UCI_FLAG_SAVED_HISTORY;
|
||||
|
||||
ctx->confdir = (char *) uci_confdir;
|
||||
@@ -86,6 +91,9 @@ void uci_free_context(struct uci_context
|
||||
uci_free_element(e);
|
||||
}
|
||||
UCI_TRAP_RESTORE(ctx);
|
||||
+ uci_foreach_element_safe(&ctx->root, tmp, e) {
|
||||
+ uci_unload_plugin(ctx, uci_to_plugin(e));
|
||||
+ }
|
||||
free(ctx);
|
||||
|
||||
ignore:
|
||||
@@ -209,9 +217,16 @@ int uci_commit(struct uci_context *ctx,
|
||||
int uci_load(struct uci_context *ctx, const char *name, struct uci_package **package)
|
||||
{
|
||||
struct uci_package *p;
|
||||
+ struct uci_element *e;
|
||||
+
|
||||
UCI_HANDLE_ERR(ctx);
|
||||
UCI_ASSERT(ctx, ctx->backend && ctx->backend->load);
|
||||
p = ctx->backend->load(ctx, name);
|
||||
+ uci_foreach_element(&ctx->hooks, e) {
|
||||
+ struct uci_hook *h = uci_to_hook(e);
|
||||
+ if (h->ops->load)
|
||||
+ h->ops->load(h->ops, p);
|
||||
+ }
|
||||
if (package)
|
||||
*package = p;
|
||||
|
||||
@@ -280,3 +295,94 @@ int uci_set_backend(struct uci_context *
|
||||
ctx->backend = uci_to_backend(e);
|
||||
return 0;
|
||||
}
|
||||
+
|
||||
+int uci_add_hook(struct uci_context *ctx, const struct uci_hook_ops *ops)
|
||||
+{
|
||||
+ struct uci_element *e;
|
||||
+ struct uci_hook *h;
|
||||
+
|
||||
+ UCI_HANDLE_ERR(ctx);
|
||||
+
|
||||
+ /* check for duplicate elements */
|
||||
+ uci_foreach_element(&ctx->hooks, e) {
|
||||
+ h = uci_to_hook(e);
|
||||
+ if (h->ops == ops)
|
||||
+ return UCI_ERR_INVAL;
|
||||
+ }
|
||||
+
|
||||
+ h = uci_alloc_element(ctx, hook, "", 0);
|
||||
+ h->ops = ops;
|
||||
+ uci_list_init(&h->e.list);
|
||||
+ uci_list_add(&ctx->hooks, &h->e.list);
|
||||
+
|
||||
+ return 0;
|
||||
+}
|
||||
+
|
||||
+int uci_remove_hook(struct uci_context *ctx, const struct uci_hook_ops *ops)
|
||||
+{
|
||||
+ struct uci_element *e;
|
||||
+
|
||||
+ uci_foreach_element(&ctx->hooks, e) {
|
||||
+ struct uci_hook *h = uci_to_hook(e);
|
||||
+ if (h->ops == ops) {
|
||||
+ uci_list_del(&e->list);
|
||||
+ return 0;
|
||||
+ }
|
||||
+ }
|
||||
+ return UCI_ERR_NOTFOUND;
|
||||
+}
|
||||
+
|
||||
+int uci_load_plugin(struct uci_context *ctx, const char *filename)
|
||||
+{
|
||||
+ struct uci_plugin *p;
|
||||
+ const struct uci_plugin_ops *ops;
|
||||
+ void *dlh;
|
||||
+
|
||||
+ UCI_HANDLE_ERR(ctx);
|
||||
+ dlh = dlopen(filename, RTLD_GLOBAL|RTLD_NOW);
|
||||
+ if (!dlh)
|
||||
+ UCI_THROW(ctx, UCI_ERR_NOTFOUND);
|
||||
+
|
||||
+ ops = dlsym(dlh, "uci_plugin");
|
||||
+ if (!ops || !ops->attach || (ops->attach(ctx) != 0)) {
|
||||
+ if (!ops)
|
||||
+ fprintf(stderr, "No ops\n");
|
||||
+ else if (!ops->attach)
|
||||
+ fprintf(stderr, "No attach\n");
|
||||
+ else
|
||||
+ fprintf(stderr, "Other weirdness\n");
|
||||
+ dlclose(dlh);
|
||||
+ UCI_THROW(ctx, UCI_ERR_INVAL);
|
||||
+ }
|
||||
+
|
||||
+ p = uci_alloc_element(ctx, plugin, filename, 0);
|
||||
+ p->dlh = dlh;
|
||||
+ p->ops = ops;
|
||||
+ uci_list_add(&ctx->plugins, &p->e.list);
|
||||
+
|
||||
+ return 0;
|
||||
+}
|
||||
+
|
||||
+static void uci_unload_plugin(struct uci_context *ctx, struct uci_plugin *p)
|
||||
+{
|
||||
+ if (p->ops->detach)
|
||||
+ p->ops->detach(ctx);
|
||||
+ dlclose(p->dlh);
|
||||
+ uci_free_element(&p->e);
|
||||
+}
|
||||
+
|
||||
+int uci_load_plugins(struct uci_context *ctx, const char *pattern)
|
||||
+{
|
||||
+ glob_t gl;
|
||||
+ int i;
|
||||
+
|
||||
+ if (!pattern)
|
||||
+ pattern = UCI_PREFIX "/lib/uci_*.so";
|
||||
+
|
||||
+ memset(&gl, 0, sizeof(gl));
|
||||
+ glob(pattern, 0, NULL, &gl);
|
||||
+ for (i = 0; i < gl.gl_pathc; i++)
|
||||
+ uci_load_plugin(ctx, gl.gl_pathv[i]);
|
||||
+
|
||||
+ return 0;
|
||||
+}
|
||||
--- a/uci.h
|
||||
+++ b/uci.h
|
||||
@@ -56,6 +56,8 @@ struct uci_list
|
||||
};
|
||||
|
||||
struct uci_ptr;
|
||||
+struct uci_plugin;
|
||||
+struct uci_hook_ops;
|
||||
struct uci_element;
|
||||
struct uci_package;
|
||||
struct uci_section;
|
||||
@@ -275,6 +277,43 @@ extern int uci_set_backend(struct uci_co
|
||||
*/
|
||||
extern bool uci_validate_text(const char *str);
|
||||
|
||||
+
|
||||
+/**
|
||||
+ * uci_add_hook: add a uci hook
|
||||
+ * @ctx: uci context
|
||||
+ * @ops: uci hook ops
|
||||
+ *
|
||||
+ * NB: allocated and freed by the caller
|
||||
+ */
|
||||
+extern int uci_add_hook(struct uci_context *ctx, const struct uci_hook_ops *ops);
|
||||
+
|
||||
+/**
|
||||
+ * uci_remove_hook: remove a uci hook
|
||||
+ * @ctx: uci context
|
||||
+ * @ops: uci hook ops
|
||||
+ */
|
||||
+extern int uci_remove_hook(struct uci_context *ctx, const struct uci_hook_ops *ops);
|
||||
+
|
||||
+/**
|
||||
+ * uci_load_plugin: load an uci plugin
|
||||
+ * @ctx: uci context
|
||||
+ * @filename: path to the uci plugin
|
||||
+ *
|
||||
+ * NB: plugin will be unloaded automatically when the context is freed
|
||||
+ */
|
||||
+int uci_load_plugin(struct uci_context *ctx, const char *filename);
|
||||
+
|
||||
+/**
|
||||
+ * uci_load_plugins: load all uci plugins from a directory
|
||||
+ * @ctx: uci context
|
||||
+ * @pattern: pattern of uci plugin files (optional)
|
||||
+ *
|
||||
+ * if pattern is NULL, then uci_load_plugins will call uci_load_plugin
|
||||
+ * for uci_*.so in <prefix>/lib/
|
||||
+ */
|
||||
+int uci_load_plugins(struct uci_context *ctx, const char *pattern);
|
||||
+
|
||||
+
|
||||
/* UCI data structures */
|
||||
enum uci_type {
|
||||
UCI_TYPE_UNSPEC = 0,
|
||||
@@ -285,6 +324,8 @@ enum uci_type {
|
||||
UCI_TYPE_PATH = 5,
|
||||
UCI_TYPE_BACKEND = 6,
|
||||
UCI_TYPE_ITEM = 7,
|
||||
+ UCI_TYPE_HOOK = 8,
|
||||
+ UCI_TYPE_PLUGIN = 9,
|
||||
};
|
||||
|
||||
enum uci_option_type {
|
||||
@@ -346,6 +387,9 @@ struct uci_context
|
||||
bool internal, nested;
|
||||
char *buf;
|
||||
int bufsz;
|
||||
+
|
||||
+ struct uci_list hooks;
|
||||
+ struct uci_list plugins;
|
||||
};
|
||||
|
||||
struct uci_package
|
||||
@@ -420,6 +464,31 @@ struct uci_ptr
|
||||
const char *value;
|
||||
};
|
||||
|
||||
+struct uci_hook_ops
|
||||
+{
|
||||
+ void (*load)(const struct uci_hook_ops *ops, struct uci_package *p);
|
||||
+ void (*set)(const struct uci_hook_ops *ops, struct uci_package *p, struct uci_history *e);
|
||||
+};
|
||||
+
|
||||
+struct uci_hook
|
||||
+{
|
||||
+ struct uci_element e;
|
||||
+ const struct uci_hook_ops *ops;
|
||||
+};
|
||||
+
|
||||
+struct uci_plugin_ops
|
||||
+{
|
||||
+ int (*attach)(struct uci_context *ctx);
|
||||
+ void (*detach)(struct uci_context *ctx);
|
||||
+};
|
||||
+
|
||||
+struct uci_plugin
|
||||
+{
|
||||
+ struct uci_element e;
|
||||
+ const struct uci_plugin_ops *ops;
|
||||
+ void *dlh;
|
||||
+};
|
||||
+
|
||||
|
||||
/* linked list handling */
|
||||
#ifndef offsetof
|
||||
@@ -490,6 +559,8 @@ struct uci_ptr
|
||||
#define uci_type_package UCI_TYPE_PACKAGE
|
||||
#define uci_type_section UCI_TYPE_SECTION
|
||||
#define uci_type_option UCI_TYPE_OPTION
|
||||
+#define uci_type_hook UCI_TYPE_HOOK
|
||||
+#define uci_type_plugin UCI_TYPE_PLUGIN
|
||||
|
||||
/* element typecasting */
|
||||
#ifdef UCI_DEBUG_TYPECAST
|
||||
@@ -499,6 +570,8 @@ static const char *uci_typestr[] = {
|
||||
[uci_type_package] = "package",
|
||||
[uci_type_section] = "section",
|
||||
[uci_type_option] = "option",
|
||||
+ [uci_type_hook] = "hook",
|
||||
+ [uci_type_plugin] = "plugin",
|
||||
};
|
||||
|
||||
static void uci_typecast_error(int from, int to)
|
||||
@@ -520,6 +593,8 @@ BUILD_CAST(history)
|
||||
BUILD_CAST(package)
|
||||
BUILD_CAST(section)
|
||||
BUILD_CAST(option)
|
||||
+BUILD_CAST(hook)
|
||||
+BUILD_CAST(plugin)
|
||||
|
||||
#else
|
||||
#define uci_to_backend(ptr) container_of(ptr, struct uci_backend, e)
|
||||
@@ -527,6 +602,8 @@ BUILD_CAST(option)
|
||||
#define uci_to_package(ptr) container_of(ptr, struct uci_package, e)
|
||||
#define uci_to_section(ptr) container_of(ptr, struct uci_section, e)
|
||||
#define uci_to_option(ptr) container_of(ptr, struct uci_option, e)
|
||||
+#define uci_to_hook(ptr) container_of(ptr, struct uci_hook, e)
|
||||
+#define uci_to_plugin(ptr) container_of(ptr, struct uci_plugin, e)
|
||||
#endif
|
||||
|
||||
/**
|
||||
--- a/lua/uci.c
|
||||
+++ b/lua/uci.c
|
||||
@@ -765,6 +765,20 @@ uci_lua_add_history(lua_State *L)
|
||||
}
|
||||
|
||||
static int
|
||||
+uci_lua_load_plugins(lua_State *L)
|
||||
+{
|
||||
+ struct uci_context *ctx;
|
||||
+ int ret, offset = 0;
|
||||
+ const char *str = NULL;
|
||||
+
|
||||
+ ctx = find_context(L, &offset);
|
||||
+ if (lua_isstring(L, -1))
|
||||
+ str = lua_tostring(L, -1);
|
||||
+ ret = uci_load_plugins(ctx, str);
|
||||
+ return uci_push_status(L, ctx, false);
|
||||
+}
|
||||
+
|
||||
+static int
|
||||
uci_lua_set_savedir(lua_State *L)
|
||||
{
|
||||
struct uci_context *ctx;
|
||||
@@ -831,6 +845,7 @@ static const luaL_Reg uci[] = {
|
||||
{ "changes", uci_lua_changes },
|
||||
{ "foreach", uci_lua_foreach },
|
||||
{ "add_history", uci_lua_add_history },
|
||||
+ { "load_plugins", uci_lua_load_plugins },
|
||||
{ "get_confdir", uci_lua_get_confdir },
|
||||
{ "set_confdir", uci_lua_set_confdir },
|
||||
{ "get_savedir", uci_lua_get_savedir },
|
|
@ -0,0 +1,182 @@
|
|||
--- /dev/null
|
||||
+++ b/trigger/Makefile
|
||||
@@ -0,0 +1,44 @@
|
||||
+include ../Makefile.inc
|
||||
+LUA_VERSION=5.1
|
||||
+PREFIX_SEARCH=/usr /usr/local /opt/local
|
||||
+LUA_PLUGINDIR=$(firstword \
|
||||
+ $(foreach ldir,$(subst ;, ,$(shell lua -e 'print(package.cpath)')), \
|
||||
+ $(if $(findstring lib/lua/,$(ldir)),$(patsubst %/?.so,%,$(ldir))) \
|
||||
+ ) \
|
||||
+)
|
||||
+
|
||||
+# find lua prefix
|
||||
+LUA_PREFIX=$(firstword \
|
||||
+ $(foreach prefix,$(PREFIX_SEARCH),\
|
||||
+ $(if $(wildcard $(prefix)/include/lua.h),$(prefix)) \
|
||||
+ ) \
|
||||
+)
|
||||
+
|
||||
+libdir=$(prefix)/libs
|
||||
+luadir=$(if $(LUA_PLUGINDIR),$(LUA_PLUGINDIR),$(libdir)/lua/$(LUA_VERSION))
|
||||
+luainc=$(shell pkg-config --silence-errors --cflags lua$(LUA_VERSION))
|
||||
+
|
||||
+CPPFLAGS=-I.. $(if $(luainc),$(luainc), -I$(LUA_PREFIX)/include)
|
||||
+LIBS=-L.. -luci $(shell pkg-config --silence-errors --libs lua$(LUA_VERSION))
|
||||
+
|
||||
+PLUGIN_LD=$(CC)
|
||||
+ifeq ($(OS),Darwin)
|
||||
+ PLUGIN_LDFLAGS=-bundle
|
||||
+else
|
||||
+ PLUGIN_LDFLAGS=-shared -Wl,-soname,$(SHLIB_FILE)
|
||||
+endif
|
||||
+
|
||||
+all: uci_trigger.so
|
||||
+
|
||||
+uci_trigger.so: uci_trigger.o
|
||||
+ $(PLUGIN_LD) $(PLUGIN_LDFLAGS) -o $@ $^ $(LIBS)
|
||||
+
|
||||
+%.o: %.c
|
||||
+ $(CC) $(CPPFLAGS) $(CFLAGS) $(FPIC) -c -o $@ $<
|
||||
+
|
||||
+install:
|
||||
+ mkdir -p $(DESTDIR)$(luadir)
|
||||
+ $(INSTALL) -m0644 uci_trigger.so $(DESTDIR)$(luadir)/
|
||||
+
|
||||
+clean:
|
||||
+ rm -f *.so *.o uci_trigger.so
|
||||
--- /dev/null
|
||||
+++ b/trigger/uci_trigger.c
|
||||
@@ -0,0 +1,132 @@
|
||||
+#include <sys/types.h>
|
||||
+#include <sys/time.h>
|
||||
+#include <stdbool.h>
|
||||
+#include <string.h>
|
||||
+#include <stdio.h>
|
||||
+#include <lualib.h>
|
||||
+#include <lauxlib.h>
|
||||
+#include "uci.h"
|
||||
+
|
||||
+// #define DEBUG
|
||||
+
|
||||
+static int refcount = 0;
|
||||
+static lua_State *gL = NULL;
|
||||
+static bool prepared = false;
|
||||
+
|
||||
+struct trigger_set_op {
|
||||
+ struct uci_package *p;
|
||||
+ struct uci_history *h;
|
||||
+};
|
||||
+
|
||||
+static int trigger_set(lua_State *L)
|
||||
+{
|
||||
+ struct trigger_set_op *so =
|
||||
+ (struct trigger_set_op *)lua_touserdata(L, 1);
|
||||
+ struct uci_package *p = so->p;
|
||||
+ struct uci_history *h = so->h;
|
||||
+ struct uci_context *ctx = p->ctx;
|
||||
+
|
||||
+ /* ignore non-standard savedirs/configdirs
|
||||
+ * in order to not trigger events on uci state changes */
|
||||
+ if (strcmp(ctx->savedir, UCI_SAVEDIR) ||
|
||||
+ strcmp(ctx->confdir, UCI_CONFDIR))
|
||||
+ return 0;
|
||||
+
|
||||
+ if (!prepared) {
|
||||
+ lua_getglobal(L, "require");
|
||||
+ lua_pushstring(L, "uci.trigger");
|
||||
+ lua_call(L, 1, 0);
|
||||
+
|
||||
+ lua_getglobal(L, "uci");
|
||||
+ lua_getfield(L, -1, "trigger");
|
||||
+ lua_getfield(L, -1, "load_modules");
|
||||
+ lua_call(L, 0, 0);
|
||||
+ prepared = true;
|
||||
+ } else {
|
||||
+ lua_getglobal(L, "uci");
|
||||
+ lua_getfield(L, -1, "trigger");
|
||||
+ }
|
||||
+
|
||||
+ lua_getfield(L, -1, "set");
|
||||
+ lua_createtable(L, 4, 0);
|
||||
+
|
||||
+ lua_pushstring(L, p->e.name);
|
||||
+ lua_rawseti(L, -2, 1);
|
||||
+ if (h->section) {
|
||||
+ lua_pushstring(L, h->section);
|
||||
+ lua_rawseti(L, -2, 2);
|
||||
+ }
|
||||
+ if (h->e.name) {
|
||||
+ lua_pushstring(L, h->e.name);
|
||||
+ lua_rawseti(L, -2, 3);
|
||||
+ }
|
||||
+ if (h->value) {
|
||||
+ lua_pushstring(L, h->value);
|
||||
+ lua_rawseti(L, -2, 4);
|
||||
+ }
|
||||
+ lua_call(L, 1, 0);
|
||||
+ lua_pop(L, 2);
|
||||
+ return 0;
|
||||
+}
|
||||
+
|
||||
+#ifdef DEBUG
|
||||
+
|
||||
+static int report (lua_State *L, int status) {
|
||||
+ if (status && !lua_isnil(L, -1)) {
|
||||
+ const char *msg = lua_tostring(L, -1);
|
||||
+ if (msg == NULL) msg = "(error object is not a string)";
|
||||
+ fprintf(stderr, "ERROR: %s\n", msg);
|
||||
+ lua_pop(L, 1);
|
||||
+ }
|
||||
+ return status;
|
||||
+}
|
||||
+
|
||||
+#else
|
||||
+
|
||||
+static inline int report(lua_State *L, int status) {
|
||||
+ return status;
|
||||
+}
|
||||
+
|
||||
+#endif
|
||||
+
|
||||
+static void trigger_set_hook(const struct uci_hook_ops *ops, struct uci_package *p, struct uci_history *h)
|
||||
+{
|
||||
+ struct trigger_set_op so;
|
||||
+
|
||||
+ so.p = p;
|
||||
+ so.h = h;
|
||||
+ report(gL, lua_cpcall(gL, &trigger_set, &so));
|
||||
+}
|
||||
+
|
||||
+static struct uci_hook_ops hook = {
|
||||
+ .set = trigger_set_hook,
|
||||
+};
|
||||
+
|
||||
+static int trigger_attach(struct uci_context *ctx)
|
||||
+{
|
||||
+ if (!gL) {
|
||||
+ gL = luaL_newstate();
|
||||
+ if (!gL)
|
||||
+ return -1;
|
||||
+ luaL_openlibs(gL);
|
||||
+
|
||||
+ refcount++;
|
||||
+ }
|
||||
+ uci_add_hook(ctx, &hook);
|
||||
+ return 0;
|
||||
+}
|
||||
+
|
||||
+static void trigger_detach(struct uci_context *ctx)
|
||||
+{
|
||||
+ if (gL && (--refcount <= 0)) {
|
||||
+ lua_close(gL);
|
||||
+ gL = NULL;
|
||||
+ refcount = 0;
|
||||
+ prepared = false;
|
||||
+ }
|
||||
+}
|
||||
+
|
||||
+struct uci_plugin_ops uci_plugin = {
|
||||
+ .attach = trigger_attach,
|
||||
+ .detach = trigger_detach,
|
||||
+};
|
|
@ -0,0 +1,44 @@
|
|||
#!/usr/bin/lua
|
||||
require("uci")
|
||||
require("uci.trigger")
|
||||
|
||||
function usage()
|
||||
print("Usage: " .. arg[0] .. " [options]")
|
||||
print("Options:")
|
||||
print(" -a: apply the config changes")
|
||||
print(" -t: show matching UCI triggers")
|
||||
print(" -s: show information about tasks to be executed")
|
||||
print(" -r: reset all triggers")
|
||||
print("")
|
||||
end
|
||||
|
||||
if arg[1] == "-s" then
|
||||
local triggers = uci.trigger.get_active()
|
||||
if #triggers > 0 then
|
||||
print("Tasks:")
|
||||
for i, a in ipairs(triggers) do
|
||||
local trigger = a[1]
|
||||
local sections = a[2]
|
||||
print(" - " .. uci.trigger.get_description(trigger, sections))
|
||||
end
|
||||
else
|
||||
print "Nothing to do"
|
||||
end
|
||||
elseif arg[1] == "-t" then
|
||||
local triggers = uci.trigger.get_active()
|
||||
for i, a in ipairs(triggers) do
|
||||
local trigger = a[1]
|
||||
local sections = a[2]
|
||||
if trigger.section_only then
|
||||
print(trigger.id .. " " .. table.concat(" ", sections))
|
||||
else
|
||||
print(trigger.id)
|
||||
end
|
||||
end
|
||||
elseif arg[1] == "-a" then
|
||||
uci.trigger.run()
|
||||
elseif arg[1] == "-r" then
|
||||
uci.trigger.reset_state()
|
||||
else
|
||||
usage()
|
||||
end
|
|
@ -0,0 +1,371 @@
|
|||
module("uci.trigger", package.seeall)
|
||||
require("posix")
|
||||
require("uci")
|
||||
|
||||
local path = "/lib/config/trigger"
|
||||
local triggers = nil
|
||||
local tmp_cursor = nil
|
||||
|
||||
function load_modules()
|
||||
if triggers ~= nil then
|
||||
return
|
||||
end
|
||||
triggers = {
|
||||
list = {},
|
||||
uci = {},
|
||||
active = {}
|
||||
}
|
||||
local modules = posix.glob(path .. "/*.lua")
|
||||
if modules == nil then
|
||||
return
|
||||
end
|
||||
local oldpath = package.path
|
||||
package.path = path .. "/?.lua"
|
||||
for i, v in ipairs(modules) do
|
||||
pcall(require(string.gsub(v, path .. "/(%w+)%.lua$", "%1")))
|
||||
end
|
||||
package.path = oldpath
|
||||
end
|
||||
|
||||
function check_table(table, name)
|
||||
if table[name] == nil then
|
||||
table[name] = {}
|
||||
end
|
||||
return table[name]
|
||||
end
|
||||
|
||||
function get_table_val(val, vtype)
|
||||
if type(val) == (vtype or "string") then
|
||||
return { val }
|
||||
elseif type(val) == "table" then
|
||||
return val
|
||||
end
|
||||
return nil
|
||||
end
|
||||
|
||||
function get_name_list(name)
|
||||
return get_table_val(name or ".all")
|
||||
end
|
||||
|
||||
function add_trigger_option(list, t)
|
||||
local name = get_name_list(t.option)
|
||||
for i, n in ipairs(name) do
|
||||
option = check_table(list, n)
|
||||
table.insert(option, t)
|
||||
end
|
||||
end
|
||||
|
||||
function add_trigger_section(list, t)
|
||||
local name = get_name_list(t.section)
|
||||
for i, n in ipairs(name) do
|
||||
section = check_table(list, n)
|
||||
add_trigger_option(section, t)
|
||||
end
|
||||
end
|
||||
|
||||
function check_insert_triggers(dest, list, tuple)
|
||||
if list == nil then
|
||||
return
|
||||
end
|
||||
for i, t in ipairs(list) do
|
||||
local add = true
|
||||
if type(t.check) == "function" then
|
||||
add = t.check(tuple)
|
||||
end
|
||||
if add then
|
||||
dest[t.id] = t
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function find_section_triggers(tlist, pos, tuple)
|
||||
if pos == nil then
|
||||
return
|
||||
end
|
||||
check_insert_triggers(tlist, pos[".all"], tuple)
|
||||
if tuple.option then
|
||||
check_insert_triggers(tlist, pos[tuple.option], tuple)
|
||||
end
|
||||
end
|
||||
|
||||
function check_recursion(name, seen)
|
||||
if seen == nil then
|
||||
seen = {}
|
||||
end
|
||||
if seen[name] then
|
||||
return nil
|
||||
end
|
||||
seen[name] = true
|
||||
return seen
|
||||
end
|
||||
|
||||
|
||||
function find_recursive_depends(list, name, seen)
|
||||
seen = check_recursion(name, seen)
|
||||
if not seen then
|
||||
return
|
||||
end
|
||||
local bt = get_table_val(triggers.list[name].belongs_to) or {}
|
||||
for i, n in ipairs(bt) do
|
||||
table.insert(list, n)
|
||||
find_recursive_depends(list, n, seen)
|
||||
end
|
||||
end
|
||||
|
||||
function check_trigger_depth(list, name)
|
||||
if name == nil then
|
||||
return
|
||||
end
|
||||
|
||||
local n = list[name]
|
||||
if n == nil then
|
||||
return
|
||||
end
|
||||
|
||||
list[name] = nil
|
||||
return check_trigger_depth(list, n)
|
||||
end
|
||||
|
||||
function find_triggers(tuple)
|
||||
local pos = triggers.uci[tuple.package]
|
||||
if pos == nil then
|
||||
return {}
|
||||
end
|
||||
|
||||
local tlist = {}
|
||||
find_section_triggers(tlist, pos[".all"], tuple)
|
||||
find_section_triggers(tlist, pos[tuple.section[".type"]], tuple)
|
||||
|
||||
for n, t in pairs(tlist) do
|
||||
local dep = {}
|
||||
find_recursive_depends(dep, t.id)
|
||||
for i, depname in ipairs(dep) do
|
||||
check_trigger_depth(tlist, depname)
|
||||
end
|
||||
end
|
||||
|
||||
local nlist = {}
|
||||
for n, t in pairs(tlist) do
|
||||
if t then
|
||||
table.insert(nlist, t)
|
||||
end
|
||||
end
|
||||
|
||||
return nlist
|
||||
end
|
||||
|
||||
function reset_state()
|
||||
assert(io.open("/var/run/uci_trigger", "w")):close()
|
||||
if tctx then
|
||||
tctx:unload("uci_trigger")
|
||||
end
|
||||
end
|
||||
|
||||
function load_state()
|
||||
-- make sure the config file exists before we attempt to load it
|
||||
-- uci doesn't like loading nonexistent config files
|
||||
local f = assert(io.open("/var/run/uci_trigger", "a")):close()
|
||||
|
||||
load_modules()
|
||||
triggers.active = {}
|
||||
if tctx then
|
||||
tctx:unload("uci_trigger")
|
||||
else
|
||||
tctx = uci.cursor()
|
||||
end
|
||||
assert(tctx:load("/var/run/uci_trigger"))
|
||||
tctx:foreach("uci_trigger", "trigger",
|
||||
function(section)
|
||||
trigger = triggers.list[section[".name"]]
|
||||
if trigger == nil then
|
||||
return
|
||||
end
|
||||
|
||||
active = {}
|
||||
triggers.active[trigger.id] = active
|
||||
|
||||
local s = get_table_val(section["sections"]) or {}
|
||||
for i, v in ipairs(s) do
|
||||
active[v] = true
|
||||
end
|
||||
end
|
||||
)
|
||||
end
|
||||
|
||||
function get_names(list)
|
||||
local slist = {}
|
||||
for name, val in pairs(list) do
|
||||
if val then
|
||||
table.insert(slist, name)
|
||||
end
|
||||
end
|
||||
return slist
|
||||
end
|
||||
|
||||
function check_cancel(name, seen)
|
||||
local t = triggers.list[name]
|
||||
local dep = get_table_val(t.belongs_to)
|
||||
seen = check_recursion(name, seen)
|
||||
|
||||
if not t or not dep or not seen then
|
||||
return false
|
||||
end
|
||||
|
||||
for i, v in ipairs(dep) do
|
||||
-- only cancel triggers for all sections
|
||||
-- if both the current and the parent trigger
|
||||
-- are per-section
|
||||
local section_only = false
|
||||
if t.section_only then
|
||||
local tdep = triggers.list[v]
|
||||
if tdep then
|
||||
section_only = tdep.section_only
|
||||
end
|
||||
end
|
||||
|
||||
if check_cancel(v, seen) then
|
||||
return true
|
||||
end
|
||||
if triggers.active[v] then
|
||||
if section_only then
|
||||
for n, active in pairs(triggers.active[v]) do
|
||||
triggers.active[name][n] = false
|
||||
end
|
||||
else
|
||||
return true
|
||||
end
|
||||
end
|
||||
end
|
||||
return false
|
||||
end
|
||||
|
||||
-- trigger api functions
|
||||
|
||||
function add(ts)
|
||||
for i,t in ipairs(ts) do
|
||||
triggers.list[t.id] = t
|
||||
match = {}
|
||||
if t.package then
|
||||
local package = check_table(triggers.uci, t.package)
|
||||
add_trigger_section(package, t)
|
||||
triggers.list[t.id] = t
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function set(data, cursor)
|
||||
assert(data ~= nil)
|
||||
if cursor == nil then
|
||||
cursor = tmp_cursor or uci.cursor()
|
||||
tmp_cursor = uci.cursor
|
||||
end
|
||||
|
||||
local tuple = {
|
||||
package = data[1],
|
||||
section = data[2],
|
||||
option = data[3],
|
||||
value = data[4]
|
||||
}
|
||||
assert(cursor:load(tuple.package))
|
||||
|
||||
load_state()
|
||||
local section = cursor:get_all(tuple.package, tuple.section)
|
||||
if (section == nil) then
|
||||
if option ~= nil then
|
||||
return
|
||||
end
|
||||
section = {
|
||||
[".type"] = value
|
||||
}
|
||||
if tuple.section == nil then
|
||||
tuple.section = ""
|
||||
section[".anonymous"] = true
|
||||
end
|
||||
section[".name"] = tuple.section
|
||||
end
|
||||
tuple.section = section
|
||||
|
||||
local ts = find_triggers(tuple)
|
||||
for i, t in ipairs(ts) do
|
||||
local active = triggers.active[t.id]
|
||||
if not active then
|
||||
active = {}
|
||||
triggers.active[t.id] = active
|
||||
tctx:set("uci_trigger", t.id, "trigger")
|
||||
end
|
||||
if section[".name"] then
|
||||
active[section[".name"]] = true
|
||||
end
|
||||
local slist = get_names(triggers.active[t.id])
|
||||
if #slist > 0 then
|
||||
tctx:set("uci_trigger", t.id, "sections", slist)
|
||||
end
|
||||
end
|
||||
tctx:save("uci_trigger")
|
||||
end
|
||||
|
||||
function get_description(trigger, sections)
|
||||
if not trigger.title then
|
||||
return trigger.id
|
||||
end
|
||||
local desc = trigger.title
|
||||
if trigger.section_only and sections and #sections > 0 then
|
||||
desc = desc .. " (" .. table.concat(sections, ", ") .. ")"
|
||||
end
|
||||
return desc
|
||||
end
|
||||
|
||||
function get_active()
|
||||
local slist = {}
|
||||
|
||||
if triggers == nil then
|
||||
load_state()
|
||||
end
|
||||
for name, val in pairs(triggers.active) do
|
||||
if val and not check_cancel(name) then
|
||||
local sections = {}
|
||||
for name, active in pairs(triggers.active[name]) do
|
||||
if active then
|
||||
table.insert(sections, name)
|
||||
end
|
||||
end
|
||||
table.insert(slist, { triggers.list[name], sections })
|
||||
end
|
||||
end
|
||||
return slist
|
||||
end
|
||||
|
||||
function run(ts)
|
||||
if ts == nil then
|
||||
ts = get_active()
|
||||
end
|
||||
for i, t in ipairs(ts) do
|
||||
local trigger = t[1]
|
||||
local sections = t[2]
|
||||
local actions = get_table_val(trigger.action, "function") or {}
|
||||
for ai, a in ipairs(actions) do
|
||||
if not trigger.section_only then
|
||||
sections = { "" }
|
||||
end
|
||||
for si, s in ipairs(sections) do
|
||||
if a(s) then
|
||||
tctx:delete("uci_trigger", trigger.id)
|
||||
tctx:save("uci_trigger")
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
-- helper functions
|
||||
|
||||
function system_command(arg)
|
||||
local cmd = arg
|
||||
return function(arg)
|
||||
return os.execute(cmd:format(arg)) == 0
|
||||
end
|
||||
end
|
||||
|
||||
function service_restart(arg)
|
||||
return system_command("/etc/init.d/" .. arg .. " restart")
|
||||
end
|
|
@ -0,0 +1,63 @@
|
|||
module("trigger.base", package.seeall)
|
||||
require("uci.trigger")
|
||||
|
||||
uci.trigger.add {
|
||||
{
|
||||
id = "dnsmasq_restart",
|
||||
title = "Restart dnsmasq",
|
||||
package = "dhcp",
|
||||
action = uci.trigger.service_restart("dnsmasq"),
|
||||
},
|
||||
{
|
||||
id = "dropbear_restart",
|
||||
title = "Restart dropbear",
|
||||
package = "dropbear",
|
||||
action = uci.trigger.service_restart("dropbear"),
|
||||
},
|
||||
{
|
||||
id = "fstab_restart",
|
||||
title = "Remount filesystems",
|
||||
package = "fstab",
|
||||
action = uci.trigger.service_restart("fstab"),
|
||||
},
|
||||
{
|
||||
id = "firewall_restart",
|
||||
title = "Reload firewall rules",
|
||||
package = "firewall",
|
||||
action = uci.trigger.service_restart("firewall"),
|
||||
},
|
||||
{
|
||||
id = "httpd_restart",
|
||||
title = "Restart the http server",
|
||||
package = "httpd",
|
||||
action = uci.trigger.service_restart("httpd")
|
||||
},
|
||||
{
|
||||
id = "led_restart",
|
||||
title = "Reload LED settings",
|
||||
package = "system",
|
||||
section = "led",
|
||||
action = uci.trigger.service_restart("led")
|
||||
},
|
||||
{
|
||||
id = "network_restart",
|
||||
title = "Restart networking and wireless",
|
||||
package = "network",
|
||||
action = uci.trigger.service_restart("network")
|
||||
},
|
||||
{
|
||||
id = "qos_restart",
|
||||
title = "Reload Quality of Service rules",
|
||||
package = "qos",
|
||||
action = uci.trigger.service_restart("qos"),
|
||||
},
|
||||
{
|
||||
id = "wireless_restart",
|
||||
title = "Restart all wireless interfaces",
|
||||
package = "wireless",
|
||||
section = { "wifi-device", "wifi-iface" },
|
||||
action = uci.trigger.system_command("wifi"),
|
||||
belongs_to = "network_restart"
|
||||
},
|
||||
}
|
||||
|
Loading…
Reference in New Issue