From dee933e614f27a8639480f65b60b526b0250eff1 Mon Sep 17 00:00:00 2001
From: zonyitoo <zonyitoo@gmail.com>
Date: Tue, 6 Apr 2021 09:10:49 +0800
Subject: [PATCH] add ipset generataor

---
 configs/genipset.py | 64 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 configs/genipset.py

diff --git a/configs/genipset.py b/configs/genipset.py
new file mode 100644
index 00000000..29885e24
--- /dev/null
+++ b/configs/genipset.py
@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+import sys
+import time
+
+
+APNIC_DELEGATED_LATEST = "https://ftp.apnic.net/apnic/stats/apnic/delegated-apnic-latest"
+
+
+def get_apnic_delegated():
+    from urllib import request
+
+    u = request.urlopen(APNIC_DELEGATED_LATEST)
+    return u.read().decode('utf-8')
+
+
+def generate_ipset(content, name, location_set, type_set, output_file):
+    cidr_trans = {}
+    for i in range(0, 32):
+        cidr_trans[2 ** (32 - i - 1)] = i + 1
+
+    for line in content.splitlines():
+        if line.startswith('#'):
+            continue
+
+        splits = line.split('|')
+        if len(splits) == 7 and splits[0] == 'apnic':
+            location = splits[1]
+            addr_type = splits[2]
+            if location in location_set and addr_type in type_set:
+                start = splits[3]
+                mask = cidr_trans[int(splits[4])] if addr_type == 'ipv4' else int(
+                    splits[4])
+                output_file.write(
+                    'add {} {}/{} -exist\n'.format(name, start, mask))
+
+
+if __name__ == '__main__':
+    import argparse
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--name', '-n', help='Name of ipset',
+                        type=str, required=True)
+    parser.add_argument('--location', '-l', help='Location to be filtered, like CN',
+                        nargs='+', type=str, required=True)
+    parser.add_argument('--address-type', '-t', help='Address type, like ipv4, ipv6',
+                        nargs='+', type=str, required=True, choices=['ipv4', 'ipv6'])
+    parser.add_argument(
+        '--output', '-o', help='Output file, default to stdout', type=str, required=False)
+
+    args = parser.parse_args()
+
+    name = args.name
+    location_set = set(args.location)
+    type_set = set(args.address_type)
+
+    start_time = time.time()
+    data = get_apnic_delegated()
+
+    if hasattr(parser, 'output'):
+        with open(parser.output, 'w') as fp:
+            generate_ipset(data, name, location_set, type_set, fp)
+    else:
+        generate_ipset(data, name, location_set, type_set, sys.stdout)