Merge branch 'u2_10_12_branch' of git://git.code.sf.net/p/undernet-ircu/ircu2
[ircu2.10.12-pk.git] / ircd / test / ircd_in_addr_t.c
diff --git a/ircd/test/ircd_in_addr_t.c b/ircd/test/ircd_in_addr_t.c
new file mode 100644 (file)
index 0000000..3814607
--- /dev/null
@@ -0,0 +1,175 @@
+/* ircd_in_addr_t.c - Test file for IP address manipulation */
+
+#include "ircd_log.h"
+#include "ircd_string.h"
+#include "numnicks.h"
+#include "res.h"
+#include <stdio.h>
+#include <string.h>
+#include <netinet/in.h>
+
+/** Structure to describe a test for IP address parsing and unparsing. */
+struct address_test {
+    const char *text; /**< Textual address to parse. */
+    const char *canonical; /**< Canonical form of address. */
+    struct irc_in_addr expected; /**< Parsed address. */
+    const char *base64_v4; /**< v4-only compatible base64 encoding. */
+    const char *base64_v6; /**< v6-compatible base64 encoding. */
+    unsigned int is_valid : 1; /**< is address valid? */
+    unsigned int is_ipv4 : 1; /**< is address ipv4? */
+    unsigned int is_loopback : 1; /**< is address loopback? */
+};
+
+/** Array of addresses to test with. */
+static struct address_test test_addrs[] = {
+    { "::", "0::",
+      {{ 0, 0, 0, 0, 0, 0, 0, 0 }},
+      "AAAAAA", "_", 0, 0, 0 },
+    { "::1", "0::1",
+      {{ 0, 0, 0, 0, 0, 0, 0, 1 }},
+      "AAAAAA", "_AAB", 1, 0, 1 },
+    { "127.0.0.1", "127.0.0.1",
+      {{ 0, 0, 0, 0, 0, 0xffff, 0x7f00, 1 }},
+      "B]AAAB", "B]AAAB", 1, 1, 1 },
+    { "::ffff:127.0.0.3", "127.0.0.3",
+      {{ 0, 0, 0, 0, 0, 0xffff, 0x7f00, 3 }},
+      "B]AAAD", "B]AAAD", 1, 1, 1 },
+    { "::127.0.0.1", "127.0.0.1",
+      {{ 0, 0, 0, 0, 0, 0, 0x7f00, 1 }},
+      "B]AAAB", "B]AAAB", 1, 1, 1 },
+    { "2002:7f00:3::1", "2002:7f00:3::1",
+      {{ 0x2002, 0x7f00, 3, 0, 0, 0, 0, 1 }},
+      "B]AAAD", "CACH8AAAD_AAB", 1, 0, 0 },
+    { "8352:0344:0:0:0:0:2001:1204", "8352:344::2001:1204",
+      {{ 0x8352, 0x344, 0, 0, 0, 0, 0x2001, 0x1204 }},
+      "AAAAAA", "INSANE_CABBIE", 1, 0, 0 },
+    { "1:2:3:4:5:6:7:8", "1:2:3:4:5:6:7:8",
+      {{ 1, 2, 3, 4, 5, 6, 7, 8 }},
+      "AAAAAA", "AABAACAADAAEAAFAAGAAHAAI", 1, 0, 0 },
+    { 0 },
+};
+
+/** Perform tests for a single IP address.
+ * @param[in] addr Address test structure.
+ */
+static void
+test_address(struct address_test *addr)
+{
+    struct irc_in_addr parsed;
+    unsigned int ii, len, val;
+    char unparsed[64], base64_v4[64], base64_v6[64];
+
+    /* Convert expected address to network order. */
+    for (ii = 0; ii < 8; ++ii)
+        addr->expected.in6_16[ii] = htons(addr->expected.in6_16[ii]);
+    /* Make sure the text form is parsed as expected. */
+    len = ircd_aton(&parsed, addr->text);
+    assert(len == strlen(addr->text));
+    assert(!irc_in_addr_cmp(&parsed, &addr->expected));
+    /* Make sure it converts back to ASCII. */
+    ircd_ntoa_r(unparsed, &parsed);
+    assert(!strcmp(unparsed, addr->canonical));
+    /* Check IP-to-base64 conversion. */
+    iptobase64(base64_v4, &parsed, sizeof(base64_v4), 0);
+    iptobase64(base64_v6, &parsed, sizeof(base64_v6), 1);
+    if (addr->base64_v4)
+        assert(!strcmp(base64_v4, addr->base64_v4));
+    if (addr->base64_v6)
+        assert(!strcmp(base64_v6, addr->base64_v6));
+    /* Check testable attributes. */
+    val = irc_in_addr_valid(&parsed);
+    assert(!!val == addr->is_valid);
+    val = irc_in_addr_is_ipv4(&parsed);
+    assert(!!val == addr->is_ipv4);
+    val = irc_in_addr_is_loopback(&parsed);
+    assert(!!val == addr->is_loopback);
+    /* Check base64-to-IP conversion. */
+    if (addr->is_ipv4) {
+        base64toip(addr->base64_v4, &parsed);
+        assert(!memcmp(parsed.in6_16+6, addr->expected.in6_16+6, 4));
+    } else {
+        base64toip(addr->base64_v6, &parsed);
+        assert(!memcmp(&parsed, &addr->expected, sizeof(parsed)));
+    }
+    /* Tests completed. */
+    printf("Passed: %s (%s/%s)\n", addr->text, base64_v4, base64_v6);
+}
+
+/** Structure to describe a test for IP mask parsing. */
+struct ipmask_test {
+    const char *text; /**< Textual IP mask to parse. */
+    struct irc_in_addr expected; /**< Canonical form of address. */
+    unsigned int is_ipmask : 1; /**< Parse as an ipmask? */
+    unsigned int is_parseable : 1; /**< Is address parseable? */
+    unsigned int exp_bits : 8; /**< Number of bits expected in netmask */
+};
+
+/** Array of ipmasks to test with. */
+static struct ipmask_test test_masks[] = {
+    { "::", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 1, 128 },
+    { "::/0", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 1, 0 },
+    { "::10.0.0.0", {{ 0, 0, 0, 0, 0, 0, 0xa00, 0 }}, 1, 1, 128 },
+    { "192.168/16", {{ 0, 0, 0, 0, 0, 0xffff, 0xc0a8, 0 }}, 1, 1, 112 },
+    { "192.*", {{ 0, 0, 0, 0, 0, 0xffff, 0xc000, 0 }}, 1, 1, 104 },
+    { "192.*.*.*", {{ 0, 0, 0, 0, 0, 0xffff, 0xc000, 0 }}, 1, 1, 104 },
+    { "192.*/8", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 0, 0 },
+    { "192*", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 0, 0 },
+    { "192.168.0.0/16", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 0, 0, 0 },
+    { "ab.*", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 0, 0 },
+    { "a*", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 0, 0 },
+    { "*", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 1, 0 },
+    { "a:b", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 0, 0 },
+    { "a::*", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 0, 0 },
+    { "a:*", {{ 0xa, 0, 0, 0, 0, 0, 0, 0 }}, 1, 1, 16 },
+    { "a:*:*", {{ 0xa, 0, 0, 0, 0, 0, 0, 0 }}, 1, 1, 16 },
+    { "a:/16", {{ 0xa, 0, 0, 0, 0, 0, 0, 0 }}, 1, 1, 16 },
+    { "0.0.0.0.0/1", {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 0, 0, 0 },
+    { "2607:3f00:1:30c:9999*",
+      {{ 0x2607, 0x3f00, 1, 0x30c, 0x9999, 0, 0, 0 }}, 1, 1, 80 },
+    { "1234:5678:9abc:def0:feed:abba:cabb:1e55::",
+      {{ 0, 0, 0, 0, 0, 0, 0, 0 }}, 1, 0, 0 },
+    { 0 }
+};
+
+/** Perform tests for a single IP mask.
+ * @param[in] mask IP mask test structure.
+ */
+static void
+test_ipmask(struct ipmask_test *mask)
+{
+    struct irc_in_addr parsed;
+    unsigned int len, ii;
+    unsigned char bits = 0;
+
+    /* Convert expected address to network order. */
+    for (ii = 0; ii < 8; ++ii)
+        mask->expected.in6_16[ii] = htons(mask->expected.in6_16[ii]);
+    /* Try to parse; make sure its parseability and netmask length are
+     * as expected. */
+    len = ipmask_parse(mask->text, &parsed, mask->is_ipmask ? &bits : 0);
+    assert(!!len == mask->is_parseable);
+    if (!len) {
+        printf("X-Fail: %s\n", mask->text);
+        return;
+    }
+    if (mask->is_ipmask)
+        assert(bits == mask->exp_bits);
+    assert(!memcmp(&parsed, &mask->expected, sizeof(parsed)));
+    printf("Passed: %s (%s/%u)\n", mask->text, ircd_ntoa(&parsed), bits);
+}
+
+int
+main(int argc, char *argv[])
+{
+    unsigned int ii;
+
+    printf("Testing address parsing..\n");
+    for (ii = 0; test_addrs[ii].text; ++ii)
+        test_address(&test_addrs[ii]);
+
+    printf("\nTesting ipmask parsing..\n");
+    for (ii = 0; test_masks[ii].text; ++ii)
+        test_ipmask(&test_masks[ii]);
+
+    return 0;
+}