diff options
Diffstat (limited to '')
| -rw-r--r-- | src/rspamadm/commands.c | 280 |
1 files changed, 280 insertions, 0 deletions
diff --git a/src/rspamadm/commands.c b/src/rspamadm/commands.c new file mode 100644 index 0000000..d64b172 --- /dev/null +++ b/src/rspamadm/commands.c @@ -0,0 +1,280 @@ +/*- + * Copyright 2016 Vsevolod Stakhov + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +#include "rspamadm.h" +#include "libutil/util.h" +#include "libserver/logger.h" +#include "lua/lua_common.h" +#include "lua/lua_thread_pool.h" + +extern struct rspamadm_command pw_command; +extern struct rspamadm_command configtest_command; +extern struct rspamadm_command configdump_command; +extern struct rspamadm_command control_command; +extern struct rspamadm_command confighelp_command; +extern struct rspamadm_command statconvert_command; +extern struct rspamadm_command fuzzyconvert_command; +extern struct rspamadm_command signtool_command; +extern struct rspamadm_command lua_command; + +const struct rspamadm_command *commands[] = { + &help_command, + &pw_command, + &configtest_command, + &configdump_command, + &control_command, + &confighelp_command, + &statconvert_command, + &fuzzyconvert_command, + &signtool_command, + &lua_command, + NULL}; + + +const struct rspamadm_command * +rspamadm_search_command(const gchar *name, GPtrArray *all_commands) +{ + const struct rspamadm_command *ret = NULL, *cmd; + const gchar *alias; + guint i, j; + + if (name == NULL) { + name = "help"; + } + + PTR_ARRAY_FOREACH(all_commands, i, cmd) + { + if (strcmp(name, cmd->name) == 0) { + ret = cmd; + break; + } + + PTR_ARRAY_FOREACH(cmd->aliases, j, alias) + { + if (strcmp(name, alias) == 0) { + ret = cmd; + break; + } + } + } + + return ret; +} + +void rspamadm_fill_internal_commands(GPtrArray *dest) +{ + guint i; + + for (i = 0; i < G_N_ELEMENTS(commands); i++) { + if (commands[i]) { + g_ptr_array_add(dest, (gpointer) commands[i]); + } + } +} + +static void +lua_thread_str_error_cb(struct thread_entry *thread, int ret, const char *msg) +{ + msg_err("call to rspamadm lua script failed (%d): %s", + ret, msg); +} + +static void +rspamadm_lua_command_run(gint argc, gchar **argv, + const struct rspamadm_command *cmd) +{ + struct thread_entry *thread = lua_thread_pool_get_for_config(rspamd_main->cfg); + + lua_State *L = thread->lua_state; + + gint table_idx = GPOINTER_TO_INT(cmd->command_data); + gint i; + + /* Function */ + lua_rawgeti(L, LUA_REGISTRYINDEX, table_idx); + lua_pushstring(L, "handler"); + lua_gettable(L, -2); + + /* Args */ + lua_createtable(L, argc + 1, 0); + + for (i = 0; i < argc; i++) { + lua_pushstring(L, argv[i]); + lua_rawseti(L, -2, i); /* Starting from zero ! */ + } + + if (lua_repl_thread_call(thread, 1, (void *) cmd, lua_thread_str_error_cb) != 0) { + exit(EXIT_FAILURE); + } + + lua_settop(L, 0); +} + +static const gchar * +rspamadm_lua_command_help(gboolean full_help, + const struct rspamadm_command *cmd) +{ + gint table_idx = GPOINTER_TO_INT(cmd->command_data); + + if (full_help) { + struct thread_entry *thread = lua_thread_pool_get_for_config(rspamd_main->cfg); + + lua_State *L = thread->lua_state; + lua_rawgeti(L, LUA_REGISTRYINDEX, table_idx); + /* Function */ + lua_pushstring(L, "handler"); + lua_gettable(L, -2); + + /* Args */ + lua_createtable(L, 2, 0); + lua_pushstring(L, cmd->name); + lua_rawseti(L, -2, 0); /* Starting from zero ! */ + + lua_pushstring(L, "--help"); + lua_rawseti(L, -2, 1); + + if (lua_repl_thread_call(thread, 1, (void *) cmd, lua_thread_str_error_cb) != 0) { + exit(EXIT_FAILURE); + } + + lua_settop(L, 0); + } + else { + lua_State *L = rspamd_main->cfg->lua_state; + lua_rawgeti(L, LUA_REGISTRYINDEX, table_idx); + lua_pushstring(L, "description"); + lua_gettable(L, -2); + + if (lua_isstring(L, -1)) { + printf(" %-18s %-60s\n", cmd->name, lua_tostring(L, -1)); + } + else { + printf(" %-18s %-60s\n", cmd->name, "no description available"); + } + + lua_settop(L, 0); + } + + return NULL; /* Must be handled in rspamadm itself */ +} + +void rspamadm_fill_lua_commands(lua_State *L, GPtrArray *dest) +{ + gint i; + + GPtrArray *lua_paths; + GError *err = NULL; + const gchar *lualibdir = RSPAMD_LUALIBDIR, *path; + struct rspamadm_command *lua_cmd; + gchar search_dir[PATH_MAX]; + + if (g_hash_table_lookup(ucl_vars, "LUALIBDIR")) { + lualibdir = g_hash_table_lookup(ucl_vars, "LUALIBDIR"); + } + + rspamd_snprintf(search_dir, sizeof(search_dir), "%s%crspamadm%c", + lualibdir, G_DIR_SEPARATOR, G_DIR_SEPARATOR); + + if ((lua_paths = rspamd_glob_path(search_dir, "*.lua", FALSE, &err)) == NULL) { + msg_err("cannot glob files in %s/*.lua: %e", search_dir, err); + g_error_free(err); + + return; + } + + PTR_ARRAY_FOREACH(lua_paths, i, path) + { + if (luaL_dofile(L, path) != 0) { + msg_err("cannot execute lua script %s: %s", + path, lua_tostring(L, -1)); + lua_settop(L, 0); + continue; + } + else { + if (lua_type(L, -1) == LUA_TTABLE) { + lua_pushstring(L, "handler"); + lua_gettable(L, -2); + } + else { + continue; /* Something goes wrong, huh */ + } + + if (lua_type(L, -1) != LUA_TFUNCTION) { + msg_err("rspamadm script %s does not have 'handler' field with type " + "function", + path); + continue; + } + + /* Pop handler */ + lua_pop(L, 1); + lua_cmd = g_malloc0(sizeof(*lua_cmd)); + + lua_pushstring(L, "name"); + lua_gettable(L, -2); + + if (lua_type(L, -1) == LUA_TSTRING) { + lua_cmd->name = g_strdup(lua_tostring(L, -1)); + } + else { + goffset ext_pos; + gchar *name; + + name = g_path_get_basename(path); + /* Remove .lua */ + ext_pos = rspamd_substring_search(path, strlen(path), ".lua", 4); + + if (ext_pos != -1) { + name[ext_pos] = '\0'; + } + + lua_cmd->name = name; + } + + lua_pop(L, 1); + + lua_pushstring(L, "aliases"); + lua_gettable(L, -2); + + if (lua_type(L, -1) == LUA_TTABLE) { + lua_cmd->aliases = g_ptr_array_new_full( + rspamd_lua_table_size(L, -1), + g_free); + + for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) { + if (lua_isstring(L, -1)) { + g_ptr_array_add(lua_cmd->aliases, + g_strdup(lua_tostring(L, -1))); + } + } + } + + lua_pop(L, 1); + + lua_pushvalue(L, -1); + /* Reference table itself */ + lua_cmd->command_data = GINT_TO_POINTER(luaL_ref(L, LUA_REGISTRYINDEX)); + lua_cmd->flags |= RSPAMADM_FLAG_LUA | RSPAMADM_FLAG_DYNAMIC; + lua_cmd->run = rspamadm_lua_command_run; + lua_cmd->help = rspamadm_lua_command_help; + + g_ptr_array_add(dest, lua_cmd); + } + + lua_settop(L, 0); + } + + g_ptr_array_free(lua_paths, TRUE); +} |