openresty+lua+redis把非正常访问的域名加入黑名单

一、验证lua geoIp2是否缺少依靠

1、执行下令

2. /usr/local/openresty/bin/opm get anjia0532/lua-resty-maxminddb

复制代码

执行安装下令报错,缺少Digest/MD5依靠
2、Digest/MD5依靠

2. yum -y install perl-Digest-MD5

复制代码

GeoIP2 lua库依靠动态库安装，lua库依靠libmaxminddb实现对mmdb的高效访问
二、安装libmaxminddb数据库

2. cd /data/file/
3. wget https://github.com/maxmind/libmaxminddb/releases/download/1.12.2/libmaxminddb-1.12.2.tar.gz
4. tar -zxf libmaxminddb-1.12.2.tar.gz
5. cd libmaxminddb-1.12.2
6. ./configure
7. make && make install
9. sh -c "echo /usr/local/lib  >> /etc/ld.so.conf.d/local.conf"
10. ldconfig

复制代码

三、编写操作geoIp的lua脚本

2. --引入Redis
3. local redis = require "resty.redis";
4. --引入rsa
5. local rsa = require "resty.rsa";
6. --引入maxminddb库
7. local geo=require 'resty.maxminddb';
8. --Redis链接ip
9. local ip = "192.168.0.145"
10. --Redis链接端口
11. local port = 6379
12. --Redis连接密码
13. local pass = "aa123456"
14. --鉴权Redis
15. local function connAuth()
16.         local red = redis:new();
17.         local connCount, err = red:connect(ip, port);
18.         if not connCount then
19.                 ngx.say("failed to connect: ", err)
20.                 close_redis(red)
21.                 return
22.         end
23.         red:set_timeouts(2000);
24.         local ok, err = red:auth(pass)
25.         if not ok then
26.                 ngx.say("failed to auth: ", err)
27.                 close_redis(red)
28.                 return
29.         end
30.         return red
31. end
33. --关闭Redis
34. local function close_redis(red)
35.     if not red then
36.         return
37.     end
38.     --释放连接(连接池实现)
39.     local pool_max_idle_time = 10000
40.     local pool_size = 100
41.     local ok, err = red:set_keepalive(pool_max_idle_time, pool_size)
42.     if not ok then
43.         ngx.say("set keepalive error : ", err)
44.     end
45. end
47. --获取请求者IP
48. local function getIp()
49.     local clientIP = ngx.req.get_headers()["X-Real-IP"]
50.     if clientIP == nil then
51.         clientIP = ngx.req.get_headers()["x_forwarded_for"]
52.     end
53.     if clientIP == nil then
54.         clientIP = ngx.var.remote_addr
55.     end
56.     return clientIP
57. end
58. --获取用户访问IP
59. local clientIP = getIp();
60. --封禁token时间（秒）
61. local token_block_time= 120
62. --指定token访问频率计数最大值（次）
63. local token_max_count = 3
65. --如果数据为空的情况下
66. local function is_empty(value)
67.     if type(value) == "table" then
68.         return next(value) == nil
69.     elseif type(value) == "string" then
70.         return #value == 0
71.     elseif type(value) == "nil" then
72.         return true
73.     end
74.         return false
75. end
77. --过滤特殊字符串(只保留字母与数字,字母不区分大小写)
78. local function filter_special_chars(s)
79.     local ss = {}
80.     local k = 1
81.     while true do
82.         if k > #s then break end
83.         local c = string.byte(s,k)
84.         if not c then break end
85.         if c<192 then
86.             if (c>=48 and c<=57) or (c>= 65 and c<=90) or (c>=97 and c<=122) then
87.                 table.insert(ss, string.char(c))
88.             end
89.             k = k + 1
90.         end
91.     end
92.     return table.concat(ss)
93. end
95. if not geo.initted() then
96.         geo.init("/usr/share/GeoIP/GeoLite2-Country.mmdb")
97. end
99. --如果头部信息没有指定的参数或是指定参数的值无法解析,加入IP黑名单
100. --如果同样的头部参数键在封禁token时间内连续访问指定token访问频率计数最大值次以上,加入IP黑名单
101. local function set_blacklist()
102.     local header = ngx.req.get_headers();
103.     local red = connAuth();
104.         -- 检测ip
105.     local res,err=geo.lookup(clientIP)
106.     if not res then
107.             local res, err =  red:sismember('black-list', clientIP);
108.         if res ~= 1 then
109.             red:sadd('black-list', clientIP)
110.         end
111.         close_redis(red)
112.         return ngx.exit(401)
113.     else
114.         for k,v in pairs(res) do
115.             --只获取国家
116.             if(k == "country") then
117.                 --获取国家编码
118.                 for key,item in pairs(v) do
119.                     if (key=="iso_code" and "CN" ~= item) then
120.                         local res, err =  red:sismember('black-list', clientIP);
121.                         if res ~= 1 then
122.                             red:sadd('black-list', clientIP)
123.                         end
124.                         close_redis(red)
125.                         return ngx.exit(401)
126.                     end
127.                 end
128.             end
129.         end
130.     end
131. end
133. -- 查看是否在黑名单里面
134. local function get_blacklist()
135.     local red = connAuth();
136.     local res, err =  red:sismember('black-list', clientIP);
137.     if res == 1 then
138.         close_redis(red);
139.         return ngx.exit(401);
140.     end
141.     close_redis(red);
142. end
144. get_blacklist();
145. set_blacklist();

复制代码

四、nginx域名配置引用

2. location / {
3.         access_by_lua_file /data/wwwroot/rsa-redis.lua;
4.         try_files $uri $uri/ /index.html;
5.     }

复制代码

在你必要应用的地方加上就行

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。

继续阅读请点击广告