def create_prefix_tree(json_data): # Create a new tree rtree = radix.Radix() # Adding a node returns a RadixNode object. You can create # arbitrary members in its 'data' dict to store your data. # Each node contains a prefix (which a router anounce) # as search value and as data --> (asn, bgp router-id, interface # name of super-prefix of that prefix) of router for i in json_data: prefixes_list = i["prefixes"] for j in prefixes_list: mask = str(IPAddress(j["mask"]).netmask_bits()) cidr = j["network"] + "/" + mask # find out in which interface name this subprefix match interface_name = None interfaces_list = i["interfaces"] for k in interfaces_list: interface_mask = str(IPAddress(k["interface_mask"]).netmask_bits()) interface_cidr = k["interface_ip"] + "/" + interface_mask s1 = IPSet([interface_cidr]) s2 = IPSet([cidr]) if s1.issuperset(s2) == True: # we found the interface of the superprefix of current subprefix interface_name = k["interface_name"] break # search if prefix already exists in tree tmp_node = rtree.search_exact(cidr) if tmp_node == None: # prefix does not exist rnode = rtree.add(cidr) rnode.data["data_list"] = [] rnode.data["data_list"].append( (str(i["origin_as"][0]["asn"]), i["bgp_router_id"][0]["router_id"], interface_name)) else: # prefix exist -> update list tmp_node.data["data_list"].append( (str(i["origin_as"][0]["asn"]), i["bgp_router_id"][0]["router_id"], interface_name)) return rtree
def test_ipset_supersets_and_subsets(): s1 = IPSet(['192.0.2.0/24', '192.0.4.0/24']) s2 = IPSet(['192.0.2.0', '192.0.4.0']) assert s1.issuperset(s2) assert s2.issubset(s1) assert not s2.issuperset(s1) assert not s1.issubset(s2) ipv4_addr_space = IPSet(['0.0.0.0/0']) private = IPSet([ '10.0.0.0/8', '172.16.0.0/12', '192.0.2.0/24', '192.168.0.0/16', '239.192.0.0/14' ]) reserved = IPSet([ '225.0.0.0/8', '226.0.0.0/7', '228.0.0.0/6', '234.0.0.0/7', '236.0.0.0/7', '238.0.0.0/8', '240.0.0.0/4' ]) unavailable = reserved | private available = ipv4_addr_space ^ unavailable assert [ tuple(map(str, (cidr, cidr[0], cidr[-1]))) for cidr in available.iter_cidrs() ] == [ ('0.0.0.0/5', '0.0.0.0', '7.255.255.255'), ('8.0.0.0/7', '8.0.0.0', '9.255.255.255'), ('11.0.0.0/8', '11.0.0.0', '11.255.255.255'), ('12.0.0.0/6', '12.0.0.0', '15.255.255.255'), ('16.0.0.0/4', '16.0.0.0', '31.255.255.255'), ('32.0.0.0/3', '32.0.0.0', '63.255.255.255'), ('64.0.0.0/2', '64.0.0.0', '127.255.255.255'), ('128.0.0.0/3', '128.0.0.0', '159.255.255.255'), ('160.0.0.0/5', '160.0.0.0', '167.255.255.255'), ('168.0.0.0/6', '168.0.0.0', '171.255.255.255'), ('172.0.0.0/12', '172.0.0.0', '172.15.255.255'), ('172.32.0.0/11', '172.32.0.0', '172.63.255.255'), ('172.64.0.0/10', '172.64.0.0', '172.127.255.255'), ('172.128.0.0/9', '172.128.0.0', '172.255.255.255'), ('173.0.0.0/8', '173.0.0.0', '173.255.255.255'), ('174.0.0.0/7', '174.0.0.0', '175.255.255.255'), ('176.0.0.0/4', '176.0.0.0', '191.255.255.255'), ('192.0.0.0/23', '192.0.0.0', '192.0.1.255'), ('192.0.3.0/24', '192.0.3.0', '192.0.3.255'), ('192.0.4.0/22', '192.0.4.0', '192.0.7.255'), ('192.0.8.0/21', '192.0.8.0', '192.0.15.255'), ('192.0.16.0/20', '192.0.16.0', '192.0.31.255'), ('192.0.32.0/19', '192.0.32.0', '192.0.63.255'), ('192.0.64.0/18', '192.0.64.0', '192.0.127.255'), ('192.0.128.0/17', '192.0.128.0', '192.0.255.255'), ('192.1.0.0/16', '192.1.0.0', '192.1.255.255'), ('192.2.0.0/15', '192.2.0.0', '192.3.255.255'), ('192.4.0.0/14', '192.4.0.0', '192.7.255.255'), ('192.8.0.0/13', '192.8.0.0', '192.15.255.255'), ('192.16.0.0/12', '192.16.0.0', '192.31.255.255'), ('192.32.0.0/11', '192.32.0.0', '192.63.255.255'), ('192.64.0.0/10', '192.64.0.0', '192.127.255.255'), ('192.128.0.0/11', '192.128.0.0', '192.159.255.255'), ('192.160.0.0/13', '192.160.0.0', '192.167.255.255'), ('192.169.0.0/16', '192.169.0.0', '192.169.255.255'), ('192.170.0.0/15', '192.170.0.0', '192.171.255.255'), ('192.172.0.0/14', '192.172.0.0', '192.175.255.255'), ('192.176.0.0/12', '192.176.0.0', '192.191.255.255'), ('192.192.0.0/10', '192.192.0.0', '192.255.255.255'), ('193.0.0.0/8', '193.0.0.0', '193.255.255.255'), ('194.0.0.0/7', '194.0.0.0', '195.255.255.255'), ('196.0.0.0/6', '196.0.0.0', '199.255.255.255'), ('200.0.0.0/5', '200.0.0.0', '207.255.255.255'), ('208.0.0.0/4', '208.0.0.0', '223.255.255.255'), ('224.0.0.0/8', '224.0.0.0', '224.255.255.255'), ('232.0.0.0/7', '232.0.0.0', '233.255.255.255'), ('239.0.0.0/9', '239.0.0.0', '239.127.255.255'), ('239.128.0.0/10', '239.128.0.0', '239.191.255.255'), ('239.196.0.0/14', '239.196.0.0', '239.199.255.255'), ('239.200.0.0/13', '239.200.0.0', '239.207.255.255'), ('239.208.0.0/12', '239.208.0.0', '239.223.255.255'), ('239.224.0.0/11', '239.224.0.0', '239.255.255.255'), ] assert ipv4_addr_space ^ available == IPSet([ '10.0.0.0/8', '172.16.0.0/12', '192.0.2.0/24', '192.168.0.0/16', '225.0.0.0/8', '226.0.0.0/7', '228.0.0.0/6', '234.0.0.0/7', '236.0.0.0/7', '238.0.0.0/8', '239.192.0.0/14', '240.0.0.0/4', ])
def test_ipset_supersets_and_subsets(): s1 = IPSet(['192.0.2.0/24', '192.0.4.0/24']) s2 = IPSet(['192.0.2.0', '192.0.4.0']) assert s1.issuperset(s2) assert s2.issubset(s1) assert not s2.issuperset(s1) assert not s1.issubset(s2) ipv4_addr_space = IPSet(['0.0.0.0/0']) private = IPSet(['10.0.0.0/8', '172.16.0.0/12', '192.0.2.0/24', '192.168.0.0/16', '239.192.0.0/14']) reserved = IPSet(['225.0.0.0/8', '226.0.0.0/7', '228.0.0.0/6', '234.0.0.0/7', '236.0.0.0/7', '238.0.0.0/8', '240.0.0.0/4']) unavailable = reserved | private available = ipv4_addr_space ^ unavailable assert [tuple(map(str, (cidr, cidr[0], cidr[-1]))) for cidr in available.iter_cidrs()] == [ ('0.0.0.0/5', '0.0.0.0', '7.255.255.255'), ('8.0.0.0/7', '8.0.0.0', '9.255.255.255'), ('11.0.0.0/8', '11.0.0.0', '11.255.255.255'), ('12.0.0.0/6', '12.0.0.0', '15.255.255.255'), ('16.0.0.0/4', '16.0.0.0', '31.255.255.255'), ('32.0.0.0/3', '32.0.0.0', '63.255.255.255'), ('64.0.0.0/2', '64.0.0.0', '127.255.255.255'), ('128.0.0.0/3', '128.0.0.0', '159.255.255.255'), ('160.0.0.0/5', '160.0.0.0', '167.255.255.255'), ('168.0.0.0/6', '168.0.0.0', '171.255.255.255'), ('172.0.0.0/12', '172.0.0.0', '172.15.255.255'), ('172.32.0.0/11', '172.32.0.0', '172.63.255.255'), ('172.64.0.0/10', '172.64.0.0', '172.127.255.255'), ('172.128.0.0/9', '172.128.0.0', '172.255.255.255'), ('173.0.0.0/8', '173.0.0.0', '173.255.255.255'), ('174.0.0.0/7', '174.0.0.0', '175.255.255.255'), ('176.0.0.0/4', '176.0.0.0', '191.255.255.255'), ('192.0.0.0/23', '192.0.0.0', '192.0.1.255'), ('192.0.3.0/24', '192.0.3.0', '192.0.3.255'), ('192.0.4.0/22', '192.0.4.0', '192.0.7.255'), ('192.0.8.0/21', '192.0.8.0', '192.0.15.255'), ('192.0.16.0/20', '192.0.16.0', '192.0.31.255'), ('192.0.32.0/19', '192.0.32.0', '192.0.63.255'), ('192.0.64.0/18', '192.0.64.0', '192.0.127.255'), ('192.0.128.0/17', '192.0.128.0', '192.0.255.255'), ('192.1.0.0/16', '192.1.0.0', '192.1.255.255'), ('192.2.0.0/15', '192.2.0.0', '192.3.255.255'), ('192.4.0.0/14', '192.4.0.0', '192.7.255.255'), ('192.8.0.0/13', '192.8.0.0', '192.15.255.255'), ('192.16.0.0/12', '192.16.0.0', '192.31.255.255'), ('192.32.0.0/11', '192.32.0.0', '192.63.255.255'), ('192.64.0.0/10', '192.64.0.0', '192.127.255.255'), ('192.128.0.0/11', '192.128.0.0', '192.159.255.255'), ('192.160.0.0/13', '192.160.0.0', '192.167.255.255'), ('192.169.0.0/16', '192.169.0.0', '192.169.255.255'), ('192.170.0.0/15', '192.170.0.0', '192.171.255.255'), ('192.172.0.0/14', '192.172.0.0', '192.175.255.255'), ('192.176.0.0/12', '192.176.0.0', '192.191.255.255'), ('192.192.0.0/10', '192.192.0.0', '192.255.255.255'), ('193.0.0.0/8', '193.0.0.0', '193.255.255.255'), ('194.0.0.0/7', '194.0.0.0', '195.255.255.255'), ('196.0.0.0/6', '196.0.0.0', '199.255.255.255'), ('200.0.0.0/5', '200.0.0.0', '207.255.255.255'), ('208.0.0.0/4', '208.0.0.0', '223.255.255.255'), ('224.0.0.0/8', '224.0.0.0', '224.255.255.255'), ('232.0.0.0/7', '232.0.0.0', '233.255.255.255'), ('239.0.0.0/9', '239.0.0.0', '239.127.255.255'), ('239.128.0.0/10', '239.128.0.0', '239.191.255.255'), ('239.196.0.0/14', '239.196.0.0', '239.199.255.255'), ('239.200.0.0/13', '239.200.0.0', '239.207.255.255'), ('239.208.0.0/12', '239.208.0.0', '239.223.255.255'), ('239.224.0.0/11', '239.224.0.0', '239.255.255.255'), ] assert ipv4_addr_space ^ available == IPSet([ '10.0.0.0/8', '172.16.0.0/12', '192.0.2.0/24', '192.168.0.0/16', '225.0.0.0/8', '226.0.0.0/7', '228.0.0.0/6', '234.0.0.0/7', '236.0.0.0/7', '238.0.0.0/8', '239.192.0.0/14', '240.0.0.0/4', ])