diff options
Diffstat (limited to 'src/rspamadm/rescore.c')
| -rw-r--r-- | src/rspamadm/rescore.c | 191 |
1 files changed, 0 insertions, 191 deletions
diff --git a/src/rspamadm/rescore.c b/src/rspamadm/rescore.c deleted file mode 100644 index 346c49c68..000000000 --- a/src/rspamadm/rescore.c +++ /dev/null @@ -1,191 +0,0 @@ -/*- - * Copyright 2017 Pragadeesh C - * - * 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 "lua/lua_common.h" - -static gchar *logdir = NULL; -static gchar *output = "new.scores"; -static gboolean score_diff = false; /* Print score diff flag */ -static gchar *config = NULL; -extern struct rspamd_main *rspamd_main; -/* Defined in modules.c */ -extern module_t *modules[]; -extern worker_t *workers[]; - -static void rspamadm_rescore (gint argc, gchar **argv, - const struct rspamadm_command *cmd); - -static const char *rspamadm_rescore_help (gboolean full_help, - const struct rspamadm_command *cmd); - -struct rspamadm_command rescore_command = { - .name = "rescore", - .flags = 0, - .help = rspamadm_rescore_help, - .run = rspamadm_rescore -}; - -static GOptionEntry entries[] = { - {"logdir", 'l', 0, G_OPTION_ARG_FILENAME, &logdir, - "Logs directory", NULL}, - {"output", 'o', 0, G_OPTION_ARG_FILENAME, &output, - "Scores output location", NULL}, - {"diff", 'd', 0, G_OPTION_ARG_NONE, &score_diff, - "Print score diff", NULL}, - {"config", 'c', 0, G_OPTION_ARG_STRING, &config, - "Config file to use", NULL}, - {NULL, 0, 0, G_OPTION_ARG_NONE, NULL, NULL, NULL} -}; - -static void -config_logger (rspamd_mempool_t *pool, gpointer ud) -{ - struct rspamd_main *rm = ud; - - rm->cfg->log_type = RSPAMD_LOG_CONSOLE; - rm->cfg->log_level = G_LOG_LEVEL_MESSAGE; - - rspamd_set_logger (rm->cfg, g_quark_try_string ("main"), &rm->logger, - rm->server_pool); - - if (rspamd_log_open_priv (rm->logger, rm->workers_uid, rm->workers_gid) == - -1) { - fprintf (stderr, "Fatal error, cannot open logfile, exiting\n"); - exit (EXIT_FAILURE); - } -} - -static const char * -rspamadm_rescore_help (gboolean full_help, const struct rspamadm_command *cmd) -{ - const char *help_str; - - if (full_help) { - help_str = "Estimate optimal symbol weights from log files\n\n" - "Usage: rspamadm rescore -l <log_directory>\n" - "Where options are:\n\n" - "-l: path to logs directory\n" - "-o: scores output file location\n" - "-d: print scores diff\n" - "-i: max iterations for perceptron\n"; - } else { - help_str = "Estimate optimal symbol weights from log files"; - } - - return help_str; -} - -static void -rspamadm_rescore (gint argc, gchar **argv, const struct rspamadm_command *cmd) -{ - GOptionContext *context; - GError *error = NULL; - struct rspamd_config *cfg = rspamd_main->cfg, **pcfg; - gboolean ret = TRUE; - worker_t **pworker; - const gchar *confdir; - -#ifndef WITH_TORCH - rspamd_fprintf (stderr, "Torch is not enabled. " - "Use -DENABLE_TORCH=ON option while running cmake.\n"); - exit (EXIT_FAILURE); -#endif - - context = g_option_context_new ( - "rescore - estimate optimal symbol weights from log files"); - - 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); - g_option_context_set_ignore_unknown_options (context, TRUE); - - if (!g_option_context_parse (context, &argc, &argv, &error)) { - rspamd_fprintf (stderr, "option parsing failed: %s\n", error->message); - g_error_free (error); - exit (EXIT_FAILURE); - } - - if (logdir == NULL) { - rspamd_fprintf (stderr, "Please specify log directory.\n"); - exit (EXIT_FAILURE); - } - - if (config == NULL) { - if ((confdir = g_hash_table_lookup (ucl_vars, "CONFDIR")) == NULL) { - confdir = RSPAMD_CONFDIR; - } - - config = g_strdup_printf ("%s%c%s", confdir, G_DIR_SEPARATOR, - "rspamd.conf"); - } - - pworker = &workers[0]; - while (*pworker) { - /* Init string quarks */ - (void) g_quark_from_static_string ((*pworker)->name); - pworker++; - } - - cfg->cache = rspamd_symbols_cache_new (cfg); - cfg->compiled_modules = modules; - cfg->compiled_workers = workers; - cfg->cfg_name = config; - - if (!rspamd_config_read (cfg, cfg->cfg_name, config_logger, rspamd_main, ucl_vars)) { - ret = FALSE; - } - else { - /* Do post-load actions */ - rspamd_lua_post_load_config (cfg); - - if (!rspamd_init_filters (cfg, FALSE)) { - ret = FALSE; - } - - if (ret) { - ret = rspamd_config_post_load (cfg, RSPAMD_CONFIG_INIT_SYMCACHE); - rspamd_symbols_cache_validate (cfg->cache, - cfg, - FALSE); - } - } - - if (ret) { - ucl_object_insert_key (cfg->rcl_obj, ucl_object_fromstring (cfg->cfg_name), - "config_path", 0, false); - ucl_object_insert_key (cfg->rcl_obj, ucl_object_fromstring (logdir), - "logdir", 0, false); - ucl_object_insert_key (cfg->rcl_obj, ucl_object_fromstring (output), - "output", 0, false); - ucl_object_insert_key (cfg->rcl_obj, ucl_object_frombool (score_diff), - "diff", 0, false); - pcfg = lua_newuserdata (L, sizeof (struct rspamd_config *)); - rspamd_lua_setclass (L, "rspamd{config}", -1); - *pcfg = cfg; - lua_setglobal (L, "rspamd_config"); - rspamadm_execute_lua_ucl_subr (L, - argc, - argv, - cfg->rcl_obj, - "rescore"); - } -} |