用命令查看web连接过高的IP地址,但是需要人工智能去封,太麻烦了,直接写个脚本自动解决。web服务器是用nginx,python为2.6
首先在nignx的config中建立空文件deny.ip, 然后在nginx.conf 的http标签中添加“include deny.ip;”。在nginx下sbin的目录中放入自动脚本。脚本可以查到连接最大的IP,并插入屏蔽列表中,验证正确性后导入配置。全部完成或者出错后发送邮件。被封ip再次访问会报403错误,如果不希望报错可以跳转到其它页面。源码如下:
check_deny_up.py 启动i脚本
#!/bin/python #-*- coding:utf-8 -*- # Filename: main.py # Revision: 1.0 # Date: 2012-06-20 # Author: simonzhang # web: www.simonzhang.net # Email: simon-zzm@163.com ### END INIT INFO importos fromstring importstrip fromemail.mime.text importMIMEText importsmtplib #### check_comm ="/bin/netstat -antp|grep :80|awk ' ''{print $5}'|awk -F: '{print $1}'|sort -r|uniq -c|sort -n -k1 -r" max_ip =100 mail_host =‘’; mail_user =‘’; mail_pwd =‘’; mail_to =‘’; mail_cc =‘’; defreboot_nginx_sendmail(ip_list): #### reboot nginx _get_check_confile =os.popen('./nginx -t').readlines() ifstr(_get_check_confile.find('ok')) !='-1': os.system('./nginx -s reload') _mail_content =ip_list else: _mail_content ='Error' #### send mail msg =MIMEText(_mail_content) msg['From'] =mail_user msg['Subject'] =' force ip.' msg['To'] =mail_to try: s =smtplib.SMTP() s.connect(mail_host) s.login(mail_user, mail_pwd) s.sendmail(mail_user, [mail_to, mail_cc], msg.as_string()) s.close() exceptException, e: printe #### force out IP defforce_out(_deny_ip): _write_status =0 _read_force_file =open('../conf/deny.ip', 'rb').read() ifstr(_read_force_file.find(_deny_ip)) =='-1': try: _get_force_file =open('../conf/deny.ip', 'ab') _get_force_file.write('deny %s ;\n'%_deny_ip) _get_force_file.close() _write_status =1 return_write_status except: return_write_status reboot_nginx_sendmail("Error !") return_write_status defmain(): get_high_ip =os.popen('%s'%check_comm).readlines() _count_force_ip =0 _force_ip_list ='' fori inxrange(3): try: _get_count =strip(get_high_ip[i]).split(' ')[0] _get_ip =strip(strip(get_high_ip[i]).split(' ')[1]) except: _get_count =0 _get_ip ='' # Maximum connection IP is Beyond the limit value if(int(_get_count) > max_ip) and(len(_get_ip) > 0): force_ip =_get_ip _get_status =force_out(force_ip) # check maximum is added in the deny.ip file ifstr(_get_status) =='1': _count_force_ip +=1 _force_ip_list +=' %s '%force_ip # if _count_force_ip > 0: # reboot_nginx_sendmail(_force_ip_list) if__name__ =='__main__': main()
check_deny_up.sh
#! /bin/bash # # make simon-zzm@163.com # # ### END INIT INFO # Source function library. . /etc/profile cd/Data/apps/nginx/sbin/ # See how we were called. case"$1"in start) /usr/local/bin/pythoncheck_ip_deny.py ;; *) echo$"Usage: $0 {start}" exit1 esac exit
将启动脚本放在crontab中运行。