diff options
Diffstat (limited to 'python')
| -rwxr-xr-x | python/vyos/firewall.py | 2 | ||||
| -rw-r--r-- | python/vyos/geoip.py | 8 |
2 files changed, 5 insertions, 5 deletions
diff --git a/python/vyos/firewall.py b/python/vyos/firewall.py index f68c317f9..08f978f39 100755 --- a/python/vyos/firewall.py +++ b/python/vyos/firewall.py @@ -259,7 +259,7 @@ def parse_rule(rule_conf, hook, fw_name, rule_id, ip_name): # for policy if hook == 'route' or hook == 'route6': hook_name = hook - output.append(f'{ip_name} {prefix}addr {operator} @GEOIP_{geoip_prefix}{def_suffix}_{hook_name}_{fw_name}_{rule_id}') + output.append(f'{ip_name} {prefix}addr {operator} @GEOIP_{geoip_prefix}{def_suffix}_{hook_name}_{fw_name}_{rule_id}_{prefix}') if 'mac_address' in side_conf: suffix = side_conf['mac_address'] diff --git a/python/vyos/geoip.py b/python/vyos/geoip.py index 70c74b04f..eb81f4ce8 100644 --- a/python/vyos/geoip.py +++ b/python/vyos/geoip.py @@ -266,13 +266,13 @@ def geoip_update(firewall=None, policy=None): version = 6 if path[0] == 'ipv6' else 4 vprefix = '6' if version == 6 else '' - set_name = f'GEOIP_CC{vprefix}_{path[1]}_{path[2]}_{path[4]}' + set_name = f'GEOIP_CC{vprefix}_{path[1]}_{path[2]}_{path[4]}_{path[5][0]}' firewall_sets[f'v{version}'][set_name] = db_return_cc_ranges(codes, version) for asns, path in dict_search_recursive(firewall, 'asn'): version = 6 if path[0] == 'ipv6' else 4 vprefix = '6' if version == 6 else '' - set_name = f'GEOIP_ASN{vprefix}_{path[1]}_{path[2]}_{path[4]}' + set_name = f'GEOIP_ASN{vprefix}_{path[1]}_{path[2]}_{path[4]}_{path[5][0]}' firewall_sets[f'v{version}'][set_name] = db_return_asn_ranges(asns, version) if policy: @@ -282,13 +282,13 @@ def geoip_update(firewall=None, policy=None): version = 6 if path[0] == 'route6' else 4 vprefix = '6' if version == 6 else '' - set_name = f'GEOIP_CC{vprefix}_{path[0]}_{path[1]}_{path[3]}' + set_name = f'GEOIP_CC{vprefix}_{path[0]}_{path[1]}_{path[3]}_{path[4][0]}' policy_sets[f'v{version}'][set_name] = db_return_cc_ranges(codes, version) for asns, path in dict_search_recursive(policy, 'asn'): version = 6 if path[0] == 'route6' else 4 vprefix = '6' if version == 6 else '' - set_name = f'GEOIP_ASN{vprefix}_{path[0]}_{path[1]}_{path[3]}' + set_name = f'GEOIP_ASN{vprefix}_{path[0]}_{path[1]}_{path[3]}_{path[4][0]}' policy_sets[f'v{version}'][set_name] = db_return_asn_ranges(asns, version) render( |