def testEgress(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER_EGRESS + GOOD_TERM_EGRESS, self.naming), EXP_INFO) self.failUnless('EGRESS' in str(acl), str(acl)) self.failUnless('INGRESS' not in str(acl), str(acl))
def testP4TagsNotPresent(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER + GOOD_TERM, self.naming), EXP_INFO) self.assertTrue('$Id:' not in str(acl))
def testDenyAction(self): self.naming.GetNetAddr.return_value = TEST_IPS acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER + GOOD_TERM_DENY, self.naming), EXP_INFO) expected = json.loads(GOOD_TERM_DENY_EXPECTED) self.assertEqual(expected, json.loads(self._StripAclHeaders(str(acl))))
def testTermWithPriority(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER + GOOD_TERM_3, self.naming), EXP_INFO) self.failUnless('"priority": "1",' in str(acl), str(acl))
def testIcmpv6Inet6(self): self.naming.GetNetAddr.return_value = TEST_IPS acl = gce.GCE( policy.ParsePolicy(GOOD_HEADER_INET6 + GOOD_TERM_ICMPV6, self.naming), EXP_INFO) self.assertIn('58', str(acl)) self.assertNotIn('icmp', str(acl))
def testIngress(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER_INGRESS + GOOD_TERM, self.naming), EXP_INFO) self.assertIn('INGRESS', str(acl)) self.assertNotIn('EGRESS', str(acl))
def testDefaultDenyIngressCreation(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy(GOOD_HEADER_INGRESS + GOOD_TERM_INGRESS_SOURCETAG + DEFAULT_DENY, self.naming), EXP_INFO) self.assertIn('"priority": 65534', str(acl))
def testDestinationRanges(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER_EGRESS + GOOD_TERM_EGRESS, self.naming), EXP_INFO) self.failUnless('destinationRanges' in str(acl), str(acl)) self.failUnless('sourceRanges' not in str(acl), str(acl)) self.failUnless('10.2.3.4/32' in str(acl), str(acl))
def testEgressTags(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER_EGRESS + GOOD_TERM_EGRESS_SOURCETAG, self.naming), EXP_INFO) self.assertIn('targetTags', str(acl)) self.assertNotIn('sourceTags', str(acl))
def testBuildTokens(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] pol1 = gce.GCE(policy.ParsePolicy(GOOD_HEADER + GOOD_TERM, self.naming), EXP_INFO) st, sst = pol1._BuildTokens() self.assertEquals(st, SUPPORTED_TOKENS) self.assertEquals(sst, SUPPORTED_SUB_TOKENS)
def testValidTermNames(self): for name in VALID_TERM_NAMES: self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] pol = policy.ParsePolicy( GOOD_HEADER + GOOD_TERM_CUSTOM_NAME % name, self.naming) acl = gce.GCE(pol, EXP_INFO) self.assertIsNotNone(str(acl))
def testInetIsDefault(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy(GOOD_HEADER + GOOD_TERM, self.naming), EXP_INFO) self.assertIn('INGRESS', str(acl)) self.assertNotIn('EGRESS', str(acl)) self.assertNotIn('2001:4860:8000::5/128', str(acl)) self.assertIn('10.2.3.4/32', str(acl))
def testIgmpMixed(self): self.naming.GetNetAddr.return_value = TEST_IPS acl = gce.GCE( policy.ParsePolicy(GOOD_HEADER_MIXED + GOOD_TERM_IGMP, self.naming), EXP_INFO) self.assertIn('2', str(acl)) self.assertIn('10.2.3.4/32', str(acl)) self.assertNotIn('2001:4860:8000::5/128', str(acl)) self.assertNotIn(gcp.GetIpv6TermName('good-term-pingv6'), str(acl))
def testTermWithLogging(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE( policy.ParsePolicy(GOOD_HEADER + GOOD_TERM_LOGGING, self.naming), EXP_INFO) rendered_acl = json.loads(str(acl))[0] self.assertIn('logConfig', rendered_acl) self.assertEqual(rendered_acl['logConfig'], {'enable': True})
def testMixedWithV4AddressesOnly(self): self.naming.GetNetAddr.return_value = TEST_IPV4_ONLY self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE( policy.ParsePolicy(GOOD_HEADER_MIXED + GOOD_TERM + DEFAULT_DENY, self.naming), EXP_INFO) self.assertIn('INGRESS', str(acl)) self.assertNotIn('EGRESS', str(acl)) self.assertNotIn('2001:4860:8000::5/128', str(acl)) self.assertIn('10.2.3.4/32', str(acl))
def testTermOwners(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE( policy.ParsePolicy(GOOD_HEADER + GOOD_TERM_OWNERS, self.naming), EXP_INFO) rendered_acl = json.loads(str(acl))[0] self.assertEqual(rendered_acl['description'], 'DNS access from corp. Owner: test-owner')
def testSkipExpiredTerm(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.return_value = ['22'] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER + GOOD_TERM_EXPIRED, self.naming), EXP_INFO) self.assertEquals(self._StripAclHeaders(str(acl)), '[]\n\n') self.naming.GetNetAddr.assert_called_once_with('CORP_EXTERNAL') self.naming.GetServiceByProto.assert_called_once_with('SSH', 'tcp')
def testInet6DefaultDenyEgressCreation(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE( policy.ParsePolicy( GOOD_HEADER_EGRESS_INET6 + GOOD_TERM_EGRESS + DEFAULT_DENY, self.naming), EXP_INFO) self.assertNotIn('INGRESS', str(acl)) self.assertIn('EGRESS', str(acl)) self.assertIn('"priority": 65534', str(acl)) self.assertIn('::/0', str(acl)) self.assertNotIn('0.0.0.0/0', str(acl))
def testMixedIsSeparateRules(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE( policy.ParsePolicy(GOOD_HEADER_MIXED + GOOD_TERM + DEFAULT_DENY, self.naming), EXP_INFO) self.assertIn('INGRESS', str(acl)) self.assertNotIn('EGRESS', str(acl)) self.assertIn('2001:4860:8000::5/128', str(acl)) self.assertIn('10.2.3.4/32', str(acl)) self.assertIn('good-term-1', str(acl)) self.assertIn(gcp.GetIpv6TermName('good-term-1'), str(acl))
def testInet6WithSourceTagAndV4Addresses(self): self.naming.GetNetAddr.return_value = TEST_IPV4_ONLY self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE( policy.ParsePolicy( GOOD_HEADER_INET6 + GOOD_TERM_INGRESS_ADDRESS_SOURCETAG + DEFAULT_DENY, self.naming), EXP_INFO) self.assertIn('INGRESS', str(acl)) self.assertNotIn('EGRESS', str(acl)) self.assertNotIn('2001:4860:8000::5/128', str(acl)) self.assertNotIn('10.2.3.4/32', str(acl)) self.assertNotIn('internal-servers', str(acl))
def testGenericTermWithoutNetwork(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER_NO_NETWORK + GOOD_TERM, self.naming), EXP_INFO) expected = json.loads(GOOD_TERM_NO_NETWORK_JSON) self.assertEqual(expected, json.loads(self._StripAclHeaders(str(acl)))) self.naming.GetNetAddr.assert_called_once_with('CORP_EXTERNAL') self.naming.GetServiceByProto.assert_has_calls([ mock.call('DNS', 'udp'), mock.call('DNS', 'tcp')])
def testMixedWithEgressSourceTag(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE( policy.ParsePolicy( GOOD_HEADER_EGRESS_MIXED + GOOD_TERM_EGRESS_SOURCETAG, self.naming), EXP_INFO) self.assertNotIn('INGRESS', str(acl)) self.assertIn('EGRESS', str(acl)) self.assertIn('10.2.3.4/32', str(acl)) self.assertIn('2001:4860:8000::5/128', str(acl)) self.assertIn('dns-servers', str(acl)) self.assertIn(gcp.GetIpv6TermName('good-term-1-e'), str(acl))
def testMixedDefaultDenyIngressCreation(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE( policy.ParsePolicy( GOOD_HEADER_MIXED + GOOD_TERM_INGRESS_SOURCETAG + DEFAULT_DENY, self.naming), EXP_INFO) self.assertIn('INGRESS', str(acl)) self.assertNotIn('EGRESS', str(acl)) self.assertIn('"priority": 65534', str(acl)) self.assertIn('default-deny', str(acl)) self.assertIn(gcp.GetIpv6TermName('default-deny'), str(acl)) self.assertIn('::/0', str(acl)) self.assertIn('0.0.0.0/0', str(acl))
def testGenericTermWithExclude(self): self.naming.GetNetAddr.side_effect = [TEST_INCLUDE_IPS, TEST_EXCLUDE_IPS] self.naming.GetServiceByProto.side_effect = [['53'], ['53']] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER + GOOD_TERM_EXCLUDE, self.naming), EXP_INFO) expected = json.loads(GOOD_TERM_JSON) self.assertEqual(expected, json.loads(self._StripAclHeaders(str(acl)))) self.naming.GetNetAddr.assert_has_calls([ mock.call('CORP_EXTERNAL'), mock.call('GUEST_WIRELESS_EXTERNAL')]) self.naming.GetServiceByProto.assert_has_calls([ mock.call('DNS', 'udp'), mock.call('DNS', 'tcp')])
def testSourceNetworkSplit(self): lots_of_ips = [] for i in range(20): for j in range(20): lots_of_ips.append(nacaddr.IP('10.%d.%d.1/32' % (i, j))) self.naming.GetNetAddr.return_value = lots_of_ips self.naming.GetServiceByProto.return_value = ['53'] acl = gce.GCE(policy.ParsePolicy( GOOD_HEADER + GOOD_TERM, self.naming), EXP_INFO) self.assertTrue('default-good-term-1-1' in str(acl)) self.assertTrue('default-good-term-1-2' in str(acl)) self.naming.GetNetAddr.assert_called_once_with('CORP_EXTERNAL') self.naming.GetServiceByProto.assert_has_calls([ mock.call('DNS', 'udp'), mock.call('DNS', 'tcp')])
def testSkipStatelessReply(self): self.naming.GetNetAddr.return_value = TEST_IPS self.naming.GetServiceByProto.return_value = ['22'] # Add stateless_reply to terms, there is no current way to include it in the # term definition. ret = policy.ParsePolicy(GOOD_HEADER + GOOD_TERM, self.naming) _, terms = ret.filters[0] for term in terms: term.stateless_reply = True acl = gce.GCE(ret, EXP_INFO) self.assertEqual(self._StripAclHeaders(str(acl)), '[]\n\n') self.naming.GetNetAddr.assert_called_once_with('CORP_EXTERNAL') self.naming.GetServiceByProto.assert_has_calls( [mock.call('DNS', 'udp'), mock.call('DNS', 'tcp')])
def testMaxAttribute(self): self.naming.GetNetAddr.return_value = [nacaddr.IP('10.2.3.4/32')] pol = policy.ParsePolicy(GOOD_HEADER_MAX_ATTRIBUTE_COUNT + GOOD_TERM_5, self.naming) acl = gce.GCE(pol, EXP_INFO) self.assertIsNotNone(str(acl))
def testNoProtocol(self): self.naming.GetNetAddr.return_value = TEST_IPS acl = gce.GCE( policy.ParsePolicy(GOOD_HEADER_MIXED + GOOD_TERM_NO_PROTOCOL, self.naming), EXP_INFO) self.assertIn('all', str(acl))
def get_acl(inputs): """Generates an ACL using Capirca. Args: inputs: Module parameters. Returns: ACL string. """ header_base = ''' header { comment:: "$comment" target:: $platform $options } ''' result = "" # Create copy of input options removing any spaces inputs['options_copy'] = [ str(elem).replace(" ", "") for elem in inputs['filter_options'] ] # Add from/to-zone to 'paloalto' and 'srx'. if inputs['platform'] in ('paloalto' 'srx'): if len(inputs['options_copy']) < 2: raise AnsibleError( "The number of options for {0} is less than 2".format( inputs['platform'])) inputs['options_copy'][0] = "from-zone " + inputs['options_copy'][0] inputs['options_copy'][1] = "to-zone " + inputs['options_copy'][1] # Create option string for header inputs['options'] = ' '.join( [str(elem) for elem in inputs['options_copy']]) header_template = Template(header_base) header = header_template.safe_substitute(inputs) defs = naming.Naming(inputs['def_folder']) terms = open(inputs['pol_file']).read() pol = policy.ParsePolicy(header + '\n' + terms, defs, optimize=True) # Exp info in weeks EXP_INFO = 2 # List from https://github.com/google/capirca/blob/master/capirca/aclgen.py#L202 # Does Python have a Switch statement? if inputs['platform'] == 'juniper': result = juniper.Juniper(pol, EXP_INFO) elif inputs['platform'] == 'cisco': result = cisco.Cisco(pol, EXP_INFO) elif inputs['platform'] == 'ciscoasa': result = ciscoasa.CiscoASA(pol, EXP_INFO) elif inputs['platform'] == 'brocade': result = brocade.Brocade(pol, EXP_INFO) elif inputs['platform'] == 'arista': result = arista.Arista(pol, EXP_INFO) elif inputs['platform'] == 'aruba': result = aruba.Aruba(pol, EXP_INFO) elif inputs['platform'] == 'ipset': result = ipset.Ipset(pol, EXP_INFO) elif inputs['platform'] == 'iptables': result = iptables.Iptables(pol, EXP_INFO) elif inputs['platform'] == 'nsxv': result = nsxv.Nsxv(pol, EXP_INFO) elif inputs['platform'] == 'packetfilter': result = packetfilter.PacketFilter(pol, EXP_INFO) elif inputs['platform'] == 'pcap': result = pcap.PcapFilter(pol, EXP_INFO) elif inputs['platform'] == 'speedway': result = speedway.Speedway(pol, EXP_INFO) elif inputs['platform'] == 'srx': result = junipersrx.JuniperSRX(pol, EXP_INFO) elif inputs['platform'] == 'srxlo': result = srxlo.SRXlo(pol, EXP_INFO) elif inputs['platform'] == 'windows_advfirewall': result = windows_advfirewall.WindowsAdvFirewall(pol, EXP_INFO) elif inputs['platform'] == 'ciscoxr': result = ciscoxr.CiscoXR(pol, EXP_INFO) elif inputs['platform'] == 'nftables': result = nftables.Nftables(pol, EXP_INFO) elif inputs['platform'] == 'gce': result = gce.GCE(pol, EXP_INFO) elif inputs['platform'] == 'paloalto': result = paloaltofw.PaloAltoFW(pol, EXP_INFO) elif inputs['platform'] == 'cloudarmor': result = cloudarmor.CloudArmor(pol, EXP_INFO) return str(result)
def RenderFile(base_directory, input_file, output_directory, definitions, exp_info, write_files): """Render a single file. Args: base_directory: The base directory to look for acls. input_file: the name of the input policy file. output_directory: the directory in which we place the rendered file. definitions: the definitions from naming.Naming(). exp_info: print a info message when a term is set to expire in that many weeks. write_files: a list of file tuples, (output_file, acl_text), to write """ logging.debug('rendering file: %s into %s', input_file, output_directory) pol = None jcl = False acl = False asacl = False aacl = False bacl = False eacl = False gca = False gcefw = False ips = False ipt = False spd = False nsx = False pcap_accept = False pcap_deny = False pf = False srx = False jsl = False nft = False win_afw = False xacl = False paloalto = False try: with open(input_file) as f: conf = f.read() logging.debug('opened and read %s', input_file) except IOError as e: logging.warning('bad file: \n%s', e) raise try: pol = policy.ParsePolicy(conf, definitions, optimize=FLAGS.optimize, base_dir=base_directory, shade_check=FLAGS.shade_check) except policy.ShadingError as e: logging.warning('shading errors for %s:\n%s', input_file, e) return except (policy.Error, naming.Error): raise ACLParserError( 'Error parsing policy file %s:\n%s%s' % (input_file, sys.exc_info()[0], sys.exc_info()[1])) platforms = set() for header in pol.headers: platforms.update(header.platforms) if 'juniper' in platforms: jcl = copy.deepcopy(pol) if 'cisco' in platforms: acl = copy.deepcopy(pol) if 'ciscoasa' in platforms: asacl = copy.deepcopy(pol) if 'brocade' in platforms: bacl = copy.deepcopy(pol) if 'arista' in platforms: eacl = copy.deepcopy(pol) if 'aruba' in platforms: aacl = copy.deepcopy(pol) if 'ipset' in platforms: ips = copy.deepcopy(pol) if 'iptables' in platforms: ipt = copy.deepcopy(pol) if 'nsxv' in platforms: nsx = copy.deepcopy(pol) if 'packetfilter' in platforms: pf = copy.deepcopy(pol) if 'pcap' in platforms: pcap_accept = copy.deepcopy(pol) pcap_deny = copy.deepcopy(pol) if 'speedway' in platforms: spd = copy.deepcopy(pol) if 'srx' in platforms: srx = copy.deepcopy(pol) if 'srxlo' in platforms: jsl = copy.deepcopy(pol) if 'windows_advfirewall' in platforms: win_afw = copy.deepcopy(pol) if 'ciscoxr' in platforms: xacl = copy.deepcopy(pol) if 'nftables' in platforms: nft = copy.deepcopy(pol) if 'gce' in platforms: gcefw = copy.deepcopy(pol) if 'paloalto' in platforms: paloalto = copy.deepcopy(pol) if 'cloudarmor' in platforms: gca = copy.deepcopy(pol) if not output_directory.endswith('/'): output_directory += '/' try: if jcl: acl_obj = juniper.Juniper(jcl, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if srx: acl_obj = junipersrx.JuniperSRX(srx, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if acl: acl_obj = cisco.Cisco(acl, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if asacl: acl_obj = ciscoasa.CiscoASA(asacl, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if aacl: acl_obj = aruba.Aruba(aacl, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if bacl: acl_obj = brocade.Brocade(bacl, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if eacl: acl_obj = arista.Arista(eacl, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if ips: acl_obj = ipset.Ipset(ips, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if ipt: acl_obj = iptables.Iptables(ipt, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if nsx: acl_obj = nsxv.Nsxv(nsx, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if spd: acl_obj = speedway.Speedway(spd, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if pcap_accept: acl_obj = pcap.PcapFilter(pcap_accept, exp_info) RenderACL(str(acl_obj), '-accept' + acl_obj.SUFFIX, output_directory, input_file, write_files) if pcap_deny: acl_obj = pcap.PcapFilter(pcap_deny, exp_info, invert=True) RenderACL(str(acl_obj), '-deny' + acl_obj.SUFFIX, output_directory, input_file, write_files) if pf: acl_obj = packetfilter.PacketFilter(pf, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if win_afw: acl_obj = windows_advfirewall.WindowsAdvFirewall(win_afw, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if jsl: acl_obj = srxlo.SRXlo(jsl, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if xacl: acl_obj = ciscoxr.CiscoXR(xacl, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if nft: acl_obj = nftables.Nftables(nft, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if gcefw: acl_obj = gce.GCE(gcefw, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if paloalto: acl_obj = paloaltofw.PaloAltoFW(paloalto, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) if gca: acl_obj = cloudarmor.CloudArmor(gca, exp_info) RenderACL(str(acl_obj), acl_obj.SUFFIX, output_directory, input_file, write_files) # TODO(robankeny) add additional errors. except (juniper.Error, junipersrx.Error, cisco.Error, ipset.Error, iptables.Error, speedway.Error, pcap.Error, aclgenerator.Error, aruba.Error, nftables.Error, gce.Error, cloudarmor.Error) as e: raise ACLGeneratorError('Error generating target ACL for %s:\n%s' % (input_file, e))