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
128
129
|
#ifndef RSPAMD_MAP_H
#define RSPAMD_MAP_H
#include "config.h"
#include "mem_pool.h"
#include "radix.h"
/**
* Maps API is designed to load lists data from different dynamic sources.
* It monitor files and HTTP locations for modifications and reload them if they are
* modified.
*/
enum fetch_proto {
MAP_PROTO_FILE,
MAP_PROTO_HTTP,
};
/**
* Callback data for async load
*/
struct map_cb_data {
gint state;
void *prev_data;
void *cur_data;
};
/**
* Data specific to file maps
*/
struct file_map_data {
const gchar *filename;
struct stat st;
};
/**
* Data specific to HTTP maps
*/
struct http_map_data {
struct addrinfo *addr;
guint16 port;
gchar *path;
gchar *host;
time_t last_checked;
gshort chunked;
gchar read_buf[BUFSIZ];
guint32 rlen;
guint32 chunk;
guint32 chunk_remain;
};
/**
* Callback types
*/
typedef gchar* (*map_cb_t)(memory_pool_t *pool, gchar *chunk, gint len, struct map_cb_data *data);
typedef void (*map_fin_cb_t)(memory_pool_t *pool, struct map_cb_data *data);
/**
* Common map object
*/
struct config_file;
struct rspamd_map {
memory_pool_t *pool;
struct config_file *cfg;
enum fetch_proto protocol;
map_cb_t read_callback;
map_fin_cb_t fin_callback;
void **user_data;
struct event ev;
struct timeval tv;
struct event_base *ev_base;
void *map_data;
gchar *uri;
gchar *description;
guint32 id;
/* Shared lock for temporary disabling of map reading (e.g. when this map is written by UI) */
gint *locked;
};
/**
* Check map protocol
*/
gboolean check_map_proto (const gchar *map_line, gint *res, const gchar **pos);
/**
* Add map from line
*/
gboolean add_map (struct config_file *cfg, const gchar *map_line, const gchar *description,
map_cb_t read_callback, map_fin_cb_t fin_callback, void **user_data);
/**
* Start watching of maps by adding events to libevent event loop
*/
void start_map_watch (struct config_file *cfg, struct event_base *ev_base);
/**
* Remove all maps watched (remove events)
*/
void remove_all_maps (struct config_file *cfg);
typedef void (*insert_func) (gpointer st, gconstpointer key, gconstpointer value);
/**
* Common callbacks for frequent types of lists
*/
/**
* Radix list is a list like ip/mask
*/
gchar* read_radix_list (memory_pool_t *pool, gchar *chunk, gint len, struct map_cb_data *data);
void fin_radix_list (memory_pool_t *pool, struct map_cb_data *data);
/**
* Host list is an ordinal list of hosts or domains
*/
gchar* read_host_list (memory_pool_t *pool, gchar *chunk, gint len, struct map_cb_data *data);
void fin_host_list (memory_pool_t *pool, struct map_cb_data *data);
/**
* Kv list is an ordinal list of keys and values separated by whitespace
*/
gchar* read_kv_list (memory_pool_t *pool, gchar *chunk, gint len, struct map_cb_data *data);
void fin_kv_list (memory_pool_t *pool, struct map_cb_data *data);
/**
* FSM for lists parsing (support comments, blank lines and partial replies)
*/
gchar * abstract_parse_list (memory_pool_t * pool, gchar * chunk, gint len, struct map_cb_data *data, insert_func func);
#endif
|
