* Add ability to lookup CDB maps from lua

* Add cdb:// map to multimap plugin

1 files changed, 152 insertions, 0 deletions

diff --git a/src/lua/lua_cdb.c b/src/lua/lua_cdb.c
new file mode 100644
index 000000000..ef6041d7c
--- /dev/null
+++ b/src/lua/lua_cdb.c
@@ -0,0 +1,152 @@
+/* Copyright (c) 2010, Vsevolod Stakhov
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *       * Redistributions of source code must retain the above copyright
+ *         notice, this list of conditions and the following disclaimer.
+ *       * Redistributions in binary form must reproduce the above copyright
+ *         notice, this list of conditions and the following disclaimer in the
+ *         documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED ''AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL Rambler BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "lua_common.h"
+#include "../cdb/cdb.h"
+
+LUA_FUNCTION_DEF (cdb, create);
+LUA_FUNCTION_DEF (cdb, lookup);
+LUA_FUNCTION_DEF (cdb, get_name);
+LUA_FUNCTION_DEF (cdb, destroy);
+
+static const struct luaL_reg    cdblib_m[] = {
+	LUA_INTERFACE_DEF (cdb, lookup),
+	LUA_INTERFACE_DEF (cdb, get_name),
+	{"__tostring", lua_class_tostring},
+	{"__gc", lua_cdb_destroy},
+	{NULL, NULL}
+};
+static const struct luaL_reg    cdblib_f[] = {
+	LUA_INTERFACE_DEF (cdb, create),
+	{NULL, NULL}
+};
+
+static struct cdb	*
+lua_check_cdb (lua_State * L)
+{
+	void                           *ud = luaL_checkudata (L, 1, "rspamd{cdb}");
+
+	luaL_argcheck (L, ud != NULL, 1, "'cdb' expected");
+	return *((struct cdb **)ud);
+}
+
+static gint
+lua_cdb_create (lua_State *L)
+{
+	struct cdb                     *cdb, **pcdb;
+	const gchar                    *filename;
+	gint                            fd;
+
+	filename = luaL_checkstring (L, 1);
+	/* If file begins with cdb://, just skip it */
+	if (g_ascii_strncasecmp (filename, "cdb://", sizeof ("cdb://") - 1) == 0) {
+		filename += sizeof ("cdb://") - 1;
+	}
+
+	if ((fd = open (filename, O_RDONLY)) == -1) {
+		msg_warn ("cannot open cdb: %s, %s", filename, strerror (errno));
+		lua_pushnil (L);
+	}
+	else {
+		cdb = g_malloc (sizeof (struct cdb));
+		cdb->filename = g_strdup (filename);
+		if (cdb_init (cdb, fd) == -1) {
+			msg_warn ("cannot open cdb: %s, %s", filename, strerror (errno));
+			lua_pushnil (L);
+		}
+		else {
+			pcdb = lua_newuserdata (L, sizeof (struct cdb*));
+			lua_setclass (L, "rspamd{cdb}", -1);
+			*pcdb = cdb;
+		}
+	}
+
+	return 1;
+}
+
+static gint
+lua_cdb_get_name (lua_State *L)
+{
+	struct cdb                     *cdb = lua_check_cdb (L);
+
+	lua_pushstring (L, cdb->filename);
+	return 1;
+}
+
+static gint
+lua_cdb_lookup (lua_State *L)
+{
+	struct cdb                     *cdb = lua_check_cdb (L);
+	const gchar                    *what;
+	gchar                          *value;
+	gsize                           vlen;
+	goffset                         vpos;
+
+	what = luaL_checkstring (L, 2);
+	if (cdb_find (cdb, what, strlen (what)) > 0) {
+		/* Extract and push value to lua as string */
+		vpos = cdb_datapos (cdb);
+		vlen = cdb_datalen (cdb);
+		value = g_malloc (vlen);
+		cdb_read (cdb, value, vlen, vpos);
+		lua_pushlstring (L, value, vlen);
+		g_free (value);
+	}
+	else {
+		lua_pushnil (L);
+	}
+
+	return 1;
+}
+
+static gint
+lua_cdb_destroy (lua_State *L)
+{
+	struct cdb                     *cdb = lua_check_cdb (L);
+
+	if (cdb) {
+		cdb_free (cdb);
+		g_free (cdb->filename);
+		g_free (cdb);
+	}
+
+	return 0;
+}
+
+gint
+luaopen_cdb (lua_State * L)
+{
+	luaL_newmetatable (L, "rspamd{cdb}");
+	lua_pushstring (L, "__index");
+	lua_pushvalue (L, -2);
+	lua_settable (L, -3);
+
+	lua_pushstring (L, "class");
+	lua_pushstring (L, "rspamd{cdb}");
+	lua_rawset (L, -3);
+
+	luaL_openlib (L, NULL, cdblib_m, 0);
+	luaL_openlib(L, "cdb", cdblib_f, 0);
+
+	return 1;
+}