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
|
/*-
* Copyright 2017 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 RSPAMD_MILTER_H
#define RSPAMD_MILTER_H
#include "config.h"
#include "fstring.h"
#include "addr.h"
#include "contrib/libucl/ucl.h"
#include "ref.h"
enum rspamd_milter_reply {
RSPAMD_MILTER_ADDRCPT = '+',
RSPAMD_MILTER_DELRCPT = '-',
RSPAMD_MILTER_ACCEPT = 'a',
RSPAMD_MILTER_CONTINUE = 'c',
RSPAMD_MILTER_DISCARD = 'd',
RSPAMD_MILTER_ADDHEADER = 'h',
RSPAMD_MILTER_CHGHEADER = 'm',
RSPAMD_MILTER_REJECT = 'r',
RSPAMD_MILTER_TEMPFAIL = 't',
RSPAMD_MILTER_REPLYCODE = 'y',
RSPAMD_MILTER_OPTNEG = 'O',
RSPAMD_MILTER_PROGRESS = 'p'
};
struct rspamd_email_address;
struct event_base;
struct rspamd_http_message;
struct rspamd_milter_session {
GHashTable *macros;
rspamd_inet_addr_t *addr;
struct rspamd_email_address *from;
GPtrArray *rcpts;
rspamd_fstring_t *helo;
rspamd_fstring_t *hostname;
rspamd_fstring_t *message;
void *priv;
ref_entry_t ref;
};
typedef void (*rspamd_milter_finish) (gint fd,
struct rspamd_milter_session *session, void *ud);
typedef void (*rspamd_milter_error) (gint fd,
struct rspamd_milter_session *session,
void *ud, GError *err);
/**
* Handles socket with milter protocol
* @param fd
* @param finish_cb
* @param error_cb
* @param ud
* @return
*/
gboolean rspamd_milter_handle_socket (gint fd, const struct timeval *tv,
rspamd_mempool_t *pool,
struct event_base *ev_base, rspamd_milter_finish finish_cb,
rspamd_milter_error error_cb, void *ud);
/**
* Updates userdata for a session, returns previous userdata
* @param session
* @param ud
* @return
*/
void * rspamd_milter_update_userdata (struct rspamd_milter_session *session,
void *ud);
/**
* Sets SMTP reply string
* @param session
* @param xcode
* @param rcode
* @param reply
* @return
*/
gboolean rspamd_milter_set_reply (struct rspamd_milter_session *session,
rspamd_fstring_t *xcode,
rspamd_fstring_t *rcode,
rspamd_fstring_t *reply);
/**
* Send some action to the MTA
* @param fd
* @param session
* @param act
* @return
*/
gboolean rspamd_milter_send_action (struct rspamd_milter_session *session,
enum rspamd_milter_reply act, ...);
/**
* Adds some header
* @param session
* @param name
* @param value
* @return
*/
gboolean rspamd_milter_add_header (struct rspamd_milter_session *session,
GString *name, GString *value);
/**
* Removes some header
* @param session
* @param name
* @return
*/
gboolean rspamd_milter_del_header (struct rspamd_milter_session *session,
GString *name);
void rspamd_milter_session_unref (struct rspamd_milter_session *session);
struct rspamd_milter_session * rspamd_milter_session_ref (
struct rspamd_milter_session *session);
/**
* Converts milter session to HTTP session that is suitable for Rspamd
* @param session
* @return
*/
struct rspamd_http_message * rspamd_milter_to_http (
struct rspamd_milter_session *session);
/**
* Sends task results to the
* @param session
* @param results
*/
void rspamd_milter_send_task_results (struct rspamd_milter_session *session,
const ucl_object_t *results);
/**
* Init internal milter context
* @param spam_header spam header name (must NOT be NULL)
*/
void rspamd_milter_init_library (const gchar *spam_header);
/**
* Returns pool for a session
* @param session
* @return
*/
rspamd_mempool_t *rspamd_milter_get_session_pool (
struct rspamd_milter_session *session);
#endif
|
