46 lines
1.0 KiB
Lua
46 lines
1.0 KiB
Lua
local dns = require "dns"
|
|
local ip_list = {%BLACKLIST_IP_LIST%}
|
|
local reverse_list = {%BLACKLIST_REVERSE_LIST%}
|
|
local ip = ngx.var.remote_addr
|
|
|
|
function ip_cached_ko ()
|
|
return ngx.shared.blacklist_ip_cache:get(ip) == "ko"
|
|
end
|
|
|
|
function reverse_cached_ko ()
|
|
return ngx.shared.blacklist_reverse_cache:get(ip) == "ko"
|
|
end
|
|
|
|
function ip_cached ()
|
|
return ngx.shared.blacklist_ip_cache:get(ip) ~= nil
|
|
end
|
|
|
|
function reverse_cached ()
|
|
return ngx.shared.blacklist_reverse_cache:get(ip) ~= nil
|
|
end
|
|
|
|
function check_ip ()
|
|
for k, v in ipairs(ip_list) do
|
|
if v == ip then
|
|
ngx.shared.blacklist_ip_cache:set(ip, "ko", 86400)
|
|
return true
|
|
end
|
|
end
|
|
ngx.shared.blacklist_ip_cache:set(ip, "ok", 86400)
|
|
return false
|
|
end
|
|
|
|
function check_reverse ()
|
|
local rdns = dns.get_reverse()
|
|
if rdns ~= "" then
|
|
for k, v in ipairs(reverse_list) do
|
|
if rdns:sub(-#v) == v then
|
|
ngx.shared.blacklist_reverse_cache:set(ip, "ko", 86400)
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
ngx.shared.blacklist_reverse_cache:set(ip, "ok", 86400)
|
|
return false
|
|
end
|