python 一个简单防攻击脚本

学习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