Save module ctx for custom command.

diff --git a/src/libserver/worker_util.h b/src/libserver/worker_util.h
index 2372375ba5f4b391dc9d2f0a93db410094d35603..2af32d1e82bf200b16d8c08a3affab5bb448a3ec 100644 (file)
--- a/src/libserver/worker_util.h
+++ b/src/libserver/worker_util.h
@@ -67,10 +67,12 @@ void rspamd_worker_stop_accept (struct rspamd_worker *worker);
 
 typedef gint (*rspamd_controller_func_t) (
                struct rspamd_http_connection_entry *conn_ent,
-               struct rspamd_http_message *msg);
+               struct rspamd_http_message *msg,
+               struct module_ctx *ctx);
 
 struct rspamd_custom_controller_command {
        const gchar *command;
+       struct module_ctx *ctx;
        gboolean privilleged;
        gboolean require_message;
        rspamd_controller_func_t handler;

diff --git a/src/webui.c b/src/webui.c
index 91bd9fe634e178e2dc1aa3664857717e1e5aba14..685e176043d7ec85113abb00a16042b8179d8657 100644 (file)
--- a/src/webui.c
+++ b/src/webui.c
@@ -1483,7 +1483,7 @@ rspamd_webui_handle_custom (struct rspamd_http_connection_entry *conn_ent,
                return 0;
        }
 
-       return cmd->handler (conn_ent, msg);
+       return cmd->handler (conn_ent, msg, cmd->ctx);
 }
 
 static void