[Feature] Add the initial version of LUA repl to rspamadm

3 files changed, 291 insertions, 0 deletions

diff --git a/src/rspamadm/CMakeLists.txt b/src/rspamadm/CMakeLists.txt
index 19d78c7bb..65b8669bc 100644
--- a/src/rspamadm/CMakeLists.txt
+++ b/src/rspamadm/CMakeLists.txt
@@ -9,6 +9,7 @@ SET(RSPAMADMSRC rspamadm.c
         confighelp.c
         stat_convert.c
         signtool.c
+        lua_repl.c
         ${CMAKE_BINARY_DIR}/src/workers.c
         ${CMAKE_BINARY_DIR}/src/modules.c
         ${CMAKE_SOURCE_DIR}/src/controller.c
@@ -43,6 +44,7 @@ IF(ENABLE_HIREDIS MATCHES "ON")
 	TARGET_LINK_LIBRARIES(rspamadm rspamd-hiredis)
 ENDIF()
 TARGET_LINK_LIBRARIES(rspamadm rspamd-actrie)
+TARGET_LINK_LIBRARIES(rspamadm rspamd-linenoise)
 
 IF (NOT DEBIAN_BUILD)
     SET_TARGET_PROPERTIES(rspamadm PROPERTIES VERSION ${RSPAMD_VERSION})
diff --git a/src/rspamadm/commands.c b/src/rspamadm/commands.c
index 0e24eaba6..0481e498b 100644
--- a/src/rspamadm/commands.c
+++ b/src/rspamadm/commands.c
@@ -24,6 +24,7 @@ extern struct rspamadm_command control_command;
 extern struct rspamadm_command confighelp_command;
 extern struct rspamadm_command statconvert_command;
 extern struct rspamadm_command signtool_command;
+extern struct rspamadm_command lua_command;
 
 const struct rspamadm_command *commands[] = {
 	&help_command,
@@ -36,6 +37,7 @@ const struct rspamadm_command *commands[] = {
 	&confighelp_command,
 	&statconvert_command,
 	&signtool_command,
+	&lua_command,
 	NULL
 };
 
diff --git a/src/rspamadm/lua_repl.c b/src/rspamadm/lua_repl.c
new file mode 100644
index 000000000..03d871df2
--- /dev/null
+++ b/src/rspamadm/lua_repl.c
@@ -0,0 +1,287 @@
+/*-
+ * 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 "config.h"
+#include "rspamadm.h"
+#include "cryptobox.h"
+#include "printf.h"
+#include "lua/lua_common.h"
+#include "linenoise.h"
+#ifdef WITH_LUAJIT
+#include <luajit.h>
+#endif
+
+static gchar **paths = NULL;
+static gchar **scripts = NULL;
+static gchar *histfile = NULL;
+static guint max_history = 2000;
+
+static const char *default_history_file = ".rspamd_repl.hist";
+
+#ifdef WITH_LUAJIT
+#define MAIN_PROMPT LUAJIT_VERSION "> "
+#else
+#define MAIN_PROMPT LUA_VERSION "> "
+#endif
+#define MULTILINE_PROMPT "... "
+
+static void rspamadm_lua (gint argc, gchar **argv);
+static const char *rspamadm_lua_help (gboolean full_help);
+
+struct rspamadm_command lua_command = {
+		.name = "lua",
+		.flags = 0,
+		.help = rspamadm_lua_help,
+		.run = rspamadm_lua
+};
+
+static GOptionEntry entries[] = {
+		{"script", 's', 0, G_OPTION_ARG_STRING_ARRAY, &scripts,
+				"Load specified scripts", NULL},
+		{"path", 'p', 0, G_OPTION_ARG_STRING_ARRAY, &paths,
+				"Add specified paths to lua paths", NULL},
+		{"history-file", 'H', 0, G_OPTION_ARG_FILENAME, &histfile,
+				"Load history from the specified file", NULL},
+		{"max-history", 'm', 0, G_OPTION_ARG_INT, &max_history,
+				"Store this number of history entries", NULL},
+		{NULL,       0,   0, G_OPTION_ARG_NONE, NULL, NULL, NULL}
+};
+
+static const char *
+rspamadm_lua_help (gboolean full_help)
+{
+	const char *help_str;
+
+	if (full_help) {
+		help_str = "Run lua read/execute/print loop\n\n"
+				"Usage: rspamadm lua [-p paths] [-s scripts]\n"
+				"Where options are:\n\n"
+				"-p: add additional lua paths (may be repeated)\n"
+				"-s: load scripts on start from specified files (may be repeated)\n"
+				"--help: shows available options and commands";
+	}
+	else {
+		help_str = "Run LUA interpreter";
+	}
+
+	return help_str;
+}
+
+static void
+rspamadm_lua_add_path (lua_State *L, const gchar *path)
+{
+	const gchar *old_path;
+	gsize len;
+	GString *new_path;
+
+	lua_getglobal (L, "package");
+	lua_getfield (L, -1, "path");
+	old_path = luaL_checklstring (L, -1, &len);
+
+	new_path = g_string_sized_new (len + strlen (path) + sizeof("/?.lua"));
+
+	if (strstr (path, "?.lua") == NULL) {
+		rspamd_printf_gstring (new_path, "%s/?.lua;%s", path, old_path);
+	}
+	else {
+		rspamd_printf_gstring (new_path, "%s;%s", path, old_path);
+	}
+
+	lua_pushlstring (L, new_path->str, new_path->len);
+	lua_setfield (L, -2, "path");
+	lua_settop (L, 0);
+	g_string_free (new_path, TRUE);
+}
+
+static gboolean
+rspamadm_lua_load_script (lua_State *L, const gchar *path)
+{
+	GString *tb;
+	gint err_idx;
+
+	lua_pushcfunction (L, &rspamd_lua_traceback);
+	err_idx = lua_gettop (L);
+
+	if (luaL_loadfile (L, path) != 0) {
+		rspamd_fprintf (stderr, "cannot load script %s: %s\n",
+				path, strerror (errno));
+		lua_settop (L, 0);
+
+		return FALSE;
+	}
+
+	if (lua_pcall (L, 0, 0, err_idx) != 0) {
+		tb = lua_touserdata (L, -1);
+		rspamd_fprintf (stderr, "call to %s failed: %v", path, tb);
+		g_string_free (tb, TRUE);
+		lua_settop (L, 0);
+
+		return FALSE;
+	}
+
+	lua_settop (L, 0);
+
+	return TRUE;
+}
+
+static void
+rspamadm_exec_input (lua_State *L, const gchar *input)
+{
+	GString *tb;
+	gint err_idx, i;
+	gchar outbuf[8192];
+
+	lua_pushcfunction (L, &rspamd_lua_traceback);
+	err_idx = lua_gettop (L);
+
+	if (luaL_loadstring (L, input) != 0) {
+		rspamd_fprintf (stderr, "cannot load string %s\n",
+				input);
+		lua_settop (L, 0);
+		return;
+	}
+
+	if (lua_pcall (L, 0, LUA_MULTRET, err_idx) != 0) {
+		tb = lua_touserdata (L, -1);
+		rspamd_fprintf (stderr, "call failed: %v", tb);
+		g_string_free (tb, TRUE);
+		lua_settop (L, 0);
+		return;
+	}
+
+	/* Print output */
+	for (i = err_idx; i < lua_gettop (L); i ++) {
+		lua_logger_out_type (L, i, outbuf, sizeof (outbuf));
+		rspamd_printf ("%s\n", outbuf);
+	}
+
+	lua_settop (L, 0);
+}
+
+static void
+rspamadm_lua_run_repl (lua_State *L)
+{
+	gchar *input;
+	gboolean is_multiline = FALSE;
+	GString *tb;
+
+	for (;;) {
+		if (!is_multiline) {
+			input = linenoise (MAIN_PROMPT);
+
+			if (input == NULL) {
+				return;
+			}
+
+			if (strcmp (input, "{{") == 0) {
+				is_multiline = TRUE;
+				linenoiseFree (input);
+				tb = g_string_sized_new (8192);
+				continue;
+			}
+
+			rspamadm_exec_input (L, input);
+			linenoiseFree (input);
+			lua_settop (L, 0);
+		}
+		else {
+			input = linenoise (MULTILINE_PROMPT);
+
+			if (input == NULL) {
+				g_string_free (tb, TRUE);
+				return;
+			}
+
+			if (strcmp (input, "}}") == 0) {
+				is_multiline = FALSE;
+				linenoiseFree (input);
+				rspamadm_exec_input (L, tb->str);
+				g_string_free (tb, TRUE);
+			}
+			else {
+				g_string_append (tb, input);
+				g_string_append (tb, "\n");
+				linenoiseFree (input);
+			}
+		}
+	}
+}
+
+static void
+rspamadm_lua (gint argc, gchar **argv)
+{
+	GOptionContext *context;
+	GError *error = NULL;
+	gchar **elt;
+	lua_State *L;
+
+	context = g_option_context_new ("lua - run lua interpreter");
+	g_option_context_set_summary (context,
+			"Summary:\n  Rspamd administration utility version "
+					RVERSION
+					"\n  Release id: "
+					RID);
+	g_option_context_add_main_entries (context, entries, NULL);
+
+	if (!g_option_context_parse (context, &argc, &argv, &error)) {
+		fprintf (stderr, "option parsing failed: %s\n", error->message);
+		g_error_free (error);
+		exit (1);
+	}
+
+	L = rspamd_lua_init ();
+	rspamd_lua_set_path (L, NULL);
+
+	if (paths) {
+		for (elt = paths; *elt != NULL; elt ++) {
+			rspamadm_lua_add_path (L, *elt);
+		}
+	}
+
+	if (scripts) {
+		for (elt = scripts; *elt != NULL; elt ++) {
+			if (!rspamadm_lua_load_script (L, *elt)) {
+				exit (EXIT_FAILURE);
+			}
+		}
+	}
+
+	if (histfile == NULL) {
+		const gchar *homedir;
+		GString *hist_path;
+
+		homedir = getenv ("HOME");
+
+		if (homedir) {
+			hist_path = g_string_sized_new (strlen (homedir) +
+					strlen (default_history_file) + 1);
+			rspamd_printf_gstring (hist_path, "%s/%s", homedir,
+					default_history_file);
+		}
+		else {
+			hist_path = g_string_sized_new (strlen (default_history_file) + 2);
+			rspamd_printf_gstring (hist_path, "./%s", default_history_file);
+		}
+
+		histfile = hist_path->str;
+		g_string_free (hist_path, FALSE);
+	}
+
+	linenoiseHistorySetMaxLen (max_history);
+	linenoiseHistoryLoad (histfile);
+	rspamadm_lua_run_repl (L);
+	linenoiseHistorySave (histfile);
+}