1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/*-
* 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 "rspamadm.h"
#include "config.h"
#include "lua/lua_common.h"
static gchar *ham_directory = NULL;
static gchar *spam_directory = NULL;
static gchar *output_location = "results.log";
static gint connections = 10;
static gdouble timeout = 60.0;
static void rspamadm_corpus_test (gint argc, gchar **argv);
static const char *rspamadm_corpus_test_help (gboolean full_help);
struct rspamadm_command corpus_test_command = {
.name = "corpus_test",
.flags = 0,
.help = rspamadm_corpus_test_help,
.run = rspamadm_corpus_test
};
// TODO add -nparellel and -o options
static GOptionEntry entries[] = {
{"ham", 'a', 0, G_OPTION_ARG_FILENAME, &ham_directory,
"Ham directory", NULL},
{"spam", 's', 0, G_OPTION_ARG_FILENAME, &spam_directory,
"Spam directory", NULL},
{"output", 'o', 0, G_OPTION_ARG_FILENAME, &output_location,
"Log output location", NULL},
{"connections", 'n', 0, G_OPTION_ARG_INT, &connections,
"Number of parellel connections [Default: 10]", NULL},
{"timeout", 't', 0, G_OPTION_ARG_DOUBLE, &timeout,
"Timeout for connections [Default: 60]", NULL},
{NULL, 0, 0, G_OPTION_ARG_NONE, NULL, NULL, NULL}
};
static const char *
rspamadm_corpus_test_help (gboolean full_help)
{
const char *help_str;
if (full_help) {
help_str = "Create logs files from email corpus\n\n"
"Usage: rspamadm corpus_test [-a <ham_directory>]"
" [-s <spam_directory>]\n"
"Where option are:\n\n"
"-a: path to ham directory\n"
"-s: path to spam directory\n"
"-n: maximum parallel connections\n"
"-o: log output file\n"
"-t: timeout for rspamc operations (default: 60)\n";
}
else {
help_str = "Create logs files from email corpus";
}
return help_str;
}
static void
rspamadm_corpus_test (gint argc, gchar **argv)
{
GOptionContext *context;
GError *error = NULL;
lua_State *L;
ucl_object_t *obj;
context = g_option_context_new (
"corpus_test - create logs files from email corpus");
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(1);
}
L = rspamd_lua_init ();
rspamd_lua_set_path(L, NULL, ucl_vars);
obj = ucl_object_typed_new (UCL_OBJECT);
ucl_object_insert_key (obj, ucl_object_fromstring (ham_directory),
"ham_directory", 0, false);
ucl_object_insert_key (obj, ucl_object_fromstring (spam_directory),
"spam_directory", 0, false);
ucl_object_insert_key (obj, ucl_object_fromstring (output_location),
"output_location", 0, false);
ucl_object_insert_key (obj, ucl_object_fromint (connections),
"connections", 0, false);
ucl_object_insert_key (obj, ucl_object_fromdouble (timeout),
"timeout", 0, false);
rspamadm_execute_lua_ucl_subr (L,
argc,
argv,
obj,
"corpus_test");
lua_close (L);
ucl_object_unref (obj);
}
|
