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
|
local rspamd_ip = require 'rspamd_ip'
local rspamd_logger = require 'rspamd_logger'
local lua_maps = require "lua_maps"
local radix_map = rspamd_config:add_map ({
url = rspamd_env.RADIX_MAP,
type = 'radix',
})
local map_map = rspamd_config:add_map ({
url = rspamd_env.MAP_MAP,
type = 'map',
})
local regexp_map = rspamd_config:add_map ({
url = rspamd_env.REGEXP_MAP,
type = 'regexp',
})
rspamd_config:register_symbol({
name = 'RADIX_KV',
score = 1.0,
callback = function()
local sip = {'8.8.8.8', '::1', '192.168.1.1', '10.0.1.1'}
local expected = {'test one', 'another', '1', false}
for i = 1, #sip do
if (radix_map:get_key(rspamd_ip.from_string(sip[i])) ~= expected[i]) then
local rip = rspamd_ip.from_string(sip[i])
local val = radix_map:get_key(rip)
return true, rspamd_logger.slog('plain: get_key(%s) [%s] -> %s [%s] [expected %s]', rip, type(rip), val, type(val), expected[i])
end
if (radix_map:get_key(sip[i]) ~= expected[i]) then
local val = radix_map:get_key(sip[i])
return true, rspamd_logger.slog('string: get_key(%s) [%s] -> %s [%s] [expected %s]', sip[i], type(sip[i]), val, type(val), expected[i])
end
end
return true, 'no worry'
end
})
rspamd_config:register_symbol({
name = 'MAP_KV',
score = 1.0,
callback = function()
local str = {'foo', 'asdf.example.com', 'asdf', 'barf'}
local expected = {'bar', 'value', '', false}
for i = 1, #str do
if (map_map:get_key(str[i]) ~= expected[i]) then
local val = map_map:get_key(str[i])
return true, rspamd_logger.slog('get_key(%s) [%s] -> %s [%s] [expected %s]', str[i], type(str[i]), val, type(val), expected[i])
end
end
return true, 'no worry'
end,
})
rspamd_config:register_symbol({
name = 'REGEXP_KV',
score = 1.0,
callback = function()
local str = {'foo', 'asdf.example.com', 'asdf', 'barf'}
local expected = {'bar', 'value', '1', false}
for i = 1, #str do
if (regexp_map:get_key(str[i]) ~= expected[i]) then
local val = regexp_map:get_key(str[i])
return true, rspamd_logger.slog('get_key(%s) [%s] -> %s [%s] [expected %s]', str[i], type(str[i]), val, type(val), expected[i])
end
end
return true, 'no worry'
end,
})
local simple_ext_map = lua_maps.map_add_from_ucl({
external = true,
backend = "http://localhost:18080/map-simple",
method = "body",
encode = "json",
}, '', 'external map')
rspamd_config:register_symbol({
name = 'EXTERNAL_MAP',
score = 1.0,
callback = function(task)
local function cb(res, data, code)
if res then
task:insert_result('EXTERNAL_MAP', string.format('+%s', data))
else
task:insert_result('EXTERNAL_MAP', string.format('-%s:%s', code, data))
end
end
simple_ext_map:get_key({
key = "value",
}, cb, task)
end,
})
|