使用Python3循环在Firewalld中添加丰富的规则

时间:2018-03-09 01:58:48

标签: ubuntu-14.04 python-3.4 firewalld

我试图使用Python3迭代IP地址列表,然后使用firewalld阻止它们。

注意:我是Python的新手,所以请原谅任何简单的错误。

import subprocess

with open("ips.txt") as ipList:
ips = ipList.readlines()

for ip in ips:
    process = subprocess.Popen(['firewall-cmd',
                            '--permanent',
                            '--add-rich-rule=\'rule family=\"ipv4\" source address=\"{0}\" reject\''.format(ip.rstrip())
                            ])

我使用format.rstrip删除列表中每个IP地址后的换行符。

运行脚本时,我收到以下错误;

root@mediaserver:~# python3 block.py 
Error: INVALID_RULE: internal error in _lexer(): rule family="ipv4" source address="1.56.0.0/13" reject
Error: INVALID_RULE: internal error in _lexer(): rule family="ipv4" source address="1.48.0.0/15" reject

此错误消息遍历列表中的所有IP块。

如果我在脚本之外运行firewall-cmd,则不会收到任何错误消息,并且规则已正确添加。

root@mediaserver:~# firewall-cmd --permanent --add-rich-rule='rule family="ipv4" source address="223.198.0.0/15" reject'
success
root@mediaserver:~# firewall-cmd --reload
success
root@mediaserver:~# firewall-cmd --zone=public --list-all
public (default, active)
  interfaces: eth0
  sources: 
  services: dhcpv6-client ssh
  ports: 
  masquerade: no
  forward-ports: 
  icmp-blocks: 
  rich rules: 
       rule family="ipv4" source address="223.198.0.0/15" reject

root@mediaserver:~# iptables -L IN_public_deny
Chain IN_public_deny (1 references)
target     prot opt source               destination         
REJECT     all  --  223.198.0.0/15       anywhere             reject-with icmp-port-unreachable


root@mediaserver:~# which python3
/usr/bin/python3

root@mediaserver:~# firewall-cmd --version
0.3.7

我认为问题可能与我在python脚本中如何转义字符有关,但据我所知,它们正在被正确转义。如果我可以提供任何其他调试信息,请告诉我。

1 个答案:

答案 0 :(得分:0)

解决方案是多部分的。我们通过声明部分命令参数并使用行继续来拆分所有内容来组织格式化。这有助于保持一切有条理并减少字符转义错误。另外,我们从Popen切换到Run,因为Popen过度使用了这个用法,并将shell = True值添加到我们的子进程中。

import subprocess

with open("ips.txt") as ip_list:
    ips = ip_list.readlines()

ips = (ip.strip() for ip in ips)
rules = ('rule family="ipv4" source address="{0}" reject'.format(ip) for ip in ips)

for rule in rules:
    process = subprocess.run("firewall-cmd "
                          "--permanent "
                          " --add-rich-rule=\'{0}\'".format(rule),
                          shell=True)