import re import subprocess import time log_file = '/var/log/suricata/fast.log' ip_set = set() while True: with open(log_file) as file: file.seek(0, 2) # 将文件指针移动到文件末尾 for line in file: match = re.search(r'\d+\.\d+\.\d+\.\d+', line) if not match: continue ip = match.group() if ip in ip_set: continue result = subprocess.call(['iptables', '-A', 'INPUT', '-s', ip, '-j', 'DROP']) if result == 0: print(f'封禁IP {ip} 成功!') ip_set.add(ip) else: print(f'封禁IP {ip} 失败!') time.sleep(5) 这段代码中并没有对Priority级别 <=2 级的预警进行主动响应,封禁ip,优化修改一下
时间: 2023-12-11 16:05:56 浏览: 29
可以考虑在程序中增加一个判断条件,来限制只对Priority级别大于2的预警进行主动响应封禁ip的操作。具体修改代码如下:
```
import re
import subprocess
import time
log_file = '/var/log/suricata/fast.log'
ip_set = set()
while True:
with open(log_file) as file:
file.seek(0, 2) # 将文件指针移动到文件末尾
for line in file:
match = re.search(r'\d+\.\d+\.\d+\.\d+', line)
if not match:
continue
ip = match.group()
priority = re.search(r'(\[.*?\])', line).group()
if ip in ip_set:
continue
if 'Priority: 1' in priority or 'Priority: 2' in priority:
continue
result = subprocess.call(['iptables', '-A', 'INPUT', '-s', ip, '-j', 'DROP'])
if result == 0:
print(f'封禁IP {ip} 成功!')
ip_set.add(ip)
else:
print(f'封禁IP {ip} 失败!')
time.sleep(5)
```
修改后的代码中,增加了一个对Priority级别的判断条件,只有当Priority级别大于2时,才会进行主动响应封禁ip的操作。这样可以避免将一些无关紧要的预警误判为攻击行为,从而减少不必要的封禁操作。