From 77c65245b7ebf4d4cad3360ce9950d51c76e1eef Mon Sep 17 00:00:00 2001
From: Jerry <isjerryxiao@outlook.com>
Date: Thu, 10 Nov 2022 15:47:25 +0800
Subject: [PATCH] support max-len, closes #236

---
 document/route.toml | 3 +++
 scripts/roa.py      | 9 ++++++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/document/route.toml b/document/route.toml
index 5485859..fcd55f3 100644
--- a/document/route.toml
+++ b/document/route.toml
@@ -20,6 +20,9 @@ name = ""
 # IP network description (optional).
 description = ""
 
+# Max length (optional). Default values: 29 for ipv4, 64 for ipv6
+max-len = 29
+
 # The supernet of this IP network (optional). Overlapping IP prefixes are
 # regarded as an error, so if this IP network is a subnet of another one, write
 # their IP prefix here.
diff --git a/scripts/roa.py b/scripts/roa.py
index 2da29c1..3da10a5 100755
--- a/scripts/roa.py
+++ b/scripts/roa.py
@@ -122,6 +122,8 @@ def route_to_roa(asn_table: dict):
                     continue
                 fields["asn"] = asn
                 fields["prefix"] = ip_network(prefix, strict=True)
+                fields["maxLength"] = fields.get("max-len", fields["prefix"].max_prefixlen)
+                assert fields["prefix"].prefixlen <= fields["maxLength"] <= fields["prefix"].max_prefixlen
                 supernet = fields.get("supernet")
                 fields["supernet"] = (
                     ip_network(supernet, strict=True) if supernet else None
@@ -132,7 +134,7 @@ def route_to_roa(asn_table: dict):
                     is_neo_network(fields["supernet"])
                     and fields["supernet"].supernet_of(fields["prefix"])
                 )
-                yield pick(fields, ["asn", "name", "type", "prefix", "supernet"])
+                yield pick(fields, ["asn", "name", "type", "prefix", "supernet", "maxLength"])
 
     entities = sorted(make_route(), key=lambda item: item["asn"])
     prefixes = [item["prefix"] for item in entities]
@@ -177,11 +179,12 @@ def prehandle_roa(asn_table: dict, args):
     ]
     roa6 = [r for r in roa6 if r["prefix"].prefixlen <= args.max6]
     for r in roa4:
-        r["maxLength"] = args.max
         if r["prefix"].prefixlen == max_prefixlen:
             r["maxLength"] = max_prefixlen
+        else:
+            r["maxLength"] = r["maxLength"] if r["maxLength"] <= args.max else args.max
     for r in roa6:
-        r["maxLength"] = args.max6
+        r["maxLength"] = r["maxLength"] if r["maxLength"] <= args.max6 else args.max6
     for r in (*roa4, *roa6):
         r["prefix"] = r["prefix"].with_prefixlen
     return roa4, roa6