修改代码

waf.lua

@@ -85,17 +85,19 @@ u_agent = ngx.req.get_headers().user_agent
 
 
 
 
 -- 白名单存在直接跳过
 -- 白名单存在直接跳过
-if wafcheck.WhiteIp(user_ip) == 'exist' or wafcheck.WhiteUrl(temp_uri) == 'exist' or wafcheck.WhiteHeader(user_ip) == 'exist' then
+if wafcheck.WhiteIp(user_ip) == 'exist' or wafcheck.WhiteUrl(temp_uri) == 'exist' or wafcheck.WhiteHeader(u_agent) == 'exist' then
 	return 
 	return 
 end
 end
 
 
 
 
+
 ngx.say(wafcheck.BlackIp(user_ip))
 ngx.say(wafcheck.BlackIp(user_ip))
+red:close()
 ngx.exit(ngx.HTTP_FORBIDDEN)
 ngx.exit(ngx.HTTP_FORBIDDEN)
 ngx.say(1)
 ngx.say(1)
 
 
 -- 黑名单存在直接302
 -- 黑名单存在直接302
-if wafcheck.BlackIp(user_ip) == 'exist' or wafcheck.BlackUrl(temp_uri) == 'exist' or wafcheck.BlackHeader(user_ip) == 'exist' then
+if wafcheck.BlackIp(user_ip) == 'exist' or wafcheck.BlackUrl(temp_uri) == 'exist' or wafcheck.BlackHeader(u_agent) == 'exist' then
 	
 	
 	ngx.say(wafcheck.BlackIp(user_ip))
 	ngx.say(wafcheck.BlackIp(user_ip))
 	ngx.exit(ngx.HTTP_FORBIDDEN)
 	ngx.exit(ngx.HTTP_FORBIDDEN)