summaryrefslogtreecommitdiff
path: root/python/vyos/firewall.py
diff options
context:
space:
mode:
authorViacheslav Hletenko <v.gletenko@vyos.io>2022-06-05 12:20:38 +0300
committerViacheslav Hletenko <v.gletenko@vyos.io>2022-06-05 09:22:31 +0000
commit654c403cebe3cdbfa7a36348c75191b47f882ac9 (patch)
treea39307f088a78d4e0b9503a2a9a0d612c949c31c /python/vyos/firewall.py
parente990b2f4c045f5d1be02915ec7d8869d5475ed4e (diff)
parentd1bdf2b9d80d2e34b7370823d6f684102d7c9f4e (diff)
downloadvyos-1x-654c403cebe3cdbfa7a36348c75191b47f882ac9.tar.gz
vyos-1x-654c403cebe3cdbfa7a36348c75191b47f882ac9.zip
Merge pull request #2 from sarthurdev/T970
firewall: T970: Maintain a domain state to fallback if resolution fails
Diffstat (limited to 'python/vyos/firewall.py')
-rw-r--r--python/vyos/firewall.py9
1 files changed, 4 insertions, 5 deletions
diff --git a/python/vyos/firewall.py b/python/vyos/firewall.py
index 8a1237ca9..b962c4f18 100644
--- a/python/vyos/firewall.py
+++ b/python/vyos/firewall.py
@@ -26,21 +26,20 @@ def get_ips_domains_dict(list_domains):
"""
Get list of IPv4 addresses by list of domains
Ex: get_ips_domains_dict(['ex1.com', 'ex2.com'])
- ['192.0.2.1', '192.0.2.2', '192.0.2.3']
+ {'ex1.com': ['192.0.2.1'], 'ex2.com': ['192.0.2.2', '192.0.2.3']}
"""
from socket import gethostbyname_ex
from socket import gaierror
- ip_list = []
+ ip_dict = {}
for domain in list_domains:
try:
_, _, ips = gethostbyname_ex(domain)
- for entry in ips:
- ip_list.append(entry)
+ ip_dict[domain] = ips
except gaierror:
pass
- return ip_list
+ return ip_dict
def nft_init_set(group_name, table="filter", family="ip"):
"""