SymbX/systemd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #33531 from neighbourhoodie/tests/dns-zone

Browse Source
This commit is contained in:
Luca Boccassi
2024-07-23 13:29:03 +01:00
parent c01267c4a9 4d3936ad2d
commit ae2bf016bc
2 changed files with 402 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/resolve/meson.build
+11
View File
@@ -210,6 +210,17 @@ executables += [
],
'include_directories' : resolve_includes,
},
test_template + {
'sources' : [
files('test-dns-zone.c'),
basic_dns_sources,
systemd_resolved_sources,
],
'dependencies' : [
systemd_resolved_dependencies,
],
'include_directories' : resolve_includes,
},
test_template + {
'sources' : [
files('test-resolved-link.c'),
src/resolve/test-dns-zone.c
+391
View File
@@ -0,0 +1,391 @@
/* SPDX-License-Identifier: LGPL-2.1-or-later */
#include "dns-type.h"
#include "resolved-dns-rr.h"
#include "resolved-dns-scope.h"
#include "resolved-dns-zone.h"
#include "resolved-link.h"
#include "resolved-manager.h"
#include "log.h"
#include "tests.h"
static void dns_scope_freep(DnsScope **s) {
if (s != NULL && *s != NULL)
dns_scope_free(*s);
}
/* ================================================================
* dns_zone_put()
* ================================================================ */
TEST(dns_zone_put_simple) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
DnsZone *zone = NULL;
DnsZoneItem *item = NULL;
_cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr = NULL;
ASSERT_OK(dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET));
ASSERT_NOT_NULL(scope);
zone = &scope->zone;
rr = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr);
ASSERT_TRUE(dns_zone_is_empty(zone));
ASSERT_OK(dns_zone_put(zone, scope, rr, 0));
ASSERT_FALSE(dns_zone_is_empty(zone));
item = dns_zone_get(zone, rr);
ASSERT_NOT_NULL(item);
ASSERT_EQ((int)item->state, DNS_ZONE_ITEM_ESTABLISHED);
}
TEST(dns_zone_put_any_class_is_invalid) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
DnsZone *zone = NULL;
_cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr = NULL;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
zone = &scope->zone;
rr = dns_resource_record_new_full(DNS_CLASS_ANY, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr);
ASSERT_ERROR(dns_zone_put(zone, scope, rr, 0), EINVAL);
ASSERT_TRUE(dns_zone_is_empty(zone));
}
TEST(dns_zone_put_any_type_is_invalid) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
DnsZone *zone = NULL;
_cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr = NULL;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
zone = &scope->zone;
rr = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_ANY, "www.example.com");
ASSERT_NOT_NULL(rr);
ASSERT_ERROR(dns_zone_put(zone, scope, rr, 0), EINVAL);
ASSERT_TRUE(dns_zone_is_empty(zone));
}
/* ================================================================
* dns_zone_remove_rr()
* ================================================================ */
TEST(dns_zone_remove_rr_match) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
DnsZone *zone = NULL;
_cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr_in = NULL, *rr_out = NULL;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
zone = &scope->zone;
rr_in = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr_in);
rr_in->a.in_addr.s_addr = htobe32(0xc0a8017f);
ASSERT_OK(dns_zone_put(zone, scope, rr_in, 0));
rr_out = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr_out);
rr_out->a.in_addr.s_addr = htobe32(0xc0a8017f);
ASSERT_NOT_NULL(dns_zone_get(zone, rr_in));
dns_zone_remove_rr(zone, rr_out);
ASSERT_NULL(dns_zone_get(zone, rr_in));
}
TEST(dns_zone_remove_rr_match_one) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
DnsZone *zone = NULL;
_cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr_in = NULL, *rr_out = NULL;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
zone = &scope->zone;
rr_in = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr_in);
rr_in->a.in_addr.s_addr = htobe32(0xc0a8017f);
ASSERT_OK(dns_zone_put(zone, scope, rr_in, 0));
dns_resource_record_unref(rr_in);
rr_in = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_CNAME, "example.com");
ASSERT_NOT_NULL(rr_in);
rr_in->cname.name = strdup("www.example.com");
ASSERT_OK(dns_zone_put(zone, scope, rr_in, 0));
rr_out = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr_out);
rr_out->a.in_addr.s_addr = htobe32(0xc0a8017f);
ASSERT_NOT_NULL(dns_zone_get(zone, rr_out));
dns_zone_remove_rr(zone, rr_out);
ASSERT_NULL(dns_zone_get(zone, rr_out));
ASSERT_NOT_NULL(dns_zone_get(zone, rr_in));
}
TEST(dns_zone_remove_rr_different_payload) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
DnsZone *zone = NULL;
_cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr_in = NULL, *rr_out = NULL;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
zone = &scope->zone;
rr_in = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr_in);
rr_in->a.in_addr.s_addr = htobe32(0xc0a8017f);
ASSERT_OK(dns_zone_put(zone, scope, rr_in, 0));
rr_out = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr_out);
rr_out->a.in_addr.s_addr = htobe32(0xc0a80180);
ASSERT_NOT_NULL(dns_zone_get(zone, rr_in));
dns_zone_remove_rr(zone, rr_out);
ASSERT_NOT_NULL(dns_zone_get(zone, rr_in));
}
/* ================================================================
* dns_zone_remove_rrs_by_key()
* ================================================================ */
TEST(dns_zone_remove_rrs_by_key) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
DnsZone *zone = NULL;
_cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr1 = NULL, *rr2 = NULL, *rr3 = NULL;
DnsResourceKey *key = NULL;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
zone = &scope->zone;
rr1 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr1);
dns_zone_put(zone, scope, rr1, 0);
rr2 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_AAAA, "www.example.com");
ASSERT_NOT_NULL(rr2);
dns_zone_put(zone, scope, rr2, 0);
rr3 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_CNAME, "example.com");
ASSERT_NOT_NULL(rr3);
rr3->cname.name = strdup("www.example.com");
dns_zone_put(zone, scope, rr3, 0);
key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_CNAME, "www.example.com");
ASSERT_NOT_NULL(key);
ASSERT_OK(dns_zone_remove_rrs_by_key(zone, key));
ASSERT_NOT_NULL(dns_zone_get(zone, rr3));
dns_resource_key_unref(key);
key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_CNAME, "example.com");
ASSERT_NOT_NULL(key);
ASSERT_OK(dns_zone_remove_rrs_by_key(zone, key));
ASSERT_NULL(dns_zone_get(zone, rr3));
dns_resource_key_unref(key);
key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_ANY, "www.example.com");
ASSERT_NOT_NULL(key);
ASSERT_OK(dns_zone_remove_rrs_by_key(zone, key));
ASSERT_NULL(dns_zone_get(zone, rr1));
ASSERT_NULL(dns_zone_get(zone, rr2));
dns_resource_key_unref(key);
}
/* ================================================================
* dns_zone_lookup()
* ================================================================ */
static void add_zone_rrs(DnsScope *scope) {
_cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr1 = NULL, *rr2 = NULL, *rr3 = NULL, *rr4 = NULL;
rr1 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(rr1);
dns_zone_put(&scope->zone, scope, rr1, 0);
rr2 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_AAAA, "www.example.com");
ASSERT_NOT_NULL(rr2);
dns_zone_put(&scope->zone, scope, rr2, 0);
rr3 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_CNAME, "example.com");
ASSERT_NOT_NULL(rr3);
rr3->cname.name = strdup("www.example.com");
dns_zone_put(&scope->zone, scope, rr3, 0);
rr4 = dns_resource_record_new_full(DNS_CLASS_IN, DNS_TYPE_NS, "app.example.com");
ASSERT_NOT_NULL(rr4);
rr4->cname.name = strdup("ns1.app.example.com");
dns_zone_put(&scope->zone, scope, rr4, 0);
}
TEST(dns_zone_lookup_match_a) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
_cleanup_(dns_resource_key_unrefp) DnsResourceKey *qkey = NULL;
_cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL, *soa = NULL;
bool tentative;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
add_zone_rrs(scope);
qkey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(qkey);
ASSERT_TRUE(dns_zone_lookup(&scope->zone, qkey, 1, &answer, &soa, &tentative));
ASSERT_FALSE(tentative);
ASSERT_EQ(dns_answer_size(answer), 1u);
ASSERT_EQ(dns_answer_size(soa), 0u);
ASSERT_TRUE(dns_answer_match_key(answer, qkey, NULL));
}
TEST(dns_zone_lookup_match_cname) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
_cleanup_(dns_resource_key_unrefp) DnsResourceKey *qkey = NULL;
_cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL, *soa = NULL;
bool tentative;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
add_zone_rrs(scope);
qkey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_CNAME, "example.com");
ASSERT_NOT_NULL(qkey);
ASSERT_TRUE(dns_zone_lookup(&scope->zone, qkey, 1, &answer, &soa, &tentative));
ASSERT_FALSE(tentative);
ASSERT_EQ(dns_answer_size(answer), 1u);
ASSERT_EQ(dns_answer_size(soa), 0u);
ASSERT_TRUE(dns_answer_match_key(answer, qkey, NULL));
}
TEST(dns_zone_lookup_match_any) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
_cleanup_(dns_resource_key_unrefp) DnsResourceKey *qkey = NULL;
DnsResourceKey *akey = NULL;
_cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL, *soa = NULL;
bool tentative;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
add_zone_rrs(scope);
qkey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_ANY, "www.example.com");
ASSERT_NOT_NULL(qkey);
ASSERT_TRUE(dns_zone_lookup(&scope->zone, qkey, 1, &answer, &soa, &tentative));
ASSERT_FALSE(tentative);
ASSERT_EQ(dns_answer_size(answer), 2u);
ASSERT_EQ(dns_answer_size(soa), 0u);
akey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(akey);
ASSERT_TRUE(dns_answer_match_key(answer, akey, NULL));
dns_resource_key_unref(akey);
akey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_AAAA, "www.example.com");
ASSERT_NOT_NULL(akey);
ASSERT_TRUE(dns_answer_match_key(answer, akey, NULL));
dns_resource_key_unref(akey);
}
TEST(dns_zone_lookup_match_any_apex) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
_cleanup_(dns_resource_key_unrefp) DnsResourceKey *qkey = NULL;
DnsResourceKey *akey = NULL;
_cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL, *soa = NULL;
bool tentative;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
add_zone_rrs(scope);
qkey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_ANY, "example.com");
ASSERT_NOT_NULL(qkey);
ASSERT_TRUE(dns_zone_lookup(&scope->zone, qkey, 1, &answer, &soa, &tentative));
ASSERT_FALSE(tentative);
ASSERT_EQ(dns_answer_size(answer), 1u);
ASSERT_EQ(dns_answer_size(soa), 0u);
akey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_CNAME, "example.com");
ASSERT_NOT_NULL(akey);
ASSERT_TRUE(dns_answer_match_key(answer, akey, NULL));
dns_resource_key_unref(akey);
}
TEST(dns_zone_lookup_match_nothing) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
_cleanup_(dns_resource_key_unrefp) DnsResourceKey *qkey = NULL;
_cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL, *soa = NULL;
bool tentative;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
add_zone_rrs(scope);
qkey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "nope.example.com");
ASSERT_NOT_NULL(qkey);
ASSERT_FALSE(dns_zone_lookup(&scope->zone, qkey, 1, &answer, &soa, &tentative));
ASSERT_FALSE(tentative);
ASSERT_EQ(dns_answer_size(answer), 0u);
ASSERT_EQ(dns_answer_size(soa), 0u);
}
TEST(dns_zone_lookup_match_nothing_with_soa) {
Manager manager = {};
_cleanup_(dns_scope_freep) DnsScope *scope = NULL;
_cleanup_(dns_resource_key_unrefp) DnsResourceKey *qkey = NULL;
DnsResourceKey *akey = NULL;
_cleanup_(dns_answer_unrefp) DnsAnswer *answer = NULL, *soa = NULL;
bool tentative;
dns_scope_new(&manager, &scope, NULL, DNS_PROTOCOL_DNS, AF_INET);
ASSERT_NOT_NULL(scope);
add_zone_rrs(scope);
qkey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "example.com");
ASSERT_NOT_NULL(qkey);
ASSERT_TRUE(dns_zone_lookup(&scope->zone, qkey, 1, &answer, &soa, &tentative));
ASSERT_FALSE(tentative);
ASSERT_EQ(dns_answer_size(answer), 0u);
ASSERT_EQ(dns_answer_size(soa), 1u);
akey = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_SOA, "example.com");
ASSERT_NOT_NULL(akey);
ASSERT_TRUE(dns_answer_match_key(soa, akey, NULL));
dns_resource_key_unref(akey);
}
DEFINE_TEST_MAIN(LOG_DEBUG);