学习python中,写了一个简单预防攻击脚本,感觉不好,mark下待留以后改进。#!/bin/env python
#-*- coding:utf-8 -*-
import sqlite3
import commands
import time
import logging
log_file=‘/var/log/ddoskill.log‘
logging.basicConfig(level=logging.INFO,format=‘%(asctime)s %(filename)s[line:%(lineno)d]%(levelname)s %(message)s‘,datefmt=‘%a,%d %b %Y %H:%M:%S‘,filename=log_file,filemode=‘a‘)
exclude_list = [‘192.168.1.56‘,‘192.168.1.200‘,‘192.168.1.100‘,‘192.168.1.300‘,‘127.0.0.1‘]
cx = sqlite3.connect(‘/tmp/ddoskill.db‘)
#查看系统防火墙是否开启
(status_4,output_4) = commands.getstatusoutput("service iptables status")
if status_4 != 0:
logging.error("iptables is closed!")
exit(100)
#取出数据库中已有IP存入ip列表中
ip_list = []
out_list = cx.execute("select ip from ddos").fetchall()
i = 0
while i < len(out_list):
ip_list.append(str(out_list[i][0]))
i+=1
#将连接数过大且不存在于数据库中的IP禁掉
command_1="netstat -n|awk ‘/^tcp/{print $5}‘|cut -d: -f1|sort|uniq -c"
output_1 = commands.getoutput(command_1)
length = len(output_1.split(‘\n‘))
x = 0
while x < length:
num = output_1.split(‘\n‘)[x].split()[0]
IP = str(output_1.split(‘\n‘)[x].split()[1])
if int(num) >= 100 and IP not in ip_list and IP not in exclude_list:
logging.warning("将 %s 写进数据库,并在iptable禁止访问!" % IP)
command_3 = "iptables -I INPUT -s "+IP+" -j DROP"
output_3 = commands.getoutput(command_3)
cx.execute("insert into ddos(ip) values(?)",(IP,))
x+=1
#删除列表中5小时之前的数据,并同时删除iptable相应条目
for ip in ip_list:
select_com ="select time from ddos where ip=‘%s‘" % ip
otime = str(cx.execute(select_com).fetchone()[0])
intv = time.time() - time.mktime(time.strptime(otime,‘%Y-%m-%d %H:%M:%S‘))
if intv/60/60 > 5:
logging.warning("从iptables和数据库中删除:%s" % ip)
command_2 = "iptables -D INPUT -s "+ip+" -j DROP"
output_2 = commands.getoutput(command_2)
delete_com = "delete from ddos where ip=‘%s‘" % ip
cx.execute(delete_com)
cx.commit()
cx.close()
原文地址:http://ganjiangpeng.blog.51cto.com/1037417/1640367