监控服务器的脚本log_agent

监控服务器脚本：

将恶意攻击IP地址加入黑名单

1、分割日志

使用os.system 执行操作系统命令，使用重定向来分割日志

2.获取访问ip

读日志文件，获取访问ip记录，使用字符串.split来获取访问ip

3.将恶意攻击ip加入黑名单

把所有访问ip加入到list，判断次数，如果超过200，加入黑名单os.system('iptables -I INPUT 1-p tcp -s %s -j DROP'%)

4.每分钟执行一次

time.sleep(60)

5.循环

while true

time模块:

import time
time_now=time.strftime('%Y%m%d%H%M%S')
print(time_now)

import os, time

while True:
    ip_list = []  # 存放所有ip的列表
    src_log = '/opt/lampp/logs/access_log'  # 源日志
    new_log = '/opt/lampp/logs/access_log_%s' % time.strftime('%Y%m%d%H%M')
    # 新日志的文件名
    os.system('cat {s_log} > {n_log} && > {s_log}'.format(s_log=src_log, n_log=new_log))  # 分割每分钟的日志
    with open(new_log) as fr:  # 打开日志文件
        for line in fr:  # 循环文件每一行的内容
            ip = line.split('-')[0].strip()  # 分割字符串，获取ip地址
            ip_list.append(ip)  # 把所有的ip地址加入到iplist中
    for ip in ip_list:  # 再循环整个iplist
        if ip_list.count(ip) > 20:  # 判断ip出现次数是否大于2的加入黑名单
            os.system('iptables -I INPUT 1 -p tcp -s %s  -j DROP' % ip)
            # 执行操行系统命令，把ip地址加入黑名单
            for i in range(ip_list.count(ip)):
                ip_list.remove(ip)  # 从ip_list中把已经加入黑名单的ip全部删掉
    time.sleep(60)

将监控服务器脚本做成一个功能函数，直接调用，入参是日志路径

#监控访问日志函数
import os, time
def log_agent(log_name):
        while True:
            ip_list = []  # 存放所有ip的列表
            src_log = log_name  # 源日志
            new_log = '%s,%s'%(src_log) % time.strftime('%Y%m%d%H%M')
            # 新日志的文件名
            os.system('cat {s_log} > {n_log} && > {s_log}'.format(s_log=src_log, n_log=new_log))  # 分割每分钟的日志
            with open(new_log) as fr:  # 打开日志文件
                for line in fr:  # 循环文件每一行的内容
                    ip = line.split('-')[0].strip()  # 分割字符串，获取ip地址
                    ip_list.append(ip)  # 把所有的ip地址加入到iplist中
            for ip in ip_list:  # 再循环整个iplist
                if ip_list.count(ip) > 20:  # 判断ip出现次数是否大于2的加入黑名单
                    os.system('iptables -I INPUT 1 -p tcp -s %s  -j DROP' % ip)
                    # 执行操行系统命令，把ip地址加入黑名单
                    for i in range(ip_list.count(ip)):
                        ip_list.remove(ip)  # 从ip_list中把已经加入黑名单的ip全部删掉
            time.sleep(60)
log_agen('日志路径')