diff options
Diffstat (limited to 'src/worker.c')
| -rw-r--r-- | src/worker.c | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/src/worker.c b/src/worker.c index e63624848..2598be415 100644 --- a/src/worker.c +++ b/src/worker.c @@ -26,6 +26,7 @@ * Rspamd worker implementation */ +#include <libserver/rspamd_control.h> #include "config.h" #include "libutil/util.h" #include "libutil/map.h" @@ -39,6 +40,7 @@ #include "keypairs_cache.h" #include "libstat/stat_api.h" #include "libserver/worker_util.h" +#include "libserver/rspamd_control.h" #include "lua/lua_common.h" @@ -280,6 +282,31 @@ accept_socket (gint fd, short what, void *arg) ctx->ev_base); } +#ifdef WITH_HYPERSCAN +static gboolean +rspamd_worker_hyperscan_ready (struct rspamd_main *rspamd_main, + struct rspamd_worker *worker, gint fd, + struct rspamd_control_command *cmd, + gpointer ud) +{ + struct rspamd_control_reply rep; + + msg_info ("loading hyperscan expressions after receiving compilation notice"); + memset (&rep, 0, sizeof (rep)); + rep.type = RSPAMD_CONTROL_HYPERSCAN_LOADED; + + rep.reply.hs_loaded.status = rspamd_re_cache_load_hyperscan ( + worker->srv->cfg->re_cache, cmd->cmd.hs_loaded.cache_dir); + + if (write (fd, &rep, sizeof (rep)) != sizeof (rep)) { + msg_err ("cannot write reply to the control socket: %s", + strerror (errno)); + } + + return TRUE; +} +#endif + gpointer init_worker (struct rspamd_config *cfg) { @@ -359,6 +386,11 @@ start_worker (struct rspamd_worker *worker) ctx->keys_cache = rspamd_keypair_cache_new (256); rspamd_stat_init (worker->srv->cfg); +#ifdef WITH_HYPERSCAN + rspamd_control_worker_add_cmd_handler (worker, RSPAMD_CONTROL_HYPERSCAN_LOADED, + rspamd_worker_hyperscan_ready, ctx); +#endif + event_base_loop (ctx->ev_base, 0); rspamd_worker_block_signals (); |