From 59406cf225c5d66124ba095f8104ad4b206bf422 Mon Sep 17 00:00:00 2001 From: "Frank.Zhang" Date: Thu, 10 Oct 2013 14:43:40 -0700 Subject: [PATCH] Change security_group_agent python side in line with default security group rules change in 4.2 --- .../security_group_agent/cs_sg_agent.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/plugins/hypervisors/baremetal/resources/security_group_agent/security_group_agent/cs_sg_agent.py b/plugins/hypervisors/baremetal/resources/security_group_agent/security_group_agent/cs_sg_agent.py index f940264611f..9bd541b0a07 100755 --- a/plugins/hypervisors/baremetal/resources/security_group_agent/security_group_agent/cs_sg_agent.py +++ b/plugins/hypervisors/baremetal/resources/security_group_agent/security_group_agent/cs_sg_agent.py @@ -323,6 +323,14 @@ class SGAgent(object): @staticmethod def start(): + + def create_rule_if_not_exists(rule): + out = sglib.ShellCmd('iptables-save')() + if rule in out: + return + + sglib.ShellCmd('iptables %s' % rule)() + def prepare_default_rules(): sglib.ShellCmd('iptables --policy INPUT DROP')() name = 'default-chain' @@ -330,7 +338,9 @@ class SGAgent(object): sglib.ShellCmd('iptables -F %s' % name)() except Exception: sglib.ShellCmd('iptables -N %s' % name)() - sglib.ShellCmd('iptables -I INPUT -p tcp --dport 9988 -j ACCEPT')() + + create_rule_if_not_exists('-I INPUT -p tcp --dport 9988 -j ACCEPT') + create_rule_if_not_exists('-I INPUT -m state --state RELATED,ESTABLISHED -j ACCEPT') prepare_default_rules()