python统计tomcat访问量高的ip并写到配置文件封禁

zephyrss

今天刚出炉的小脚本，眼下有这么个需求，在一台win2003服务器上跑了一个tomcat网站，经常被人爬，一时找不到好的解决方法，就在网上搜罗相关内容，最后凝聚成了这个python小脚本。
大致思路就是读取日志，截取统计ip，判断ip访问量大于一个数，然后给写到tomcat的配置文件里那条封禁ip的参数，最后重启tomcat。
然后写个计划任务和bat就搞定了
格式乱了。贴个图吧。





import osimport linecacheimport datetime
#2019-11-12 #zephyrss
dateTime_p = datetime.datetime.now()  
date_p = dateTime_p.date()
date_p = str(date_p)
def shutdown():
print("tomcat正在重启，请稍等...")
os.system("d:\i.bat")
print("关闭tomcat成功")#shutdown()
ips = []
fr = open('d:\\'+date_p+'.txt')
fr.seek(0)for line in fr:   
  ip = line.split()[0]    
ips.append(ip)
new_ips = set(ips)
for new_ip in new_ips:   
if ips.count(new_ip)>200:         
  var1=str(new_ip)         
  with open("d:\server.xml", 'r') as f:         
   lines = f.readlines()     
        lines.insert(28, '<Valve className=\"org.apache.catalina.valves.RemoteAddrValve\" deny=\"' + var1 + '\"/>\n')             #lines.insert(28, var1)      
      s = ''.join(lines).strip('\n')           
  with open("d:\server.xml", 'w') as f:         
   f.write(s)           
  shutdown()     
        print('程序结束')

whh19921024

重启tomcat不担心业务频繁中断？ 还是建议拒绝策略加到防火墙上面

zephyrss

whh19921024 发表于 2019-11-13 09:59
重启tomcat不担心业务频繁中断？ 还是建议拒绝策略加到防火墙上面

防火墙服务起不来。估计系统有问题了。。。临时解决吧。