234 lines
8.2 KiB
Lua
234 lines
8.2 KiB
Lua
-- WAF Action
|
|
require "config"
|
|
require "lib"
|
|
|
|
-- args
|
|
local rulematch = ngx.re.find
|
|
local unescape = ngx.unescape_uri
|
|
|
|
-- allow white ip
|
|
function white_ip_check()
|
|
if config_white_ip_check == "on" then
|
|
local IP_WHITE_RULE = get_rule("whiteip")
|
|
local WHITE_IP = get_client_ip()
|
|
if IP_WHITE_RULE ~= nil then
|
|
for _,rule in pairs(IP_WHITE_RULE) do
|
|
if rule ~= "" and rulematch(WHITE_IP,rule,"jo") then
|
|
log_record("White_IP",ngx.var.request_uri,"_","_")
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
-- deny black ip
|
|
function black_ip_check()
|
|
if config_black_ip_check == "on" then
|
|
local IP_BLACK_RULE = get_rule("blackip")
|
|
local BLACK_IP = get_client_ip()
|
|
if IP_BLACK_RULE ~= nil then
|
|
for _,rule in pairs(IP_BLACK_RULE) do
|
|
if rule ~= "" and rulematch(BLACK_IP,rule,"jo") then
|
|
-- log_record('BlackList_IP',ngx.var.request_uri,"_","_")
|
|
if config_waf_enable == "on" then
|
|
ngx.header.content_type = "text/html"
|
|
ngx.say('Your IP blacklist, Please contact the administrator! ')
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
-- allow white url
|
|
function white_url_check()
|
|
if config_white_url_check == "on" then
|
|
local URL_WHITE_RULES = get_rule("whiteurl")
|
|
local REQ_URI = string.lower(ngx.var.request_uri)
|
|
if URL_WHITE_RULES ~= nil then
|
|
for _,rule in pairs(URL_WHITE_RULES) do
|
|
if rule ~= "" then
|
|
local REQ_URI_LEN = string.len(REQ_URI)
|
|
local rule_str = string.sub(rule,1,2)
|
|
local from, to, err = rulematch(REQ_URI,string.lower(rule),"jo")
|
|
if rule_str == "\\." then
|
|
local wfrom, wto, werr = rulematch(REQ_URI,"%?","jo")
|
|
if from and REQ_URI_LEN == to and wfrom == nil then
|
|
return true
|
|
end
|
|
elseif from and rule_str == "\\/" and from == 1 then
|
|
return true
|
|
elseif from and from == 2 then
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
-- deny cc attack
|
|
function cc_attack_check()
|
|
if config_cc_check == "on" then
|
|
local USER_AGENT = get_user_agent()
|
|
local ARGS = ngx.var.args or ""
|
|
|
|
local ATTACK_URL = ngx.var.host .. ngx.var.uri
|
|
-- local ATTACK_URL = ngx.var.host .. ngx.var.request_uri
|
|
-- local ATTACK_URL = ngx.var.host .. ngx.var.uri .. '?' .. ARGS
|
|
|
|
local CC_TOKEN = get_client_ip() .. "." .. ngx.md5(string.lower(ATTACK_URL) .. USER_AGENT)
|
|
local limit = ngx.shared.limit
|
|
local CCcount=tonumber(string.match(config_cc_rate,'(.*)/'))
|
|
local CCseconds=tonumber(string.match(config_cc_rate,'/(.*)'))
|
|
local req,_ = limit:get(CC_TOKEN)
|
|
if req then
|
|
-- write("/data/wwwlogs/info.log",CC_TOKEN .."\t".. ATTACK_URL .. "\t".. "req: " .. req .."\n")
|
|
if req > CCcount then
|
|
log_record("CC_Attack",ngx.var.request_uri,"-","-")
|
|
log_post_http("CC_Attack",ngx.var.request_uri,"-","-")
|
|
if config_waf_enable == "on" then
|
|
local source = ngx.encode_base64(ngx.var.scheme.."://"..ngx.var.host..ngx.var.request_uri)
|
|
local dest = '/captcha-waf.html' .. '?continue=' .. source
|
|
local CCcountcode,_ = math.modf(CCcount/2);
|
|
limit:set(CC_TOKEN,CCcountcode)
|
|
ngx.redirect(dest,302)
|
|
end
|
|
else
|
|
limit:incr(CC_TOKEN,1)
|
|
end
|
|
else
|
|
limit:set(CC_TOKEN,1,CCseconds)
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
-- deny cookie
|
|
function cookie_attack_check()
|
|
if config_cookie_check == "on" then
|
|
local COOKIE_RULES = get_rule("cookie")
|
|
local USER_COOKIE = ngx.var.http_cookie
|
|
if USER_COOKIE ~= nil then
|
|
for _,rule in pairs(COOKIE_RULES) do
|
|
if rule ~="" and rulematch(string.lower(USER_COOKIE),string.lower(rule),"jo") then
|
|
log_record("Deny_Cookie",ngx.var.request_uri,"-",rule)
|
|
if config_waf_enable == "on" then
|
|
waf_output()
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
-- deny url
|
|
function url_attack_check()
|
|
if config_url_check == "on" then
|
|
local URL_RULES = get_rule("blackurl")
|
|
local REQ_URI = ngx.var.request_uri
|
|
for _,rule in pairs(URL_RULES) do
|
|
if rule ~="" and rulematch(string.lower(REQ_URI),string.lower(rule),"jo") then
|
|
log_record("Deny_URL",REQ_URI,"-",rule)
|
|
if config_waf_enable == "on" then
|
|
waf_output()
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
-- deny url args
|
|
function url_args_attack_check()
|
|
if config_url_args_check == "on" then
|
|
local ARGS_RULES = get_rule('args')
|
|
for _,rule in pairs(ARGS_RULES) do
|
|
--local REQ_ARGS = ngx.req.get_uri_args()
|
|
local REQ_ARGS, err = ngx.req.get_uri_args()
|
|
if err == "truncated" then
|
|
log_record("URL_ARGS_MANY",ngx.var.request_uri,"-",rule)
|
|
if config_waf_enable == "on" then
|
|
waf_output()
|
|
return true
|
|
end
|
|
end
|
|
for key, val in pairs(REQ_ARGS) do
|
|
if type(val) == "table" then
|
|
ARGS_DATA = string.lower(table.concat(val, " "))
|
|
else
|
|
ARGS_DATA = string.lower(val)
|
|
end
|
|
if ARGS_DATA and type(ARGS_DATA) ~= "boolean" and rule ~="" and rulematch(unescape(ARGS_DATA),string.lower(rule),"jo") then
|
|
log_record("Deny_URL_Args",ngx.var.request_uri,"-",rule)
|
|
if config_waf_enable == "on" then
|
|
waf_output()
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
-- deny user agent
|
|
function user_agent_attack_check()
|
|
if config_user_agent_check == "on" then
|
|
local USER_AGENT_RULES = get_rule("useragent")
|
|
local USER_AGENT = ngx.var.http_user_agent
|
|
if USER_AGENT ~= nil then
|
|
for _,rule in pairs(USER_AGENT_RULES) do
|
|
if rule ~="" and rulematch(string.lower(USER_AGENT),string.lower(rule),"jo") then
|
|
log_record("Deny_USER_AGENT",ngx.var.request_uri,"-",rule)
|
|
if config_waf_enable == "on" then
|
|
waf_output()
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
-- deny post
|
|
function post_attack_check()
|
|
if config_post_check == "on" then
|
|
local POST_RULES = get_rule("post")
|
|
ngx.req.read_body()
|
|
for _,rule in pairs(POST_RULES) do
|
|
-- local REQ_POST = ngx.req.get_post_args()
|
|
local REQ_POST, err = ngx.req.get_post_args()
|
|
if err == "truncated" then
|
|
log_record("DENY_POST_MANY",ngx.var.request_uri,"-",rule)
|
|
if config_waf_enable == "on" then
|
|
waf_output()
|
|
return true
|
|
end
|
|
end
|
|
for key, val in pairs(REQ_POST) do
|
|
if type(val) == "table" then
|
|
POST_DATA = string.lower(table.concat(val, " "))
|
|
else
|
|
POST_DATA = string.lower(val)
|
|
end
|
|
if POST_DATA and type(POST_DATA) ~= "boolean" and rule ~="" and rulematch(unescape(POST_DATA),string.lower(rule),"jo") then
|
|
log_record("Deny_POST",ngx.var.request_uri,"-",rule)
|
|
if config_waf_enable == "on" then
|
|
waf_output()
|
|
return true
|
|
end
|
|
end
|
|
end
|
|
end
|
|
return true
|
|
end
|
|
return false
|
|
end
|