b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen#include "alloc-util.h"

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen#include "dns-domain.h"

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen#include "macro.h"

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen#include "string-util.h"

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_label_unescape_one(const char *what, const char *expect, size_t buffer_sz, int ret) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen char buffer[buffer_sz];

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen int r;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_label_unescape(&what, buffer, buffer_sz);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen if (r < 0)

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen return;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(streq(buffer, expect));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_label_unescape(void) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one("hallo", "hallo", 6, 5);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one("hallo", "hallo", 4, -ENOSPC);

79008bddf679a5e0900369950eb346c9fa687107Lennart Poettering test_dns_label_unescape_one("", "", 10, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one("hallo\\.foobar", "hallo.foobar", 20, 12);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one("hallo.foobar", "hallo", 10, 5);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one("hallo\n.foobar", "hallo", 20, -EINVAL);

79008bddf679a5e0900369950eb346c9fa687107Lennart Poettering test_dns_label_unescape_one("hallo\\", "hallo", 20, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one("hallo\\032 ", "hallo ", 20, 7);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one(".", "", 20, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one("..", "", 20, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one(".foobar", "", 20, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_one("foobar.", "foobar", 20, 6);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_to_wire_format_one(const char *what, const char *expect, size_t buffer_sz, int ret) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen uint8_t buffer[buffer_sz];

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen int r;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_name_to_wire_format(what, buffer, buffer_sz);

79008bddf679a5e0900369950eb346c9fa687107Lennart Poettering assert_se(r == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen if (r < 0)

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen return;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(!memcmp(buffer, expect, r));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_to_wire_format(void) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen const char out1[] = { 3, 'f', 'o', 'o', 0 };

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen const char out2[] = { 5, 'h', 'a', 'l', 'l', 'o', 3, 'f', 'o', 'o', 3, 'b', 'a', 'r', 0 };

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen const char out3[] = { 4, ' ', 'f', 'o', 'o', 3, 'b', 'a', 'r', 0 };

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_to_wire_format_one("", NULL, 0, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_to_wire_format_one("foo", out1, sizeof(out1), sizeof(out1));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_to_wire_format_one("foo", out1, sizeof(out1) + 1, sizeof(out1));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_to_wire_format_one("foo", out1, sizeof(out1) - 1, -ENOBUFS);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_to_wire_format_one("hallo.foo.bar", out2, sizeof(out2), sizeof(out2));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_to_wire_format_one("hallo.foo..bar", NULL, 32, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_to_wire_format_one("\\032foo.bar", out3, sizeof(out3), sizeof(out3));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

79008bddf679a5e0900369950eb346c9fa687107Lennart Poetteringstatic void test_dns_label_unescape_suffix_one(const char *what, const char *expect1, const char *expect2, size_t buffer_sz, int ret1, int ret2) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen char buffer[buffer_sz];

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen const char *label;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen int r;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen label = what + strlen(what);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_label_unescape_suffix(what, &label, buffer, buffer_sz);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret1);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen if (r >= 0)

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(streq(buffer, expect1));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_label_unescape_suffix(what, &label, buffer, buffer_sz);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret2);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen if (r >= 0)

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(streq(buffer, expect2));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_label_unescape_suffix(void) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("hallo", "hallo", "", 6, 5, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("hallo", "hallo", "", 4, -ENOSPC, -ENOSPC);

79008bddf679a5e0900369950eb346c9fa687107Lennart Poettering test_dns_label_unescape_suffix_one("", "", "", 10, 0, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("hallo\\.foobar", "hallo.foobar", "", 20, 12, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("hallo.foobar", "foobar", "hallo", 10, 6, 5);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("hallo.foobar\n", "foobar", "foobar", 20, -EINVAL, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("hallo\\", "hallo", "hallo", 20, -EINVAL, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("hallo\\032 ", "hallo ", "", 20, 7, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one(".", "", "", 20, 0, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("..", "", "", 20, 0, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one(".foobar", "foobar", "", 20, 6, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("foobar.", "", "foobar", 20, 0, 6);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("foo\\\\bar", "foo\\bar", "", 20, 7, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("foo.bar", "bar", "foo", 20, 3, 3);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("foo..bar", "bar", "", 20, 3, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("foo...bar", "bar", "", 20, 3, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("foo\\.bar", "foo.bar", "", 20, 7, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("foo\\\\.bar", "bar", "foo\\", 20, 3, 4);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_unescape_suffix_one("foo\\\\\\.bar", "foo\\.bar", "", 20, 8, 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_label_escape_one(const char *what, size_t l, const char *expect, int ret) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen _cleanup_free_ char *t = NULL;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen int r;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_label_escape(what, l, &t);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen if (r < 0)

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen return;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(streq_ptr(expect, t));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

79008bddf679a5e0900369950eb346c9fa687107Lennart Poettering

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_label_escape(void) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_escape_one("", 0, "", 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_escape_one("hallo", 5, "hallo", 5);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_escape_one("hallo", 6, NULL, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_label_escape_one("hallo hallo.foobar,waldi", 24, "hallo\\032hallo\\.foobar\\044waldi", 31);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_normalize_one(const char *what, const char *expect, int ret) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen _cleanup_free_ char *t = NULL;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen int r;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_name_normalize(what, &t);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen if (r < 0)

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen return;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(streq_ptr(expect, t));

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_normalize(void) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one("", "", 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one("f", "f", 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one("f.waldi", "f.waldi", 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one("f \\032.waldi", "f\\032\\032.waldi", 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one("\\000", NULL, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one("..", NULL, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one(".foobar", NULL, -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one("foobar.", "foobar", 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_normalize_one(".", "", 0);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_equal_one(const char *a, const char *b, int ret) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen int r;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_name_equal(a, b);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_name_equal(b, a);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_equal(void) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("", "", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("x", "x", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("x", "x.", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("abc.def", "abc.def", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("abc.def", "ABC.def", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("abc.def", "CBA.def", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("", "xxx", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("ab", "a", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("\\000", "xxxx", -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one(".", "", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one(".", ".", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_equal_one("..", "..", -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_between_one(const char *a, const char *b, const char *c, int ret) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen int r;

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_name_between(a, b, c);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen r = dns_name_between(c, b, a);

79008bddf679a5e0900369950eb346c9fa687107Lennart Poettering if (ret >= 0)

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == 0);

79008bddf679a5e0900369950eb346c9fa687107Lennart Poettering else

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(r == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_between(void) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen /* see https://tools.ietf.org/html/rfc4034#section-6.1

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("example", "a.example", "yljkjljk.a.example", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("a.example", "yljkjljk.a.example", "Z.a.example", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("yljkjljk.a.example", "Z.a.example", "zABC.a.EXAMPLE", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("Z.a.example", "zABC.a.EXAMPLE", "z.example", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("zABC.a.EXAMPLE", "z.example", "\\033.z.example", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("z.example", "\\033.z.example", "*.z.example", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("\\033.z.example", "*.z.example", "\\200.z.example", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("*.z.example", "\\200.z.example", "example", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("\\200.z.example", "example", "a.example", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("example", "a.example", "example", -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("example", "example", "yljkjljk.a.example", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_between_one("example", "yljkjljk.a.example", "yljkjljk.a.example", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_endswith_one(const char *a, const char *b, int ret) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen assert_se(dns_name_endswith(a, b) == ret);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersenstatic void test_dns_name_endswith(void) {

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("", "", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("", "xxx", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("xxx", "", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x", "x", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x", "y", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y", "y", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y", "Y", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y", "x", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y.z", "Z", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y.z", "y.Z", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y.z", "x.y.Z", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y.z", "waldo", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y.z.u.v.w", "y.z", false);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y.z.u.v.w", "u.v.w", true);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen test_dns_name_endswith_one("x.y\001.z", "waldo", -EINVAL);

b22d8a00f48f3c5fc4510b4acd3e1a43e731e592Tom Gundersen}

static void test_dns_name_root(void) {

assert_se(dns_name_root("") == true);

assert_se(dns_name_root(".") == true);

assert_se(dns_name_root("xxx") == false);

assert_se(dns_name_root("xxx.") == false);

assert_se(dns_name_root("..") == -EINVAL);

}

static void test_dns_name_single_label(void) {

assert_se(dns_name_single_label("") == false);

assert_se(dns_name_single_label(".") == false);

assert_se(dns_name_single_label("..") == -EINVAL);

assert_se(dns_name_single_label("x") == true);

assert_se(dns_name_single_label("x.") == true);

assert_se(dns_name_single_label("xx.yy") == false);

}

static void test_dns_name_reverse_one(const char *address, const char *name) {

_cleanup_free_ char *p = NULL;

union in_addr_union a, b = {};

int familya, familyb;

assert_se(in_addr_from_string_auto(address, &familya, &a) >= 0);

assert_se(dns_name_reverse(familya, &a, &p) >= 0);

assert_se(streq(p, name));

assert_se(dns_name_address(p, &familyb, &b) > 0);

assert_se(familya == familyb);

assert_se(in_addr_equal(familya, &a, &b));

}

static void test_dns_name_reverse(void) {

test_dns_name_reverse_one("47.11.8.15", "15.8.11.47.in-addr.arpa");

test_dns_name_reverse_one("fe80::47", "7.4.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.8.e.f.ip6.arpa");

test_dns_name_reverse_one("127.0.0.1", "1.0.0.127.in-addr.arpa");

test_dns_name_reverse_one("::1", "1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa");

}

static void test_dns_name_concat_one(const char *a, const char *b, int r, const char *result) {

_cleanup_free_ char *p = NULL;

assert_se(dns_name_concat(a, b, &p) == r);

assert_se(streq_ptr(p, result));

}

static void test_dns_name_concat(void) {

test_dns_name_concat_one("foo", "bar", 0, "foo.bar");

test_dns_name_concat_one("foo.foo", "bar.bar", 0, "foo.foo.bar.bar");

test_dns_name_concat_one("foo", NULL, 0, "foo");

test_dns_name_concat_one("foo.", "bar.", 0, "foo.bar");

}

static void test_dns_name_is_valid_one(const char *s, int ret) {

assert_se(dns_name_is_valid(s) == ret);

}

static void test_dns_name_is_valid(void) {

test_dns_name_is_valid_one("foo", 1);

test_dns_name_is_valid_one("foo.", 1);

test_dns_name_is_valid_one("Foo", 1);

test_dns_name_is_valid_one("foo.bar", 1);

test_dns_name_is_valid_one("foo.bar.baz", 1);

test_dns_name_is_valid_one("", 1);

test_dns_name_is_valid_one("foo..bar", 0);

test_dns_name_is_valid_one(".foo.bar", 0);

test_dns_name_is_valid_one("foo.bar.", 1);

test_dns_name_is_valid_one("\\zbar", 0);

test_dns_name_is_valid_one("ä", 1);

test_dns_name_is_valid_one("\n", 0);

}

static void test_dns_service_name_is_valid(void) {

assert_se(dns_service_name_is_valid("Lennart's Compüter"));

assert_se(dns_service_name_is_valid("piff.paff"));

assert_se(!dns_service_name_is_valid(NULL));

assert_se(!dns_service_name_is_valid(""));

assert_se(!dns_service_name_is_valid("foo\nbar"));

assert_se(!dns_service_name_is_valid("foo\201bar"));

assert_se(!dns_service_name_is_valid("this is an overly long string that is certainly longer than 63 characters"));

}

static void test_dns_srv_type_verify(void) {

assert_se(dns_srv_type_verify("_http._tcp") > 0);

assert_se(dns_srv_type_verify("_foo-bar._tcp") > 0);

assert_se(dns_srv_type_verify("_w._udp") > 0);

assert_se(dns_srv_type_verify("_piep._sub._w._udp") > 0);

assert_se(dns_srv_type_verify("_a800._tcp") > 0);

assert_se(dns_srv_type_verify("_a-800._tcp") > 0);

assert_se(dns_srv_type_verify(NULL) == 0);

assert_se(dns_srv_type_verify("") == 0);

assert_se(dns_srv_type_verify("x") == 0);

assert_se(dns_srv_type_verify("_foo") == 0);

assert_se(dns_srv_type_verify("_tcp") == 0);

assert_se(dns_srv_type_verify("_") == 0);

assert_se(dns_srv_type_verify("_foo.") == 0);

assert_se(dns_srv_type_verify("_föo._tcp") == 0);

assert_se(dns_srv_type_verify("_f\no._tcp") == 0);

assert_se(dns_srv_type_verify("_800._tcp") == 0);

assert_se(dns_srv_type_verify("_-800._tcp") == 0);

assert_se(dns_srv_type_verify("_-foo._tcp") == 0);

}

int main(int argc, char *argv[]) {

test_dns_label_unescape();

test_dns_label_unescape_suffix();

test_dns_label_escape();

test_dns_name_normalize();

test_dns_name_equal();

test_dns_name_endswith();

test_dns_name_between();

test_dns_name_root();

test_dns_name_single_label();

test_dns_name_reverse();

test_dns_name_concat();

test_dns_name_is_valid();

test_dns_name_to_wire_format();

test_dns_service_name_is_valid();

test_dns_srv_type_verify();

return 0;

}