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
|
#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 {
PROTO_FILE,
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 in_addr addr;
guint16 port;
gchar *path;
gchar *host;
time_t last_checked;
gboolean chunked;
u_char read_buf[BUFSIZ];
guint32 rlen;
guint32 chunk;
guint32 chunk_read;
};
/**
* Callback types
*/
typedef u_char* (*map_cb_t)(memory_pool_t *pool, u_char *chunk, size_t 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 rspamd_map {
memory_pool_t *pool;
enum fetch_proto protocol;
map_cb_t read_callback;
map_fin_cb_t fin_callback;
void **user_data;
struct event ev;
struct timeval tv;
void *map_data;
};
/**
* Check map protocol
*/
gboolean check_map_proto (const gchar *map_line, gint *res, const gchar **pos);
/**
* Add map from line
*/
gboolean add_map (const gchar *map_line, 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 (void);
/**
* Remove all maps watched (remove events)
*/
void remove_all_maps (void);
typedef void (*insert_func) (gpointer st, gconstpointer key, gpointer value);
/**
* Common callbacks for frequent types of lists
*/
/**
* Radix list is a list like ip/mask
*/
u_char* read_radix_list (memory_pool_t *pool, u_char *chunk, size_t 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
*/
u_char* read_host_list (memory_pool_t *pool, u_char *chunk, size_t len, struct map_cb_data *data);
void fin_host_list (memory_pool_t *pool, struct map_cb_data *data);
/**
* FSM for lists parsing (support comments, blank lines and partial replies)
*/
u_char * abstract_parse_list (memory_pool_t * pool, u_char * chunk, size_t len, struct map_cb_data *data, insert_func func);
#endif
|
