Exemple #1
0
    def test_create_firewall_no_rules(self):
        apply_list = self._fake_apply_list()
        firewall = self._fake_firewall_no_rule()
        self.firewall.create_firewall(apply_list, firewall)
        invalid_rule = '-m state --state INVALID -j DROP'
        est_rule = '-m state --state ESTABLISHED,RELATED -j ACCEPT'
        bname = fwaas.iptables_manager.binary_name

        for ip_version in (4, 6):
            ingress_chain = ('iv%s%s' % (ip_version, firewall['id']))
            egress_chain = ('ov%s%s' % (ip_version, firewall['id']))
            calls = [
                call.ensure_remove_chain('iv%sfake-fw-uuid' % ip_version),
                call.ensure_remove_chain('ov%sfake-fw-uuid' % ip_version),
                call.ensure_remove_chain('fwaas-default-policy'),
                call.add_chain('fwaas-default-policy'),
                call.add_rule('fwaas-default-policy', '-j DROP'),
                call.add_chain(ingress_chain),
                call.add_rule(ingress_chain, invalid_rule),
                call.add_rule(ingress_chain, est_rule),
                call.add_chain(egress_chain),
                call.add_rule(egress_chain, invalid_rule),
                call.add_rule(egress_chain, est_rule),
                call.add_rule('FORWARD', '-o qr-+ -j %s-fwaas-defau' % bname),
                call.add_rule('FORWARD', '-i qr-+ -j %s-fwaas-defau' % bname)
            ]
            if ip_version == 4:
                v4filter_inst = apply_list[0].iptables_manager.ipv4['filter']
                v4filter_inst.assert_has_calls(calls)
            else:
                v6filter_inst = apply_list[0].iptables_manager.ipv6['filter']
                v6filter_inst.assert_has_calls(calls)
 def _setup_firewall_with_rules(self, func):
     apply_list = self._fake_apply_list()
     rule_list = self._fake_rules_v4(FAKE_FW_ID)
     firewall = self._fake_firewall(rule_list)
     func(apply_list, firewall)
     invalid_rule = '-m state --state INVALID -j DROP'
     est_rule = '-m state --state ESTABLISHED,RELATED -j ACCEPT'
     rule1 = '-p tcp --dport 80  -s 10.24.4.2  -j ACCEPT'
     rule2 = '-p tcp --dport 22    -j DROP'
     ingress_chain = 'iv4%s' % firewall['id']
     egress_chain = 'ov4%s' % firewall['id']
     bname = fwaas.iptables_manager.binary_name
     ipt_mgr_ichain = '%s-%s' % (bname, ingress_chain[:11])
     ipt_mgr_echain = '%s-%s' % (bname, egress_chain[:11])
     calls = [call.ensure_remove_chain('iv4fake-fw-uuid'),
              call.ensure_remove_chain('ov4fake-fw-uuid'),
              call.ensure_remove_chain('fwaas-default-policy'),
              call.add_chain('fwaas-default-policy'),
              call.add_rule('fwaas-default-policy', '-j DROP'),
              call.add_chain(ingress_chain),
              call.add_rule(ingress_chain, invalid_rule),
              call.add_rule(ingress_chain, est_rule),
              call.add_chain(egress_chain),
              call.add_rule(egress_chain, invalid_rule),
              call.add_rule(egress_chain, est_rule),
              call.add_rule(ingress_chain, rule1),
              call.add_rule(egress_chain, rule1),
              call.add_rule(ingress_chain, rule2),
              call.add_rule(egress_chain, rule2),
              call.add_rule('FORWARD', '-o qr-+ -j %s' % ipt_mgr_ichain),
              call.add_rule('FORWARD', '-i qr-+ -j %s' % ipt_mgr_echain),
              call.add_rule('FORWARD', '-o qr-+ -j %s-fwaas-defau' % bname),
              call.add_rule('FORWARD', '-i qr-+ -j %s-fwaas-defau' % bname)]
     self.v4filter_inst.assert_has_calls(calls)
    def test_create_firewall_no_rules(self):
        apply_list = self._fake_apply_list()
        firewall = self._fake_firewall_no_rule()
        self.firewall.create_firewall(apply_list, firewall)
        invalid_rule = '-m state --state INVALID -j DROP'
        est_rule = '-m state --state ESTABLISHED,RELATED -j ACCEPT'
        bname = fwaas.iptables_manager.binary_name

        for ip_version in (4, 6):
            ingress_chain = ('iv%s%s' % (ip_version, firewall['id']))
            egress_chain = ('ov%s%s' % (ip_version, firewall['id']))
            calls = [call.ensure_remove_chain('iv%sfake-fw-uuid' % ip_version),
                     call.ensure_remove_chain('ov%sfake-fw-uuid' % ip_version),
                     call.ensure_remove_chain('fwaas-default-policy'),
                     call.add_chain('fwaas-default-policy'),
                     call.add_rule('fwaas-default-policy', '-j DROP'),
                     call.add_chain(ingress_chain),
                     call.add_rule(ingress_chain, invalid_rule),
                     call.add_rule(ingress_chain, est_rule),
                     call.add_chain(egress_chain),
                     call.add_rule(egress_chain, invalid_rule),
                     call.add_rule(egress_chain, est_rule),
                     call.add_rule('FORWARD',
                                   '-o qr-+ -j %s-fwaas-defau' % bname),
                     call.add_rule('FORWARD',
                                   '-i qr-+ -j %s-fwaas-defau' % bname)]
            if ip_version == 4:
                v4filter_inst = apply_list[0].iptables_manager.ipv4['filter']
                v4filter_inst.assert_has_calls(calls)
            else:
                v6filter_inst = apply_list[0].iptables_manager.ipv6['filter']
                v6filter_inst.assert_has_calls(calls)
 def test_create_firewall_no_rules(self):
     apply_list = self._fake_apply_list()
     firewall = self._fake_firewall_no_rule()
     self.firewall.create_firewall(apply_list, firewall)
     invalid_rule = '-m state --state INVALID -j DROP'
     est_rule = '-m state --state ESTABLISHED,RELATED -j ACCEPT'
     ingress_chain = ('iv4%s' % firewall['id'])
     egress_chain = ('ov4%s' % firewall['id'])
     bname = fwaas.iptables_manager.binary_name
     calls = [
         call.ensure_remove_chain('iv4fake-fw-uuid'),
         call.ensure_remove_chain('ov4fake-fw-uuid'),
         call.ensure_remove_chain('fwaas-default-policy'),
         call.add_chain('fwaas-default-policy'),
         call.add_rule('fwaas-default-policy', '-j DROP'),
         call.add_chain(ingress_chain),
         call.add_rule(ingress_chain, invalid_rule),
         call.add_rule(ingress_chain, est_rule),
         call.add_chain(egress_chain),
         call.add_rule(egress_chain, invalid_rule),
         call.add_rule(egress_chain, est_rule),
         call.add_rule('FORWARD', '-o qr-+ -j %s-fwaas-defau' % bname),
         call.add_rule('FORWARD', '-i qr-+ -j %s-fwaas-defau' % bname)
     ]
     apply_list[0].iptables_manager.ipv4['filter'].assert_has_calls(calls)
 def test_delete_firewall(self):
     apply_list = self._fake_apply_list()
     firewall = self._fake_firewall_no_rule()
     self.firewall.delete_firewall(apply_list, firewall)
     ingress_chain = 'iv4%s' % firewall['id']
     egress_chain = 'ov4%s' % firewall['id']
     calls = [call.ensure_remove_chain(ingress_chain),
              call.ensure_remove_chain(egress_chain),
              call.ensure_remove_chain('fwaas-default-policy')]
     apply_list[0].iptables_manager.ipv4['filter'].assert_has_calls(calls)
 def test_create_firewall_with_admin_down(self):
     rule_list = self._fake_rules_v4(FAKE_FW_ID)
     apply_list = self._fake_apply_list()
     firewall = self._fake_firewall_with_admin_down(rule_list)
     self.firewall.create_firewall(apply_list, firewall)
     calls = [call.ensure_remove_chain('iv4fake-fw-uuid'),
              call.ensure_remove_chain('ov4fake-fw-uuid'),
              call.ensure_remove_chain('fwaas-default-policy'),
              call.add_chain('fwaas-default-policy'),
              call.add_rule('fwaas-default-policy', '-j DROP')]
     self.v4filter_inst.assert_has_calls(calls)
 def test_create_firewall_with_admin_down(self):
     apply_list = self._fake_apply_list()
     rule_list = self._fake_rules_v4(FAKE_FW_ID, apply_list)
     firewall = self._fake_firewall_with_admin_down(rule_list)
     self.firewall.create_firewall(apply_list, firewall)
     calls = [call.ensure_remove_chain('iv4fake-fw-uuid'),
              call.ensure_remove_chain('ov4fake-fw-uuid'),
              call.ensure_remove_chain('fwaas-default-policy'),
              call.add_chain('fwaas-default-policy'),
              call.add_rule('fwaas-default-policy', '-j DROP')]
     apply_list[0].iptables_manager.ipv4['filter'].assert_has_calls(calls)
