ipset update scripts
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

97 lines
2.6 KiB

#!/usr/bin/python
import os
import json
import requests
import re
import ipaddress
print('Reading lists of URL to download')
with open("lists.json","r") as f:
data = json.load(f)
def save_to_file(text,filename):
print("= Saved {} elements to {}".format(len(text),filename))
with open('./{}'.format(filename), mode='w') as myfile:
myfile.write('\n'.join(text))
def is_subnet(subnet):
if re.match('^([0-9]{1,3}\.){3}[0-9]{1,3}(\/([0-9]|[1-2][0-9]|3[0-2]))?$',subnet):
return True
else:
return False
def is_ipv4(ip):
if re.match("^(\d{0,3})\.(\d{0,3})\.(\d{0,3})\.(\d{0,3})$", ip):
return True
else:
return False
def is_ipv6(ip):
try:
ip_addr = ipaddress.IPv6Address(ip)
except ipaddress.AddressValueError:
return False
return ip_addr.version == 6
def extract_ipv6(source):
return re.findall(regex_ipv6_cidr,source)
def is_ipv6_subnet(ip):
try:
addr = ipaddress.IPv6Network(ip)
except Exception as e:
return False
return True
def download_list(list_name,list_url):
print("- Downloading {}".format(list_name))
filedata = ""
try:
filedata = requests.get(list_url).content.split('\n')
except Exception as e:
print("Error downloading {} : {}".format(list_name,e))
return filedata
def update_ipset_files():
ipv4_list = []
ipv6_list = []
net_list = []
for elem in ['ipv4','net']:
print("= Update {}".format(elem))
elem_list = []
for key,url in data['lists'][elem].iteritems():
dl_list = download_list(key,url)
ipv4_tab = []
ipv6_tab = []
subnet_tab = []
for line in dl_list:
if elem == "ipv4":
if is_ipv4(line):
ipv4_tab.append("add blacklist_ipv4 {} -exist".format(line))
if elem == "ipv6":
if is_ipv6_subnet(line):
ipv6_tab.append("add blacklist_ipv6 {} -exist".format(line))
if elem == "net":
if is_subnet(line):
subnet_tab.append("add blacklist_net {} -exist".format(line))
if elem == "ipv4": ipv4_list.extend(ipv4_tab)
if elem == "ipv6": ipv6_list.extend(ipv6_tab)
if elem == "net": net_list.extend(subnet_tab)
ipv4_list = sorted(set(ipv4_list))
ipv6_list = sorted(set(ipv6_list))
inet_list = sorted(set(net_list))
save_to_file(ipv4_list,"ipset_ipv4.set")
#save_to_file(ipv6_list,"ipset_ipv6.set")
save_to_file(net_list,"ipset_subnets.set")
update_ipset_files()