import re import subprocess import time while True: with open('/var/log/suricata/fast.log', 'r') as f: for line in f: # 使用正则表达式匹配IP地址 match = re.search(r'\d+\.\d+\.\d+\.\d+', line) if not match: continue ip = match.group() # 调用iptables命令封禁IP result = subprocess.call(['iptables', '-A', 'INPUT', '-s', ip, '-j', 'DROP']) if result == 0: print(f'封禁IP {ip} 成功!') else: print(f'封禁IP {ip} 失败!') time.sleep(5) # 每隔5秒读取一次日志 将这两行代码加入到上面代码中,应该如何修改优化file = open('/var/log/suricata/eve.json') file.seek(0, 2)
时间: 2023-12-11 19:03:06 浏览: 90
adworld-wargame:我对https://adworld.xctf.org.cntask上的pwn挑战的利用
可以将文件操作部分放入一个函数中,每次读取完文件后关闭文件,避免文件句柄泄漏。同时,可以使用 `time.sleep()` 函数控制读取日志的时间间隔。修改后代码如下:
```python
import re
import subprocess
import time
def process_log_file():
# 打开日志文件
with open('/var/log/suricata/fast.log', 'r') as f:
for line in f:
# 使用正则表达式匹配IP地址
match = re.search(r'\d+\.\d+\.\d+\.\d+', line)
if not match:
continue
ip = match.group()
# 调用iptables命令封禁IP
result = subprocess.call(['iptables', '-A', 'INPUT', '-s', ip, '-j', 'DROP'])
if result == 0:
print(f'封禁IP {ip} 成功!')
else:
print(f'封禁IP {ip} 失败!')
while True:
process_log_file()
time.sleep(5) # 每隔5秒读取一次日志
# 打开日志文件
with open('/var/log/suricata/eve.json', 'r') as file:
file.seek(0, 2)
# 在文件结尾处添加新内容
# ...
```
阅读全文