]> git.ipfire.org Git - people/ms/libloc.git/blobdiff - src/network.c
Drop loc_network_eq in favour of loc_network_cmp
[people/ms/libloc.git] / src / network.c
index b8bc0f7d7a23dc8a4268e65eb2c9ce9427f90fdf..503bf3d02e0eb47c8af8b91b744ae3a428d01ff2 100644 (file)
 
 #include <arpa/inet.h>
 #include <assert.h>
-#include <endian.h>
 #include <errno.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 
+#ifdef HAVE_ENDIAN_H
+#  include <endian.h>
+#endif
+
 #include <loc/libloc.h>
+#include <loc/compat.h>
+#include <loc/country.h>
 #include <loc/network.h>
+#include <loc/network-list.h>
 #include <loc/private.h>
 
 struct loc_network {
        struct loc_ctx* ctx;
        int refcount;
 
-       struct in6_addr start_address;
+       int family;
+       struct in6_addr first_address;
+       struct in6_addr last_address;
        unsigned int prefix;
 
        char country_code[3];
        uint32_t asn;
+       enum loc_network_flags flags;
 };
 
 static int valid_prefix(struct in6_addr* address, unsigned int prefix) {
@@ -69,24 +78,37 @@ static struct in6_addr prefix_to_bitmask(unsigned int prefix) {
        return bitmask;
 }
 
-static struct in6_addr make_start_address(const struct in6_addr* address, unsigned int prefix) {
+static struct in6_addr make_first_address(const struct in6_addr* address, const struct in6_addr* bitmask) {
        struct in6_addr a;
-       struct in6_addr bitmask = prefix_to_bitmask(prefix);
 
        // Perform bitwise AND
        for (unsigned int i = 0; i < 4; i++)
-               a.s6_addr32[i] = address->s6_addr32[i] & bitmask.s6_addr32[i];
+               a.s6_addr32[i] = address->s6_addr32[i] & bitmask->s6_addr32[i];
 
        return a;
 }
 
-static struct in6_addr make_last_address(const struct in6_addr* address, unsigned int prefix) {
+static struct in6_addr make_last_address(const struct in6_addr* address, const struct in6_addr* bitmask) {
        struct in6_addr a;
-       struct in6_addr bitmask = prefix_to_bitmask(prefix);
 
        // Perform bitwise OR
        for (unsigned int i = 0; i < 4; i++)
-               a.s6_addr32[i] = address->s6_addr32[i] | ~bitmask.s6_addr32[i];
+               a.s6_addr32[i] = address->s6_addr32[i] | ~bitmask->s6_addr32[i];
+
+       return a;
+}
+
+static struct in6_addr address_increment(const struct in6_addr* address) {
+       struct in6_addr a = *address;
+
+       for (int octet = 15; octet >= 0; octet--) {
+               if (a.s6_addr[octet] < 255) {
+                       a.s6_addr[octet]++;
+                       break;
+               } else {
+                       a.s6_addr[octet] = 0;
+               }
+       }
 
        return a;
 }
@@ -130,28 +152,35 @@ LOC_EXPORT int loc_network_new(struct loc_ctx* ctx, struct loc_network** network
        n->ctx = loc_ref(ctx);
        n->refcount = 1;
 
-       // Store the first address in the network
-       n->start_address = make_start_address(address, prefix);
+       // Store the prefix
        n->prefix = prefix;
 
+       // Convert the prefix into a bitmask
+       struct in6_addr bitmask = prefix_to_bitmask(n->prefix);
+
+       // Store the first and last address in the network
+       n->first_address = make_first_address(address, &bitmask);
+       n->last_address = make_last_address(&n->first_address, &bitmask);
+
+       // Set family
+       if (IN6_IS_ADDR_V4MAPPED(&n->first_address))
+               n->family = AF_INET;
+       else
+               n->family = AF_INET6;
+
        DEBUG(n->ctx, "Network allocated at %p\n", n);
        *network = n;
        return 0;
 }
 
-static int loc_network_address_family(struct loc_network* network) {
-       if (IN6_IS_ADDR_V4MAPPED(&network->start_address))
-               return AF_INET;
-
-       return AF_INET6;
-}
-
 LOC_EXPORT int loc_network_new_from_string(struct loc_ctx* ctx, struct loc_network** network,
                const char* address_string) {
-       struct in6_addr start_address;
-       unsigned int prefix = 0;
+       struct in6_addr first_address;
        char* prefix_string;
-       int r = 1;
+       unsigned int prefix = 128;
+       int r = -EINVAL;
+
+       DEBUG(ctx, "Attempting to parse network %s\n", address_string);
 
        // Make a copy of the string to work on it
        char* buffer = strdup(address_string);
@@ -160,29 +189,40 @@ LOC_EXPORT int loc_network_new_from_string(struct loc_ctx* ctx, struct loc_netwo
        // Split address and prefix
        address_string = strsep(&prefix_string, "/");
 
-       // Did we find a prefix?
+       DEBUG(ctx, "  Split into address = %s, prefix = %s\n", address_string, prefix_string);
+
+       // Parse the address
+       r = loc_parse_address(ctx, address_string, &first_address);
+       if (r) {
+               DEBUG(ctx, "The address could not be parsed\n");
+               goto FAIL;
+       }
+
+       // If a prefix was given, we will try to parse it
        if (prefix_string) {
                // Convert prefix to integer
                prefix = strtol(prefix_string, NULL, 10);
 
-               if (prefix) {
-                       // Parse the address
-                       r = loc_parse_address(ctx, address_string, &start_address);
-
-                       // Map the prefix to IPv6 if needed
-                       if (IN6_IS_ADDR_V4MAPPED(&start_address))
-                               prefix += 96;
+               if (!prefix) {
+                       DEBUG(ctx, "The prefix was not parsable: %s\n", prefix_string);
+                       goto FAIL;
                }
+
+               // Map the prefix to IPv6 if needed
+               if (IN6_IS_ADDR_V4MAPPED(&first_address))
+                       prefix += 96;
        }
 
+FAIL:
        // Free temporary buffer
        free(buffer);
 
-       if (r == 0) {
-               r = loc_network_new(ctx, network, &start_address, prefix);
-       }
+       // Exit if the parsing was unsuccessful
+       if (r)
+               return r;
 
-       return r;
+       // Create a new network
+       return loc_network_new(ctx, network, &first_address, prefix);
 }
 
 LOC_EXPORT struct loc_network* loc_network_ref(struct loc_network* network) {
@@ -238,14 +278,13 @@ LOC_EXPORT char* loc_network_str(struct loc_network* network) {
 
        unsigned int prefix = network->prefix;
 
-       int family = loc_network_address_family(network);
-       switch (family) {
+       switch (network->family) {
                case AF_INET6:
-                       r = format_ipv6_address(&network->start_address, string, length);
+                       r = format_ipv6_address(&network->first_address, string, length);
                        break;
 
                case AF_INET:
-                       r = format_ipv4_address(&network->start_address, string, length);
+                       r = format_ipv4_address(&network->first_address, string, length);
                        prefix -= 96;
                        break;
 
@@ -267,16 +306,78 @@ LOC_EXPORT char* loc_network_str(struct loc_network* network) {
        return string;
 }
 
+LOC_EXPORT int loc_network_address_family(struct loc_network* network) {
+       return network->family;
+}
+
+LOC_EXPORT unsigned int loc_network_prefix(struct loc_network* network) {
+       switch (network->family) {
+               case AF_INET6:
+                       return network->prefix;
+
+               case AF_INET:
+                       return network->prefix - 96;
+       }
+
+       return 0;
+}
+
+static char* loc_network_format_address(struct loc_network* network, const struct in6_addr* address) {
+       const size_t length = INET6_ADDRSTRLEN;
+
+       char* string = malloc(length);
+       if (!string)
+               return NULL;
+
+       int r = 0;
+
+       switch (network->family) {
+               case AF_INET6:
+                       r = format_ipv6_address(address, string, length);
+                       break;
+
+               case AF_INET:
+                       r = format_ipv4_address(address, string, length);
+                       break;
+
+               default:
+                       r = -1;
+                       break;
+       }
+
+       if (r) {
+               ERROR(network->ctx, "Could not format IP address to string: %s\n", strerror(errno));
+               free(string);
+
+               return NULL;
+       }
+
+       return string;
+}
+
+LOC_EXPORT const struct in6_addr* loc_network_get_first_address(struct loc_network* network) {
+       return &network->first_address;
+}
+
+LOC_EXPORT char* loc_network_format_first_address(struct loc_network* network) {
+       return loc_network_format_address(network, &network->first_address);
+}
+
+LOC_EXPORT const struct in6_addr* loc_network_get_last_address(struct loc_network* network) {
+       return &network->last_address;
+}
+
+LOC_EXPORT char* loc_network_format_last_address(struct loc_network* network) {
+       return loc_network_format_address(network, &network->last_address);
+}
+
 LOC_EXPORT int loc_network_match_address(struct loc_network* network, const struct in6_addr* address) {
        // Address must be larger than the start address
-       if (in6_addr_cmp(&network->start_address, address) > 0)
+       if (in6_addr_cmp(&network->first_address, address) > 0)
                return 1;
 
-       // Determine the last address in this network
-       struct in6_addr last_address = make_last_address(&network->start_address, network->prefix);
-
        // Address must be smaller than the last address
-       if (in6_addr_cmp(address, &last_address) > 0)
+       if (in6_addr_cmp(&network->last_address, address) < 0)
                return 1;
 
        // The address is inside this network
@@ -294,20 +395,18 @@ LOC_EXPORT int loc_network_set_country_code(struct loc_network* network, const c
                return 0;
        }
 
-       // Country codes must be two characters
-       if (strlen(country_code) != 2)
+       // Check country code
+       if (!loc_country_code_is_valid(country_code))
                return -EINVAL;
 
-       for (unsigned int i = 0; i < 3; i++) {
-               network->country_code[i] = country_code[i];
-       }
+       loc_country_code_copy(network->country_code, country_code);
 
        return 0;
 }
 
 LOC_EXPORT int loc_network_match_country_code(struct loc_network* network, const char* country_code) {
-       // Country codes must be two characters
-       if (strlen(country_code) != 2)
+       // Check country code
+       if (!loc_country_code_is_valid(country_code))
                return -EINVAL;
 
        return (network->country_code[0] == country_code[0])
@@ -324,39 +423,404 @@ LOC_EXPORT int loc_network_set_asn(struct loc_network* network, uint32_t asn) {
        return 0;
 }
 
-LOC_EXPORT int loc_network_to_database_v0(struct loc_network* network, struct loc_database_network_v0* dbobj) {
-       // Add country code
-       for (unsigned int i = 0; i < 2; i++) {
-               dbobj->country_code[i] = network->country_code[i] ? network->country_code[i] : '\0';
+LOC_EXPORT int loc_network_match_asn(struct loc_network* network, uint32_t asn) {
+       return network->asn == asn;
+}
+
+LOC_EXPORT int loc_network_has_flag(struct loc_network* network, uint32_t flag) {
+       return network->flags & flag;
+}
+
+LOC_EXPORT int loc_network_set_flag(struct loc_network* network, uint32_t flag) {
+       network->flags |= flag;
+
+       return 0;
+}
+
+LOC_EXPORT int loc_network_match_flag(struct loc_network* network, uint32_t flag) {
+       return loc_network_has_flag(network, flag);
+}
+
+LOC_EXPORT int loc_network_cmp(struct loc_network* self, struct loc_network* other) {
+       // Compare family
+       if (self->family > other->family)
+               return 1;
+       else if (self->family < other->family)
+               return -1;
+
+       // Compare address
+       int r = in6_addr_cmp(&self->first_address, &other->first_address);
+       if (r)
+               return r;
+
+       // Compare prefix
+       if (self->prefix > other->prefix)
+               return 1;
+       else if (self->prefix < other->prefix)
+               return -1;
+
+       // Both networks are equal
+       return 0;
+}
+
+LOC_EXPORT int loc_network_gt(struct loc_network* self, struct loc_network* other) {
+       // Families must match
+       if (self->family != other->family)
+               return -1;
+
+       int r = in6_addr_cmp(&self->first_address, &other->first_address);
+
+       switch (r) {
+               // Smaller
+               case -1:
+                       return 0;
+
+               // Larger
+               case 1:
+                       return 1;
+
+               default:
+                       break;
+       }
+
+       if (self->prefix > other->prefix)
+               return 1;
+
+       // Dunno
+       return 0;
+}
+
+LOC_EXPORT int loc_network_overlaps(struct loc_network* self, struct loc_network* other) {
+       if (loc_network_match_address(self, &other->first_address) == 0)
+               return 1;
+
+       if (loc_network_match_address(self, &other->last_address) == 0)
+               return 1;
+
+       if (loc_network_match_address(other, &self->first_address) == 0)
+               return 1;
+
+       if (loc_network_match_address(other, &self->last_address) == 0)
+               return 1;
+
+       return 0;
+}
+
+LOC_EXPORT int loc_network_is_subnet(struct loc_network* self, struct loc_network* other) {
+       // Check family
+       if (self->family != other->family)
+               return 0;
+
+       // The prefix must be smaller (this avoids the more complex comparisons later)
+       if (self->prefix > other->prefix)
+               return 0;
+
+       // If the start address of the other network is smaller than this network,
+       // it cannot be a subnet.
+       if (in6_addr_cmp(&self->first_address, &other->first_address) > 0)
+               return 0;
+
+       // If the end address of the other network is greater than this network,
+       // it cannot be a subnet.
+       if (in6_addr_cmp(&self->last_address, &other->last_address) < 0)
+               return 0;
+
+       return 1;
+}
+
+LOC_EXPORT int loc_network_subnets(struct loc_network* network,
+               struct loc_network** subnet1, struct loc_network** subnet2) {
+       int r;
+       *subnet1 = NULL;
+       *subnet2 = NULL;
+
+       // New prefix length
+       unsigned int prefix = network->prefix + 1;
+
+       // Check if the new prefix is valid
+       if (valid_prefix(&network->first_address, prefix))
+               return -1;
+
+       // Create the first half of the network
+       r = loc_network_new(network->ctx, subnet1, &network->first_address, prefix);
+       if (r)
+               return r;
+
+       // The next subnet starts after the first one
+       struct in6_addr first_address = address_increment(&(*subnet1)->last_address);
+
+       // Create the second half of the network
+       r = loc_network_new(network->ctx, subnet2, &first_address, prefix);
+       if (r)
+               return r;
+
+       // Copy country code
+       const char* country_code = loc_network_get_country_code(network);
+       if (country_code) {
+               loc_network_set_country_code(*subnet1, country_code);
+               loc_network_set_country_code(*subnet2, country_code);
+       }
+
+       // Copy ASN
+       uint32_t asn = loc_network_get_asn(network);
+       if (asn) {
+               loc_network_set_asn(*subnet1, asn);
+               loc_network_set_asn(*subnet2, asn);
+       }
+
+       return 0;
+}
+
+static int __loc_network_exclude(struct loc_network* network,
+               struct loc_network* other, struct loc_network_list* list) {
+       struct loc_network* subnet1 = NULL;
+       struct loc_network* subnet2 = NULL;
+
+       int r = loc_network_subnets(network, &subnet1, &subnet2);
+       if (r)
+               goto ERROR;
+
+       if (loc_network_cmp(other, subnet1) == 0) {
+               r = loc_network_list_push(list, subnet2);
+               if (r)
+                       goto ERROR;
+
+       } else if (loc_network_cmp(other, subnet2) == 0) {
+               r = loc_network_list_push(list, subnet1);
+               if (r)
+                       goto ERROR;
+
+       } else  if (loc_network_is_subnet(subnet1, other)) {
+               r = loc_network_list_push(list, subnet2);
+               if (r)
+                       goto ERROR;
+
+               r = __loc_network_exclude(subnet1, other, list);
+               if (r)
+                       goto ERROR;
+
+       } else if (loc_network_is_subnet(subnet2, other)) {
+               r = loc_network_list_push(list, subnet1);
+               if (r)
+                       goto ERROR;
+
+               r = __loc_network_exclude(subnet2, other, list);
+               if (r)
+                       goto ERROR;
+
+       } else {
+               ERROR(network->ctx, "We should never get here\n");
+               r = 1;
+               goto ERROR;
+       }
+
+ERROR:
+       if (subnet1)
+               loc_network_unref(subnet1);
+
+       if (subnet2)
+               loc_network_unref(subnet2);
+
+       return r;
+}
+
+static int __loc_network_exclude_to_list(struct loc_network* self,
+               struct loc_network* other, struct loc_network_list* list) {
+       // Family must match
+       if (self->family != other->family) {
+               DEBUG(self->ctx, "Family mismatch\n");
+
+               return 1;
+       }
+
+       // Other must be a subnet of self
+       if (!loc_network_is_subnet(self, other)) {
+               DEBUG(self->ctx, "Network %p is not contained in network %p\n", other, self);
+
+               return 1;
+       }
+
+       // We cannot perform this operation if both networks equal
+       if (loc_network_cmp(self, other) == 0) {
+               DEBUG(self->ctx, "Networks %p and %p are equal\n", self, other);
+
+               return 1;
+       }
+
+       return __loc_network_exclude(self, other, list);
+}
+
+LOC_EXPORT struct loc_network_list* loc_network_exclude(
+               struct loc_network* self, struct loc_network* other) {
+       struct loc_network_list* list;
+
+#ifdef ENABLE_DEBUG
+       char* n1 = loc_network_str(self);
+       char* n2 = loc_network_str(other);
+
+       DEBUG(self->ctx, "Returning %s excluding %s...\n", n1, n2);
+
+       free(n1);
+       free(n2);
+#endif
+
+       // Create a new list with the result
+       int r = loc_network_list_new(self->ctx, &list);
+       if (r) {
+               ERROR(self->ctx, "Could not create network list: %d\n", r);
+
+               return NULL;
+       }
+
+       r = __loc_network_exclude_to_list(self, other, list);
+       if (r) {
+               loc_network_list_unref(list);
+
+               return NULL;
+       }
+
+       // Return the result
+       return list;
+}
+
+LOC_EXPORT struct loc_network_list* loc_network_exclude_list(
+               struct loc_network* network, struct loc_network_list* list) {
+       struct loc_network_list* to_check;
+
+       // Create a new list with all networks to look at
+       int r = loc_network_list_new(network->ctx, &to_check);
+       if (r)
+               return NULL;
+
+       struct loc_network* subnet = NULL;
+       struct loc_network_list* subnets = NULL;
+
+       for (unsigned int i = 0; i < loc_network_list_size(list); i++) {
+               subnet = loc_network_list_get(list, i);
+
+               // Find all excluded networks
+               if (!loc_network_list_contains(to_check, subnet)) {
+                       r = __loc_network_exclude_to_list(network, subnet, to_check);
+                       if (r) {
+                               loc_network_list_unref(to_check);
+                               loc_network_unref(subnet);
+
+                               return NULL;
+                       }
+               }
+
+               // Cleanup
+               loc_network_unref(subnet);
+       }
+
+       r = loc_network_list_new(network->ctx, &subnets);
+       if (r) {
+               loc_network_list_unref(to_check);
+               return NULL;
        }
 
+       while (!loc_network_list_empty(to_check)) {
+               struct loc_network* subnet_to_check = loc_network_list_pop(to_check);
+
+               // Marks whether this subnet passed all checks
+               int passed = 1;
+
+               for (unsigned int i = 0; i < loc_network_list_size(list); i++) {
+                       subnet = loc_network_list_get(list, i);
+
+                       // Drop this subnet if is is already in list
+                       if (loc_network_cmp(subnet_to_check, subnet) == 0) {
+                               passed = 0;
+                               loc_network_unref(subnet);
+                               break;
+                       }
+
+                       // Drop this subnet if is a subnet of another subnet
+                       if (loc_network_is_subnet(subnet_to_check, subnet)) {
+                               passed = 0;
+                               loc_network_unref(subnet);
+                               break;
+                       }
+
+                       // Break it down if it overlaps
+                       if (loc_network_overlaps(subnet_to_check, subnet)) {
+                               passed = 0;
+
+                               struct loc_network_list* excluded = loc_network_exclude(subnet_to_check, subnet);
+                               if (excluded) {
+                                       loc_network_list_merge(to_check, excluded);
+                                       loc_network_list_unref(excluded);
+                               }
+
+                               loc_network_unref(subnet);
+                               break;
+                       }
+
+                       loc_network_unref(subnet);
+               }
+
+               if (passed) {
+                       r = loc_network_list_push(subnets, subnet_to_check);
+               }
+
+               loc_network_unref(subnet_to_check);
+       }
+
+       loc_network_list_unref(to_check);
+
+       // Sort the result
+       loc_network_list_sort(subnets);
+
+       return subnets;
+}
+
+LOC_EXPORT int loc_network_to_database_v1(struct loc_network* network, struct loc_database_network_v1* dbobj) {
+       // Add country code
+       loc_country_code_copy(dbobj->country_code, network->country_code);
+
        // Add ASN
        dbobj->asn = htobe32(network->asn);
 
+       // Flags
+       dbobj->flags = htobe16(network->flags);
+
        return 0;
 }
 
-LOC_EXPORT int loc_network_new_from_database_v0(struct loc_ctx* ctx, struct loc_network** network,
-               struct in6_addr* address, unsigned int prefix, const struct loc_database_network_v0* dbobj) {
+LOC_EXPORT int loc_network_new_from_database_v1(struct loc_ctx* ctx, struct loc_network** network,
+               struct in6_addr* address, unsigned int prefix, const struct loc_database_network_v1* dbobj) {
+       char country_code[3] = "\0\0";
+
        int r = loc_network_new(ctx, network, address, prefix);
-       if (r)
+       if (r) {
+               ERROR(ctx, "Could not allocate a new network: %s", strerror(-r));
                return r;
+       }
 
        // Import country code
-       char country_code[3];
-       for (unsigned int i = 0; i < 2; i++) {
-               country_code[i] = dbobj->country_code[i];
-       }
-       country_code[2] = '\0';
+       loc_country_code_copy(country_code, dbobj->country_code);
 
        r = loc_network_set_country_code(*network, country_code);
-       if (r)
+       if (r) {
+               ERROR(ctx, "Could not set country code: %s\n", country_code);
                return r;
+       }
 
        // Import ASN
-       r = loc_network_set_asn(*network, be32toh(dbobj->asn));
-       if (r)
+       uint32_t asn = be32toh(dbobj->asn);
+       r = loc_network_set_asn(*network, asn);
+       if (r) {
+               ERROR(ctx, "Could not set ASN: %d\n", asn);
                return r;
+       }
+
+       // Import flags
+       int flags = be16toh(dbobj->flags);
+       r = loc_network_set_flag(*network, flags);
+       if (r) {
+               ERROR(ctx, "Could not set flags: %d\n", flags);
+               return r;
+       }
 
        return 0;
 }
@@ -378,7 +842,7 @@ struct loc_network_tree_node {
        struct loc_network* network;
 };
 
-LOC_EXPORT int loc_network_tree_new(struct loc_ctx* ctx, struct loc_network_tree** tree) {
+int loc_network_tree_new(struct loc_ctx* ctx, struct loc_network_tree** tree) {
        struct loc_network_tree* t = calloc(1, sizeof(*t));
        if (!t)
                return -ENOMEM;
@@ -398,7 +862,7 @@ LOC_EXPORT int loc_network_tree_new(struct loc_ctx* ctx, struct loc_network_tree
        return 0;
 }
 
-LOC_EXPORT struct loc_network_tree_node* loc_network_tree_get_root(struct loc_network_tree* tree) {
+struct loc_network_tree_node* loc_network_tree_get_root(struct loc_network_tree* tree) {
        return loc_network_tree_node_ref(tree->root);
 }
 
@@ -470,7 +934,7 @@ static int __loc_network_tree_walk(struct loc_ctx* ctx, struct loc_network_tree_
        return 0;
 }
 
-LOC_EXPORT int loc_network_tree_walk(struct loc_network_tree* tree,
+int loc_network_tree_walk(struct loc_network_tree* tree,
                int(*filter_callback)(struct loc_network* network, void* data),
                int(*callback)(struct loc_network* network, void* data), void* data) {
        return __loc_network_tree_walk(tree->ctx, tree->root, filter_callback, callback, data);
@@ -485,7 +949,7 @@ static void loc_network_tree_free(struct loc_network_tree* tree) {
        free(tree);
 }
 
-LOC_EXPORT struct loc_network_tree* loc_network_tree_unref(struct loc_network_tree* tree) {
+struct loc_network_tree* loc_network_tree_unref(struct loc_network_tree* tree) {
        if (--tree->refcount > 0)
                return tree;
 
@@ -506,17 +970,17 @@ static int __loc_network_tree_dump(struct loc_network* network, void* data) {
        return 0;
 }
 
-LOC_EXPORT int loc_network_tree_dump(struct loc_network_tree* tree) {
+int loc_network_tree_dump(struct loc_network_tree* tree) {
        DEBUG(tree->ctx, "Dumping network tree at %p\n", tree);
 
        return loc_network_tree_walk(tree, NULL, __loc_network_tree_dump, NULL);
 }
 
-LOC_EXPORT int loc_network_tree_add_network(struct loc_network_tree* tree, struct loc_network* network) {
+int loc_network_tree_add_network(struct loc_network_tree* tree, struct loc_network* network) {
        DEBUG(tree->ctx, "Adding network %p to tree %p\n", network, tree);
 
        struct loc_network_tree_node* node = loc_network_tree_get_path(tree,
-                       &network->start_address, network->prefix);
+                       &network->first_address, network->prefix);
        if (!node) {
                ERROR(tree->ctx, "Could not find a node\n");
                return -ENOMEM;
@@ -543,7 +1007,7 @@ static int __loc_network_tree_count(struct loc_network* network, void* data) {
        return 0;
 }
 
-LOC_EXPORT size_t loc_network_tree_count_networks(struct loc_network_tree* tree) {
+size_t loc_network_tree_count_networks(struct loc_network_tree* tree) {
        size_t counter = 0;
 
        int r = loc_network_tree_walk(tree, NULL, __loc_network_tree_count, &counter);
@@ -565,11 +1029,11 @@ static size_t __loc_network_tree_count_nodes(struct loc_network_tree_node* node)
        return counter;
 }
 
-LOC_EXPORT size_t loc_network_tree_count_nodes(struct loc_network_tree* tree) {
+size_t loc_network_tree_count_nodes(struct loc_network_tree* tree) {
        return __loc_network_tree_count_nodes(tree->root);
 }
 
-LOC_EXPORT int loc_network_tree_node_new(struct loc_ctx* ctx, struct loc_network_tree_node** node) {
+int loc_network_tree_node_new(struct loc_ctx* ctx, struct loc_network_tree_node** node) {
        struct loc_network_tree_node* n = calloc(1, sizeof(*n));
        if (!n)
                return -ENOMEM;
@@ -584,7 +1048,7 @@ LOC_EXPORT int loc_network_tree_node_new(struct loc_ctx* ctx, struct loc_network
        return 0;
 }
 
-LOC_EXPORT struct loc_network_tree_node* loc_network_tree_node_ref(struct loc_network_tree_node* node) {
+struct loc_network_tree_node* loc_network_tree_node_ref(struct loc_network_tree_node* node) {
        if (node)
                node->refcount++;
 
@@ -607,7 +1071,7 @@ static void loc_network_tree_node_free(struct loc_network_tree_node* node) {
        free(node);
 }
 
-LOC_EXPORT struct loc_network_tree_node* loc_network_tree_node_unref(struct loc_network_tree_node* node) {
+struct loc_network_tree_node* loc_network_tree_node_unref(struct loc_network_tree_node* node) {
        if (!node)
                return NULL;
 
@@ -618,7 +1082,7 @@ LOC_EXPORT struct loc_network_tree_node* loc_network_tree_node_unref(struct loc_
        return NULL;
 }
 
-LOC_EXPORT struct loc_network_tree_node* loc_network_tree_node_get(struct loc_network_tree_node* node, unsigned int index) {
+struct loc_network_tree_node* loc_network_tree_node_get(struct loc_network_tree_node* node, unsigned int index) {
        if (index == 0)
                node = node->zero;
        else
@@ -630,10 +1094,10 @@ LOC_EXPORT struct loc_network_tree_node* loc_network_tree_node_get(struct loc_ne
        return loc_network_tree_node_ref(node);
 }
 
-LOC_EXPORT int loc_network_tree_node_is_leaf(struct loc_network_tree_node* node) {
+int loc_network_tree_node_is_leaf(struct loc_network_tree_node* node) {
        return (!!node->network);
 }
 
-LOC_EXPORT struct loc_network* loc_network_tree_node_get_network(struct loc_network_tree_node* node) {
+struct loc_network* loc_network_tree_node_get_network(struct loc_network_tree_node* node) {
        return loc_network_ref(node->network);
 }