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
|
#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);
/* Write pid to file */
gint write_pid (struct rspamd_main *);
/* Make specified socket non-blocking */
gint make_socket_nonblocking (gint);
gint make_socket_blocking (gint);
/* Poll 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
gint init_title(gint argc, gchar *argv[], gchar *envp[]);
gint setproctitle(const gchar *fmt, ...);
#endif
#ifndef HAVE_PIDFILE
struct pidfh {
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;
};
struct pidfh *pidfile_open(const gchar *path, mode_t mode, pid_t *pidptr);
gint pidfile_write(struct pidfh *pfh);
gint pidfile_close(struct pidfh *pfh);
gint pidfile_remove(struct pidfh *pfh);
#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
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
double set_counter (const gchar *name, guint32 value);
gboolean lock_file (gint fd, gboolean async);
gboolean unlock_file (gint fd, gboolean async);
guint rspamd_strcase_hash (gconstpointer key);
gboolean rspamd_strcase_equal (gconstpointer v, gconstpointer v2);
guint fstr_strcase_hash (gconstpointer key);
gboolean fstr_strcase_equal (gconstpointer v, gconstpointer v2);
void gperf_profiler_init (struct config_file *cfg, const gchar *descr);
stat_file_t* get_statfile_by_symbol (statfile_pool_t *pool, struct classifier_config *ccf,
const gchar *symbol, struct statfile **st, gboolean try_create);
#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
|
