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
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
|
--[[
Copyright (c) 2011-2015, Vsevolod Stakhov <vsevolod@highsecure.ru>
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:
1. Redistributions of source code must retain the above copyright notice, this
list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright notice,
this list of conditions and the following disclaimer in the documentation
and/or other materials provided with the distribution.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
]]--
-- This plugin implements user dynamic settings
-- Settings documentation can be found here:
-- https://rspamd.com/doc/configuration/settings.html
local set_section = rspamd_config:get_all_opt("settings")
local settings = {
[1] = {},
[2] = {},
[3] = {}
}
local settings_initialized = false
local max_pri = 0
local rspamd_logger = require "rspamd_logger"
local rspamd_ip = require "rspamd_ip"
local rspamd_regexp = require "rspamd_regexp"
local ucl = require "ucl"
require "fun" ()
-- Checks for overrided settings within query params and returns 'true' if
-- settings are overrided
local function check_query_settings(task)
-- Try 'settings' attribute
local query_set = task:get_request_header('settings')
if query_set then
local parser = ucl.parser()
local res,err = parser:parse_string(tostring(query_set))
if res then
task:set_settings(parser:get_object())
return true
end
end
local query_maxscore = task:get_request_header('maxscore')
if query_maxscore then
-- We have score limits redefined by request
local ms = tonumber(tostring(query_maxscore))
if ms then
local nset = {
default = {
actions = {
reject = ms
}
}
}
local query_softscore = task:get_request_header('softscore')
if query_softscore then
local ss = tonumber(tostring(query_softscore))
nset['default']['actions']['add header'] = ss
end
task:set_settings(nset)
return true
end
end
return false
end
-- Check limit for a task
local function check_settings(task)
local function check_addr_setting(rule, addr)
local function check_specific_addr(elt)
if rule['name'] then
if elt['addr'] == rule['name'] then
return true
end
end
if rule['user'] then
if rule['user'] == elt['user'] then
return true
end
end
if rule['domain'] then
if rule['domain'] == elt['domain'] then
return true
end
end
if rule['regexp'] then
if rule['regexp']:match(elt['addr']) then
return true
end
end
return false
end
for _, e in ipairs(addr) do
if check_specific_addr(e) then
return true
end
end
return false
end
local function check_ip_setting(rule, ip)
if rule[2] ~= 0 then
local nip = ip:apply_mask(rule[2])
if nip and nip == rule[1] then
return true
end
elseif ip == rule[1] then
return true
end
return false
end
local function check_specific_setting(name, rule, ip, from, rcpt, user)
local res = false
if rule['ip'] then
if not ip then
return nil
end
for _, i in ipairs(rule['ip']) do
res = check_ip_setting(i, ip)
if res then
break
end
end
if not res then
return nil
end
end
if rule['from'] then
if not from then
return nil
end
for _, i in ipairs(rule['from']) do
res = check_addr_setting(i, from)
if res then
break
end
end
if not res then
return nil
end
end
if rule['rcpt'] then
if not rcpt then
return nil
end
for _, i in ipairs(rule['rcpt']) do
res = check_addr_setting(i, rcpt)
if res then
break
end
end
if not res then
return nil
end
end
if rule['user'] then
if not user then
return nil
end
for _, i in ipairs(rule['user']) do
res = check_addr_setting(i, user)
if res then
break
end
end
if not res then
return nil
end
end
if res then
if rule['whitelist'] then
rule['apply'] = {whitelist = true}
end
return rule
end
return nil
end
-- Check if we have override as query argument
if check_query_settings(task) then
return
end
-- Do not waste resources
if not settings_initialized then
return
end
rspamd_logger.infox(task, "check for settings")
local ip = task:get_from_ip()
local from = task:get_from()
local rcpt = task:get_recipients()
local uname = task:get_user()
local user = {}
if uname then
user[1] = {}
for localpart, domainpart in string.gmatch(uname, "(.+)@(.+)") do
user[1]["user"] = localpart
user[1]["domain"] = domainpart
user[1]["addr"] = uname
break
end
if not user[1]["addr"] then
user[1]["user"] = uname
user[1]["addr"] = uname
end
end
-- Match rules according their order
for pri = max_pri,1,-1 do
if settings[pri] then
for name, r in pairs(settings[pri]) do
local rule = check_specific_setting(name, r, ip, from, rcpt, user)
if rule then
rspamd_logger.infox(task, "<%1> apply settings according to rule %2",
task:get_message_id(), name)
if rule['apply'] then
task:set_settings(rule['apply'])
end
if rule['symbols'] then
-- Add symbols, specified in the settings
each(function(val)
task:insert_result(val, 1.0)
end, rule['symbols'])
end
end
end
end
end
end
-- Process settings based on their priority
local function process_settings_table(tbl)
local get_priority = function(elt)
local pri_tonum = function(p)
if p then
if type(p) == "number" then
return tonumber(p)
elseif type(p) == "string" then
if p == "high" then
return 3
elseif p == "medium" then
return 2
end
end
end
return 1
end
return pri_tonum(elt['priority'])
end
-- Check the setting element internal data
local process_setting_elt = function(name, elt)
-- Process IP address
local function process_ip(ip)
local out = {}
if type(ip) == "table" then
for i,v in ipairs(ip) do
table.insert(out, process_ip(v))
end
elseif type(ip) == "string" then
local slash = string.find(ip, '/')
if not slash then
-- Just a plain IP address
local res = rspamd_ip.from_string(ip)
if res:is_valid() then
out[1] = res
out[2] = 0
else
rspamd_logger.errx(rspamd_config, "bad IP address: " .. ip)
return nil
end
else
local res = rspamd_ip.from_string(string.sub(ip, 1, slash - 1))
local mask = tonumber(string.sub(ip, slash + 1))
if res:is_valid() then
out[1] = res
out[2] = mask
else
rspamd_logger.errx(rspamd_config, "bad IP address: " .. ip)
return nil
end
end
else
return nil
end
return out
end
local function process_addr(addr)
local out = {}
if type(addr) == "table" then
for i,v in ipairs(addr) do
table.insert(out, process_addr(v))
end
elseif type(addr) == "string" then
local start = string.sub(addr, 1, 1)
if start == '/' then
-- It is a regexp
local re = rspamd_regexp.create(addr)
if re then
out['regexp'] = re
else
rspamd_logger.errx(rspamd_config, "bad regexp: " .. addr)
return nil
end
elseif start == '@' then
-- It is a domain if form @domain
out['domain'] = string.sub(addr, 2)
else
-- Check user@domain parts
local at = string.find(addr, '@')
if at then
-- It is full address
out['name'] = addr
else
-- It is a user
out['user'] = addr
end
end
else
return nil
end
return out
end
local check_table = function(elt, out)
if type(elt) == 'string' then
return {out}
end
return out
end
local out = {}
if elt['ip'] then
local ip = process_ip(elt['ip'])
if ip then
out['ip'] = check_table(elt['ip'], ip)
end
end
if elt['from'] then
local from = process_addr(elt['from'])
if from then
out['from'] = check_table(elt['from'], from)
end
end
if elt['rcpt'] then
local rcpt = process_addr(elt['rcpt'])
if rcpt then
out['rcpt'] = check_table(elt['rcpt'], rcpt)
end
end
if elt['user'] then
local user = process_addr(elt['user'])
if user then
out['user'] = check_table(elt['user'], user)
end
end
-- Now we must process actions
if elt['symbols'] then out['symbols'] = elt['symbols'] end
if elt['apply'] then
-- Just insert all metric results to the action key
out['apply'] = elt['apply']
elseif elt['whitelist'] or elt['want_spam'] then
out['whitelist'] = true
else
rspamd_logger.errx(rspamd_config, "no actions in settings: " .. name)
return nil
end
return out
end
settings_initialized = false
-- filter trash in the input
local ft = filter(
function(_, elt)
if type(elt) == "table" then
return true
end
return false
end, tbl)
-- clear all settings
max_pri = 0
local nrules = 0
for k,v in pairs(settings) do settings[k]={} end
-- fill new settings by priority
for_each(function(k, v)
local pri = get_priority(v)
if pri > max_pri then max_pri = pri end
if not settings[pri] then
settings[pri] = {}
end
local s = process_setting_elt(k, v)
if s then
settings[pri][k] = s
nrules = nrules + 1
end
end, ft)
settings_initialized = true
rspamd_logger.infox(rspamd_config, 'loaded %1 elements of settings', nrules)
return true
end
-- Parse settings map from the ucl line
local function process_settings_map(string)
local ucl = require "ucl"
local parser = ucl.parser()
local res,err = parser:parse_string(string)
if not res then
rspamd_logger.warnx(rspamd_config, 'cannot parse settings map: ' .. err)
else
local obj = parser:get_object()
if obj['settings'] then
process_settings_table(obj['settings'])
else
process_settings_table(obj)
end
end
return res
end
if set_section and set_section[1] and type(set_section[1]) == "string" then
-- Just a map of ucl
if not rspamd_config:add_map(set_section[1], "settings map", process_settings_map) then
rspamd_logger.errx(rspamd_config, 'cannot load settings from %1', set_section)
end
elseif set_section and type(set_section) == "table" then
process_settings_table(set_section)
end
rspamd_config:register_pre_filter(check_settings)
|