Exemple #8
0
 def test_delete_firewall(self):
     apply_list = self._fake_apply_list()
     firewall = self._fake_firewall_no_rule()
     self.firewall.delete_firewall(apply_list, firewall)
     ingress_chain = 'iv4%s' % firewall['id']
     egress_chain = 'ov4%s' % firewall['id']
     calls = [
         call.ensure_remove_chain(ingress_chain),
         call.ensure_remove_chain(egress_chain),
         call.ensure_remove_chain('fwaas-default-policy')
     ]
     apply_list[0].iptables_manager.ipv4['filter'].assert_has_calls(calls)
Exemple #9
0
 def test_create_firewall_with_admin_down(self):
     apply_list = self._fake_apply_list()
     rule_list = self._fake_rules_v4(FAKE_FW_ID, apply_list)
     firewall = self._fake_firewall_with_admin_down(rule_list)
     self.firewall.create_firewall(apply_list, firewall)
     calls = [
         call.ensure_remove_chain('iv4fake-fw-uuid'),
         call.ensure_remove_chain('ov4fake-fw-uuid'),
         call.ensure_remove_chain('fwaas-default-policy'),
         call.add_chain('fwaas-default-policy'),
         call.add_rule('fwaas-default-policy', '-j DROP')
     ]
     apply_list[0].iptables_manager.ipv4['filter'].assert_has_calls(calls)
    def _test_prepare_port_filter(self, rule, ingress_expected_call=None, egress_expected_call=None):
        port = self._fake_port()
        ethertype = rule["ethertype"]
        prefix = FAKE_IP[ethertype]
        filter_inst = self.v4filter_inst
        dhcp_rule = call.add_rule("ofake_dev", "-p udp -m udp --sport 68 --dport 67 -j RETURN")

        if ethertype == "IPv6":
            filter_inst = self.v6filter_inst
            dhcp_rule = call.add_rule("ofake_dev", "-p icmpv6 -j RETURN")

        sg = [rule]
        port["security_group_rules"] = sg
        self.firewall.prepare_port_filter(port)
        calls = [
            call.add_chain("sg-fallback"),
            call.add_rule("sg-fallback", "-j DROP"),
            call.ensure_remove_chain("sg-chain"),
            call.add_chain("sg-chain"),
            call.add_chain("ifake_dev"),
            call.add_rule("FORWARD", "-m physdev --physdev-out tapfake_dev " "--physdev-is-bridged " "-j $sg-chain"),
            call.add_rule("sg-chain", "-m physdev --physdev-out tapfake_dev " "--physdev-is-bridged " "-j $ifake_dev"),
        ]
        if ethertype == "IPv6":
            for icmp6_type in constants.ICMPV6_ALLOWED_TYPES:
                calls.append(call.add_rule("ifake_dev", "-p icmpv6 --icmpv6-type %s -j RETURN" % icmp6_type))
        calls += [
            call.add_rule("ifake_dev", "-m state --state INVALID -j DROP"),
            call.add_rule("ifake_dev", "-m state --state RELATED,ESTABLISHED -j RETURN"),
        ]

        if ingress_expected_call:
            calls.append(ingress_expected_call)

        calls += [
            call.add_rule("ifake_dev", "-j $sg-fallback"),
            call.add_chain("ofake_dev"),
            call.add_rule("FORWARD", "-m physdev --physdev-in tapfake_dev " "--physdev-is-bridged " "-j $sg-chain"),
            call.add_rule("sg-chain", "-m physdev --physdev-in tapfake_dev " "--physdev-is-bridged " "-j $ofake_dev"),
            call.add_rule("INPUT", "-m physdev --physdev-in tapfake_dev " "--physdev-is-bridged " "-j $ofake_dev"),
            call.add_chain("sfake_dev"),
            call.add_rule("sfake_dev", "-m mac --mac-source ff:ff:ff:ff -s %s -j RETURN" % prefix),
            call.add_rule("sfake_dev", "-j DROP"),
            dhcp_rule,
            call.add_rule("ofake_dev", "-j $sfake_dev"),
        ]
        if ethertype == "IPv4":
            calls.append(call.add_rule("ofake_dev", "-p udp -m udp --sport 67 --dport 68 -j DROP"))

        calls += [
            call.add_rule("ofake_dev", "-m state --state INVALID -j DROP"),
            call.add_rule("ofake_dev", "-m state --state RELATED,ESTABLISHED -j RETURN"),
        ]

        if egress_expected_call:
            calls.append(egress_expected_call)

        calls += [call.add_rule("ofake_dev", "-j $sg-fallback"), call.add_rule("sg-chain", "-j ACCEPT")]

        filter_inst.assert_has_calls(calls)
 def test_ip_spoofing_no_fixed_ips(self):
     port = {"device": "tapfake_dev", "mac_address": "ff:ff:ff:ff", "fixed_ips": []}
     self.firewall.prepare_port_filter(port)
     calls = [
         call.add_chain("sg-fallback"),
         call.add_rule("sg-fallback", "-j DROP"),
         call.ensure_remove_chain("sg-chain"),
         call.add_chain("sg-chain"),
         call.add_chain("ifake_dev"),
         call.add_rule("FORWARD", "-m physdev --physdev-out tapfake_dev " "--physdev-is-bridged " "-j $sg-chain"),
         call.add_rule("sg-chain", "-m physdev --physdev-out tapfake_dev " "--physdev-is-bridged " "-j $ifake_dev"),
         call.add_rule("ifake_dev", "-m state --state INVALID -j DROP"),
         call.add_rule("ifake_dev", "-m state --state RELATED,ESTABLISHED -j RETURN"),
         call.add_rule("ifake_dev", "-j $sg-fallback"),
         call.add_chain("ofake_dev"),
         call.add_rule("FORWARD", "-m physdev --physdev-in tapfake_dev " "--physdev-is-bridged " "-j $sg-chain"),
         call.add_rule("sg-chain", "-m physdev --physdev-in tapfake_dev " "--physdev-is-bridged " "-j $ofake_dev"),
         call.add_rule("INPUT", "-m physdev --physdev-in tapfake_dev " "--physdev-is-bridged " "-j $ofake_dev"),
         call.add_chain("sfake_dev"),
         call.add_rule("sfake_dev", "-m mac --mac-source ff:ff:ff:ff -j RETURN"),
         call.add_rule("sfake_dev", "-j DROP"),
         call.add_rule("ofake_dev", "-p udp -m udp --sport 68 --dport 67 -j RETURN"),
         call.add_rule("ofake_dev", "-j $sfake_dev"),
         call.add_rule("ofake_dev", "-p udp -m udp --sport 67 --dport 68 -j DROP"),
         call.add_rule("ofake_dev", "-m state --state INVALID -j DROP"),
         call.add_rule("ofake_dev", "-m state --state RELATED,ESTABLISHED -j RETURN"),
         call.add_rule("ofake_dev", "-j $sg-fallback"),
         call.add_rule("sg-chain", "-j ACCEPT"),
     ]
     self.v4filter_inst.assert_has_calls(calls)
 def test_ip_spoofing_filter_with_multiple_ips(self):
     port = {'device': 'tapfake_dev',
             'mac_address': 'ff:ff:ff:ff',
             'fixed_ips': ['10.0.0.1', 'fe80::1', '10.0.0.2']}
     self.firewall.prepare_port_filter(port)
     calls = [call.add_chain('sg-fallback'),
              call.add_rule('sg-fallback', '-j DROP'),
              call.ensure_remove_chain('sg-chain'),
              call.add_chain('sg-chain'),
              call.add_chain('ifake_dev'),
              call.add_rule('FORWARD',
                            '-m physdev --physdev-out tapfake_dev '
                            '--physdev-is-bridged '
                            '-j $sg-chain'),
              call.add_rule('sg-chain',
                            '-m physdev --physdev-out tapfake_dev '
                            '--physdev-is-bridged '
                            '-j $ifake_dev'),
              call.add_rule(
                  'ifake_dev', '-m state --state INVALID -j DROP'),
              call.add_rule(
                  'ifake_dev',
                  '-m state --state RELATED,ESTABLISHED -j RETURN'),
              call.add_rule('ifake_dev', '-j $sg-fallback'),
              call.add_chain('ofake_dev'),
              call.add_rule('FORWARD',
                            '-m physdev --physdev-in tapfake_dev '
                            '--physdev-is-bridged '
                            '-j $sg-chain'),
              call.add_rule('sg-chain',
                            '-m physdev --physdev-in tapfake_dev '
                            '--physdev-is-bridged '
                            '-j $ofake_dev'),
              call.add_rule('INPUT',
                            '-m physdev --physdev-in tapfake_dev '
                            '--physdev-is-bridged '
                            '-j $ofake_dev'),
              call.add_chain('sfake_dev'),
              call.add_rule(
                  'sfake_dev',
                  '-m mac --mac-source ff:ff:ff:ff -s 10.0.0.1 -j RETURN'),
              call.add_rule(
                  'sfake_dev',
                  '-m mac --mac-source ff:ff:ff:ff -s 10.0.0.2 -j RETURN'),
              call.add_rule('sfake_dev', '-j DROP'),
              call.add_rule(
                  'ofake_dev',
                  '-p udp -m udp --sport 68 --dport 67 -j RETURN'),
              call.add_rule('ofake_dev', '-j $sfake_dev'),
              call.add_rule(
                  'ofake_dev',
                  '-p udp -m udp --sport 67 --dport 68 -j DROP'),
              call.add_rule(
                  'ofake_dev', '-m state --state INVALID -j DROP'),
              call.add_rule(
                  'ofake_dev',
                  '-m state --state RELATED,ESTABLISHED -j RETURN'),
              call.add_rule('ofake_dev', '-j $sg-fallback'),
              call.add_rule('sg-chain', '-j ACCEPT')]
     self.v4filter_inst.assert_has_calls(calls)
    def test_prepare_port_filter_with_no_sg(self):
        port = self._fake_port()
        self.firewall.prepare_port_filter(port)
        calls = [
            call.add_chain("sg-fallback"),
            call.add_rule("sg-fallback", "-j DROP"),
            call.ensure_remove_chain("sg-chain"),
            call.add_chain("sg-chain"),
            call.add_chain("ifake_dev"),
            call.add_rule("FORWARD", "-m physdev --physdev-is-bridged " "--physdev-out tapfake_dev " "-j $sg-chain"),
            call.add_rule("sg-chain", "-m physdev --physdev-is-bridged " "--physdev-out tapfake_dev " "-j $ifake_dev"),
            call.add_rule("ifake_dev", "-m state --state INVALID -j DROP"),
            call.add_rule("ifake_dev", "-m state --state ESTABLISHED,RELATED -j RETURN"),
            call.add_rule("ifake_dev", "-j $sg-fallback"),
            call.add_chain("ofake_dev"),
            call.add_rule("FORWARD", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev " "-j $sg-chain"),
            call.add_rule("sg-chain", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev " "-j $ofake_dev"),
            call.add_rule("INPUT", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev " "-j $ofake_dev"),
            call.add_rule("ofake_dev", "-m mac ! --mac-source ff:ff:ff:ff -j DROP"),
            call.add_rule("ofake_dev", "-p udp --sport 68 --dport 67 -j RETURN"),
            call.add_rule("ofake_dev", "! -s 10.0.0.1 -j DROP"),
            call.add_rule("ofake_dev", "-p udp --sport 67 --dport 68 -j DROP"),
            call.add_rule("ofake_dev", "-m state --state INVALID -j DROP"),
            call.add_rule("ofake_dev", "-m state --state ESTABLISHED,RELATED -j RETURN"),
            call.add_rule("ofake_dev", "-j $sg-fallback"),
            call.add_rule("sg-chain", "-j ACCEPT"),
        ]

        self.v4filter_inst.assert_has_calls(calls)
    def test_prepare_port_filter_with_no_sg(self):
        port = self._fake_port()
        self.firewall.prepare_port_filter(port)
        calls = [call.add_chain('sg-fallback'),
                 call.add_rule('sg-fallback', '-j DROP'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain'),
                 call.add_chain('ifake_dev'),
                 call.add_rule('FORWARD',
                               '-m physdev --physdev-out tapfake_dev '
                               '--physdev-is-bridged '
                               '-j $sg-chain'),
                 call.add_rule('sg-chain',
                               '-m physdev --physdev-out tapfake_dev '
                               '--physdev-is-bridged '
                               '-j $ifake_dev'),
                 call.add_rule(
                     'ifake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state RELATED,ESTABLISHED -j RETURN'),
                 call.add_rule('ifake_dev', '-j $sg-fallback'),
                 call.add_chain('ofake_dev'),
                 call.add_rule('FORWARD',
                               '-m physdev --physdev-in tapfake_dev '
                               '--physdev-is-bridged '
                               '-j $sg-chain'),
                 call.add_rule('sg-chain',
                               '-m physdev --physdev-in tapfake_dev '
                               '--physdev-is-bridged '
                               '-j $ofake_dev'),
                 call.add_rule('INPUT',
                               '-m physdev --physdev-in tapfake_dev '
                               '--physdev-is-bridged '
                               '-j $ofake_dev'),
                 call.add_chain('sfake_dev'),
                 call.add_rule(
                     'sfake_dev', '-m mac --mac-source ff:ff:ff:ff '
                     '-s 10.0.0.1 -j RETURN'),
                 call.add_rule('sfake_dev', '-j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-p udp -m udp --sport 68 --dport 67 -j RETURN'),
                 call.add_rule('ofake_dev', '-j $sfake_dev'),
                 call.add_rule(
                     'ofake_dev',
                     '-p udp -m udp --sport 67 --dport 68 -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-m state --state RELATED,ESTABLISHED -j RETURN'),
                 call.add_rule('ofake_dev', '-j $sg-fallback'),
                 call.add_rule('sg-chain', '-j ACCEPT')]

        self.v4filter_inst.assert_has_calls(calls)
Exemple #15
0
 def _setup_firewall_with_rules(self, func, router_count=1):
     apply_list = self._fake_apply_list(router_count=router_count)
     rule_list = self._fake_rules_v4(FAKE_FW_ID, apply_list)
     firewall = self._fake_firewall(rule_list)
     func(apply_list, firewall)
     invalid_rule = '-m state --state INVALID -j DROP'
     est_rule = '-m state --state ESTABLISHED,RELATED -j ACCEPT'
     rule1 = '-p tcp --dport 80  -s 10.24.4.2  -j ACCEPT'
     rule2 = '-p tcp --dport 22    -j DROP'
     ingress_chain = 'iv4%s' % firewall['id']
     egress_chain = 'ov4%s' % firewall['id']
     bname = fwaas.iptables_manager.binary_name
     ipt_mgr_ichain = '%s-%s' % (bname, ingress_chain[:11])
     ipt_mgr_echain = '%s-%s' % (bname, egress_chain[:11])
     for router_info_inst in apply_list:
         v4filter_inst = router_info_inst.iptables_manager.ipv4['filter']
         calls = [
             call.ensure_remove_chain('iv4fake-fw-uuid'),
             call.ensure_remove_chain('ov4fake-fw-uuid'),
             call.ensure_remove_chain('fwaas-default-policy'),
             call.add_chain('fwaas-default-policy'),
             call.add_rule('fwaas-default-policy', '-j DROP'),
             call.add_chain(ingress_chain),
             call.add_rule(ingress_chain, invalid_rule),
             call.add_rule(ingress_chain, est_rule),
             call.add_chain(egress_chain),
             call.add_rule(egress_chain, invalid_rule),
             call.add_rule(egress_chain, est_rule),
             call.add_rule(ingress_chain, rule1),
             call.add_rule(egress_chain, rule1),
             call.add_rule(ingress_chain, rule2),
             call.add_rule(egress_chain, rule2),
             call.add_rule('FORWARD', '-o qr-+ -j %s' % ipt_mgr_ichain),
             call.add_rule('FORWARD', '-i qr-+ -j %s' % ipt_mgr_echain),
             call.add_rule('FORWARD', '-o qr-+ -j %s-fwaas-defau' % bname),
             call.add_rule('FORWARD', '-i qr-+ -j %s-fwaas-defau' % bname)
         ]
         v4filter_inst.assert_has_calls(calls)
 def test_create_firewall_no_rules(self):
     apply_list = self._fake_apply_list()
     firewall = self._fake_firewall_no_rule()
     self.firewall.create_firewall(apply_list, firewall)
     invalid_rule = '-m state --state INVALID -j DROP'
     est_rule = '-m state --state ESTABLISHED,RELATED -j ACCEPT'
     ingress_chain = ('iv4%s' % firewall['id'])
     egress_chain = ('ov4%s' % firewall['id'])
     bname = fwaas.iptables_manager.binary_name
     calls = [call.ensure_remove_chain('iv4fake-fw-uuid'),
              call.ensure_remove_chain('ov4fake-fw-uuid'),
              call.ensure_remove_chain('fwaas-default-policy'),
              call.add_chain('fwaas-default-policy'),
              call.add_rule('fwaas-default-policy', '-j DROP'),
              call.add_chain(ingress_chain),
              call.add_rule(ingress_chain, invalid_rule),
              call.add_rule(ingress_chain, est_rule),
              call.add_chain(egress_chain),
              call.add_rule(egress_chain, invalid_rule),
              call.add_rule(egress_chain, est_rule),
              call.add_rule('FORWARD', '-o qr-+ -j %s-fwaas-defau' % bname),
              call.add_rule('FORWARD', '-i qr-+ -j %s-fwaas-defau' % bname)]
     self.v4filter_inst.assert_has_calls(calls)
    def test_update_delete_port_filter(self):
        port = self._fake_port()
        port["security_group_rules"] = [{"ethertype": "IPv4", "direction": "ingress"}]
        self.firewall.prepare_port_filter(port)
        port["security_group_rules"] = [{"ethertype": "IPv4", "direction": "egress"}]
        self.firewall.update_port_filter(port)
        self.firewall.update_port_filter({"device": "no-exist-device"})
        self.firewall.remove_port_filter(port)
        self.firewall.remove_port_filter({"device": "no-exist-device"})
        calls = [
            call.add_chain("sg-fallback"),
            call.add_rule("sg-fallback", "-j DROP"),
            call.ensure_remove_chain("sg-chain"),
            call.add_chain("sg-chain"),
            call.add_chain("ifake_dev"),
            call.add_rule("FORWARD", "-m physdev --physdev-is-bridged " "--physdev-out tapfake_dev -j $sg-chain"),
            call.add_rule("sg-chain", "-m physdev --physdev-is-bridged " "--physdev-out tapfake_dev -j $ifake_dev"),
            call.add_rule("ifake_dev", "-m state --state INVALID -j DROP"),
            call.add_rule("ifake_dev", "-m state --state ESTABLISHED,RELATED -j RETURN"),
            call.add_rule("ifake_dev", "-j RETURN"),
            call.add_rule("ifake_dev", "-j $sg-fallback"),
            call.add_chain("ofake_dev"),
            call.add_rule("FORWARD", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev -j $sg-chain"),
            call.add_rule("sg-chain", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev -j $ofake_dev"),
            call.add_rule("INPUT", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev -j $ofake_dev"),
            call.add_rule("ofake_dev", "-m mac ! --mac-source ff:ff:ff:ff -j DROP"),
            call.add_rule("ofake_dev", "-p udp --sport 68 --dport 67 -j RETURN"),
            call.add_rule("ofake_dev", "! -s 10.0.0.1 -j DROP"),
            call.add_rule("ofake_dev", "-p udp --sport 67 --dport 68 -j DROP"),
            call.add_rule("ofake_dev", "-m state --state INVALID -j DROP"),
            call.add_rule("ofake_dev", "-m state --state ESTABLISHED,RELATED -j RETURN"),
            call.add_rule("ofake_dev", "-j $sg-fallback"),
            call.add_rule("sg-chain", "-j ACCEPT"),
            call.ensure_remove_chain("ifake_dev"),
            call.ensure_remove_chain("ofake_dev"),
            call.ensure_remove_chain("sg-chain"),
            call.add_chain("sg-chain"),
            call.add_chain("ifake_dev"),
            call.add_rule("FORWARD", "-m physdev --physdev-is-bridged " "--physdev-out tapfake_dev -j $sg-chain"),
            call.add_rule("sg-chain", "-m physdev --physdev-is-bridged " "--physdev-out tapfake_dev -j $ifake_dev"),
            call.add_rule("ifake_dev", "-m state --state INVALID -j DROP"),
            call.add_rule("ifake_dev", "-m state --state ESTABLISHED,RELATED -j RETURN"),
            call.add_rule("ifake_dev", "-j $sg-fallback"),
            call.add_chain("ofake_dev"),
            call.add_rule("FORWARD", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev -j $sg-chain"),
            call.add_rule("sg-chain", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev -j $ofake_dev"),
            call.add_rule("INPUT", "-m physdev --physdev-is-bridged " "--physdev-in tapfake_dev -j $ofake_dev"),
            call.add_rule("ofake_dev", "-m mac ! --mac-source ff:ff:ff:ff -j DROP"),
            call.add_rule("ofake_dev", "-p udp --sport 68 --dport 67 -j RETURN"),
            call.add_rule("ofake_dev", "! -s 10.0.0.1 -j DROP"),
            call.add_rule("ofake_dev", "-p udp --sport 67 --dport 68 -j DROP"),
            call.add_rule("ofake_dev", "-m state --state INVALID -j DROP"),
            call.add_rule("ofake_dev", "-m state --state ESTABLISHED,RELATED -j RETURN"),
            call.add_rule("ofake_dev", "-j RETURN"),
            call.add_rule("ofake_dev", "-j $sg-fallback"),
            call.add_rule("sg-chain", "-j ACCEPT"),
            call.ensure_remove_chain("ifake_dev"),
            call.ensure_remove_chain("ofake_dev"),
            call.ensure_remove_chain("sg-chain"),
            call.add_chain("sg-chain"),
        ]

        self.v4filter_inst.assert_has_calls(calls)
    def test_update_delete_port_filter(self):
        port = self._fake_port()
        port['security_group_rules'] = [{'ethertype': 'IPv4',
                                         'direction': 'ingress'}]
        self.firewall.prepare_port_filter(port)
        port['security_group_rules'] = [{'ethertype': 'IPv4',
                                         'direction': 'egress'}]
        self.firewall.update_port_filter(port)
        self.firewall.update_port_filter({'device': 'no-exist-device'})
        self.firewall.remove_port_filter(port)
        self.firewall.remove_port_filter({'device': 'no-exist-device'})
        calls = [call.add_chain('sg-fallback'),
                 call.add_rule('sg-fallback', '-j DROP'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain'),
                 call.add_chain('ifake_dev'),
                 call.add_rule(
                     'FORWARD',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-out tapfake_dev -j $sg-chain'),
                 call.add_rule(
                     'sg-chain',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-out tapfake_dev -j $ifake_dev'),
                 call.add_rule(
                     'ifake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN'),
                 call.add_rule('ifake_dev', '-j RETURN'),
                 call.add_rule('ifake_dev', '-j $sg-fallback'),
                 call.add_chain('ofake_dev'),
                 call.add_rule(
                     'FORWARD',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $sg-chain'),
                 call.add_rule(
                     'sg-chain',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $ofake_dev'),
                 call.add_rule(
                     'INPUT',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $ofake_dev'),
                 call.add_rule(
                     'ofake_dev',
                     '-m mac ! --mac-source ff:ff:ff:ff -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-p udp --sport 68 --dport 67 -j RETURN'),
                 call.add_rule(
                     'ofake_dev',
                     '! -s 10.0.0.1 -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-p udp --sport 67 --dport 68 -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN'),
                 call.add_rule('ofake_dev', '-j $sg-fallback'),
                 call.add_rule('sg-chain', '-j ACCEPT'),
                 call.ensure_remove_chain('ifake_dev'),
                 call.ensure_remove_chain('ofake_dev'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain'),
                 call.add_chain('ifake_dev'),
                 call.add_rule(
                     'FORWARD',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-out tapfake_dev -j $sg-chain'),
                 call.add_rule(
                     'sg-chain',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-out tapfake_dev -j $ifake_dev'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN'),
                 call.add_rule('ifake_dev', '-j $sg-fallback'),
                 call.add_chain('ofake_dev'),
                 call.add_rule(
                     'FORWARD',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $sg-chain'),
                 call.add_rule(
                     'sg-chain',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $ofake_dev'),
                 call.add_rule(
                     'INPUT',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $ofake_dev'),
                 call.add_rule(
                     'ofake_dev',
                     '-m mac ! --mac-source ff:ff:ff:ff -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-p udp --sport 68 --dport 67 -j RETURN'),
                 call.add_rule(
                     'ofake_dev', '! -s 10.0.0.1 -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-p udp --sport 67 --dport 68 -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN'),
                 call.add_rule('ofake_dev', '-j RETURN'),
                 call.add_rule('ofake_dev', '-j $sg-fallback'),
                 call.add_rule('sg-chain', '-j ACCEPT'),
                 call.ensure_remove_chain('ifake_dev'),
                 call.ensure_remove_chain('ofake_dev'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain')]

        self.v4filter_inst.assert_has_calls(calls)
    def _test_prepare_port_filter(self,
                                  rule,
                                  ingress_expected_call=None,
                                  egress_expected_call=None):
        port = self._fake_port()
        ethertype = rule['ethertype']
        prefix = FAKE_IP[ethertype]
        filter_inst = self.v4filter_inst
        dhcp_rule = call.add_rule(
            'ofake_dev',
            '-p udp --sport 68 --dport 67 -j RETURN')

        if ethertype == 'IPv6':
            filter_inst = self.v6filter_inst
            dhcp_rule = call.add_rule('ofake_dev', '-p icmpv6 -j RETURN')

        sg = [rule]
        port['security_group_rules'] = sg
        self.firewall.prepare_port_filter(port)
        calls = [call.add_chain('sg-fallback'),
                 call.add_rule('sg-fallback', '-j DROP'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain'),
                 call.add_chain('ifake_dev'),
                 call.add_rule('FORWARD',
                               '-m physdev --physdev-is-bridged '
                               '--physdev-out tapfake_dev '
                               '-j $sg-chain'),
                 call.add_rule('sg-chain',
                               '-m physdev --physdev-is-bridged '
                               '--physdev-out tapfake_dev '
                               '-j $ifake_dev'),
                 call.add_rule(
                     'ifake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN')]

        if ingress_expected_call:
            calls.append(ingress_expected_call)

        calls += [call.add_rule('ifake_dev', '-j $sg-fallback'),
                  call.add_chain('ofake_dev'),
                  call.add_rule('FORWARD',
                                '-m physdev --physdev-is-bridged '
                                '--physdev-in tapfake_dev '
                                '-j $sg-chain'),
                  call.add_rule('sg-chain',
                                '-m physdev --physdev-is-bridged '
                                '--physdev-in tapfake_dev '
                                '-j $ofake_dev'),
                  call.add_rule('INPUT',
                                '-m physdev --physdev-is-bridged '
                                '--physdev-in tapfake_dev '
                                '-j $ofake_dev'),
                  call.add_rule(
                      'ofake_dev',
                      '-m mac ! --mac-source ff:ff:ff:ff -j DROP'),
                  dhcp_rule,
                  call.add_rule('ofake_dev', '! -s %s -j DROP' % prefix)]

        if ethertype == 'IPv4':
            calls.append(call.add_rule(
                'ofake_dev',
                '-p udp --sport 67 --dport 68 -j DROP'))

        calls += [call.add_rule(
                  'ofake_dev', '-m state --state INVALID -j DROP'),
                  call.add_rule(
                  'ofake_dev',
                  '-m state --state ESTABLISHED,RELATED -j RETURN')]

        if egress_expected_call:
            calls.append(egress_expected_call)

        calls += [call.add_rule('ofake_dev', '-j $sg-fallback'),
                  call.add_rule('sg-chain', '-j ACCEPT')]

        filter_inst.assert_has_calls(calls)
    def test_update_delete_port_filter(self):
        port = self._fake_port()
        port['security_group_rules'] = [{'ethertype': 'IPv4',
                                         'direction': 'ingress'}]
        self.firewall.prepare_port_filter(port)
        port['security_group_rules'] = [{'ethertype': 'IPv4',
                                         'direction': 'egress'}]
        self.firewall.update_port_filter(port)
        self.firewall.update_port_filter({'device': 'no-exist-device'})
        self.firewall.remove_port_filter(port)
        self.firewall.remove_port_filter({'device': 'no-exist-device'})
        calls = [call.add_chain('sg-fallback'),
                 call.add_rule('sg-fallback', '-j DROP'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain'),
                 call.add_chain('ifake_dev'),
                 call.add_rule(
                     'FORWARD',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-out tapfake_dev -j $sg-chain'),
                 call.add_rule(
                     'sg-chain',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-out tapfake_dev -j $ifake_dev'),
                 call.add_rule(
                     'ifake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN'),
                 call.add_rule('ifake_dev', '-j RETURN'),
                 call.add_rule('ifake_dev', '-j $sg-fallback'),
                 call.add_chain('ofake_dev'),
                 call.add_rule(
                     'FORWARD',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $sg-chain'),
                 call.add_rule(
                     'sg-chain',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $ofake_dev'),
                 call.add_rule(
                     'INPUT',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $ofake_dev'),
                 call.add_rule(
                     'ofake_dev',
                     '-m mac ! --mac-source ff:ff:ff:ff -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-p udp --sport 68 --dport 67 -j RETURN'),
                 call.add_rule(
                     'ofake_dev',
                     '! -s 10.0.0.1 -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-p udp --sport 67 --dport 68 -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN'),
                 call.add_rule('ofake_dev', '-j $sg-fallback'),
                 call.add_rule('sg-chain', '-j ACCEPT'),
                 call.ensure_remove_chain('ifake_dev'),
                 call.ensure_remove_chain('ofake_dev'),
                 call.ensure_remove_chain('sfake_dev'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain'),
                 call.add_chain('ifake_dev'),
                 call.add_rule(
                     'FORWARD',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-out tapfake_dev -j $sg-chain'),
                 call.add_rule(
                     'sg-chain',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-out tapfake_dev -j $ifake_dev'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN'),
                 call.add_rule('ifake_dev', '-j $sg-fallback'),
                 call.add_chain('ofake_dev'),
                 call.add_rule(
                     'FORWARD',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $sg-chain'),
                 call.add_rule(
                     'sg-chain',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $ofake_dev'),
                 call.add_rule(
                     'INPUT',
                     '-m physdev --physdev-is-bridged '
                     '--physdev-in tapfake_dev -j $ofake_dev'),
                 call.add_rule(
                     'ofake_dev',
                     '-m mac ! --mac-source ff:ff:ff:ff -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-p udp --sport 68 --dport 67 -j RETURN'),
                 call.add_rule(
                     'ofake_dev', '! -s 10.0.0.1 -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-p udp --sport 67 --dport 68 -j DROP'),
                 call.add_rule(
                     'ofake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ofake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN'),
                 call.add_rule('ofake_dev', '-j RETURN'),
                 call.add_rule('ofake_dev', '-j $sg-fallback'),
                 call.add_rule('sg-chain', '-j ACCEPT'),
                 call.ensure_remove_chain('ifake_dev'),
                 call.ensure_remove_chain('ofake_dev'),
                 call.ensure_remove_chain('sfake_dev'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain')]

        self.v4filter_inst.assert_has_calls(calls)
    def _test_prepare_port_filter(self,
                                  rule,
                                  ingress_expected_call=None,
                                  egress_expected_call=None):
        port = self._fake_port()
        ethertype = rule['ethertype']
        prefix = FAKE_IP[ethertype]
        filter_inst = self.v4filter_inst
        dhcp_rule = call.add_rule(
            'ofake_dev',
            '-p udp --sport 68 --dport 67 -j RETURN')

        if ethertype == 'IPv6':
            filter_inst = self.v6filter_inst
            dhcp_rule = call.add_rule('ofake_dev', '-p icmpv6 -j RETURN')

        sg = [rule]
        port['security_group_rules'] = sg
        self.firewall.prepare_port_filter(port)
        calls = [call.add_chain('sg-fallback'),
                 call.add_rule('sg-fallback', '-j DROP'),
                 call.ensure_remove_chain('sg-chain'),
                 call.add_chain('sg-chain'),
                 call.add_chain('ifake_dev'),
                 call.add_rule('FORWARD',
                               '-m physdev --physdev-is-bridged '
                               '--physdev-out tapfake_dev '
                               '-j $sg-chain'),
                 call.add_rule('sg-chain',
                               '-m physdev --physdev-is-bridged '
                               '--physdev-out tapfake_dev '
                               '-j $ifake_dev'),
                 call.add_rule(
                     'ifake_dev', '-m state --state INVALID -j DROP'),
                 call.add_rule(
                     'ifake_dev',
                     '-m state --state ESTABLISHED,RELATED -j RETURN')]

        if ingress_expected_call:
            calls.append(ingress_expected_call)

        calls += [call.add_rule('ifake_dev', '-j $sg-fallback'),
                  call.add_chain('ofake_dev'),
                  call.add_rule('FORWARD',
                                '-m physdev --physdev-is-bridged '
                                '--physdev-in tapfake_dev '
                                '-j $sg-chain'),
                  call.add_rule('sg-chain',
                                '-m physdev --physdev-is-bridged '
                                '--physdev-in tapfake_dev '
                                '-j $ofake_dev'),
                  call.add_rule('INPUT',
                                '-m physdev --physdev-is-bridged '
                                '--physdev-in tapfake_dev '
                                '-j $ofake_dev'),
                  call.add_rule(
                      'ofake_dev',
                      '-m mac ! --mac-source ff:ff:ff:ff -j DROP'),
                  dhcp_rule,
                  call.add_rule('ofake_dev', '! -s %s -j DROP' % prefix)]

        if ethertype == 'IPv4':
            calls.append(call.add_rule(
                'ofake_dev',
                '-p udp --sport 67 --dport 68 -j DROP'))

        calls += [call.add_rule(
                  'ofake_dev', '-m state --state INVALID -j DROP'),
                  call.add_rule(
                  'ofake_dev',
                  '-m state --state ESTABLISHED,RELATED -j RETURN')]

        if egress_expected_call:
            calls.append(egress_expected_call)

        calls += [call.add_rule('ofake_dev', '-j $sg-fallback'),
                  call.add_rule('sg-chain', '-j ACCEPT')]

        filter_inst.assert_has_calls(calls)