]> git.ipfire.org Git - location/libloc.git/blobdiff - src/test-network.c
address: Add some simple tests
[location/libloc.git] / src / test-network.c
index 85eca000d5e36a92bf5fbbdff584057eebd372c2..f27a06ba19c6244e7973823252d305bbcdeb583e 100644 (file)
@@ -14,6 +14,7 @@
        GNU General Public License for more details.
 */
 
+#include <arpa/inet.h>
 #include <errno.h>
 #include <stdio.h>
 #include <stddef.h>
 #include <string.h>
 #include <syslog.h>
 
-#include <loc/libloc.h>
-#include <loc/database.h>
-#include <loc/network.h>
-#include <loc/writer.h>
+#include <libloc/libloc.h>
+#include <libloc/address.h>
+#include <libloc/database.h>
+#include <libloc/network.h>
+#include <libloc/private.h>
+#include <libloc/writer.h>
 
 int main(int argc, char** argv) {
        int err;
@@ -37,12 +40,21 @@ int main(int argc, char** argv) {
        // Enable debug logging
        loc_set_log_priority(ctx, LOG_DEBUG);
 
+#if 0
        struct loc_network_tree* tree;
        err = loc_network_tree_new(ctx, &tree);
        if (err) {
                fprintf(stderr, "Could not create the network tree\n");
                exit(EXIT_FAILURE);
        }
+#endif
+
+       struct in6_addr address;
+       err = inet_pton(AF_INET6, "2001:db8::1", &address);
+       if (err != 1) {
+               fprintf(stderr, "Could not parse IP address\n");
+               exit(EXIT_FAILURE);
+       }
 
        // Create a network
        struct loc_network* network1;
@@ -52,21 +64,23 @@ int main(int argc, char** argv) {
                exit(EXIT_FAILURE);
        }
 
-       err = loc_network_set_country_code(network1, "XX");
+       err = loc_network_set_country_code(network1, "DE");
        if (err) {
                fprintf(stderr, "Could not set country code\n");
                exit(EXIT_FAILURE);
        }
 
+#if 0
        // Adding network to the tree
        err = loc_network_tree_add_network(tree, network1);
        if (err) {
                fprintf(stderr, "Could not add network to the tree\n");
                exit(EXIT_FAILURE);
        }
+#endif
 
        // Check if the first and last addresses are correct
-       char* string = loc_network_format_first_address(network1);
+       const char* string = loc_network_format_first_address(network1);
        if (!string) {
                fprintf(stderr, "Did get NULL instead of a string for the first address\n");
                exit(EXIT_FAILURE);
@@ -88,6 +102,12 @@ int main(int argc, char** argv) {
                exit(EXIT_FAILURE);
        }
 
+       err = loc_network_matches_address(network1, &address);
+       if (!err) {
+               fprintf(stderr, "Network1 does not match address\n");
+               exit(EXIT_FAILURE);
+       }
+
        struct loc_network* network2;
        err = loc_network_new_from_string(ctx, &network2, "2001:db8:ffff::/48");
        if (err) {
@@ -95,12 +115,13 @@ int main(int argc, char** argv) {
                exit(EXIT_FAILURE);
        }
 
-       err = loc_network_set_country_code(network2, "XY");
+       err = loc_network_set_country_code(network2, "DE");
        if (err) {
                fprintf(stderr, "Could not set country code\n");
                exit(EXIT_FAILURE);
        }
 
+#if 0
        // Adding network to the tree
        err = loc_network_tree_add_network(tree, network2);
        if (err) {
@@ -117,20 +138,82 @@ int main(int argc, char** argv) {
 
        size_t nodes = loc_network_tree_count_nodes(tree);
        printf("The tree has %zu nodes\n", nodes);
+#endif
+
+       // Check equals function
+       err = loc_network_cmp(network1, network1);
+       if (err) {
+               fprintf(stderr, "Network is not equal with itself\n");
+               exit(EXIT_FAILURE);
+       }
+
+       err = loc_network_cmp(network1, network2);
+       if (!err) {
+               fprintf(stderr, "Networks equal unexpectedly\n");
+               exit(EXIT_FAILURE);
+       }
 
        // Check subnet function
-       err = loc_network_is_subnet_of(network1, network2);
-       if (err != 0) {
+       err = loc_network_is_subnet(network1, network2);
+       if (!err) {
                fprintf(stderr, "Subnet check 1 failed: %d\n", err);
                exit(EXIT_FAILURE);
        }
 
-       err = loc_network_is_subnet_of(network2, network1);
-       if (err != 1) {
+       err = loc_network_is_subnet(network2, network1);
+       if (err) {
                fprintf(stderr, "Subnet check 2 failed: %d\n", err);
                exit(EXIT_FAILURE);
        }
 
+       // Make subnets
+       struct loc_network* subnet1 = NULL;
+       struct loc_network* subnet2 = NULL;
+
+       err  = loc_network_subnets(network1, &subnet1, &subnet2);
+       if (err || !subnet1 || !subnet2) {
+               fprintf(stderr, "Could not find subnets of network: %d\n", err);
+               exit(EXIT_FAILURE);
+       }
+
+       const char* s = loc_network_str(subnet1);
+       printf("Received subnet1 = %s\n", s);
+
+       s = loc_network_str(subnet2);
+       printf("Received subnet2 = %s\n", s);
+
+       if (!loc_network_is_subnet(network1, subnet1)) {
+               fprintf(stderr, "Subnet1 is not a subnet\n");
+               exit(EXIT_FAILURE);
+       }
+
+       if (!loc_network_is_subnet(network1, subnet2)) {
+               fprintf(stderr, "Subnet2 is not a subnet\n");
+               exit(EXIT_FAILURE);
+       }
+
+       if (!loc_network_overlaps(network1, subnet1)) {
+               fprintf(stderr, "Network1 does not seem to contain subnet1\n");
+               exit(EXIT_FAILURE);
+       }
+
+       if (!loc_network_overlaps(network1, subnet2)) {
+               fprintf(stderr, "Network1 does not seem to contain subnet2\n");
+               exit(EXIT_FAILURE);
+       }
+
+       loc_network_unref(subnet1);
+       loc_network_unref(subnet2);
+
+       struct loc_network_list* excluded = loc_network_exclude(network1, network2);
+       if (!excluded) {
+               fprintf(stderr, "Could not create excluded list\n");
+               exit(EXIT_FAILURE);
+       }
+
+       loc_network_list_dump(excluded);
+       loc_network_list_unref(excluded);
+
        // Create a database
        struct loc_writer* writer;
        err = loc_writer_new(ctx, &writer, NULL, NULL);
@@ -170,8 +253,8 @@ int main(int argc, char** argv) {
 
        // Try adding a single address
        err = loc_writer_add_network(writer, &network, "2001:db8::");
-       if (err != -EINVAL) {
-               fprintf(stderr, "It was possible to add an invalid network (err = %d)\n", err);
+       if (err) {
+               fprintf(stderr, "It was impossible to add an single IP address (err = %d)\n", err);
                exit(EXIT_FAILURE);
        }
 
@@ -192,7 +275,10 @@ int main(int argc, char** argv) {
        loc_network_unref(network2);
        loc_network_unref(network3);
        loc_network_unref(network4);
+
+#if 0
        loc_network_tree_unref(tree);
+#endif
 
        // And open it again from disk
        struct loc_database* db;
@@ -218,6 +304,39 @@ int main(int argc, char** argv) {
        }
        loc_network_unref(network1);
 
+       const struct bit_length_test {
+               const char* network;
+               unsigned int bit_length;
+       } bit_length_tests[] = {
+               { "::/0", 0 },
+               { "2001::/128", 126 },
+               { "1.0.0.0/32", 25 },
+               { "0.0.0.1/32", 1 },
+               { "255.255.255.255/32", 32 },
+               { NULL, 0, },
+       };
+
+       for (const struct bit_length_test* t = bit_length_tests; t->network; t++) {
+               err = loc_network_new_from_string(ctx, &network1, t->network);
+               if (err) {
+                       fprintf(stderr, "Could not create network %s: %m\n", t->network);
+                       exit(EXIT_FAILURE);
+               }
+
+               const struct in6_addr* addr = loc_network_get_first_address(network1);
+
+               unsigned int bit_length = loc_address_bit_length(addr);
+
+               if (bit_length != t->bit_length) {
+                       printf("Bit length of %s didn't match: %u != %u\n",
+                               t->network, t->bit_length, bit_length);
+                       loc_network_unref(network1);
+                       exit(EXIT_FAILURE);
+               }
+
+               loc_network_unref(network1);
+       }
+
        loc_unref(ctx);
        fclose(f);