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
|
#ifndef RPOP_MAIN_H
#define RPOP_MAIN_H
#include "config.h"
#include <sys/types.h>
#include <sys/socket.h>
#ifndef HAVE_OWN_QUEUE_H
#include <sys/queue.h>
#else
#include "queue.h"
#endif
#include <sys/time.h>
#include <sys/un.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>
#include <event.h>
#include "fstring.h"
#include "mem_pool.h"
#include "url.h"
#include "memcached.h"
#include "protocol.h"
#include <glib.h>
#include <gmime/gmime.h>
/* Default values */
#define FIXED_CONFIG_FILE "./rspamd.conf"
/* Time in seconds to exit for old worker */
#define SOFT_SHUTDOWN_TIME 60
/* Logging in postfix style */
#define msg_err g_error
#define msg_warn g_warning
#define msg_info g_message
#define msg_debug g_debug
/* Process type: main or worker */
enum process_type {
TYPE_MAIN,
TYPE_WORKER,
};
/* Worker process structure */
struct rspamd_worker {
pid_t pid;
char is_initialized;
char is_dying;
TAILQ_ENTRY (rspamd_worker) next;
struct rspamd_main *srv;
enum process_type type;
struct event sig_ev;
struct event bind_ev;
};
struct pidfh;
struct config_file;
struct filter_chain;
/* Struct that determine main server object (for logging purposes) */
struct rspamd_main {
struct config_file *cfg;
pid_t pid;
/* Pid file structure */
struct pidfh *pfh;
enum process_type type;
unsigned ev_initialized:1;
TAILQ_HEAD (workq, rspamd_worker) workers;
};
struct filter_result {
const char *symbol;
struct filter_chain *chain;
int mark;
TAILQ_ENTRY (filter_result) next;
};
struct chain_result {
struct filter_chain *chain;
int *marks;
unsigned int marks_num;
int result_mark;
TAILQ_ENTRY (chain_result) next;
};
struct mime_part {
GMimeContentType *type;
GByteArray *content;
TAILQ_ENTRY (mime_part) next;
};
struct save_point {
enum { C_FILTER, PERL_FILTER } save_type;
void *entry;
void *chain;
unsigned int saved;
};
struct worker_task {
struct rspamd_worker *worker;
enum {
READ_COMMAND,
READ_HEADER,
READ_MESSAGE,
WRITE_REPLY,
WRITE_ERROR,
WAIT_FILTER,
} state;
size_t content_length;
enum rspamd_protocol proto;
enum rspamd_command cmd;
char *helo;
char *from;
char *rcpt;
unsigned int nrcpt;
struct in_addr from_addr;
f_str_buf_t *msg;
struct bufferevent *bev;
/* Memcached connection for this task */
memcached_ctx_t *memc_ctx;
unsigned memc_busy:1;
/* Number of mime parts */
int parts_count;
/* Headers */
GMimeMessage *message;
/* All parts of message */
TAILQ_HEAD (mime_partq, mime_part) parts;
/* URLs extracted from message */
TAILQ_HEAD (uriq, uri) urls;
/* List of filter results */
TAILQ_HEAD (resultsq, filter_result) results;
/* Results of all chains */
TAILQ_HEAD (chainsq, chain_result) chain_results;
struct config_file *cfg;
struct save_point save;
/* Memory pool that is associated with this task */
memory_pool_t *task_pool;
};
struct module_ctx {
int (*header_filter)(struct worker_task *task);
int (*mime_filter)(struct worker_task *task);
int (*message_filter)(struct worker_task *task);
int (*url_filter)(struct worker_task *task);
};
struct c_module {
const char *name;
struct module_ctx *ctx;
LIST_ENTRY (c_module) next;
};
void start_worker (struct rspamd_worker *worker, int listen_sock);
int process_filters (struct worker_task *task);
#endif
/*
* vi:ts=4
*/
|
