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

resolved: tests for dns_cache_dump()

Browse Source
This commit is contained in:
James Coglan
2024-07-23 14:08:31 +01:00
parent e936ec5e4f
commit 6dd377ad70
1 changed files with 90 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/resolve/test-dns-cache.c
+90
View File
@@ -3,6 +3,9 @@
#include <sys/socket.h>
#include "dns-type.h"
#include "fd-util.h"
#include "fileio.h"
#include "fs-util.h"
#include "log.h"
#include "resolve-util.h"
#include "resolved-def.h"
@@ -12,6 +15,7 @@
#include "resolved-dns-packet.h"
#include "resolved-dns-rr.h"
#include "tests.h"
#include "tmpfile-util.h"
static DnsCache new_cache(void) {
return (DnsCache) {};
@@ -449,6 +453,92 @@ TEST(dns_cache_lookup_any_always_misses) {
ASSERT_EQ(dns_answer_size(ret_answer), 0u);
}
/* ================================================================
* dns_cache_dump()
* ================================================================ */
static int cmpstring(const void *a, const void *b) {
ASSERT_NOT_NULL(a);
ASSERT_NOT_NULL(b);
return strcmp(*(const char **)a, *(const char **)b);
}
static void check_dump_contents(FILE *f, const char **expected, size_t n) {
char *actual[n];
rewind(f);
for (size_t i = 0; i < n; i++) {
size_t length = read_line(f, 1024, &actual[i]);
ASSERT_GT(length, 0u);
}
qsort(actual, n, sizeof(char *), cmpstring);
for (size_t i = 0; i < n; i++)
ASSERT_STREQ(actual[i], expected[i]);
for (size_t i = 0; i < n; i++)
free(actual[i]);
}
TEST(dns_cache_dump_single_a) {
_cleanup_(dns_cache_unrefp) DnsCache cache = new_cache();
_cleanup_(put_args_unrefp) PutArgs put_args = mk_put_args();
put_args.key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(put_args.key);
put_args.rcode = DNS_RCODE_SUCCESS;
answer_add_a(&put_args, put_args.key, 0xc0a8017f, 3600, DNS_ANSWER_CACHEABLE);
cache_put(&cache, &put_args);
ASSERT_EQ(dns_cache_size(&cache), 1u);
_cleanup_(unlink_tempfilep) char p[] = "/tmp/dns-cache-dump-single-a-XXXXXX";
_cleanup_fclose_ FILE *f = NULL;
fmkostemp_safe(p, "r+", &f);
dns_cache_dump(&cache, f);
const char *expected[] = {
"\twww.example.com IN A 192.168.1.127"
};
check_dump_contents(f, expected, 1);
}
TEST(dns_cache_dump_a_with_cname) {
_cleanup_(dns_cache_unrefp) DnsCache cache = new_cache();
_cleanup_(put_args_unrefp) PutArgs put_args = mk_put_args();
_cleanup_(dns_resource_key_unrefp) DnsResourceKey *key = NULL;
put_args.key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "www.example.com");
ASSERT_NOT_NULL(put_args.key);
put_args.rcode = DNS_RCODE_SUCCESS;
key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_CNAME, "www.example.com");
ASSERT_NOT_NULL(key);
answer_add_cname(&put_args, key, "example.com", 3600, DNS_ANSWER_CACHEABLE);
dns_resource_key_unref(key);
key = dns_resource_key_new(DNS_CLASS_IN, DNS_TYPE_A, "example.com");
ASSERT_NOT_NULL(key);
answer_add_a(&put_args, key, 0xc0a8017f, 3600, DNS_ANSWER_CACHEABLE);
cache_put(&cache, &put_args);
ASSERT_EQ(dns_cache_size(&cache), 2u);
_cleanup_(unlink_tempfilep) char p[] = "/tmp/dns-cache-dump-a-with-cname-XXXXXX";
_cleanup_fclose_ FILE *f = NULL;
fmkostemp_safe(p, "r+", &f);
dns_cache_dump(&cache, f);
const char *expected[] = {
"\texample.com IN A 192.168.1.127",
"\twww.example.com IN CNAME example.com"
};
check_dump_contents(f, expected, 2);
}
/* ================================================================
* dns_cache_dump_to_json()
* ================================================================ */