1 files changed, 89 insertions, 0 deletions
diff --git a/scripts/rpki/resource_set.py b/scripts/rpki/resource_set.py
new file mode 100644
index 00000000..bc1e7c16
--- /dev/null
+++ b/scripts/rpki/resource_set.py
@@ -0,0 +1,89 @@
+# $Id$
+
+import re, ipaddrs
+
+class resource_range(object):
+
+  def __init__(self, min, max):
+    assert min <= max, "Mis-ordered range: %s before %s" % (str(min), str(max))
+    self.min = min
+    self.max = max
+
+  def __cmp__(self, other):
+    c = self.min - other.min
+    if c == 0:
+      c = self.max - other.max
+    return c
+
+class resource_range_as(resource_range):
+
+  def __str__(self):
+    if self.min == self.max:
+      return str(self.min)
+    else:
+      return str(self.min) + "-" + str(self.max)
+
+class resource_range_ip(resource_range):
+
+  def __str__(self):
+    mask = self.min ^ self.max
+    prefixlen = self.min.bits
+    while mask & 1:
+      prefixlen -= 1
+      mask >>= 1
+    if mask:
+      return str(self.min) + "-" + str(self.max)
+    else:
+      return str(self.min) + "/" + str(prefixlen)
+
+class resource_range_ipv4(resource_range_ip):
+  pass
+
+class resource_range_ipv6(resource_range_ip):
+  pass
+
+class resource_set(list):
+
+  def __init__(self, s):
+    if s:
+      self.extend(map(self.parse, s.split(",")))
+      self.sort()
+      if __debug__:
+        for i in range(0, len(self) - 1):
+          assert self[i].max < self[i + 1].min, 'Resource overlap "%s"' % (s)
+
+  def __str__(self):
+    return ",".join(map(str, self))
+
+class resource_set_as(resource_set):
+
+  def parse(self, x):
+    r = re.match("^([0-9]+)-([0-9]+)$", x)
+    if r:
+      return resource_range_as(long(r.group(1)), long(r.group(2)))
+    else:
+      return resource_range_as(long(x), long(x))
+
+class resource_set_ip(resource_set):
+
+  def parse(self, x):
+    r = re.match("^([0-9:.a-fA-F]+)-([0-9:.a-fA-F]+)$", x)
+    if r:
+      return self.range_type(self.addr_type(r.group(1)), self.addr_type(r.group(2)))
+    r = re.match("^([0-9:.a-fA-F]+)/([0-9]+)$", x)
+    if r:
+      min = self.addr_type(r.group(1))
+      prefixlen = int(r.group(2))
+      mask = (1 << (self.addr_type.bits - prefixlen)) - 1
+      assert (min & mask) == 0, "Resource not in canonical form: %s" % (x)
+      max = min | mask
+      return self.range_type(min, max)
+    raise RuntimeError, 'Bad IP resource "%s"' % (x)
+
+class resource_set_ipv4(resource_set_ip):
+  addr_type = ipaddrs.v4addr
+  range_type = resource_range_ipv4
+
+class resource_set_ipv6(resource_set_ip):
+  addr_type = ipaddrs.v6addr
+  range_type = resource_range_ipv6