lua: Create a simple iterator for all networks

[people/ms/libloc.git] / src / test-network.c

diff --git a/src/test-network.c b/src/test-network.c
index 8a6763c953d1e8247dd0d0451129246f3617ab42..f6ccb626a9e9c75192f2867365c42110f51c232f 100644 (file)
--- a/src/test-network.c
+++ b/src/test-network.c
@@ -14,6 +14,7 @@
        GNU General Public License for more details.
 */
 
+#include <arpa/inet.h>
 #include <errno.h>
 #include <stdio.h>
 #include <stddef.h>
@@ -21,10 +22,12 @@
 #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;
@@ -46,6 +49,13 @@ int main(int argc, char** argv) {
        }
 #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;
        err = loc_network_new_from_string(ctx, &network1, "2001:db8::1/32");
@@ -54,7 +64,7 @@ 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);
@@ -70,7 +80,7 @@ int main(int argc, char** argv) {
 #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);
@@ -92,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) {
@@ -99,7 +115,7 @@ 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);
@@ -125,14 +141,14 @@ int main(int argc, char** argv) {
 #endif
 
        // Check equals function
-       err = loc_network_eq(network1, network1);
-       if (!err) {
+       err = loc_network_cmp(network1, network1);
+       if (err) {
                fprintf(stderr, "Network is not equal with itself\n");
                exit(EXIT_FAILURE);
        }
 
-       err = loc_network_eq(network1, network2);
-       if (err) {
+       err = loc_network_cmp(network1, network2);
+       if (!err) {
                fprintf(stderr, "Networks equal unexpectedly\n");
                exit(EXIT_FAILURE);
        }
@@ -160,13 +176,11 @@ int main(int argc, char** argv) {
                exit(EXIT_FAILURE);
        }
 
-       char* s = loc_network_str(subnet1);
+       const char* s = loc_network_str(subnet1);
        printf("Received subnet1 = %s\n", s);
-       free(s);
 
        s = loc_network_str(subnet2);
        printf("Received subnet2 = %s\n", s);
-       free(s);
 
        if (!loc_network_is_subnet(network1, subnet1)) {
                fprintf(stderr, "Subnet1 is not a subnet\n");
@@ -178,6 +192,16 @@ int main(int argc, char** argv) {
                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);
 
@@ -222,7 +246,7 @@ int main(int argc, char** argv) {
        // Try adding an invalid network
        struct loc_network* network;
        err = loc_writer_add_network(writer, &network, "xxxx:xxxx::/32");
-       if (err != -EINVAL) {
+       if (!err) {
                fprintf(stderr, "It was possible to add an invalid network (err = %d)\n", err);
                exit(EXIT_FAILURE);
        }
@@ -234,22 +258,15 @@ int main(int argc, char** argv) {
                exit(EXIT_FAILURE);
        }
 
-       // Try adding localhost
-       err = loc_writer_add_network(writer, &network, "::1/128");
-       if (err != -EINVAL) {
-               fprintf(stderr, "It was possible to add localhost (::1/128): %d\n", err);
-               exit(EXIT_FAILURE);
-       }
-
        FILE* f = tmpfile();
        if (!f) {
-               fprintf(stderr, "Could not open file for writing: %s\n", strerror(errno));
+               fprintf(stderr, "Could not open file for writing: %m\n");
                exit(EXIT_FAILURE);
        }
 
        err = loc_writer_write(writer, f, LOC_DATABASE_VERSION_UNSET);
        if (err) {
-               fprintf(stderr, "Could not write database: %s\n", strerror(-err));
+               fprintf(stderr, "Could not write database: %m\n");
                exit(EXIT_FAILURE);
        }
        loc_writer_unref(writer);
@@ -267,7 +284,7 @@ int main(int argc, char** argv) {
        struct loc_database* db;
        err = loc_database_new(ctx, &db, f);
        if (err) {
-               fprintf(stderr, "Could not open database: %s\n", strerror(-err));
+               fprintf(stderr, "Could not open database: %m\n");
                exit(EXIT_FAILURE);
        }
 
@@ -285,7 +302,39 @@ int main(int argc, char** argv) {
                fprintf(stderr, "Could look up 2001:db8:fffe:1::, but I shouldn't\n");
                exit(EXIT_FAILURE);
        }
-       loc_network_unref(network1);
+
+       const struct bit_length_test {
+               const char* network;
+               unsigned int bit_length;
+       } bit_length_tests[] = {
+               { "::/0", 0 },
+               { "2001::/128", 16 },
+               { "1.0.0.0/32", 8 },
+               { "0.0.0.1/32", 32 },
+               { "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);