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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
|
#ifndef RSPAMD_UTIL_H
#define RSPAMD_UTIL_H
#include "config.h"
#include "mem_pool.h"
#include "radix.h"
#include "statfile.h"
#include "printf.h"
#include "fstring.h"
struct config_file;
struct rspamd_main;
struct workq;
struct statfile;
struct classifier_config;
/**
* Process type: main or worker
*/
enum process_type {
TYPE_UNKNOWN=-1,
TYPE_MAIN,
TYPE_WORKER,
TYPE_CONTROLLER,
TYPE_LMTP,
TYPE_SMTP,
TYPE_FUZZY,
TYPE_GREYLIST,
TYPE_MAX=255
};
/*
* Create socket and bind or connect it to specified address and port
*/
gint make_tcp_socket (struct in_addr *, u_short, gboolean is_server, gboolean async);
/*
* Create socket and bind or connect it to specified address and port
*/
gint make_udp_socket (struct in_addr *, u_short, gboolean is_server, gboolean async);
/*
* Accept from socket
*/
gint accept_from_socket (gint listen_sock, struct sockaddr *addr, socklen_t *len);
/*
* Create and bind or connect unix socket
*/
gint make_unix_socket (const gchar *, struct sockaddr_un *, gboolean is_server, gboolean async);
/**
* Make universal stream socket
* @param credits host, ip or path to unix socket
* @param port port (used for network sockets)
* @param async make this socket asynced
* @param is_server make this socket as server socket
* @param try_resolve try name resolution for a socket (BLOCKING)
*/
gint make_universal_stream_socket (const gchar *credits, guint16 port,
gboolean async, gboolean is_server, gboolean try_resolve);
/*
* Create socketpair
*/
gint make_socketpair (gint pair[2]);
/*
* Write pid to file
*/
gint write_pid (struct rspamd_main *);
/*
* Make specified socket non-blocking
*/
gint make_socket_nonblocking (gint);
/*
* Make specified socket blocking
*/
gint make_socket_blocking (gint);
/*
* Poll a sync socket for specified events
*/
gint poll_sync_socket (gint fd, gint timeout, short events);
/*
* Init signals
*/
#ifdef HAVE_SA_SIGINFO
void init_signals (struct sigaction *sa, void (*sig_handler)(gint, siginfo_t *, void *));
#else
void init_signals (struct sigaction *sa, sighandler_t);
#endif
/*
* Send specified signal to each worker
*/
void pass_signal_worker (GHashTable *, gint );
/*
* Convert string to lowercase
*/
void convert_to_lowercase (gchar *str, guint size);
#ifndef HAVE_SETPROCTITLE
/*
* Process title utility functions
*/
gint init_title(gint argc, gchar *argv[], gchar *envp[]);
gint setproctitle(const gchar *fmt, ...);
#endif
#ifndef HAVE_PIDFILE
/*
* Pidfile functions from FreeBSD libutil code
*/
typedef struct rspamd_pidfh_s {
gint pf_fd;
#ifdef HAVE_PATH_MAX
gchar pf_path[PATH_MAX + 1];
#elif defined(HAVE_MAXPATHLEN)
gchar pf_path[MAXPATHLEN + 1];
#else
gchar pf_path[1024 + 1];
#endif
dev_t pf_dev;
ino_t pf_ino;
} rspamd_pidfh_t;
rspamd_pidfh_t *rspamd_pidfile_open(const gchar *path, mode_t mode, pid_t *pidptr);
gint rspamd_pidfile_write(rspamd_pidfh_t *pfh);
gint rspamd_pidfile_close(rspamd_pidfh_t *pfh);
gint rspamd_pidfile_remove(rspamd_pidfh_t *pfh);
#else
typedef struct pidfh rspamd_pidfh_t;
#define rspamd_pidfile_open pidfile_open
#define rspamd_pidfile_write pidfile_write
#define rspamd_pidfile_close pidfile_close
#define rspamd_pidfile_remove pidfile_remove
#endif
/*
* Replace %r with rcpt value and %f with from value, new string is allocated in pool
*/
gchar* resolve_stat_filename (memory_pool_t *pool, gchar *pattern, gchar *rcpt, gchar *from);
#ifdef HAVE_CLOCK_GETTIME
/*
* Calculate check time with specified resolution of timer
*/
const gchar* calculate_check_time (struct timeval *tv, struct timespec *begin, gint resolution);
#else
const gchar* calculate_check_time (struct timeval *begin, gint resolution);
#endif
/*
* Set counter for a symbol
*/
double set_counter (const gchar *name, guint32 value);
/*
* File locking functions
*/
gboolean lock_file (gint fd, gboolean async);
gboolean unlock_file (gint fd, gboolean async);
/*
* Hash table utility functions for case insensitive hashing
*/
guint rspamd_strcase_hash (gconstpointer key);
gboolean rspamd_strcase_equal (gconstpointer v, gconstpointer v2);
/*
* Hash table utility functions for hashing fixed strings
*/
guint fstr_strcase_hash (gconstpointer key);
gboolean fstr_strcase_equal (gconstpointer v, gconstpointer v2);
/*
* Google perf-tools initialization function
*/
void gperf_profiler_init (struct config_file *cfg, const gchar *descr);
/*
* Get a statfile by symbol
*/
stat_file_t* get_statfile_by_symbol (statfile_pool_t *pool, struct classifier_config *ccf,
const gchar *symbol, struct statfile **st, gboolean try_create);
/*
* Workaround for older versions of glib
*/
#if ((GLIB_MAJOR_VERSION == 2) && (GLIB_MINOR_VERSION < 22))
void g_ptr_array_unref (GPtrArray *array);
#endif
/*
* Copy src to dest limited to len, in compare with standart strlcpy(3) rspamd strlcpy does not
* traverse the whole string and it is possible to use it for non NULL terminated strings. This is
* more like memccpy(dst, src, size, '\0')
*
* @param dst destination string
* @param src source string
* @param siz length of destination buffer
* @return bytes copied
*/
gsize rspamd_strlcpy (gchar *dst, const gchar *src, gsize siz);
/*
* Convert process type to its name
*
* @param type numeric type
* @return string representation of type
*/
const gchar * process_to_str (enum process_type type);
/*
* Convert string to process type
*
* @param type numeric type
* @return string representation of type
*/
enum process_type str_to_process (const gchar *str);
/*
* Strip <> from email address
*/
gchar * escape_braces_addr_fstr (memory_pool_t *pool, f_str_t *in);
/*
* Convert milliseconds to timeval fields
*/
#define msec_to_tv(msec, tv) do { (tv)->tv_sec = (msec) / 1000; (tv)->tv_usec = ((msec) - (tv)->tv_sec * 1000) * 1000; } while(0)
struct worker_task;
struct rspamd_worker;
/**
* Construct new task for worker
*/
struct worker_task* construct_task (struct rspamd_worker *worker);
/**
* Destroy task object and remove its IO dispatcher if it exists
*/
void free_task (struct worker_task *task, gboolean is_soft);
void free_task_hard (gpointer ud);
void free_task_soft (gpointer ud);
/*
* Find string find in string s ignoring case
*/
gchar* rspamd_strncasestr (const gchar *s, const gchar *find, gint len);
#endif
|
