天天看点

python攻击脚本执行命令_python 执行shell 命令,自动化添加攻击IP地址到iptables

通过python执行shell命令的方法有4种,在这里介绍一种常用的。

os.system、 os.popen、 commands、 subprocess

接下来介绍subprocess的使用

通过python 日志分析,获取到攻击源IP地址,收集写入到mysql数据库中

mysql如下:

python攻击脚本执行命令_python 执行shell 命令,自动化添加攻击IP地址到iptables

iptables.py脚本

#!/usr/bin/env python

# -*- coding:utf-8 -*-

import os

import MySQLdb

import subprocess

try:

connection = MySQLdb.connect(user=‘soul‘,passwd=‘soul‘,host=‘127.0.0.1‘,db=‘iptable‘,port=3306,use_unicode=1,charset="utf8")

cursor = connection.cursor()

cursor.execute(‘select * from iptables‘)

data = cursor.fetchall()

cursor.close()

connection.close()

except MySQLdb.Error,err_msg:

print ‘MySQLdb error msg:‘ ,err_msg

def insert():

for info in data:

ipaddress = info[1]

print ‘成功插入IP黑名单:%s‘ %ipaddress

subprocess.call([‘/sbin/iptables -I INPUT -s %s -p tcp --dport 8000 -j ACCEPT‘ % ipaddress],  shell=True)

def save():

subprocess.call([‘service iptables save‘], shell=True)

if __name__ == ‘__main__‘:

insert()

save()

python iptables.py

result:

python攻击脚本执行命令_python 执行shell 命令,自动化添加攻击IP地址到iptables

原文:http://soul455879510.blog.51cto.com/6180012/1943514