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
|
/*-
* Copyright 2021 Vsevolod Stakhov
*
* 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.
*/
#ifndef RSPAMD_COMPOSITES_INTERNAL_HXX
#define RSPAMD_COMPOSITES_INTERNAL_HXX
#pragma once
#include <string>
#include "libutil/expression.h"
#include "libutil/cxx/hash_util.hxx"
#include "libserver/cfg_file.h"
namespace rspamd::composites {
/**
* Subr for composite expressions
*/
extern const struct rspamd_atom_subr composite_expr_subr;
enum class rspamd_composite_policy {
RSPAMD_COMPOSITE_POLICY_REMOVE_ALL = 0,
RSPAMD_COMPOSITE_POLICY_REMOVE_SYMBOL,
RSPAMD_COMPOSITE_POLICY_REMOVE_WEIGHT,
RSPAMD_COMPOSITE_POLICY_LEAVE,
RSPAMD_COMPOSITE_POLICY_UNKNOWN
};
/**
* Static composites structure
*/
struct rspamd_composite {
std::string str_expr;
std::string sym;
struct rspamd_expression *expr;
gint id;
rspamd_composite_policy policy;
};
#define COMPOSITE_MANAGER_FROM_PTR(ptr) (reinterpret_cast<rspamd::composites::composites_manager *>(ptr))
class composites_manager {
public:
composites_manager(struct rspamd_config *_cfg)
: cfg(_cfg)
{
rspamd_mempool_add_destructor(_cfg->cfg_pool, composites_manager_dtor, this);
}
auto size(void) const -> std::size_t
{
return all_composites.size();
}
auto find(std::string_view name) const -> const rspamd_composite *
{
auto found = composites.find(std::string(name));
if (found != composites.end()) {
return found->second.get();
}
return nullptr;
}
auto add_composite(std::string_view, const ucl_object_t *, bool silent_duplicate) -> rspamd_composite *;
auto add_composite(std::string_view name, std::string_view expression, bool silent_duplicate, double score = NAN) -> rspamd_composite *;
private:
~composites_manager() = default;
static void composites_manager_dtor(void *ptr)
{
delete COMPOSITE_MANAGER_FROM_PTR(ptr);
}
auto new_composite(std::string_view composite_name, rspamd_expression *expr,
std::string_view composite_expression) -> auto
{
auto &composite = all_composites.emplace_back(std::make_shared<rspamd_composite>());
composite->expr = expr;
composite->id = all_composites.size() - 1;
composite->str_expr = composite_expression;
composite->sym = composite_name;
composites[composite->sym] = composite;
return composite;
}
ankerl::unordered_dense::map<std::string,
std::shared_ptr<rspamd_composite>, rspamd::smart_str_hash, rspamd::smart_str_equal>
composites;
/* Store all composites here, even if we have duplicates */
std::vector<std::shared_ptr<rspamd_composite>> all_composites;
struct rspamd_config *cfg;
};
}// namespace rspamd::composites
#endif//RSPAMD_COMPOSITES_INTERNAL_HXX
|
