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
|
/**
* @file protocol.h
* Rspamd protocol definition
*/
#ifndef RSPAMD_PROTOCOL_H
#define RSPAMD_PROTOCOL_H
#include "config.h"
#include "filter.h"
#include "http_connection.h"
#include "task.h"
#define RSPAMD_BASE_ERROR 500
#define RSPAMD_FILTER_ERROR RSPAMD_BASE_ERROR + 1
#define RSPAMD_NETWORK_ERROR RSPAMD_BASE_ERROR + 2
#define RSPAMD_PROTOCOL_ERROR RSPAMD_BASE_ERROR + 3
#define RSPAMD_LENGTH_ERROR RSPAMD_BASE_ERROR + 4
#define RSPAMD_STATFILE_ERROR RSPAMD_BASE_ERROR + 5
struct rspamd_protocol_log_symbol_result {
guint32 id;
float score;
};
struct rspamd_protocol_log_message_sum {
guint32 nresults;
guint32 nextra;
guint32 settings_id;
gdouble score;
gdouble required_score;
struct rspamd_protocol_log_symbol_result results[];
};
struct rspamd_metric;
/**
* Process headers into HTTP message and set appropriate task fields
* @param task
* @param msg
* @return
*/
gboolean rspamd_protocol_handle_headers (struct rspamd_task *task,
struct rspamd_http_message *msg);
/**
* Process control chunk and update task structure accordingly
* @param task
* @param control
* @return
*/
gboolean rspamd_protocol_handle_control (struct rspamd_task *task,
const ucl_object_t *control);
/**
* Process HTTP request to the task structure
* @param task
* @param msg
* @return
*/
gboolean rspamd_protocol_handle_request (struct rspamd_task *task,
struct rspamd_http_message *msg);
/**
* Write task results to http message
* @param msg
* @param task
*/
void rspamd_protocol_http_reply (struct rspamd_http_message *msg,
struct rspamd_task *task, ucl_object_t **pobj);
/**
* Write data to log pipes
* @param task
*/
void rspamd_protocol_write_log_pipe (struct rspamd_task *task);
enum rspamd_protocol_flags {
RSPAMD_PROTOCOL_BASIC = 1 << 0,
RSPAMD_PROTOCOL_METRICS = 1 << 1,
RSPAMD_PROTOCOL_MESSAGES = 1 << 2,
RSPAMD_PROTOCOL_RMILTER = 1 << 3,
RSPAMD_PROTOCOL_DKIM = 1 << 4,
RSPAMD_PROTOCOL_URLS = 1 << 5,
RSPAMD_PROTOCOL_EXTRA = 1 << 6,
};
#define RSPAMD_PROTOCOL_DEFAULT (RSPAMD_PROTOCOL_BASIC| \
RSPAMD_PROTOCOL_METRICS| \
RSPAMD_PROTOCOL_MESSAGES| \
RSPAMD_PROTOCOL_RMILTER| \
RSPAMD_PROTOCOL_DKIM| \
RSPAMD_PROTOCOL_EXTRA)
/**
* Write reply to ucl object filling log buffer
* @param task
* @param logbuf
* @return
*/
ucl_object_t * rspamd_protocol_write_ucl (struct rspamd_task *task,
enum rspamd_protocol_flags flags);
/**
* Write reply for specified task command
* @param task task object
* @return 0 if we wrote reply and -1 if there was some error
*/
void rspamd_protocol_write_reply (struct rspamd_task *task);
/**
* Convert rspamd output to legacy protocol reply
* @param task
* @param top
* @param out
*/
void rspamd_ucl_torspamc_output (const ucl_object_t *top,
rspamd_fstring_t **out);
void rspamd_ucl_tospamc_output (const ucl_object_t *top,
rspamd_fstring_t **out);
#endif
|
