Set lua path according to rspamd settings.

diff --git a/src/lua/lua_common.c b/src/lua/lua_common.c
index 8df878585691dd693b7aad34084f5e2759f1e04c..0f7c79ce7d35af75efbb41ad7d19d5c6308da72b 100644 (file)
--- a/src/lua/lua_common.c
+++ b/src/lua/lua_common.c
@@ -281,6 +281,40 @@ lua_add_actions_global (lua_State *L)
        lua_setglobal (L, "rspamd_actions");
 }
 
+static void
+rspamd_lua_set_path (lua_State *L, struct rspamd_config *cfg)
+{
+       const gchar *old_path, *additional_path = NULL;
+       const ucl_object_t *opts;
+       gchar path_buf[PATH_MAX];
+
+       lua_getglobal (L, "package");
+       lua_getfield (L, -1, "path");
+       old_path = luaL_checkstring (L, -1);
+
+       opts = ucl_object_find_key (cfg->rcl_obj, "options");
+       if (opts != NULL) {
+               opts = ucl_object_find_key (opts, "lua_path");
+               if (opts != NULL && ucl_object_type (opts) == UCL_STRING) {
+                       additional_path = ucl_object_tostring (opts);
+               }
+       }
+
+       if (additional_path) {
+               rspamd_snprintf (path_buf, sizeof (path_buf), "%s;%s/?.lua;%s/?.lua;%s",
+                               old_path, RSPAMD_PLUGINSDIR, RSPAMD_CONFDIR, additional_path);
+       }
+       else {
+               rspamd_snprintf (path_buf, sizeof (path_buf), "%s;%s/?.lua;%s/?.lua",
+                               old_path, RSPAMD_PLUGINSDIR, RSPAMD_CONFDIR);
+       }
+
+       lua_pop (L, 1);
+       lua_pushstring (L, path_buf);
+       lua_setfield (L, -2, "path");
+       lua_pop (L, 1);
+}
+
 lua_State *
 rspamd_lua_init (struct rspamd_config *cfg)
 {
@@ -317,6 +351,8 @@ rspamd_lua_init (struct rspamd_config *cfg)
 
        rspamd_lua_add_preload (L, "ucl", luaopen_ucl);
 
+       rspamd_lua_set_path (L, cfg);
+
        return L;
 }
 
@@ -361,6 +397,7 @@ rspamd_init_lua_filters (struct rspamd_config *cfg)
        lua_State *L = cfg->lua_state;
 
        cur = g_list_first (cfg->script_modules);
+
        while (cur) {
                module = cur->data;
                if (module->path) {