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
130
131
132
133
134
135
136
137
138
139
|
/*-
* Copyright 2016 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 SRC_LIBUTIL_MAP_PRIVATE_H_
#define SRC_LIBUTIL_MAP_PRIVATE_H_
#include "config.h"
#include "mem_pool.h"
#include "keypair.h"
#include "unix-std.h"
#include "ref.h"
typedef void (*rspamd_map_dtor) (gpointer p);
#define msg_err_map(...) rspamd_default_log_function (G_LOG_LEVEL_CRITICAL, \
"map", map->tag, \
G_STRFUNC, \
__VA_ARGS__)
#define msg_warn_map(...) rspamd_default_log_function (G_LOG_LEVEL_WARNING, \
"map", map->tag, \
G_STRFUNC, \
__VA_ARGS__)
#define msg_info_map(...) rspamd_default_log_function (G_LOG_LEVEL_INFO, \
"map", map->tag, \
G_STRFUNC, \
__VA_ARGS__)
#define msg_debug_map(...) rspamd_default_log_function (G_LOG_LEVEL_DEBUG, \
"map", map->tag, \
G_STRFUNC, \
__VA_ARGS__)
enum fetch_proto {
MAP_PROTO_FILE,
MAP_PROTO_HTTP,
};
struct rspamd_map_backend {
enum fetch_proto protocol;
gboolean is_signed;
struct rspamd_cryptobox_pubkey *trusted_pubkey;
union {
struct file_map_data *fd;
struct http_map_data *hd;
} data;
gchar *uri;
ref_entry_t ref;
};
struct rspamd_map {
struct rspamd_dns_resolver *r;
struct rspamd_config *cfg;
GPtrArray *backends;
map_cb_t read_callback;
map_fin_cb_t fin_callback;
void **user_data;
struct event_base *ev_base;
gchar *description;
gchar *name;
guint32 id;
struct event ev;
struct timeval tv;
gdouble poll_timeout;
/* Shared lock for temporary disabling of map reading (e.g. when this map is written by UI) */
gint *locked;
gchar tag[MEMPOOL_UID_LEN];
rspamd_map_dtor dtor;
gpointer dtor_data;
};
/**
* Data specific to file maps
*/
struct file_map_data {
gchar *filename;
struct stat st;
};
/**
* Data specific to HTTP maps
*/
struct http_map_data {
gchar *path;
gchar *host;
gchar *last_signature;
time_t last_checked;
gboolean request_sent;
guint16 port;
};
enum rspamd_map_http_stage {
map_resolve_host2 = 0, /* 2 requests sent */
map_resolve_host1, /* 1 requests sent */
map_load_file,
map_load_pubkey,
map_load_signature
};
struct map_periodic_cbdata {
struct rspamd_map *map;
struct map_cb_data cbdata;
gboolean need_modify;
gboolean errored;
guint cur_backend;
ref_entry_t ref;
};
struct http_callback_data {
struct event_base *ev_base;
struct rspamd_http_connection *conn;
rspamd_inet_addr_t *addr;
struct rspamd_map *map;
struct rspamd_map_backend *bk;
struct http_map_data *data;
struct map_periodic_cbdata *periodic;
struct rspamd_cryptobox_pubkey *pk;
gboolean check;
gchar *tmpfile;
enum rspamd_map_http_stage stage;
gint out_fd;
gint fd;
struct timeval tv;
ref_entry_t ref;
};
#endif /* SRC_LIBUTIL_MAP_PRIVATE_H_ */
|
