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
|
#ifndef RSPAMD_SYMBOLS_CACHE_H
#define RSPAMD_SYMBOLS_CACHE_H
#include "config.h"
#include "radix.h"
#define MAX_SYMBOL 128
struct worker_task;
struct config_file;
typedef void (*symbol_func_t)(struct worker_task *task, gpointer user_data);
struct saved_cache_item {
char symbol[MAX_SYMBOL];
double weight;
uint32_t frequency;
double avg_time;
};
struct dynamic_map_item {
struct in_addr addr;
uint32_t mask;
gboolean negative;
};
struct cache_item {
/* Static item's data */
struct saved_cache_item *s;
/* For dynamic rules */
struct dynamic_map_item *networks;
uint32_t networks_number;
gboolean is_dynamic;
/* Callback data */
symbol_func_t func;
gpointer user_data;
};
struct symbols_cache {
/* Normal cache items */
GList *static_items;
/* Items that have negative weights */
GList *negative_items;
/* Radix map of dynamic rules with ip mappings */
radix_tree_t *dynamic_map;
radix_tree_t *negative_dynamic_map;
/* Common dynamic rules */
GList *dynamic_items;
memory_pool_t *static_pool;
guint cur_items;
guint used_items;
guint uses;
gpointer map;
memory_pool_rwlock_t *lock;
struct config_file *cfg;
};
/**
* Load symbols cache from file, must be called _after_ init_symbols_cache
*/
gboolean init_symbols_cache (memory_pool_t *pool, struct symbols_cache *cache, struct config_file *cfg, const char *filename);
/**
* Register function for symbols parsing
* @param name name of symbol
* @param func pointer to handler
* @param user_data pointer to user_data
*/
void register_symbol (struct symbols_cache **cache, const char *name, double weight, symbol_func_t func, gpointer user_data);
/**
* Register function for dynamic symbols parsing
* @param name name of symbol
* @param func pointer to handler
* @param user_data pointer to user_data
*/
void register_dynamic_symbol (memory_pool_t *pool, struct symbols_cache **cache, const char *name,
double weight, symbol_func_t func,
gpointer user_data, GList *networks);
/**
* Call function for cached symbol using saved callback
* @param task task object
* @param cache symbols cache
* @param saved_item pointer to currently saved item
*/
gboolean call_symbol_callback (struct worker_task *task, struct symbols_cache *cache, gpointer *save);
/**
* Remove all dynamic rules from cache
* @param cache symbols cache
*/
void remove_dynamic_rules (struct symbols_cache *cache);
#endif
|
