bunkerweb/lua/dnsbl.lua

33 lines
724 B
Lua

local M = {}
local dns = require "dns"
local dnsbls = {%DNSBL_LIST%}
local ip = ngx.var.remote_addr
function M.cached_ko ()
return ngx.shared.dnsbl_cache:get(ip) == "ko"
end
function M.cached ()
return ngx.shared.dnsbl_cache:get(ip) ~= nil
end
function M.check ()
local rip = dns.ip_to_arpa()
for k, v in ipairs(dnsbls) do
local req = rip .. "." .. v
local ips = dns.get_ips(req)
for k2, v2 in ipairs(ips) do
local a,b,c,d = v2:match("([%d]+).([%d]+).([%d]+).([%d]+)")
if a == "127" then
ngx.shared.dnsbl_cache:set(ip, "ko", 86400)
ngx.log(ngx.NOTICE, "ip " .. ip .. " is in DNSBL " .. v)
return true
end
end
end
ngx.shared.dnsbl_cache:set(ip, "ok", 86400)
return false
end
return M